apunkt/rowboat
1
0
Fork 0
mirror of https://github.com/rowboatlabs/rowboat.git synced 2026-04-27 17:36:25 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

fixed token_used error for other providers

Browse source
This commit is contained in:
arkml 2025-05-20 17:12:06 +05:30
parent 148299d6f8
commit 37cac79da7
1 changed files with 11 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -153,8 +153,8 @@ async def run_turn_streamed(
# Initialize agents and get external tools
new_agents = get_agents(
agent_configs=agent_configs,
tool_configs=tool_configs,
agent_configs=agent_configs,
tool_configs=tool_configs,
complete_request=complete_request
)
new_agents = add_child_transfer_related_instructions_to_agents(new_agents)
@ -196,12 +196,15 @@ async def run_turn_streamed(
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)
try:
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)
except Exception as e:
print(f"Warning: Tokens used is likely not available for your chosen model: {e}")
web_search_messages = handle_web_search_event(event, current_agent)
for message in web_search_messages: