| 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) | |