Update agent.py

agent.py

@@ -140,11 +140,7 @@ def build_graph():
         # Clean the answer
         clean_text = clean_answer(llm_response.content)
         
-        # Format the response properly
-        task_id = str(state.get("task_id", "1"))
-        formatted_response = [{"task_id": task_id, "submitted_answer": clean_text}]
-        
-        return {"messages": [AIMessage(content=json.dumps(formatted_response, ensure_ascii=False))]}
+        return {"messages": [AIMessage(content=json.dumps(clean_text, ensure_ascii=False))]}
 
     # --- Graph Definition ---
     builder = StateGraph(MessagesState)