fix(status): account cached prompt tokens

This commit is contained in:
Peter Steinberger
2025-12-12 23:22:05 +00:00
parent e502ad13f9
commit c3aed2543e
6 changed files with 171 additions and 27 deletions

View File

@@ -1,5 +1,7 @@
import fs from "node:fs";
import os from "node:os";
import path from "node:path";
import { afterEach, describe, expect, it, vi } from "vitest";
import { buildStatusMessage } from "./status.js";
afterEach(() => {
@@ -60,4 +62,55 @@ describe("buildStatusMessage", () => {
expect(text).toContain("Context:");
expect(text).toContain("Web: not linked");
});
it("prefers cached prompt tokens from the session log", () => {
const dir = fs.mkdtempSync(path.join(os.tmpdir(), "clawdis-status-"));
const storePath = path.join(dir, "sessions.json");
const sessionId = "sess-1";
const logPath = path.join(dir, `${sessionId}.jsonl`);
fs.writeFileSync(
logPath,
[
JSON.stringify({
type: "message",
message: {
role: "assistant",
model: "claude-opus-4-5",
usage: {
input: 1,
output: 2,
cacheRead: 1000,
cacheWrite: 0,
totalTokens: 1003,
},
},
}),
].join("\n"),
"utf-8",
);
const text = buildStatusMessage({
reply: {
mode: "command",
command: ["echo", "{{Body}}"],
agent: { kind: "pi", model: "claude-opus-4-5", contextTokens: 32_000 },
session: { scope: "per-sender" },
},
sessionEntry: {
sessionId,
updatedAt: 0,
totalTokens: 3, // would be wrong if cached prompt tokens exist
contextTokens: 32_000,
},
sessionKey: "main",
sessionScope: "per-sender",
storePath,
webLinked: true,
});
expect(text).toContain("Context: 1.0k/32k");
fs.rmSync(dir, { recursive: true, force: true });
});
});