from smolagents import (
    CodeAgent,
    DuckDuckGoSearchTool,
    VisitWebpageTool,
    InferenceClientModel,
)


class BasicAgent:
    def __init__(self):
        token = os.getenv("HF_TOKEN")
        if not token:
            raise ValueError("Add HF_TOKEN in your Space's Settings → Secrets.")

        model = InferenceClientModel(token=token)

        self.agent = CodeAgent(
            tools=[
                DuckDuckGoSearchTool(),
                VisitWebpageTool(),
            ],
            model=model,
            max_steps=12,
        )

    def __call__(self, question: str) -> str:
        prompt = (
            "Answer the following question accurately. "
            "Use web search and visit source pages when needed. "
            "Follow the requested answer format exactly. "
            "Return only the answer, without explanations, citations, "
            "or a 'FINAL ANSWER' prefix.\n\n"
            + question
        )
        return str(self.agent.run(prompt, reset=True)).strip()