muonroi-cli 1.2.0

package/dist/orchestrator/abort.js

@@ -0,0 +1,31 @@
+/**
+ * AbortContext — single-owner AbortController wrapper for the orchestrator.
+ *
+ * Rules:
+ * - One AbortContext per user message turn (created fresh by the orchestrator).
+ * - The signal is threaded into streamText, tool runner, and EE intercept calls.
+ * - abort() is idempotent — safe to call from multiple signal paths (SIGINT,
+ *   EE timeout, etc.). First call wins; subsequent calls are no-ops.
+ *
+ * Mitigates: Pitfall 2 (orchestrator-level abort safely terminates all I/O).
+ * References: TUI-04, 00-CONTEXT.md decision — orchestrator owns controller.
+ */
+/**
+ * Creates a fresh AbortContext.  Call once per user message turn.
+ */
+export function createAbortContext() {
+    const controller = new AbortController();
+    let _reason;
+    return {
+        signal: controller.signal,
+        abort(reason) {
+            if (controller.signal.aborted)
+                return; // idempotent
+            _reason = reason;
+            controller.abort(reason);
+        },
+        isAborted: () => controller.signal.aborted,
+        reason: () => _reason,
+    };
+}
+//# sourceMappingURL=abort.js.map

package/dist/orchestrator/abort.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"abort.js","sourceRoot":"","sources":["../../src/orchestrator/abort.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;GAWG;AAmBH;;GAEG;AACH,MAAM,UAAU,kBAAkB;IAChC,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;IACzC,IAAI,OAA2B,CAAC;IAEhC,OAAO;QACL,MAAM,EAAE,UAAU,CAAC,MAAM;QAEzB,KAAK,CAAC,MAAe;YACnB,IAAI,UAAU,CAAC,MAAM,CAAC,OAAO;gBAAE,OAAO,CAAC,aAAa;YACpD,OAAO,GAAG,MAAM,CAAC;YACjB,UAAU,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;QAC3B,CAAC;QAED,SAAS,EAAE,GAAG,EAAE,CAAC,UAAU,CAAC,MAAM,CAAC,OAAO;QAE1C,MAAM,EAAE,GAAG,EAAE,CAAC,OAAO;KACtB,CAAC;AACJ,CAAC"}

package/dist/orchestrator/abort.test.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/orchestrator/abort.test.js

@@ -0,0 +1,34 @@
+import { describe, expect, it } from "vitest";
+import { createAbortContext } from "./abort.js";
+describe("AbortContext", () => {
+    it("Test 1: initial state — not aborted", () => {
+        const ctx = createAbortContext();
+        expect(ctx.signal.aborted).toBe(false);
+        expect(ctx.isAborted()).toBe(false);
+        expect(ctx.reason()).toBeUndefined();
+    });
+    it("Test 1: after abort — signal is aborted and reason is captured", () => {
+        const ctx = createAbortContext();
+        ctx.abort("test");
+        expect(ctx.signal.aborted).toBe(true);
+        expect(ctx.isAborted()).toBe(true);
+        expect(ctx.reason()).toBe("test");
+    });
+    it("Test 2: abort is idempotent — second call does not throw", () => {
+        const ctx = createAbortContext();
+        expect(() => {
+            ctx.abort("first");
+            ctx.abort("second");
+        }).not.toThrow();
+        // Reason captured from first call only
+        expect(ctx.reason()).toBe("first");
+    });
+    it("Test 2: abort with no reason is idempotent", () => {
+        const ctx = createAbortContext();
+        ctx.abort();
+        ctx.abort("later");
+        expect(ctx.isAborted()).toBe(true);
+        expect(ctx.reason()).toBeUndefined();
+    });
+});
+//# sourceMappingURL=abort.test.js.map

