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

fix: fix tests

Browse source
This commit is contained in:
Krrish Dholakia 2024-08-07 15:02:04 -07:00
parent d832327ccf
commit ff373663a3
4 changed files with 24 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

9
litellm/exceptions.py
View file

@ -730,10 +730,15 @@ LITELLM_EXCEPTION_TYPES = [
class BudgetExceededError(Exception):
def __init__(self, current_cost, max_budget):
def __init__(
self, current_cost: float, max_budget: float, message: Optional[str] = None
):
self.current_cost = current_cost
self.max_budget = max_budget
message = f"Budget has been exceeded! Current cost: {current_cost}, Max budget: {max_budget}"
message = (
message
or f"Budget has been exceeded! Current cost: {current_cost}, Max budget: {max_budget}"
)
self.message = message
super().__init__(message)