fix: show budget reset date in BudgetExceededError message

This commit is contained in:
Patrick Decat 2025-04-15 16:57:31 +02:00
parent aff0d1a18c
commit 731263b32e
No known key found for this signature in database
GPG key ID: FD55B9BD5687D8FF
10 changed files with 33 additions and 8 deletions

View file

@ -832,6 +832,7 @@ async def _user_api_key_auth_builder( # noqa: PLR0915
raise litellm.BudgetExceededError(
current_cost=valid_token.team_member_spend,
max_budget=team_member_budget,
budget_reset_at=valid_token.budget_reset_at,
)
# Check 3. If token is expired