Update agent.py

agent.py

@@ -3,12 +3,14 @@ import datetime
 
 class SmartDuckDuckGoAgent:
     def __init__(self):
-        # Herramientas de búsqueda
+        # Definir los tools de búsqueda
         self.wikipedia_tool = WikipediaSearchTool()
         self.duckduckgo_tool = DuckDuckGoSearchTool()
+
         # Modelo
         self.model = HfApiModel()
-        # Agente con herramientas y modelo
+        
+        # Crear el agente con los tools y el modelo
         self.agent = CodeAgent(
             tools=[self.wikipedia_tool, self.duckduckgo_tool],
             model=self.model,
@@ -16,17 +18,23 @@ class SmartDuckDuckGoAgent:
         )
 
     def __call__(self, question: str) -> str:
-        # 1. Intentar buscar en Wikipedia
-        wiki_response = self.wikipedia_tool.search(question)
-        if wiki_response and "No results found" not in wiki_response:
-            return wiki_response
-
-        # 2. Si Wikipedia falla, intentar DuckDuckGo
-        duck_response = self.duckduckgo_tool.run(question)
-        if duck_response and "No results found" not in duck_response:
-            return duck_response
+        # Intentar buscar en Wikipedia primero
+        try:
+            wiki_response = self.wikipedia_tool.search(question)  # Usa 'search' en lugar de 'run'
+            if wiki_response:
+                return wiki_response
+        except Exception as e:
+            print(f"Error buscando en Wikipedia: {e}")
+        
+        # Si Wikipedia no tiene resultados, intentar con DuckDuckGo
+        try:
+            duck_response = self.duckduckgo_tool.run(question)
+            if duck_response:
+                return duck_response
+        except Exception as e:
+            print(f"Error buscando en DuckDuckGo: {e}")
 
-        # 3. Si ambos fallan, usar el modelo para generar una respuesta educada
+        # Si ambos fallan, usar el modelo para generar una respuesta
         fallback_prompt = f"""I couldn't find a direct answer to the following question:
         {question}