phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-27 03:34:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat(_types.py): allow team admin to delete member from team

Browse source
This commit is contained in:
Krrish Dholakia 2024-08-20 16:25:13 -07:00
parent 64affd0d6b
commit c305eb3321
2 changed files with 7 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

5
litellm/proxy/_types.py
View file

@ -342,7 +342,10 @@ class LiteLLMRoutes(enum.Enum):
+ sso_only_routes + 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): # class LiteLLMAllowedRoutes(LiteLLMBase):

5
litellm/tests/test_proxy_server.py
View file

@ -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_member_role", ["admin", "user"])
@pytest.mark.parametrize("team_route", ["/team/member_add", "/team/member_delete"])
@pytest.mark.asyncio @pytest.mark.asyncio
async def test_create_team_member_add_team_admin_user_api_key_auth( 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 import time
@ -981,7 +982,7 @@ async def test_create_team_member_add_team_admin_user_api_key_auth(
from starlette.datastructures import URL from starlette.datastructures import URL
request = Request(scope={"type": "http"}) request = Request(scope={"type": "http"})
request._url = URL(url="/team/member_add") request._url = URL(url=team_route)
body = {} body = {}
json_bytes = json.dumps(body).encode("utf-8") json_bytes = json.dumps(body).encode("utf-8")