Spaces:
Runtime error
Runtime error
Update agents/project_manager_agent.py
Browse files
agents/project_manager_agent.py
CHANGED
@@ -15,7 +15,7 @@ model = AutoModelForCausalLM.from_pretrained(
|
|
15 |
def run(state: dict) -> dict:
|
16 |
"""Creates project plan based on product requirements."""
|
17 |
messages = state["messages"]
|
18 |
-
prompt = messages[-1]
|
19 |
|
20 |
input_ids = tokenizer(prompt, return_tensors="pt").input_ids.to(model.device)
|
21 |
output_ids = model.generate(input_ids, max_new_tokens=3000)
|
|
|
15 |
def run(state: dict) -> dict:
|
16 |
"""Creates project plan based on product requirements."""
|
17 |
messages = state["messages"]
|
18 |
+
prompt = messages[-1].content
|
19 |
|
20 |
input_ids = tokenizer(prompt, return_tensors="pt").input_ids.to(model.device)
|
21 |
output_ids = model.generate(input_ids, max_new_tokens=3000)
|