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

Update agents/project_manager_agent.py

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