From 5d9f2e7115d546d7359b0a4641197b29beba5e0d Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Fri, 6 Sep 2024 14:10:02 -0700
Subject: [PATCH] working regen flow

---
 ui/litellm-dashboard/src/components/networking.tsx     | 4 ++--
 ui/litellm-dashboard/src/components/view_key_table.tsx | 4 ++--
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/ui/litellm-dashboard/src/components/networking.tsx b/ui/litellm-dashboard/src/components/networking.tsx
index a6bd5d32cb0..27d1096992e 100644
--- a/ui/litellm-dashboard/src/components/networking.tsx
+++ b/ui/litellm-dashboard/src/components/networking.tsx
@@ -771,7 +771,7 @@ export const claimOnboardingToken = async (
   }
 };
 
-export const regenerateKeyCall = async (accessToken: string, keyToRegenerate: string) => {
+export const regenerateKeyCall = async (accessToken: string, keyToRegenerate: string, formData: any) => {
   try {
     const url = proxyBaseUrl
       ? `${proxyBaseUrl}/key/${keyToRegenerate}/regenerate`
@@ -783,7 +783,7 @@ export const regenerateKeyCall = async (accessToken: string, keyToRegenerate: st
         [globalLitellmHeaderName]: `Bearer ${accessToken}`,
         "Content-Type": "application/json",
       },
-      body: JSON.stringify({}),
+      body: JSON.stringify(formData),
     });
 
     if (!response.ok) {
diff --git a/ui/litellm-dashboard/src/components/view_key_table.tsx b/ui/litellm-dashboard/src/components/view_key_table.tsx
index d22e844d44e..5c482fcd4dd 100644
--- a/ui/litellm-dashboard/src/components/view_key_table.tsx
+++ b/ui/litellm-dashboard/src/components/view_key_table.tsx
@@ -749,7 +749,7 @@ const ViewKeyTable: React.FC<ViewKeyTableProps> = ({
 
     try {
       const formValues = await regenerateForm.validateFields();
-      const response = await regenerateKeyCall(accessToken, { ...formValues, token: selectedToken?.token });
+      const response = await regenerateKeyCall(accessToken, selectedToken.token, formValues);
       setRegeneratedKey(response.key);
 
       // Update the data state with the new key_name
@@ -1173,7 +1173,7 @@ const ViewKeyTable: React.FC<ViewKeyTableProps> = ({
         }}
       >
           <Form.Item name="key_alias" label="Key Alias">
-            <TextInput />
+            <TextInput disabled={true} />
           </Form.Item>
           <Form.Item name="max_budget" label="Max Budget (USD)">
             <InputNumber step={0.01} precision={2} style={{ width: '100%' }} />