Update BasicAgent.py

BasicAgent.py

@@ -10,17 +10,6 @@ from smolagents import (
     GoogleSearchTool,
     )
 
-#model_id = "Qwen/Qwen2.5-Coder-32B-Instruct" 
-#model_id = 'meta-llama/Llama-3.3-70B-Instruct'
-
-#model = HfApiModel(model_id=model_id, token="HUGGINGFACEHUB_API_TOKEN") 
-
-#agent = CodeAgent(tools=[], model=model, add_base_tools=True)
-
-#max_steps=5
-#import os
-#from smolagents import CodeAgent, HfApiModel, InferenceClientModel, WebSearchTool
-
 class newAgent:
     """Adapts smolagents.CodeAgent to the HF course template API."""
     def __init__(self):
@@ -30,22 +19,21 @@ class newAgent:
             raise RuntimeError("HUGGINGFACEHUB_API_TOKEN not set in Space secrets")
 
         model = HfApiModel(model_id=model_id, token=hf_token)
-        # add_base_tools=True already gives you search, python, etc.
-        self.agent = CodeAgent(tools=[], model=model, add_base_tools=True)
-    #*
-            # include FinalAnswerTool in tools so agent knows when to stop
+        
+        # include FinalAnswerTool in tools so agent knows when to stop
         tools = [FinalAnswerTool()]
         self.agent = CodeAgent(
             tools=tools,
             model=model,
             add_base_tools=True
+            max_steps=5 # limit reasoning time
         )
-            #*
+   
     def __call__(self, question: str) -> str:
         """ONE question in → ONE pure-text answer out."""
         #↓ Replace .run with whatever method actually returns the answer string.
         result = self.agent.run(question)
-        return answer
+        return result
 
         #answer = self.run