@openharness/core 0.4.0 → 0.4.2

package/dist/__tests__/middleware/retry.test.js

@@ -0,0 +1,151 @@
+import { describe, it, expect } from "vitest";
+import { withRetry } from "../../middleware/retry.js";
+// ── Helpers ─────────────────────────────────────────────────────────
+async function collect(gen) {
+    const result = [];
+    for await (const event of gen)
+        result.push(event);
+    return result;
+}
+const doneEvent = {
+    type: "done",
+    result: "complete",
+    messages: [],
+    totalUsage: { inputTokens: 10, outputTokens: 5, totalTokens: 15 },
+};
+// ── Tests ───────────────────────────────────────────────────────────
+describe("withRetry", () => {
+    it("passes through events on success", async () => {
+        const runner = async function* () {
+            yield { type: "text.delta", text: "hello" };
+            yield doneEvent;
+        };
+        const retried = withRetry({ maxRetries: 3, initialDelayMs: 1 })(runner);
+        const events = await collect(retried([], "test"));
+        expect(events).toHaveLength(2);
+        expect(events[0].type).toBe("text.delta");
+        expect(events[1].type).toBe("done");
+    });
+    it("retries on retryable error before content", async () => {
+        let callCount = 0;
+        const runner = async function* () {
+            callCount++;
+            if (callCount === 1) {
+                yield { type: "error", error: new Error("429 rate limit") };
+                yield {
+                    type: "done",
+                    result: "error",
+                    messages: [],
+                    totalUsage: { inputTokens: 0, outputTokens: 0, totalTokens: 0 },
+                };
+                return;
+            }
+            yield { type: "text.delta", text: "success" };
+            yield doneEvent;
+        };
+        const retried = withRetry({ maxRetries: 3, initialDelayMs: 1 })(runner);
+        const events = await collect(retried([], "test"));
+        expect(callCount).toBe(2);
+        const retryEvents = events.filter((e) => e.type === "retry");
+        expect(retryEvents).toHaveLength(1);
+        expect(retryEvents[0].attempt).toBe(0);
+    });
+    it("does not retry after content has been yielded", async () => {
+        let callCount = 0;
+        const runner = async function* () {
+            callCount++;
+            yield { type: "text.delta", text: "content" };
+            yield { type: "error", error: new Error("429 rate limit") };
+            yield doneEvent;
+        };
+        const retried = withRetry({ maxRetries: 3, initialDelayMs: 1 })(runner);
+        const events = await collect(retried([], "test"));
+        expect(callCount).toBe(1);
+        expect(events.filter((e) => e.type === "retry")).toHaveLength(0);
+        // Error should pass through
+        expect(events.find((e) => e.type === "error")).toBeDefined();
+    });
+    it("does not retry non-retryable errors", async () => {
+        let callCount = 0;
+        const runner = async function* () {
+            callCount++;
+            yield { type: "error", error: new Error("invalid input") };
+            yield {
+                type: "done",
+                result: "error",
+                messages: [],
+                totalUsage: { inputTokens: 0, outputTokens: 0, totalTokens: 0 },
+            };
+        };
+        const retried = withRetry({ maxRetries: 3, initialDelayMs: 1 })(runner);
+        const events = await collect(retried([], "test"));
+        expect(callCount).toBe(1);
+        expect(events.filter((e) => e.type === "retry")).toHaveLength(0);
+    });
+    it("retries on thrown retryable errors", async () => {
+        let callCount = 0;
+        const runner = async function* () {
+            callCount++;
+            if (callCount === 1) {
+                throw new Error("503 service unavailable");
+            }
+            yield doneEvent;
+        };
+        const retried = withRetry({ maxRetries: 3, initialDelayMs: 1 })(runner);
+        const events = await collect(retried([], "test"));
+        expect(callCount).toBe(2);
+        expect(events.filter((e) => e.type === "retry")).toHaveLength(1);
+    });
+    it("throws non-retryable thrown errors", async () => {
+        const runner = async function* () {
+            throw new Error("bug in code");
+        };
+        const retried = withRetry({ maxRetries: 3, initialDelayMs: 1 })(runner);
+        await expect(collect(retried([], "test"))).rejects.toThrow("bug in code");
+    });
+    it("gives up after maxRetries", async () => {
+        let callCount = 0;
+        const runner = async function* () {
+            callCount++;
+            yield { type: "error", error: new Error("429 rate limit") };
+            yield {
+                type: "done",
+                result: "error",
+                messages: [],
+                totalUsage: { inputTokens: 0, outputTokens: 0, totalTokens: 0 },
+            };
+        };
+        const retried = withRetry({ maxRetries: 2, initialDelayMs: 1 })(runner);
+        const events = await collect(retried([], "test"));
+        // 1 initial + 2 retries = 3 calls
+        expect(callCount).toBe(3);
+        // Last attempt's error should pass through (not retried)
+        const errorEvents = events.filter((e) => e.type === "error");
+        expect(errorEvents).toHaveLength(1);
+    });
+    it("supports custom isRetryable", async () => {
+        let callCount = 0;
+        const runner = async function* () {
+            callCount++;
+            if (callCount === 1) {
+                yield { type: "error", error: new Error("custom-retryable") };
+                yield {
+                    type: "done",
+                    result: "error",
+                    messages: [],
+                    totalUsage: { inputTokens: 0, outputTokens: 0, totalTokens: 0 },
+                };
+                return;
+            }
+            yield doneEvent;
+        };
+        const retried = withRetry({
+            maxRetries: 3,
+            initialDelayMs: 1,
+            isRetryable: (e) => e.message.includes("custom-retryable"),
+        })(runner);
+        await collect(retried([], "test"));
+        expect(callCount).toBe(2);
+    });
+});
+//# sourceMappingURL=retry.test.js.map

package/dist/__tests__/middleware/retry.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"retry.test.js","sourceRoot":"","sources":["../../../src/__tests__/middleware/retry.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAM,MAAM,QAAQ,CAAC;AAGlD,OAAO,EAAE,SAAS,EAAE,MAAM,2BAA2B,CAAC;AAEtD,uEAAuE;AAEvE,KAAK,UAAU,OAAO,CAAC,GAAwB;IAC7C,MAAM,MAAM,GAAU,EAAE,CAAC;IACzB,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,GAAG;QAAE,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAClD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,MAAM,SAAS,GAAe;IAC5B,IAAI,EAAE,MAAM;IACZ,MAAM,EAAE,UAAU;IAClB,QAAQ,EAAE,EAAE;IACZ,UAAU,EAAE,EAAE,WAAW,EAAE,EAAE,EAAE,YAAY,EAAE,CAAC,EAAE,WAAW,EAAE,EAAE,EAAE;CAClE,CAAC;AAEF,uEAAuE;AAEvE,QAAQ,CAAC,WAAW,EAAE,GAAG,EAAE;IACzB,EAAE,CAAC,kCAAkC,EAAE,KAAK,IAAI,EAAE;QAChD,MAAM,MAAM,GAAW,KAAK,SAAS,CAAC;YACpC,MAAM,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,OAAO,EAAgB,CAAC;YAC1D,MAAM,SAAS,CAAC;QAClB,CAAC,CAAC;QAEF,MAAM,OAAO,GAAG,SAAS,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,cAAc,EAAE,CAAC,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC;QACxE,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,OAAO,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC,CAAC;QAElD,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC/B,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QAC1C,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IACtC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;QACzD,IAAI,SAAS,GAAG,CAAC,CAAC;QAClB,MAAM,MAAM,GAAW,KAAK,SAAS,CAAC;YACpC,SAAS,EAAE,CAAC;YACZ,IAAI,SAAS,KAAK,CAAC,EAAE,CAAC;gBACpB,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,IAAI,KAAK,CAAC,gBAAgB,CAAC,EAAgB,CAAC;gBAC1E,MAAM;oBACJ,IAAI,EAAE,MAAM;oBACZ,MAAM,EAAE,OAAO;oBACf,QAAQ,EAAE,EAAE;oBACZ,UAAU,EAAE,EAAE,WAAW,EAAE,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE,WAAW,EAAE,CAAC,EAAE;iBAClD,CAAC;gBAChB,OAAO;YACT,CAAC;YACD,MAAM,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,SAAS,EAAgB,CAAC;YAC5D,MAAM,SAAS,CAAC;QAClB,CAAC,CAAC;QAEF,MAAM,OAAO,GAAG,SAAS,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,cAAc,EAAE,CAAC,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC;QACxE,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,OAAO,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC,CAAC;QAElD,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC1B,MAAM,WAAW,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,OAAO,CAAC,CAAC;QAClE,MAAM,CAAC,WAAW,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACpC,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;QAC7D,IAAI,SAAS,GAAG,CAAC,CAAC;QAClB,MAAM,MAAM,GAAW,KAAK,SAAS,CAAC;YACpC,SAAS,EAAE,CAAC;YACZ,MAAM,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,SAAS,EAAgB,CAAC;YAC5D,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,IAAI,KAAK,CAAC,gBAAgB,CAAC,EAAgB,CAAC;YAC1E,MAAM,SAAS,CAAC;QAClB,CAAC,CAAC;QAEF,MAAM,OAAO,GAAG,SAAS,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,cAAc,EAAE,CAAC,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC;QACxE,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,OAAO,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC,CAAC;QAElD,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC1B,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,OAAO,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACtE,4BAA4B;QAC5B,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,OAAO,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC;IACpE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qCAAqC,EAAE,KAAK,IAAI,EAAE;QACnD,IAAI,SAAS,GAAG,CAAC,CAAC;QAClB,MAAM,MAAM,GAAW,KAAK,SAAS,CAAC;YACpC,SAAS,EAAE,CAAC;YACZ,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,IAAI,KAAK,CAAC,eAAe,CAAC,EAAgB,CAAC;YACzE,MAAM;gBACJ,IAAI,EAAE,MAAM;gBACZ,MAAM,EAAE,OAAO;gBACf,QAAQ,EAAE,EAAE;gBACZ,UAAU,EAAE,EAAE,WAAW,EAAE,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE,WAAW,EAAE,CAAC,EAAE;aAClD,CAAC;QAClB,CAAC,CAAC;QAEF,MAAM,OAAO,GAAG,SAAS,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,cAAc,EAAE,CAAC,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC;QACxE,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,OAAO,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC,CAAC;QAElD,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC1B,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,OAAO,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IACxE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;QAClD,IAAI,SAAS,GAAG,CAAC,CAAC;QAClB,MAAM,MAAM,GAAW,KAAK,SAAS,CAAC;YACpC,SAAS,EAAE,CAAC;YACZ,IAAI,SAAS,KAAK,CAAC,EAAE,CAAC;gBACpB,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;YAC7C,CAAC;YACD,MAAM,SAAS,CAAC;QAClB,CAAC,CAAC;QAEF,MAAM,OAAO,GAAG,SAAS,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,cAAc,EAAE,CAAC,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC;QACxE,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,OAAO,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC,CAAC;QAElD,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC1B,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,OAAO,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IACxE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;QAClD,MAAM,MAAM,GAAW,KAAK,SAAS,CAAC;YACpC,MAAM,IAAI,KAAK,CAAC,aAAa,CAAC,CAAC;QACjC,CAAC,CAAC;QAEF,MAAM,OAAO,GAAG,SAAS,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,cAAc,EAAE,CAAC,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC;QACxE,MAAM,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;IAC5E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2BAA2B,EAAE,KAAK,IAAI,EAAE;QACzC,IAAI,SAAS,GAAG,CAAC,CAAC;QAClB,MAAM,MAAM,GAAW,KAAK,SAAS,CAAC;YACpC,SAAS,EAAE,CAAC;YACZ,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,IAAI,KAAK,CAAC,gBAAgB,CAAC,EAAgB,CAAC;YAC1E,MAAM;gBACJ,IAAI,EAAE,MAAM;gBACZ,MAAM,EAAE,OAAO;gBACf,QAAQ,EAAE,EAAE;gBACZ,UAAU,EAAE,EAAE,WAAW,EAAE,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE,WAAW,EAAE,CAAC,EAAE;aAClD,CAAC;QAClB,CAAC,CAAC;QAEF,MAAM,OAAO,GAAG,SAAS,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,cAAc,EAAE,CAAC,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC;QACxE,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,OAAO,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC,CAAC;QAElD,kCAAkC;QAClC,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC1B,yDAAyD;QACzD,MAAM,WAAW,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,OAAO,CAAC,CAAC;QAClE,MAAM,CAAC,WAAW,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IACtC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6BAA6B,EAAE,KAAK,IAAI,EAAE;QAC3C,IAAI,SAAS,GAAG,CAAC,CAAC;QAClB,MAAM,MAAM,GAAW,KAAK,SAAS,CAAC;YACpC,SAAS,EAAE,CAAC;YACZ,IAAI,SAAS,KAAK,CAAC,EAAE,CAAC;gBACpB,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,IAAI,KAAK,CAAC,kBAAkB,CAAC,EAAgB,CAAC;gBAC5E,MAAM;oBACJ,IAAI,EAAE,MAAM;oBACZ,MAAM,EAAE,OAAO;oBACf,QAAQ,EAAE,EAAE;oBACZ,UAAU,EAAE,EAAE,WAAW,EAAE,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE,WAAW,EAAE,CAAC,EAAE;iBAClD,CAAC;gBAChB,OAAO;YACT,CAAC;YACD,MAAM,SAAS,CAAC;QAClB,CAAC,CAAC;QAEF,MAAM,OAAO,GAAG,SAAS,CAAC;YACxB,UAAU,EAAE,CAAC;YACb,cAAc,EAAE,CAAC;YACjB,WAAW,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,kBAAkB,CAAC;SAC3D,CAAC,CAAC,MAAM,CAAC,CAAC;QACX,MAAM,OAAO,CAAC,OAAO,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC,CAAC;QAEnC,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAC5B,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/__tests__/middleware/turn-tracking.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=turn-tracking.test.d.ts.map

package/dist/__tests__/middleware/turn-tracking.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"turn-tracking.test.d.ts","sourceRoot":"","sources":["../../../src/__tests__/middleware/turn-tracking.test.ts"],"names":[],"mappings":""}

