test(streaming): build thinking chunks with typed Delta/StreamingChoices

2026-02-08 05:08:50 -06:00 · 2026-02-08 05:08:50 -06:00 · c63d5fa0b5
commit c63d5fa0b5
parent f7d03f8a43
1 changed files with 8 additions and 10 deletions
--- 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 = [