Update agent.py

agent.py

@@ -81,8 +81,7 @@ def build_graph():
             answer_text = answer_text.split(":", 1)[1].strip()
         # Get task_id from state or set a placeholder
         task_id = state.get("task_id", "1")  # Replace with actual logic if needed
-        formatted = f'Answers (answers): [{{"task_id": "{task_id}", "submitted_answer": "{answer_text}"}}]'
-        formatted = extract_flat_answer(formatted)
+        formatted = [{"task_id": task_id, "submitted_answer": answer_text}]
         return {"messages": [formatted]}
 
     # --- Graph Definition ---