Skip to content

Commit 5daac34

Browse files
authored
fix: Token usage tracking fails in streaming mode (#3603)
1 parent b7a2fda commit 5daac34

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

camel/agents/chat_agent.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -3357,9 +3357,11 @@ def _update_token_usage_tracker(
33573357
tracker (Dict[str, int]): The token usage tracker to update.
33583358
usage_dict (Dict[str, int]): The usage dictionary with new values.
33593359
"""
3360-
tracker["prompt_tokens"] += usage_dict.get("prompt_tokens", 0)
3361-
tracker["completion_tokens"] += usage_dict.get("completion_tokens", 0)
3362-
tracker["total_tokens"] += usage_dict.get("total_tokens", 0)
3360+
tracker["prompt_tokens"] += usage_dict.get("prompt_tokens") or 0
3361+
tracker["completion_tokens"] += (
3362+
usage_dict.get("completion_tokens") or 0
3363+
)
3364+
tracker["total_tokens"] += usage_dict.get("total_tokens") or 0
33633365

33643366
def _convert_to_chatagent_response(
33653367
self,

0 commit comments

Comments
 (0)