Spaces:
Runtime error
Runtime error
Update agents/product_manager_agent.py
Browse files
agents/product_manager_agent.py
CHANGED
@@ -14,7 +14,7 @@ model = AutoModelForCausalLM.from_pretrained(
|
|
14 |
def run(state: dict) -> dict:
|
15 |
"""Generates structured product requirements from user input prompt."""
|
16 |
messages = state["messages"]
|
17 |
-
prompt = messages[-1]
|
18 |
|
19 |
input_ids = tokenizer(prompt, return_tensors="pt").input_ids.to(model.device)
|
20 |
output_ids = model.generate(input_ids, max_new_tokens=3000)
|
|
|
14 |
def run(state: dict) -> dict:
|
15 |
"""Generates structured product requirements from user input prompt."""
|
16 |
messages = state["messages"]
|
17 |
+
prompt = messages[-1].content
|
18 |
|
19 |
input_ids = tokenizer(prompt, return_tensors="pt").input_ids.to(model.device)
|
20 |
output_ids = model.generate(input_ids, max_new_tokens=3000)
|