@dbos-inc/vercel-ai 0.1.5 → 0.3.7

package/dist/durable-stream.js

@@ -0,0 +1,344 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ModelStreamWriter = void 0;
+exports.resolveDurableStream = resolveDurableStream;
+exports.writeToolRecord = writeToolRecord;
+exports.writeDurableStream = writeDurableStream;
+exports.closeDurableStream = closeDurableStream;
+exports.readDurableStream = readDurableStream;
+const node_crypto_1 = require("node:crypto");
+const dbos_sdk_1 = require("@dbos-inc/dbos-sdk");
+function resolveDurableStream(options) {
+    if (options === undefined)
+        return undefined;
+    const config = typeof options === 'string' ? { key: options } : options;
+    return { key: config.key, maxBatchParts: config.maxBatchParts ?? 20, maxBatchDelayMs: config.maxBatchDelayMs ?? 25 };
+}
+function stepInfo() {
+    return { step: dbos_sdk_1.DBOS.stepID ?? -1, attempt: dbos_sdk_1.DBOS.stepStatus?.currentAttempt ?? 1 };
+}
+// Parts the reader can render; framing, metadata and terminal parts are recorded elsewhere or not at all.
+function isContentPart(part) {
+    switch (part.type) {
+        case 'stream-start':
+        case 'response-metadata':
+        case 'finish':
+        case 'error':
+        case 'raw':
+            return false;
+        default:
+            return true;
+    }
+}
+// Bytes become base64 so a file part stays compact in Postgres.
+function encodePart(part) {
+    if (part.type === 'file' && part.data.type === 'data' && part.data.data instanceof Uint8Array) {
+        return { ...part, data: { type: 'data', data: Buffer.from(part.data.data).toString('base64') } };
+    }
+    return part;
+}
+// A transient write error (the SDK already retries offset conflicts) gets a few attempts before it fails the model call.
+async function writeWithRetry(key, record) {
+    for (let attempt = 1;; attempt++) {
+        try {
+            return await dbos_sdk_1.DBOS.writeStream(key, record);
+        }
+        catch (error) {
+            if (attempt === 3)
+                throw error;
+            await new Promise((resolve) => setTimeout(resolve, 100 * attempt));
+        }
+    }
+}
+/** Batches a live model step's parts into step-scope stream writes; nothing is written on replay because the step body does not run. */
+class ModelStreamWriter {
+    config;
+    pending = [];
+    chain = Promise.resolve();
+    failure;
+    timer;
+    step = dbos_sdk_1.DBOS.stepID ?? -1;
+    // Unique per execution of the step, so a recovered run's re-execution is distinguishable from the crashed one.
+    attempt = (0, node_crypto_1.randomUUID)();
+    constructor(config) {
+        this.config = config;
+    }
+    push(part) {
+        if (!isContentPart(part))
+            return;
+        this.pending.push(encodePart(part));
+        if (this.pending.length >= this.config.maxBatchParts)
+            this.flush();
+        else
+            this.timer ??= setTimeout(() => this.flush(), this.config.maxBatchDelayMs);
+    }
+    /** Flushes, records how the call ended, and resolves once every write is durable; a write that failed after retries fails the call here. */
+    async end(outcome) {
+        this.flush();
+        this.write({ kind: 'model-end', step: this.step, attempt: this.attempt, ...outcome });
+        await this.chain;
+        if (this.failure !== undefined)
+            throw this.failure;
+    }
+    /** After a failure: flush what streamed so the record matches what the consumer saw; the stream's end then comes from the workflow's status. */
+    async abandon() {
+        this.flush();
+        await this.chain;
+    }
+    flush() {
+        if (this.timer !== undefined)
+            clearTimeout(this.timer);
+        this.timer = undefined;
+        if (this.pending.length === 0)
+            return;
+        const parts = this.pending;
+        this.pending = [];
+        this.write({ kind: 'model', step: this.step, attempt: this.attempt, parts });
+    }
+    // Every link has a handler, so a rejection can never sit unobserved; after one failure later writes are skipped.
+    write(record) {
+        this.chain = this.chain
+            .then(() => (this.failure === undefined ? writeWithRetry(this.config.key, record) : undefined))
+            .catch((error) => {
+            this.failure ??= error;
+        });
+    }
+}
+exports.ModelStreamWriter = ModelStreamWriter;
+/** Records a tool call's outcome from inside its step. */
+function writeToolRecord(key, toolCallId, outcome) {
+    const record = { kind: 'tool', ...stepInfo(), toolCallId, ...outcome };
+    return dbos_sdk_1.DBOS.writeStream(key, record);
+}
+/**
+ * Appends UI message chunks to a durable stream. From a step the write is cheap and at-least-once, so give data parts
+ * stable ids; from workflow code it is a checkpointed step, so the number of calls must be deterministic.
+ */
+function writeDurableStream(key, chunks) {
+    const status = dbos_sdk_1.DBOS.stepStatus;
+    const record = { kind: 'ui', step: status?.stepID, attempt: status?.currentAttempt, chunks };
+    return dbos_sdk_1.DBOS.writeStream(key, record);
+}
+/** Marks the end of the turn explicitly and closes the stream; without it the reader infers the end from the last model call or the workflow's status. */
+async function closeDurableStream(key, finishReason = 'stop') {
+    const record = { kind: 'end', finishReason };
+    await dbos_sdk_1.DBOS.writeStream(key, record);
+    await dbos_sdk_1.DBOS.closeStream(key);
+}
+/** Reads a durable stream as AI SDK UI message chunks, live or after the fact, resuming from `offset`. */
+function readDurableStream(options) {
+    const iterator = uiChunks(options);
+    return new ReadableStream({
+        async pull(controller) {
+            const { value, done } = await iterator.next();
+            if (done)
+                controller.close();
+            else
+                controller.enqueue(value);
+        },
+        async cancel() {
+            await iterator.return(undefined);
+        },
+    });
+}
+async function* uiChunks(options) {
+    const { workflowID, key, sendReasoning = true, sendSources = false, onError = () => 'An error occurred.' } = options;
+    const client = options.client ?? dbos_sdk_1.DBOS;
+    const state = {
+        offset: options.offset ?? 0,
+        resumed: (options.offset ?? 0) > 0,
+        openParts: new Map(),
+        ended: false,
+    };
+    if (state.offset === 0)
+        yield { type: 'start', messageId: options.messageId };
+    const emit = (record) => emitRecord(state, record, { sendReasoning, sendSources, onError });
+    // Phase 1: everything already stored, one value per query until an offset is empty; a superseded attempt is skipped whole.
+    const history = [];
+    for (;;) {
+        try {
+            history.push(await client.readStreamOffset(workflowID, key, state.offset + history.length, { timeoutSeconds: 0 }));
+        }
+        catch (error) {
+            if (!dbos_sdk_1.Error.isStreamTimeoutError(error))
+                throw error;
+            break;
+        }
+    }
+    const finalAttempt = new Map();
+    for (const record of history) {
+        if (record.kind === 'model' || record.kind === 'model-end')
+            finalAttempt.set(record.step, record.attempt);
+    }
+    for (const record of history) {
+        const stale = (record.kind === 'model' || record.kind === 'model-end') && finalAttempt.get(record.step) !== record.attempt;
+        yield* stale ? skipRecord(state) : emit(record);
+        if (state.ended)
+            return;
+    }
+    // Phase 2: live; a re-executed step shows up as a new attempt and is handed off in place.
+    for await (const record of client.readStream(workflowID, key, { offset: state.offset })) {
+        yield* emit(record);
+        if (state.ended)
+            return;
+    }
+    // No end record: the workflow's status decides how the turn ended (a stream closed while it still runs counts as finished).
+    const status = (await client.retrieveWorkflow(workflowID).getStatus())?.status;
+    yield* closeStep(state);
+    if (status === dbos_sdk_1.StatusString.CANCELLED) {
+        yield { type: 'abort' };
+    }
+    else if (status === undefined || status === dbos_sdk_1.StatusString.SUCCESS || status === dbos_sdk_1.StatusString.PENDING || status === dbos_sdk_1.StatusString.ENQUEUED) {
+        // The AI SDK's finish schema has no 'unknown'; a turn with no model call ends with the reason omitted.
+        yield state.finishReason === undefined ? { type: 'finish' } : { type: 'finish', finishReason: state.finishReason };
+    }
+    else {
+        const error = (await client.retrieveWorkflow(workflowID).getStatus())?.error;
+        yield { type: 'error', errorText: onError(error ?? new Error('The workflow ended before the response completed.')) };
+        yield { type: 'finish', finishReason: 'error' };
+    }
+}
+function offsetChunk(state) {
+    return { type: 'data-dbos-offset', data: { offset: state.offset }, transient: true };
+}
+function* skipRecord(state) {
+    state.offset += 1;
+    yield offsetChunk(state);
+}
+function* closeStep(state) {
+    if (state.openStep !== undefined)
+        yield { type: 'finish-step' };
+    state.openStep = undefined;
+    state.openAttempt = undefined;
+    state.openParts.clear();
+}
+// A live re-execution of the open step: end the stale attempt's parts and tell the client which ones to discard.
+function* supersede(state, attempt) {
+    for (const [id, kind] of state.openParts)
+        yield { type: kind === 'text' ? 'text-end' : 'reasoning-end', id };
+    yield {
+        type: 'data-dbos-superseded',
+        data: { attempt: state.openAttempt, parts: [...state.openParts.keys()] },
+        transient: true,
+    };
+    state.openParts.clear();
+    state.openAttempt = attempt;
+}
+function* emitRecord(state, record, filter) {
+    state.offset += 1;
+    switch (record.kind) {
+        case 'model': {
+            if (state.openStep !== record.step) {
+                yield* closeStep(state);
+                if (!state.resumed)
+                    yield { type: 'start-step' };
+                state.resumed = false;
+                state.openStep = record.step;
+                state.openAttempt = record.attempt;
+            }
+            else if (state.openAttempt !== record.attempt) {
+                yield* supersede(state, record.attempt);
+            }
+            for (const part of record.parts) {
+                if (!filter.sendReasoning && part.type.startsWith('reasoning-'))
+                    continue;
+                if (!filter.sendSources && part.type === 'source')
+                    continue;
+                const chunk = toUIChunk(part, record.attempt);
+                if (!chunk)
+                    continue;
+                if (chunk.type === 'text-start' || chunk.type === 'reasoning-start')
+                    state.openParts.set(chunk.id, chunk.type === 'text-start' ? 'text' : 'reasoning');
+                if (chunk.type === 'text-end' || chunk.type === 'reasoning-end')
+                    state.openParts.delete(chunk.id);
+                yield chunk;
+            }
+            break;
+        }
+        case 'model-end':
+            // The stream outlives the call: the workflow may run more calls, so only its end (or closeDurableStream) ends the turn.
+            if (record.attempt === state.openAttempt)
+                state.finishReason = record.aborted ? 'other' : record.finishReason?.unified;
+            break;
+        case 'tool':
+            // Local tool errors are masked like the AI SDK does; provider-executed ones (in model records) pass through verbatim.
+            yield record.errorText !== undefined
+                ? { type: 'tool-output-error', toolCallId: record.toolCallId, errorText: filter.onError(new Error(record.errorText)) }
+                : { type: 'tool-output-available', toolCallId: record.toolCallId, output: record.output };
+            break;
+        case 'ui':
+            yield* record.chunks;
+            break;
+        case 'end':
+            // The terminal chunk comes last, so the offset goes out first.
+            yield offsetChunk(state);
+            yield* closeStep(state);
+            yield { type: 'finish', finishReason: record.finishReason };
+            state.ended = true;
+            return;
+    }
+    yield offsetChunk(state);
+}
+function parseInput(input) {
+    try {
+        return JSON.parse(input);
+    }
+    catch {
+        return input;
+    }
+}
+// Text and reasoning ids are only unique within one model call; the attempt id keeps calls, and re-executions, apart in one message.
+function toUIChunk(part, attempt) {
+    const id = 'id' in part ? `${attempt}:${part.id}` : '';
+    switch (part.type) {
+        case 'text-start':
+            return { type: 'text-start', id, providerMetadata: part.providerMetadata };
+        case 'text-delta':
+            return { type: 'text-delta', id, delta: part.delta, providerMetadata: part.providerMetadata };
+        case 'text-end':
+            return { type: 'text-end', id, providerMetadata: part.providerMetadata };
+        case 'reasoning-start':
+            return { type: 'reasoning-start', id, providerMetadata: part.providerMetadata };
+        case 'reasoning-delta':
+            return { type: 'reasoning-delta', id, delta: part.delta, providerMetadata: part.providerMetadata };
+        case 'reasoning-end':
+            return { type: 'reasoning-end', id, providerMetadata: part.providerMetadata };
+        case 'tool-input-start':
+            return {
+                type: 'tool-input-start',
+                toolCallId: part.id,
+                toolName: part.toolName,
+                providerExecuted: part.providerExecuted,
+                dynamic: part.dynamic,
+                title: part.title,
+                providerMetadata: part.providerMetadata,
+            };
+        case 'tool-input-delta':
+            return { type: 'tool-input-delta', toolCallId: part.id, inputTextDelta: part.delta };
+        case 'tool-call':
+            return {
+                type: 'tool-input-available',
+                toolCallId: part.toolCallId,
+                toolName: part.toolName,
+                input: parseInput(part.input),
+                providerExecuted: part.providerExecuted,
+                dynamic: part.dynamic,
+                providerMetadata: part.providerMetadata,
+            };
+        case 'tool-result':
+            return part.isError
+                ? { type: 'tool-output-error', toolCallId: part.toolCallId, errorText: JSON.stringify(part.result), providerExecuted: true, dynamic: part.dynamic }
+                : { type: 'tool-output-available', toolCallId: part.toolCallId, output: part.result, providerExecuted: true, dynamic: part.dynamic, preliminary: part.preliminary };
+        case 'source':
+            return part.sourceType === 'url'
+                ? { type: 'source-url', sourceId: part.id, url: part.url, title: part.title, providerMetadata: part.providerMetadata }
+                : { type: 'source-document', sourceId: part.id, mediaType: part.mediaType, title: part.title, filename: part.filename, providerMetadata: part.providerMetadata };
+        case 'file': {
+            const url = part.data.type === 'url' ? String(part.data.url) : part.data.type === 'data' ? `data:${part.mediaType};base64,${String(part.data.data)}` : undefined;
+            return url === undefined ? undefined : { type: 'file', url, mediaType: part.mediaType, providerMetadata: part.providerMetadata };
+        }
+        default:
+            return undefined;
+    }
+}
+//# sourceMappingURL=durable-stream.js.map

