From 5299e9ad4ae9acae59decb5eb995693ec7920c2b Mon Sep 17 00:00:00 2001
From: tanjiro <56165694+NANDINI-star@users.noreply.github.com>
Date: Thu, 21 Aug 2025 06:34:38 +0900
Subject: [PATCH] fix key edit settings after regenerating key (#13815)

---
 .../components/templates/key_edit_view.tsx    | 21 +++++++++++++++++++
 1 file changed, 21 insertions(+)

diff --git a/ui/litellm-dashboard/src/components/templates/key_edit_view.tsx b/ui/litellm-dashboard/src/components/templates/key_edit_view.tsx
index 9462a9e105..8e5c5e7069 100644
--- a/ui/litellm-dashboard/src/components/templates/key_edit_view.tsx
+++ b/ui/litellm-dashboard/src/components/templates/key_edit_view.tsx
@@ -130,6 +130,7 @@ export function KeyEditView({
   // Set initial form values
   const initialValues = {
     ...keyData,
+    token: keyData.token || keyData.token_id,
     budget_duration: getBudgetDuration(keyData.budget_duration),
     metadata: formatMetadataForDisplay(keyData.metadata),
     guardrails: keyData.metadata?.guardrails || [],
@@ -145,6 +146,26 @@ export function KeyEditView({
       : [],
   }
 
+  useEffect(() => {
+    form.setFieldsValue({
+      ...keyData,
+      token: keyData.token || keyData.token_id,
+      budget_duration: getBudgetDuration(keyData.budget_duration),
+      metadata: formatMetadataForDisplay(keyData.metadata),
+      guardrails: keyData.metadata?.guardrails || [],
+      prompts: keyData.metadata?.prompts || [],
+      vector_stores: keyData.object_permission?.vector_stores || [],
+      mcp_servers_and_groups: {
+        servers: keyData.object_permission?.mcp_servers || [],
+        accessGroups: keyData.object_permission?.mcp_access_groups || [],
+      },
+      logging_settings: extractLoggingSettings(keyData.metadata),
+      disabled_callbacks: Array.isArray(keyData.metadata?.litellm_disabled_callbacks)
+        ? mapInternalToDisplayNames(keyData.metadata.litellm_disabled_callbacks)
+        : [],
+    })
+  }, [keyData, form])
+
   console.log("premiumUser:", premiumUser)
 
   return (