diff --git a/litellm/proxy/_types.py b/litellm/proxy/_types.py index fa77809368..25cf2f56d3 100644 --- a/litellm/proxy/_types.py +++ b/litellm/proxy/_types.py @@ -342,7 +342,10 @@ class LiteLLMRoutes(enum.Enum): + sso_only_routes ) - team_admin_routes: List = ["/team/member_add"] + internal_user_routes + team_admin_routes: List = [ + "/team/member_add", + "/team/member_delete", + ] + internal_user_routes # class LiteLLMAllowedRoutes(LiteLLMBase): diff --git a/litellm/tests/test_proxy_server.py b/litellm/tests/test_proxy_server.py index 7caf1ecbc1..a1d6d9dee1 100644 --- a/litellm/tests/test_proxy_server.py +++ b/litellm/tests/test_proxy_server.py @@ -931,9 +931,10 @@ async def test_create_team_member_add(prisma_client, new_member_method): @pytest.mark.parametrize("team_member_role", ["admin", "user"]) +@pytest.mark.parametrize("team_route", ["/team/member_add", "/team/member_delete"]) @pytest.mark.asyncio async def test_create_team_member_add_team_admin_user_api_key_auth( - prisma_client, team_member_role + prisma_client, team_member_role, team_route ): import time @@ -981,7 +982,7 @@ async def test_create_team_member_add_team_admin_user_api_key_auth( from starlette.datastructures import URL request = Request(scope={"type": "http"}) - request._url = URL(url="/team/member_add") + request._url = URL(url=team_route) body = {} json_bytes = json.dumps(body).encode("utf-8")