From 8dc1af666b417895c877b9df8aad2a0417abb159 Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Wed, 28 Aug 2024 11:51:47 -0700
Subject: [PATCH] ui fix viewing budget info

---
 ui/litellm-dashboard/src/components/view_key_table.tsx | 7 ++++---
 1 file changed, 4 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 9e765e81d87..7d6242098d3 100644
--- a/ui/litellm-dashboard/src/components/view_key_table.tsx
+++ b/ui/litellm-dashboard/src/components/view_key_table.tsx
@@ -88,7 +88,8 @@ interface ItemData {
   metadata: any;
   user_id: string | null;
   expires: any;
-  budget_reset_at?: string | null;
+  budget_duration: string | null;
+  budget_reset_at: string | null;
   // Add any other properties that exist in the item data
 }
 
@@ -306,7 +307,7 @@ const ViewKeyTable: React.FC<ViewKeyTableProps> = ({
                 name="budget_duration"
                 help={`Current Reset Budget: ${
                   token.budget_duration
-                }, budget will be reset at ${new Date(token.budget_reset_at).toLocaleString()}`}
+                }, budget will be reset: ${token.budget_reset_at ? new Date(token.budget_reset_at).toLocaleString() : 'Never'}`}
               >
                 <Select placeholder="n/a">
                   <Select.Option value="daily">daily</Select.Option>
@@ -921,7 +922,7 @@ const ViewKeyTable: React.FC<ViewKeyTableProps> = ({
                     {selectedToken.budget_duration && (
                       <>
                         <br />
-                        Budget will be reset at {new Date(selectedToken.budget_reset_at).toLocaleString()}
+                        Budget will be reset at {selectedToken.budget_reset_at ? new Date(selectedToken.budget_reset_at).toLocaleString() : 'Never'}
                       </>
                     )}
                   </>