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

fix - raise Exception when trying to update/delete a non-existtent team

Browse source
This commit is contained in:
Ishaan Jaff 2024-05-20 17:36:08 -07:00
parent dd81ab4a61
commit 367d86047c
1 changed files with 16 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

16
litellm/proxy/proxy_server.py
View file

@ -7444,6 +7444,11 @@ async def update_team(
existing_team_row = await prisma_client.get_data( existing_team_row = await prisma_client.get_data(
team_id=data.team_id, table_name="team", query_type="find_unique" team_id=data.team_id, table_name="team", query_type="find_unique"
) )
if existing_team_row is None:
raise HTTPException(
status_code=404,
detail={"error": f"Team not found, passed team_id={data.team_id}"},
)
updated_kv = data.json(exclude_none=True) updated_kv = data.json(exclude_none=True)
team_row = await prisma_client.update_data( team_row = await prisma_client.update_data(
@ -7697,6 +7702,17 @@ async def delete_team(
if data.team_ids is None: if data.team_ids is None:
raise HTTPException(status_code=400, detail={"error": "No team id passed in"}) raise HTTPException(status_code=400, detail={"error": "No team id passed in"})
# check that all teams passed exist
for team_id in data.team_ids:
team_row = await prisma_client.get_data( # type: ignore
team_id=team_id, table_name="team", query_type="find_unique"
)
if team_row is None:
raise HTTPException(
status_code=404,
detail={"error": f"Team not found, passed team_id={team_id}"},
)
## DELETE ASSOCIATED KEYS ## DELETE ASSOCIATED KEYS
await prisma_client.delete_data(team_id_list=data.team_ids, table_name="key") await prisma_client.delete_data(team_id_list=data.team_ids, table_name="key")
## DELETE TEAMS ## DELETE TEAMS