From c9958eb329ff0f80cd8d5b9abdb91a22b216fcbf Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Sat, 3 Feb 2024 10:40:59 -0800 Subject: [PATCH] (feat) raise error when user_role not in known roles --- litellm/proxy/proxy_server.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py index 97a740072e..919499bdfb 100644 --- a/litellm/proxy/proxy_server.py +++ b/litellm/proxy/proxy_server.py @@ -3005,6 +3005,16 @@ async def new_user(data: NewUserRequest): - max_budget: (float|None) Max budget for given user. """ 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) return NewUserResponse( key=response["token"],