diff --git a/tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_server_manager.py b/tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_server_manager.py index 7a936461823..9edfe9b0291 100644 --- a/tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_server_manager.py +++ b/tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_server_manager.py @@ -10914,7 +10914,10 @@ class TestKeyAttachedGuardrailsReachTheMCPPreCallHook: seen.update(data) return None - with patch.object(proxy_logging_obj, "pre_call_hook", side_effect=capture_pre_call_hook): + with ( + patch("litellm.proxy.utils._premium_user_check"), + patch.object(proxy_logging_obj, "pre_call_hook", side_effect=capture_pre_call_hook), + ): await MCPServerManager().pre_call_tool_check( name="echo", arguments={"text": "hi"},