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

(Testing) e2e testing for team budget enforcement checks (#7988)

Browse source 
* test_team_and_key_budget_enforcement

* test_team_budget_update

* test_gemini_pro_json_schema_httpx_content_policy_error
This commit is contained in:
Ishaan Jaff 2025-01-24 18:18:12 -08:00 • committed by GitHub
parent d7f862783d
commit bf46ae7346
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 175 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/proxy/auth/auth_checks.py
View file

@ -1037,7 +1037,7 @@ async def _team_max_budget_check(
raise litellm.BudgetExceededError(
current_cost=team_object.spend,
max_budget=team_object.max_budget,
message=f"Team={team_object.team_id} over budget. Spend={team_object.spend}, Budget={team_object.max_budget}",
message=f"Budget has been exceeded! Team={team_object.team_id} Current cost: {team_object.spend}, Max budget: {team_object.max_budget}",
)