@trigger.dev/core 4.4.5 → 4.5.0-rc.0

package/dist/esm/v3/test/test-realtime-streams-manager.js

@@ -0,0 +1,136 @@
+import { createAsyncIterableStreamFromAsyncIterable, } from "../streams/asyncIterableStream.js";
+export class TestRealtimeStreamsManager {
+    buffers = new Map();
+    pipeWaits = new Map();
+    writeListeners = new Set();
+    pipe(key, source, _options) {
+        const buffer = this.getBuffer(key);
+        const self = this;
+        // Eagerly drain the source in the background so chunks land in the
+        // buffer + notify listeners even when the caller never consumes the
+        // returned stream. This mirrors the real SDK behavior: `streams.writer`
+        // awaits `instance.wait()`, it doesn't read the returned stream.
+        //
+        // The source is read ONCE (into a chunks array) and replayed into a
+        // ReadableStream so the caller can still consume it if they want.
+        const readChunks = [];
+        let resolveDone;
+        const done = new Promise((resolve) => {
+            resolveDone = resolve;
+        });
+        (async () => {
+            try {
+                const iter = source instanceof ReadableStream
+                    ? (async function* () {
+                        const reader = source.getReader();
+                        try {
+                            while (true) {
+                                const { done: d, value } = await reader.read();
+                                if (d)
+                                    return;
+                                yield value;
+                            }
+                        }
+                        finally {
+                            reader.releaseLock();
+                        }
+                    })()
+                    : source;
+                for await (const chunk of iter) {
+                    readChunks.push(chunk);
+                    buffer.push(chunk);
+                    self.notify(key, chunk);
+                }
+            }
+            catch {
+                // Swallow — tests can inspect what made it into the buffer
+            }
+            finally {
+                resolveDone();
+            }
+        })();
+        const replayStream = (async function* () {
+            // Wait for all chunks to be drained, then replay from our snapshot
+            await done;
+            for (const chunk of readChunks)
+                yield chunk;
+        })();
+        const wrappedStream = createAsyncIterableStreamFromAsyncIterable(replayStream);
+        if (!this.pipeWaits.has(key))
+            this.pipeWaits.set(key, []);
+        this.pipeWaits.get(key).push(done);
+        return {
+            wait: () => done.then(() => ({})),
+            get stream() {
+                return wrappedStream;
+            },
+        };
+    }
+    async append(key, part, _options) {
+        this.getBuffer(key).push(part);
+        this.notify(key, part);
+    }
+    /**
+     * Register a listener fired for every chunk written to any stream.
+     * Returns an unsubscribe function.
+     *
+     * Intended for test harnesses that need to react to writes synchronously
+     * (e.g. resolving a "turn complete" latch).
+     */
+    onWrite(listener) {
+        this.writeListeners.add(listener);
+        return () => {
+            this.writeListeners.delete(listener);
+        };
+    }
+    notify(key, chunk) {
+        for (const listener of this.writeListeners) {
+            try {
+                listener(key, chunk);
+            }
+            catch {
+                // Never let a listener error break stream writes
+            }
+        }
+    }
+    // ── Test driver API (not part of RealtimeStreamsManager interface) ──────
+    /**
+     * Return all chunks written to the given stream key in order of write.
+     */
+    __chunksFromTest(key) {
+        return (this.buffers.get(key) ?? []).slice();
+    }
+    /**
+     * Return all chunks across every stream, keyed by stream id.
+     */
+    __allChunksFromTest() {
+        const result = {};
+        for (const [key, chunks] of this.buffers.entries()) {
+            result[key] = chunks.slice();
+        }
+        return result;
+    }
+    /**
+     * Clear the buffer for a specific stream or all streams.
+     */
+    __clearFromTest(key) {
+        if (key === undefined) {
+            this.buffers.clear();
+        }
+        else {
+            this.buffers.delete(key);
+        }
+    }
+    reset() {
+        this.buffers.clear();
+        this.pipeWaits.clear();
+        this.writeListeners.clear();
+    }
+    getBuffer(key) {
+        if (!this.buffers.has(key)) {
+            this.buffers.set(key, []);
+        }
+        return this.buffers.get(key);
+    }
+}
+//# sourceMappingURL=test-realtime-streams-manager.js.map

