From a261722efb36316fa967e36666caca86ba1314fc Mon Sep 17 00:00:00 2001 From: ryan-crabbe-berri Date: Mon, 3 Aug 2026 15:01:47 -0700 Subject: [PATCH] test(team): stub premium_user so budget validation tests pass without a license --- .../proxy/management_endpoints/test_team_endpoints.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/test_litellm/proxy/management_endpoints/test_team_endpoints.py b/tests/test_litellm/proxy/management_endpoints/test_team_endpoints.py index 10f2e932d95..1cdc8a89261 100644 --- a/tests/test_litellm/proxy/management_endpoints/test_team_endpoints.py +++ b/tests/test_litellm/proxy/management_endpoints/test_team_endpoints.py @@ -10755,7 +10755,8 @@ class TestValidateTeamModelMaxBudget: _validate_team_model_max_budget, ) - _validate_team_model_max_budget(model_max_budget) + with patch("litellm.proxy.proxy_server.premium_user", True): + _validate_team_model_max_budget(model_max_budget) def test_normalized_duplicate_entries_rejected(self): from litellm.proxy._types import ProxyException