refactor: openai token

This commit is contained in:
wxg0103 2025-05-22 09:11:12 +08:00
parent 2ec0d22b14
commit 81a3af2c8b

View File

@ -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