Rahul-8799 commited on
Commit
4aaf03d
·
verified ·
1 Parent(s): 77d2314

Update agents/product_manager_agent.py

Browse files
Files changed (1) hide show
  1. agents/product_manager_agent.py +1 -1
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]["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)
 
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)