scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-04 02:31:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 69

Fix audio token usage formatting

Browse source
This commit is contained in:
Liam McDonald 2026-05-12 13:30:03 -07:00
parent 8a3c56786a
commit a220078c51
1 changed files with 8 additions and 17 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

25
litellm/responses/utils.py
View file

@ -958,23 +958,14 @@ class ResponseAPILoggingUtils:
response_api_usage, "output_tokens_details", None
)
if output_tokens_details:
if isinstance(output_tokens_details, dict):
completion_tokens_details = CompletionTokensDetailsWrapper(
**output_tokens_details
)
else:
completion_tokens_details = CompletionTokensDetailsWrapper(
reasoning_tokens=getattr(
output_tokens_details, "reasoning_tokens", None
),
image_tokens=getattr(
output_tokens_details, "image_tokens", None
),
text_tokens=getattr(output_tokens_details, "text_tokens", None),
audio_tokens=getattr(
output_tokens_details, "audio_tokens", None
),
)
completion_tokens_details = CompletionTokensDetailsWrapper(
reasoning_tokens=getattr(
output_tokens_details, "reasoning_tokens", None
),
image_tokens=getattr(output_tokens_details, "image_tokens", None),
text_tokens=getattr(output_tokens_details, "text_tokens", None),
audio_tokens=getattr(output_tokens_details, "audio_tokens", None),
)
chat_usage = Usage(
prompt_tokens=prompt_tokens,