From 981efffa3b7a4105f67af3138d32e403ae4562cb Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Sat, 14 Dec 2024 14:12:39 -0800
Subject: [PATCH] ui fix key table

---
 .../src/components/view_key_table.tsx         | 37 +++++++++++++++++--
 1 file changed, 33 insertions(+), 4 deletions(-)

diff --git a/ui/litellm-dashboard/src/components/view_key_table.tsx b/ui/litellm-dashboard/src/components/view_key_table.tsx
index 3ef50bc603..1bdc31500f 100644
--- a/ui/litellm-dashboard/src/components/view_key_table.tsx
+++ b/ui/litellm-dashboard/src/components/view_key_table.tsx
@@ -261,6 +261,23 @@ const ViewKeyTable: React.FC<ViewKeyTableProps> = ({
     const [errorModels, setErrorModels] = useState<string[]>([]);
     const [errorBudget, setErrorBudget] = useState<boolean>(false);
 
+    let metadataString = '';
+    try {
+      metadataString = JSON.stringify(token.metadata, null, 2);
+    } catch (error) {
+      console.error("Error stringifying metadata:", error);
+      // You can choose a fallback, such as an empty string or a warning message
+      metadataString = '';
+    }
+
+    // Ensure token is defined and handle gracefully if not
+    const initialValues = token ? {
+      ...token,
+      budget_duration: token.budget_duration,
+      metadata: metadataString
+    } : { metadata: metadataString };
+
+
     const handleOk = () => {
       form
         .validateFields()
@@ -286,7 +303,7 @@ const ViewKeyTable: React.FC<ViewKeyTableProps> = ({
         <Form
           form={form}
           onFinish={handleEditSubmit}
-          initialValues={{...token, budget_duration: token.budget_duration}} // Pass initial values here
+          initialValues={initialValues}
           labelCol={{ span: 8 }}
           wrapperCol={{ span: 16 }}
           labelAlign="left"
@@ -441,13 +458,14 @@ const ViewKeyTable: React.FC<ViewKeyTableProps> = ({
                 <InputNumber step={1} precision={1} width={200} />
               </Form.Item>
               <Form.Item
-                label="Metadata"
+                label="Metadata (ensure this is valid JSON)"
                 name="metadata"
-                initialValue={token.metadata}
               >
                 <TextArea
-                  value={String(token.metadata)}
                   rows={10}
+                  onChange={(e) => {
+                    form.setFieldsValue({ metadata: e.target.value });
+                  }}
                 />
               </Form.Item>
             </>
@@ -675,6 +693,17 @@ const ViewKeyTable: React.FC<ViewKeyTableProps> = ({
   const currentKey = formValues.token; 
   formValues.key = currentKey;
 
+  // Convert metadata back to an object if it exists and is a string
+  if (formValues.metadata && typeof formValues.metadata === 'string') {
+    try {
+      formValues.metadata = JSON.parse(formValues.metadata);
+    } catch (error) {
+      console.error("Error parsing metadata JSON:", error);
+      message.error("Invalid metadata JSON for formValue " + formValues.metadata);
+      return;
+    }
+  }
+
   // Convert the budget_duration back to the API expected format
   if (formValues.budget_duration) {
     switch (formValues.budget_duration) {