package/dist/esm/v3/test/test-realtime-streams-manager.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"test-realtime-streams-manager.js","sourceRoot":"","sources":["../../../../src/v3/test/test-realtime-streams-manager.ts"],"names":[],"mappings":"AAAA,OAAO,EAEL,0CAA0C,GAC3C,MAAM,mCAAmC,CAAC;AAgB3C,MAAM,OAAO,0BAA0B;IAC7B,OAAO,GAAG,IAAI,GAAG,EAAqB,CAAC;IACvC,SAAS,GAAG,IAAI,GAAG,EAA2B,CAAC;IAC/C,cAAc,GAAG,IAAI,GAAG,EAAiB,CAAC;IAElD,IAAI,CACF,GAAW,EACX,MAA4C,EAC5C,QAAyC;QAEzC,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;QACnC,MAAM,IAAI,GAAG,IAAI,CAAC;QAElB,mEAAmE;QACnE,oEAAoE;QACpE,wEAAwE;QACxE,iEAAiE;QACjE,EAAE;QACF,oEAAoE;QACpE,kEAAkE;QAClE,MAAM,UAAU,GAAQ,EAAE,CAAC;QAC3B,IAAI,WAAwB,CAAC;QAC7B,MAAM,IAAI,GAAG,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,EAAE;YACzC,WAAW,GAAG,OAAO,CAAC;QACxB,CAAC,CAAC,CAAC;QAEH,CAAC,KAAK,IAAI,EAAE;YACV,IAAI,CAAC;gBACH,MAAM,IAAI,GACR,MAAM,YAAY,cAAc;oBAC9B,CAAC,CAAC,CAAC,KAAK,SAAS,CAAC;wBACd,MAAM,MAAM,GAAG,MAAM,CAAC,SAAS,EAAE,CAAC;wBAClC,IAAI,CAAC;4BACH,OAAO,IAAI,EAAE,CAAC;gCACZ,MAAM,EAAE,IAAI,EAAE,CAAC,EAAE,KAAK,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;gCAC/C,IAAI,CAAC;oCAAE,OAAO;gCACd,MAAM,KAAU,CAAC;4BACnB,CAAC;wBACH,CAAC;gCAAS,CAAC;4BACT,MAAM,CAAC,WAAW,EAAE,CAAC;wBACvB,CAAC;oBACH,CAAC,CAAC,EAAE;oBACN,CAAC,CAAC,MAAM,CAAC;gBAEb,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,IAAI,EAAE,CAAC;oBAC/B,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;oBACvB,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;oBACnB,IAAI,CAAC,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;gBAC1B,CAAC;YACH,CAAC;YAAC,MAAM,CAAC;gBACP,2DAA2D;YAC7D,CAAC;oBAAS,CAAC;gBACT,WAAW,EAAE,CAAC;YAChB,CAAC;QACH,CAAC,CAAC,EAAE,CAAC;QAEL,MAAM,YAAY,GAAG,CAAC,KAAK,SAAS,CAAC;YACnC,mEAAmE;YACnE,MAAM,IAAI,CAAC;YACX,KAAK,MAAM,KAAK,IAAI,UAAU;gBAAE,MAAM,KAAK,CAAC;QAC9C,CAAC,CAAC,EAAE,CAAC;QACL,MAAM,aAAa,GAAG,0CAA0C,CAAC,YAAY,CAAC,CAAC;QAE/E,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,GAAG,CAAC;YAAE,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;QAC1D,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,GAAG,CAAE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAEpC,OAAO;YACL,IAAI,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC;YACjC,IAAI,MAAM;gBACR,OAAO,aAAa,CAAC;YACvB,CAAC;SACF,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,MAAM,CACV,GAAW,EACX,IAAW,EACX,QAAyC;QAEzC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC/B,IAAI,CAAC,MAAM,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;IACzB,CAAC;IAED;;;;;;OAMG;IACH,OAAO,CAAC,QAAuB;QAC7B,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAClC,OAAO,GAAG,EAAE;YACV,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QACvC,CAAC,CAAC;IACJ,CAAC;IAEO,MAAM,CAAC,GAAW,EAAE,KAAc;QACxC,KAAK,MAAM,QAAQ,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YAC3C,IAAI,CAAC;gBACH,QAAQ,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;YACvB,CAAC;YAAC,MAAM,CAAC;gBACP,iDAAiD;YACnD,CAAC;QACH,CAAC;IACH,CAAC;IAED,2EAA2E;IAE3E;;OAEG;IACH,gBAAgB,CAAc,GAAW;QACvC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC,CAAC,KAAK,EAAS,CAAC;IACtD,CAAC;IAED;;OAEG;IACH,mBAAmB;QACjB,MAAM,MAAM,GAA8B,EAAE,CAAC;QAC7C,KAAK,MAAM,CAAC,GAAG,EAAE,MAAM,CAAC,IAAI,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,EAAE,CAAC;YACnD,MAAM,CAAC,GAAG,CAAC,GAAG,MAAM,CAAC,KAAK,EAAE,CAAC;QAC/B,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;OAEG;IACH,eAAe,CAAC,GAAY;QAC1B,IAAI,GAAG,KAAK,SAAS,EAAE,CAAC;YACtB,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;QACvB,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QAC3B,CAAC;IACH,CAAC;IAED,KAAK;QACH,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;QACrB,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC;QACvB,IAAI,CAAC,cAAc,CAAC,KAAK,EAAE,CAAC;IAC9B,CAAC;IAEO,SAAS,CAAC,GAAW;QAC3B,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC;YAC3B,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;QAC5B,CAAC;QACD,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAE,CAAC;IAChC,CAAC;CACF"}

package/dist/esm/v3/test/test-run-metadata-manager.d.ts

@@ -0,0 +1,31 @@
+import type { DeserializedJson } from "../../schemas/json.js";
+import type { AsyncIterableStream } from "../streams/asyncIterableStream.js";
+import type { RunMetadataManager, RunMetadataUpdater } from "../runMetadata/types.js";
+/**
+ * In-memory implementation of `RunMetadataManager` for unit tests.
+ *
+ * Just stores metadata in a Map — no API calls, no queue. Good enough
+ * for tests that read/write metadata via `runMetadata.getKey()` /
+ * `runMetadata.set()`, including the IDLE_TIMEOUT and TURN_TIMEOUT
+ * checks inside `chat.agent()`.
+ */
+export declare class TestRunMetadataManager implements RunMetadataManager {
+    private store;
+    enterWithMetadata(metadata: Record<string, DeserializedJson>): void;
+    current(): Record<string, DeserializedJson> | undefined;
+    getKey(key: string): DeserializedJson | undefined;
+    set(key: string, value: DeserializedJson): this;
+    del(key: string): this;
+    append(key: string, value: DeserializedJson): this;
+    remove(key: string, value: DeserializedJson): this;
+    increment(key: string, value: number): this;
+    decrement(key: string, value: number): this;
+    update(metadata: Record<string, DeserializedJson>): this;
+    flush(): Promise<void>;
+    refresh(): Promise<void>;
+    stream<T>(_key: string, value: AsyncIterable<T> | ReadableStream<T>): Promise<AsyncIterable<T>>;
+    fetchStream<T>(_key: string): Promise<AsyncIterableStream<T>>;
+    get parent(): RunMetadataUpdater;
+    get root(): RunMetadataUpdater;
+    reset(): void;
+}

