diff --git a/litellm/litellm_core_utils/streaming_handler.py b/litellm/litellm_core_utils/streaming_handler.py index 861dccd0099..959cdd27791 100644 --- a/litellm/litellm_core_utils/streaming_handler.py +++ b/litellm/litellm_core_utils/streaming_handler.py @@ -1571,9 +1571,7 @@ class CustomStreamWrapper: if hasattr(chunk, "usage") and chunk.usage is not None: model_response.usage = ( - litellm.Usage(**chunk.usage.model_dump()) - if isinstance(chunk.usage, BaseModel) - else chunk.usage + litellm.Usage(**chunk.usage.model_dump()) if isinstance(chunk.usage, BaseModel) else chunk.usage ) ## RETURN ARG