package/dist/durable-stream.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"durable-stream.js","sourceRoot":"","sources":["../src/durable-stream.ts"],"names":[],"mappings":";;;AAsBA,oDAIC;AA6FD,0CAGC;AAMD,gDAIC;AAGD,gDAIC;AA2BD,8CAYC;AAlLD,6CAAyC;AACzC,iDAA6E;AAqB7E,SAAgB,oBAAoB,CAAC,OAAyC;IAC5E,IAAI,OAAO,KAAK,SAAS;QAAE,OAAO,SAAS,CAAC;IAC5C,MAAM,MAAM,GAAG,OAAO,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,EAAE,GAAG,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC;IACxE,OAAO,EAAE,GAAG,EAAE,MAAM,CAAC,GAAG,EAAE,aAAa,EAAE,MAAM,CAAC,aAAa,IAAI,EAAE,EAAE,eAAe,EAAE,MAAM,CAAC,eAAe,IAAI,EAAE,EAAE,CAAC;AACvH,CAAC;AAED,SAAS,QAAQ;IACf,OAAO,EAAE,IAAI,EAAE,eAAI,CAAC,MAAM,IAAI,CAAC,CAAC,EAAE,OAAO,EAAE,eAAI,CAAC,UAAU,EAAE,cAAc,IAAI,CAAC,EAAE,CAAC;AACpF,CAAC;AAED,0GAA0G;AAC1G,SAAS,aAAa,CAAC,IAA+B;IACpD,QAAQ,IAAI,CAAC,IAAI,EAAE,CAAC;QAClB,KAAK,cAAc,CAAC;QACpB,KAAK,mBAAmB,CAAC;QACzB,KAAK,QAAQ,CAAC;QACd,KAAK,OAAO,CAAC;QACb,KAAK,KAAK;YACR,OAAO,KAAK,CAAC;QACf;YACE,OAAO,IAAI,CAAC;IAChB,CAAC;AACH,CAAC;AAED,gEAAgE;AAChE,SAAS,UAAU,CAAC,IAA+B;IACjD,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,IAAI,IAAI,CAAC,IAAI,CAAC,IAAI,KAAK,MAAM,IAAI,IAAI,CAAC,IAAI,CAAC,IAAI,YAAY,UAAU,EAAE,CAAC;QAC9F,OAAO,EAAE,GAAG,IAAI,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;IACnG,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAED,yHAAyH;AACzH,KAAK,UAAU,cAAc,CAAC,GAAW,EAAE,MAA2B;IACpE,KAAK,IAAI,OAAO,GAAG,CAAC,GAAI,OAAO,EAAE,EAAE,CAAC;QAClC,IAAI,CAAC;YACH,OAAO,MAAM,eAAI,CAAC,WAAW,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;QAC7C,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,OAAO,KAAK,CAAC;gBAAE,MAAM,KAAK,CAAC;YAC/B,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,GAAG,GAAG,OAAO,CAAC,CAAC,CAAC;QACrE,CAAC;IACH,CAAC;AACH,CAAC;AAED,wIAAwI;AACxI,MAAa,iBAAiB;IASC;IARrB,OAAO,GAAgC,EAAE,CAAC;IAC1C,KAAK,GAAkB,OAAO,CAAC,OAAO,EAAE,CAAC;IACzC,OAAO,CAAU;IACjB,KAAK,CAA4C;IACxC,IAAI,GAAG,eAAI,CAAC,MAAM,IAAI,CAAC,CAAC,CAAC;IAC1C,+GAA+G;IAC9F,OAAO,GAAG,IAAA,wBAAU,GAAE,CAAC;IAExC,YAA6B,MAA6B;QAA7B,WAAM,GAAN,MAAM,CAAuB;IAAG,CAAC;IAE9D,IAAI,CAAC,IAA+B;QAClC,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC;YAAE,OAAO;QACjC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC;QACpC,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,IAAI,IAAI,CAAC,MAAM,CAAC,aAAa;YAAE,IAAI,CAAC,KAAK,EAAE,CAAC;;YAC9D,IAAI,CAAC,KAAK,KAAK,UAAU,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,KAAK,EAAE,EAAE,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC;IAClF,CAAC;IAED,4IAA4I;IAC5I,KAAK,CAAC,GAAG,CAAC,OAA0E;QAClF,IAAI,CAAC,KAAK,EAAE,CAAC;QACb,IAAI,CAAC,KAAK,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,GAAG,OAAO,EAAE,CAAC,CAAC;QACtF,MAAM,IAAI,CAAC,KAAK,CAAC;QACjB,IAAI,IAAI,CAAC,OAAO,KAAK,SAAS;YAAE,MAAM,IAAI,CAAC,OAAO,CAAC;IACrD,CAAC;IAED,gJAAgJ;IAChJ,KAAK,CAAC,OAAO;QACX,IAAI,CAAC,KAAK,EAAE,CAAC;QACb,MAAM,IAAI,CAAC,KAAK,CAAC;IACnB,CAAC;IAEO,KAAK;QACX,IAAI,IAAI,CAAC,KAAK,KAAK,SAAS;YAAE,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACvD,IAAI,CAAC,KAAK,GAAG,SAAS,CAAC;QACvB,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO;QACtC,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC;QAC3B,IAAI,CAAC,OAAO,GAAG,EAAE,CAAC;QAClB,IAAI,CAAC,KAAK,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC;IAC/E,CAAC;IAED,iHAAiH;IACzG,KAAK,CAAC,MAA2B;QACvC,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK;aACpB,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC,IAAI,CAAC,OAAO,KAAK,SAAS,CAAC,CAAC,CAAC,cAAc,CAAC,IAAI,CAAC,MAAM,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;aAC9F,KAAK,CAAC,CAAC,KAAc,EAAE,EAAE;YACxB,IAAI,CAAC,OAAO,KAAK,KAAK,CAAC;QACzB,CAAC,CAAC,CAAC;IACP,CAAC;CACF;AAjDD,8CAiDC;AAED,0DAA0D;AAC1D,SAAgB,eAAe,CAAC,GAAW,EAAE,UAAkB,EAAE,OAAoD;IACnH,MAAM,MAAM,GAAwB,EAAE,IAAI,EAAE,MAAM,EAAE,GAAG,QAAQ,EAAE,EAAE,UAAU,EAAE,GAAG,OAAO,EAAE,CAAC;IAC5F,OAAO,eAAI,CAAC,WAAW,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;AACvC,CAAC;AAED;;;GAGG;AACH,SAAgB,kBAAkB,CAAC,GAAW,EAAE,MAAwB;IACtE,MAAM,MAAM,GAAG,eAAI,CAAC,UAAU,CAAC;IAC/B,MAAM,MAAM,GAAwB,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,cAAc,EAAE,MAAM,EAAE,CAAC;IAClH,OAAO,eAAI,CAAC,WAAW,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;AACvC,CAAC;AAED,0JAA0J;AACnJ,KAAK,UAAU,kBAAkB,CAAC,GAAW,EAAE,YAAY,GAAG,MAAM;IACzE,MAAM,MAAM,GAAwB,EAAE,IAAI,EAAE,KAAK,EAAE,YAAY,EAAE,CAAC;IAClE,MAAM,eAAI,CAAC,WAAW,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;IACpC,MAAM,eAAI,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;AAC9B,CAAC;AA0BD,0GAA0G;AAC1G,SAAgB,iBAAiB,CAAC,OAAiC;IACjE,MAAM,QAAQ,GAAG,QAAQ,CAAC,OAAO,CAAC,CAAC;IACnC,OAAO,IAAI,cAAc,CAAiB;QACxC,KAAK,CAAC,IAAI,CAAC,UAAU;YACnB,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;YAC9C,IAAI,IAAI;gBAAE,UAAU,CAAC,KAAK,EAAE,CAAC;;gBACxB,UAAU,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QACjC,CAAC;QACD,KAAK,CAAC,MAAM;YACV,MAAM,QAAQ,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QACnC,CAAC;KACF,CAAC,CAAC;AACL,CAAC;AAED,KAAK,SAAS,CAAC,CAAC,QAAQ,CAAC,OAAiC;IACxD,MAAM,EAAE,UAAU,EAAE,GAAG,EAAE,aAAa,GAAG,IAAI,EAAE,WAAW,GAAG,KAAK,EAAE,OAAO,GAAG,GAAG,EAAE,CAAC,oBAAoB,EAAE,GAAG,OAAO,CAAC;IACrH,MAAM,MAAM,GAAwB,OAAO,CAAC,MAAM,IAAI,eAAI,CAAC;IAC3D,MAAM,KAAK,GAAgB;QACzB,MAAM,EAAE,OAAO,CAAC,MAAM,IAAI,CAAC;QAC3B,OAAO,EAAE,CAAC,OAAO,CAAC,MAAM,IAAI,CAAC,CAAC,GAAG,CAAC;QAClC,SAAS,EAAE,IAAI,GAAG,EAAE;QACpB,KAAK,EAAE,KAAK;KACb,CAAC;IACF,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;QAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,SAAS,EAAE,OAAO,CAAC,SAAS,EAAE,CAAC;IAC9E,MAAM,IAAI,GAAG,CAAC,MAA2B,EAAE,EAAE,CAAC,UAAU,CAAC,KAAK,EAAE,MAAM,EAAE,EAAE,aAAa,EAAE,WAAW,EAAE,OAAO,EAAE,CAAC,CAAC;IAEjH,2HAA2H;IAC3H,MAAM,OAAO,GAA0B,EAAE,CAAC;IAC1C,SAAS,CAAC;QACR,IAAI,CAAC;YACH,OAAO,CAAC,IAAI,CAAC,MAAM,MAAM,CAAC,gBAAgB,CAAsB,UAAU,EAAE,GAAG,EAAE,KAAK,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,EAAE,EAAE,cAAc,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;QAC1I,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,gBAAU,CAAC,oBAAoB,CAAC,KAAK,CAAC;gBAAE,MAAM,KAAK,CAAC;YACzD,MAAM;QACR,CAAC;IACH,CAAC;IACD,MAAM,YAAY,GAAG,IAAI,GAAG,EAAkB,CAAC;IAC/C,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;QAC7B,IAAI,MAAM,CAAC,IAAI,KAAK,OAAO,IAAI,MAAM,CAAC,IAAI,KAAK,WAAW;YAAE,YAAY,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC;IAC5G,CAAC;IACD,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;QAC7B,MAAM,KAAK,GAAG,CAAC,MAAM,CAAC,IAAI,KAAK,OAAO,IAAI,MAAM,CAAC,IAAI,KAAK,WAAW,CAAC,IAAI,YAAY,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,MAAM,CAAC,OAAO,CAAC;QAC3H,KAAK,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAChD,IAAI,KAAK,CAAC,KAAK;YAAE,OAAO;IAC1B,CAAC;IAED,0FAA0F;IAC1F,IAAI,KAAK,EAAE,MAAM,MAAM,IAAI,MAAM,CAAC,UAAU,CAAsB,UAAU,EAAE,GAAG,EAAE,EAAE,MAAM,EAAE,KAAK,CAAC,MAAM,EAAE,CAAC,EAAE,CAAC;QAC7G,KAAK,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACpB,IAAI,KAAK,CAAC,KAAK;YAAE,OAAO;IAC1B,CAAC;IAED,4HAA4H;IAC5H,MAAM,MAAM,GAAG,CAAC,MAAM,MAAM,CAAC,gBAAgB,CAAC,UAAU,CAAC,CAAC,SAAS,EAAE,CAAC,EAAE,MAAM,CAAC;IAC/E,KAAK,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;IACxB,IAAI,MAAM,KAAK,uBAAY,CAAC,SAAS,EAAE,CAAC;QACtC,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;IAC1B,CAAC;SAAM,IAAI,MAAM,KAAK,SAAS,IAAI,MAAM,KAAK,uBAAY,CAAC,OAAO,IAAI,MAAM,KAAK,uBAAY,CAAC,OAAO,IAAI,MAAM,KAAK,uBAAY,CAAC,QAAQ,EAAE,CAAC;QAC1I,uGAAuG;QACvG,MAAM,KAAK,CAAC,YAAY,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,YAAY,EAAE,KAAK,CAAC,YAA8B,EAAE,CAAC;IACvI,CAAC;SAAM,CAAC;QACN,MAAM,KAAK,GAAG,CAAC,MAAM,MAAM,CAAC,gBAAgB,CAAC,UAAU,CAAC,CAAC,SAAS,EAAE,CAAC,EAAE,KAAK,CAAC;QAC7E,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,SAAS,EAAE,OAAO,CAAC,KAAK,IAAI,IAAI,KAAK,CAAC,mDAAmD,CAAC,CAAC,EAAE,CAAC;QACrH,MAAM,EAAE,IAAI,EAAE,QAAQ,EAAE,YAAY,EAAE,OAAO,EAAE,CAAC;IAClD,CAAC;AACH,CAAC;AAaD,SAAS,WAAW,CAAC,KAAkB;IACrC,OAAO,EAAE,IAAI,EAAE,kBAAkB,EAAE,IAAI,EAAE,EAAE,MAAM,EAAE,KAAK,CAAC,MAAM,EAAE,EAAE,SAAS,EAAE,IAAI,EAAoB,CAAC;AACzG,CAAC;AAED,QAAQ,CAAC,CAAC,UAAU,CAAC,KAAkB;IACrC,KAAK,CAAC,MAAM,IAAI,CAAC,CAAC;IAClB,MAAM,WAAW,CAAC,KAAK,CAAC,CAAC;AAC3B,CAAC;AAED,QAAQ,CAAC,CAAC,SAAS,CAAC,KAAkB;IACpC,IAAI,KAAK,CAAC,QAAQ,KAAK,SAAS;QAAE,MAAM,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC;IAChE,KAAK,CAAC,QAAQ,GAAG,SAAS,CAAC;IAC3B,KAAK,CAAC,WAAW,GAAG,SAAS,CAAC;IAC9B,KAAK,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC;AAC1B,CAAC;AAED,iHAAiH;AACjH,QAAQ,CAAC,CAAC,SAAS,CAAC,KAAkB,EAAE,OAAe;IACrD,KAAK,MAAM,CAAC,EAAE,EAAE,IAAI,CAAC,IAAI,KAAK,CAAC,SAAS;QAAE,MAAM,EAAE,IAAI,EAAE,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,eAAe,EAAE,EAAE,EAAE,CAAC;IAC7G,MAAM;QACJ,IAAI,EAAE,sBAAsB;QAC5B,IAAI,EAAE,EAAE,OAAO,EAAE,KAAK,CAAC,WAAW,EAAE,KAAK,EAAE,CAAC,GAAG,KAAK,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC,EAAE;QACxE,SAAS,EAAE,IAAI;KACE,CAAC;IACpB,KAAK,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC;IACxB,KAAK,CAAC,WAAW,GAAG,OAAO,CAAC;AAC9B,CAAC;AAED,QAAQ,CAAC,CAAC,UAAU,CAClB,KAAkB,EAClB,MAA2B,EAC3B,MAA6F;IAE7F,KAAK,CAAC,MAAM,IAAI,CAAC,CAAC;IAClB,QAAQ,MAAM,CAAC,IAAI,EAAE,CAAC;QACpB,KAAK,OAAO,CAAC,CAAC,CAAC;YACb,IAAI,KAAK,CAAC,QAAQ,KAAK,MAAM,CAAC,IAAI,EAAE,CAAC;gBACnC,KAAK,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;gBACxB,IAAI,CAAC,KAAK,CAAC,OAAO;oBAAE,MAAM,EAAE,IAAI,EAAE,YAAY,EAAE,CAAC;gBACjD,KAAK,CAAC,OAAO,GAAG,KAAK,CAAC;gBACtB,KAAK,CAAC,QAAQ,GAAG,MAAM,CAAC,IAAI,CAAC;gBAC7B,KAAK,CAAC,WAAW,GAAG,MAAM,CAAC,OAAO,CAAC;YACrC,CAAC;iBAAM,IAAI,KAAK,CAAC,WAAW,KAAK,MAAM,CAAC,OAAO,EAAE,CAAC;gBAChD,KAAK,CAAC,CAAC,SAAS,CAAC,KAAK,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC;YAC1C,CAAC;YACD,KAAK,MAAM,IAAI,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;gBAChC,IAAI,CAAC,MAAM,CAAC,aAAa,IAAI,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,YAAY,CAAC;oBAAE,SAAS;gBAC1E,IAAI,CAAC,MAAM,CAAC,WAAW,IAAI,IAAI,CAAC,IAAI,KAAK,QAAQ;oBAAE,SAAS;gBAC5D,MAAM,KAAK,GAAG,SAAS,CAAC,IAAI,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC;gBAC9C,IAAI,CAAC,KAAK;oBAAE,SAAS;gBACrB,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,IAAI,KAAK,CAAC,IAAI,KAAK,iBAAiB;oBAAE,KAAK,CAAC,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,EAAE,KAAK,CAAC,IAAI,KAAK,YAAY,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC;gBACvJ,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU,IAAI,KAAK,CAAC,IAAI,KAAK,eAAe;oBAAE,KAAK,CAAC,SAAS,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;gBAClG,MAAM,KAAK,CAAC;YACd,CAAC;YACD,MAAM;QACR,CAAC;QACD,KAAK,WAAW;YACd,wHAAwH;YACxH,IAAI,MAAM,CAAC,OAAO,KAAK,KAAK,CAAC,WAAW;gBAAE,KAAK,CAAC,YAAY,GAAG,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,YAAY,EAAE,OAAO,CAAC;YACvH,MAAM;QACR,KAAK,MAAM;YACT,sHAAsH;YACtH,MAAM,MAAM,CAAC,SAAS,KAAK,SAAS;gBAClC,CAAC,CAAC,EAAE,IAAI,EAAE,mBAAmB,EAAE,UAAU,EAAE,MAAM,CAAC,UAAU,EAAE,SAAS,EAAE,MAAM,CAAC,OAAO,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,EAAE;gBACtH,CAAC,CAAC,EAAE,IAAI,EAAE,uBAAuB,EAAE,UAAU,EAAE,MAAM,CAAC,UAAU,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,CAAC;YAC5F,MAAM;QACR,KAAK,IAAI;YACP,KAAK,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC;YACrB,MAAM;QACR,KAAK,KAAK;YACR,+DAA+D;YAC/D,MAAM,WAAW,CAAC,KAAK,CAAC,CAAC;YACzB,KAAK,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;YACxB,MAAM,EAAE,IAAI,EAAE,QAAQ,EAAE,YAAY,EAAE,MAAM,CAAC,YAA8B,EAAE,CAAC;YAC9E,KAAK,CAAC,KAAK,GAAG,IAAI,CAAC;YACnB,OAAO;IACX,CAAC;IACD,MAAM,WAAW,CAAC,KAAK,CAAC,CAAC;AAC3B,CAAC;AAID,SAAS,UAAU,CAAC,KAAa;IAC/B,IAAI,CAAC;QACH,OAAO,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;IAC3B,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,KAAK,CAAC;IACf,CAAC;AACH,CAAC;AAED,qIAAqI;AACrI,SAAS,SAAS,CAAC,IAA+B,EAAE,OAAe;IACjE,MAAM,EAAE,GAAG,IAAI,IAAI,IAAI,CAAC,CAAC,CAAC,GAAG,OAAO,IAAI,IAAI,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;IACvD,QAAQ,IAAI,CAAC,IAAI,EAAE,CAAC;QAClB,KAAK,YAAY;YACf,OAAO,EAAE,IAAI,EAAE,YAAY,EAAE,EAAE,EAAE,gBAAgB,EAAE,IAAI,CAAC,gBAAgB,EAAE,CAAC;QAC7E,KAAK,YAAY;YACf,OAAO,EAAE,IAAI,EAAE,YAAY,EAAE,EAAE,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,gBAAgB,EAAE,IAAI,CAAC,gBAAgB,EAAE,CAAC;QAChG,KAAK,UAAU;YACb,OAAO,EAAE,IAAI,EAAE,UAAU,EAAE,EAAE,EAAE,gBAAgB,EAAE,IAAI,CAAC,gBAAgB,EAAE,CAAC;QAC3E,KAAK,iBAAiB;YACpB,OAAO,EAAE,IAAI,EAAE,iBAAiB,EAAE,EAAE,EAAE,gBAAgB,EAAE,IAAI,CAAC,gBAAgB,EAAE,CAAC;QAClF,KAAK,iBAAiB;YACpB,OAAO,EAAE,IAAI,EAAE,iBAAiB,EAAE,EAAE,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,gBAAgB,EAAE,IAAI,CAAC,gBAAgB,EAAE,CAAC;QACrG,KAAK,eAAe;YAClB,OAAO,EAAE,IAAI,EAAE,eAAe,EAAE,EAAE,EAAE,gBAAgB,EAAE,IAAI,CAAC,gBAAgB,EAAE,CAAC;QAChF,KAAK,kBAAkB;YACrB,OAAO;gBACL,IAAI,EAAE,kBAAkB;gBACxB,UAAU,EAAE,IAAI,CAAC,EAAE;gBACnB,QAAQ,EAAE,IAAI,CAAC,QAAQ;gBACvB,gBAAgB,EAAE,IAAI,CAAC,gBAAgB;gBACvC,OAAO,EAAE,IAAI,CAAC,OAAO;gBACrB,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,gBAAgB,EAAE,IAAI,CAAC,gBAAgB;aACxC,CAAC;QACJ,KAAK,kBAAkB;YACrB,OAAO,EAAE,IAAI,EAAE,kBAAkB,EAAE,UAAU,EAAE,IAAI,CAAC,EAAE,EAAE,cAAc,EAAE,IAAI,CAAC,KAAK,EAAE,CAAC;QACvF,KAAK,WAAW;YACd,OAAO;gBACL,IAAI,EAAE,sBAAsB;gBAC5B,UAAU,EAAE,IAAI,CAAC,UAAU;gBAC3B,QAAQ,EAAE,IAAI,CAAC,QAAQ;gBACvB,KAAK,EAAE,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC;gBAC7B,gBAAgB,EAAE,IAAI,CAAC,gBAAgB;gBACvC,OAAO,EAAE,IAAI,CAAC,OAAO;gBACrB,gBAAgB,EAAE,IAAI,CAAC,gBAAgB;aACxC,CAAC;QACJ,KAAK,aAAa;YAChB,OAAO,IAAI,CAAC,OAAO;gBACjB,CAAC,CAAC,EAAE,IAAI,EAAE,mBAAmB,EAAE,UAAU,EAAE,IAAI,CAAC,UAAU,EAAE,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,gBAAgB,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE;gBACnJ,CAAC,CAAC,EAAE,IAAI,EAAE,uBAAuB,EAAE,UAAU,EAAE,IAAI,CAAC,UAAU,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,gBAAgB,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,WAAW,EAAE,IAAI,CAAC,WAAW,EAAE,CAAC;QACxK,KAAK,QAAQ;YACX,OAAO,IAAI,CAAC,UAAU,KAAK,KAAK;gBAC9B,CAAC,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,QAAQ,EAAE,IAAI,CAAC,EAAE,EAAE,GAAG,EAAE,IAAI,CAAC,GAAG,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,gBAAgB,EAAE,IAAI,CAAC,gBAAgB,EAAE;gBACtH,CAAC,CAAC,EAAE,IAAI,EAAE,iBAAiB,EAAE,QAAQ,EAAE,IAAI,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,CAAC,SAAS,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,QAAQ,EAAE,IAAI,CAAC,QAAQ,EAAE,gBAAgB,EAAE,IAAI,CAAC,gBAAgB,EAAE,CAAC;QACrK,KAAK,MAAM,CAAC,CAAC,CAAC;YACZ,MAAM,GAAG,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,KAAK,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,QAAQ,IAAI,CAAC,SAAS,WAAW,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;YACjK,OAAO,GAAG,KAAK,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,GAAG,EAAE,SAAS,EAAE,IAAI,CAAC,SAAS,EAAE,gBAAgB,EAAE,IAAI,CAAC,gBAAgB,EAAE,CAAC;QACnI,CAAC;QACD;YACE,OAAO,SAAS,CAAC;IACrB,CAAC;AACH,CAAC"}