package/dist/esm/v3/test/test-run-metadata-manager.js

@@ -0,0 +1,82 @@
+/**
+ * In-memory implementation of `RunMetadataManager` for unit tests.
+ *
+ * Just stores metadata in a Map — no API calls, no queue. Good enough
+ * for tests that read/write metadata via `runMetadata.getKey()` /
+ * `runMetadata.set()`, including the IDLE_TIMEOUT and TURN_TIMEOUT
+ * checks inside `chat.agent()`.
+ */
+export class TestRunMetadataManager {
+    store = {};
+    enterWithMetadata(metadata) {
+        this.store = { ...metadata };
+    }
+    current() {
+        return { ...this.store };
+    }
+    getKey(key) {
+        return this.store[key];
+    }
+    set(key, value) {
+        this.store[key] = value;
+        return this;
+    }
+    del(key) {
+        delete this.store[key];
+        return this;
+    }
+    append(key, value) {
+        const existing = this.store[key];
+        if (Array.isArray(existing)) {
+            existing.push(value);
+        }
+        else {
+            this.store[key] = [value];
+        }
+        return this;
+    }
+    remove(key, value) {
+        const existing = this.store[key];
+        if (Array.isArray(existing)) {
+            this.store[key] = existing.filter((v) => v !== value);
+        }
+        return this;
+    }
+    increment(key, value) {
+        const existing = this.store[key];
+        const current = typeof existing === "number" ? existing : 0;
+        this.store[key] = current + value;
+        return this;
+    }
+    decrement(key, value) {
+        return this.increment(key, -value);
+    }
+    update(metadata) {
+        this.store = { ...metadata };
+        return this;
+    }
+    async flush() { }
+    async refresh() { }
+    async stream(_key, value) {
+        return value;
+    }
+    async fetchStream(_key) {
+        // Return an empty async iterable — tests can override if needed
+        const empty = {
+            [Symbol.asyncIterator]: () => ({
+                next: () => Promise.resolve({ done: true, value: undefined }),
+            }),
+        };
+        return empty;
+    }
+    get parent() {
+        return this;
+    }
+    get root() {
+        return this;
+    }
+    reset() {
+        this.store = {};
+    }
+}
+//# sourceMappingURL=test-run-metadata-manager.js.map

package/dist/esm/v3/test/test-run-metadata-manager.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"test-run-metadata-manager.js","sourceRoot":"","sources":["../../../../src/v3/test/test-run-metadata-manager.ts"],"names":[],"mappings":"AAIA;;;;;;;GAOG;AACH,MAAM,OAAO,sBAAsB;IACzB,KAAK,GAAqC,EAAE,CAAC;IAErD,iBAAiB,CAAC,QAA0C;QAC1D,IAAI,CAAC,KAAK,GAAG,EAAE,GAAG,QAAQ,EAAE,CAAC;IAC/B,CAAC;IAED,OAAO;QACL,OAAO,EAAE,GAAG,IAAI,CAAC,KAAK,EAAE,CAAC;IAC3B,CAAC;IAED,MAAM,CAAC,GAAW;QAChB,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IACzB,CAAC;IAED,GAAG,CAAC,GAAW,EAAE,KAAuB;QACtC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;QACxB,OAAO,IAAI,CAAC;IACd,CAAC;IAED,GAAG,CAAC,GAAW;QACb,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QACvB,OAAO,IAAI,CAAC;IACd,CAAC;IAED,MAAM,CAAC,GAAW,EAAE,KAAuB;QACzC,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QACjC,IAAI,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC5B,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACvB,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QAC5B,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,MAAM,CAAC,GAAW,EAAE,KAAuB;QACzC,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QACjC,IAAI,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC5B,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,KAAK,CAAqB,CAAC;QAC5E,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,SAAS,CAAC,GAAW,EAAE,KAAa;QAClC,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QACjC,MAAM,OAAO,GAAG,OAAO,QAAQ,KAAK,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;QAC5D,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,OAAO,GAAG,KAAK,CAAC;QAClC,OAAO,IAAI,CAAC;IACd,CAAC;IAED,SAAS,CAAC,GAAW,EAAE,KAAa;QAClC,OAAO,IAAI,CAAC,SAAS,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC,CAAC;IACrC,CAAC;IAED,MAAM,CAAC,QAA0C;QAC/C,IAAI,CAAC,KAAK,GAAG,EAAE,GAAG,QAAQ,EAAE,CAAC;QAC7B,OAAO,IAAI,CAAC;IACd,CAAC;IAED,KAAK,CAAC,KAAK,KAAmB,CAAC;IAC/B,KAAK,CAAC,OAAO,KAAmB,CAAC;IAEjC,KAAK,CAAC,MAAM,CACV,IAAY,EACZ,KAA2C;QAE3C,OAAO,KAAyB,CAAC;IACnC,CAAC;IAED,KAAK,CAAC,WAAW,CAAI,IAAY;QAC/B,gEAAgE;QAChE,MAAM,KAAK,GAAG;YACZ,CAAC,MAAM,CAAC,aAAa,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;gBAC7B,IAAI,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,IAAa,EAAE,KAAK,EAAE,SAAc,EAAE,CAAC;aAC5E,CAAC;SACH,CAAC;QACF,OAAO,KAA0C,CAAC;IACpD,CAAC;IAED,IAAI,MAAM;QACR,OAAO,IAAI,CAAC;IACd,CAAC;IAED,IAAI,IAAI;QACN,OAAO,IAAI,CAAC;IACd,CAAC;IAED,KAAK;QACH,IAAI,CAAC,KAAK,GAAG,EAAE,CAAC;IAClB,CAAC;CACF"}

