scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-21 00:21:49 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 74

fix(ui): hide Top Virtual Keys user column when rows carry no user

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
Devin AI 2026-09-16 01:31:38 +00:00 • committed by ryan-crabbe-berri
parent 3d805e5166
commit e93fe60512
3 changed files with 36 additions and 16 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

28
ui/litellm-dashboard/src/components/UsagePage/components/EntityUsage/TopKeyView.test.tsx
View file

@ -119,8 +119,24 @@ describe("TopKeyView", () => {
expect(screen.getByText("$100.00")).toBeInTheDocument();
});
it("should display user attribution when the key has no alias", () => {
render(
it("should render User column only when a row has user attribution", () => {
const { rerender } = render(
<TopKeyView
{...baseProps}
topKeys={[
{
api_key: "key-123",
key_alias: "Key without user",
user_email: null,
spend: 100,
},
]}
/>,
);
expect(screen.queryByText("User")).not.toBeInTheDocument();
rerender(
<TopKeyView
{...baseProps}
topKeys={[
@ -243,7 +259,7 @@ describe("TopKeyView", () => {
]}
/>,
);
expect(screen.getAllByText("-")).toHaveLength(2);
expect(screen.getAllByText("-")).toHaveLength(1);
});
it("should format spend values with two decimal places", () => {
@ -294,7 +310,7 @@ describe("TopKeyView", () => {
]}
/>,
);
expect(screen.getAllByText("-")).toHaveLength(2);
expect(screen.getAllByText("-")).toHaveLength(1);
expect(screen.queryByText("$0.00")).not.toBeInTheDocument();
});
@ -697,13 +713,13 @@ describe("TopKeyView", () => {
topKeys={[
{
api_key: "key-123",
key_alias: "",
key_alias: null,
user_email: null,
spend: 100,
},
]}
/>,
);
expect(screen.getAllByText("-")).toHaveLength(2);
expect(screen.getAllByText("-")).toHaveLength(1);
});
});

18
ui/litellm-dashboard/src/components/UsagePage/components/EntityUsage/TopKeyView.tsx
View file

@ -17,8 +17,8 @@ const TOP_KEYS_LIMITS = [5, 10, 25, 50] as const;
export interface TopKeyItem {
api_key: string;
key_alias: string;
user_email: string | null;
key_alias: string | null;
user_email?: string | null;
tags?: TagUsage[] | null;
spend: number;
}
@ -103,11 +103,15 @@ const TopKeyView: React.FC<TopKeyViewProps> = ({ topKeys, teams, showTags = fals
accessorKey: "key_alias",
cell: (info: any) => info.getValue() || "-",
},
{
header: "User",
accessorKey: "user_email",
cell: (info: any) => info.getValue() || "-",
},
...(topKeys.some((k) => k.user_email)
? [
{
header: "User",
accessorKey: "user_email",
cell: (info: any) => info.getValue() || "-",
},
]
: []),
];
const tagsColumn = {

6
ui/litellm-dashboard/tests/top_key_view.test.tsx
View file

@ -197,7 +197,7 @@ describe("TopKeyView", () => {
];
renderWithProviders(<TopKeyView {...mockProps} topKeys={keysWithoutTags} showTags={true} />);
expect(screen.getAllByText("-")).toHaveLength(2);
expect(screen.getAllByText("-")).toHaveLength(1);
});
it("should handle keys with undefined tags", () => {
@ -212,7 +212,7 @@ describe("TopKeyView", () => {
];
renderWithProviders(<TopKeyView {...mockProps} topKeys={keysWithUndefinedTags} showTags={true} />);
expect(screen.getAllByText("-")).toHaveLength(2);
expect(screen.getAllByText("-")).toHaveLength(1);
});
it("should handle keys with null tags", () => {
@ -227,7 +227,7 @@ describe("TopKeyView", () => {
];
renderWithProviders(<TopKeyView {...mockProps} topKeys={keysWithNullTags} showTags={true} />);
expect(screen.getAllByText("-")).toHaveLength(2);
expect(screen.getAllByText("-")).toHaveLength(1);
});
});