From 27a77dc657f3dccfdcea6dc0b806ac366ee465e6 Mon Sep 17 00:00:00 2001 From: wxg0103 <727495428@qq.com> Date: Wed, 21 May 2025 18:33:56 +0800 Subject: [PATCH] refactor: openai token --- apps/setting/models_provider/impl/base_chat_open_ai.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/apps/setting/models_provider/impl/base_chat_open_ai.py b/apps/setting/models_provider/impl/base_chat_open_ai.py index c96bfacf2..7b5f3355c 100644 --- a/apps/setting/models_provider/impl/base_chat_open_ai.py +++ b/apps/setting/models_provider/impl/base_chat_open_ai.py @@ -133,7 +133,7 @@ class BaseChatOpenAI(ChatOpenAI): ) usage_metadata: Optional[UsageMetadata] = ( - _create_usage_metadata(token_usage) if token_usage else None + _create_usage_metadata(token_usage) if token_usage.get("prompt_tokens") else None ) if len(choices) == 0: # logprobs is implicitly None