package/dist/esm/v3/test/test-session-stream-manager.d.ts

@@ -0,0 +1,60 @@
+import { InputStreamOncePromise } from "../inputStreams/types.js";
+import type { InputStreamOnceOptions } from "../realtimeStreams/types.js";
+import type { SessionChannelIO, SessionStreamManager } from "../sessionStreams/types.js";
+type Handler = (data: unknown) => void | Promise<void>;
+/**
+ * In-memory implementation of `SessionStreamManager` for unit tests. Same
+ * shape as {@link TestInputStreamManager} but keyed on `(sessionId, io)`.
+ *
+ * Tests push data via `__sendFromTest(sessionId, io, data)` — any pending
+ * `once()` waiters resolve immediately, and all `on()` handlers fire (awaited
+ * if they return a promise). Records that arrive before a listener is
+ * registered are buffered so the first `once()` picks them up.
+ */
+export declare class TestSessionStreamManager implements SessionStreamManager {
+    private handlers;
+    private onceWaiters;
+    private buffer;
+    private seqNums;
+    on(sessionId: string, io: SessionChannelIO, handler: Handler): {
+        off: () => void;
+    };
+    once(sessionId: string, io: SessionChannelIO, options?: InputStreamOnceOptions): InputStreamOncePromise<unknown>;
+    peek(sessionId: string, io: SessionChannelIO): unknown | undefined;
+    lastSeqNum(sessionId: string, io: SessionChannelIO): number | undefined;
+    setLastSeqNum(sessionId: string, io: SessionChannelIO, seqNum: number): void;
+    lastDispatchedSeqNum(_sessionId: string, _io: SessionChannelIO): number | undefined;
+    setLastDispatchedSeqNum(_sessionId: string, _io: SessionChannelIO, _seqNum: number): void;
+    setMinTimestamp(_sessionId: string, _io: SessionChannelIO, _minTimestamp: number | undefined): void;
+    shiftBuffer(sessionId: string, io: SessionChannelIO): boolean;
+    disconnectStream(_sessionId: string, _io: SessionChannelIO): void;
+    clearHandlers(): void;
+    reset(): void;
+    disconnect(): void;
+    /**
+     * Push a record onto the given channel.
+     *
+     * Dispatch rules — similar to the production manager, but with a tweak
+     * that makes unit tests deterministic:
+     *
+     * 1. **Handlers always observe** (like production). A session-level `.on`
+     *    is a filter-observer — it fires every time a record arrives,
+     *    regardless of whether a `.once` waiter is also active.
+     * 2. **First waiter consumes** the record if present (like production).
+     * 3. **If no waiter, the record is buffered for the next `.once` call.**
+     *    Production discards records that only match handlers — but in
+     *    production the SSE tail introduces enough latency that the next
+     *    `.once` is usually registered before the next record arrives. Tests
+     *    send synchronously right after `turn-complete`, so without this
+     *    buffer the next `waitWithIdleTimeout` would race and lose the
+     *    message. The buffer is the only deviation from production semantics.
+     */
+    __sendFromTest(sessionId: string, io: SessionChannelIO, data: unknown): Promise<void>;
+    /**
+     * Immediately resolve every pending `once()` waiter for the given channel
+     * with a timeout error. Simulates a closed stream (e.g. session closed).
+     */
+    __closeFromTest(sessionId: string, io: SessionChannelIO): void;
+    private removeWaiter;
+}
+export {};

package/dist/esm/v3/test/test-session-stream-manager.js

@@ -0,0 +1,243 @@
+import { InputStreamOncePromise, InputStreamTimeoutError, } from "../inputStreams/types.js";
+function keyFor(sessionId, io) {
+    return `${sessionId}:${io}`;
+}
+/**
+ * In-memory implementation of `SessionStreamManager` for unit tests. Same
+ * shape as {@link TestInputStreamManager} but keyed on `(sessionId, io)`.
+ *
+ * Tests push data via `__sendFromTest(sessionId, io, data)` — any pending
+ * `once()` waiters resolve immediately, and all `on()` handlers fire (awaited
+ * if they return a promise). Records that arrive before a listener is
+ * registered are buffered so the first `once()` picks them up.
+ */
+export class TestSessionStreamManager {
+    handlers = new Map();
+    onceWaiters = new Map();
+    buffer = new Map();
+    seqNums = new Map();
+    on(sessionId, io, handler) {
+        const key = keyFor(sessionId, io);
+        let set = this.handlers.get(key);
+        if (!set) {
+            set = new Set();
+            this.handlers.set(key, set);
+        }
+        set.add(handler);
+        // Note: we intentionally do NOT replay buffered records into the
+        // newly-registered handler, and we do NOT drain the buffer. The
+        // buffer is owned by `once()` — registering a passive observer
+        // (`on`) must not consume records destined for a future `once`
+        // waiter. This matches production SSE semantics where handlers
+        // observe records as they arrive, not retroactively.
+        //
+        // Earlier versions drained the buffer here, which caused user
+        // messages buffered during the runtime's `runFn` boot phase to be
+        // silently swallowed by the `stopInput.on()` handler registered at
+        // ai.ts:4806 (the stop handler ignores `kind: "message"` chunks).
+        // The next `messagesInput.waitWithIdleTimeout` then waited 30s for
+        // a record that had already been "delivered" to a handler that
+        // didn't want it.
+        return {
+            off: () => {
+                this.handlers.get(key)?.delete(handler);
+            },
+        };
+    }
+    once(sessionId, io, options) {
+        const key = keyFor(sessionId, io);
+        return new InputStreamOncePromise((resolve) => {
+            if (options?.signal?.aborted) {
+                resolve({
+                    ok: false,
+                    error: new InputStreamTimeoutError(key, options.timeoutMs ?? 0),
+                });
+                return;
+            }
+            const buffered = this.buffer.get(key);
+            if (buffered && buffered.length > 0) {
+                const next = buffered.shift();
+                if (buffered.length === 0)
+                    this.buffer.delete(key);
+                resolve({ ok: true, output: next });
+                return;
+            }
+            const waiter = { resolve, signal: options?.signal };
+            if (options?.timeoutMs !== undefined) {
+                waiter.timer = setTimeout(() => {
+                    this.removeWaiter(key, waiter);
+                    resolve({
+                        ok: false,
+                        error: new InputStreamTimeoutError(key, options.timeoutMs),
+                    });
+                }, options.timeoutMs);
+            }
+            if (options?.signal) {
+                const abortHandler = () => {
+                    this.removeWaiter(key, waiter);
+                    if (waiter.timer)
+                        clearTimeout(waiter.timer);
+                    resolve({
+                        ok: false,
+                        error: new InputStreamTimeoutError(key, options.timeoutMs ?? 0),
+                    });
+                };
+                waiter.abortHandler = abortHandler;
+                options.signal.addEventListener("abort", abortHandler, { once: true });
+            }
+            let waiters = this.onceWaiters.get(key);
+            if (!waiters) {
+                waiters = [];
+                this.onceWaiters.set(key, waiters);
+            }
+            waiters.push(waiter);
+        });
+    }
+    peek(sessionId, io) {
+        const buffered = this.buffer.get(keyFor(sessionId, io));
+        if (buffered && buffered.length > 0)
+            return buffered[0];
+        return undefined;
+    }
+    lastSeqNum(sessionId, io) {
+        return this.seqNums.get(keyFor(sessionId, io));
+    }
+    setLastSeqNum(sessionId, io, seqNum) {
+        this.seqNums.set(keyFor(sessionId, io), seqNum);
+    }
+    lastDispatchedSeqNum(_sessionId, _io) {
+        // The test harness drives records via `__sendFromTest` without seq
+        // numbers, so the committed-consume cursor stays undefined. Tests
+        // that need cursor behaviour exercise it via the real manager.
+        return undefined;
+    }
+    setLastDispatchedSeqNum(_sessionId, _io, _seqNum) {
+        // no-op — see comment on `lastDispatchedSeqNum`.
+    }
+    setMinTimestamp(_sessionId, _io, _minTimestamp) {
+        // No filter applied in tests; the test harness drives records directly
+        // and the chat.agent retry path is exercised separately.
+    }
+    shiftBuffer(sessionId, io) {
+        const key = keyFor(sessionId, io);
+        const buffered = this.buffer.get(key);
+        if (buffered && buffered.length > 0) {
+            buffered.shift();
+            if (buffered.length === 0)
+                this.buffer.delete(key);
+            return true;
+        }
+        return false;
+    }
+    disconnectStream(_sessionId, _io) {
+        // no-op — no real SSE tail in tests
+    }
+    clearHandlers() {
+        this.handlers.clear();
+    }
+    reset() {
+        for (const waiters of this.onceWaiters.values()) {
+            for (const w of waiters) {
+                if (w.timer)
+                    clearTimeout(w.timer);
+                if (w.signal && w.abortHandler) {
+                    w.signal.removeEventListener("abort", w.abortHandler);
+                }
+            }
+        }
+        this.onceWaiters.clear();
+        this.handlers.clear();
+        this.buffer.clear();
+        this.seqNums.clear();
+    }
+    disconnect() {
+        this.reset();
+    }
+    // ── Test driver API (not part of SessionStreamManager interface) ──────
+    /**
+     * Push a record onto the given channel.
+     *
+     * Dispatch rules — similar to the production manager, but with a tweak
+     * that makes unit tests deterministic:
+     *
+     * 1. **Handlers always observe** (like production). A session-level `.on`
+     *    is a filter-observer — it fires every time a record arrives,
+     *    regardless of whether a `.once` waiter is also active.
+     * 2. **First waiter consumes** the record if present (like production).
+     * 3. **If no waiter, the record is buffered for the next `.once` call.**
+     *    Production discards records that only match handlers — but in
+     *    production the SSE tail introduces enough latency that the next
+     *    `.once` is usually registered before the next record arrives. Tests
+     *    send synchronously right after `turn-complete`, so without this
+     *    buffer the next `waitWithIdleTimeout` would race and lose the
+     *    message. The buffer is the only deviation from production semantics.
+     */
+    async __sendFromTest(sessionId, io, data) {
+        const key = keyFor(sessionId, io);
+        const handlers = this.handlers.get(key);
+        if (handlers && handlers.size > 0) {
+            // Awaited so test code can rely on handlers having completed by the
+            // time `__sendFromTest` resolves. Wrapped per-handler so a
+            // throwing/rejecting handler doesn't poison Promise.all and break
+            // unrelated test state.
+            await Promise.all(Array.from(handlers).map(async (h) => {
+                try {
+                    await h(data);
+                }
+                catch {
+                    // Never let a handler error break test state
+                }
+            }));
+        }
+        const waiters = this.onceWaiters.get(key);
+        if (waiters && waiters.length > 0) {
+            const w = waiters.shift();
+            if (waiters.length === 0)
+                this.onceWaiters.delete(key);
+            if (w.timer)
+                clearTimeout(w.timer);
+            if (w.signal && w.abortHandler) {
+                w.signal.removeEventListener("abort", w.abortHandler);
+            }
+            w.resolve({ ok: true, output: data });
+            return;
+        }
+        let buffered = this.buffer.get(key);
+        if (!buffered) {
+            buffered = [];
+            this.buffer.set(key, buffered);
+        }
+        buffered.push(data);
+    }
+    /**
+     * Immediately resolve every pending `once()` waiter for the given channel
+     * with a timeout error. Simulates a closed stream (e.g. session closed).
+     */
+    __closeFromTest(sessionId, io) {
+        const key = keyFor(sessionId, io);
+        const waiters = this.onceWaiters.get(key);
+        if (!waiters)
+            return;
+        const pending = waiters.splice(0);
+        for (const w of pending) {
+            if (w.timer)
+                clearTimeout(w.timer);
+            if (w.signal && w.abortHandler) {
+                w.signal.removeEventListener("abort", w.abortHandler);
+            }
+            w.resolve({
+                ok: false,
+                error: new InputStreamTimeoutError(key, 0),
+            });
+        }
+    }
+    removeWaiter(key, waiter) {
+        const waiters = this.onceWaiters.get(key);
+        if (!waiters)
+            return;
+        const idx = waiters.indexOf(waiter);
+        if (idx >= 0)
+            waiters.splice(idx, 1);
+    }
+}
+//# sourceMappingURL=test-session-stream-manager.js.map

