Spaces:
Running
Running
File size: 1,164 Bytes
c02d868 b015e1c 667e88a c02d868 61e90fc c02d868 427cae4 c02d868 61e90fc b015e1c 61e90fc b015e1c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 |
import requests
from duckduckgo_search import DDGS
# --- Clase de Agente ---
class BasicAgent:
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]}...")
# 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."
print(f"Agent returning answer: {fixed_answer}")
return fixed_answer
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)}"
|