From 9ba1183e93430ec2c24384043ca27c8a3655daeb Mon Sep 17 00:00:00 2001 From: daleselaji-dev <265319989+daleselaji-dev@users.noreply.github.com> Date: Sun, 23 Aug 2026 10:58:10 +0800 Subject: [PATCH] style: format streaming usage changes --- .../litellm_core_utils/streaming_chunk_builder_utils.py | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) diff --git a/litellm/litellm_core_utils/streaming_chunk_builder_utils.py b/litellm/litellm_core_utils/streaming_chunk_builder_utils.py index f706cc18f0b..51e2f51401c 100644 --- a/litellm/litellm_core_utils/streaming_chunk_builder_utils.py +++ b/litellm/litellm_core_utils/streaming_chunk_builder_utils.py @@ -798,9 +798,7 @@ class ChunkProcessor: usage_chunk_dict = self._usage_chunk_calculation_helper(usage_chunk) if "prompt_tokens" in usage_chunk: prompt_tokens_provided = True - if "prompt_tokens" in usage_chunk and ( - usage_chunk_dict["prompt_tokens"] > 0 or prompt_tokens == 0 - ): + if "prompt_tokens" in usage_chunk and (usage_chunk_dict["prompt_tokens"] > 0 or prompt_tokens == 0): prompt_tokens = usage_chunk_dict["prompt_tokens"] if "completion_tokens" in usage_chunk: completion_tokens_provided = True @@ -965,9 +963,7 @@ class ChunkProcessor: try: returned_usage.prompt_tokens = ( - prompt_tokens - if prompt_tokens_provided - else token_counter(model=model, messages=messages) + prompt_tokens if prompt_tokens_provided else token_counter(model=model, messages=messages) ) except Exception: # don't allow this failing to block a complete streaming response from being returned print_verbose("token_counter failed, assuming prompt tokens is 0")