package/dist/esm/v3/test/test-session-stream-manager.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"test-session-stream-manager.js","sourceRoot":"","sources":["../../../../src/v3/test/test-session-stream-manager.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,sBAAsB,EAEtB,uBAAuB,GACxB,MAAM,0BAA0B,CAAC;AAgBlC,SAAS,MAAM,CAAC,SAAiB,EAAE,EAAoB;IACrD,OAAO,GAAG,SAAS,IAAI,EAAE,EAAE,CAAC;AAC9B,CAAC;AAED;;;;;;;;GAQG;AACH,MAAM,OAAO,wBAAwB;IAC3B,QAAQ,GAAG,IAAI,GAAG,EAAwB,CAAC;IAC3C,WAAW,GAAG,IAAI,GAAG,EAAwB,CAAC;IAC9C,MAAM,GAAG,IAAI,GAAG,EAAqB,CAAC;IACtC,OAAO,GAAG,IAAI,GAAG,EAAkB,CAAC;IAE5C,EAAE,CACA,SAAiB,EACjB,EAAoB,EACpB,OAAgB;QAEhB,MAAM,GAAG,GAAG,MAAM,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;QAElC,IAAI,GAAG,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QACjC,IAAI,CAAC,GAAG,EAAE,CAAC;YACT,GAAG,GAAG,IAAI,GAAG,EAAE,CAAC;YAChB,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;QAC9B,CAAC;QACD,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QAEjB,iEAAiE;QACjE,gEAAgE;QAChE,+DAA+D;QAC/D,+DAA+D;QAC/D,+DAA+D;QAC/D,qDAAqD;QACrD,EAAE;QACF,8DAA8D;QAC9D,kEAAkE;QAClE,mEAAmE;QACnE,kEAAkE;QAClE,mEAAmE;QACnE,+DAA+D;QAC/D,kBAAkB;QAElB,OAAO;YACL,GAAG,EAAE,GAAG,EAAE;gBACR,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC;YAC1C,CAAC;SACF,CAAC;IACJ,CAAC;IAED,IAAI,CACF,SAAiB,EACjB,EAAoB,EACpB,OAAgC;QAEhC,MAAM,GAAG,GAAG,MAAM,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;QAElC,OAAO,IAAI,sBAAsB,CAAU,CAAC,OAAO,EAAE,EAAE;YACrD,IAAI,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC;gBAC7B,OAAO,CAAC;oBACN,EAAE,EAAE,KAAK;oBACT,KAAK,EAAE,IAAI,uBAAuB,CAAC,GAAG,EAAE,OAAO,CAAC,SAAS,IAAI,CAAC,CAAC;iBAChE,CAAC,CAAC;gBACH,OAAO;YACT,CAAC;YAED,MAAM,QAAQ,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YACtC,IAAI,QAAQ,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACpC,MAAM,IAAI,GAAG,QAAQ,CAAC,KAAK,EAAE,CAAC;gBAC9B,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC;oBAAE,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;gBACnD,OAAO,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;gBACpC,OAAO;YACT,CAAC;YAED,MAAM,MAAM,GAAe,EAAE,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC;YAEhE,IAAI,OAAO,EAAE,SAAS,KAAK,SAAS,EAAE,CAAC;gBACrC,MAAM,CAAC,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE;oBAC7B,IAAI,CAAC,YAAY,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;oBAC/B,OAAO,CAAC;wBACN,EAAE,EAAE,KAAK;wBACT,KAAK,EAAE,IAAI,uBAAuB,CAAC,GAAG,EAAE,OAAO,CAAC,SAAU,CAAC;qBAC5D,CAAC,CAAC;gBACL,CAAC,EAAE,OAAO,CAAC,SAAS,CAAC,CAAC;YACxB,CAAC;YAED,IAAI,OAAO,EAAE,MAAM,EAAE,CAAC;gBACpB,MAAM,YAAY,GAAG,GAAG,EAAE;oBACxB,IAAI,CAAC,YAAY,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;oBAC/B,IAAI,MAAM,CAAC,KAAK;wBAAE,YAAY,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;oBAC7C,OAAO,CAAC;wBACN,EAAE,EAAE,KAAK;wBACT,KAAK,EAAE,IAAI,uBAAuB,CAAC,GAAG,EAAE,OAAO,CAAC,SAAS,IAAI,CAAC,CAAC;qBAChE,CAAC,CAAC;gBACL,CAAC,CAAC;gBACF,MAAM,CAAC,YAAY,GAAG,YAAY,CAAC;gBACnC,OAAO,CAAC,MAAM,CAAC,gBAAgB,CAAC,OAAO,EAAE,YAAY,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;YACzE,CAAC;YAED,IAAI,OAAO,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YACxC,IAAI,CAAC,OAAO,EAAE,CAAC;gBACb,OAAO,GAAG,EAAE,CAAC;gBACb,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;YACrC,CAAC;YACD,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACvB,CAAC,CAAC,CAAC;IACL,CAAC;IAED,IAAI,CAAC,SAAiB,EAAE,EAAoB;QAC1C,MAAM,QAAQ,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC,CAAC;QACxD,IAAI,QAAQ,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC;YAAE,OAAO,QAAQ,CAAC,CAAC,CAAC,CAAC;QACxD,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,UAAU,CAAC,SAAiB,EAAE,EAAoB;QAChD,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC,CAAC;IACjD,CAAC;IAED,aAAa,CAAC,SAAiB,EAAE,EAAoB,EAAE,MAAc;QACnE,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,SAAS,EAAE,EAAE,CAAC,EAAE,MAAM,CAAC,CAAC;IAClD,CAAC;IAED,oBAAoB,CAAC,UAAkB,EAAE,GAAqB;QAC5D,mEAAmE;QACnE,kEAAkE;QAClE,+DAA+D;QAC/D,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,uBAAuB,CACrB,UAAkB,EAClB,GAAqB,EACrB,OAAe;QAEf,iDAAiD;IACnD,CAAC;IAED,eAAe,CACb,UAAkB,EAClB,GAAqB,EACrB,aAAiC;QAEjC,uEAAuE;QACvE,yDAAyD;IAC3D,CAAC;IAED,WAAW,CAAC,SAAiB,EAAE,EAAoB;QACjD,MAAM,GAAG,GAAG,MAAM,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;QAClC,MAAM,QAAQ,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QACtC,IAAI,QAAQ,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACpC,QAAQ,CAAC,KAAK,EAAE,CAAC;YACjB,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC;gBAAE,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YACnD,OAAO,IAAI,CAAC;QACd,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED,gBAAgB,CAAC,UAAkB,EAAE,GAAqB;QACxD,oCAAoC;IACtC,CAAC;IAED,aAAa;QACX,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC;IACxB,CAAC;IAED,KAAK;QACH,KAAK,MAAM,OAAO,IAAI,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,EAAE,CAAC;YAChD,KAAK,MAAM,CAAC,IAAI,OAAO,EAAE,CAAC;gBACxB,IAAI,CAAC,CAAC,KAAK;oBAAE,YAAY,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;gBACnC,IAAI,CAAC,CAAC,MAAM,IAAI,CAAC,CAAC,YAAY,EAAE,CAAC;oBAC/B,CAAC,CAAC,MAAM,CAAC,mBAAmB,CAAC,OAAO,EAAE,CAAC,CAAC,YAAY,CAAC,CAAC;gBACxD,CAAC;YACH,CAAC;QACH,CAAC;QACD,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,CAAC;QACzB,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC;QACtB,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;QACpB,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;IACvB,CAAC;IAED,UAAU;QACR,IAAI,CAAC,KAAK,EAAE,CAAC;IACf,CAAC;IAED,yEAAyE;IAEzE;;;;;;;;;;;;;;;;;OAiBG;IACH,KAAK,CAAC,cAAc,CAClB,SAAiB,EACjB,EAAoB,EACpB,IAAa;QAEb,MAAM,GAAG,GAAG,MAAM,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;QAElC,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QACxC,IAAI,QAAQ,IAAI,QAAQ,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;YAClC,oEAAoE;YACpE,2DAA2D;YAC3D,kEAAkE;YAClE,wBAAwB;YACxB,MAAM,OAAO,CAAC,GAAG,CACf,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,KAAK,EAAE,CAAC,EAAE,EAAE;gBACnC,IAAI,CAAC;oBACH,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC;gBAChB,CAAC;gBAAC,MAAM,CAAC;oBACP,6CAA6C;gBAC/C,CAAC;YACH,CAAC,CAAC,CACH,CAAC;QACJ,CAAC;QAED,MAAM,OAAO,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QAC1C,IAAI,OAAO,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAClC,MAAM,CAAC,GAAG,OAAO,CAAC,KAAK,EAAG,CAAC;YAC3B,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC;gBAAE,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YACvD,IAAI,CAAC,CAAC,KAAK;gBAAE,YAAY,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;YACnC,IAAI,CAAC,CAAC,MAAM,IAAI,CAAC,CAAC,YAAY,EAAE,CAAC;gBAC/B,CAAC,CAAC,MAAM,CAAC,mBAAmB,CAAC,OAAO,EAAE,CAAC,CAAC,YAAY,CAAC,CAAC;YACxD,CAAC;YACD,CAAC,CAAC,OAAO,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;YACtC,OAAO;QACT,CAAC;QAED,IAAI,QAAQ,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QACpC,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,QAAQ,GAAG,EAAE,CAAC;YACd,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,GAAG,EAAE,QAAQ,CAAC,CAAC;QACjC,CAAC;QACD,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACtB,CAAC;IAED;;;OAGG;IACH,eAAe,CAAC,SAAiB,EAAE,EAAoB;QACrD,MAAM,GAAG,GAAG,MAAM,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;QAClC,MAAM,OAAO,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QAC1C,IAAI,CAAC,OAAO;YAAE,OAAO;QACrB,MAAM,OAAO,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;QAClC,KAAK,MAAM,CAAC,IAAI,OAAO,EAAE,CAAC;YACxB,IAAI,CAAC,CAAC,KAAK;gBAAE,YAAY,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;YACnC,IAAI,CAAC,CAAC,MAAM,IAAI,CAAC,CAAC,YAAY,EAAE,CAAC;gBAC/B,CAAC,CAAC,MAAM,CAAC,mBAAmB,CAAC,OAAO,EAAE,CAAC,CAAC,YAAY,CAAC,CAAC;YACxD,CAAC;YACD,CAAC,CAAC,OAAO,CAAC;gBACR,EAAE,EAAE,KAAK;gBACT,KAAK,EAAE,IAAI,uBAAuB,CAAC,GAAG,EAAE,CAAC,CAAC;aAC3C,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAEO,YAAY,CAAC,GAAW,EAAE,MAAkB;QAClD,MAAM,OAAO,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QAC1C,IAAI,CAAC,OAAO;YAAE,OAAO;QACrB,MAAM,GAAG,GAAG,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QACpC,IAAI,GAAG,IAAI,CAAC;YAAE,OAAO,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC;IACvC,CAAC;CACF"}

