From 5e1e25d98d8e09e198ec320e0328e9bec15aff38 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Mon, 17 Mar 2025 17:55:35 -0700 Subject: [PATCH] is_internal_user_role --- litellm/proxy/_types.py | 8 ++++++++ .../proxy/management_endpoints/internal_user_endpoints.py | 2 +- 2 files changed, 9 insertions(+), 1 deletion(-) diff --git a/litellm/proxy/_types.py b/litellm/proxy/_types.py index 041ee78838a..3a66e13dae2 100644 --- a/litellm/proxy/_types.py +++ b/litellm/proxy/_types.py @@ -124,6 +124,14 @@ class LitellmUserRoles(str, enum.Enum): } return ui_labels.get(self.value, "") + @property + def is_internal_user_role(self) -> bool: + """returns true if this role is an `internal_user` or `internal_user_viewer` role""" + return self.value in [ + self.INTERNAL_USER.value, + self.INTERNAL_USER_VIEW_ONLY.value, + ] + class LitellmTableNames(str, enum.Enum): """ diff --git a/litellm/proxy/management_endpoints/internal_user_endpoints.py b/litellm/proxy/management_endpoints/internal_user_endpoints.py index a638f4c7438..250495dadd1 100644 --- a/litellm/proxy/management_endpoints/internal_user_endpoints.py +++ b/litellm/proxy/management_endpoints/internal_user_endpoints.py @@ -86,7 +86,7 @@ def _update_internal_new_user_params(data_json: dict, data: NewUserRequest) -> d ) is_internal_user = False - if data.user_role == LitellmUserRoles.INTERNAL_USER: + if data.user_role and data.user_role._is_internal_user_role: is_internal_user = True if litellm.default_internal_user_params: for key, value in litellm.default_internal_user_params.items():