phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix(view_key_table_tsx): show abbreviated key name instead of hashed token

Browse source
This commit is contained in:
Krrish Dholakia 2024-02-02 14:56:19 -08:00
parent 276d203aa2
commit 24b8abe10d
2 changed files with 13 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

5
litellm/proxy/proxy_server.py
View file

@ -1429,7 +1429,10 @@ async def generate_key_helper_fn(
"budget_reset_at": key_reset_at, "budget_reset_at": key_reset_at,
"allowed_cache_controls": allowed_cache_controls, "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:]}" key_data["key_name"] = f"sk-...{token[-4:]}"
saved_token = copy.deepcopy(key_data) saved_token = copy.deepcopy(key_data)
if isinstance(saved_token["aliases"], str): if isinstance(saved_token["aliases"], str):

16
ui/litellm-dashboard/src/components/view_key_table.tsx
View file

@ -55,6 +55,7 @@ const ViewKeyTable: React.FC<ViewKeyTableProps> = ({
<Table className="mt-5"> <Table className="mt-5">
<TableHead> <TableHead>
<TableRow> <TableRow>
<TableHeaderCell>Key Alias</TableHeaderCell>
<TableHeaderCell>Secret Key</TableHeaderCell> <TableHeaderCell>Secret Key</TableHeaderCell>
<TableHeaderCell>Spend (USD)</TableHeaderCell> <TableHeaderCell>Spend (USD)</TableHeaderCell>
<TableHeaderCell>Key Budget (USD)</TableHeaderCell> <TableHeaderCell>Key Budget (USD)</TableHeaderCell>
@ -72,9 +73,11 @@ const ViewKeyTable: React.FC<ViewKeyTableProps> = ({
{item.key_alias != null ? ( {item.key_alias != null ? (
<Text>{item.key_alias}</Text> <Text>{item.key_alias}</Text>
) : ( ) : (
<Text>{item.token}</Text> <Text>Not Set</Text>
) )}
} </TableCell>
<TableCell>
<Text>{item.key_name}</Text>
</TableCell> </TableCell>
<TableCell> <TableCell>
<Text>{item.spend}</Text> <Text>{item.spend}</Text>
@ -84,13 +87,12 @@ const ViewKeyTable: React.FC<ViewKeyTableProps> = ({
<Text>{item.max_budget}</Text> <Text>{item.max_budget}</Text>
) : ( ) : (
<Text>Unlimited Budget</Text> <Text>Unlimited Budget</Text>
) )}
}
</TableCell> </TableCell>
<TableCell > <TableCell>
<Text>{item.team_id}</Text> <Text>{item.team_id}</Text>
</TableCell> </TableCell>
<TableCell > <TableCell>
<Text>{JSON.stringify(item.metadata)}</Text> <Text>{JSON.stringify(item.metadata)}</Text>
</TableCell> </TableCell>
<TableCell> <TableCell>