package/dist/orchestrator/abort.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"abort.test.js","sourceRoot":"","sources":["../../src/orchestrator/abort.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,kBAAkB,EAAE,MAAM,YAAY,CAAC;AAEhD,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;IAC5B,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;QAC7C,MAAM,GAAG,GAAG,kBAAkB,EAAE,CAAC;QACjC,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACvC,MAAM,CAAC,GAAG,CAAC,SAAS,EAAE,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACpC,MAAM,CAAC,GAAG,CAAC,MAAM,EAAE,CAAC,CAAC,aAAa,EAAE,CAAC;IACvC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gEAAgE,EAAE,GAAG,EAAE;QACxE,MAAM,GAAG,GAAG,kBAAkB,EAAE,CAAC;QACjC,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;QAClB,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACtC,MAAM,CAAC,GAAG,CAAC,SAAS,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACnC,MAAM,CAAC,GAAG,CAAC,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IACpC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0DAA0D,EAAE,GAAG,EAAE;QAClE,MAAM,GAAG,GAAG,kBAAkB,EAAE,CAAC;QACjC,MAAM,CAAC,GAAG,EAAE;YACV,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YACnB,GAAG,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;QACtB,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;QACjB,uCAAuC;QACvC,MAAM,CAAC,GAAG,CAAC,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACrC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;QACpD,MAAM,GAAG,GAAG,kBAAkB,EAAE,CAAC;QACjC,GAAG,CAAC,KAAK,EAAE,CAAC;QACZ,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QACnB,MAAM,CAAC,GAAG,CAAC,SAAS,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACnC,MAAM,CAAC,GAAG,CAAC,MAAM,EAAE,CAAC,CAAC,aAAa,EAAE,CAAC;IACvC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/orchestrator/agent.test.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/orchestrator/agent.test.js

@@ -0,0 +1,126 @@
+import { afterEach, describe, expect, it, vi } from "vitest";
+async function importAgentModule() {
+    vi.resetModules();
+    vi.doMock("../storage/index", () => ({
+        appendCompaction: vi.fn(),
+        appendMessages: vi.fn(() => []),
+        appendSystemMessage: vi.fn(() => 0),
+        buildChatEntries: vi.fn(() => []),
+        getNextMessageSequence: vi.fn(() => 0),
+        getSessionTotalTokens: vi.fn(() => 0),
+        loadTranscript: vi.fn(() => []),
+        loadTranscriptState: vi.fn(() => ({ messages: [], seqs: [] })),
+        recordUsageEvent: vi.fn(),
+        SessionStore: class {
+            getWorkspace() {
+                return null;
+            }
+            openSession() {
+                return null;
+            }
+            createSession() {
+                return null;
+            }
+            setModel() { }
+            getRequiredSession() {
+                return null;
+            }
+            setMode() { }
+            touchSession() { }
+        },
+    }));
+    return import("./orchestrator");
+}
+afterEach(() => {
+    vi.restoreAllMocks();
+    vi.resetModules();
+    vi.doUnmock("../storage/index");
+});
+describe("Agent class", { timeout: 15_000 }, () => {
+    it("constructs with default options", async () => {
+        const { Agent } = await importAgentModule();
+        const agent = new Agent(undefined, undefined, undefined, undefined, {
+            persistSession: false,
+        });
+        expect(agent).toBeDefined();
+        expect(agent.getMode()).toBe("agent");
+    });
+    it("can switch mode", async () => {
+        const { Agent } = await importAgentModule();
+        const agent = new Agent(undefined, undefined, undefined, undefined, {
+            persistSession: false,
+        });
+        agent.setMode("plan");
+        expect(agent.getMode()).toBe("plan");
+        agent.setMode("ask");
+        expect(agent.getMode()).toBe("ask");
+        agent.setMode("agent");
+        expect(agent.getMode()).toBe("agent");
+    });
+    it("returns a model string", async () => {
+        const { Agent } = await importAgentModule();
+        const agent = new Agent(undefined, undefined, undefined, undefined, {
+            persistSession: false,
+        });
+        const model = agent.getModel();
+        expect(typeof model).toBe("string");
+        expect(model.length).toBeGreaterThan(0);
+    });
+    it("can set model", async () => {
+        const { Agent } = await importAgentModule();
+        const agent = new Agent(undefined, undefined, undefined, undefined, {
+            persistSession: false,
+        });
+        agent.setModel("claude-sonnet-4-6-20250514");
+        expect(agent.getModel()).toBe("claude-sonnet-4-6-20250514");
+    });
+    it("constructs with sandbox mode", async () => {
+        const { Agent } = await importAgentModule();
+        const agent = new Agent(undefined, undefined, undefined, undefined, {
+            persistSession: false,
+            sandboxMode: "shuru",
+        });
+        expect(agent.getSandboxMode()).toBe("shuru");
+    });
+    it("defaults sandbox mode to off", async () => {
+        const { Agent } = await importAgentModule();
+        const agent = new Agent(undefined, undefined, undefined, undefined, {
+            persistSession: false,
+        });
+        expect(agent.getSandboxMode()).toBe("off");
+    });
+    it("constructs with sandbox settings", async () => {
+        const { Agent } = await importAgentModule();
+        const settings = { allowNet: true, cpus: 4 };
+        const agent = new Agent(undefined, undefined, undefined, undefined, {
+            persistSession: false,
+            sandboxMode: "shuru",
+            sandboxSettings: settings,
+        });
+        expect(agent.getSandboxSettings()).toEqual(settings);
+    });
+    it("respects MUONROI_MAX_TOKENS env var", async () => {
+        vi.stubEnv("MUONROI_MAX_TOKENS", "32768");
+        const { Agent } = await importAgentModule();
+        const agent = new Agent(undefined, undefined, undefined, undefined, {
+            persistSession: false,
+        });
+        expect(agent).toBeDefined();
+    });
+    it("constructs with permission mode", async () => {
+        const { Agent } = await importAgentModule();
+        const agent = new Agent(undefined, undefined, undefined, undefined, {
+            persistSession: false,
+            permissionMode: "yolo",
+        });
+        expect(agent).toBeDefined();
+    });
+    it("constructs with explicit model parameter", async () => {
+        const { Agent } = await importAgentModule();
+        const agent = new Agent(undefined, undefined, "claude-sonnet-4-6-20250514", undefined, {
+            persistSession: false,
+        });
+        expect(agent.getModel()).toBe("claude-sonnet-4-6-20250514");
+    });
+});
+//# sourceMappingURL=agent.test.js.map

package/dist/orchestrator/agent.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"agent.test.js","sourceRoot":"","sources":["../../src/orchestrator/agent.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAE7D,KAAK,UAAU,iBAAiB;IAC9B,EAAE,CAAC,YAAY,EAAE,CAAC;IAClB,EAAE,CAAC,MAAM,CAAC,kBAAkB,EAAE,GAAG,EAAE,CAAC,CAAC;QACnC,gBAAgB,EAAE,EAAE,CAAC,EAAE,EAAE;QACzB,cAAc,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC;QAC/B,mBAAmB,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC;QACnC,gBAAgB,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC;QACjC,sBAAsB,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC;QACtC,qBAAqB,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC;QACrC,cAAc,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC;QAC/B,mBAAmB,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,QAAQ,EAAE,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC;QAC9D,gBAAgB,EAAE,EAAE,CAAC,EAAE,EAAE;QACzB,YAAY,EAAE;YACZ,YAAY;gBACV,OAAO,IAAI,CAAC;YACd,CAAC;YACD,WAAW;gBACT,OAAO,IAAI,CAAC;YACd,CAAC;YACD,aAAa;gBACX,OAAO,IAAI,CAAC;YACd,CAAC;YACD,QAAQ,KAAI,CAAC;YACb,kBAAkB;gBAChB,OAAO,IAAI,CAAC;YACd,CAAC;YACD,OAAO,KAAI,CAAC;YACZ,YAAY,KAAI,CAAC;SAClB;KACF,CAAC,CAAC,CAAC;IAEJ,OAAO,MAAM,CAAC,gBAAgB,CAAC,CAAC;AAClC,CAAC;AAED,SAAS,CAAC,GAAG,EAAE;IACb,EAAE,CAAC,eAAe,EAAE,CAAC;IACrB,EAAE,CAAC,YAAY,EAAE,CAAC;IAClB,EAAE,CAAC,QAAQ,CAAC,kBAAkB,CAAC,CAAC;AAClC,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,aAAa,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE,EAAE,GAAG,EAAE;IAChD,EAAE,CAAC,iCAAiC,EAAE,KAAK,IAAI,EAAE;QAC/C,MAAM,EAAE,KAAK,EAAE,GAAG,MAAM,iBAAiB,EAAE,CAAC;QAC5C,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE;YAClE,cAAc,EAAE,KAAK;SACtB,CAAC,CAAC;QACH,MAAM,CAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC;QAC5B,MAAM,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACxC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iBAAiB,EAAE,KAAK,IAAI,EAAE;QAC/B,MAAM,EAAE,KAAK,EAAE,GAAG,MAAM,iBAAiB,EAAE,CAAC;QAC5C,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE;YAClE,cAAc,EAAE,KAAK;SACtB,CAAC,CAAC;QACH,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QACtB,MAAM,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACrC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QACrB,MAAM,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACpC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QACvB,MAAM,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACxC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wBAAwB,EAAE,KAAK,IAAI,EAAE;QACtC,MAAM,EAAE,KAAK,EAAE,GAAG,MAAM,iBAAiB,EAAE,CAAC;QAC5C,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE;YAClE,cAAc,EAAE,KAAK;SACtB,CAAC,CAAC;QACH,MAAM,KAAK,GAAG,KAAK,CAAC,QAAQ,EAAE,CAAC;QAC/B,MAAM,CAAC,OAAO,KAAK,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACpC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;IAC1C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,eAAe,EAAE,KAAK,IAAI,EAAE;QAC7B,MAAM,EAAE,KAAK,EAAE,GAAG,MAAM,iBAAiB,EAAE,CAAC;QAC5C,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE;YAClE,cAAc,EAAE,KAAK;SACtB,CAAC,CAAC;QACH,KAAK,CAAC,QAAQ,CAAC,4BAA4B,CAAC,CAAC;QAC7C,MAAM,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC,4BAA4B,CAAC,CAAC;IAC9D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8BAA8B,EAAE,KAAK,IAAI,EAAE;QAC5C,MAAM,EAAE,KAAK,EAAE,GAAG,MAAM,iBAAiB,EAAE,CAAC;QAC5C,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE;YAClE,cAAc,EAAE,KAAK;YACrB,WAAW,EAAE,OAAO;SACrB,CAAC,CAAC;QACH,MAAM,CAAC,KAAK,CAAC,cAAc,EAAE,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAC/C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8BAA8B,EAAE,KAAK,IAAI,EAAE;QAC5C,MAAM,EAAE,KAAK,EAAE,GAAG,MAAM,iBAAiB,EAAE,CAAC;QAC5C,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE;YAClE,cAAc,EAAE,KAAK;SACtB,CAAC,CAAC;QACH,MAAM,CAAC,KAAK,CAAC,cAAc,EAAE,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAC7C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kCAAkC,EAAE,KAAK,IAAI,EAAE;QAChD,MAAM,EAAE,KAAK,EAAE,GAAG,MAAM,iBAAiB,EAAE,CAAC;QAC5C,MAAM,QAAQ,GAAG,EAAE,QAAQ,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC;QAC7C,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE;YAClE,cAAc,EAAE,KAAK;YACrB,WAAW,EAAE,OAAO;YACpB,eAAe,EAAE,QAAQ;SAC1B,CAAC,CAAC;QACH,MAAM,CAAC,KAAK,CAAC,kBAAkB,EAAE,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;IACvD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qCAAqC,EAAE,KAAK,IAAI,EAAE;QACnD,EAAE,CAAC,OAAO,CAAC,oBAAoB,EAAE,OAAO,CAAC,CAAC;QAC1C,MAAM,EAAE,KAAK,EAAE,GAAG,MAAM,iBAAiB,EAAE,CAAC;QAC5C,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE;YAClE,cAAc,EAAE,KAAK;SACtB,CAAC,CAAC;QACH,MAAM,CAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC;IAC9B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iCAAiC,EAAE,KAAK,IAAI,EAAE;QAC/C,MAAM,EAAE,KAAK,EAAE,GAAG,MAAM,iBAAiB,EAAE,CAAC;QAC5C,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE;YAClE,cAAc,EAAE,KAAK;YACrB,cAAc,EAAE,MAAM;SACvB,CAAC,CAAC;QACH,MAAM,CAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC;IAC9B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;QACxD,MAAM,EAAE,KAAK,EAAE,GAAG,MAAM,iBAAiB,EAAE,CAAC;QAC5C,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,SAAS,EAAE,SAAS,EAAE,4BAA4B,EAAE,SAAS,EAAE;YACrF,cAAc,EAAE,KAAK;SACtB,CAAC,CAAC;QACH,MAAM,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC,4BAA4B,CAAC,CAAC;IAC9D,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/orchestrator/cleanup.test.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/orchestrator/cleanup.test.js

@@ -0,0 +1,67 @@
+import { describe, expect, it } from "vitest";
+import { createEEClient } from "../ee/client.js";
+import { setDefaultEEClient } from "../ee/intercept.js";
+import { extractSession } from "../ee/extract-session.js";
+import { startStubEEServer } from "../__test-stubs__/ee-server.js";
+// ─── Helpers ──────────────────────────────────────────────────────────────────
+function buildTestMessages(userCount) {
+    const msgs = [];
+    for (let i = 0; i < userCount; i++) {
+        msgs.push({ role: "user", content: `Message ${i + 1}` });
+        msgs.push({ role: "assistant", content: `Response ${i + 1}` });
+    }
+    return msgs;
+}
+// ─── Tests ────────────────────────────────────────────────────────────────────
+describe("cleanup -> extractSession -> /api/extract integration", () => {
+    // Stub server started fresh per test via beforeEach / per-test setup
+    // afterAll cleanup is handled within each test using try/finally
+    it("Test 1 (EXTRACT-01): extractSession hits /api/extract on stub server with correct source", async () => {
+        const stub = await startStubEEServer();
+        const client = createEEClient({ baseUrl: `http://127.0.0.1:${stub.port}` });
+        setDefaultEEClient(client);
+        try {
+            // 6 user messages — exceeds 5-message threshold
+            await extractSession(buildTestMessages(6), "/tmp/test-project", "cli-exit", "session-abc");
+            expect(stub.calls.extract.length).toBe(1);
+            expect(stub.calls.extract[0].meta?.source).toBe("cli-exit");
+            expect(stub.calls.extract[0].projectPath).toBe("/tmp/test-project");
+        }
+        finally {
+            setDefaultEEClient(null);
+            await stub.stop();
+        }
+    });
+    it("Test 2 (EXTRACT-03): extractSession completes within 2.5s against 3s-latency stub", async () => {
+        const stub = await startStubEEServer({ latencyMs: 3000 });
+        const client = createEEClient({ baseUrl: `http://127.0.0.1:${stub.port}` });
+        setDefaultEEClient(client);
+        try {
+            const start = Date.now();
+            // Should complete quickly because AbortSignal.timeout(2000) fires
+            await extractSession(buildTestMessages(6), "/tmp/test-project", "cli-exit");
+            const elapsed = Date.now() - start;
+            // AbortSignal(2000ms) fires and error is swallowed — must complete in < 2500ms
+            expect(elapsed).toBeLessThan(2500);
+            // Request was aborted before server responded — no extract calls logged
+            expect(stub.calls.extract.length).toBe(0);
+        }
+        finally {
+            setDefaultEEClient(null);
+            await stub.stop();
+        }
+    });
+    it("Test 3 (D-05): extractSession resolves without throwing when EE server is unreachable", async () => {
+        // Point at port 1 — nothing listening there
+        const client = createEEClient({ baseUrl: "http://127.0.0.1:1" });
+        setDefaultEEClient(client);
+        try {
+            // Must resolve (not throw) even when server is unreachable
+            await expect(extractSession(buildTestMessages(6), "/tmp/test-project", "cli-exit")).resolves.toBeUndefined();
+        }
+        finally {
+            setDefaultEEClient(null);
+        }
+    });
+});
+//# sourceMappingURL=cleanup.test.js.map

package/dist/orchestrator/cleanup.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cleanup.test.js","sourceRoot":"","sources":["../../src/orchestrator/cleanup.test.ts"],"names":[],"mappings":"AAWA,OAAO,EAAY,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AACxD,OAAO,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAC;AACjD,OAAO,EAAE,kBAAkB,EAAE,MAAM,oBAAoB,CAAC;AACxD,OAAO,EAAE,cAAc,EAAE,MAAM,0BAA0B,CAAC;AAC1D,OAAO,EAAE,iBAAiB,EAAE,MAAM,gCAAgC,CAAC;AAEnE,iFAAiF;AAEjF,SAAS,iBAAiB,CAAC,SAAiB;IAC1C,MAAM,IAAI,GAAmB,EAAE,CAAC;IAChC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,SAAS,EAAE,CAAC,EAAE,EAAE,CAAC;QACnC,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,WAAW,CAAC,GAAG,CAAC,EAAE,EAA6B,CAAC,CAAC;QACpF,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,YAAY,CAAC,GAAG,CAAC,EAAE,EAA6B,CAAC,CAAC;IAC5F,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAED,iFAAiF;AAEjF,QAAQ,CAAC,uDAAuD,EAAE,GAAG,EAAE;IACrE,qEAAqE;IACrE,iEAAiE;IAEjE,EAAE,CAAC,0FAA0F,EAAE,KAAK,IAAI,EAAE;QACxG,MAAM,IAAI,GAAG,MAAM,iBAAiB,EAAE,CAAC;QACvC,MAAM,MAAM,GAAG,cAAc,CAAC,EAAE,OAAO,EAAE,oBAAoB,IAAI,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;QAC5E,kBAAkB,CAAC,MAAa,CAAC,CAAC;QAElC,IAAI,CAAC;YACH,gDAAgD;YAChD,MAAM,cAAc,CAAC,iBAAiB,CAAC,CAAC,CAAC,EAAE,mBAAmB,EAAE,UAAU,EAAE,aAAa,CAAC,CAAC;YAE3F,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAC1C,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YAC5D,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;QACtE,CAAC;gBAAS,CAAC;YACT,kBAAkB,CAAC,IAAW,CAAC,CAAC;YAChC,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC;QACpB,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mFAAmF,EAAE,KAAK,IAAI,EAAE;QACjG,MAAM,IAAI,GAAG,MAAM,iBAAiB,CAAC,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAC1D,MAAM,MAAM,GAAG,cAAc,CAAC,EAAE,OAAO,EAAE,oBAAoB,IAAI,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;QAC5E,kBAAkB,CAAC,MAAa,CAAC,CAAC;QAElC,IAAI,CAAC;YACH,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YACzB,kEAAkE;YAClE,MAAM,cAAc,CAAC,iBAAiB,CAAC,CAAC,CAAC,EAAE,mBAAmB,EAAE,UAAU,CAAC,CAAC;YAC5E,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,CAAC;YAEnC,+EAA+E;YAC/E,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;YACnC,wEAAwE;YACxE,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC5C,CAAC;gBAAS,CAAC;YACT,kBAAkB,CAAC,IAAW,CAAC,CAAC;YAChC,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC;QACpB,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uFAAuF,EAAE,KAAK,IAAI,EAAE;QACrG,4CAA4C;QAC5C,MAAM,MAAM,GAAG,cAAc,CAAC,EAAE,OAAO,EAAE,oBAAoB,EAAE,CAAC,CAAC;QACjE,kBAAkB,CAAC,MAAa,CAAC,CAAC;QAElC,IAAI,CAAC;YACH,2DAA2D;YAC3D,MAAM,MAAM,CACV,cAAc,CAAC,iBAAiB,CAAC,CAAC,CAAC,EAAE,mBAAmB,EAAE,UAAU,CAAC,CACtE,CAAC,QAAQ,CAAC,aAAa,EAAE,CAAC;QAC7B,CAAC;gBAAS,CAAC;YACT,kBAAkB,CAAC,IAAW,CAAC,CAAC;QAClC,CAAC;IACH,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/orchestrator/compaction.d.ts

@@ -0,0 +1,36 @@
+import { type ModelMessage } from "ai";
+import type { ProviderFactory as LegacyProvider } from "../providers/runtime.js";
+export interface CompactionSettings {
+    reserveTokens: number;
+    keepRecentTokens: number;
+}
+export interface CutPointResult {
+    firstKeptIndex: number;
+    turnStartIndex: number;
+    isSplitTurn: boolean;
+}
+export interface PreparedCompaction {
+    previousSummary?: string;
+    messagesToSummarize: ModelMessage[];
+    turnPrefixMessages: ModelMessage[];
+    keptMessages: ModelMessage[];
+    firstKeptIndex: number;
+    isSplitTurn: boolean;
+    tokensBefore: number;
+    settings: CompactionSettings;
+}
+export declare const DEFAULT_RESERVE_TOKENS = 16384;
+export declare const DEFAULT_KEEP_RECENT_TOKENS = 20000;
+export declare const POST_TURN_MIN_TOKENS = 2000;
+export declare const COMPACTION_SUMMARY_HEADER = "[Context checkpoint summary]";
+export declare function createCompactionSummaryMessage(summary: string): ModelMessage;
+export declare function isCompactionSummaryMessage(message: ModelMessage | undefined): boolean;
+export declare function getCompactionSummaryText(message: ModelMessage | undefined): string | null;
+export declare function estimateMessageTokens(message: ModelMessage): number;
+export declare function estimateConversationTokens(systemPrompt: string, messages: ModelMessage[], inFlightText?: string): number;
+export declare function shouldCompactContext(contextTokens: number, contextWindow: number, settings: CompactionSettings): boolean;
+export declare function findCutPoint(messages: ModelMessage[], startIndex: number, keepRecentTokens: number): CutPointResult;
+export declare function prepareCompaction(messages: ModelMessage[], systemPrompt: string, settings: CompactionSettings): PreparedCompaction | null;
+export declare function relaxCompactionSettings(settings: CompactionSettings): CompactionSettings;
+export declare function serializeConversation(messages: ModelMessage[]): string;
+export declare function generateCompactionSummary(provider: LegacyProvider, modelId: string, preparation: PreparedCompaction, customInstructions?: string, signal?: AbortSignal): Promise<string>;