From c351c691c3bd9697cdcfe36acb1027ccb1efa645 Mon Sep 17 00:00:00 2001 From: Ishaan Jaffer Date: Mon, 13 Apr 2026 14:59:34 -0700 Subject: [PATCH] style: black format streaming_chunk_builder_utils.py --- .../streaming_chunk_builder_utils.py | 22 +++++++++---------- 1 file changed, 10 insertions(+), 12 deletions(-) diff --git a/litellm/litellm_core_utils/streaming_chunk_builder_utils.py b/litellm/litellm_core_utils/streaming_chunk_builder_utils.py index 00159dfe9b7..fe7c62c3842 100644 --- a/litellm/litellm_core_utils/streaming_chunk_builder_utils.py +++ b/litellm/litellm_core_utils/streaming_chunk_builder_utils.py @@ -119,9 +119,7 @@ class ChunkProcessor: model = ChunkProcessor._get_model_from_chunks(chunks, first_chunk_model) system_fingerprint = chunk.get("system_fingerprint", None) - first_chunk_with_choices = next( - (c for c in chunks if c.get("choices")), chunk - ) + first_chunk_with_choices = next((c for c in chunks if c.get("choices")), chunk) role = first_chunk_with_choices["choices"][0]["delta"]["role"] finish_reason = "stop" for chunk in chunks: @@ -166,9 +164,9 @@ class ChunkProcessor: self, tool_call_chunks: List[Dict[str, Any]] ) -> List[ChatCompletionMessageToolCall]: tool_calls_list: List[ChatCompletionMessageToolCall] = [] - tool_call_map: Dict[ - int, Dict[str, Any] - ] = {} # Map to store tool calls by index + tool_call_map: Dict[int, Dict[str, Any]] = ( + {} + ) # Map to store tool calls by index for chunk in tool_call_chunks: choices = chunk["choices"] @@ -649,12 +647,12 @@ class ChunkProcessor: web_search_requests: Optional[int] = calculated_usage_per_chunk[ "web_search_requests" ] - completion_tokens_details: Optional[ - CompletionTokensDetails - ] = calculated_usage_per_chunk["completion_tokens_details"] - prompt_tokens_details: Optional[ - PromptTokensDetailsWrapper - ] = calculated_usage_per_chunk["prompt_tokens_details"] + completion_tokens_details: Optional[CompletionTokensDetails] = ( + calculated_usage_per_chunk["completion_tokens_details"] + ) + prompt_tokens_details: Optional[PromptTokensDetailsWrapper] = ( + calculated_usage_per_chunk["prompt_tokens_details"] + ) try: returned_usage.prompt_tokens = prompt_tokens or token_counter(