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(proxy_server.py): fix end user object check when master key used

Browse source 
check if end user max budget exceeded for master key
This commit is contained in:
Krrish Dholakia 2024-05-29 17:20:59 -07:00
parent e9b5bf2d2f
commit cfcf5969c8
4 changed files with 30 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

1
litellm/exceptions.py
View file

@ -314,6 +314,7 @@ class BudgetExceededError(Exception):
self.current_cost = current_cost
self.max_budget = max_budget
message = f"Budget has been exceeded! Current cost: {current_cost}, Max budget: {max_budget}"
self.message = message
super().__init__(message)