From bb16abc0437e688300cbc1016ab99d85dd32343e Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Sat, 16 Nov 2024 19:58:26 -0800
Subject: [PATCH] (UI) fix - allow editing key alias on Admin UI  (#6776)

* fix allow editing key alias on UI

* fix non type for budget duration
---
 .../key_management_endpoints.py                    | 14 +++++++++-----
 .../src/components/view_key_table.tsx              |  4 ++++
 2 files changed, 13 insertions(+), 5 deletions(-)

diff --git a/litellm/proxy/management_endpoints/key_management_endpoints.py b/litellm/proxy/management_endpoints/key_management_endpoints.py
index e38236e9b..941fb3b36 100644
--- a/litellm/proxy/management_endpoints/key_management_endpoints.py
+++ b/litellm/proxy/management_endpoints/key_management_endpoints.py
@@ -288,11 +288,15 @@ def prepare_key_update_data(
             non_default_values["expires"] = expires
 
     if "budget_duration" in non_default_values:
-        duration_s = _duration_in_seconds(
-            duration=non_default_values["budget_duration"]
-        )
-        key_reset_at = datetime.now(timezone.utc) + timedelta(seconds=duration_s)
-        non_default_values["budget_reset_at"] = key_reset_at
+        budget_duration = non_default_values.pop("budget_duration")
+        if (
+            budget_duration
+            and (isinstance(budget_duration, str))
+            and len(budget_duration) > 0
+        ):
+            duration_s = _duration_in_seconds(duration=budget_duration)
+            key_reset_at = datetime.now(timezone.utc) + timedelta(seconds=duration_s)
+            non_default_values["budget_reset_at"] = key_reset_at
 
     _metadata = existing_key_row.metadata or {}
 
diff --git a/ui/litellm-dashboard/src/components/view_key_table.tsx b/ui/litellm-dashboard/src/components/view_key_table.tsx
index 474a308e9..b657ed47c 100644
--- a/ui/litellm-dashboard/src/components/view_key_table.tsx
+++ b/ui/litellm-dashboard/src/components/view_key_table.tsx
@@ -291,6 +291,10 @@ const ViewKeyTable: React.FC<ViewKeyTableProps> = ({
         >
                 <>
 
+                <Form.Item name="key_alias" label="Key Alias">
+                  <TextInput />
+                </Form.Item>
+
               <Form.Item label="Models" name="models" rules={[
                 {
                   validator: (rule, value) => {