Update app.py

app.py

@@ -148,20 +148,19 @@ class AgentState(TypedDict):
     messages: Annotated[list[AnyMessage], add_messages]
 
 
-
-
-def build_agent(chat):
-    def assistant(state: AgentState):
-        return {
+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)
 
     # Define nodes: these do the work
-    builder.add_node("assistant", assistant)
+    builder.add_node("assistant", assistant(chat))
     builder.add_node("tools", ToolNode([get_file_path_tool]))
     
     # Define edges: these determine how the control flow moves