From 4795fced10b5846b95c90d59719e282b228733a6 Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Wed, 3 Apr 2024 18:33:16 -0700
Subject: [PATCH] fix deleting a key on ui

---
 .../src/components/view_key_table.tsx                | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/ui/litellm-dashboard/src/components/view_key_table.tsx b/ui/litellm-dashboard/src/components/view_key_table.tsx
index e330f7427..c1e9f0d56 100644
--- a/ui/litellm-dashboard/src/components/view_key_table.tsx
+++ b/ui/litellm-dashboard/src/components/view_key_table.tsx
@@ -299,13 +299,19 @@ const handleEditSubmit = async (formValues: Record<string, any>) => {
 
   
 
-  const handleDelete = async (token: string) => {
+  const handleDelete = async (token: any) => {
+    console.log("handleDelete:", token);
+    if (token.token == null) {
+      if (token.token_id !== null) {
+        token.token = token.token_id;
+      }
+    }
     if (data == null) {
       return;
     }
 
     // Set the key to delete and open the confirmation modal
-    setKeyToDelete(token);
+    setKeyToDelete(token.token);
     localStorage.removeItem("userData" + userID);
     setIsDeleteModalOpen(true);
   };
@@ -597,7 +603,7 @@ const handleEditSubmit = async (formValues: Record<string, any>) => {
                     onClick={() => handleEditClick(item)}
                   />
                   <Icon
-                    onClick={() => handleDelete(item.token)}
+                    onClick={() => handleDelete(item)}
                     icon={TrashIcon}
                     size="sm"
                   />