From cce1c2fcaf2c9fa55f23af5c9f95af35e803ebd3 Mon Sep 17 00:00:00 2001
From: ishaan-jaff <ishaanjaffer0324@gmail.com>
Date: Mon, 29 Jan 2024 13:44:51 -0800
Subject: [PATCH] (ui) show key budget on ui

---
 .../src/components/view_key_table.tsx          | 18 ++++++++++++++++--
 1 file changed, 16 insertions(+), 2 deletions(-)

diff --git a/ui/litellm-dashboard/src/components/view_key_table.tsx b/ui/litellm-dashboard/src/components/view_key_table.tsx
index 5471ad05ec..e500ffe726 100644
--- a/ui/litellm-dashboard/src/components/view_key_table.tsx
+++ b/ui/litellm-dashboard/src/components/view_key_table.tsx
@@ -58,7 +58,8 @@ const ViewKeyTable: React.FC<ViewKeyTableProps> = ({
         <TableHead>
           <TableRow>
             <TableHeaderCell>Secret Key</TableHeaderCell>
-            <TableHeaderCell>Spend</TableHeaderCell>
+            <TableHeaderCell>Spend (USD)</TableHeaderCell>
+            <TableHeaderCell>Key Budget (USD)</TableHeaderCell>
             <TableHeaderCell>Expires</TableHeaderCell>
           </TableRow>
         </TableHead>
@@ -68,11 +69,24 @@ const ViewKeyTable: React.FC<ViewKeyTableProps> = ({
             return (
               <TableRow key={item.token}>
                 <TableCell>
-                  <Text>{item.key_name}</Text>
+                  {item.key_name != null ? (
+                    <Text>{item.key_name}</Text>
+                  ) : (
+                    <Text>{item.token}</Text>
+                  )
+                }
                 </TableCell>
                 <TableCell>
                   <Text>{item.spend}</Text>
                 </TableCell>
+                <TableCell>
+                  {item.max_budget != null ? (
+                    <Text>{item.max_budget}</Text>
+                  ) : (
+                    <Text>Unlimited Budget</Text>
+                  )
+                }
+                </TableCell>
                 <TableCell>
                   {item.expires != null ? (
                     <Text>{item.expires}</Text>