remove budget reservation disable flag

This commit is contained in:
user
2026-04-30 20:57:20 -07:00
parent f30bfcf36a
commit ce17639cf7
2 changed files with 1 additions and 34 deletions
+1 -5
View File
@@ -1909,7 +1909,7 @@ async def _reserve_budget_after_common_checks(
end_user_object: Optional[LiteLLM_EndUserTable] = None,
) -> None:
user_api_key_auth_obj.budget_reservation = None
if skip_budget_checks or _is_budget_reservation_disabled():
if skip_budget_checks:
return
from litellm.proxy.spend_tracking.budget_reservation import (
@@ -1931,10 +1931,6 @@ async def _reserve_budget_after_common_checks(
)
def _is_budget_reservation_disabled() -> bool:
return get_secret_bool("LITELLM_DISABLE_BUDGET_RESERVATION", False) is True
def _should_skip_budget_checks(
request_data: dict,
route: str,
@@ -77,35 +77,6 @@ async def test_should_clear_stale_budget_reservation_when_budget_checks_skip():
assert user_api_key_auth_obj.budget_reservation is None
@pytest.mark.asyncio
async def test_should_skip_budget_reservation_when_disabled():
user_api_key_auth_obj = UserAPIKeyAuth(
token="test_token",
spend=0.0,
max_budget=1.0,
budget_reservation={
"reserved_cost": 0.5,
"entries": [{"counter_key": "spend:key:test_token"}],
},
)
with patch.dict(os.environ, {"LITELLM_DISABLE_BUDGET_RESERVATION": "true"}):
await _reserve_budget_after_common_checks(
user_api_key_auth_obj=user_api_key_auth_obj,
request_data={"model": "gpt-4"},
route="/v1/chat/completions",
llm_router=None,
team_object=None,
user_object=None,
prisma_client=None,
user_api_key_cache=DualCache(),
proxy_logging_obj=MagicMock(),
skip_budget_checks=False,
)
assert user_api_key_auth_obj.budget_reservation is None
@pytest.mark.asyncio
async def test_should_not_reuse_cached_key_object_for_request_state():
key_cache = DualCache()