From 9ff6fc34bb3777cc68785a376706a146719325e5 Mon Sep 17 00:00:00 2001 From: shivam Date: Sun, 8 Nov 2026 03:14:09 +0000 Subject: [PATCH] fix(auth): preserve apply-guardrail budget exemptions Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> --- litellm/proxy/auth/auth_checks.py | 2 ++ .../proxy/auth/test_auth_checks_object_access_and_lookup.py | 2 ++ 2 files changed, 4 insertions(+) diff --git a/litellm/proxy/auth/auth_checks.py b/litellm/proxy/auth/auth_checks.py index b40eb451979..8e1850602db 100644 --- a/litellm/proxy/auth/auth_checks.py +++ b/litellm/proxy/auth/auth_checks.py @@ -1111,6 +1111,8 @@ def route_skips_budget_checks(route: str) -> bool: return False if route in MODEL_DISCOVERY_ROUTES: return True + if route in LiteLLMRoutes.apply_guardrail_routes.value: + return True if RouteChecks.is_llm_api_route(route=route): return False from litellm.proxy.pass_through_endpoints.pass_through_endpoints import ( # noqa: PLC0415 # pass-through endpoints import the auth stack diff --git a/tests/unit/proxy/auth/test_auth_checks_object_access_and_lookup.py b/tests/unit/proxy/auth/test_auth_checks_object_access_and_lookup.py index 78ce9728f3b..95938b79108 100644 --- a/tests/unit/proxy/auth/test_auth_checks_object_access_and_lookup.py +++ b/tests/unit/proxy/auth/test_auth_checks_object_access_and_lookup.py @@ -11089,6 +11089,8 @@ def test_route_skips_budget_checks_marks_only_spend_free_routes() -> None: assert route_skips_budget_checks(route="/v1/models") is True assert route_skips_budget_checks(route="/spend/logs") is True assert route_skips_budget_checks(route="/utils/model_info") is True + assert route_skips_budget_checks(route="/guardrails/apply_guardrail") is True + assert route_skips_budget_checks(route="/apply_guardrail") is True assert RouteChecks.is_llm_api_route(route="/utils/model_info") is True assert route_skips_budget_checks(route="/health") is False assert route_skips_budget_checks(route="/v1/chat/completions") is False