Update agent.py

agent.py

@@ -32,6 +32,97 @@ class GaiaAgent:
             api_key=api_key
         )
         
+        # Definizione dei tools necessari
+        @tool("search_wikipedia", "Cerca informazioni su Wikipedia")
+        def search_wikipedia(query: str) -> str:
+            """
+            Cerca informazioni su Wikipedia.
+            
+            Args:
+                query: La query di ricerca
+                
+            Returns:
+                I risultati della ricerca
+            """
+            import wikipedia
+            try:
+                return wikipedia.summary(query, sentences=5)
+            except Exception as e:
+                return f"Errore nella ricerca su Wikipedia: {str(e)}"
+        
+        @tool("search_web", "Cerca informazioni sul web usando DuckDuckGo")
+        def search_web(query: str) -> str:
+            """
+            Cerca informazioni sul web usando DuckDuckGo.
+            
+            Args:
+                query: La query di ricerca
+                
+            Returns:
+                I risultati della ricerca
+            """
+            from duckduckgo_search import DDGS
+            try:
+                with DDGS() as ddgs:
+                    results = list(ddgs.text(query, max_results=5))
+                return str(results)
+            except Exception as e:
+                return f"Errore nella ricerca web: {str(e)}"
+        
+        @tool("search_arxiv", "Cerca articoli scientifici su arXiv")
+        def search_arxiv(query: str) -> str:
+            """
+            Cerca articoli scientifici su arXiv.
+            
+            Args:
+                query: La query di ricerca
+                
+            Returns:
+                I risultati della ricerca
+            """
+            import arxiv
+            try:
+                search = arxiv.Search(query=query, max_results=5)
+                results = []
+                for result in search.results():
+                    results.append({
+                        "title": result.title,
+                        "authors": ", ".join(author.name for author in result.authors),
+                        "summary": result.summary[:200] + "..." if len(result.summary) > 200 else result.summary
+                    })
+                return str(results)
+            except Exception as e:
+                return f"Errore nella ricerca su arXiv: {str(e)}"
+        
+        @tool("execute_code", "Esegui codice Python")
+        def execute_code(code: str) -> str:
+            """
+            Esegue codice Python.
+            
+            Args:
+                code: Il codice da eseguire
+                
+            Returns:
+                Il risultato dell'esecuzione
+            """
+            import sys
+            from io import StringIO
+            import traceback
+            
+            old_stdout = sys.stdout
+            redirected_output = sys.stdout = StringIO()
+            
+            try:
+                exec(code)
+                sys.stdout = old_stdout
+                return redirected_output.getvalue()
+            except Exception as e:
+                sys.stdout = old_stdout
+                return f"Errore nell'esecuzione del codice: {str(e)}\n{traceback.format_exc()}"
+        
+        # Lista dei tools
+        tools = [search_wikipedia, search_web, search_arxiv, execute_code]
+        
         # 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.
@@ -42,11 +133,11 @@ Return ONLY the final answer line."""
         # Creazione dell'agente
         self.agent = CodeAgent(
             model=self.model,
+            tools=tools,
             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: