phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix(proxy_server.py): allow updating tpm/rpm limits via /team/update

Browse source
This commit is contained in:
Krrish Dholakia 2024-03-28 15:59:35 -07:00
parent 569e660a67
commit 58bdbecbc6
3 changed files with 10 additions and 18 deletions
Download patch file Download diff file Expand all files Collapse all files

7
litellm/proxy/_types.py
View file

@ -427,13 +427,8 @@ class TeamMemberDeleteRequest(LiteLLMBase):
return values return values
class UpdateTeamRequest(LiteLLMBase): class UpdateTeamRequest(TeamBase):
team_id: str # required team_id: str # required
team_alias: Optional[str] = None
admins: Optional[list] = None
members: Optional[list] = None
members_with_roles: Optional[List[Member]] = None
metadata: Optional[dict] = None
class DeleteTeamRequest(LiteLLMBase): class DeleteTeamRequest(LiteLLMBase):

1
litellm/proxy/proxy_server.py
View file

@ -5826,7 +5826,6 @@ async def update_team(
user_api_key_dict: UserAPIKeyAuth = Depends(user_api_key_auth), user_api_key_dict: UserAPIKeyAuth = Depends(user_api_key_auth),
): ):
""" """
[BETA]
[RECOMMENDED] - use `/team/member_add` to add new team members instead [RECOMMENDED] - use `/team/member_add` to add new team members instead
You can now update team budget / rate limits via /team/update You can now update team budget / rate limits via /team/update

20
tests/test_team.py
View file

@ -157,18 +157,10 @@ async def new_team(session, i, user_id=None, member_list=None, model_aliases=Non
return await response.json() return await response.json()
async def update_team( async def update_team(session, i, team_id, user_id=None, member_list=None, **kwargs):
session,
i,
team_id,
user_id=None,
member_list=None,
):
url = "http://0.0.0.0:4000/team/update" url = "http://0.0.0.0:4000/team/update"
headers = {"Authorization": "Bearer sk-1234", "Content-Type": "application/json"} headers = {"Authorization": "Bearer sk-1234", "Content-Type": "application/json"}
data = { data = {"team_id": team_id, **kwargs}
"team_id": team_id,
}
if user_id is not None: if user_id is not None:
data["members_with_roles"] = [{"role": "user", "user_id": user_id}] data["members_with_roles"] = [{"role": "user", "user_id": user_id}]
elif member_list is not None: elif member_list is not None:
@ -284,9 +276,15 @@ async def test_team_update():
{"role": "user", "user_id": new_normal_user}, {"role": "user", "user_id": new_normal_user},
] ]
team_data = await update_team( team_data = await update_team(
session=session, i=0, member_list=member_list, team_id=team_data["team_id"] session=session,
i=0,
member_list=member_list,
team_id=team_data["team_id"],
tpm_limit=100,
) )
assert team_data["data"]["tpm_limit"] == 100
@pytest.mark.asyncio @pytest.mark.asyncio
async def test_team_delete(): async def test_team_delete():