From c8e49dec35eddd41975c8f7fb2e056f8e415cd9f Mon Sep 17 00:00:00 2001 From: shijingjian Date: Tue, 5 Aug 2025 13:55:36 +0800 Subject: [PATCH] fix: token count statistical problem fixing --- owl/utils/enhanced_role_playing.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/owl/utils/enhanced_role_playing.py b/owl/utils/enhanced_role_playing.py index f6aaa553c..0fbb61f31 100644 --- a/owl/utils/enhanced_role_playing.py +++ b/owl/utils/enhanced_role_playing.py @@ -520,7 +520,7 @@ async def arun_society( assistant_response, user_response = await society.astep(input_msg) # Check if usage info is available before accessing it if assistant_response.info.get("usage") and user_response.info.get("usage"): - overall_prompt_token_count += assistant_response.info["usage"].get( + overall_completion_token_count += assistant_response.info["usage"].get( "completion_tokens", 0 ) overall_prompt_token_count += assistant_response.info["usage"].get(