From 81a3af2c8b56f6f826efa528818d80cd7b03008e Mon Sep 17 00:00:00 2001 From: wxg0103 <727495428@qq.com> Date: Thu, 22 May 2025 09:11:12 +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 efdc22a8f..ccafc0e0b 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.get("prompt_tokens") else {} + _create_usage_metadata(token_usage) if token_usage and token_usage.get("prompt_tokens") else None ) if len(choices) == 0: # logprobs is implicitly None