mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-21 00:21:49 +00:00
refactor(proxy): drop comments from team member temp budget helper
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
parent
7f3f8fae2d
commit
25e7253fda
2 changed files with 0 additions and 4 deletions
|
|
@ -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:
|
||||
|
|
|
|||
|
|
@ -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(
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue