fix(runtime): derive unattributed cost from usage status

This commit is contained in:
ZacharyZhang-NY committed 2026-04-22 16:18:03 -04:00
1 parent f107415f4f
commit ebcc15c68d
8 files changed
+61 -7

No files matched your search

+43
View File
@@ -0,0 +1,43 @@
import { describe, expect, it } from "vitest";
import { workerMessageSchema } from "../../src/main/worker/entry/types";
const RUN_ID = "00000000-0000-4000-8000-000000000001";
describe("worker message schema", () => {
it("accepts run-finished messages keyed by usage status", () => {
expect(
workerMessageSchema.safeParse({
type: "run-finished",
runId: RUN_ID,
status: "failed",
summary: null,
errorMessage: "Agent stopped with error.",
exitCode: null,
signal: null,
model: "gpt-5.4",
usageStatus: "error_stop",
inputTokens: 12,
outputTokens: 1,
}).success,
).toBe(true);
});
it("rejects legacy unattributed-cost truth in run-finished messages", () => {
expect(
workerMessageSchema.safeParse({
type: "run-finished",
runId: RUN_ID,
status: "failed",
summary: null,
errorMessage: "Agent stopped with error.",
exitCode: null,
signal: null,
model: "gpt-5.4",
usageStatus: "error_stop",
unattributedCost: false,
inputTokens: 12,
outputTokens: 1,
}).success,
).toBe(false);
});
});