From 5ba517819cef4bf315117fd1e9d3ba115bfefc15 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Wed, 21 Aug 2024 08:37:04 -0700 Subject: [PATCH] test(test_proxy_server.py): fix test to specify user role --- litellm/proxy/management_endpoints/team_endpoints.py | 3 ++- litellm/tests/test_proxy_server.py | 2 +- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/litellm/proxy/management_endpoints/team_endpoints.py b/litellm/proxy/management_endpoints/team_endpoints.py index 8f8ad610a80..d3c2e3e839b 100644 --- a/litellm/proxy/management_endpoints/team_endpoints.py +++ b/litellm/proxy/management_endpoints/team_endpoints.py @@ -488,7 +488,8 @@ async def team_member_add( status_code=403, detail={ "error": "Call not allowed. User not proxy admin OR team admin. route={}, team_id={}".format( - "/team/member_add", complete_team_data.team_id + "/team/member_add", + complete_team_data.team_id, ) }, ) diff --git a/litellm/tests/test_proxy_server.py b/litellm/tests/test_proxy_server.py index 79da5bf7c55..c4d9afbbdb3 100644 --- a/litellm/tests/test_proxy_server.py +++ b/litellm/tests/test_proxy_server.py @@ -909,7 +909,7 @@ async def test_create_team_member_add(prisma_client, new_member_method): await team_member_add( data=team_member_add_request, - user_api_key_dict=UserAPIKeyAuth(), + user_api_key_dict=UserAPIKeyAuth(user_role="proxy_admin"), http_request=Request( scope={"type": "http", "path": "/user/new"}, ),