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

(feat) raise error when user_role not in known roles

Browse source
This commit is contained in:
ishaan-jaff 2024-02-03 10:40:59 -08:00
parent 29885bbdc1
commit c9958eb329
1 changed files with 10 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

10
litellm/proxy/proxy_server.py
View file

@ -3005,6 +3005,16 @@ async def new_user(data: NewUserRequest):
- max_budget: (float|None) Max budget for given user. - max_budget: (float|None) Max budget for given user.
""" """
data_json = data.json() # type: ignore data_json = data.json() # type: ignore
if "user_role" in data_json:
user_role = data_json["user_role"]
if user_role is not None:
if user_role not in ["proxy_admin", "app_owner", "app_user"]:
raise ProxyException(
message=f"Invalid user role, passed in {user_role}. Must be one of 'proxy_admin', 'app_owner', 'app_user'",
type="invalid_user_role",
param="user_role",
code=status.HTTP_400_BAD_REQUEST,
)
response = await generate_key_helper_fn(**data_json) response = await generate_key_helper_fn(**data_json)
return NewUserResponse( return NewUserResponse(
key=response["token"], key=response["token"],