Update agent.py

agent.py

@@ -1,31 +1,31 @@
-from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, WikipediaSearchTool
+from smolagents import CodeAgent, HfApiModel, DuckDuckGoSearchTool, WikipediaSearchTool
 import datetime
 
-
-
-class DuckDuckGoAgent:
+class SmartDuckDuckGoAgent:
     def __init__(self):
-
+        self.wikipedia_tool = WikipediaSearchTool()
+        self.duckduckgo_tool = DuckDuckGoSearchTool()
+        self.model = HfApiModel()
         self.agent = CodeAgent(
-            tools=[DuckDuckGoSearchTool(), WikipediaSearchTool()],
-            model=HfApiModel(),
+            tools=[self.wikipedia_tool, self.duckduckgo_tool],
+            model=self.model,
             additional_authorized_imports=['datetime']
         )
 
     def __call__(self, question: str) -> str:
-        """Busca la respuesta usando Internet o Wikipedia, luego limita la longitud."""
-        print(f"Running search for question: {question}")
-        try:
-            response = self.agent.run(question)
-            if not response:
-                return "No information found."
+        # 1. Intentar buscar en Wikipedia
+        wiki_response = self.wikipedia_tool.run(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
 
-            max_length = 200
-            response = response.strip()
-            if len(response) > max_length:
-                response = response[:max_length].rsplit('.', 1)[0] + '.'
-            return response
-        except Exception as e:
-            print(f"Error during agent search: {e}")
-            return "Sorry, I couldn't find an answer."
+        # 3. Si ambos fallan, usar el modelo para generar una respuesta educada
+        fallback_prompt = f"""I couldn't find a direct answer to the following question:
+        {question}
+        
+        Based on your general knowledge, provide a reasonable and well-written answer."""
+        return self.model.run(fallback_prompt)