Update app.py

app.py

@@ -148,19 +148,19 @@ class AgentState(TypedDict):
     messages: Annotated[list[AnyMessage], add_messages]
 
 
-def assistant(state: AgentState, chat):
-    return {
-            "messages": chat.invoke(state["messages"]),
-        }
+
 
 def build_agent(chat):    
     tools = [get_file_path_tool, get_content_tool, search_web_tool, extract_text_tool, play_chess_tool, run_code_tool]
     chat_with_tools = chat.bind_tools(tools, parallel_tool_calls=False)
     # The graph
     builder = StateGraph(AgentState)
-
+    def assistant(state: AgentState):
+        return {
+            "messages": chat.invoke(state["messages"]),
+            }
     # Define nodes: these do the work
-    builder.add_node("assistant", assistant(chat))
+    builder.add_node("assistant", assistant)
     builder.add_node("tools", ToolNode([get_file_path_tool]))
     
     # Define edges: these determine how the control flow moves