From c63d5fa0b5bc165dd6ae11a50b22358262078dbd Mon Sep 17 00:00:00 2001
From: Emerson Gomes <emerson.gomes@thalesgroup.com>
Date: Sun, 8 Feb 2026 05:08:50 -0600
Subject: [PATCH] test(streaming): build thinking chunks with typed
 Delta/StreamingChoices

---
 .../test_streaming_chunk_builder_utils.py      | 18 ++++++++----------
 1 file changed, 8 insertions(+), 10 deletions(-)

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 = [