File size: 365 Bytes
1004317 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
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)
|