From a220078c51b25d019ad2774ce521a057ffb7841d Mon Sep 17 00:00:00 2001 From: Liam McDonald Date: Tue, 12 May 2026 13:30:03 -0700 Subject: [PATCH] Fix audio token usage formatting --- litellm/responses/utils.py | 25 ++++++++----------------- 1 file changed, 8 insertions(+), 17 deletions(-) diff --git a/litellm/responses/utils.py b/litellm/responses/utils.py index a35aaac6b3c..5ee91c70483 100644 --- a/litellm/responses/utils.py +++ b/litellm/responses/utils.py @@ -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,