@superdurable/dex-agents 0.2.0 → 0.3.0

package/dist/src/testing/context-window.d.ts

@@ -0,0 +1,194 @@
+import { type CompactionPolicy } from "../agent-context-compaction.js";
+import type { AgentMessage, AgentSnapshot } from "../agent-session-values.js";
+import { AgentSessionTestClient } from "./agent-session-test-client.js";
+import { KillableWorkerProcess } from "./agent-worker-process.js";
+/** Environment variable holding the JSON of {@link ContextWindowSettings}. */
+export declare const contextWindowSettingsVariable = "DEX_AGENTS_CONTEXT_WINDOW_SETTINGS";
+/** Tool that the scripted model calls once per turn. */
+export declare const contextWindowToolName = "lookup";
+/** Start of the error that the scripted model throws for an oversized request. */
+export declare const contextWindowExceededError = "scripted model: input exceeds the context window";
+/** Gate that holds a summary call of a range that names `gate-summary`. */
+export declare const contextWindowSummaryGateName = "summary";
+/**
+ * Sizes of the scripted model and the session of one context window test. A Worker child process reads them from
+ * {@link contextWindowSettingsVariable}.
+ */
+export interface ContextWindowSettings {
+    /** Context window of the scripted model, at four characters per token. */
+    readonly windowTokens: number;
+    /** Whether the session gets a compaction policy. */
+    readonly isCompactionEnabled: boolean;
+    /** Size of each tool result in characters. */
+    readonly toolOutputCharacters: number;
+    /** Size of each model answer in characters. */
+    readonly answerCharacters: number;
+    /** `AgentSessionOptions.messageWindowSize`. */
+    readonly messageWindowSize: number;
+    /** `AgentSessionOptions.messageChunkSize`. */
+    readonly messageChunkSize: number;
+    /** `AgentSessionOptions.retainedMessageLimit`; `null` keeps every message. */
+    readonly retainedMessageLimit: number | null;
+}
+/** The settings of a test that changes nothing. */
+export declare const defaultContextWindowSettings: ContextWindowSettings;
+/**
+ * Returns the Worker environment that carries the settings.
+ * @param settings - The test's settings.
+ * @returns One entry under {@link contextWindowSettingsVariable}.
+ */
+export declare function contextWindowSettingsEnvironment(settings: ContextWindowSettings): Record<string, string>;
+/**
+ * Reads the settings a test gave this Worker.
+ * @returns The test's settings, or the defaults outside a Worker child.
+ */
+export declare function readContextWindowSettings(): ContextWindowSettings;
+/**
+ * Returns the `AgentSessionOptions` fields for the settings.
+ * @param settings - The test's settings.
+ * @returns The compaction policy and the message storage options.
+ */
+export declare function contextWindowSessionOptions(settings: ContextWindowSettings): {
+    readonly compaction?: CompactionPolicy;
+    readonly messageWindowSize: number;
+    readonly messageChunkSize: number;
+    readonly retainedMessageLimit: number | null;
+};
+/** What the scripted model does for one request: call the tool with a tag, or answer with a text. */
+export type ContextWindowReply = {
+    readonly kind: "tool_call";
+    readonly tag: string;
+} | {
+    readonly kind: "text";
+    readonly text: string;
+};
+/** One request the scripted model saw. */
+export interface ContextWindowRequest {
+    /** `"model"` for an agent's request, `"summary"` for a summary call. */
+    readonly kind: "model" | "summary";
+    /** Characters of the request, instructions included. */
+    readonly characters: number;
+    /** The whole request as text, in the framework's format. */
+    readonly input: string;
+}
+/**
+ * Answers model requests of a framework adapter and logs every request, like a provider with a small context
+ * window. Each user message is `turn <tag>`: the model calls {@link contextWindowToolName} once with the tag, then
+ * answers {@link answerText}. It holds no state between calls, so it survives a Worker restart.
+ */
+export declare class ContextWindowModelScript {
+    private readonly stateDirectory;
+    private readonly settings;
+    /**
+     * @param stateDirectory - Directory shared with the test.
+     * @param settings - Sizes of the test.
+     */
+    constructor(stateDirectory: string, settings: ContextWindowSettings);
+    /**
+     * Returns what the tool answers.
+     * @param tag - The tool call's `tag` argument.
+     * @returns Text of `toolOutputCharacters` that names the tag.
+     */
+    toolOutput(tag: string): string;
+    /**
+     * Returns the answer of the turn with this tag.
+     * @param tag - The tag of the user message.
+     * @returns Text of `answerCharacters` that names the tag.
+     */
+    answerText(tag: string): string;
+    /**
+     * Answers one model request.
+     * @param request - The system prompt, the whole request as text, and whether the running turn has its result.
+     * @returns The tool call or the answer.
+     * @throws {Error} When the request exceeds the window or names no turn.
+     */
+    reply(request: {
+        readonly instructions: string | undefined;
+        readonly inputText: string;
+        readonly hasToolResult: boolean;
+    }): Promise<ContextWindowReply>;
+}
+/**
+ * Returns every request the scripted model saw, in order.
+ * @param stateDirectory - The test's state directory.
+ * @returns The logged requests.
+ */
+export declare function readContextWindowRequests(stateDirectory: string): ContextWindowRequest[];
+/** Options of {@link ContextWindowSessionScenario}. */
+export interface ContextWindowSessionScenarioOptions {
+    /** Dex Server address of the test's engine. */
+    readonly serverAddress: string;
+    /** Session client for the adapter's Flow. */
+    readonly sessions: AgentSessionTestClient;
+    /** Directory shared with the Worker child processes. */
+    readonly stateDirectory: string;
+    /** Sizes of the test. */
+    readonly settings: ContextWindowSettings;
+    /** Compiled module of the Worker child, which calls `runAgentWorkerProcess`. */
+    readonly workerEntryModulePath: string;
+    /** Environment that tells the Worker child where {@link stateDirectory} is. */
+    readonly workerEnvironment: Readonly<Record<string, string>>;
+}
+/**
+ * Drives an adapter's session past a context window in Worker child processes, and checks what happened. Every
+ * adapter so proves one contract: the session keeps going, the model input shrinks, and the record stays readable.
+ */
+export declare class ContextWindowSessionScenario {
+    private readonly options;
+    /** The session ID of this scenario. */
+    readonly sessionId: string;
+    private readonly workers;
+    /** @param options - The engine, the session client, and the Worker child. */
+    constructor(options: ContextWindowSessionScenarioOptions);
+    /**
+     * Starts a Worker child with the scenario's settings, or with other settings.
+     * @param settings - Settings of this Worker. Defaults to the scenario's.
+     * @returns The Worker.
+     */
+    startWorker(settings?: ContextWindowSettings): Promise<KillableWorkerProcess>;
+    /** Kills every Worker child the scenario started. */
+    close(): Promise<void>;
+    /**
+     * Returns the logged requests of one kind.
+     * @param kind - `"model"` or `"summary"`.
+     * @returns The requests in order.
+     */
+    requests(kind: ContextWindowRequest["kind"]): ContextWindowRequest[];
+    /**
+     * Sends one turn per tag and waits until the session finished them.
+     * @param tags - The turns' tags.
+     * @param isFirst - Whether the first tag starts the session.
+     * @returns The snapshot after the last turn.
+     */
+    runTurns(tags: readonly string[], isFirst?: boolean): Promise<AgentSnapshot>;
+    /**
+     * Reads every message history still holds, through the public RPC.
+     * @returns The messages in sequence order.
+     */
+    readStoredMessages(): Promise<AgentMessage[]>;
+    /**
+     * Counts the finished `CompactContext` Steps of the session.
+     * @returns The count.
+     */
+    countCompactContextSteps(): Promise<number>;
+    /**
+     * Checks the whole contract on a session longer than the window.
+     * @param turnCount - Turns to run.
+     * @returns The final snapshot.
+     */
+    assertOutlivesTheWindow(turnCount?: number): Promise<AgentSnapshot>;
+    /**
+     * Checks that retention deletes only chunks behind the summary. The settings need a small `retainedMessageLimit`.
+     * @param turnCount - Turns to run.
+     */
+    assertRetentionFollowsTheSummary(turnCount?: number): Promise<void>;
+    /**
+     * Checks that a session without a policy is unchanged and reaches the window.
+     * @param turnCount - Turns to run; the window must end the session first.
+     */
+    assertFailsAtTheWindowWithoutAPolicy(turnCount?: number): Promise<void>;
+    /** Checks that a session started without a policy compacts once a Worker has one. */
+    assertStartsCompactingAfterAnUpgrade(): Promise<void>;
+    /** Checks SIGKILL during `CompactContext`: one more summary call, one stored summary, no repeated model call. */
+    assertWorkerKillRerunsOnlyTheSummary(): Promise<void>;
+}

