Update agent.py

agent.py

@@ -2,13 +2,7 @@ import os
 import logging
 from typing import List, Dict, Any, Optional
 
-from smolagents.agent import Agent
-from smolagents.llms import OpenAILLM
-from smolagents.tools import Tool
-from smolagents.tools.web_search import DuckDuckGoSearchTool
-from smolagents.tools.wikipedia import WikipediaSearchTool
-from smolagents.tools.arxiv import ArxivSearchTool
-from smolagents.tools.code_interpreter import CodeInterpreterTool
+from smolagents import CodeAgent, tool, InferenceClientModel
 
 # Configurazione del logging
 logging.basicConfig(
@@ -33,19 +27,11 @@ class GaiaAgent:
             raise ValueError("OPENAI_API_KEY non trovata. Configura il secret in Hugging Face.")
         
         # Configurazione del modello LLM
-        self.llm = OpenAILLM(
+        self.model = InferenceClientModel(
             model="gpt-4.1",
             api_key=api_key
         )
         
-        # Configurazione degli strumenti
-        self.tools: List[Tool] = [
-            WikipediaSearchTool(),
-            DuckDuckGoSearchTool(),
-            ArxivSearchTool(),
-            CodeInterpreterTool()
-        ]
-        
         # Istruzioni specifiche per il formato delle risposte
         system_prompt = """You are a helpful assistant tasked with answering questions using a set of tools.
 Your final answer should be a number OR as few words as possible OR a comma separated list of numbers and/or strings.
@@ -54,13 +40,13 @@ If you are asked for a string, don't use articles, neither abbreviations, and wr
 Return ONLY the final answer line."""
         
         # Creazione dell'agente
-        self.agent = Agent(
-            llm=self.llm,
-            tools=self.tools,
+        self.agent = CodeAgent(
+            model=self.model,
             system_prompt=system_prompt,
             verbose=True  # Per logging essenziale
         )
         
+        # Configurazione degli strumenti avviene automaticamente in CodeAgent
         logger.info("GaiaAgent inizializzato con successo")
     
     def __call__(self, question: str) -> str: