diff --git a/tests/test_litellm/responses/litellm_completion_transformation/test_reasoning_content_transformation.py b/tests/test_litellm/responses/litellm_completion_transformation/test_reasoning_content_transformation.py index d1926bbc93..020b5de0a2 100644 --- a/tests/test_litellm/responses/litellm_completion_transformation/test_reasoning_content_transformation.py +++ b/tests/test_litellm/responses/litellm_completion_transformation/test_reasoning_content_transformation.py @@ -281,6 +281,7 @@ def test_streaming_chunk_id_raw(): ) iterator = LiteLLMCompletionStreamingIterator( + model="test-model", litellm_custom_stream_wrapper=AsyncMock(), request_input="Test input", responses_api_request={},