Update app.py

app.py

@@ -35,27 +35,25 @@ class BasicAgent:
             search_tool
         ]
         chat_with_tools = chat.bind_tools(self.tools)
-        self.graph = self._build_graph()
+        self._build_graph()
         
         print("BasicAgent initialized.")
     def __call__(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
         messages = [HumanMessage(content="Tell me about our guest named 'Lady Ada Lovelace'.")]
-        response = self.invoke({"messages": messages})
+        response = self.graph.invoke({"messages": messages})
         response = response['messages'][-1].content
         print(f"Agent returning answer: {response}")
         return response
         
     def _build_graph(self):
-        self.builder = StateGraph(AgentState)
-        self.builder.add_node("assistant", assistant)
-        self.builder.add_node("tools", ToolNode(self.tools))
-        self.builder.add_edge(START, "assistant")
-        self.builder.add_conditional_edges("assistant", tools_condition)
-        self.builder.add_edge("tools", "assistant")
-        self.builder.compile()
-        
-        
+        self.graph = StateGraph(AgentState)
+        self.graph.add_node("assistant", assistant)
+        self.graph.add_node("tools", ToolNode(self.tools))
+        self.graph.add_edge(START, "assistant")
+        self.graph.add_conditional_edges("assistant", tools_condition)
+        self.graph.add_edge("tools", "assistant")
+        self.graph.compile()
 
     
 def run_and_submit_all( profile: gr.OAuthProfile | None):