package/dist/index.d.ts

@@ -1,3 +1,6 @@
-export { durableCalls, durableEmbeddingCalls, durableImageCalls } from './middleware';
+export { durableCalls, DurableCallsOptions, durableEmbeddingCalls, durableImageCalls } from './middleware';
 export { durableMCPTools, DurableMCPToolsOptions, MCPClientLike } from './mcp';
+export { durableTools, DurableToolsOptions } from './tools';
+export { closeDurableStream, DurableStreamOptions, DurableStreamRecord, DurableStreamSource, ReadDurableStreamOptions, readDurableStream, writeDurableStream, } from './durable-stream';
+export { agentTool, AgentTool, AgentToolOptions } from './agent-tool';
 //# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,qBAAqB,EAAE,iBAAiB,EAAE,MAAM,cAAc,CAAC;AACtF,OAAO,EAAE,eAAe,EAAE,sBAAsB,EAAE,aAAa,EAAE,MAAM,OAAO,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,mBAAmB,EAAE,qBAAqB,EAAE,iBAAiB,EAAE,MAAM,cAAc,CAAC;AAC3G,OAAO,EAAE,eAAe,EAAE,sBAAsB,EAAE,aAAa,EAAE,MAAM,OAAO,CAAC;AAC/E,OAAO,EAAE,YAAY,EAAE,mBAAmB,EAAE,MAAM,SAAS,CAAC;AAC5D,OAAO,EACL,kBAAkB,EAClB,oBAAoB,EACpB,mBAAmB,EACnB,mBAAmB,EACnB,wBAAwB,EACxB,iBAAiB,EACjB,kBAAkB,GACnB,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EAAE,SAAS,EAAE,SAAS,EAAE,gBAAgB,EAAE,MAAM,cAAc,CAAC"}

