Update agent.py

agent.py

@@ -12,15 +12,15 @@ tools = [add, substract, multiply, divide, web_search]
 class AgentState(TypedDict):
     question: str
 
-def assistant(state: AgentState):
-    return {
-        "answer": [chat_with_tools.invoke(state["question"])],
-    }
-
 def build_agent():
     llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0)
     chat_with_tools = llm.bind_tools(tools)
 
+    def assistant(state: AgentState):
+        return {
+            "answer": [chat_with_tools.invoke(state["question"])],
+        }
+
     ## The graph
     builder = StateGraph(AgentState)