From 24b8abe10df338dcedf8489e1f5477c463b5265a Mon Sep 17 00:00:00 2001
From: Krrish Dholakia <krrishdholakia@gmail.com>
Date: Fri, 2 Feb 2024 14:56:19 -0800
Subject: [PATCH] fix(view_key_table_tsx): show abbreviated key name instead of
 hashed token

---
 litellm/proxy/proxy_server.py                    |  5 ++++-
 .../src/components/view_key_table.tsx            | 16 +++++++++-------
 2 files changed, 13 insertions(+), 8 deletions(-)

diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py
index 85a5b2d8a5..7498c45ac4 100644
--- a/litellm/proxy/proxy_server.py
+++ b/litellm/proxy/proxy_server.py
@@ -1429,7 +1429,10 @@ async def generate_key_helper_fn(
             "budget_reset_at": key_reset_at,
             "allowed_cache_controls": allowed_cache_controls,
         }
-        if general_settings.get("allow_user_auth", False) == True:
+        if (
+            general_settings.get("allow_user_auth", False) == True
+            or _has_user_setup_sso() == True
+        ):
             key_data["key_name"] = f"sk-...{token[-4:]}"
         saved_token = copy.deepcopy(key_data)
         if isinstance(saved_token["aliases"], str):
diff --git a/ui/litellm-dashboard/src/components/view_key_table.tsx b/ui/litellm-dashboard/src/components/view_key_table.tsx
index a69ff5c93d..bcfa8a23e6 100644
--- a/ui/litellm-dashboard/src/components/view_key_table.tsx
+++ b/ui/litellm-dashboard/src/components/view_key_table.tsx
@@ -55,6 +55,7 @@ const ViewKeyTable: React.FC<ViewKeyTableProps> = ({
       <Table className="mt-5">
         <TableHead>
           <TableRow>
+            <TableHeaderCell>Key Alias</TableHeaderCell>
             <TableHeaderCell>Secret Key</TableHeaderCell>
             <TableHeaderCell>Spend (USD)</TableHeaderCell>
             <TableHeaderCell>Key Budget (USD)</TableHeaderCell>
@@ -72,9 +73,11 @@ const ViewKeyTable: React.FC<ViewKeyTableProps> = ({
                   {item.key_alias != null ? (
                     <Text>{item.key_alias}</Text>
                   ) : (
-                    <Text>{item.token}</Text>
-                  )
-                }
+                    <Text>Not Set</Text>
+                  )}
+                </TableCell>
+                <TableCell>
+                  <Text>{item.key_name}</Text>
                 </TableCell>
                 <TableCell>
                   <Text>{item.spend}</Text>
@@ -84,13 +87,12 @@ const ViewKeyTable: React.FC<ViewKeyTableProps> = ({
                     <Text>{item.max_budget}</Text>
                   ) : (
                     <Text>Unlimited Budget</Text>
-                  )
-                }
+                  )}
                 </TableCell>
-                <TableCell >
+                <TableCell>
                   <Text>{item.team_id}</Text>
                 </TableCell>
-                <TableCell >
+                <TableCell>
                   <Text>{JSON.stringify(item.metadata)}</Text>
                 </TableCell>
                 <TableCell>