diff --git a/tests/test_litellm/litellm_core_utils/test_streaming_chunk_builder_utils.py b/tests/test_litellm/litellm_core_utils/test_streaming_chunk_builder_utils.py index 34c9efd838..da6d802792 100644 --- a/tests/test_litellm/litellm_core_utils/test_streaming_chunk_builder_utils.py +++ b/tests/test_litellm/litellm_core_utils/test_streaming_chunk_builder_utils.py @@ -168,16 +168,14 @@ def test_get_combined_thinking_content_preserves_interleaved_blocks(): def make_chunk(**delta_kwargs): return ModelResponseStream( - **{ - **base_chunk, - "choices": [ - { - "index": 0, - "delta": delta_kwargs, - "finish_reason": None, - } - ], - } + **base_chunk, + choices=[ + StreamingChoices( + index=0, + delta=Delta(**delta_kwargs), + finish_reason=None, + ) + ], ) chunks = [