Update agents/project_manager_agent.py

agents/project_manager_agent.py

@@ -19,10 +19,9 @@ def run(state: dict) -> dict:
     input_ids = tokenizer(prompt, return_tensors="pt").input_ids.to(model.device)
     output_ids = model.generate(input_ids, max_new_tokens=3000)
     output = tokenizer.decode(output_ids[0], skip_special_tokens=True)
-    ai_message = {"role": "assistant", "content": output}
 
     return {
-        "messages": [ai_message],
+        "messages": [AIMessage(content=output)],
         "chat_log": state["chat_log"] + [{"role": "Project Manager", "content": output}],
         "proj_output": output,
     }