Spaces:
Paused
Paused
File size: 1,875 Bytes
df56b50 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 | /**
* Hooks for fetching usage stats data.
*/
import { useState, useEffect, useCallback } from "preact/hooks";
export interface UsageSummary {
total_input_tokens: number;
total_output_tokens: number;
total_request_count: number;
total_accounts: number;
active_accounts: number;
}
export interface UsageDataPoint {
timestamp: string;
input_tokens: number;
output_tokens: number;
request_count: number;
}
export type Granularity = "raw" | "hourly" | "daily";
export function useUsageSummary(refreshIntervalMs = 30_000) {
const [summary, setSummary] = useState<UsageSummary | null>(null);
const [loading, setLoading] = useState(true);
const load = useCallback(async () => {
try {
const resp = await fetch("/admin/usage-stats/summary");
if (resp.ok) setSummary(await resp.json());
} catch { /* ignore */ }
setLoading(false);
}, []);
useEffect(() => {
load();
const id = setInterval(load, refreshIntervalMs);
return () => clearInterval(id);
}, [load, refreshIntervalMs]);
return { summary, loading };
}
export function useUsageHistory(granularity: Granularity, hours: number, refreshIntervalMs = 60_000) {
const [dataPoints, setDataPoints] = useState<UsageDataPoint[]>([]);
const [loading, setLoading] = useState(true);
const load = useCallback(async () => {
try {
const resp = await fetch(
`/admin/usage-stats/history?granularity=${granularity}&hours=${hours}`,
);
if (resp.ok) {
const body = await resp.json();
setDataPoints(body.data_points);
}
} catch { /* ignore */ }
setLoading(false);
}, [granularity, hours]);
useEffect(() => {
setLoading(true);
load();
const id = setInterval(load, refreshIntervalMs);
return () => clearInterval(id);
}, [load, refreshIntervalMs]);
return { dataPoints, loading };
}
|