package/dist/__tests__/middleware/turn-tracking.test.js

@@ -0,0 +1,70 @@
+import { describe, it, expect } from "vitest";
+import { withTurnTracking } from "../../middleware/turn-tracking.js";
+// ── Helpers ─────────────────────────────────────────────────────────
+function createMockRunner(events) {
+    return async function* () {
+        for (const event of events)
+            yield event;
+    };
+}
+async function collect(gen) {
+    const result = [];
+    for await (const event of gen)
+        result.push(event);
+    return result;
+}
+const doneEvent = {
+    type: "done",
+    result: "complete",
+    messages: [],
+    totalUsage: { inputTokens: 10, outputTokens: 5, totalTokens: 15 },
+};
+// ── Tests ───────────────────────────────────────────────────────────
+describe("withTurnTracking", () => {
+    it("wraps events with turn.start and turn.done", async () => {
+        const runner = createMockRunner([
+            { type: "text.delta", text: "hi" },
+            doneEvent,
+        ]);
+        const tracked = withTurnTracking()(runner);
+        const events = await collect(tracked([], "test"));
+        expect(events[0]).toEqual({ type: "turn.start", turnNumber: 1 });
+        expect(events[events.length - 1]).toMatchObject({
+            type: "turn.done",
+            turnNumber: 1,
+        });
+    });
+    it("increments turn counter across calls", async () => {
+        const runner = createMockRunner([doneEvent]);
+        const tracked = withTurnTracking()(runner);
+        const events1 = await collect(tracked([], "first"));
+        const events2 = await collect(tracked([], "second"));
+        expect(events1[0]).toEqual({ type: "turn.start", turnNumber: 1 });
+        expect(events2[0]).toEqual({ type: "turn.start", turnNumber: 2 });
+    });
+    it("reports usage from done event in turn.done", async () => {
+        const runner = createMockRunner([doneEvent]);
+        const tracked = withTurnTracking()(runner);
+        const events = await collect(tracked([], "test"));
+        const turnDone = events.find((e) => e.type === "turn.done");
+        expect(turnDone.usage).toEqual({
+            inputTokens: 10,
+            outputTokens: 5,
+            totalTokens: 15,
+        });
+    });
+    it("passes through all inner events", async () => {
+        const innerEvents = [
+            { type: "text.delta", text: "a" },
+            { type: "text.done", text: "a" },
+            doneEvent,
+        ];
+        const runner = createMockRunner(innerEvents);
+        const tracked = withTurnTracking()(runner);
+        const events = await collect(tracked([], "test"));
+        // turn.start + 3 inner events + turn.done
+        expect(events).toHaveLength(5);
+        expect(events.slice(1, 4)).toEqual(innerEvents);
+    });
+});
+//# sourceMappingURL=turn-tracking.test.js.map

