From bc910253073c2cc800755ab466367ba17d45d113 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Sat, 13 Jul 2024 12:29:15 -0700 Subject: [PATCH] use wrapper on /user endpoints --- .../internal_user_endpoints.py | 16 ++++++++++++++-- 1 file changed, 14 insertions(+), 2 deletions(-) diff --git a/litellm/proxy/management_endpoints/internal_user_endpoints.py b/litellm/proxy/management_endpoints/internal_user_endpoints.py index 6bf2102d51c..09146596896 100644 --- a/litellm/proxy/management_endpoints/internal_user_endpoints.py +++ b/litellm/proxy/management_endpoints/internal_user_endpoints.py @@ -29,6 +29,10 @@ from litellm.proxy.auth.user_api_key_auth import user_api_key_auth from litellm.proxy.management_endpoints.key_management_endpoints import ( generate_key_helper_fn, ) +from litellm.proxy.management_helpers.utils import ( + add_new_member, + management_endpoint_wrapper, +) router = APIRouter() @@ -39,7 +43,10 @@ router = APIRouter() dependencies=[Depends(user_api_key_auth)], response_model=NewUserResponse, ) -async def new_user(data: NewUserRequest): +@management_endpoint_wrapper +async def new_user( + data: NewUserRequest, +): """ Use this to create a new INTERNAL user with a budget. Internal Users can access LiteLLM Admin UI to make keys, request access to models. @@ -254,6 +261,7 @@ async def ui_get_available_role( tags=["Internal User management"], dependencies=[Depends(user_api_key_auth)], ) +@management_endpoint_wrapper async def user_info( user_id: Optional[str] = fastapi.Query( default=None, description="User ID in the request parameters" @@ -441,7 +449,10 @@ async def user_info( tags=["Internal User management"], dependencies=[Depends(user_api_key_auth)], ) -async def user_update(data: UpdateUserRequest): +@management_endpoint_wrapper +async def user_update( + data: UpdateUserRequest, +): """ Example curl @@ -683,6 +694,7 @@ async def get_users( tags=["Internal User management"], dependencies=[Depends(user_api_key_auth)], ) +@management_endpoint_wrapper async def delete_user( data: DeleteUserRequest, user_api_key_dict: UserAPIKeyAuth = Depends(user_api_key_auth),