fix(proxy): skip key max-budget check on discovery routes

This commit is contained in:
Jeevan Babu Gotru 2026-10-29 09:49:07 -05:00 • committed by GitHub
parent 9d2a7f1160
commit 364c7cd378
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -3051,8 +3051,8 @@ async def validate_resolved_virtual_key( # noqa: C901 # Preserve ordering of e
user_obj=user_obj,
)
# Check 5. Token Spend is under budget
if RouteChecks.is_llm_api_route(route=route):
# Check 5. Token Spend is under budget (discovery routes are OpenAI routes, so they need the route_skips_budget_checks exemption too, #48809)
if RouteChecks.is_llm_api_route(route=route) and not route_skips_budget_checks(route=route):
await virtual_key_max_budget_check(
valid_token=valid_token,
proxy_logging_obj=proxy_logging_obj,