From 1f6303677c03f30c128a851cd26e7ec25e44e5dc Mon Sep 17 00:00:00 2001 From: Julio Quinteros Pro Date: Fri, 20 Feb 2026 11:12:38 -0300 Subject: [PATCH] fix(tests): set premium_user=True in test_aasync_call_with_key_over_model_budget generate_key_fn calls validate_model_max_budget which raises "You must have an enterprise license to set model_max_budget" unless premium_user=True. The test was hitting this gate before it could exercise the actual per-model budget logic, causing all three variants to fail with an enterprise-check error instead of the expected budget exceeded / pass-through result. Co-Authored-By: Claude Sonnet 4.6 --- tests/proxy_unit_tests/test_key_generate_prisma.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/proxy_unit_tests/test_key_generate_prisma.py b/tests/proxy_unit_tests/test_key_generate_prisma.py index d0559a007a7..9d001952cb2 100644 --- a/tests/proxy_unit_tests/test_key_generate_prisma.py +++ b/tests/proxy_unit_tests/test_key_generate_prisma.py @@ -1779,6 +1779,7 @@ async def test_aasync_call_with_key_over_model_budget( # 12. Make a call with a key over budget, expect to fail setattr(litellm.proxy.proxy_server, "prisma_client", prisma_client) setattr(litellm.proxy.proxy_server, "master_key", "sk-1234") + setattr(litellm.proxy.proxy_server, "premium_user", True) await litellm.proxy.proxy_server.prisma_client.connect() verbose_proxy_logger.setLevel(logging.DEBUG)