package/dist/__tests__/middleware/turn-tracking.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"turn-tracking.test.js","sourceRoot":"","sources":["../../../src/__tests__/middleware/turn-tracking.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAG9C,OAAO,EAAE,gBAAgB,EAAE,MAAM,mCAAmC,CAAC;AAErE,uEAAuE;AAEvE,SAAS,gBAAgB,CAAC,MAAoB;IAC5C,OAAO,KAAK,SAAS,CAAC;QACpB,KAAK,MAAM,KAAK,IAAI,MAAM;YAAE,MAAM,KAAK,CAAC;IAC1C,CAAC,CAAC;AACJ,CAAC;AAED,KAAK,UAAU,OAAO,CAAC,GAAwB;IAC7C,MAAM,MAAM,GAAU,EAAE,CAAC;IACzB,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,GAAG;QAAE,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAClD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,MAAM,SAAS,GAAe;IAC5B,IAAI,EAAE,MAAM;IACZ,MAAM,EAAE,UAAU;IAClB,QAAQ,EAAE,EAAE;IACZ,UAAU,EAAE,EAAE,WAAW,EAAE,EAAE,EAAE,YAAY,EAAE,CAAC,EAAE,WAAW,EAAE,EAAE,EAAE;CAClE,CAAC;AAEF,uEAAuE;AAEvE,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;IAChC,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;QAC1D,MAAM,MAAM,GAAG,gBAAgB,CAAC;YAC9B,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,IAAI,EAAE;YAClC,SAAS;SACV,CAAC,CAAC;QACH,MAAM,OAAO,GAAG,gBAAgB,EAAE,CAAC,MAAM,CAAC,CAAC;QAC3C,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,OAAO,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC,CAAC;QAElD,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,UAAU,EAAE,CAAC,EAAE,CAAC,CAAC;QACjE,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC;YAC9C,IAAI,EAAE,WAAW;YACjB,UAAU,EAAE,CAAC;SACd,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;QACpD,MAAM,MAAM,GAAG,gBAAgB,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC;QAC7C,MAAM,OAAO,GAAG,gBAAgB,EAAE,CAAC,MAAM,CAAC,CAAC;QAE3C,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,OAAO,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC,CAAC;QACpD,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,OAAO,CAAC,EAAE,EAAE,QAAQ,CAAC,CAAC,CAAC;QAErD,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,UAAU,EAAE,CAAC,EAAE,CAAC,CAAC;QAClE,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,UAAU,EAAE,CAAC,EAAE,CAAC,CAAC;IACpE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;QAC1D,MAAM,MAAM,GAAG,gBAAgB,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC;QAC7C,MAAM,OAAO,GAAG,gBAAgB,EAAE,CAAC,MAAM,CAAC,CAAC;QAC3C,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,OAAO,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC,CAAC;QAElD,MAAM,QAAQ,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,WAAW,CAAC,CAAC;QACjE,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC;YAC7B,WAAW,EAAE,EAAE;YACf,YAAY,EAAE,CAAC;YACf,WAAW,EAAE,EAAE;SAChB,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iCAAiC,EAAE,KAAK,IAAI,EAAE;QAC/C,MAAM,WAAW,GAAiB;YAChC,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,GAAG,EAAE;YACjC,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,GAAG,EAAE;YAChC,SAAS;SACV,CAAC;QACF,MAAM,MAAM,GAAG,gBAAgB,CAAC,WAAW,CAAC,CAAC;QAC7C,MAAM,OAAO,GAAG,gBAAgB,EAAE,CAAC,MAAM,CAAC,CAAC;QAC3C,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,OAAO,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC,CAAC;QAElD,0CAA0C;QAC1C,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC/B,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;IAClD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/__tests__/runner.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=runner.test.d.ts.map

package/dist/__tests__/runner.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"runner.test.d.ts","sourceRoot":"","sources":["../../src/__tests__/runner.test.ts"],"names":[],"mappings":""}

package/dist/__tests__/runner.test.js

@@ -0,0 +1,111 @@
+import { describe, it, expect, vi } from "vitest";
+import { toRunner, pipe, apply } from "../runner.js";
+// ── Helpers ─────────────────────────────────────────────────────────
+function createMockRunner(events) {
+    return async function* () {
+        for (const event of events)
+            yield event;
+    };
+}
+async function collect(gen) {
+    const result = [];
+    for await (const event of gen)
+        result.push(event);
+    return result;
+}
+const doneEvent = {
+    type: "done",
+    result: "complete",
+    messages: [],
+    totalUsage: { inputTokens: 10, outputTokens: 5, totalTokens: 15 },
+};
+// ── Tests ───────────────────────────────────────────────────────────
+describe("runner", () => {
+    describe("toRunner", () => {
+        it("wraps an Agent-like object into a Runner", async () => {
+            const mockAgent = {
+                run: vi.fn(async function* () {
+                    yield { type: "text.delta", text: "hello" };
+                    yield doneEvent;
+                }),
+            };
+            const runner = toRunner(mockAgent);
+            const events = await collect(runner([], "test"));
+            expect(events).toHaveLength(2);
+            expect(events[0]).toEqual({ type: "text.delta", text: "hello" });
+            expect(mockAgent.run).toHaveBeenCalledWith([], "test", undefined);
+        });
+        it("passes history, input, and options through", async () => {
+            const mockAgent = {
+                run: vi.fn(async function* () {
+                    yield doneEvent;
+                }),
+            };
+            const history = [{ role: "user", content: "hi" }];
+            const signal = new AbortController().signal;
+            const runner = toRunner(mockAgent);
+            await collect(runner(history, "test", { signal }));
+            expect(mockAgent.run).toHaveBeenCalledWith(history, "test", { signal });
+        });
+    });
+    describe("pipe", () => {
+        it("returns identity when no middleware provided", async () => {
+            const runner = createMockRunner([doneEvent]);
+            const piped = pipe()(runner);
+            const events = await collect(piped([], "test"));
+            expect(events).toEqual([doneEvent]);
+        });
+        it("composes middleware outermost first", async () => {
+            const order = [];
+            const outer = (inner) => async function* (h, i, o) {
+                order.push("outer-before");
+                yield* inner(h, i, o);
+                order.push("outer-after");
+            };
+            const inner = (inner) => async function* (h, i, o) {
+                order.push("inner-before");
+                yield* inner(h, i, o);
+                order.push("inner-after");
+            };
+            const runner = createMockRunner([doneEvent]);
+            const piped = pipe(outer, inner)(runner);
+            await collect(piped([], "test"));
+            expect(order).toEqual([
+                "outer-before",
+                "inner-before",
+                "inner-after",
+                "outer-after",
+            ]);
+        });
+        it("composes three middleware in correct order", async () => {
+            const tags = [];
+            const mw = (tag) => (inner) => async function* (h, i, o) {
+                tags.push(`${tag}-in`);
+                yield* inner(h, i, o);
+                tags.push(`${tag}-out`);
+            };
+            const runner = createMockRunner([doneEvent]);
+            const piped = pipe(mw("a"), mw("b"), mw("c"))(runner);
+            await collect(piped([], "test"));
+            expect(tags).toEqual(["a-in", "b-in", "c-in", "c-out", "b-out", "a-out"]);
+        });
+    });
+    describe("apply", () => {
+        it("is shorthand for pipe(...mw)(runner)", async () => {
+            const addPrefix = (inner) => async function* (h, i, o) {
+                yield { type: "text.delta", text: "[prefix] " };
+                yield* inner(h, i, o);
+            };
+            const runner = createMockRunner([
+                { type: "text.delta", text: "hello" },
+                doneEvent,
+            ]);
+            const applied = apply(runner, addPrefix);
+            const events = await collect(applied([], "test"));
+            expect(events).toHaveLength(3);
+            expect(events[0]).toEqual({ type: "text.delta", text: "[prefix] " });
+            expect(events[1]).toEqual({ type: "text.delta", text: "hello" });
+        });
+    });
+});
+//# sourceMappingURL=runner.test.js.map

