Update app.py

app.py

@@ -17,26 +17,26 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 # --- Basic Agent Definition ---
 
-# Generate the AgentState and Agent graph
-class AgentState(TypedDict):
-    messages: Annotated[list[AnyMessage], add_messages]
+llm = HuggingFaceEndpoint(repo_id=llm_checkpoint)
+chat = ChatHuggingFace(llm=llm, verbose=True)
+tools = [
+    search_tool
+]
+chat_with_tools = chat.bind_tools(self.tools)
 
 def assistant(state: AgentState):
     return {
         "messages": [chat_with_tools.invoke(state["messages"])],
-    }
+    }   
+
+# Generate the AgentState and Agent graph
+class AgentState(TypedDict):
+    messages: Annotated[list[AnyMessage], add_messages]
     
 class BasicAgent:
     
     def __init__(self, llm_checkpoint):
-        llm = HuggingFaceEndpoint(repo_id=llm_checkpoint)
-        chat = ChatHuggingFace(llm=llm, verbose=True)
-        self.tools = [
-            search_tool
-        ]
-        chat_with_tools = chat.bind_tools(self.tools)
         self.graph = self._build_graph()
-        
         print("BasicAgent initialized.")
         
     def __call__(self, question: str) -> str:
@@ -50,7 +50,7 @@ class BasicAgent:
     def _build_graph(self):
         graph = StateGraph(AgentState)
         graph.add_node("assistant", assistant)
-        graph.add_node("tools", ToolNode(self.tools))
+        graph.add_node("tools", ToolNode(tools))
         graph.add_edge(START, "assistant")
         graph.add_conditional_edges("assistant", tools_condition)
         graph.add_edge("tools", "assistant")