Update agent.py

agent.py

@@ -7,7 +7,7 @@ Rules: ALL workflow knowledge in prompt. Code is just wiring.
 
 import os
 from langchain_mistralai import ChatMistralAI
-from langgraph.prebuilt import create_react_agent
+from langgraph.prebuilt import create_react_agent, ToolNode
 from langgraph.checkpoint.memory import MemorySaver
 from tools import ALL_TOOLS
 
@@ -300,9 +300,10 @@ def create_agent():
         temperature=0.1,
     )
     memory = MemorySaver()
+    tool_node = ToolNode(ALL_TOOLS, handle_tool_errors=True)
     agent = create_react_agent(
         llm,
-        ALL_TOOLS,
+        tool_node,
         prompt=SYSTEM_PROMPT,
         checkpointer=memory,
     )
@@ -328,4 +329,4 @@ def invoke_agent(message: str, thread_id: str = "default") -> str:
     result = agent.invoke({"messages": [("user", message)]}, config=config)
     messages = result.get("messages", [])
     last = messages[-1] if messages else None
-    return last.content if last and hasattr(last, "content") else str(last)
+    return last.content if last and hasattr(last, "content") else str(last)