package/dist/__tests__/runner.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"runner.test.js","sourceRoot":"","sources":["../../src/__tests__/runner.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAGlD,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE,KAAK,EAAE,MAAM,cAAc,CAAC;AAErD,uEAAuE;AAEvE,SAAS,gBAAgB,CAAC,MAAoB;IAC5C,OAAO,KAAK,SAAS,CAAC;QACpB,KAAK,MAAM,KAAK,IAAI,MAAM;YAAE,MAAM,KAAK,CAAC;IAC1C,CAAC,CAAC;AACJ,CAAC;AAED,KAAK,UAAU,OAAO,CAAC,GAA+B;IACpD,MAAM,MAAM,GAAiB,EAAE,CAAC;IAChC,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,GAAG;QAAE,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAClD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,MAAM,SAAS,GAAe;IAC5B,IAAI,EAAE,MAAM;IACZ,MAAM,EAAE,UAAU;IAClB,QAAQ,EAAE,EAAE;IACZ,UAAU,EAAE,EAAE,WAAW,EAAE,EAAE,EAAE,YAAY,EAAE,CAAC,EAAE,WAAW,EAAE,EAAE,EAAE;CAClE,CAAC;AAEF,uEAAuE;AAEvE,QAAQ,CAAC,QAAQ,EAAE,GAAG,EAAE;IACtB,QAAQ,CAAC,UAAU,EAAE,GAAG,EAAE;QACxB,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;YACxD,MAAM,SAAS,GAAG;gBAChB,GAAG,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,SAAS,CAAC;oBACxB,MAAM,EAAE,IAAI,EAAE,YAAqB,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;oBACrD,MAAM,SAAS,CAAC;gBAClB,CAAC,CAAC;aACH,CAAC;YAEF,MAAM,MAAM,GAAG,QAAQ,CAAC,SAAgB,CAAC,CAAC;YAC1C,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,MAAM,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC,CAAC;YAEjD,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAC/B,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC;YACjE,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,oBAAoB,CAAC,EAAE,EAAE,MAAM,EAAE,SAAS,CAAC,CAAC;QACpE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;YAC1D,MAAM,SAAS,GAAG;gBAChB,GAAG,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,SAAS,CAAC;oBACxB,MAAM,SAAS,CAAC;gBAClB,CAAC,CAAC;aACH,CAAC;YAEF,MAAM,OAAO,GAAG,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;YAC3D,MAAM,MAAM,GAAG,IAAI,eAAe,EAAE,CAAC,MAAM,CAAC;YAC5C,MAAM,MAAM,GAAG,QAAQ,CAAC,SAAgB,CAAC,CAAC;YAC1C,MAAM,OAAO,CAAC,MAAM,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC;YAEnD,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,oBAAoB,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;QAC1E,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,MAAM,EAAE,GAAG,EAAE;QACpB,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;YAC5D,MAAM,MAAM,GAAG,gBAAgB,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC;YAC7C,MAAM,KAAK,GAAG,IAAI,EAAE,CAAC,MAAM,CAAC,CAAC;YAC7B,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,KAAK,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC,CAAC;YAEhD,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC;QACtC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qCAAqC,EAAE,KAAK,IAAI,EAAE;YACnD,MAAM,KAAK,GAAa,EAAE,CAAC;YAE3B,MAAM,KAAK,GAAe,CAAC,KAAK,EAAE,EAAE,CAClC,KAAK,SAAS,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC;gBACtB,KAAK,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;gBAC3B,KAAK,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;gBACtB,KAAK,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;YAC5B,CAAC,CAAC;YAEJ,MAAM,KAAK,GAAe,CAAC,KAAK,EAAE,EAAE,CAClC,KAAK,SAAS,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC;gBACtB,KAAK,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;gBAC3B,KAAK,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;gBACtB,KAAK,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;YAC5B,CAAC,CAAC;YAEJ,MAAM,MAAM,GAAG,gBAAgB,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC;YAC7C,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC,MAAM,CAAC,CAAC;YACzC,MAAM,OAAO,CAAC,KAAK,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC,CAAC;YAEjC,MAAM,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC;gBACpB,cAAc;gBACd,cAAc;gBACd,aAAa;gBACb,aAAa;aACd,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;YAC1D,MAAM,IAAI,GAAa,EAAE,CAAC;YAE1B,MAAM,EAAE,GAAG,CAAC,GAAW,EAAc,EAAE,CAAC,CAAC,KAAK,EAAE,EAAE,CAChD,KAAK,SAAS,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC;gBACtB,IAAI,CAAC,IAAI,CAAC,GAAG,GAAG,KAAK,CAAC,CAAC;gBACvB,KAAK,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;gBACtB,IAAI,CAAC,IAAI,CAAC,GAAG,GAAG,MAAM,CAAC,CAAC;YAC1B,CAAC,CAAC;YAEJ,MAAM,MAAM,GAAG,gBAAgB,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC;YAC7C,MAAM,KAAK,GAAG,IAAI,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC;YACtD,MAAM,OAAO,CAAC,KAAK,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC,CAAC;YAEjC,MAAM,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC;QAC5E,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,OAAO,EAAE,GAAG,EAAE;QACrB,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;YACpD,MAAM,SAAS,GAAe,CAAC,KAAK,EAAE,EAAE,CACtC,KAAK,SAAS,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC;gBACtB,MAAM,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,WAAW,EAAgB,CAAC;gBAC9D,KAAK,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;YACxB,CAAC,CAAC;YAEJ,MAAM,MAAM,GAAG,gBAAgB,CAAC;gBAC9B,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,OAAO,EAAE;gBACrC,SAAS;aACV,CAAC,CAAC;YAEH,MAAM,OAAO,GAAG,KAAK,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;YACzC,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,OAAO,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC,CAAC;YAElD,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAC/B,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,WAAW,EAAE,CAAC,CAAC;YACrE,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC;QACnE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/__tests__/stream.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=stream.test.d.ts.map

package/dist/__tests__/stream.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"stream.test.d.ts","sourceRoot":"","sources":["../../src/__tests__/stream.test.ts"],"names":[],"mappings":""}

package/dist/__tests__/stream.test.js

