fix(tests): set premium_user=True in test_aasync_call_with_key_over_model_budget

[jquinter]

Summary

• test_aasync_call_with_key_over_model_budget (all 3 parametrized variants) was failing because generate_key_fn calls validate_model_max_budget, which raises "You must have an enterprise license to set model_max_budget" when premium_user is not True
• The enterprise gate fires before the test can exercise the actual per-model budget logic, so:
  • [gpt-4o-mini-False] / [openai/gpt-4o-mini-False]: got an enterprise error instead of "exceeded budget for model=..."
  • [gpt-4o-True]: caught the enterprise exception in the except block, triggering assert (should_pass is False) which failed because should_pass=True
• Fix: add setattr(litellm.proxy.proxy_server, "premium_user", True) alongside the other proxy-server setatrs at the top of the test, matching the pattern used in JWT tests (PR fix(tests): set premium_user=True in JWT tests that call user_api_key_auth #21641)

Not related to PR #21107 (bedrock) or PR #21664 (pricing JSON).

Test plan

• One-line fix, same pattern as merged PR fix(test): mock enterprise license check in JWT test #21285 / fix(tests): set premium_user=True in JWT tests that call user_api_key_auth #21641

🤖 Generated with Claude Code

[vercel]

The latest updates on your projects. Learn more about Vercel for GitHub.

Project	Deployment	Actions	Updated (UTC)
litellm	Ready	Preview, Comment	Feb 20, 2026 2:14pm

[greptile-apps]

Greptile Summary

Added premium_user=True flag to test_aasync_call_with_key_over_model_budget to bypass enterprise license check in validate_model_max_budget. The test was failing because the enterprise gate was triggered before the actual per-model budget logic could be exercised. This follows the same pattern established in PR #21641 for JWT tests.

Confidence Score: 5/5

• This PR is safe to merge with minimal risk
• Single-line test fix that follows established pattern from PR fix(tests): set premium_user=True in JWT tests that call user_api_key_auth #21641, directly addresses the root cause of test failures, and has no impact on production code
• No files require special attention

Important Files Changed

Filename	Overview
tests/proxy_unit_tests/test_key_generate_prisma.py	Added premium_user=True flag to fix enterprise license check in test

_{Last reviewed commit: 1f63036}

[greptile-apps]

_{1 file reviewed, no comments}

_{Edit Code Review Agent Settings | Greptile}