phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix(proxy_server.py): support checking openai user param

Browse source
This commit is contained in:
Krrish Dholakia 2024-03-13 12:00:23 -07:00
parent 7924700df6
commit b3493269b3
2 changed files with 41 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

7
enterprise/enterprise_hooks/blocked_user_list.py
View file

@ -66,12 +66,13 @@ class _ENTERPRISE_BlockedUserList(CustomLogger):
- check if user id part of blocked list
"""
self.print_verbose(f"Inside Blocked User List Pre-Call Hook")
if "user_id" in data:
if data["user_id"] in self.blocked_user_list:
if "user_id" in data or "user" in data:
user = data.get("user_id", data.get("user", ""))
if user in self.blocked_user_list:
raise HTTPException(
status_code=400,
detail={
"error": f"User blocked from making LLM API Calls. User={data['user_id']}"
"error": f"User blocked from making LLM API Calls. User={user}"
},
)
except HTTPException as e: