From 7563d94e65591c2ed3b909c196a3eb76e61625fd Mon Sep 17 00:00:00 2001
From: mateo-berri <277851410+mateo-berri@users.noreply.github.com>
Date: Wed, 9 Sep 2026 16:29:55 -0700
Subject: [PATCH] test(responses): assert the streamed response.completed event
 echoes the named tool_choice

---
 .../test_streaming_iterator_transformation.py              | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/tests/test_litellm/responses/litellm_completion_transformation/test_streaming_iterator_transformation.py b/tests/test_litellm/responses/litellm_completion_transformation/test_streaming_iterator_transformation.py
index 29061db97dc..70f0957890d 100644
--- a/tests/test_litellm/responses/litellm_completion_transformation/test_streaming_iterator_transformation.py
+++ b/tests/test_litellm/responses/litellm_completion_transformation/test_streaming_iterator_transformation.py
@@ -679,6 +679,9 @@ def test_streamed_named_tool_choice_is_echoed_in_responses_api_shape() -> None:
         "response.in_progress",
         "response.completed",
     ]
-    assert response_events[0].response.tool_choice == {"type": "function", "name": "run_command"}
-    assert response_events[1].response.tool_choice == {"type": "function", "name": "run_command"}
+    assert [event.response.tool_choice for event in response_events] == [
+        {"type": "function", "name": "run_command"},
+        {"type": "function", "name": "run_command"},
+        {"type": "function", "name": "run_command"},
+    ]
     assert any(getattr(event, "type", None) == "response.output_item.done" for event in events)