diff --git a/tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_hook_extra_headers.py b/tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_hook_extra_headers.py index 649a08e8744..18b2d5cce22 100644 --- a/tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_hook_extra_headers.py +++ b/tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_hook_extra_headers.py @@ -549,7 +549,7 @@ class TestHookHeaderMergePriority: captured_extra_headers: Dict[str, Any] = {} async def fake_create_mcp_client( - server, mcp_auth_header=None, extra_headers=None, stdio_env=None + server, mcp_auth_header=None, extra_headers=None, stdio_env=None, subject_token=None ): captured_extra_headers["value"] = extra_headers mock_client = MagicMock() @@ -589,7 +589,7 @@ class TestHookHeaderMergePriority: captured_extra_headers: Dict[str, Any] = {} async def fake_create_mcp_client( - server, mcp_auth_header=None, extra_headers=None, stdio_env=None + server, mcp_auth_header=None, extra_headers=None, stdio_env=None, subject_token=None ): captured_extra_headers["value"] = extra_headers mock_client = MagicMock() @@ -635,7 +635,7 @@ class TestHookHeaderMergePriority: captured_extra_headers: Dict[str, Any] = {} async def fake_create_mcp_client( - server, mcp_auth_header=None, extra_headers=None, stdio_env=None + server, mcp_auth_header=None, extra_headers=None, stdio_env=None, subject_token=None ): captured_extra_headers["value"] = extra_headers mock_client = MagicMock() @@ -691,7 +691,7 @@ class TestHookHeaderMergePriority: captured_extra_headers: Dict[str, Any] = {} async def fake_create_mcp_client( - server, mcp_auth_header=None, extra_headers=None, stdio_env=None + server, mcp_auth_header=None, extra_headers=None, stdio_env=None, subject_token=None ): captured_extra_headers["value"] = extra_headers mock_client = MagicMock() @@ -739,7 +739,7 @@ class TestHookHeaderMergePriority: captured_extra_headers: Dict[str, Any] = {} async def fake_create_mcp_client( - server, mcp_auth_header=None, extra_headers=None, stdio_env=None + server, mcp_auth_header=None, extra_headers=None, stdio_env=None, subject_token=None ): captured_extra_headers["value"] = extra_headers mock_client = MagicMock()