File tree Expand file tree Collapse file tree 1 file changed +5
-3
lines changed
Expand file tree Collapse file tree 1 file changed +5
-3
lines changed Original file line number Diff line number Diff 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 ,
You can’t perform that action at this time.
0 commit comments