from smolagents import CodeAgent, InferenceClientModel, WikipediaSearchTool


class GAIAAgent:
    def __init__(self):
        self.agent = CodeAgent(
            tools=[WikipediaSearchTool()],
            model=InferenceClientModel(),
            add_base_tools=True
        )

    def __call__(self, question: str) -> str:
        return self.agent.run(question)