apunkt/rowboat
1
0
Fork 0
mirror of https://github.com/rowboatlabs/rowboat.git synced 2026-05-28 19:05:31 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add back token usage counting in agents code

Browse source
This commit is contained in:
akhisud3195 2025-05-08 10:36:19 +05:30
parent 93869fddd8
commit 9a72540536
1 changed files with 9 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

9
apps/rowboat_agents/src/graph/core.py
View file

@ -194,6 +194,15 @@ async def run_turn_streamed(
try:
# Handle web search events
if event.type == "raw_response_event":
# Handle token usage counting
if hasattr(event.data, 'type') and event.data.type == "response.completed" and hasattr(event.data.response, 'usage'):
tokens_used["total"] += event.data.response.usage.total_tokens
tokens_used["prompt"] += event.data.response.usage.input_tokens
tokens_used["completion"] += event.data.response.usage.output_tokens
print('-'*50)
print(f"Found usage information. Updated cumulative tokens: {tokens_used}")
print('-'*50)
web_search_messages = handle_web_search_event(event, current_agent)
for message in web_search_messages:
message['response_type'] = ResponseType.INTERNAL.value