package/dist/index.js

@@ -1,10 +1,18 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.durableMCPTools = exports.durableImageCalls = exports.durableEmbeddingCalls = exports.durableCalls = void 0;
+exports.agentTool = exports.writeDurableStream = exports.readDurableStream = exports.closeDurableStream = exports.durableTools = exports.durableMCPTools = exports.durableImageCalls = exports.durableEmbeddingCalls = exports.durableCalls = void 0;
 var middleware_1 = require("./middleware");
 Object.defineProperty(exports, "durableCalls", { enumerable: true, get: function () { return middleware_1.durableCalls; } });
 Object.defineProperty(exports, "durableEmbeddingCalls", { enumerable: true, get: function () { return middleware_1.durableEmbeddingCalls; } });
 Object.defineProperty(exports, "durableImageCalls", { enumerable: true, get: function () { return middleware_1.durableImageCalls; } });
 var mcp_1 = require("./mcp");
 Object.defineProperty(exports, "durableMCPTools", { enumerable: true, get: function () { return mcp_1.durableMCPTools; } });
+var tools_1 = require("./tools");
+Object.defineProperty(exports, "durableTools", { enumerable: true, get: function () { return tools_1.durableTools; } });
+var durable_stream_1 = require("./durable-stream");
+Object.defineProperty(exports, "closeDurableStream", { enumerable: true, get: function () { return durable_stream_1.closeDurableStream; } });
+Object.defineProperty(exports, "readDurableStream", { enumerable: true, get: function () { return durable_stream_1.readDurableStream; } });
+Object.defineProperty(exports, "writeDurableStream", { enumerable: true, get: function () { return durable_stream_1.writeDurableStream; } });
+var agent_tool_1 = require("./agent-tool");
+Object.defineProperty(exports, "agentTool", { enumerable: true, get: function () { return agent_tool_1.agentTool; } });
 //# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";;;AAAA,2CAAsF;AAA7E,0GAAA,YAAY,OAAA;AAAE,mHAAA,qBAAqB,OAAA;AAAE,+GAAA,iBAAiB,OAAA;AAC/D,6BAA+E;AAAtE,sGAAA,eAAe,OAAA"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";;;AAAA,2CAA2G;AAAlG,0GAAA,YAAY,OAAA;AAAuB,mHAAA,qBAAqB,OAAA;AAAE,+GAAA,iBAAiB,OAAA;AACpF,6BAA+E;AAAtE,sGAAA,eAAe,OAAA;AACxB,iCAA4D;AAAnD,qGAAA,YAAY,OAAA;AACrB,mDAQ0B;AAPxB,oHAAA,kBAAkB,OAAA;AAKlB,mHAAA,iBAAiB,OAAA;AACjB,oHAAA,kBAAkB,OAAA;AAEpB,2CAAsE;AAA7D,uGAAA,SAAS,OAAA"}

