Update agent.py

agent.py

@@ -3,9 +3,12 @@ import datetime
 
 class SmartDuckDuckGoAgent:
     def __init__(self):
+        # Herramientas de búsqueda
         self.wikipedia_tool = WikipediaSearchTool()
         self.duckduckgo_tool = DuckDuckGoSearchTool()
+        # Modelo
         self.model = HfApiModel()
+        # Agente con herramientas y modelo
         self.agent = CodeAgent(
             tools=[self.wikipedia_tool, self.duckduckgo_tool],
             model=self.model,
@@ -14,7 +17,7 @@ class SmartDuckDuckGoAgent:
 
     def __call__(self, question: str) -> str:
         # 1. Intentar buscar en Wikipedia
-        wiki_response = self.wikipedia_tool.run(question)
+        wiki_response = self.wikipedia_tool.search(question)
         if wiki_response and "No results found" not in wiki_response:
             return wiki_response