mirror of
https://github.com/fosrl/pangolin.git
synced 2026-08-19 10:42:57 +02:00
send, process, store, display virtual api key ai information in usage and sessions
This commit is contained in:
@@ -29,6 +29,8 @@ import { ProvidersTab } from "./ai-usage-analytics/ProvidersTab";
|
||||
import { ResourcesTab } from "./ai-usage-analytics/ResourcesTab";
|
||||
import { RolesTab } from "./ai-usage-analytics/RolesTab";
|
||||
import { UsersTab } from "./ai-usage-analytics/UsersTab";
|
||||
import { VirtualApiKeysTab } from "./ai-usage-analytics/VirtualApiKeysTab";
|
||||
import { formatVirtualApiKeyPreview } from "@app/lib/virtualApiKeyFormat";
|
||||
|
||||
export type AiUsageAnalyticsDataProps = {
|
||||
orgId: string;
|
||||
@@ -140,13 +142,20 @@ export function AiUsageAnalyticsData(props: AiUsageAnalyticsDataProps) {
|
||||
value: u.id,
|
||||
label: u.email ?? u.id
|
||||
}));
|
||||
const virtualApiKeyOptions = (filterOptions?.virtualApiKeys ?? []).map(
|
||||
(k) => ({
|
||||
value: k.id,
|
||||
label: k.name ?? formatVirtualApiKeyPreview(k.id, k.lastChars)
|
||||
})
|
||||
);
|
||||
|
||||
const tabs: TabItem[] = [
|
||||
{ title: t("aiUsageTabOverview"), href: "#" },
|
||||
{ title: t("aiUsageTabProviders"), href: "#" },
|
||||
{ title: t("aiUsageTabResources"), href: "#" },
|
||||
{ title: t("aiUsageRolesTab"), href: "#" },
|
||||
{ title: t("aiUsageUsersTab"), href: "#" }
|
||||
{ title: t("aiUsageUsersTab"), href: "#" },
|
||||
{ title: t("aiUsageVirtualApiKeysTab"), href: "#" }
|
||||
];
|
||||
|
||||
return (
|
||||
@@ -180,7 +189,9 @@ export function AiUsageAnalyticsData(props: AiUsageAnalyticsDataProps) {
|
||||
value={filters.providerId?.toString()}
|
||||
options={providerOptions}
|
||||
placeholder={t("aiUsageFilterAllProviders")}
|
||||
onValueChange={(v) => setFilter("providerId", v)}
|
||||
onValueChange={(v) =>
|
||||
setFilter("providerId", v)
|
||||
}
|
||||
/>
|
||||
<FilterSelect
|
||||
id="model"
|
||||
@@ -196,7 +207,9 @@ export function AiUsageAnalyticsData(props: AiUsageAnalyticsDataProps) {
|
||||
value={filters.resourceId?.toString()}
|
||||
options={resourceOptions}
|
||||
placeholder={t("aiUsageFilterAllResources")}
|
||||
onValueChange={(v) => setFilter("resourceId", v)}
|
||||
onValueChange={(v) =>
|
||||
setFilter("resourceId", v)
|
||||
}
|
||||
/>
|
||||
<FilterSelect
|
||||
id="roleId"
|
||||
@@ -214,6 +227,18 @@ export function AiUsageAnalyticsData(props: AiUsageAnalyticsDataProps) {
|
||||
placeholder={t("aiUsageFilterAllUsers")}
|
||||
onValueChange={(v) => setFilter("userId", v)}
|
||||
/>
|
||||
<FilterSelect
|
||||
id="virtualApiKeyId"
|
||||
label={t("aiUsageFilterVirtualApiKey")}
|
||||
value={filters.virtualApiKeyId}
|
||||
options={virtualApiKeyOptions}
|
||||
placeholder={t(
|
||||
"aiUsageFilterAllVirtualApiKeys"
|
||||
)}
|
||||
onValueChange={(v) =>
|
||||
setFilter("virtualApiKeyId", v)
|
||||
}
|
||||
/>
|
||||
|
||||
{!isEmptySearchParams && (
|
||||
<Button
|
||||
@@ -259,6 +284,7 @@ export function AiUsageAnalyticsData(props: AiUsageAnalyticsDataProps) {
|
||||
<ResourcesTab orgId={props.orgId} filters={filters} />
|
||||
<RolesTab orgId={props.orgId} filters={filters} />
|
||||
<UsersTab orgId={props.orgId} filters={filters} />
|
||||
<VirtualApiKeysTab orgId={props.orgId} filters={filters} />
|
||||
</HorizontalTabs>
|
||||
</div>
|
||||
);
|
||||
|
||||
@@ -0,0 +1,108 @@
|
||||
"use client";
|
||||
|
||||
import { useQuery } from "@tanstack/react-query";
|
||||
import { aiUsageAnalyticsQueries } from "@app/lib/queries";
|
||||
import type { AiUsageAnalyticsFilters } from "@app/lib/queries";
|
||||
import { formatVirtualApiKeyPreview } from "@app/lib/virtualApiKeyFormat";
|
||||
import { useTranslations } from "next-intl";
|
||||
import { Card, CardContent, CardHeader } from "@app/components/ui/card";
|
||||
import { ToggleableTrendChart } from "./ToggleableTrendChart";
|
||||
import { TopEntitiesList, type TopEntity } from "./TopEntitiesList";
|
||||
import { buildSeriesFromData, formatCost } from "./shared";
|
||||
|
||||
type VirtualApiKeysTabProps = {
|
||||
orgId: string;
|
||||
filters: AiUsageAnalyticsFilters;
|
||||
};
|
||||
|
||||
const UNKNOWN_VIRTUAL_API_KEY_KEY = "unknown";
|
||||
|
||||
export function VirtualApiKeysTab(props: VirtualApiKeysTabProps) {
|
||||
const t = useTranslations();
|
||||
const { data, isLoading } = useQuery(
|
||||
aiUsageAnalyticsQueries.virtualApiKeys({
|
||||
orgId: props.orgId,
|
||||
filters: props.filters
|
||||
})
|
||||
);
|
||||
|
||||
const labelByKey = new Map<string, string>();
|
||||
for (const k of data?.topVirtualApiKeys ?? []) {
|
||||
if (k.virtualApiKeyId) {
|
||||
labelByKey.set(k.virtualApiKeyId, k.name ?? k.virtualApiKeyId);
|
||||
}
|
||||
}
|
||||
const virtualApiKeyLabelFor = (key: string) =>
|
||||
key === UNKNOWN_VIRTUAL_API_KEY_KEY
|
||||
? t("aiUsageUnknownVirtualApiKey")
|
||||
: (labelByKey.get(key) ?? key);
|
||||
|
||||
const virtualApiKeyCostSeries = buildSeriesFromData(
|
||||
data?.virtualApiKeyCostPerDay ?? [],
|
||||
virtualApiKeyLabelFor,
|
||||
t("aiUsageOther")
|
||||
);
|
||||
const virtualApiKeyTokensSeries = buildSeriesFromData(
|
||||
data?.virtualApiKeyTokensPerDay ?? [],
|
||||
virtualApiKeyLabelFor,
|
||||
t("aiUsageOther")
|
||||
);
|
||||
|
||||
const topVirtualApiKeys: TopEntity[] = (data?.topVirtualApiKeys ?? []).map(
|
||||
(k) => ({
|
||||
key: k.virtualApiKeyId ?? UNKNOWN_VIRTUAL_API_KEY_KEY,
|
||||
label: k.virtualApiKeyId
|
||||
? (k.name ?? t("aiUsageUnnamedVirtualApiKey"))
|
||||
: t("aiUsageUnknownVirtualApiKey"),
|
||||
sublabel:
|
||||
k.virtualApiKeyId && k.lastChars
|
||||
? formatVirtualApiKeyPreview(k.virtualApiKeyId, k.lastChars)
|
||||
: undefined,
|
||||
requests: k.requests,
|
||||
totalTokens: k.totalTokens,
|
||||
costUsd: k.costUsd
|
||||
})
|
||||
);
|
||||
|
||||
return (
|
||||
<div className="flex flex-col gap-5">
|
||||
<Card>
|
||||
<CardHeader>
|
||||
<h3 className="font-semibold">
|
||||
{t("aiUsageTopVirtualApiKeys")}
|
||||
</h3>
|
||||
</CardHeader>
|
||||
<CardContent>
|
||||
<TopEntitiesList
|
||||
entities={topVirtualApiKeys}
|
||||
isLoading={isLoading}
|
||||
nameColumnLabel={t("aiUsageFilterVirtualApiKey")}
|
||||
/>
|
||||
</CardContent>
|
||||
</Card>
|
||||
<div className="grid lg:grid-cols-2 gap-5">
|
||||
<Card>
|
||||
<CardContent className="p-6">
|
||||
<ToggleableTrendChart
|
||||
title={t("aiUsageVirtualApiKeyCost")}
|
||||
data={data?.virtualApiKeyCostPerDay ?? []}
|
||||
series={virtualApiKeyCostSeries}
|
||||
isLoading={isLoading}
|
||||
valueFormatter={(v) => formatCost(v)}
|
||||
/>
|
||||
</CardContent>
|
||||
</Card>
|
||||
<Card>
|
||||
<CardContent className="p-6">
|
||||
<ToggleableTrendChart
|
||||
title={t("aiUsageVirtualApiKeyTokenUsage")}
|
||||
data={data?.virtualApiKeyTokensPerDay ?? []}
|
||||
series={virtualApiKeyTokensSeries}
|
||||
isLoading={isLoading}
|
||||
/>
|
||||
</CardContent>
|
||||
</Card>
|
||||
</div>
|
||||
</div>
|
||||
);
|
||||
}
|
||||
Reference in New Issue
Block a user