tbenandClaude Opus 5.5
c13b9ea826
Every bot message gets an info button that opens its details: the provider and model behind it, tokens and cost, the job that ran it, each model used and every call in order. Each model call now records how it was made in token_usages.settings (generation settings sent, whose settings they were, pinned or automatic model, fallbacks), shown per model in the modal. Live replies now carry their token usage, from the job's push and the live relay alike, so it shows without a reload. Co-Authored-By: Claude Opus 5.5 <[email protected]>
30 lines
1.1 KiB
TypeScript
30 lines
1.1 KiB
TypeScript
import type { BUN_SQLITE_COBBLE_BOT_MESSAGES } from "@/db/types/db";
|
|
import type { TokenUsageSummary } from "@/src/harness/usage/types";
|
|
import grabMessageUsage from "./grab-message-usage";
|
|
|
|
type Params<T extends BUN_SQLITE_COBBLE_BOT_MESSAGES> = {
|
|
messages: T[];
|
|
};
|
|
|
|
/**
|
|
* The messages with what their model calls spent (`usage`), for a live
|
|
* push: a pushed bot message must carry it like a loaded one
|
|
* (`grabBotMessages`), or the thread shows its tokens only after a reload.
|
|
* Messages with no calls (the user's own among them) are left as they are.
|
|
*/
|
|
export default async function withMessageUsage<T extends BUN_SQLITE_COBBLE_BOT_MESSAGES>({
|
|
messages,
|
|
}: Params<T>): Promise<(T & { usage?: TokenUsageSummary })[]> {
|
|
const ids = messages.filter((m) => m.id && m.author == "bot").map((m) => Number(m.id));
|
|
|
|
if (!ids.length) return messages;
|
|
|
|
const usage = await grabMessageUsage({ message_ids: ids });
|
|
|
|
return messages.map((message) => {
|
|
const spent = usage.get(Number(message.id));
|
|
|
|
return spent ? { ...message, usage: spent } : message;
|
|
});
|
|
}
|