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

fix(proxy_server.py): use consistent naming schema - move to /team/block

Browse source
This commit is contained in:
Krrish Dholakia 2024-03-26 16:59:36 -07:00
parent ce44645000
commit 94c5d9f82f
5 changed files with 42 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -30,15 +30,15 @@ def common_checks(
"""
Common checks across jwt + key-based auth.
1. If team is disabled
1. If team is blocked
2. If team can call model
3. If team is in budget
4. If end_user ('user' passed to /chat/completions, /embeddings endpoint) is in budget
"""
_model = request_body.get("model", None)
if team_object.disabled == True:
if team_object.blocked == True:
raise Exception(
f"Team={team_object.team_id} is disabled. Update via `/team/update`."
f"Team={team_object.team_id} is blocked. Update via `/team/unblock` if your admin."
)
# 2. If user can call model
if (