Update agent.py

agent.py

@@ -2,13 +2,14 @@ from smolagents import CodeAgent, HfApiModel
 from smolagents import VisitWebpageTool, WebSearchTool, WikipediaSearchTool, PythonInterpreterTool,FinalAnswerTool
 
 
-
-model = HfApiModel(
-    max_token = 2096,
-    temperature = 0.5, 
-    model_id = 'https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud',
-    custom_role_conversations=None
-)
+class GaiaAgent:
+    def _init_(self):
+        self.model = HfApiModel(
+            max_token = 2096,
+            temperature = 0.5, 
+            model_id = 'https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud',
+            custom_role_conversations=None
+        )
 #or 
 #model=InferenceClientModel(model_id="Qwen/Qwen2.5-Coder-32B-Instruct", provider="together")
 
@@ -16,23 +17,28 @@ model = HfApiModel(
  #   prompt_templates = yaml.safe_load(stream)
 
 #Creating the agent
-agent = CodeAgent(
-    model=model,
-    tools=[
-        VisitWebpageTool(),
-        WebSearchTool(),
-        WikipediaSearchTool(),
-        PythonInterpreterTool(),
-        FinalAnswerTool(),
-    ], 
-    max_steps=10,
-    verbosity_level = 1,        #or other parameter
-    grammar = None,             
-    planning_interval=None,      
-    #name="web_agent",            
-    #description="Brows the web to find information",      
-    #additional_autorized_imports = ["pandas"],
-)
+        self.agent = CodeAgent(
+            model=self.model,
+            tools=[
+                VisitWebpageTool(),
+                WebSearchTool(),
+                WikipediaSearchTool(),
+                PythonInterpreterTool(),
+                FinalAnswerTool(),
+            ], 
+            max_steps=10,
+            verbosity_level = 1,        #or other parameter
+            grammar = None,             
+            planning_interval=None,      
+            #name="web_agent",            
+            #description="Brows the web to find information",      
+            #additional_autorized_imports = ["pandas"],
+        )
+    def _call_(self, question: str) -> str:
+        try:
+            return self.agent.run(question)
+        except Exception as e:
+            return f"ERROR: {e}"
 
 #manager_agent = CodeAgent(
  #   model=InferenceClientModel("deepseek-ai/DeepSeek-R1", provider="together", max_tokens=8096),