diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py index 7917f14c3..86fd69fa6 100644 --- a/litellm/proxy/proxy_server.py +++ b/litellm/proxy/proxy_server.py @@ -3785,10 +3785,12 @@ async def delete_key_fn( number_deleted_keys = await delete_verification_token( 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: - assert len(keys) == number_deleted_keys + assert len(keys) == number_deleted_keys["deleted_keys"] except Exception as e: raise HTTPException( status_code=400, diff --git a/litellm/proxy/utils.py b/litellm/proxy/utils.py index d95f5a550..4bfb87058 100644 --- a/litellm/proxy/utils.py +++ b/litellm/proxy/utils.py @@ -1379,9 +1379,11 @@ class PrismaClient: } else: filter_query = {"token": {"in": hashed_tokens}} + deleted_tokens = await self.db.litellm_verificationtoken.delete_many( where=filter_query # type: ignore ) + verbose_proxy_logger.debug(f"deleted_tokens: {deleted_tokens}") return {"deleted_keys": deleted_tokens} elif ( table_name == "team"