package/dist/internal.d.ts

@@ -1,6 +1,8 @@
 import { StepConfig } from '@dbos-inc/dbos-sdk';
 export declare function isInWorkflowFunction(): boolean;
 export declare function assertNotInTransaction(operation: string): void;
+export declare function runDurableStep<T>(name: string, fn: () => Promise<T>, config: StepConfig): Promise<T>;
+export declare function isAsyncIterable(value: unknown): value is AsyncIterable<unknown>;
 export declare function restoreAISDKErrorIdentity(error: unknown): unknown;
 export declare function withErrorClassification(options: StepConfig): StepConfig;
 //# sourceMappingURL=internal.d.ts.map

package/dist/internal.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"internal.d.ts","sourceRoot":"","sources":["../src/internal.ts"],"names":[],"mappings":"AAAA,OAAO,EAAQ,UAAU,EAAE,MAAM,oBAAoB,CAAC;AAEtD,wBAAgB,oBAAoB,IAAI,OAAO,CAG9C;AAED,wBAAgB,sBAAsB,CAAC,SAAS,EAAE,MAAM,GAAG,IAAI,CAI9D;AA2BD,wBAAgB,yBAAyB,CAAC,KAAK,EAAE,OAAO,GAAG,OAAO,CAejE;AAKD,wBAAgB,uBAAuB,CAAC,OAAO,EAAE,UAAU,GAAG,UAAU,CAMvE"}
+{"version":3,"file":"internal.d.ts","sourceRoot":"","sources":["../src/internal.ts"],"names":[],"mappings":"AAAA,OAAO,EAAQ,UAAU,EAAE,MAAM,oBAAoB,CAAC;AAEtD,wBAAgB,oBAAoB,IAAI,OAAO,CAG9C;AAED,wBAAgB,sBAAsB,CAAC,SAAS,EAAE,MAAM,GAAG,IAAI,CAI9D;AAGD,wBAAgB,cAAc,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,OAAO,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,UAAU,GAAG,OAAO,CAAC,CAAC,CAAC,CAOpG;AAED,wBAAgB,eAAe,CAAC,KAAK,EAAE,OAAO,GAAG,KAAK,IAAI,aAAa,CAAC,OAAO,CAAC,CAE/E;AA2BD,wBAAgB,yBAAyB,CAAC,KAAK,EAAE,OAAO,GAAG,OAAO,CAejE;AAKD,wBAAgB,uBAAuB,CAAC,OAAO,EAAE,UAAU,GAAG,UAAU,CAMvE"}

package/dist/internal.js

@@ -2,6 +2,8 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.isInWorkflowFunction = isInWorkflowFunction;
 exports.assertNotInTransaction = assertNotInTransaction;
+exports.runDurableStep = runDurableStep;
+exports.isAsyncIterable = isAsyncIterable;
 exports.restoreAISDKErrorIdentity = restoreAISDKErrorIdentity;
 exports.withErrorClassification = withErrorClassification;
 const dbos_sdk_1 = require("@dbos-inc/dbos-sdk");
@@ -14,6 +16,19 @@ function assertNotInTransaction(operation) {
         throw new Error(`Cannot call ${operation} inside a DBOS transaction; run it in workflow or step code.`);
     }
 }
