Codex
38 lines
1.6 KiB
TypeScript
38 lines
1.6 KiB
TypeScript
import assert from "node:assert/strict";
|
|
import { test } from "bun:test";
|
|
|
|
import { createCodeAgentTraceStore } from "./code-agent-trace-store.ts";
|
|
|
|
test("code agent trace store keeps assistant deltas outside event count", () => {
|
|
const traceStore = createCodeAgentTraceStore({ maxEvents: 6000 });
|
|
const traceId = "trc_trace-store-assistant-stream";
|
|
traceStore.append(traceId, { type: "request", status: "accepted", label: "request:accepted" });
|
|
for (let index = 0; index < 6200; index += 1) {
|
|
traceStore.appendAssistantDelta(traceId, {
|
|
itemId: "item_assistant_stream",
|
|
chunk: `chunk-${index} `,
|
|
waitingFor: "turn/completed"
|
|
});
|
|
}
|
|
const snapshot = traceStore.snapshot(traceId);
|
|
assert.equal(snapshot.eventCount, 1);
|
|
assert.equal(snapshot.events.length, 1);
|
|
assert.equal(snapshot.events[0].label, "request:accepted");
|
|
assert.equal(snapshot.assistantStreams.length, 1);
|
|
assert.equal(snapshot.assistantStreams[0].chunkCount, 6200);
|
|
assert.equal(snapshot.waitingFor, "turn/completed");
|
|
});
|
|
|
|
test("code agent trace store retains six thousand regular events", () => {
|
|
const traceStore = createCodeAgentTraceStore({ maxEvents: 6000 });
|
|
const traceId = "trc_trace-store-regular-events";
|
|
for (let index = 0; index < 6001; index += 1) {
|
|
traceStore.append(traceId, { type: "trace", status: "observed", label: `event:${index}` });
|
|
}
|
|
const snapshot = traceStore.snapshot(traceId);
|
|
assert.equal(snapshot.eventCount, 6000);
|
|
assert.equal(snapshot.events.length, 6000);
|
|
assert.equal(snapshot.events[0].label, "event:1");
|
|
assert.equal(snapshot.events.at(-1).label, "event:6000");
|
|
});
|