migrate agentic system

This commit is contained in:
Xi Yan 2024-09-11 13:57:39 -07:00
parent 4b34f741d0
commit 6049aada71
5 changed files with 39 additions and 7 deletions

View file

@ -73,9 +73,7 @@ class AgenticSystemClient(AgenticSystem):
async with client.stream(
"POST",
f"{self.base_url}/agentic_system/turn/create",
json={
"request": encodable_dict(request),
},
json=encodable_dict(request),
headers={"Content-Type": "application/json"},
timeout=20,
) as response:
@ -134,7 +132,7 @@ async def run_main(host: str, port: int):
api = AgenticSystemClient(f"http://{host}:{port}")
tool_definitions = [
SearchToolDefinition(engine=SearchEngineType.bing),
SearchToolDefinition(engine=SearchEngineType.brave),
WolframAlphaToolDefinition(),
CodeInterpreterToolDefinition(),
]