+// Run fn as a durable step named `name` inside a workflow; elsewhere call it directly.
+function runDurableStep(name, fn, config) {
+    assertNotInTransaction(name);
+    if (!isInWorkflowFunction())
+        return fn();
+    // Restore the AI SDK error identity a replay revival strips, so the SDK's retry/catch logic behaves the same.
+    return dbos_sdk_1.DBOS.runStep(fn, { ...config, name }).catch((error) => {
+        throw restoreAISDKErrorIdentity(error);
+    });
+}
+function isAsyncIterable(value) {
+    return typeof value?.[Symbol.asyncIterator] === 'function';
+}
 // Aborts/timeouts are deliberate cancellations, never transient; retrying just re-runs an already-cancelled call.
 function isAbortError(error) {
     const name = error?.name;

package/dist/internal.js.map

@@ -1 +1 @@
-{"version":3,"file":"internal.js","sourceRoot":"","sources":["../src/internal.ts"],"names":[],"mappings":";;AAEA,oDAGC;AAED,wDAIC;AA2BD,8DAeC;AAKD,0DAMC;AAhED,iDAAsD;AAEtD,SAAgB,oBAAoB;IAClC,kFAAkF;IAClF,OAAO,eAAI,CAAC,YAAY,EAAE,CAAC;AAC7B,CAAC;AAED,SAAgB,sBAAsB,CAAC,SAAiB;IACtD,IAAI,eAAI,CAAC,eAAe,EAAE,EAAE,CAAC;QAC3B,MAAM,IAAI,KAAK,CAAC,eAAe,SAAS,8DAA8D,CAAC,CAAC;IAC1G,CAAC;AACH,CAAC;AAED,kHAAkH;AAClH,SAAS,YAAY,CAAC,KAAc;IAClC,MAAM,IAAI,GAAI,KAAmC,EAAE,IAAI,CAAC;IACxD,OAAO,IAAI,KAAK,YAAY,IAAI,IAAI,KAAK,cAAc,CAAC;AAC1D,CAAC;AAED,8HAA8H;AAC9H,SAAS,cAAc,CAAC,KAAc;IACpC,IAAI,CAAC;QACH,OAAO,CACL,YAAY,CAAC,KAAK,CAAC;YACnB,CAAC,OAAO,KAAK,KAAK,QAAQ;gBACxB,KAAK,KAAK,IAAI;gBACd,aAAa,IAAI,KAAK;gBACrB,KAAmC,CAAC,WAAW,KAAK,KAAK,CAAC,CAC9D,CAAC;IACJ,CAAC;IAAC,MAAM,CAAC;QACP,kFAAkF;QAClF,OAAO,KAAK,CAAC;IACf,CAAC;AACH,CAAC;AAED,qHAAqH;AACrH,sHAAsH;AACtH,qHAAqH;AACrH,SAAgB,yBAAyB,CAAC,KAAc;IACtD,+GAA+G;IAC/G,iHAAiH;IACjH,IAAI,CAAC;QACH,MAAM,IAAI,GAAI,KAAmC,EAAE,IAAI,CAAC;QACxD,IAAI,KAAK,KAAK,IAAI,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,OAAO,IAAI,KAAK,QAAQ;YAAE,OAAO,KAAK,CAAC;QAC1F,MAAM,IAAI,GAAG,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,iBAAiB,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,yBAAyB,CAAC,CAAC,CAAC,SAAS,CAAC;QAC7H,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;YACtB,KAAiC,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,GAAG,IAAI,CAAC;YAC3D,KAAiC,CAAC,MAAM,CAAC,GAAG,CAAC,GAAG,IAAI,IAAI,IAAI,EAAE,CAAC,CAAC,GAAG,IAAI,CAAC;QAC3E,CAAC;IACH,CAAC;IAAC,MAAM,CAAC;QACP,2CAA2C;IAC7C,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC;AAED,oHAAoH;AACpH,oHAAoH;AACpH,qGAAqG;AACrG,SAAgB,uBAAuB,CAAC,OAAmB;IACzD,OAAO;QACL,GAAG,OAAO;QACV,cAAc,EAAE,OAAO,CAAC,cAAc,IAAI,IAAI;QAC9C,WAAW,EAAE,OAAO,CAAC,WAAW,IAAI,CAAC,CAAC,KAAc,EAAE,EAAE,CAAC,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC;KACjF,CAAC;AACJ,CAAC"}
+{"version":3,"file":"internal.js","sourceRoot":"","sources":["../src/internal.ts"],"names":[],"mappings":";;AAEA,oDAGC;AAED,wDAIC;AAGD,wCAOC;AAED,0CAEC;AA2BD,8DAeC;AAKD,0DAMC;AA9ED,iDAAsD;AAEtD,SAAgB,oBAAoB;IAClC,kFAAkF;IAClF,OAAO,eAAI,CAAC,YAAY,EAAE,CAAC;AAC7B,CAAC;AAED,SAAgB,sBAAsB,CAAC,SAAiB;IACtD,IAAI,eAAI,CAAC,eAAe,EAAE,EAAE,CAAC;QAC3B,MAAM,IAAI,KAAK,CAAC,eAAe,SAAS,8DAA8D,CAAC,CAAC;IAC1G,CAAC;AACH,CAAC;AAED,uFAAuF;AACvF,SAAgB,cAAc,CAAI,IAAY,EAAE,EAAoB,EAAE,MAAkB;IACtF,sBAAsB,CAAC,IAAI,CAAC,CAAC;IAC7B,IAAI,CAAC,oBAAoB,EAAE;QAAE,OAAO,EAAE,EAAE,CAAC;IACzC,8GAA8G;IAC9G,OAAO,eAAI,CAAC,OAAO,CAAC,EAAE,EAAE,EAAE,GAAG,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,KAAc,EAAE,EAAE;QACpE,MAAM,yBAAyB,CAAC,KAAK,CAAC,CAAC;IACzC,CAAC,CAAC,CAAC;AACL,CAAC;AAED,SAAgB,eAAe,CAAC,KAAc;IAC5C,OAAO,OAAQ,KAAmD,EAAE,CAAC,MAAM,CAAC,aAAa,CAAC,KAAK,UAAU,CAAC;AAC5G,CAAC;AAED,kHAAkH;AAClH,SAAS,YAAY,CAAC,KAAc;IAClC,MAAM,IAAI,GAAI,KAAmC,EAAE,IAAI,CAAC;IACxD,OAAO,IAAI,KAAK,YAAY,IAAI,IAAI,KAAK,cAAc,CAAC;AAC1D,CAAC;AAED,8HAA8H;AAC9H,SAAS,cAAc,CAAC,KAAc;IACpC,IAAI,CAAC;QACH,OAAO,CACL,YAAY,CAAC,KAAK,CAAC;YACnB,CAAC,OAAO,KAAK,KAAK,QAAQ;gBACxB,KAAK,KAAK,IAAI;gBACd,aAAa,IAAI,KAAK;gBACrB,KAAmC,CAAC,WAAW,KAAK,KAAK,CAAC,CAC9D,CAAC;IACJ,CAAC;IAAC,MAAM,CAAC;QACP,kFAAkF;QAClF,OAAO,KAAK,CAAC;IACf,CAAC;AACH,CAAC;AAED,qHAAqH;AACrH,sHAAsH;AACtH,qHAAqH;AACrH,SAAgB,yBAAyB,CAAC,KAAc;IACtD,+GAA+G;IAC/G,iHAAiH;IACjH,IAAI,CAAC;QACH,MAAM,IAAI,GAAI,KAAmC,EAAE,IAAI,CAAC;QACxD,IAAI,KAAK,KAAK,IAAI,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,OAAO,IAAI,KAAK,QAAQ;YAAE,OAAO,KAAK,CAAC;QAC1F,MAAM,IAAI,GAAG,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,iBAAiB,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,yBAAyB,CAAC,CAAC,CAAC,SAAS,CAAC;QAC7H,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;YACtB,KAAiC,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,GAAG,IAAI,CAAC;YAC3D,KAAiC,CAAC,MAAM,CAAC,GAAG,CAAC,GAAG,IAAI,IAAI,IAAI,EAAE,CAAC,CAAC,GAAG,IAAI,CAAC;QAC3E,CAAC;IACH,CAAC;IAAC,MAAM,CAAC;QACP,2CAA2C;IAC7C,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC;AAED,oHAAoH;AACpH,oHAAoH;AACpH,qGAAqG;AACrG,SAAgB,uBAAuB,CAAC,OAAmB;IACzD,OAAO;QACL,GAAG,OAAO;QACV,cAAc,EAAE,OAAO,CAAC,cAAc,IAAI,IAAI;QAC9C,WAAW,EAAE,OAAO,CAAC,WAAW,IAAI,CAAC,CAAC,KAAc,EAAE,EAAE,CAAC,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC;KACjF,CAAC;AACJ,CAAC"}

package/dist/mcp.d.ts

@@ -7,6 +7,8 @@ export interface MCPClientLike {
 export interface DurableMCPToolsOptions extends StepConfig {
     /** Forwarded to client.tools() on listing and on each call (e.g. { schemas } for subsetting and output schemas). */
     toolOptions?: unknown;
+    /** Write each tool call's output (or error) to this durable stream from inside its step. */
+    durableStream?: string;
 }
 /**
  * Wraps an MCP client so its tool listing and each tool call run as durable DBOS steps: the tool

package/dist/mcp.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"mcp.d.ts","sourceRoot":"","sources":["../src/mcp.ts"],"names":[],"mappings":"AAAA,OAAO,EAAQ,UAAU,EAAE,MAAM,oBAAoB,CAAC;AACtD,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,IAAI,CAAC,OAAO,iBAAiB,EAAE,QAAQ,EAAE,CAAC;AAKvE,MAAM,WAAW,aAAa;IAC5B,KAAK,CAAC,OAAO,CAAC,EAAE,OAAO,GAAG,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC;IAC3D,KAAK,CAAC,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;CACzB;AAaD,MAAM,WAAW,sBAAuB,SAAQ,UAAU;IACxD,oHAAoH;IACpH,WAAW,CAAC,EAAE,OAAO,CAAC;CACvB;AAqCD;;;;GAIG;AACH,wBAAsB,eAAe,CAAC,MAAM,EAAE,aAAa,EAAE,OAAO,GAAE,sBAA2B,GAAG,OAAO,CAAC,OAAO,CAAC,CAwEnH"}
+{"version":3,"file":"mcp.d.ts","sourceRoot":"","sources":["../src/mcp.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,oBAAoB,CAAC;AAChD,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,IAAI,CAAC,OAAO,iBAAiB,EAAE,QAAQ,EAAE,CAAC;AAMvE,MAAM,WAAW,aAAa;IAC5B,KAAK,CAAC,OAAO,CAAC,EAAE,OAAO,GAAG,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC;IAC3D,KAAK,CAAC,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;CACzB;AAaD,MAAM,WAAW,sBAAuB,SAAQ,UAAU;IACxD,oHAAoH;IACpH,WAAW,CAAC,EAAE,OAAO,CAAC;IACtB,4FAA4F;IAC5F,aAAa,CAAC,EAAE,MAAM,CAAC;CACxB;AAiCD;;;;GAIG;AACH,wBAAsB,eAAe,CAAC,MAAM,EAAE,aAAa,EAAE,OAAO,GAAE,sBAA2B,GAAG,OAAO,CAAC,OAAO,CAAC,CA4EnH"}

package/dist/mcp.js

@@ -1,11 +1,8 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.durableMCPTools = durableMCPTools;
-const dbos_sdk_1 = require("@dbos-inc/dbos-sdk");
 const internal_1 = require("./internal");
-function isAsyncIterable(value) {
-    return typeof value?.[Symbol.asyncIterator] === 'function';
-}
+const durable_stream_1 = require("./durable-stream");
 // Mirror of @ai-sdk/mcp's toModelOutput: MCP content becomes model content (text stays text, images become files).
 function mcpToolOutput(output) {
     const result = output;
@@ -31,18 +28,10 @@ function mcpToolOutput(output) {
  * checkpointed so a recovered workflow replays results instead of re-invoking the tool.
  */
 async function durableMCPTools(client, options = {}) {
-    const { toolOptions, ...stepOptions } = options;
+    const { toolOptions, durableStream, ...stepOptions } = options;
     const stepConfig = (0, internal_1.withErrorClassification)(stepOptions);
     const { asSchema, dynamicTool, jsonSchema } = await import('ai');
-    const run = (name, fn, config = stepConfig) => {
-        (0, internal_1.assertNotInTransaction)(name);
-        if (!(0, internal_1.isInWorkflowFunction)())
-            return fn();
-        // Restore the AI SDK error identity a replay revival strips, so the SDK's retry/catch logic behaves the same.
-        return dbos_sdk_1.DBOS.runStep(fn, { ...config, name }).catch((error) => {
-            throw (0, internal_1.restoreAISDKErrorIdentity)(error);
-        });
-    };
+    const run = (name, fn, config = stepConfig) => (0, internal_1.runDurableStep)(name, fn, config);
     // Checkpoint the tool list as plain JSON schemas, so replay reconstructs tools without the live client.
     const listed = await run('mcp.listTools', async () => {
         const tools = await client.tools(toolOptions);
@@ -72,24 +61,36 @@ async function durableMCPTools(client, options = {}) {
             toModelOutput: def.convertsOutput ? ({ output }) => mcpToolOutput(output) : undefined,
             // Re-fetch the live tool inside the step (its execute closure can't be checkpointed); replay returns the recorded result.
             execute: (input, execOptions) => {
-                const signal = execOptions?.abortSignal;
+                const { abortSignal: signal, toolCallId } = (execOptions ?? {});
                 // An aborted consumer is done with this call, whatever the failure looks like; a retry would re-run a cancelled side effect.
                 const callConfig = {
                     ...stepConfig,
                     shouldRetry: async (error) => !signal?.aborted && (stepConfig.shouldRetry ? await stepConfig.shouldRetry(error) : true),
                 };
-                return run(`mcp.tool.${name}`, async () => {
-                    const tool = (await client.tools(toolOptions))[name];
-                    if (typeof tool?.execute !== 'function')
-                        throw new Error(`MCP tool "${name}" is not executable.`);
-                    const output = await tool.execute(input, execOptions);
-                    // A streaming execute can't checkpoint mid-flight; drain it and record the final value (the last yield).
-                    if (isAsyncIterable(output)) {
-                        let last;
-                        for await (last of output)
-                            ;
-                        return last;
+                // The tool call id comes from the checkpointed model result, so a reordered parallel step fails replay instead of swapping results.
+                return run(`mcp.tool.${name}.${toolCallId ?? 'call'}`, async () => {
+                    let output;
+                    try {
+                        const tool = (await client.tools(toolOptions))[name];
+                        if (typeof tool?.execute !== 'function')
+                            throw new Error(`MCP tool "${name}" is not executable.`);
+                        output = await tool.execute(input, execOptions);
+                        // A streaming execute can't checkpoint mid-flight; drain it and record the final value (the last yield).
+                        if ((0, internal_1.isAsyncIterable)(output)) {
+                            let last;
+                            for await (last of output)
+                                ;
+                            output = last;
+                        }
+                    }
+                    catch (error) {
+                        if (durableStream && toolCallId) {
+                            await (0, durable_stream_1.writeToolRecord)(durableStream, toolCallId, { errorText: error instanceof Error ? error.message : String(error) });
+                        }
+                        throw error;
                     }
+                    if (durableStream && toolCallId)
+                        await (0, durable_stream_1.writeToolRecord)(durableStream, toolCallId, { output });
                     return output;
                 }, callConfig);
             },

package/dist/mcp.js.map

@@ -1 +1 @@
-{"version":3,"file":"mcp.js","sourceRoot":"","sources":["../src/mcp.ts"],"names":[],"mappings":";;AAmEA,0CAwEC;AA3ID,iDAAsD;AAEtD,yCAA8H;AAoC9H,SAAS,eAAe,CAAC,KAAc;IACrC,OAAO,OAAQ,KAAmD,EAAE,CAAC,MAAM,CAAC,aAAa,CAAC,KAAK,UAAU,CAAC;AAC5G,CAAC;AAED,mHAAmH;AACnH,SAAS,aAAa,CAAC,MAAe;IACpC,MAAM,MAAM,GAAG,MAA+B,CAAC;IAC/C,IAAI,MAAM,KAAK,IAAI,IAAI,OAAO,MAAM,KAAK,QAAQ,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC;QACpF,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAqB,CAAC;IAC5D,CAAC;IACD,OAAO;QACL,IAAI,EAAE,SAAS;QACf,KAAK,EAAE,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,IAAwE,EAAE,EAAE;YACrG,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,IAAI,OAAO,IAAI,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;gBAC1D,OAAO,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC;YACpD,CAAC;YACD,IAAI,IAAI,CAAC,IAAI,KAAK,OAAO,IAAI,IAAI,CAAC,IAAI,KAAK,SAAS,IAAI,IAAI,CAAC,QAAQ,KAAK,SAAS,EAAE,CAAC;gBACpF,OAAO,EAAE,IAAI,EAAE,MAAe,EAAE,SAAS,EAAE,IAAI,CAAC,QAAQ,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,EAAE,CAAC;YAC/G,CAAC;YACD,OAAO,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC;QAC/D,CAAC,CAAC;KACH,CAAC;AACJ,CAAC;AAED;;;;GAIG;AACI,KAAK,UAAU,eAAe,CAAC,MAAqB,EAAE,UAAkC,EAAE;IAC/F,MAAM,EAAE,WAAW,EAAE,GAAG,WAAW,EAAE,GAAG,OAAO,CAAC;IAChD,MAAM,UAAU,GAAG,IAAA,kCAAuB,EAAC,WAAW,CAAC,CAAC;IACxD,MAAM,EAAE,QAAQ,EAAE,WAAW,EAAE,UAAU,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,CAAC;IACjE,MAAM,GAAG,GAAG,CAAI,IAAY,EAAE,EAAoB,EAAE,SAAqB,UAAU,EAAc,EAAE;QACjG,IAAA,iCAAsB,EAAC,IAAI,CAAC,CAAC;QAC7B,IAAI,CAAC,IAAA,+BAAoB,GAAE;YAAE,OAAO,EAAE,EAAE,CAAC;QACzC,8GAA8G;QAC9G,OAAO,eAAI,CAAC,OAAO,CAAC,EAAE,EAAE,EAAE,GAAG,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,KAAc,EAAE,EAAE;YACpE,MAAM,IAAA,oCAAyB,EAAC,KAAK,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC;IAEF,wGAAwG;IACxG,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,eAAe,EAAE,KAAK,IAAI,EAAE;QACnD,MAAM,KAAK,GAAG,MAAM,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;QAC9C,MAAM,IAAI,GAAmC,EAAE,CAAC;QAChD,KAAK,MAAM,CAAC,IAAI,EAAE,OAAO,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;YACpD,MAAM,IAAI,GAAG,OAAsB,CAAC;YACpC,IAAI,CAAC,IAAI,CAAC,GAAG;gBACX,WAAW,EAAE,OAAO,IAAI,CAAC,WAAW,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,SAAS;gBAChF,KAAK,EAAE,OAAO,IAAI,CAAC,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS;gBAC9D,QAAQ,EAAE,IAAI,CAAC,QAAQ;gBACvB,IAAI,EAAE,IAAI,CAAC,KAAK;gBAChB,cAAc,EAAE,OAAO,IAAI,CAAC,aAAa,KAAK,UAAU;gBACxD,mHAAmH;gBACnH,eAAe,EAAE,MAAM,QAAQ,CAAC,IAAI,CAAC,WAA6C,CAAC,CAAC,UAAU;aAC/F,CAAC;QACJ,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC,CAAC,CAAC;IAEH,MAAM,OAAO,GAAY,EAAE,CAAC;IAC5B,KAAK,MAAM,CAAC,IAAI,EAAE,GAAG,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;QACjD,MAAM,aAAa,GAAG,WAAW,CAAC;YAChC,WAAW,EAAE,GAAG,CAAC,WAAW;YAC5B,KAAK,EAAE,GAAG,CAAC,KAAK;YAChB,QAAQ,EAAE,GAAG,CAAC,QAAQ;YACtB,WAAW,EAAE,UAAU,CAAC,GAAG,CAAC,eAAmD,CAAC;YAChF,mIAAmI;YACnI,aAAa,EAAE,GAAG,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,SAAS;YACrF,0HAA0H;YAC1H,OAAO,EAAE,CAAC,KAAc,EAAE,WAAW,EAAE,EAAE;gBACvC,MAAM,MAAM,GAAI,WAAyD,EAAE,WAAW,CAAC;gBACvF,6HAA6H;gBAC7H,MAAM,UAAU,GAAe;oBAC7B,GAAG,UAAU;oBACb,WAAW,EAAE,KAAK,EAAE,KAAc,EAAE,EAAE,CACpC,CAAC,MAAM,EAAE,OAAO,IAAI,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC,CAAC,MAAM,UAAU,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;iBAC5F,CAAC;gBACF,OAAO,GAAG,CACR,YAAY,IAAI,EAAE,EAClB,KAAK,IAAI,EAAE;oBACT,MAAM,IAAI,GAAG,CAAC,MAAM,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAA4B,CAAC;oBAChF,IAAI,OAAO,IAAI,EAAE,OAAO,KAAK,UAAU;wBAAE,MAAM,IAAI,KAAK,CAAC,aAAa,IAAI,sBAAsB,CAAC,CAAC;oBAClG,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,WAAW,CAAC,CAAC;oBACtD,yGAAyG;oBACzG,IAAI,eAAe,CAAC,MAAM,CAAC,EAAE,CAAC;wBAC5B,IAAI,IAAa,CAAC;wBAClB,IAAI,KAAK,EAAE,IAAI,IAAI,MAAM;4BAAC,CAAC;wBAC3B,OAAO,IAAI,CAAC;oBACd,CAAC;oBACD,OAAO,MAAM,CAAC;gBAChB,CAAC,EACD,UAAU,CACX,CAAC;YACJ,CAAC;SACF,CAAC,CAAC;QACH,kGAAkG;QAClG,OAAO,CAAC,IAAI,CAAC,GAAG,GAAG,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,aAAa,EAAE,EAAE,KAAK,EAAE,GAAG,CAAC,IAAI,EAAE,CAAC,CAAC;IAC7G,CAAC;IACD,OAAO,OAAO,CAAC;AACjB,CAAC"}
+{"version":3,"file":"mcp.js","sourceRoot":"","sources":["../src/mcp.ts"],"names":[],"mappings":";;AAkEA,0CA4EC;AA5ID,yCAAsF;AACtF,qDAAmD;AAsCnD,mHAAmH;AACnH,SAAS,aAAa,CAAC,MAAe;IACpC,MAAM,MAAM,GAAG,MAA+B,CAAC;IAC/C,IAAI,MAAM,KAAK,IAAI,IAAI,OAAO,MAAM,KAAK,QAAQ,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC;QACpF,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAqB,CAAC;IAC5D,CAAC;IACD,OAAO;QACL,IAAI,EAAE,SAAS;QACf,KAAK,EAAE,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,IAAwE,EAAE,EAAE;YACrG,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,IAAI,OAAO,IAAI,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;gBAC1D,OAAO,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC;YACpD,CAAC;YACD,IAAI,IAAI,CAAC,IAAI,KAAK,OAAO,IAAI,IAAI,CAAC,IAAI,KAAK,SAAS,IAAI,IAAI,CAAC,QAAQ,KAAK,SAAS,EAAE,CAAC;gBACpF,OAAO,EAAE,IAAI,EAAE,MAAe,EAAE,SAAS,EAAE,IAAI,CAAC,QAAQ,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,EAAE,CAAC;YAC/G,CAAC;YACD,OAAO,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC;QAC/D,CAAC,CAAC;KACH,CAAC;AACJ,CAAC;AAED;;;;GAIG;AACI,KAAK,UAAU,eAAe,CAAC,MAAqB,EAAE,UAAkC,EAAE;IAC/F,MAAM,EAAE,WAAW,EAAE,aAAa,EAAE,GAAG,WAAW,EAAE,GAAG,OAAO,CAAC;IAC/D,MAAM,UAAU,GAAG,IAAA,kCAAuB,EAAC,WAAW,CAAC,CAAC;IACxD,MAAM,EAAE,QAAQ,EAAE,WAAW,EAAE,UAAU,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,CAAC;IACjE,MAAM,GAAG,GAAG,CAAI,IAAY,EAAE,EAAoB,EAAE,SAAqB,UAAU,EAAc,EAAE,CACjG,IAAA,yBAAc,EAAC,IAAI,EAAE,EAAE,EAAE,MAAM,CAAC,CAAC;IAEnC,wGAAwG;IACxG,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,eAAe,EAAE,KAAK,IAAI,EAAE;QACnD,MAAM,KAAK,GAAG,MAAM,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;QAC9C,MAAM,IAAI,GAAmC,EAAE,CAAC;QAChD,KAAK,MAAM,CAAC,IAAI,EAAE,OAAO,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;YACpD,MAAM,IAAI,GAAG,OAAsB,CAAC;YACpC,IAAI,CAAC,IAAI,CAAC,GAAG;gBACX,WAAW,EAAE,OAAO,IAAI,CAAC,WAAW,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,SAAS;gBAChF,KAAK,EAAE,OAAO,IAAI,CAAC,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS;gBAC9D,QAAQ,EAAE,IAAI,CAAC,QAAQ;gBACvB,IAAI,EAAE,IAAI,CAAC,KAAK;gBAChB,cAAc,EAAE,OAAO,IAAI,CAAC,aAAa,KAAK,UAAU;gBACxD,mHAAmH;gBACnH,eAAe,EAAE,MAAM,QAAQ,CAAC,IAAI,CAAC,WAA6C,CAAC,CAAC,UAAU;aAC/F,CAAC;QACJ,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC,CAAC,CAAC;IAEH,MAAM,OAAO,GAAY,EAAE,CAAC;IAC5B,KAAK,MAAM,CAAC,IAAI,EAAE,GAAG,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;QACjD,MAAM,aAAa,GAAG,WAAW,CAAC;YAChC,WAAW,EAAE,GAAG,CAAC,WAAW;YAC5B,KAAK,EAAE,GAAG,CAAC,KAAK;YAChB,QAAQ,EAAE,GAAG,CAAC,QAAQ;YACtB,WAAW,EAAE,UAAU,CAAC,GAAG,CAAC,eAAmD,CAAC;YAChF,mIAAmI;YACnI,aAAa,EAAE,GAAG,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,SAAS;YACrF,0HAA0H;YAC1H,OAAO,EAAE,CAAC,KAAc,EAAE,WAAW,EAAE,EAAE;gBACvC,MAAM,EAAE,WAAW,EAAE,MAAM,EAAE,UAAU,EAAE,GAAG,CAAC,WAAW,IAAI,EAAE,CAAuD,CAAC;gBACtH,6HAA6H;gBAC7H,MAAM,UAAU,GAAe;oBAC7B,GAAG,UAAU;oBACb,WAAW,EAAE,KAAK,EAAE,KAAc,EAAE,EAAE,CACpC,CAAC,MAAM,EAAE,OAAO,IAAI,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC,CAAC,MAAM,UAAU,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;iBAC5F,CAAC;gBACF,oIAAoI;gBACpI,OAAO,GAAG,CACR,YAAY,IAAI,IAAI,UAAU,IAAI,MAAM,EAAE,EAC1C,KAAK,IAAI,EAAE;oBACT,IAAI,MAAe,CAAC;oBACpB,IAAI,CAAC;wBACH,MAAM,IAAI,GAAG,CAAC,MAAM,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAA4B,CAAC;wBAChF,IAAI,OAAO,IAAI,EAAE,OAAO,KAAK,UAAU;4BAAE,MAAM,IAAI,KAAK,CAAC,aAAa,IAAI,sBAAsB,CAAC,CAAC;wBAClG,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,WAAW,CAAC,CAAC;wBAChD,yGAAyG;wBACzG,IAAI,IAAA,0BAAe,EAAC,MAAM,CAAC,EAAE,CAAC;4BAC5B,IAAI,IAAa,CAAC;4BAClB,IAAI,KAAK,EAAE,IAAI,IAAI,MAAM;gCAAC,CAAC;4BAC3B,MAAM,GAAG,IAAI,CAAC;wBAChB,CAAC;oBACH,CAAC;oBAAC,OAAO,KAAK,EAAE,CAAC;wBACf,IAAI,aAAa,IAAI,UAAU,EAAE,CAAC;4BAChC,MAAM,IAAA,gCAAe,EAAC,aAAa,EAAE,UAAU,EAAE,EAAE,SAAS,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;wBAC1H,CAAC;wBACD,MAAM,KAAK,CAAC;oBACd,CAAC;oBACD,IAAI,aAAa,IAAI,UAAU;wBAAE,MAAM,IAAA,gCAAe,EAAC,aAAa,EAAE,UAAU,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;oBAC9F,OAAO,MAAM,CAAC;gBAChB,CAAC,EACD,UAAU,CACX,CAAC;YACJ,CAAC;SACF,CAAC,CAAC;QACH,kGAAkG;QAClG,OAAO,CAAC,IAAI,CAAC,GAAG,GAAG,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,aAAa,EAAE,EAAE,KAAK,EAAE,GAAG,CAAC,IAAI,EAAE,CAAC,CAAC;IAC7G,CAAC;IACD,OAAO,OAAO,CAAC;AACjB,CAAC"}

package/dist/middleware.d.ts

@@ -1,7 +1,12 @@
 import { StepConfig } from '@dbos-inc/dbos-sdk';
 import type { EmbeddingModelMiddleware, ImageModelMiddleware, LanguageModelMiddleware } from 'ai' with { 'resolution-mode': 'import' };
+import { type DurableStreamOptions } from './durable-stream';
+export interface DurableCallsOptions extends StepConfig {
+    /** Write each streamed model call's parts to this durable stream from inside its step (see readDurableStream). */
+    durableStream?: DurableStreamOptions;
+}
 /** AI SDK language-model middleware that runs each model call as a durable, checkpointed DBOS step (replayed on recovery); outside a workflow it calls the model directly. */
-export declare function durableCalls(options?: StepConfig): LanguageModelMiddleware;
+export declare function durableCalls(options?: DurableCallsOptions): LanguageModelMiddleware;
 /** AI SDK embedding-model middleware that runs each embedding call as a durable DBOS step, like {@link durableCalls}. */
 export declare function durableEmbeddingCalls(options?: StepConfig): EmbeddingModelMiddleware;
 /**

package/dist/middleware.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"middleware.d.ts","sourceRoot":"","sources":["../src/middleware.ts"],"names":[],"mappings":"AACA,OAAO,EAAQ,UAAU,EAAE,MAAM,oBAAoB,CAAC;AAItD,OAAO,KAAK,EAAE,wBAAwB,EAAE,oBAAoB,EAAE,uBAAuB,EAAE,MAAM,IAAI,CAAC,OAAO,iBAAiB,EAAE,QAAQ,EAAE,CAAC;AAgDvI,8KAA8K;AAC9K,wBAAgB,YAAY,CAAC,OAAO,GAAE,UAAe,GAAG,uBAAuB,CAsK9E;AAED,yHAAyH;AACzH,wBAAgB,qBAAqB,CAAC,OAAO,GAAE,UAAe,GAAG,wBAAwB,CAsBxF;AAED;;;;GAIG;AACH,wBAAgB,iBAAiB,CAAC,OAAO,GAAE,UAAe,GAAG,oBAAoB,CAmBhF"}
+{"version":3,"file":"middleware.d.ts","sourceRoot":"","sources":["../src/middleware.ts"],"names":[],"mappings":"AACA,OAAO,EAAQ,UAAU,EAAE,MAAM,oBAAoB,CAAC;AAItD,OAAO,KAAK,EAAE,wBAAwB,EAAE,oBAAoB,EAAE,uBAAuB,EAAE,MAAM,IAAI,CAAC,OAAO,iBAAiB,EAAE,QAAQ,EAAE,CAAC;AAkBvI,OAAO,EAAE,KAAK,oBAAoB,EAA2C,MAAM,kBAAkB,CAAC;AAEtG,MAAM,WAAW,mBAAoB,SAAQ,UAAU;IACrD,kHAAkH;IAClH,aAAa,CAAC,EAAE,oBAAoB,CAAC;CACtC;AA8BD,8KAA8K;AAC9K,wBAAgB,YAAY,CAAC,OAAO,GAAE,mBAAwB,GAAG,uBAAuB,CA2MvF;AAED,yHAAyH;AACzH,wBAAgB,qBAAqB,CAAC,OAAO,GAAE,UAAe,GAAG,wBAAwB,CA2BxF;AAED;;;;GAIG;AACH,wBAAgB,iBAAiB,CAAC,OAAO,GAAE,UAAe,GAAG,oBAAoB,CAmBhF"}