phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 11:14:04 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Merge pull request #3048 from BerriAI/litellm_better_error_msgs

Browse source 
[Fix]  better error msgs when `/key/delete` raises an error
This commit is contained in:
Ishaan Jaff 2024-04-15 17:43:05 -07:00 • committed by GitHub
commit f843a0d654
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 8 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

11
litellm/proxy/proxy_server.py
View file

@ -2817,12 +2817,17 @@ async def delete_verification_token(tokens: List, user_id: Optional[str] = None)
deleted_tokens = await prisma_client.delete_data( deleted_tokens = await prisma_client.delete_data(
tokens=tokens, user_id=user_id tokens=tokens, user_id=user_id
) )
_num_deleted_tokens = deleted_tokens.get("deleted_keys", 0)
if _num_deleted_tokens != len(tokens):
raise Exception(
"Failed to delete all tokens. Tried to delete tokens that don't belong to user: "
+ str(user_id)
)
else: else:
raise Exception raise Exception("DB not connected. prisma_client is None")
except Exception as e: except Exception as e:
traceback.print_exc() traceback.print_exc()
raise HTTPException(status_code=status.HTTP_500_INTERNAL_SERVER_ERROR) raise e
return deleted_tokens return deleted_tokens