Update agent.py

agent.py

@@ -1,17 +1,30 @@
-from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel
+from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, WikipediaSearchTool
+import datetime
+
+
 
 class DuckDuckGoAgent:
     def __init__(self):
+
         self.agent = CodeAgent(
-            tools=[DuckDuckGoSearchTool()],
-            model=HfApiModel()
+            tools=[DuckDuckGoSearchTool(), WikipediaSearchTool()],
+            model=HfApiModel(),
+            additional_authorized_imports=['datetime']
         )
 
     def __call__(self, question: str) -> str:
-        """Recibe una pregunta, busca en DuckDuckGo y devuelve una respuesta."""
+        """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."
+
+
+            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}")