fixed agent architecture

agent.py

@@ -147,6 +147,8 @@ class BasicAgent:
         self.hf_model_provider = hf_model_provider
         self.tools_list = tools_list
         print("BasicAgent initialized.")
+        # Create agent with all the tools
+        self.agent_graph = self.build_agent_graph()
 
     def build_llm_with_tools(self):
         print("Building Hugging Face model and tools...")
@@ -156,7 +158,7 @@ class BasicAgent:
             provider=self.hf_model_provider,
             max_new_tokens=8192,
             do_sample=False,
-            temperature=0.4,
+            temperature=0.2,
         )
 
         chat_model = ChatHuggingFace(llm=llm)
@@ -166,7 +168,7 @@ class BasicAgent:
         llm_with_tools = chat_model.bind_tools(
             self.tools_list
         )
-        print("Agent built successfully.")
+        print("LLM with tools built successfully.")
         return llm_with_tools
     
     def build_agent_graph(self):    
@@ -187,14 +189,14 @@ class BasicAgent:
             tools_condition,
         )
         builder.add_edge("tools", "assistant")
-        return builder.compile()
+        agent_graph = builder.compile()
+        print("Agent graph built successfully.")
+        return agent_graph
 
     async def __call__(self, question: str) -> str:
         print(f"Agent received question (first 100 chars): {question[:100]}...")
         # fixed_answer = "This is a default answer."
         # print(f"Agent returning fixed answer: {fixed_answer}")
-        # Create agent with all the tools
-        agent_graph = self.build_agent_graph()
         # Example query agent might receive
         # fixed_answer = await agent.run(question)
         messages = [
@@ -204,7 +206,7 @@ class BasicAgent:
                 content=question # + '/nothink'
             )
         ]
-        response = await agent_graph.ainvoke(
+        response = await self.agent_graph.ainvoke(
             {"messages": messages}, 
             config={
                 "recursion_limit": 8,

app.py

@@ -9,7 +9,6 @@ import re
 
 from tools import fetch_website, get_wiki_full, youtube_transcript, python_repl_tool, duckduckgo_search_results
 
-
 # (Keep Constants as is)
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"