package/dist/src/testing/context-window.js

@@ -0,0 +1,360 @@
+// Copyright (c) 2026 Super Durable, Inc.
+//
+// Licensed under the Sustainable Use License 1.0.
+// You may not use this file except in compliance with the License.
+// See the LICENSE file in the repository root.
+//
+// SPDX-License-Identifier: LicenseRef-Sustainable-Use-1.0
+import assert from "node:assert/strict";
+import { randomUUID } from "node:crypto";
+import { appendFileSync, existsSync, readFileSync } from "node:fs";
+import { join } from "node:path";
+import { contextSummaryInstructions, contextSummaryNoteHeader, } from "../agent-context-compaction.js";
+import { KillableWorkerProcess } from "./agent-worker-process.js";
+import { FileGate } from "./coordination.js";
+import { pollUntil } from "./polling.js";
+import { executeCompletedEvent, readStepHistory } from "./step-history.js";
+/** Environment variable holding the JSON of {@link ContextWindowSettings}. */
+export const contextWindowSettingsVariable = "DEX_AGENTS_CONTEXT_WINDOW_SETTINGS";
+/** Tool that the scripted model calls once per turn. */
+export const contextWindowToolName = "lookup";
+/** Start of the error that the scripted model throws for an oversized request. */
+export const contextWindowExceededError = "scripted model: input exceeds the context window";
+/** Gate that holds a summary call of a range that names `gate-summary`. */
+export const contextWindowSummaryGateName = "summary";
+const requestLogName = "model-requests.jsonl";
+const summaryGateTag = "gate-summary";
+const turnCommandPattern = /turn ([A-Za-z0-9-]+)/gu;
+const providerCharactersPerToken = 4;
+const summaryTagLimit = 10;
+/** The settings of a test that changes nothing. */
+export const defaultContextWindowSettings = Object.freeze({
+    windowTokens: 6000,
+    isCompactionEnabled: true,
+    toolOutputCharacters: 2000,
+    answerCharacters: 1000,
+    messageWindowSize: 8,
+    messageChunkSize: 4,
+    retainedMessageLimit: 1000,
+});
+/**
+ * Returns the Worker environment that carries the settings.
+ * @param settings - The test's settings.
+ * @returns One entry under {@link contextWindowSettingsVariable}.
+ */
+export function contextWindowSettingsEnvironment(settings) {
+    return { [contextWindowSettingsVariable]: JSON.stringify(settings) };
+}
+/**
+ * Reads the settings a test gave this Worker.
+ * @returns The test's settings, or the defaults outside a Worker child.
+ */
+export function readContextWindowSettings() {
+    const value = process.env[contextWindowSettingsVariable];
+    return value === undefined ? defaultContextWindowSettings : JSON.parse(value);
+}
+/**
+ * Returns the `AgentSessionOptions` fields for the settings.
+ * @param settings - The test's settings.
+ * @returns The compaction policy and the message storage options.
+ */
+export function contextWindowSessionOptions(settings) {
+    return {
+        ...(settings.isCompactionEnabled ? { compaction: { contextWindowTokens: settings.windowTokens } } : {}),
+        messageWindowSize: settings.messageWindowSize,
+        messageChunkSize: settings.messageChunkSize,
+        retainedMessageLimit: settings.retainedMessageLimit,
+    };
+}
+/**
+ * Answers model requests of a framework adapter and logs every request, like a provider with a small context
+ * window. Each user message is `turn <tag>`: the model calls {@link contextWindowToolName} once with the tag, then
+ * answers {@link answerText}. It holds no state between calls, so it survives a Worker restart.
+ */
+export class ContextWindowModelScript {
+    stateDirectory;
+    settings;
+    /**
+     * @param stateDirectory - Directory shared with the test.
+     * @param settings - Sizes of the test.
+     */
+    constructor(stateDirectory, settings) {
+        this.stateDirectory = stateDirectory;
+        this.settings = settings;
+    }
+    /**
+     * Returns what the tool answers.
+     * @param tag - The tool call's `tag` argument.
+     * @returns Text of `toolOutputCharacters` that names the tag.
+     */
+    toolOutput(tag) {
+        return `lookup-body-${tag}:${"y".repeat(this.settings.toolOutputCharacters)}`;
+    }
+    /**
+     * Returns the answer of the turn with this tag.
+     * @param tag - The tag of the user message.
+     * @returns Text of `answerCharacters` that names the tag.
+     */
+    answerText(tag) {
+        return `answer-body-${tag}:${"x".repeat(this.settings.answerCharacters)}`;
+    }
+    /**
+     * Answers one model request.
+     * @param request - The system prompt, the whole request as text, and whether the running turn has its result.
+     * @returns The tool call or the answer.
+     * @throws {Error} When the request exceeds the window or names no turn.
+     */
+    async reply(request) {
+        const instructions = request.instructions ?? "";
+        const isSummary = instructions.includes(contextSummaryInstructions);
+        const characters = request.inputText.length + instructions.length;
+        const entry = { kind: isSummary ? "summary" : "model", characters, input: request.inputText };
+        appendFileSync(join(this.stateDirectory, requestLogName), `${JSON.stringify(entry)}\n`);
+        const windowCharacters = this.settings.windowTokens * providerCharactersPerToken;
+        if (characters > windowCharacters) {
+            throw new Error(`${contextWindowExceededError}: ${characters} characters, limit ${windowCharacters}`);
+        }
+        const tags = [...request.inputText.matchAll(turnCommandPattern)].map((match) => match[1] ?? "");
+        if (isSummary) {
+            if (tags.includes(summaryGateTag)) {
+                const gate = new FileGate(this.stateDirectory, contextWindowSummaryGateName);
+                gate.markReached();
+                await gate.waitForRelease();
+            }
+            return { kind: "text", text: `summary of turns: ${[...new Set(tags)].slice(-summaryTagLimit).join(" ")}` };
+        }
+        const tag = tags.at(-1);
+        if (tag === undefined) {
+            throw new Error("the scripted model needs a turn command");
+        }
+        return request.hasToolResult ? { kind: "text", text: this.answerText(tag) } : { kind: "tool_call", tag };
+    }
+}
+/**
+ * Returns every request the scripted model saw, in order.
+ * @param stateDirectory - The test's state directory.
+ * @returns The logged requests.
+ */
+export function readContextWindowRequests(stateDirectory) {
+    const path = join(stateDirectory, requestLogName);
+    if (!existsSync(path)) {
+        return [];
+    }
+    return readFileSync(path, "utf8")
+        .split("\n")
+        .filter((line) => line !== "")
+        .map((line) => JSON.parse(line));
+}
+/**
+ * Drives an adapter's session past a context window in Worker child processes, and checks what happened. Every
+ * adapter so proves one contract: the session keeps going, the model input shrinks, and the record stays readable.
+ */
+export class ContextWindowSessionScenario {
+    options;
+    /** The session ID of this scenario. */
+    sessionId = `session-${randomUUID()}`;
+    workers = [];
+    /** @param options - The engine, the session client, and the Worker child. */
+    constructor(options) {
+        this.options = options;
+    }
+    /**
+     * Starts a Worker child with the scenario's settings, or with other settings.
+     * @param settings - Settings of this Worker. Defaults to the scenario's.
+     * @returns The Worker.
+     */
+    async startWorker(settings = this.options.settings) {
+        const worker = await KillableWorkerProcess.start({
+            entryModulePath: this.options.workerEntryModulePath,
+            serverAddress: this.options.serverAddress,
+            logPath: join(this.options.stateDirectory, "worker.log"),
+            environment: { ...this.options.workerEnvironment, ...contextWindowSettingsEnvironment(settings) },
+        });
+        this.workers.push(worker);
+        return worker;
+    }
+    /** Kills every Worker child the scenario started. */
+    async close() {
+        for (const worker of this.workers.splice(0)) {
+            await worker.kill();
+        }
+    }
+    /**
+     * Returns the logged requests of one kind.
+     * @param kind - `"model"` or `"summary"`.
+     * @returns The requests in order.
+     */
+    requests(kind) {
+        return readContextWindowRequests(this.options.stateDirectory).filter((request) => request.kind === kind);
+    }
+    /**
+     * Sends one turn per tag and waits until the session finished them.
+     * @param tags - The turns' tags.
+     * @param isFirst - Whether the first tag starts the session.
+     * @returns The snapshot after the last turn.
+     */
+    async runTurns(tags, isFirst = true) {
+        const { sessions } = this.options;
+        let commands = tags.map((tag) => `turn ${tag}`);
+        let before = 0;
+        if (isFirst) {
+            await sessions.startSession(this.sessionId, commands[0] ?? "");
+            commands = commands.slice(1);
+        }
+        else {
+            before = (await sessions.readAgentState(this.sessionId))?.turnCount ?? 0;
+        }
+        for (const command of commands) {
+            assert.equal(await sessions.sendUserMessage(this.sessionId, command), true);
+        }
+        await sessions.waitForCompletedTurns(this.sessionId, before + tags.length, { timeoutMs: 180_000 });
+        return sessions.getAgentSnapshot(this.sessionId);
+    }
+    /**
+     * Reads every message history still holds, through the public RPC.
+     * @returns The messages in sequence order.
+     */
+    async readStoredMessages() {
+        const messages = [];
+        let beforeSequence = null;
+        for (;;) {
+            const page = await this.options.sessions.listAgentMessages(this.sessionId, beforeSequence, 200);
+            messages.unshift(...page.messages);
+            if (page.nextBeforeSequence === null) {
+                return messages;
+            }
+            beforeSequence = page.nextBeforeSequence;
+        }
+    }
+    /**
+     * Counts the finished `CompactContext` Steps of the session.
+     * @returns The count.
+     */
+    async countCompactContextSteps() {
+        const history = await readStepHistory(this.options.serverAddress, this.sessionId);
+        return history.filter((entry) => entry.stepType === "CompactContext" && entry.eventType === executeCompletedEvent)
+            .length;
+    }
+    /**
+     * Checks the whole contract on a session longer than the window.
+     * @param turnCount - Turns to run.
+     * @returns The final snapshot.
+     */
+    async assertOutlivesTheWindow(turnCount = 12) {
+        await this.startWorker();
+        const snapshot = await this.runTurns(numberedTags(turnCount));
+        assert.equal(snapshot.state?.lastTurnError ?? null, null);
+        assert.equal(snapshot.state?.status, "waiting_for_user_message");
+        assert.equal(snapshot.state?.turnCount, turnCount);
+        const summary = snapshot.contextSummary;
+        assert.ok(summary);
+        assert.match(summary.text, /^summary of turns:/u);
+        assert.ok((await this.countCompactContextSteps()) >= 1);
+        const limit = this.options.settings.windowTokens * providerCharactersPerToken;
+        const modelRequests = this.requests("model");
+        assert.ok(modelRequests.every((request) => request.characters <= limit));
+        const allRequests = readContextWindowRequests(this.options.stateDirectory);
+        allRequests.forEach((request, index) => {
+            if (request.kind === "summary") {
+                assert.ok((allRequests[index + 1]?.characters ?? 0) < (allRequests[index - 1]?.characters ?? 0));
+            }
+        });
+        const lastInput = modelRequests.at(-1)?.input ?? "";
+        assert.ok(lastInput.includes(contextSummaryNoteHeader));
+        assert.ok(lastInput.includes(summary.text.split("\n")[0] ?? ""));
+        // Turn 1 is behind the summary, so its bodies left the model input.
+        assert.ok(!lastInput.includes("answer-body-t1:"));
+        assert.ok(!lastInput.includes("lookup-body-t1:"));
+        const messages = await this.readStoredMessages();
+        assert.deepEqual(messages.map((message) => message.sequence), messages.map((_, index) => index + 1));
+        assert.equal(messages[0]?.text, "turn t1");
+        assert.ok(!messages.some((message) => message.frameworkItem.includes(contextSummaryNoteHeader)));
+        assert.ok(messages.some((message) => (message.text ?? "").includes("answer-body-t1:")));
+        const summaryCalls = this.requests("summary").length;
+        assert.equal(summaryCalls, await this.countCompactContextSteps());
+        assert.equal(snapshot.usage.modelCalls, modelRequests.length + summaryCalls);
+        return snapshot;
+    }
+    /**
+     * Checks that retention deletes only chunks behind the summary. The settings need a small `retainedMessageLimit`.
+     * @param turnCount - Turns to run.
+     */
+    async assertRetentionFollowsTheSummary(turnCount = 24) {
+        await this.startWorker();
+        const snapshot = await this.runTurns(numberedTags(turnCount));
+        assert.equal(snapshot.state?.lastTurnError ?? null, null);
+        const summary = snapshot.contextSummary;
+        assert.ok(summary);
+        const page = await this.options.sessions.listAgentMessages(this.sessionId, null, 200);
+        const deletedBefore = page.deletedBeforeSequence ?? null;
+        assert.ok(deletedBefore !== null && deletedBefore > 1);
+        assert.ok(deletedBefore - 1 <= summary.lastSequence);
+        const messages = await this.readStoredMessages();
+        assert.equal(messages[0]?.sequence, deletedBefore);
+        assert.deepEqual(messages.map((message) => message.sequence), messages.map((_, index) => deletedBefore + index));
+        const limit = this.options.settings.retainedMessageLimit;
+        assert.ok(limit !== null && messages.length >= limit);
+        assert.ok((messages.at(-1)?.text ?? "").startsWith(`answer-body-t${turnCount}:`));
+    }
+    /**
+     * Checks that a session without a policy is unchanged and reaches the window.
+     * @param turnCount - Turns to run; the window must end the session first.
+     */
+    async assertFailsAtTheWindowWithoutAPolicy(turnCount = 12) {
+        await this.startWorker();
+        const snapshot = await this.runTurns(numberedTags(turnCount));
+        assert.ok((snapshot.state?.lastTurnError ?? "").includes(contextWindowExceededError));
+        assert.equal(snapshot.contextSummary ?? null, null);
+        assert.equal(await this.countCompactContextSteps(), 0);
+        assert.equal(this.requests("summary").length, 0);
+        assert.ok(!this.requests("model").some((request) => request.input.includes(contextSummaryNoteHeader)));
+    }
+    /** Checks that a session started without a policy compacts once a Worker has one. */
+    async assertStartsCompactingAfterAnUpgrade() {
+        const worker = await this.startWorker({ ...this.options.settings, isCompactionEnabled: false });
+        await this.runTurns(["t1", "t2", "t3"]);
+        assert.equal(this.requests("summary").length, 0);
+        await worker.kill();
+        await this.startWorker();
+        const snapshot = await this.runTurns(numberedTags(12).slice(3), false);
+        assert.equal(snapshot.state?.lastTurnError ?? null, null);
+        assert.ok(snapshot.contextSummary);
+        assert.equal((await this.readStoredMessages())[0]?.text, "turn t1");
+    }
+    /** Checks SIGKILL during `CompactContext`: one more summary call, one stored summary, no repeated model call. */
+    async assertWorkerKillRerunsOnlyTheSummary() {
+        const { sessions, stateDirectory } = this.options;
+        const worker = await this.startWorker();
+        const commands = ["gate-summary", ...numberedTags(12).slice(1)].map((tag) => `turn ${tag}`);
+        await sessions.startSession(this.sessionId, commands[0] ?? "");
+        for (const command of commands.slice(1)) {
+            assert.equal(await sessions.sendUserMessage(this.sessionId, command), true);
+        }
+        const gate = new FileGate(stateDirectory, contextWindowSummaryGateName);
+        await gate.waitUntilReached();
+        const compacting = await pollUntil(async () => {
+            const snapshot = await sessions.getAgentSnapshot(this.sessionId);
+            return snapshot.state?.status === "compacting_context" ? snapshot : undefined;
+        }, { description: "the session compacting" });
+        assert.equal(compacting.contextSummary ?? null, null);
+        const modelCallsBeforeKill = this.requests("model").length;
+        await worker.kill();
+        gate.release();
+        await this.startWorker();
+        await sessions.waitForCompletedTurns(this.sessionId, commands.length, { timeoutMs: 180_000 });
+        const snapshot = await sessions.getAgentSnapshot(this.sessionId);
+        assert.equal(gate.reachedCount(), 2);
+        assert.equal(snapshot.state?.lastTurnError ?? null, null);
+        assert.ok(snapshot.contextSummary);
+        const modelRequests = this.requests("model");
+        // The model call that started compaction never ran, so the kill repeats no model call.
+        assert.ok(modelCallsBeforeKill < modelRequests.length);
+        const summaryCalls = this.requests("summary").length;
+        assert.equal(await this.countCompactContextSteps(), summaryCalls - 1);
+        assert.equal(snapshot.usage.modelCalls, modelRequests.length + summaryCalls - 1);
+    }
+}
+function numberedTags(count) {
+    return Array.from({ length: count }, (_, index) => `t${index + 1}`);
+}
+//# sourceMappingURL=context-window.js.map

