From 2addd663939a88b02eab5ccb217cfc70f269fcb6 Mon Sep 17 00:00:00 2001
From: Krrish Dholakia <krrishdholakia@gmail.com>
Date: Mon, 11 Mar 2024 13:54:58 -0700
Subject: [PATCH] fix(proxy_server.py): bug fix

---
 litellm/proxy/proxy_server.py | 6 ++++--
 litellm/proxy/utils.py        | 2 ++
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py
index 7917f14c3f..86fd69fa6f 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 d95f5a5500..4bfb87058b 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"