Update app.py

app.py

@@ -42,7 +42,13 @@ def assistant(state: MessagesState):
 class BasicAgent:
     
     def __init__(self):
-        self.graph = self._build_graph()
+        graph = StateGraph(MessagesState)
+        graph.add_node("assistant", assistant)
+        graph.add_node("tools", ToolNode(tools))
+        graph.add_edge(START, "assistant")
+        graph.add_conditional_edges("assistant", tools_condition)
+        graph.add_edge("tools", "assistant")
+        self.graph = graph.compile()
         print("BasicAgent initialized.")
         
     def __call__(self, question: str) -> str:
@@ -55,13 +61,7 @@ class BasicAgent:
         return response
         
     def _build_graph(self):
-        graph = StateGraph(MessagesState)
-        graph.add_node("assistant", assistant)
-        graph.add_node("tools", ToolNode(tools))
-        graph.add_edge(START, "assistant")
-        graph.add_conditional_edges("assistant", tools_condition)
-        graph.add_edge("tools", "assistant")
-        return graph.compile()
+
 
     
 def run_and_submit_all( profile: gr.OAuthProfile | None):