@@ -0,0 +1,100 @@
+import { describe, it, expect } from "vitest";
+import { tap, filter, map, takeUntil } from "../stream.js";
+// ── Helpers ─────────────────────────────────────────────────────────
+async function collect(gen) {
+    const result = [];
+    for await (const event of gen)
+        result.push(event);
+    return result;
+}
+async function* fromArray(events) {
+    for (const event of events)
+        yield event;
+}
+const doneEvent = {
+    type: "done",
+    result: "complete",
+    messages: [],
+    totalUsage: { inputTokens: 0, outputTokens: 0, totalTokens: 0 },
+};
+// ── Tests ───────────────────────────────────────────────────────────
+describe("stream combinators", () => {
+    describe("tap", () => {
+        it("calls fn for each event without modifying the stream", async () => {
+            const seen = [];
+            const events = [
+                { type: "text.delta", text: "a" },
+                { type: "text.delta", text: "b" },
+                doneEvent,
+            ];
+            const result = await collect(tap((e) => seen.push(e.type))(fromArray(events)));
+            expect(result).toEqual(events);
+            expect(seen).toEqual(["text.delta", "text.delta", "done"]);
+        });
+        it("returns all events unchanged", async () => {
+            const events = [
+                { type: "step.start", stepNumber: 1 },
+                doneEvent,
+            ];
+            const result = await collect(tap(() => { })(fromArray(events)));
+            expect(result).toEqual(events);
+        });
+    });
+    describe("filter", () => {
+        it("drops events that don't match the predicate", async () => {
+            const events = [
+                { type: "text.delta", text: "a" },
+                { type: "reasoning.delta", text: "think" },
+                { type: "text.delta", text: "b" },
+                doneEvent,
+            ];
+            const result = await collect(filter((e) => e.type !== "reasoning.delta")(fromArray(events)));
+            expect(result).toHaveLength(3);
+            expect(result.map((e) => e.type)).toEqual(["text.delta", "text.delta", "done"]);
+        });
+        it("never filters done events", async () => {
+            const events = [
+                { type: "text.delta", text: "a" },
+                doneEvent,
+            ];
+            // Predicate rejects everything
+            const result = await collect(filter(() => false)(fromArray(events)));
+            expect(result).toHaveLength(1);
+            expect(result[0].type).toBe("done");
+        });
+    });
+    describe("map", () => {
+        it("transforms each event", async () => {
+            const events = [
+                { type: "text.delta", text: "hello" },
+                doneEvent,
+            ];
+            const result = await collect(map((e) => (e.type === "text.delta" ? { ...e, text: e.text.toUpperCase() } : e))(fromArray(events)));
+            expect(result[0]).toEqual({ type: "text.delta", text: "HELLO" });
+            expect(result[1]).toEqual(doneEvent);
+        });
+    });
+    describe("takeUntil", () => {
+        it("stops after predicate matches (inclusive)", async () => {
+            const events = [
+                { type: "text.delta", text: "a" },
+                { type: "text.delta", text: "b" },
+                { type: "text.done", text: "ab" },
+                { type: "text.delta", text: "c" },
+                doneEvent,
+            ];
+            const result = await collect(takeUntil((e) => e.type === "text.done")(fromArray(events)));
+            expect(result).toHaveLength(3);
+            expect(result[2]).toEqual({ type: "text.done", text: "ab" });
+        });
+        it("yields all events if predicate never matches", async () => {
+            const events = [
+                { type: "text.delta", text: "a" },
+                doneEvent,
+            ];
+            const result = await collect(takeUntil(() => false)(fromArray(events)));
+            expect(result).toEqual(events);
+        });
+    });
+});
+//# sourceMappingURL=stream.test.js.map

