phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 03:04:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: show budget reset date in BudgetExceededError message

Browse source
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
Download patch file Download diff file Expand all files Collapse all files

1
tests/litellm/proxy/db/test_exception_handler.py
View file

@ -143,6 +143,7 @@ def test_handle_db_exception_with_non_db_error():
regular_error = litellm.BudgetExceededError(
current_cost=10,
max_budget=10,
budget_reset_at="2025-01-01T00:00:00Z",
)
with pytest.raises(litellm.BudgetExceededError):
PrismaDBExceptionHandler.handle_db_exception(regular_error)