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

fix(proxy_server.py): return updated member list in /team/member_delete

Browse source
This commit is contained in:
Krrish Dholakia 2024-06-08 16:33:07 -07:00
parent 5c4abcbdb4
commit dddd4a73fe
5 changed files with 11 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

9
litellm/proxy/proxy_server.py
View file

@ -10295,7 +10295,7 @@ async def team_member_delete(
existing_team_row = LiteLLM_TeamTable(**_existing_team_row.model_dump())
## DELETE MEMBER FROM TEAM
new_team_members: List[dict] = []
new_team_members: List[Member] = []
for m in existing_team_row.members_with_roles:
if (
data.user_id is not None
@ -10309,13 +10309,16 @@ async def team_member_delete(
and data.user_email == m.user_email
):
continue
new_team_members.append(m.model_dump())
new_team_members.append(m)
existing_team_row.members_with_roles = new_team_members
_db_new_team_members: List[dict] = [m.model_dump() for m in new_team_members]
_ = await prisma_client.db.litellm_teamtable.update(
where={
"team_id": data.team_id,
},
data={"members_with_roles": json.dumps(new_team_members)}, # type: ignore
data={"members_with_roles": json.dumps(_db_new_team_members)}, # type: ignore
)
## DELETE TEAM ID from USER ROW, IF EXISTS ##