From cfda9ec3511125b5989f6647a1cfcc6f7784d839 Mon Sep 17 00:00:00 2001 From: Ishaan Jaffer Date: Tue, 3 Feb 2026 18:43:02 -0800 Subject: [PATCH] test_async_realtime_url_contains_model --- .../llms/openai/realtime/test_openai_realtime_handler.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/tests/test_litellm/llms/openai/realtime/test_openai_realtime_handler.py b/tests/test_litellm/llms/openai/realtime/test_openai_realtime_handler.py index 87923a8093b..a8382311b87 100644 --- a/tests/test_litellm/llms/openai/realtime/test_openai_realtime_handler.py +++ b/tests/test_litellm/llms/openai/realtime/test_openai_realtime_handler.py @@ -195,10 +195,10 @@ async def test_async_realtime_url_contains_model(): # Verify proper headers were set called_kwargs = mock_ws_connect.call_args[1] - assert "additional_headers" in called_kwargs - additional_headers = called_kwargs["additional_headers"] - assert additional_headers["Authorization"] == f"Bearer {api_key}" - assert additional_headers["OpenAI-Beta"] == "realtime=v1" + assert "extra_headers" in called_kwargs + extra_headers = called_kwargs["extra_headers"] + assert extra_headers["Authorization"] == f"Bearer {api_key}" + assert extra_headers["OpenAI-Beta"] == "realtime=v1" assert called_kwargs["ssl"] is shared_context mock_realtime_streaming.assert_called_once()