Update agent.py

agent.py

@@ -10,7 +10,7 @@ from typing import TypedDict
 tools = [add, substract, multiply, divide, web_search]
 
 class AgentState(TypedDict):
-    question: str
+    messages: str
 
 def build_agent():
     llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0)
@@ -18,7 +18,7 @@ def build_agent():
 
     def assistant(state: AgentState):
         return {
-            "answer": [chat_with_tools.invoke(state["question"])],
+            "messages": [chat_with_tools.invoke(state["messages"])],
         }
 
     ## The graph