Update agent.py

agent.py

@@ -1,33 +1,32 @@
 import requests
-from duckduckgo_search import DDGS
+import urllib.parse
 
-# --- Clase de Agente ---
-class BasicAgent:
+class DuckDuckGoAgent:
     def __init__(self):
-        print("BasicAgent initialized.")
-
-    def __call__(self, question: str) -> str:
-        """
-        Este agente ahora realizará una búsqueda en DuckDuckGo
-        si detecta que la consulta parece una pregunta sobre búsqueda web.
-        """
-        print(f"Agent received question (first 50 chars): {question[:50]}...")
+        print("DuckDuckGoAgent initialized.")
         
-        # Si la pregunta contiene "buscar", realizamos una búsqueda en DuckDuckGo
-        if "buscar" in question.lower():
-            fixed_answer = self.web_search(question)
-        else:
-            fixed_answer = "This is a default answer."
+    def get_duckduckgo_answer(self, query: str) -> str:
+        # Crear una consulta de búsqueda para DuckDuckGo
+        search_query = urllib.parse.quote(query)
+        url = f"https://api.duckduckgo.com/?q={search_query}&format=json&no_html=1&skip_disambig=1"
         
-        print(f"Agent returning answer: {fixed_answer}")
-        return fixed_answer
+        # Realizar la solicitud a DuckDuckGo
+        response = requests.get(url)
+        if response.status_code == 200:
+            data = response.json()
+            # Tomamos el primer resultado de la respuesta (si existe)
+            if 'AbstractText' in data and data['AbstractText']:
+                answer = data['AbstractText']
+                # Limitamos la respuesta a 200 caracteres
+                return answer[:200]
+            else:
+                return "No se encontró una respuesta adecuada en DuckDuckGo."
+        else:
+            return "Error al contactar con DuckDuckGo."
 
-    def web_search(self, query: str) -> str:
-        """
-        Realiza una búsqueda en DuckDuckGo y devuelve los primeros resultados.
-        """
-        try:
-            results = DDGS().text(query, max_results=3)
-            return "\n".join([r["body"] for r in results])
-        except Exception as e:
-            return f"Error en la búsqueda: {str(e)}"
+    def __call__(self, question: str) -> str:
+        print(f"Agent received question (first 50 chars): {question[:50]}...")
+        # Obtener respuesta de DuckDuckGo
+        answer = self.get_duckduckgo_answer(question)
+        print(f"Agent returning answer: {answer}")
+        return answer