diff --git a/litellm/proxy/auth/auth_checks.py b/litellm/proxy/auth/auth_checks.py index fcb35fc41a6..a33cda43758 100644 --- a/litellm/proxy/auth/auth_checks.py +++ b/litellm/proxy/auth/auth_checks.py @@ -5297,9 +5297,6 @@ async def _virtual_key_max_budget_alert_check( def _effective_team_member_budget(budget: LiteLLM_BudgetTable, now: datetime) -> float | None: - """Per-member cap including an unexpired temp_budget_increase. Naive - temp_budget_expiry values are treated as UTC (same convention as - _get_temp_budget_increase for keys).""" if budget.max_budget is None: return None if budget.temp_budget_increase is None or budget.temp_budget_expiry is None: diff --git a/tests/test_litellm/proxy/auth/test_auth_checks.py b/tests/test_litellm/proxy/auth/test_auth_checks.py index fbaa8c371f2..a773a75eb1e 100644 --- a/tests/test_litellm/proxy/auth/test_auth_checks.py +++ b/tests/test_litellm/proxy/auth/test_auth_checks.py @@ -8531,7 +8531,6 @@ async def test_team_member_budget_check_temp_budget_increase_extends_cap(): return 150.0 return fallback_spend - # $150 spend is over the $100 cap but under the $200 temp-extended cap. with ( patch("litellm.proxy.proxy_server.get_current_spend", mock_get_current_spend), patch(