Added token counting to TeamOne (#264)

* Added token counting to TeamOne
This commit is contained in:
afourney
2024-07-26 09:45:41 -07:00
committed by GitHub
parent 25b9a91328
commit 61dec0a3ca
3 changed files with 70 additions and 37 deletions

View File

@@ -35,9 +35,13 @@ async def main() -> None:
run_context = runtime.start()
actual_surfer = runtime._get_agent(web_surfer.id) # type: ignore
assert isinstance(actual_surfer, MultimodalWebSurfer)
await actual_surfer.init(model_client=client, downloads_folder=os.getcwd(), browser_channel="chromium")
actual_surfer = await runtime.try_get_underlying_agent_instance(web_surfer.id, type=MultimodalWebSurfer)
await actual_surfer.init(
model_client=client,
downloads_folder=os.getcwd(),
start_page="https://www.adamfourney.com",
browser_channel="chromium",
)
await runtime.send_message(RequestReplyMessage(), user_proxy.id)
await run_context.stop_when_idle()