package/dist/esm/v3/types/tasks.d.ts

@@ -280,6 +280,12 @@ type CommonTaskOptions<TIdentifier extends string, TPayload = void, TOutput = un
      * Should be a valid JSON Schema Draft 7 object.
      */
     jsonSchema?: JSONSchema;
+    /** @internal Set by SDK internals (e.g. `chat.agent()`, `schedules.task()`). */
+    triggerSource?: string;
+    /** @internal Agent configuration, only set when `triggerSource` is `"agent"`. */
+    agentConfig?: {
+        type: string;
+    };
 };
 export type TaskOptions<TIdentifier extends string, TPayload = void, TOutput = unknown, TInitOutput extends InitOutput = any> = CommonTaskOptions<TIdentifier, TPayload, TOutput, TInitOutput>;
 export type TaskOptionsWithSchema<TIdentifier extends string, TOutput = unknown, TInitOutput extends InitOutput = any> = CommonTaskOptions<TIdentifier, any, TOutput, TInitOutput> & {
@@ -424,6 +430,26 @@ export interface Task<TIdentifier extends string, TInput = void, TOutput = any>
      * ```
      */
     triggerAndWait: (payload: TInput, options?: TriggerAndWaitOptions, requestOptions?: TriggerApiRequestOptions) => TaskRunPromise<TIdentifier, TOutput>;
+    /**
+     * Trigger a task and subscribe to its updates via realtime. Unlike `triggerAndWait`,
+     * this does NOT suspend the parent run — the parent stays alive and polls for updates.
+     * This enables parallel tool calls and proper abort signal handling.
+     *
+     * @param payload
+     * @param options - Options for the task run, including an optional `signal` to cancel the subscription and child run
+     * @returns TaskRunPromise
+     * @example
+     * ```
+     * const result = await task.triggerAndSubscribe({ foo: "bar" }, { signal: abortSignal });
+     *
+     * if (result.ok) {
+     *   console.log(result.output);
+     * } else {
+     *   console.error(result.error);
+     * }
+     * ```
+     */
+    triggerAndSubscribe: (payload: TInput, options?: TriggerAndSubscribeOptions) => TaskRunPromise<TIdentifier, TOutput>;
     /**
      * Batch trigger multiple task runs with the given payloads, and wait for the results. Returns the results of the task runs.
      * @param items - Array, AsyncIterable, or ReadableStream of batch items
@@ -713,6 +739,15 @@ export type TriggerOptions = {
     };
 };
 export type TriggerAndWaitOptions = Omit<TriggerOptions, "version">;
+export type TriggerAndSubscribeOptions = Omit<TriggerOptions, "version"> & {
+    /** An AbortSignal to cancel the subscription. When fired, the subscription closes and the promise rejects. */
+    signal?: AbortSignal;
+    /**
+     * Whether to cancel the child run when the abort signal fires.
+     * @default true
+     */
+    cancelOnAbort?: boolean;
+};
 export type BatchTriggerOptions = {
     /**
      * If no idempotencyKey is set on an individual item in the batch, it will use this key on each item + the array index.

package/dist/esm/v3/utils/gitBranch.d.ts

@@ -0,0 +1,2 @@
+export declare function isValidGitBranchName(branch: string): boolean;
+export declare function sanitizeBranchName(ref: string | null | undefined): string | null;