package/dist/src/testing/context-window.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"context-window.js","sourceRoot":"","sources":["../../../src/testing/context-window.ts"],"names":[],"mappings":"AAAA,yCAAyC;AACzC,EAAE;AACF,kDAAkD;AAClD,mEAAmE;AACnE,+CAA+C;AAC/C,EAAE;AACF,0DAA0D;AAE1D,OAAO,MAAM,MAAM,oBAAoB,CAAC;AACxC,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,YAAY,EAAE,MAAM,SAAS,CAAC;AACnE,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AAEjC,OAAO,EACL,0BAA0B,EAC1B,wBAAwB,GAEzB,MAAM,gCAAgC,CAAC;AAGxC,OAAO,EAAE,qBAAqB,EAAE,MAAM,2BAA2B,CAAC;AAClE,OAAO,EAAE,QAAQ,EAAE,MAAM,mBAAmB,CAAC;AAC7C,OAAO,EAAE,SAAS,EAAE,MAAM,cAAc,CAAC;AACzC,OAAO,EAAE,qBAAqB,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AAE3E,8EAA8E;AAC9E,MAAM,CAAC,MAAM,6BAA6B,GAAG,oCAAoC,CAAC;AAClF,wDAAwD;AACxD,MAAM,CAAC,MAAM,qBAAqB,GAAG,QAAQ,CAAC;AAC9C,kFAAkF;AAClF,MAAM,CAAC,MAAM,0BAA0B,GAAG,kDAAkD,CAAC;AAC7F,2EAA2E;AAC3E,MAAM,CAAC,MAAM,4BAA4B,GAAG,SAAS,CAAC;AAEtD,MAAM,cAAc,GAAG,sBAAsB,CAAC;AAC9C,MAAM,cAAc,GAAG,cAAc,CAAC;AACtC,MAAM,kBAAkB,GAAG,wBAAwB,CAAC;AACpD,MAAM,0BAA0B,GAAG,CAAC,CAAC;AACrC,MAAM,eAAe,GAAG,EAAE,CAAC;AAuB3B,mDAAmD;AACnD,MAAM,CAAC,MAAM,4BAA4B,GAA0B,MAAM,CAAC,MAAM,CAAC;IAC/E,YAAY,EAAE,IAAI;IAClB,mBAAmB,EAAE,IAAI;IACzB,oBAAoB,EAAE,IAAI;IAC1B,gBAAgB,EAAE,IAAI;IACtB,iBAAiB,EAAE,CAAC;IACpB,gBAAgB,EAAE,CAAC;IACnB,oBAAoB,EAAE,IAAI;CAC3B,CAAC,CAAC;AAEH;;;;GAIG;AACH,MAAM,UAAU,gCAAgC,CAAC,QAA+B;IAC9E,OAAO,EAAE,CAAC,6BAA6B,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,EAAE,CAAC;AACvE,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,yBAAyB;IACvC,MAAM,KAAK,GAAG,OAAO,CAAC,GAAG,CAAC,6BAA6B,CAAC,CAAC;IACzD,OAAO,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,4BAA4B,CAAC,CAAC,CAAE,IAAI,CAAC,KAAK,CAAC,KAAK,CAA2B,CAAC;AAC3G,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,2BAA2B,CAAC,QAA+B;IAMzE,OAAO;QACL,GAAG,CAAC,QAAQ,CAAC,mBAAmB,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,EAAE,mBAAmB,EAAE,QAAQ,CAAC,YAAY,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QACvG,iBAAiB,EAAE,QAAQ,CAAC,iBAAiB;QAC7C,gBAAgB,EAAE,QAAQ,CAAC,gBAAgB;QAC3C,oBAAoB,EAAE,QAAQ,CAAC,oBAAoB;KACpD,CAAC;AACJ,CAAC;AAiBD;;;;GAIG;AACH,MAAM,OAAO,wBAAwB;IAMhB;IACA;IANnB;;;OAGG;IACH,YACmB,cAAsB,EACtB,QAA+B;QAD/B,mBAAc,GAAd,cAAc,CAAQ;QACtB,aAAQ,GAAR,QAAQ,CAAuB;IAC/C,CAAC;IAEJ;;;;OAIG;IACI,UAAU,CAAC,GAAW;QAC3B,OAAO,eAAe,GAAG,IAAI,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,oBAAoB,CAAC,EAAE,CAAC;IAChF,CAAC;IAED;;;;OAIG;IACI,UAAU,CAAC,GAAW;QAC3B,OAAO,eAAe,GAAG,IAAI,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,gBAAgB,CAAC,EAAE,CAAC;IAC5E,CAAC;IAED;;;;;OAKG;IACI,KAAK,CAAC,KAAK,CAAC,OAIlB;QACC,MAAM,YAAY,GAAG,OAAO,CAAC,YAAY,IAAI,EAAE,CAAC;QAChD,MAAM,SAAS,GAAG,YAAY,CAAC,QAAQ,CAAC,0BAA0B,CAAC,CAAC;QACpE,MAAM,UAAU,GAAG,OAAO,CAAC,SAAS,CAAC,MAAM,GAAG,YAAY,CAAC,MAAM,CAAC;QAClE,MAAM,KAAK,GAAyB,EAAE,IAAI,EAAE,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,OAAO,EAAE,UAAU,EAAE,KAAK,EAAE,OAAO,CAAC,SAAS,EAAE,CAAC;QACpH,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,cAAc,CAAC,EAAE,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QACxF,MAAM,gBAAgB,GAAG,IAAI,CAAC,QAAQ,CAAC,YAAY,GAAG,0BAA0B,CAAC;QACjF,IAAI,UAAU,GAAG,gBAAgB,EAAE,CAAC;YAClC,MAAM,IAAI,KAAK,CAAC,GAAG,0BAA0B,KAAK,UAAU,sBAAsB,gBAAgB,EAAE,CAAC,CAAC;QACxG,CAAC;QACD,MAAM,IAAI,GAAG,CAAC,GAAG,OAAO,CAAC,SAAS,CAAC,QAAQ,CAAC,kBAAkB,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;QAChG,IAAI,SAAS,EAAE,CAAC;YACd,IAAI,IAAI,CAAC,QAAQ,CAAC,cAAc,CAAC,EAAE,CAAC;gBAClC,MAAM,IAAI,GAAG,IAAI,QAAQ,CAAC,IAAI,CAAC,cAAc,EAAE,4BAA4B,CAAC,CAAC;gBAC7E,IAAI,CAAC,WAAW,EAAE,CAAC;gBACnB,MAAM,IAAI,CAAC,cAAc,EAAE,CAAC;YAC9B,CAAC;YACD,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,qBAAqB,CAAC,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,eAAe,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC;QAC7G,CAAC;QACD,MAAM,GAAG,GAAG,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;QACxB,IAAI,GAAG,KAAK,SAAS,EAAE,CAAC;YACtB,MAAM,IAAI,KAAK,CAAC,yCAAyC,CAAC,CAAC;QAC7D,CAAC;QACD,OAAO,OAAO,CAAC,aAAa,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,GAAG,EAAE,CAAC;IAC3G,CAAC;CACF;AAED;;;;GAIG;AACH,MAAM,UAAU,yBAAyB,CAAC,cAAsB;IAC9D,MAAM,IAAI,GAAG,IAAI,CAAC,cAAc,EAAE,cAAc,CAAC,CAAC;IAClD,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC;QACtB,OAAO,EAAE,CAAC;IACZ,CAAC;IACD,OAAO,YAAY,CAAC,IAAI,EAAE,MAAM,CAAC;SAC9B,KAAK,CAAC,IAAI,CAAC;SACX,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,KAAK,EAAE,CAAC;SAC7B,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAyB,CAAC,CAAC;AAC7D,CAAC;AAkBD;;;GAGG;AACH,MAAM,OAAO,4BAA4B;IAMH;IALpC,uCAAuC;IACvB,SAAS,GAAG,WAAW,UAAU,EAAE,EAAE,CAAC;IACrC,OAAO,GAA4B,EAAE,CAAC;IAEvD,6EAA6E;IAC7E,YAAoC,OAA4C;QAA5C,YAAO,GAAP,OAAO,CAAqC;IAAG,CAAC;IAEpF;;;;OAIG;IACI,KAAK,CAAC,WAAW,CAAC,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ;QACvD,MAAM,MAAM,GAAG,MAAM,qBAAqB,CAAC,KAAK,CAAC;YAC/C,eAAe,EAAE,IAAI,CAAC,OAAO,CAAC,qBAAqB;YACnD,aAAa,EAAE,IAAI,CAAC,OAAO,CAAC,aAAa;YACzC,OAAO,EAAE,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,cAAc,EAAE,YAAY,CAAC;YACxD,WAAW,EAAE,EAAE,GAAG,IAAI,CAAC,OAAO,CAAC,iBAAiB,EAAE,GAAG,gCAAgC,CAAC,QAAQ,CAAC,EAAE;SAClG,CAAC,CAAC;QACH,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC1B,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,qDAAqD;IAC9C,KAAK,CAAC,KAAK;QAChB,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC;YAC5C,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;QACtB,CAAC;IACH,CAAC;IAED;;;;OAIG;IACI,QAAQ,CAAC,IAAkC;QAChD,OAAO,yBAAyB,CAAC,IAAI,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,IAAI,KAAK,IAAI,CAAC,CAAC;IAC3G,CAAC;IAED;;;;;OAKG;IACI,KAAK,CAAC,QAAQ,CAAC,IAAuB,EAAE,OAAO,GAAG,IAAI;QAC3D,MAAM,EAAE,QAAQ,EAAE,GAAG,IAAI,CAAC,OAAO,CAAC;QAClC,IAAI,QAAQ,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,QAAQ,GAAG,EAAE,CAAC,CAAC;QAChD,IAAI,MAAM,GAAG,CAAC,CAAC;QACf,IAAI,OAAO,EAAE,CAAC;YACZ,MAAM,QAAQ,CAAC,YAAY,CAAC,IAAI,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;YAC/D,QAAQ,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAC/B,CAAC;aAAM,CAAC;YACN,MAAM,GAAG,CAAC,MAAM,QAAQ,CAAC,cAAc,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,EAAE,SAAS,IAAI,CAAC,CAAC;QAC3E,CAAC;QACD,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;YAC/B,MAAM,CAAC,KAAK,CAAC,MAAM,QAAQ,CAAC,eAAe,CAAC,IAAI,CAAC,SAAS,EAAE,OAAO,CAAC,EAAE,IAAI,CAAC,CAAC;QAC9E,CAAC;QACD,MAAM,QAAQ,CAAC,qBAAqB,CAAC,IAAI,CAAC,SAAS,EAAE,MAAM,GAAG,IAAI,CAAC,MAAM,EAAE,EAAE,SAAS,EAAE,OAAO,EAAE,CAAC,CAAC;QACnG,OAAO,QAAQ,CAAC,gBAAgB,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IACnD,CAAC;IAED;;;OAGG;IACI,KAAK,CAAC,kBAAkB;QAC7B,MAAM,QAAQ,GAAmB,EAAE,CAAC;QACpC,IAAI,cAAc,GAAkB,IAAI,CAAC;QACzC,SAAS,CAAC;YACR,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,iBAAiB,CAAC,IAAI,CAAC,SAAS,EAAE,cAAc,EAAE,GAAG,CAAC,CAAC;YAChG,QAAQ,CAAC,OAAO,CAAC,GAAG,IAAI,CAAC,QAAQ,CAAC,CAAC;YACnC,IAAI,IAAI,CAAC,kBAAkB,KAAK,IAAI,EAAE,CAAC;gBACrC,OAAO,QAAQ,CAAC;YAClB,CAAC;YACD,cAAc,GAAG,IAAI,CAAC,kBAAkB,CAAC;QAC3C,CAAC;IACH,CAAC;IAED;;;OAGG;IACI,KAAK,CAAC,wBAAwB;QACnC,MAAM,OAAO,GAAG,MAAM,eAAe,CAAC,IAAI,CAAC,OAAO,CAAC,aAAa,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;QAClF,OAAO,OAAO,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,QAAQ,KAAK,gBAAgB,IAAI,KAAK,CAAC,SAAS,KAAK,qBAAqB,CAAC;aAC/G,MAAM,CAAC;IACZ,CAAC;IAED;;;;OAIG;IACI,KAAK,CAAC,uBAAuB,CAAC,SAAS,GAAG,EAAE;QACjD,MAAM,IAAI,CAAC,WAAW,EAAE,CAAC;QACzB,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC,CAAC;QAE9D,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,KAAK,EAAE,aAAa,IAAI,IAAI,EAAE,IAAI,CAAC,CAAC;QAC1D,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,KAAK,EAAE,MAAM,EAAE,0BAA0B,CAAC,CAAC;QACjE,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,KAAK,EAAE,SAAS,EAAE,SAAS,CAAC,CAAC;QACnD,MAAM,OAAO,GAAG,QAAQ,CAAC,cAAc,CAAC;QACxC,MAAM,CAAC,EAAE,CAAC,OAAO,CAAC,CAAC;QACnB,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,EAAE,qBAAqB,CAAC,CAAC;QAClD,MAAM,CAAC,EAAE,CAAC,CAAC,MAAM,IAAI,CAAC,wBAAwB,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC;QAExD,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,YAAY,GAAG,0BAA0B,CAAC;QAC9E,MAAM,aAAa,GAAG,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;QAC7C,MAAM,CAAC,EAAE,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,UAAU,IAAI,KAAK,CAAC,CAAC,CAAC;QACzE,MAAM,WAAW,GAAG,yBAAyB,CAAC,IAAI,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC;QAC3E,WAAW,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,KAAK,EAAE,EAAE;YACrC,IAAI,OAAO,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;gBAC/B,MAAM,CAAC,EAAE,CAAC,CAAC,WAAW,CAAC,KAAK,GAAG,CAAC,CAAC,EAAE,UAAU,IAAI,CAAC,CAAC,GAAG,CAAC,WAAW,CAAC,KAAK,GAAG,CAAC,CAAC,EAAE,UAAU,IAAI,CAAC,CAAC,CAAC,CAAC;YACnG,CAAC;QACH,CAAC,CAAC,CAAC;QACH,MAAM,SAAS,GAAG,aAAa,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,KAAK,IAAI,EAAE,CAAC;QACpD,MAAM,CAAC,EAAE,CAAC,SAAS,CAAC,QAAQ,CAAC,wBAAwB,CAAC,CAAC,CAAC;QACxD,MAAM,CAAC,EAAE,CAAC,SAAS,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;QACjE,oEAAoE;QACpE,MAAM,CAAC,EAAE,CAAC,CAAC,SAAS,CAAC,QAAQ,CAAC,iBAAiB,CAAC,CAAC,CAAC;QAClD,MAAM,CAAC,EAAE,CAAC,CAAC,SAAS,CAAC,QAAQ,CAAC,iBAAiB,CAAC,CAAC,CAAC;QAElD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,kBAAkB,EAAE,CAAC;QACjD,MAAM,CAAC,SAAS,CACd,QAAQ,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,QAAQ,CAAC,EAC3C,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE,CAAC,KAAK,GAAG,CAAC,CAAC,CACtC,CAAC;QACF,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,SAAS,CAAC,CAAC;QAC3C,MAAM,CAAC,EAAE,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,aAAa,CAAC,QAAQ,CAAC,wBAAwB,CAAC,CAAC,CAAC,CAAC;QACjG,MAAM,CAAC,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC,QAAQ,CAAC,iBAAiB,CAAC,CAAC,CAAC,CAAC;QACxF,MAAM,YAAY,GAAG,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,MAAM,CAAC;QACrD,MAAM,CAAC,KAAK,CAAC,YAAY,EAAE,MAAM,IAAI,CAAC,wBAAwB,EAAE,CAAC,CAAC;QAClE,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,UAAU,EAAE,aAAa,CAAC,MAAM,GAAG,YAAY,CAAC,CAAC;QAC7E,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED;;;OAGG;IACI,KAAK,CAAC,gCAAgC,CAAC,SAAS,GAAG,EAAE;QAC1D,MAAM,IAAI,CAAC,WAAW,EAAE,CAAC;QACzB,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC,CAAC;QAC9D,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,KAAK,EAAE,aAAa,IAAI,IAAI,EAAE,IAAI,CAAC,CAAC;QAC1D,MAAM,OAAO,GAAG,QAAQ,CAAC,cAAc,CAAC;QACxC,MAAM,CAAC,EAAE,CAAC,OAAO,CAAC,CAAC;QACnB,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,iBAAiB,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,EAAE,GAAG,CAAC,CAAC;QACtF,MAAM,aAAa,GAAG,IAAI,CAAC,qBAAqB,IAAI,IAAI,CAAC;QACzD,MAAM,CAAC,EAAE,CAAC,aAAa,KAAK,IAAI,IAAI,aAAa,GAAG,CAAC,CAAC,CAAC;QACvD,MAAM,CAAC,EAAE,CAAC,aAAa,GAAG,CAAC,IAAI,OAAO,CAAC,YAAY,CAAC,CAAC;QACrD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,kBAAkB,EAAE,CAAC;QACjD,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,aAAa,CAAC,CAAC;QACnD,MAAM,CAAC,SAAS,CACd,QAAQ,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,QAAQ,CAAC,EAC3C,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE,CAAC,aAAa,GAAG,KAAK,CAAC,CAClD,CAAC;QACF,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,oBAAoB,CAAC;QACzD,MAAM,CAAC,EAAE,CAAC,KAAK,KAAK,IAAI,IAAI,QAAQ,CAAC,MAAM,IAAI,KAAK,CAAC,CAAC;QACtD,MAAM,CAAC,EAAE,CAAC,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,IAAI,EAAE,CAAC,CAAC,UAAU,CAAC,gBAAgB,SAAS,GAAG,CAAC,CAAC,CAAC;IACpF,CAAC;IAED;;;OAGG;IACI,KAAK,CAAC,oCAAoC,CAAC,SAAS,GAAG,EAAE;QAC9D,MAAM,IAAI,CAAC,WAAW,EAAE,CAAC;QACzB,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC,CAAC;QAC9D,MAAM,CAAC,EAAE,CAAC,CAAC,QAAQ,CAAC,KAAK,EAAE,aAAa,IAAI,EAAE,CAAC,CAAC,QAAQ,CAAC,0BAA0B,CAAC,CAAC,CAAC;QACtF,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,cAAc,IAAI,IAAI,EAAE,IAAI,CAAC,CAAC;QACpD,MAAM,CAAC,KAAK,CAAC,MAAM,IAAI,CAAC,wBAAwB,EAAE,EAAE,CAAC,CAAC,CAAC;QACvD,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QACjD,MAAM,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,KAAK,CAAC,QAAQ,CAAC,wBAAwB,CAAC,CAAC,CAAC,CAAC;IACzG,CAAC;IAED,qFAAqF;IAC9E,KAAK,CAAC,oCAAoC;QAC/C,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,EAAE,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,mBAAmB,EAAE,KAAK,EAAE,CAAC,CAAC;QAChG,MAAM,IAAI,CAAC,QAAQ,CAAC,CAAC,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC,CAAC;QACxC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QACjD,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;QAEpB,MAAM,IAAI,CAAC,WAAW,EAAE,CAAC;QACzB,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;QACvE,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,KAAK,EAAE,aAAa,IAAI,IAAI,EAAE,IAAI,CAAC,CAAC;QAC1D,MAAM,CAAC,EAAE,CAAC,QAAQ,CAAC,cAAc,CAAC,CAAC;QACnC,MAAM,CAAC,KAAK,CAAC,CAAC,MAAM,IAAI,CAAC,kBAAkB,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,SAAS,CAAC,CAAC;IACtE,CAAC;IAED,iHAAiH;IAC1G,KAAK,CAAC,oCAAoC;QAC/C,MAAM,EAAE,QAAQ,EAAE,cAAc,EAAE,GAAG,IAAI,CAAC,OAAO,CAAC;QAClD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,WAAW,EAAE,CAAC;QACxC,MAAM,QAAQ,GAAG,CAAC,cAAc,EAAE,GAAG,YAAY,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,QAAQ,GAAG,EAAE,CAAC,CAAC;QAC5F,MAAM,QAAQ,CAAC,YAAY,CAAC,IAAI,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;QAC/D,KAAK,MAAM,OAAO,IAAI,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC;YACxC,MAAM,CAAC,KAAK,CAAC,MAAM,QAAQ,CAAC,eAAe,CAAC,IAAI,CAAC,SAAS,EAAE,OAAO,CAAC,EAAE,IAAI,CAAC,CAAC;QAC9E,CAAC;QACD,MAAM,IAAI,GAAG,IAAI,QAAQ,CAAC,cAAc,EAAE,4BAA4B,CAAC,CAAC;QACxE,MAAM,IAAI,CAAC,gBAAgB,EAAE,CAAC;QAC9B,MAAM,UAAU,GAAG,MAAM,SAAS,CAAC,KAAK,IAAI,EAAE;YAC5C,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,gBAAgB,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YACjE,OAAO,QAAQ,CAAC,KAAK,EAAE,MAAM,KAAK,oBAAoB,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC;QAChF,CAAC,EAAE,EAAE,WAAW,EAAE,wBAAwB,EAAE,CAAC,CAAC;QAC9C,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,cAAc,IAAI,IAAI,EAAE,IAAI,CAAC,CAAC;QACtD,MAAM,oBAAoB,GAAG,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,MAAM,CAAC;QAE3D,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;QACpB,IAAI,CAAC,OAAO,EAAE,CAAC;QACf,MAAM,IAAI,CAAC,WAAW,EAAE,CAAC;QACzB,MAAM,QAAQ,CAAC,qBAAqB,CAAC,IAAI,CAAC,SAAS,EAAE,QAAQ,CAAC,MAAM,EAAE,EAAE,SAAS,EAAE,OAAO,EAAE,CAAC,CAAC;QAC9F,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,gBAAgB,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAEjE,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,YAAY,EAAE,EAAE,CAAC,CAAC,CAAC;QACrC,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,KAAK,EAAE,aAAa,IAAI,IAAI,EAAE,IAAI,CAAC,CAAC;QAC1D,MAAM,CAAC,EAAE,CAAC,QAAQ,CAAC,cAAc,CAAC,CAAC;QACnC,MAAM,aAAa,GAAG,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;QAC7C,uFAAuF;QACvF,MAAM,CAAC,EAAE,CAAC,oBAAoB,GAAG,aAAa,CAAC,MAAM,CAAC,CAAC;QACvD,MAAM,YAAY,GAAG,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,MAAM,CAAC;QACrD,MAAM,CAAC,KAAK,CAAC,MAAM,IAAI,CAAC,wBAAwB,EAAE,EAAE,YAAY,GAAG,CAAC,CAAC,CAAC;QACtE,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,UAAU,EAAE,aAAa,CAAC,MAAM,GAAG,YAAY,GAAG,CAAC,CAAC,CAAC;IACnF,CAAC;CACF;AAED,SAAS,YAAY,CAAC,KAAa;IACjC,OAAO,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE,CAAC,IAAI,KAAK,GAAG,CAAC,EAAE,CAAC,CAAC;AACtE,CAAC"}

