From 3527e47b59fe169a261028aec8fd2e8e3b348f6d Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Mon, 26 Aug 2024 18:33:18 -0700
Subject: [PATCH] ui regenerate an api key

---
 .../src/components/view_key_table.tsx         | 20 ++++++++++++++++---
 1 file changed, 17 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 752aad37f9..620c8ca1c1 100644
--- a/ui/litellm-dashboard/src/components/view_key_table.tsx
+++ b/ui/litellm-dashboard/src/components/view_key_table.tsx
@@ -620,6 +620,10 @@ const ViewKeyTable: React.FC<ViewKeyTableProps> = ({
 
   const handleRegenerateKey = async () => {
     try {
+      if (selectedToken == null) {
+        message.error("Please select a key to regenerate");
+        return;
+      }
       const response = await regenerateKeyCall(accessToken, selectedToken.token);
       setRegeneratedKey(response.key);
       setRegenerateDialogVisible(false);
@@ -974,8 +978,15 @@ const ViewKeyTable: React.FC<ViewKeyTableProps> = ({
     <Modal
       title="Regenerate API Key"
       visible={regenerateDialogVisible}
-      onOk={handleRegenerateKey}
       onCancel={() => setRegenerateDialogVisible(false)}
+      footer={[
+        <Button key="cancel" onClick={() => setRegenerateDialogVisible(false)} className="mr-2">
+          Cancel
+        </Button>,
+        <Button key="regenerate" onClick={handleRegenerateKey}>
+          Regenerate
+        </Button>
+      ]}
     >
       <p>Are you sure you want to regenerate this key?</p>
       <p>Key Alias:</p>
@@ -986,9 +997,12 @@ const ViewKeyTable: React.FC<ViewKeyTableProps> = ({
     {regeneratedKey && (
       <Modal
         visible={!!regeneratedKey}
-        onOk={() => setRegeneratedKey(null)}
         onCancel={() => setRegeneratedKey(null)}
-        footer={null}
+        footer={[
+          <Button key="close" onClick={() => setRegeneratedKey(null)}>
+            Close
+          </Button>
+        ]}
       >
         <Grid numItems={1} className="gap-2 w-full">
           <Title>Save your New Key</Title>