From 20b479725dbcac91a071fad70da685f0fb988aa3 Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Tue, 2 Apr 2024 21:38:18 -0700
Subject: [PATCH] fix edit key flow on admin ui

---
 ui/litellm-dashboard/src/components/view_key_table.tsx | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/ui/litellm-dashboard/src/components/view_key_table.tsx b/ui/litellm-dashboard/src/components/view_key_table.tsx
index f500def6a..a330c0c74 100644
--- a/ui/litellm-dashboard/src/components/view_key_table.tsx
+++ b/ui/litellm-dashboard/src/components/view_key_table.tsx
@@ -204,6 +204,11 @@ const ViewKeyTable: React.FC<ViewKeyTableProps> = ({
   
   const handleEditClick = (token: any) => {
     console.log("handleEditClick:", token);
+
+    // set token.token to token.token_id if token_id is not null
+    if (token.token_id !== null) {
+      token.token = token.token_id;
+    }
     setSelectedToken(token);
     setEditModalVisible(true);
   };