package/dist/__tests__/stream.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"stream.test.js","sourceRoot":"","sources":["../../src/__tests__/stream.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAM,MAAM,QAAQ,CAAC;AAElD,OAAO,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE,SAAS,EAAE,MAAM,cAAc,CAAC;AAE3D,uEAAuE;AAEvE,KAAK,UAAU,OAAO,CAAC,GAA8B;IACnD,MAAM,MAAM,GAAiB,EAAE,CAAC;IAChC,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,GAAG;QAAE,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAClD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,KAAK,SAAS,CAAC,CAAC,SAAS,CAAC,MAAoB;IAC5C,KAAK,MAAM,KAAK,IAAI,MAAM;QAAE,MAAM,KAAK,CAAC;AAC1C,CAAC;AAED,MAAM,SAAS,GAAe;IAC5B,IAAI,EAAE,MAAM;IACZ,MAAM,EAAE,UAAU;IAClB,QAAQ,EAAE,EAAE;IACZ,UAAU,EAAE,EAAE,WAAW,EAAE,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE,WAAW,EAAE,CAAC,EAAE;CAChE,CAAC;AAEF,uEAAuE;AAEvE,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;IAClC,QAAQ,CAAC,KAAK,EAAE,GAAG,EAAE;QACnB,EAAE,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;YACpE,MAAM,IAAI,GAAa,EAAE,CAAC;YAC1B,MAAM,MAAM,GAAiB;gBAC3B,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,GAAG,EAAE;gBACjC,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,GAAG,EAAE;gBACjC,SAAS;aACV,CAAC;YAEF,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;YAE/E,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;YAC/B,MAAM,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,YAAY,EAAE,YAAY,EAAE,MAAM,CAAC,CAAC,CAAC;QAC7D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8BAA8B,EAAE,KAAK,IAAI,EAAE;YAC5C,MAAM,MAAM,GAAiB;gBAC3B,EAAE,IAAI,EAAE,YAAY,EAAE,UAAU,EAAE,CAAC,EAAE;gBACrC,SAAS;aACV,CAAC;YAEF,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;YAC/D,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QACjC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,QAAQ,EAAE,GAAG,EAAE;QACtB,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;YAC3D,MAAM,MAAM,GAAiB;gBAC3B,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,GAAG,EAAE;gBACjC,EAAE,IAAI,EAAE,iBAAiB,EAAE,IAAI,EAAE,OAAO,EAAE;gBAC1C,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,GAAG,EAAE;gBACjC,SAAS;aACV,CAAC;YAEF,MAAM,MAAM,GAAG,MAAM,OAAO,CAC1B,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,iBAAiB,CAAC,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAC/D,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAC/B,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,YAAY,EAAE,YAAY,EAAE,MAAM,CAAC,CAAC,CAAC;QAClF,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2BAA2B,EAAE,KAAK,IAAI,EAAE;YACzC,MAAM,MAAM,GAAiB;gBAC3B,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,GAAG,EAAE;gBACjC,SAAS;aACV,CAAC;YAEF,+BAA+B;YAC/B,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;YAErE,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAC/B,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACtC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,KAAK,EAAE,GAAG,EAAE;QACnB,EAAE,CAAC,uBAAuB,EAAE,KAAK,IAAI,EAAE;YACrC,MAAM,MAAM,GAAiB;gBAC3B,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,OAAO,EAAE;gBACrC,SAAS;aACV,CAAC;YAEF,MAAM,MAAM,GAAG,MAAM,OAAO,CAC1B,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,YAAY,CAAC,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAC9E,SAAS,CAAC,MAAM,CAAC,CAClB,CACF,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC;YACjE,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QACvC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,WAAW,EAAE,GAAG,EAAE;QACzB,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;YACzD,MAAM,MAAM,GAAiB;gBAC3B,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,GAAG,EAAE;gBACjC,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,GAAG,EAAE;gBACjC,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,IAAI,EAAE;gBACjC,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,GAAG,EAAE;gBACjC,SAAS;aACV,CAAC;YAEF,MAAM,MAAM,GAAG,MAAM,OAAO,CAC1B,SAAS,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,WAAW,CAAC,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAC5D,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAC/B,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;QAC/D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;YAC5D,MAAM,MAAM,GAAiB;gBAC3B,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,GAAG,EAAE;gBACjC,SAAS;aACV,CAAC;YAEF,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,SAAS,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;YACxE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QACjC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/__tests__/ui-stream.test.js

@@ -257,6 +257,111 @@ describe("sessionEventsToUIStream", () => {
             expect(subError).toHaveLength(1);
             expect(subError[0].data.error).toBe("Agent crashed");
         });
+        it("emits data-oh:subagent.start for background task tool calls", async () => {
+            const chunks = await collectChunks([
+                {
+                    type: "tool.start",
+                    toolCallId: "tc-bg",
+                    toolName: "task",
+                    input: { agent: "researcher", prompt: "find info", background: true },
+                },
+            ]);
+            const subStart = findChunks(chunks, "data-oh:subagent.start");
+            expect(subStart).toHaveLength(1);
+            expect(subStart[0].data).toEqual({
+                agentName: "researcher",
+                task: "find info",
+                path: ["researcher"],
+            });
+        });
+        it("suppresses data-oh:subagent.done for background task tool calls", async () => {
+            const chunks = await collectChunks([
+                {
+                    type: "tool.start",
+                    toolCallId: "tc-bg",
+                    toolName: "task",
+                    input: { agent: "researcher", prompt: "find info", background: true },
+                },
+                {
+                    type: "tool.done",
+                    toolCallId: "tc-bg",
+                    toolName: "task",
+                    output: '<background_spawn agent_id="bg-1">spawned</background_spawn>',
+                },
+            ]);
+            const subDone = findChunks(chunks, "data-oh:subagent.done");
+            expect(subDone).toHaveLength(0);
+        });
+        it("still emits data-oh:subagent.done for foreground task tool calls", async () => {
+            const chunks = await collectChunks([
+                {
+                    type: "tool.start",
+                    toolCallId: "tc-fg",
+                    toolName: "task",
+                    input: { agent: "coder", prompt: "fix bug" },
+                },
+                {
+                    type: "tool.done",
+                    toolCallId: "tc-fg",
+                    toolName: "task",
+                    output: "<task_result>fixed</task_result>",
+                },
+            ]);
+            const subDone = findChunks(chunks, "data-oh:subagent.done");
+            expect(subDone).toHaveLength(1);
+        });
+        it("emits data-oh:subagent.error for background task tool errors", async () => {
+            const chunks = await collectChunks([
+                {
+                    type: "tool.start",
+                    toolCallId: "tc-bg",
+                    toolName: "task",
+                    input: { agent: "researcher", prompt: "find info", background: true },
+                },
+                {
+                    type: "tool.error",
+                    toolCallId: "tc-bg",
+                    toolName: "task",
+                    error: "Spawn failed",
+                },
+            ]);
+            const subError = findChunks(chunks, "data-oh:subagent.error");
+            expect(subError).toHaveLength(1);
+            expect(subError[0].data.error).toBe("Spawn failed");
+        });
+        it("handles mixed foreground and background task calls", async () => {
+            const chunks = await collectChunks([
+                {
+                    type: "tool.start",
+                    toolCallId: "tc-bg",
+                    toolName: "task",
+                    input: { agent: "researcher", prompt: "bg work", background: true },
+                },
+                {
+                    type: "tool.start",
+                    toolCallId: "tc-fg",
+                    toolName: "task",
+                    input: { agent: "coder", prompt: "fg work" },
+                },
+                {
+                    type: "tool.done",
+                    toolCallId: "tc-bg",
+                    toolName: "task",
+                    output: "spawned",
+                },
+                {
+                    type: "tool.done",
+                    toolCallId: "tc-fg",
+                    toolName: "task",
+                    output: "completed",
+                },
+            ]);
+            const subStart = findChunks(chunks, "data-oh:subagent.start");
+            expect(subStart).toHaveLength(2);
+            const subDone = findChunks(chunks, "data-oh:subagent.done");
+            expect(subDone).toHaveLength(1);
+            expect(subDone[0].data.agentName).toBe("coder");
+        });
         it("does not emit subagent events for regular tools", async () => {
             const chunks = await collectChunks([
                 {