agent.py

from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel

class DuckDuckGoAgent:
    def __init__(self):
        self.agent = CodeAgent(
            tools=[DuckDuckGoSearchTool()],
            model=HfApiModel()
        )

    def __call__(self, question: str) -> str:
        """Recibe una pregunta, busca en DuckDuckGo y devuelve una respuesta."""
        print(f"Running search for question: {question}")
        try:
            response = self.agent.run(question)
            return response
        except Exception as e:
            print(f"Error during agent search: {e}")
            return "Sorry, I couldn't find an answer."