scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-17 23:51:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 66

is_internal_user_role

Browse source
This commit is contained in:
Ishaan Jaff 2025-03-17 17:55:35 -07:00
parent 8ff8ad82c9
commit 5e1e25d98d
2 changed files with 9 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
litellm/proxy/_types.py
View file

@ -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):
"""

2
litellm/proxy/management_endpoints/internal_user_endpoints.py
View file

@ -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():