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

Merge pull request #3750 from BerriAI/litellm_raise_exception_when_deleting_updating_team

Browse source 
[Fix] - raise Exception when trying to update/delete a non-existent team
This commit is contained in:
Ishaan Jaff 2024-05-20 18:42:27 -07:00 • committed by GitHub
commit 829ca84f07
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
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

@ -7522,6 +7522,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(
@ -7775,6 +7780,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