phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 18:54:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

add auth on all scim endpoints

Browse source
This commit is contained in:
Ishaan Jaff 2025-04-16 17:12:08 -07:00
parent a208598b04
commit 119ea80f60
1 changed files with 13 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

23
litellm/proxy/management_endpoints/scim/scim_v2.py
View file

@ -30,6 +30,7 @@ from litellm.proxy._types import (
NewUserResponse, NewUserResponse,
UserAPIKeyAuth, UserAPIKeyAuth,
) )
from litellm.proxy.auth.user_api_key_auth import user_api_key_auth
from litellm.proxy.management_endpoints.internal_user_endpoints import new_user from litellm.proxy.management_endpoints.internal_user_endpoints import new_user
from litellm.proxy.management_endpoints.team_endpoints import new_team from litellm.proxy.management_endpoints.team_endpoints import new_team
from litellm.types.proxy.management_endpoints.scim_v2 import * from litellm.types.proxy.management_endpoints.scim_v2 import *
@ -198,7 +199,7 @@ class ScimTransformations:
"/Users", "/Users",
response_model=SCIMListResponse, response_model=SCIMListResponse,
status_code=200, status_code=200,
dependencies=[Depends(set_scim_content_type)], dependencies=[Depends(user_api_key_auth), Depends(set_scim_content_type)],
) )
async def get_users( async def get_users(
startIndex: int = Query(1, ge=1), startIndex: int = Query(1, ge=1),
@ -265,7 +266,7 @@ async def get_users(
"/Users/{user_id}", "/Users/{user_id}",
response_model=SCIMUser, response_model=SCIMUser,
status_code=200, status_code=200,
dependencies=[Depends(set_scim_content_type)], dependencies=[Depends(user_api_key_auth), Depends(set_scim_content_type)],
) )
async def get_user( async def get_user(
user_id: str = Path(..., title="User ID"), user_id: str = Path(..., title="User ID"),
@ -304,7 +305,7 @@ async def get_user(
"/Users", "/Users",
response_model=SCIMUser, response_model=SCIMUser,
status_code=201, status_code=201,
dependencies=[Depends(set_scim_content_type)], dependencies=[Depends(user_api_key_auth), Depends(set_scim_content_type)],
) )
async def create_user( async def create_user(
user: SCIMUser = Body(...), user: SCIMUser = Body(...),
@ -371,7 +372,7 @@ async def create_user(
"/Users/{user_id}", "/Users/{user_id}",
response_model=SCIMUser, response_model=SCIMUser,
status_code=200, status_code=200,
dependencies=[Depends(set_scim_content_type)], dependencies=[Depends(user_api_key_auth), Depends(set_scim_content_type)],
) )
async def update_user( async def update_user(
user_id: str = Path(..., title="User ID"), user_id: str = Path(..., title="User ID"),
@ -397,6 +398,7 @@ async def update_user(
@scim_router.delete( @scim_router.delete(
"/Users/{user_id}", "/Users/{user_id}",
status_code=204, status_code=204,
dependencies=[Depends(user_api_key_auth)],
) )
async def delete_user( async def delete_user(
user_id: str = Path(..., title="User ID"), user_id: str = Path(..., title="User ID"),
@ -456,7 +458,7 @@ async def delete_user(
"/Users/{user_id}", "/Users/{user_id}",
response_model=SCIMUser, response_model=SCIMUser,
status_code=200, status_code=200,
dependencies=[Depends(set_scim_content_type)], dependencies=[Depends(user_api_key_auth), Depends(set_scim_content_type)],
) )
async def patch_user( async def patch_user(
user_id: str = Path(..., title="User ID"), user_id: str = Path(..., title="User ID"),
@ -498,7 +500,7 @@ async def patch_user(
"/Groups", "/Groups",
response_model=SCIMListResponse, response_model=SCIMListResponse,
status_code=200, status_code=200,
dependencies=[Depends(set_scim_content_type)], dependencies=[Depends(user_api_key_auth), Depends(set_scim_content_type)],
) )
async def get_groups( async def get_groups(
startIndex: int = Query(1, ge=1), startIndex: int = Query(1, ge=1),
@ -584,7 +586,7 @@ async def get_groups(
"/Groups/{group_id}", "/Groups/{group_id}",
response_model=SCIMGroup, response_model=SCIMGroup,
status_code=200, status_code=200,
dependencies=[Depends(set_scim_content_type)], dependencies=[Depends(user_api_key_auth), Depends(set_scim_content_type)],
) )
async def get_group( async def get_group(
group_id: str = Path(..., title="Group ID"), group_id: str = Path(..., title="Group ID"),
@ -625,7 +627,7 @@ async def get_group(
"/Groups", "/Groups",
response_model=SCIMGroup, response_model=SCIMGroup,
status_code=201, status_code=201,
dependencies=[Depends(set_scim_content_type)], dependencies=[Depends(user_api_key_auth), Depends(set_scim_content_type)],
) )
async def create_group( async def create_group(
group: SCIMGroup = Body(...), group: SCIMGroup = Body(...),
@ -691,7 +693,7 @@ async def create_group(
"/Groups/{group_id}", "/Groups/{group_id}",
response_model=SCIMGroup, response_model=SCIMGroup,
status_code=200, status_code=200,
dependencies=[Depends(set_scim_content_type)], dependencies=[Depends(user_api_key_auth), Depends(set_scim_content_type)],
) )
async def update_group( async def update_group(
group_id: str = Path(..., title="Group ID"), group_id: str = Path(..., title="Group ID"),
@ -810,6 +812,7 @@ async def update_group(
@scim_router.delete( @scim_router.delete(
"/Groups/{group_id}", "/Groups/{group_id}",
status_code=204, status_code=204,
dependencies=[Depends(user_api_key_auth)],
) )
async def delete_group( async def delete_group(
group_id: str = Path(..., title="Group ID"), group_id: str = Path(..., title="Group ID"),
@ -864,7 +867,7 @@ async def delete_group(
"/Groups/{group_id}", "/Groups/{group_id}",
response_model=SCIMGroup, response_model=SCIMGroup,
status_code=200, status_code=200,
dependencies=[Depends(set_scim_content_type)], dependencies=[Depends(user_api_key_auth), Depends(set_scim_content_type)],
) )
async def patch_group( async def patch_group(
group_id: str = Path(..., title="Group ID"), group_id: str = Path(..., title="Group ID"),