From 93c7e1d457113fa0f5c22bd2843b44f137e3b7f3 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Sat, 3 Feb 2024 10:15:03 -0800 Subject: [PATCH] (feat) add user_role to _types --- litellm/proxy/_types.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/litellm/proxy/_types.py b/litellm/proxy/_types.py index f7d1e8ad0c4..e8e01602097 100644 --- a/litellm/proxy/_types.py +++ b/litellm/proxy/_types.py @@ -194,6 +194,7 @@ class DeleteKeyRequest(LiteLLMBase): class NewUserRequest(GenerateKeyRequest): max_budget: Optional[float] = None user_email: Optional[str] = None + user_role: Optional[str] = None class NewUserResponse(GenerateKeyResponse): @@ -206,6 +207,7 @@ class UpdateUserRequest(GenerateRequestBase): user_id: str spend: Optional[float] = None metadata: Optional[dict] = None + user_role: Optional[str] = None class KeyManagementSystem(enum.Enum):