Update agents/software_engineer_agent.py

agents/software_engineer_agent.py

@@ -14,7 +14,7 @@ model = AutoModelForCausalLM.from_pretrained(
 def run(state: dict) -> dict:
     """Software Engineer generates code from architecture specification"""
     messages = state["messages"]
-    prompt = messages[-1]["content"]
+    prompt = messages[-1].content
 
     input_ids = tokenizer(prompt, return_tensors="pt").input_ids.to(model.device)
     output_ids = model.generate(input_ids, max_new_tokens=3000)