mirror of
https://github.com/1Panel-dev/MaxKB.git
synced 2025-12-25 17:22:55 +00:00
refactor: openai token
This commit is contained in:
parent
187e9c1e4e
commit
27a77dc657
|
|
@ -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
|
||||
|
|
|
|||
Loading…
Reference in New Issue