33 lines
853 B
TypeScript
33 lines
853 B
TypeScript
import { describe, expect, test } from "bun:test";
|
|
import { getDetailStats } from "./detail-stats";
|
|
|
|
describe("getDetailStats", () => {
|
|
test("includes completion tokens in detail page stat cards", () => {
|
|
const stats = getDetailStats({
|
|
calls: 12,
|
|
total_tokens: 1000,
|
|
quota: 250000,
|
|
prompt_tokens: 400,
|
|
completion_tokens: 350,
|
|
cache_creation_tokens: 150,
|
|
cache_read_tokens: 100,
|
|
});
|
|
|
|
expect(stats.map((stat) => stat.key)).toEqual([
|
|
"calls",
|
|
"total_tokens",
|
|
"quota",
|
|
"prompt_tokens",
|
|
"completion_tokens",
|
|
"cache_creation_tokens",
|
|
"cache_read_tokens",
|
|
]);
|
|
expect(stats.find((stat) => stat.key === "completion_tokens")).toEqual({
|
|
key: "completion_tokens",
|
|
labelKey: "th.output",
|
|
value: 350,
|
|
format: "tokens",
|
|
});
|
|
});
|
|
});
|