Update agent.py

agent.py

@@ -1,47 +1,110 @@
-"""
-GAIA Level-1 agent powered by smolagents ≥ 0.6
-"""
-from __future__ import annotations
-import os, textwrap
-from smolagents import CodeAgent, DuckDuckGoSearchTool
-from tools import WikipediaTool, ArxivTool
-import openai
+import os
+import logging
+from typing import List, Dict, Any, Optional
 
-# ─── API key ───────────────────────────────────────
-openai.api_key = os.getenv("OPENAI_API_KEY") or ""
-if not openai.api_key:
-    raise EnvironmentError("OPENAI_API_KEY non trovata – aggiungila nei Secrets dello Space.")
+from smalagents.agent import Agent
+from smalagents.llms import OpenAILLM
+from smalagents.tools import Tool
+from smalagents.tools.web_search import DuckDuckGoSearchTool
+from smalagents.tools.wikipedia import WikipediaSearchTool
+from smalagents.tools.arxiv import ArxivSearchTool
+from smalagents.tools.code_interpreter import CodeInterpreterTool
 
-# ─── System prompt (exact-match) ───────────────────
-SYSTEM_PROMPT = textwrap.dedent("""
-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.
-If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise.
-If you are asked for a string, don't use articles, neither abbreviations, and write digits in plain text unless specified otherwise.
-Return ONLY the final answer line.
-""").strip()
-
-# ─── Build CodeAgent ───────────────────────────────
-MODEL_ID = os.getenv("GAIA_MODEL_ID", "gpt-4o")   # override se hai “gpt-4.1”
-
-core_agent = CodeAgent(
-    model=MODEL_ID,
-    tools=[
-        DuckDuckGoSearchTool(),
-        WikipediaTool(),
-        ArxivTool(),
-    ],
-    max_steps=6,
-    temperature=0,
-    system_prompt=SYSTEM_PROMPT,
-    stream_outputs=False,
+# Configurazione del logging
+logging.basicConfig(
+    level=logging.INFO,
+    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s'
 )
+logger = logging.getLogger("GaiaAgent")
 
-# ─── Thin wrapper per app.py ───────────────────────
-class BasicAgent:
+class GaiaAgent:
+    """
+    Agente AI per superare il test GAIA livello 1.
+    Utilizza smalagents con capacità di ricerca Wikipedia, DuckDuckGo, arXiv e code execution.
+    """
+    
     def __init__(self):
-        print("✅ smolagents BasicAgent avviato")
-
+        logger.info("Inizializzazione GaiaAgent")
+        
+        # Verifica della chiave API
+        api_key = os.environ.get("OPENAI_API_KEY")
+        if not api_key:
+            logger.error("OPENAI_API_KEY non trovata nelle variabili d'ambiente")
+            raise ValueError("OPENAI_API_KEY non trovata. Configura il secret in Hugging Face.")
+        
+        # Configurazione del modello LLM
+        self.llm = OpenAILLM(
+            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.
+If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise.
+If you are asked for a string, don't use articles, neither abbreviations, and write digits in plain text unless specified otherwise.
+Return ONLY the final answer line."""
+        
+        # Creazione dell'agente
+        self.agent = Agent(
+            llm=self.llm,
+            tools=self.tools,
+            system_prompt=system_prompt,
+            verbose=True  # Per logging essenziale
+        )
+        
+        logger.info("GaiaAgent inizializzato con successo")
+    
     def __call__(self, question: str) -> str:
-        raw = core_agent.run(question)
-        return raw.strip().split("\n", 1)[0]
+        """
+        Processa una domanda e restituisce la risposta.
+        
+        Args:
+            question: La domanda da processare
+            
+        Returns:
+            La risposta alla domanda
+        """
+        logger.info(f"Elaborazione domanda: {question[:50]}...")
+        
+        try:
+            # Esecuzione dell'agente sulla domanda
+            response = self.agent.run(question)
+            
+            # Estrazione della risposta finale
+            answer = self._extract_final_answer(response)
+            
+            logger.info(f"Risposta generata: {answer[:50]}...")
+            return answer
+            
+        except Exception as e:
+            logger.error(f"Errore durante l'elaborazione della domanda: {str(e)}")
+            return f"Errore: {str(e)}"
+    
+    def _extract_final_answer(self, response: Dict[str, Any]) -> str:
+        """
+        Estrae la risposta finale dal risultato dell'agente.
+        
+        Args:
+            response: Il risultato completo dell'agente
+            
+        Returns:
+            La risposta finale estratta
+        """
+        # Estrazione della risposta finale dal risultato dell'agente
+        if isinstance(response, dict) and "output" in response:
+            return response["output"].strip()
+        elif isinstance(response, str):
+            return response.strip()
+        else:
+            logger.warning(f"Formato di risposta inatteso: {type(response)}")
+            # Tentativo di conversione a stringa
+            return str(response).strip()