package/dist/src/testing/index.d.ts

@@ -1,6 +1,7 @@
 export { AgentSessionTestClient, type AgentSessionWaitOptions, } from "./agent-session-test-client.js";
 export { AgentTestEnvironment, type AgentTestEnvironmentOptions, } from "./agent-test-environment.js";
 export { KillableWorkerProcess, agentWorkerBlobCacheDirectoryVariable, agentWorkerReadyLine, agentWorkerServerAddressVariable, runAgentWorkerProcess, type KillableWorkerProcessOptions, } from "./agent-worker-process.js";
+export { ContextWindowModelScript, ContextWindowSessionScenario, contextWindowExceededError, contextWindowSessionOptions, contextWindowSettingsEnvironment, contextWindowSettingsVariable, contextWindowSummaryGateName, contextWindowToolName, defaultContextWindowSettings, readContextWindowRequests, readContextWindowSettings, type ContextWindowReply, type ContextWindowRequest, type ContextWindowSessionScenarioOptions, type ContextWindowSettings, } from "./context-window.js";
 export { FileGate, InvocationLog } from "./coordination.js";
 export { pollUntil, type PollOptions } from "./polling.js";
 export { ScriptedModelClient, scriptedModelApi, scriptedModelProvider, type ScriptedModelReply, type ScriptedModelResponder, type ScriptedToolCall, } from "./scripted-model-client.js";

