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

fix(proxy_server.py): bug fix

Browse source
This commit is contained in:
Krrish Dholakia 2024-03-11 13:54:58 -07:00
parent 4eb244c3ca
commit 2addd66393
2 changed files with 6 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

6
litellm/proxy/proxy_server.py
View file

@ -3785,10 +3785,12 @@ async def delete_key_fn(
number_deleted_keys = await delete_verification_token( number_deleted_keys = await delete_verification_token(
tokens=keys, user_id=user_id tokens=keys, user_id=user_id
) )
verbose_proxy_logger.debug("/key/delete - deleted_keys=", number_deleted_keys) verbose_proxy_logger.debug(
f"/key/delete - deleted_keys={number_deleted_keys['deleted_keys']}"
)
try: try:
assert len(keys) == number_deleted_keys assert len(keys) == number_deleted_keys["deleted_keys"]
except Exception as e: except Exception as e:
raise HTTPException( raise HTTPException(
status_code=400, status_code=400,

2
litellm/proxy/utils.py
View file

@ -1379,9 +1379,11 @@ class PrismaClient:
} }
else: else:
filter_query = {"token": {"in": hashed_tokens}} filter_query = {"token": {"in": hashed_tokens}}
deleted_tokens = await self.db.litellm_verificationtoken.delete_many( deleted_tokens = await self.db.litellm_verificationtoken.delete_many(
where=filter_query # type: ignore where=filter_query # type: ignore
) )
verbose_proxy_logger.debug(f"deleted_tokens: {deleted_tokens}")
return {"deleted_keys": deleted_tokens} return {"deleted_keys": deleted_tokens}
elif ( elif (
table_name == "team" table_name == "team"