Update app.py

app.py

@@ -61,7 +61,7 @@ from langchain_core.messages import HumanMessage, SystemMessage
 #     """Multiply a and b."""
 #     return a * b
 
-def assistant(state: AgentState):
+def assistant(state: AgentState, llm_with_tools):
     # System message
     textual_description_of_tool = """
 DuckDuckGoSearchResults():
@@ -114,10 +114,12 @@ class BasicAgent:
         # self.visionLLM = ChatOpenAI(model="gpt-4o",api_key=api_key) # multi-modal LLM
         self.LLM = ChatOpenAI(model="gpt-4o",api_key=api_key) # manager LLM
         self.LLM_with_tools = self.LLM.bind_tools(tools, parallel_tool_calls=False)
+
+        assistant_with_llm = lambda state:assistant(state, self.LLM_with_tools)
         # Graph
         self.Builder = StateGraph(AgentState)
         # Define nodes: these do the work
-        self.Builder.add_node("assistant", assistant)
+        self.Builder.add_node("assistant", assistant_with_llm)
         self.Builder.add_node("tools",ToolNode(tools))
         # Define edges: these determine how the control flow moves
         self.Builder.add_edge(START,"assistant")