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)
|