Skip to content

Commit 27a77dc

Browse files
committed
refactor: openai token
1 parent 187e9c1 commit 27a77dc

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

apps/setting/models_provider/impl/base_chat_open_ai.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -133,7 +133,7 @@ def _convert_chunk_to_generation_chunk(
133133
)
134134

135135
usage_metadata: Optional[UsageMetadata] = (
136-
_create_usage_metadata(token_usage) if token_usage else None
136+
_create_usage_metadata(token_usage) if token_usage.get("prompt_tokens") else None
137137
)
138138
if len(choices) == 0:
139139
# logprobs is implicitly None

0 commit comments

Comments
 (0)