package/dist/src/testing/index.js

@@ -8,6 +8,7 @@
 export { AgentSessionTestClient, } from "./agent-session-test-client.js";
 export { AgentTestEnvironment, } from "./agent-test-environment.js";
 export { KillableWorkerProcess, agentWorkerBlobCacheDirectoryVariable, agentWorkerReadyLine, agentWorkerServerAddressVariable, runAgentWorkerProcess, } from "./agent-worker-process.js";
+export { ContextWindowModelScript, ContextWindowSessionScenario, contextWindowExceededError, contextWindowSessionOptions, contextWindowSettingsEnvironment, contextWindowSettingsVariable, contextWindowSummaryGateName, contextWindowToolName, defaultContextWindowSettings, readContextWindowRequests, readContextWindowSettings, } from "./context-window.js";
 export { FileGate, InvocationLog } from "./coordination.js";
 export { pollUntil } from "./polling.js";
 export { ScriptedModelClient, scriptedModelApi, scriptedModelProvider, } from "./scripted-model-client.js";

package/dist/src/testing/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/testing/index.ts"],"names":[],"mappings":"AAAA,yCAAyC;AACzC,EAAE;AACF,kDAAkD;AAClD,mEAAmE;AACnE,+CAA+C;AAC/C,EAAE;AACF,0DAA0D;AAE1D,OAAO,EACL,sBAAsB,GAEvB,MAAM,gCAAgC,CAAC;AACxC,OAAO,EACL,oBAAoB,GAErB,MAAM,6BAA6B,CAAC;AACrC,OAAO,EACL,qBAAqB,EACrB,qCAAqC,EACrC,oBAAoB,EACpB,gCAAgC,EAChC,qBAAqB,GAEtB,MAAM,2BAA2B,CAAC;AACnC,OAAO,EAAE,QAAQ,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AAC5D,OAAO,EAAE,SAAS,EAAoB,MAAM,cAAc,CAAC;AAC3D,OAAO,EACL,mBAAmB,EACnB,gBAAgB,EAChB,qBAAqB,GAItB,MAAM,4BAA4B,CAAC;AACpC,OAAO,EACL,qBAAqB,EACrB,eAAe,EACf,qBAAqB,GAGtB,MAAM,mBAAmB,CAAC;AAC3B,OAAO,EACL,gBAAgB,EAChB,oBAAoB,EACpB,qBAAqB,GAEtB,MAAM,yBAAyB,CAAC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/testing/index.ts"],"names":[],"mappings":"AAAA,yCAAyC;AACzC,EAAE;AACF,kDAAkD;AAClD,mEAAmE;AACnE,+CAA+C;AAC/C,EAAE;AACF,0DAA0D;AAE1D,OAAO,EACL,sBAAsB,GAEvB,MAAM,gCAAgC,CAAC;AACxC,OAAO,EACL,oBAAoB,GAErB,MAAM,6BAA6B,CAAC;AACrC,OAAO,EACL,qBAAqB,EACrB,qCAAqC,EACrC,oBAAoB,EACpB,gCAAgC,EAChC,qBAAqB,GAEtB,MAAM,2BAA2B,CAAC;AACnC,OAAO,EACL,wBAAwB,EACxB,4BAA4B,EAC5B,0BAA0B,EAC1B,2BAA2B,EAC3B,gCAAgC,EAChC,6BAA6B,EAC7B,4BAA4B,EAC5B,qBAAqB,EACrB,4BAA4B,EAC5B,yBAAyB,EACzB,yBAAyB,GAK1B,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAAE,QAAQ,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AAC5D,OAAO,EAAE,SAAS,EAAoB,MAAM,cAAc,CAAC;AAC3D,OAAO,EACL,mBAAmB,EACnB,gBAAgB,EAChB,qBAAqB,GAItB,MAAM,4BAA4B,CAAC;AACpC,OAAO,EACL,qBAAqB,EACrB,eAAe,EACf,qBAAqB,GAGtB,MAAM,mBAAmB,CAAC;AAC3B,OAAO,EACL,gBAAgB,EAChB,oBAAoB,EACpB,qBAAqB,GAEtB,MAAM,yBAAyB,CAAC"}

package/dist/src/ui-stream/agent-event-sse.d.ts

@@ -63,8 +63,9 @@ export declare function formatAgentEventSseFrame(message: StreamMessage<AgentEve
  * Each event becomes one frame whose `id` is its Dex resume token and whose data is the `AgentEvent` wire
  * JSON: `type`, `turn_id`, `attempt`, `step_execution_id`, and `payload` text. Reconnecting with the last
  * `id` continues with the next event, with no repeated or missing event while the Stream retains them.
- * After each empty long poll it yields a keep-alive comment. The stream never ends on its own; stop
- * iterating, or abort `signal`, when the browser goes away.
+ * After each empty long poll it yields a keep-alive comment. The stream ends on its own only once the
+ * session's Flow failed and every event was sent; otherwise stop iterating, or abort `signal`, when the
+ * browser goes away.
  *
  * The browser applies the event rules of the session contract. Within one `step_execution_id` it keeps
  * only the highest `attempt` and drops the partial text of lower attempts. After `turn_completed`,