@dbos-inc/vercel-ai 0.1.5

package/dist/middleware.js

@@ -0,0 +1,496 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.durableCalls = durableCalls;
+exports.durableEmbeddingCalls = durableEmbeddingCalls;
+exports.durableImageCalls = durableImageCalls;
+const node_crypto_1 = require("node:crypto");
+const dbos_sdk_1 = require("@dbos-inc/dbos-sdk");
+const internal_1 = require("./internal");
+// In-flight durable model calls per workflow; concurrent calls have a nondeterministic DBOS step order on replay, so we reject them.
+const inflightModelCalls = new Map();
+function enterDurableModelCall(operation) {
+    const workflowID = dbos_sdk_1.DBOS.workflowID;
+    const inflight = inflightModelCalls.get(workflowID) ?? 0;
+    if (inflight > 0) {
+        // embedMany parallelizes its batches; maxParallelCalls: 1 serializes them deterministically. Other callers use child workflows.
+        const remedy = operation === 'embed'
+            ? 'pass maxParallelCalls: 1 to embedMany, or run each call in its own child workflow with DBOS.startWorkflow'
+            : 'run each call in its own child workflow with DBOS.startWorkflow';
+        throw new Error(`Concurrent durable model calls in workflow "${workflowID}" are not supported because their step order is nondeterministic on replay; ${remedy}.`);
+    }
+    inflightModelCalls.set(workflowID, inflight + 1);
+    return workflowID;
+}
+function exitDurableModelCall(workflowID) {
+    const inflight = (inflightModelCalls.get(workflowID) ?? 1) - 1;
+    if (inflight > 0) {
+        inflightModelCalls.set(workflowID, inflight);
+    }
+    else {
+        inflightModelCalls.delete(workflowID);
+    }
+}
+/** 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. */
+function durableCalls(options = {}) {
+    const stepConfig = (0, internal_1.withErrorClassification)(options);
+    return {
+        specificationVersion: 'v4',
+        wrapGenerate: async ({ doGenerate, model }) => {
+            (0, internal_1.assertNotInTransaction)('generate');
+            if (!(0, internal_1.isInWorkflowFunction)()) {
+                return await doGenerate();
+            }
+            const workflowID = enterDurableModelCall('generate');
+            try {
+                return await dbos_sdk_1.DBOS.runStep(async () => ensureResponseMetadata(encodeBinaryContent(await doGenerate())), {
+                    ...stepConfig,
+                    name: stepConfig.name ?? stepName(model, 'generate'),
+                });
+            }
+            catch (error) {
+                // Restore the AI SDK error identity a replay revival strips, so the SDK's retry/catch logic behaves the same.
+                throw (0, internal_1.restoreAISDKErrorIdentity)(error);
+            }
+            finally {
+                exitDurableModelCall(workflowID);
+            }
+        },
+        wrapStream: async ({ doStream, params, model }) => {
+            (0, internal_1.assertNotInTransaction)('stream');
+            if (!(0, internal_1.isInWorkflowFunction)()) {
+                return await doStream();
+            }
+            const workflowID = enterDurableModelCall('stream');
+            // An aborted consumer is done with this call, like a cancelled one: post-abort failures must checkpoint as a (partial) success, or replay would fail where the live run ended gracefully.
+            const aborted = () => params.abortSignal?.aborted === true;
+            // Free the concurrency guard as soon as the consumer detaches (abort/cancel), not only when the step
+            // settles: this call's step is already sequenced, so a sequential follow-up in the same workflow is
+            // deterministic on replay and must not be rejected as concurrent while this step drains in the background.
+            let guardReleased = false;
+            const releaseGuard = () => {
+                if (guardReleased)
+                    return;
+                guardReleased = true;
+                params.abortSignal?.removeEventListener('abort', releaseGuard);
+                exitDurableModelCall(workflowID);
+            };
+            params.abortSignal?.addEventListener('abort', releaseGuard, { once: true });
+            let executed = false;
+            let cancelled = false;
+            let emittedLive = false;
+            let controller;
+            let step;
+            const stream = new ReadableStream({
+                start(c) {
+                    controller = c;
+                },
+                // Await the step so an early cancel still blocks until the model result is checkpointed.
+                async cancel() {
+                    cancelled = true;
+                    releaseGuard();
+                    // Post-cancel failures normally checkpoint as a success; anything else (e.g. a failed checkpoint write) is only visible here.
+                    await step.catch((error) => dbos_sdk_1.DBOS.logger.warn(`Durable model call step failed after consumer cancel: ${String(error)}`));
+                },
+            });
+            const emit = (part) => {
+                if (!cancelled) {
+                    controller.enqueue(part);
+                    // stream-start and response-metadata merge idempotently downstream, so they alone don't preclude a retry.
+                    if (part.type !== 'stream-start' && part.type !== 'response-metadata')
+                        emittedLive = true;
+                }
+            };
+            // Once any output part has streamed live, a retry would re-stream from scratch and duplicate output, so stop retrying.
+            const streamStepConfig = {
+                ...stepConfig,
+                shouldRetry: async (error) => !emittedLive && (stepConfig.shouldRetry ? await stepConfig.shouldRetry(error) : true),
+            };
+            try {
+                step = dbos_sdk_1.DBOS.runStep(async () => {
+                    executed = true;
+                    const accumulator = new StreamAccumulator();
+                    // A timed-out attempt is abandoned by DBOS (its outcome is discarded) but keeps running; stop it so it can't emit alongside a retry.
+                    const timeoutSignal = dbos_sdk_1.DBOS.stepStatus?.timeoutSignal;
+                    let streamResult;
+                    let reader;
+                    let sawFinish = false;
+                    const abandon = () => void reader?.cancel().catch(() => { });
+                    timeoutSignal?.addEventListener('abort', abandon, { once: true });
+                    // A consumer abort detaches this call; stop draining now so the checkpoint lands promptly and matches what streamed.
+                    params.abortSignal?.addEventListener('abort', abandon, { once: true });
+                    try {
+                        streamResult = await doStream();
+                        reader = streamResult.stream.getReader();
+                        for (;;) {
+                            const { done, value: part } = await reader.read();
+                            if (timeoutSignal?.aborted)
+                                throw (timeoutSignal.reason ?? new Error('step attempt timed out'));
+                            if (done)
+                                break;
+                            if (part.type === 'error') {
+                                // A cancelled or aborted consumer abandoned this call; don't let a late failure become the step outcome, or replay would fail where the live run succeeded.
+                                if (cancelled || aborted())
+                                    break;
+                                throw toStepError(part.error);
+                            }
+                            // Stream deltas live but withhold 'finish' until the checkpoint is durable: the AI SDK runs tool calls (and their durable steps) on 'finish', which must not checkpoint before this model step.
+                            if (part.type === 'finish')
+                                sawFinish = true;
+                            else
+                                emit(part);
+                            accumulator.add(part);
+                        }
+                        // No terminal part and no output: fail (retryably) like the AI SDK's NoOutputGeneratedError, instead of checkpointing a permanent empty success.
+                        if (!sawFinish && !accumulator.hasContent && !cancelled && !aborted()) {
+                            throw new Error('Model stream ended without a finish part or any output.');
+                        }
+                    }
+                    catch (error) {
+                        // Same rule for stream-level failures (doStream or a read rejecting) after a cancel or abort.
+                        if (!cancelled && !aborted())
+                            throw error;
+                    }
+                    finally {
+                        timeoutSignal?.removeEventListener('abort', abandon);
+                        params.abortSignal?.removeEventListener('abort', abandon);
+                        // Tear down the provider stream on early exits (error part, post-cancel break); a no-op after a clean drain.
+                        void reader?.cancel().catch(() => { });
+                    }
+                    // Give the response a durable id/timestamp when the provider sent none, and emit it live (before the
+                    // withheld 'finish') so the SDK sees the same values live and on replay instead of a fresh fallback.
+                    // Skip the live emit for a timed-out (abandoned) attempt so it can't interleave with its retry.
+                    const responseMetadataPart = accumulator.fillResponseMetadata((0, node_crypto_1.randomUUID)(), new Date());
+                    if (responseMetadataPart && !timeoutSignal?.aborted)
+                        emit(responseMetadataPart);
+                    return encodeBinaryContent(accumulator.result(streamResult?.request, streamResult?.response));
+                }, { ...streamStepConfig, name: stepConfig.name ?? stepName(model, 'stream') });
+            }
+            catch (error) {
+                // runStep can throw synchronously (e.g. a shutdown race); don't leak the guard entry.
+                releaseGuard();
+                throw error;
+            }
+            // Drive the returned stream from the settled step: a live run emits only the withheld 'finish' (deltas already
+            // streamed); a recovered run synthesizes the whole stream from the checkpoint. Either way consumers finish
+            // only after the result is durable.
+            void step
+                .then((recorded) => {
+                if (cancelled)
+                    return;
+                if (executed) {
+                    emit({
+                        type: 'finish',
+                        finishReason: recorded.finishReason,
+                        usage: recorded.usage,
+                        providerMetadata: recorded.providerMetadata,
+                    });
+                }
+                else {
+                    for (const part of replayParts(recorded))
+                        emit(part);
+                }
+                controller.close();
+            }, (error) => {
+                if (!cancelled)
+                    controller.error((0, internal_1.restoreAISDKErrorIdentity)(error));
+            })
+                .finally(releaseGuard);
+            return { stream };
+        },
+    };
+}
+/** AI SDK embedding-model middleware that runs each embedding call as a durable DBOS step, like {@link durableCalls}. */
+function durableEmbeddingCalls(options = {}) {
+    const stepConfig = (0, internal_1.withErrorClassification)(options);
+    return {
+        specificationVersion: 'v4',
+        wrapEmbed: async ({ doEmbed, model }) => {
+            (0, internal_1.assertNotInTransaction)('embed');
+            if (!(0, internal_1.isInWorkflowFunction)()) {
+                return await doEmbed();
+            }
+            const workflowID = enterDurableModelCall('embed');
+            try {
+                return await dbos_sdk_1.DBOS.runStep(async () => doEmbed(), {
+                    ...stepConfig,
+                    name: stepConfig.name ?? stepName(model, 'embed'),
+                });
+            }
+            catch (error) {
+                throw (0, internal_1.restoreAISDKErrorIdentity)(error);
+            }
+            finally {
+                exitDurableModelCall(workflowID);
+            }
+        },
+    };
+}
+/**
+ * AI SDK image-model middleware that runs each image generation as a durable DBOS step, like {@link durableCalls}.
+ * No concurrency guard: generateImage splits `n > maxImagesPerCall` into batches it dispatches synchronously (no
+ * await before doGenerate), so their step order is deterministic on replay — unlike embedMany's parallel batches.
+ */
+function durableImageCalls(options = {}) {
+    const stepConfig = (0, internal_1.withErrorClassification)(options);
+    return {
+        specificationVersion: 'v4',
+        wrapGenerate: async ({ doGenerate, model }) => {
+            (0, internal_1.assertNotInTransaction)('generateImage');
+            if (!(0, internal_1.isInWorkflowFunction)()) {
+                return await doGenerate();
+            }
+            try {
+                return await dbos_sdk_1.DBOS.runStep(async () => encodeImageResult(await doGenerate()), {
+                    ...stepConfig,
+                    name: stepConfig.name ?? stepName(model, 'image'),
+                });
+            }
+            catch (error) {
+                throw (0, internal_1.restoreAISDKErrorIdentity)(error);
+            }
+        },
+    };
+}
+/** Convert generated image bytes (Uint8Array) to base64 (spec-allowed) to keep checkpoints compact. */
+function encodeImageResult(result) {
+    const images = result.images;
+    if (images.every((image) => typeof image === 'string')) {
+        return result;
+    }
+    return {
+        ...result,
+        images: images.map((image) => (typeof image === 'string' ? image : Buffer.from(image).toString('base64'))),
+    };
+}
+function stepName(model, operation) {
+    return `${model.provider}.${model.modelId}.${operation}`;
+}
+/** Normalize an error-part payload to an Error, preserving the payload (JSON message, cause, isRetryable). */
+function toStepError(error) {
+    if (error instanceof Error)
+        return error;
+    let message;
+    try {
+        message = typeof error === 'string' ? error : (JSON.stringify(error) ?? String(error));
+    }
+    catch {
+        message = String(error);
+    }
+    const result = new Error(message, { cause: error });
+    // Carry the payload's name so an abort/timeout error-part is classified terminal (isAbortError reads .name), not retried.
+    const name = error?.name;
+    if (typeof name === 'string')
+        result.name = name;
+    const isRetryable = error?.isRetryable;
+    return isRetryable === undefined ? result : Object.assign(result, { isRetryable });
+}
+// generateText fills response.id/timestamp with generateId()/new Date() OUTSIDE this step when the provider omits
+// them, so they'd differ on every replay. Populate them here (checkpointed once) so that fallback never runs.
+function ensureResponseMetadata(result) {
+    const response = result.response;
+    // Match the AI SDK's `?? generateId()` fallback (nullish, not just undefined): a null id would regenerate too.
+    if (response?.id != null && response?.timestamp != null) {
+        return result;
+    }
+    return {
+        ...result,
+        response: { ...response, id: response?.id ?? (0, node_crypto_1.randomUUID)(), timestamp: response?.timestamp ?? new Date() },
+    };
+}
+/** Convert generated-file bytes (Uint8Array) to base64 (spec-allowed) to keep checkpoints compact. */
+function encodeBinaryContent(result) {
+    const content = result.content.map(encodeBinaryPart);
+    return content.some((part, i) => part !== result.content[i]) ? { ...result, content } : result;
+}
+function encodeBinaryPart(part) {
+    if ((part.type === 'file' || part.type === 'reasoning-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;
+}
+/** Assembles stream parts into a LanguageModelV4GenerateResult so a stream can be checkpointed as one step result. */
+class StreamAccumulator {
+    content = [];
+    finishReason = { unified: 'other', raw: undefined };
+    usage = {
+        inputTokens: { total: undefined, noCache: undefined, cacheRead: undefined, cacheWrite: undefined },
+        outputTokens: { total: undefined, text: undefined, reasoning: undefined },
+    };
+    warnings = [];
+    providerMetadata;
+    responseMetadata;
+    // Text/reasoning content objects (also in content), appended at -start and mutated in place so content keeps arrival order.
+    textBlocks = new Map();
+    reasoningBlocks = new Map();
+    add(part) {
+        switch (part.type) {
+            case 'stream-start':
+                this.warnings.push(...part.warnings);
+                break;
+            case 'text-start': {
+                const block = { type: 'text', text: '', providerMetadata: part.providerMetadata };
+                this.textBlocks.set(part.id, block);
+                this.content.push(block);
+                break;
+            }
+            case 'text-delta': {
+                const block = this.getOrCreateText(part.id);
+                block.text += part.delta;
+                if (part.providerMetadata)
+                    block.providerMetadata = part.providerMetadata;
+                break;
+            }
+            case 'text-end': {
+                const block = this.textBlocks.get(part.id);
+                if (block && part.providerMetadata)
+                    block.providerMetadata = part.providerMetadata;
+                this.textBlocks.delete(part.id);
+                break;
+            }
+            case 'reasoning-start': {
+                const block = {
+                    type: 'reasoning',
+                    text: '',
+                    providerMetadata: part.providerMetadata,
+                };
+                this.reasoningBlocks.set(part.id, block);
+                this.content.push(block);
+                break;
+            }
+            case 'reasoning-delta': {
+                const block = this.getOrCreateReasoning(part.id);
+                block.text += part.delta;
+                if (part.providerMetadata)
+                    block.providerMetadata = part.providerMetadata;
+                break;
+            }
+            case 'reasoning-end': {
+                const block = this.reasoningBlocks.get(part.id);
+                if (block && part.providerMetadata)
+                    block.providerMetadata = part.providerMetadata;
+                this.reasoningBlocks.delete(part.id);
+                break;
+            }
+            case 'response-metadata':
+                // Merge per-field like the AI SDK: later parts override only the fields they carry.
+                this.responseMetadata = {
+                    id: part.id ?? this.responseMetadata?.id,
+                    timestamp: part.timestamp ?? this.responseMetadata?.timestamp,
+                    modelId: part.modelId ?? this.responseMetadata?.modelId,
+                };
+                break;
+            case 'finish':
+                this.finishReason = part.finishReason;
+                this.usage = part.usage;
+                this.providerMetadata = part.providerMetadata;
+                break;
+            case 'tool-input-start':
+            case 'tool-input-delta':
+            case 'tool-input-end':
+            case 'raw':
+            case 'error':
+                // Transient parts; the tool-call part carries the complete input.
+                break;
+            default:
+                // Complete content parts: tool-call, tool-result, tool-approval-request, file, reasoning-file, source, custom.
+                this.content.push(part);
+                break;
+        }
+    }
+    get hasContent() {
+        return this.content.length > 0;
+    }
+    // Fill any missing response id/timestamp so the checkpoint carries them; returns the response-metadata part to
+    // emit live (or undefined if the provider already supplied both). replayParts re-emits it from the checkpoint.
+    fillResponseMetadata(id, timestamp) {
+        // Nullish check to match the AI SDK's `?? generateId()` fallback (a null id/timestamp would regenerate too).
+        if (this.responseMetadata?.id != null && this.responseMetadata?.timestamp != null) {
+            return undefined;
+        }
+        this.responseMetadata = {
+            id: this.responseMetadata?.id ?? id,
+            timestamp: this.responseMetadata?.timestamp ?? timestamp,
+            modelId: this.responseMetadata?.modelId,
+        };
+        return { type: 'response-metadata', ...this.responseMetadata };
+    }
+    // If a delta arrives with no preceding start, create the block in arrival position rather than dropping the text.
+    getOrCreateText(id) {
+        let block = this.textBlocks.get(id);
+        if (!block) {
+            block = { type: 'text', text: '' };
+            this.textBlocks.set(id, block);
+            this.content.push(block);
+        }
+        return block;
+    }
+    getOrCreateReasoning(id) {
+        let block = this.reasoningBlocks.get(id);
+        if (!block) {
+            block = { type: 'reasoning', text: '' };
+            this.reasoningBlocks.set(id, block);
+            this.content.push(block);
+        }
+        return block;
+    }
+    result(request, response) {
+        return {
+            content: this.content,
+            finishReason: this.finishReason,
+            usage: this.usage,
+            warnings: this.warnings,
+            providerMetadata: this.providerMetadata,
+            request,
+            response: this.responseMetadata ? { ...this.responseMetadata, ...response } : response,
+        };
+    }
+}
+/** Synthesizes a stream from a checkpointed result on recovery; text/reasoning come back as one delta per block. */
+function* replayParts(result) {
+    yield { type: 'stream-start', warnings: result.warnings ?? [] };
+    if (result.response?.id !== undefined || result.response?.timestamp !== undefined || result.response?.modelId !== undefined) {
+        yield {
+            type: 'response-metadata',
+            id: result.response.id,
+            timestamp: result.response.timestamp,
+            modelId: result.response.modelId,
+        };
+    }
+    let blockIndex = 0;
+    for (const part of result.content) {
+        const id = `replay-${blockIndex++}`;
+        if (part.type === 'text') {
+            yield { type: 'text-start', id, providerMetadata: part.providerMetadata };
+            if (part.text.length > 0)
+                yield { type: 'text-delta', id, delta: part.text };
+            yield { type: 'text-end', id, providerMetadata: part.providerMetadata };
+        }
+        else if (part.type === 'reasoning') {
+            yield { type: 'reasoning-start', id, providerMetadata: part.providerMetadata };
+            if (part.text.length > 0)
+                yield { type: 'reasoning-delta', id, delta: part.text };
+            yield { type: 'reasoning-end', id, providerMetadata: part.providerMetadata };
+        }
+        else if (part.type === 'tool-call') {
+            // Re-synthesize the tool-input grammar: consumer tool callbacks (onInputStart/onInputAvailable) are keyed off tool-input-start.
+            const { toolCallId, toolName, providerExecuted, dynamic } = part;
+            yield { type: 'tool-input-start', id: toolCallId, toolName, providerExecuted, dynamic };
+            if (part.input.length > 0)
+                yield { type: 'tool-input-delta', id: toolCallId, delta: part.input };
+            yield { type: 'tool-input-end', id: toolCallId };
+            yield part;
+        }
+        else {
+            yield part;
+        }
+    }
+    yield {
+        type: 'finish',
+        finishReason: result.finishReason,
+        usage: result.usage,
+        providerMetadata: result.providerMetadata,
+    };
+}
+//# sourceMappingURL=middleware.js.map

package/dist/middleware.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"middleware.js","sourceRoot":"","sources":["../src/middleware.ts"],"names":[],"mappings":";;AAsDA,oCAsKC;AAGD,sDAsBC;AAOD,8CAmBC;AA/QD,6CAAyC;AACzC,iDAAsD;AAqBtD,yCAA8H;AAE9H,qIAAqI;AACrI,MAAM,kBAAkB,GAAG,IAAI,GAAG,EAAkB,CAAC;AAErD,SAAS,qBAAqB,CAAC,SAA0C;IACvE,MAAM,UAAU,GAAG,eAAI,CAAC,UAAW,CAAC;IACpC,MAAM,QAAQ,GAAG,kBAAkB,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;IACzD,IAAI,QAAQ,GAAG,CAAC,EAAE,CAAC;QACjB,gIAAgI;QAChI,MAAM,MAAM,GACV,SAAS,KAAK,OAAO;YACnB,CAAC,CAAC,2GAA2G;YAC7G,CAAC,CAAC,iEAAiE,CAAC;QACxE,MAAM,IAAI,KAAK,CACb,+CAA+C,UAAU,+EAA+E,MAAM,GAAG,CAClJ,CAAC;IACJ,CAAC;IACD,kBAAkB,CAAC,GAAG,CAAC,UAAU,EAAE,QAAQ,GAAG,CAAC,CAAC,CAAC;IACjD,OAAO,UAAU,CAAC;AACpB,CAAC;AAED,SAAS,oBAAoB,CAAC,UAAkB;IAC9C,MAAM,QAAQ,GAAG,CAAC,kBAAkB,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC;IAC/D,IAAI,QAAQ,GAAG,CAAC,EAAE,CAAC;QACjB,kBAAkB,CAAC,GAAG,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;IAC/C,CAAC;SAAM,CAAC;QACN,kBAAkB,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;IACxC,CAAC;AACH,CAAC;AAED,8KAA8K;AAC9K,SAAgB,YAAY,CAAC,UAAsB,EAAE;IACnD,MAAM,UAAU,GAAG,IAAA,kCAAuB,EAAC,OAAO,CAAC,CAAC;IACpD,OAAO;QACL,oBAAoB,EAAE,IAAI;QAE1B,YAAY,EAAE,KAAK,EAAE,EAAE,UAAU,EAAE,KAAK,EAAE,EAAE,EAAE;YAC5C,IAAA,iCAAsB,EAAC,UAAU,CAAC,CAAC;YACnC,IAAI,CAAC,IAAA,+BAAoB,GAAE,EAAE,CAAC;gBAC5B,OAAO,MAAM,UAAU,EAAE,CAAC;YAC5B,CAAC;YACD,MAAM,UAAU,GAAG,qBAAqB,CAAC,UAAU,CAAC,CAAC;YACrD,IAAI,CAAC;gBACH,OAAO,MAAM,eAAI,CAAC,OAAO,CAAC,KAAK,IAAI,EAAE,CAAC,sBAAsB,CAAC,mBAAmB,CAAC,MAAM,UAAU,EAAE,CAAC,CAAC,EAAE;oBACrG,GAAG,UAAU;oBACb,IAAI,EAAE,UAAU,CAAC,IAAI,IAAI,QAAQ,CAAC,KAAK,EAAE,UAAU,CAAC;iBACrD,CAAC,CAAC;YACL,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,8GAA8G;gBAC9G,MAAM,IAAA,oCAAyB,EAAC,KAAK,CAAC,CAAC;YACzC,CAAC;oBAAS,CAAC;gBACT,oBAAoB,CAAC,UAAU,CAAC,CAAC;YACnC,CAAC;QACH,CAAC;QAED,UAAU,EAAE,KAAK,EAAE,EAAE,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,EAAE,EAAE;YAChD,IAAA,iCAAsB,EAAC,QAAQ,CAAC,CAAC;YACjC,IAAI,CAAC,IAAA,+BAAoB,GAAE,EAAE,CAAC;gBAC5B,OAAO,MAAM,QAAQ,EAAE,CAAC;YAC1B,CAAC;YACD,MAAM,UAAU,GAAG,qBAAqB,CAAC,QAAQ,CAAC,CAAC;YACnD,0LAA0L;YAC1L,MAAM,OAAO,GAAG,GAAG,EAAE,CAAC,MAAM,CAAC,WAAW,EAAE,OAAO,KAAK,IAAI,CAAC;YAE3D,qGAAqG;YACrG,oGAAoG;YACpG,2GAA2G;YAC3G,IAAI,aAAa,GAAG,KAAK,CAAC;YAC1B,MAAM,YAAY,GAAG,GAAG,EAAE;gBACxB,IAAI,aAAa;oBAAE,OAAO;gBAC1B,aAAa,GAAG,IAAI,CAAC;gBACrB,MAAM,CAAC,WAAW,EAAE,mBAAmB,CAAC,OAAO,EAAE,YAAY,CAAC,CAAC;gBAC/D,oBAAoB,CAAC,UAAU,CAAC,CAAC;YACnC,CAAC,CAAC;YACF,MAAM,CAAC,WAAW,EAAE,gBAAgB,CAAC,OAAO,EAAE,YAAY,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;YAE5E,IAAI,QAAQ,GAAG,KAAK,CAAC;YACrB,IAAI,SAAS,GAAG,KAAK,CAAC;YACtB,IAAI,WAAW,GAAG,KAAK,CAAC;YACxB,IAAI,UAAuE,CAAC;YAC5E,IAAI,IAA6C,CAAC;YAClD,MAAM,MAAM,GAAG,IAAI,cAAc,CAA4B;gBAC3D,KAAK,CAAC,CAAC;oBACL,UAAU,GAAG,CAAC,CAAC;gBACjB,CAAC;gBACD,yFAAyF;gBACzF,KAAK,CAAC,MAAM;oBACV,SAAS,GAAG,IAAI,CAAC;oBACjB,YAAY,EAAE,CAAC;oBACf,8HAA8H;oBAC9H,MAAM,IAAI,CAAC,KAAK,CAAC,CAAC,KAAc,EAAE,EAAE,CAClC,eAAI,CAAC,MAAM,CAAC,IAAI,CAAC,yDAAyD,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAC3F,CAAC;gBACJ,CAAC;aACF,CAAC,CAAC;YACH,MAAM,IAAI,GAAG,CAAC,IAA+B,EAAE,EAAE;gBAC/C,IAAI,CAAC,SAAS,EAAE,CAAC;oBACf,UAAU,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;oBACzB,0GAA0G;oBAC1G,IAAI,IAAI,CAAC,IAAI,KAAK,cAAc,IAAI,IAAI,CAAC,IAAI,KAAK,mBAAmB;wBAAE,WAAW,GAAG,IAAI,CAAC;gBAC5F,CAAC;YACH,CAAC,CAAC;YAEF,uHAAuH;YACvH,MAAM,gBAAgB,GAAe;gBACnC,GAAG,UAAU;gBACb,WAAW,EAAE,KAAK,EAAE,KAAc,EAAE,EAAE,CACpC,CAAC,WAAW,IAAI,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC,CAAC,MAAM,UAAU,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;aACxF,CAAC;YAEF,IAAI,CAAC;gBACH,IAAI,GAAG,eAAI,CAAC,OAAO,CACjB,KAAK,IAAI,EAAE;oBACT,QAAQ,GAAG,IAAI,CAAC;oBAChB,MAAM,WAAW,GAAG,IAAI,iBAAiB,EAAE,CAAC;oBAC5C,qIAAqI;oBACrI,MAAM,aAAa,GAAG,eAAI,CAAC,UAAU,EAAE,aAAa,CAAC;oBACrD,IAAI,YAA8D,CAAC;oBACnE,IAAI,MAA0E,CAAC;oBAC/E,IAAI,SAAS,GAAG,KAAK,CAAC;oBACtB,MAAM,OAAO,GAAG,GAAG,EAAE,CAAC,KAAK,MAAM,EAAE,MAAM,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;oBAC5D,aAAa,EAAE,gBAAgB,CAAC,OAAO,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;oBAClE,qHAAqH;oBACrH,MAAM,CAAC,WAAW,EAAE,gBAAgB,CAAC,OAAO,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;oBACvE,IAAI,CAAC;wBACH,YAAY,GAAG,MAAM,QAAQ,EAAE,CAAC;wBAChC,MAAM,GAAG,YAAY,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC;wBACzC,SAAS,CAAC;4BACR,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;4BAClD,IAAI,aAAa,EAAE,OAAO;gCAAE,MAAM,CAAC,aAAa,CAAC,MAAM,IAAI,IAAI,KAAK,CAAC,wBAAwB,CAAC,CAAC,CAAC;4BAChG,IAAI,IAAI;gCAAE,MAAM;4BAChB,IAAI,IAAI,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;gCAC1B,4JAA4J;gCAC5J,IAAI,SAAS,IAAI,OAAO,EAAE;oCAAE,MAAM;gCAClC,MAAM,WAAW,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;4BAChC,CAAC;4BACD,gMAAgM;4BAChM,IAAI,IAAI,CAAC,IAAI,KAAK,QAAQ;gCAAE,SAAS,GAAG,IAAI,CAAC;;gCACxC,IAAI,CAAC,IAAI,CAAC,CAAC;4BAChB,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;wBACxB,CAAC;wBACD,iJAAiJ;wBACjJ,IAAI,CAAC,SAAS,IAAI,CAAC,WAAW,CAAC,UAAU,IAAI,CAAC,SAAS,IAAI,CAAC,OAAO,EAAE,EAAE,CAAC;4BACtE,MAAM,IAAI,KAAK,CAAC,yDAAyD,CAAC,CAAC;wBAC7E,CAAC;oBACH,CAAC;oBAAC,OAAO,KAAK,EAAE,CAAC;wBACf,8FAA8F;wBAC9F,IAAI,CAAC,SAAS,IAAI,CAAC,OAAO,EAAE;4BAAE,MAAM,KAAK,CAAC;oBAC5C,CAAC;4BAAS,CAAC;wBACT,aAAa,EAAE,mBAAmB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;wBACrD,MAAM,CAAC,WAAW,EAAE,mBAAmB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;wBAC1D,6GAA6G;wBAC7G,KAAK,MAAM,EAAE,MAAM,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;oBACxC,CAAC;oBACD,qGAAqG;oBACrG,qGAAqG;oBACrG,gGAAgG;oBAChG,MAAM,oBAAoB,GAAG,WAAW,CAAC,oBAAoB,CAAC,IAAA,wBAAU,GAAE,EAAE,IAAI,IAAI,EAAE,CAAC,CAAC;oBACxF,IAAI,oBAAoB,IAAI,CAAC,aAAa,EAAE,OAAO;wBAAE,IAAI,CAAC,oBAAoB,CAAC,CAAC;oBAChF,OAAO,mBAAmB,CAAC,WAAW,CAAC,MAAM,CAAC,YAAY,EAAE,OAAO,EAAE,YAAY,EAAE,QAAQ,CAAC,CAAC,CAAC;gBAChG,CAAC,EACD,EAAE,GAAG,gBAAgB,EAAE,IAAI,EAAE,UAAU,CAAC,IAAI,IAAI,QAAQ,CAAC,KAAK,EAAE,QAAQ,CAAC,EAAE,CAC5E,CAAC;YACJ,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,sFAAsF;gBACtF,YAAY,EAAE,CAAC;gBACf,MAAM,KAAK,CAAC;YACd,CAAC;YAED,+GAA+G;YAC/G,2GAA2G;YAC3G,oCAAoC;YACpC,KAAK,IAAI;iBACN,IAAI,CACH,CAAC,QAAQ,EAAE,EAAE;gBACX,IAAI,SAAS;oBAAE,OAAO;gBACtB,IAAI,QAAQ,EAAE,CAAC;oBACb,IAAI,CAAC;wBACH,IAAI,EAAE,QAAQ;wBACd,YAAY,EAAE,QAAQ,CAAC,YAAY;wBACnC,KAAK,EAAE,QAAQ,CAAC,KAAK;wBACrB,gBAAgB,EAAE,QAAQ,CAAC,gBAAgB;qBAC5C,CAAC,CAAC;gBACL,CAAC;qBAAM,CAAC;oBACN,KAAK,MAAM,IAAI,IAAI,WAAW,CAAC,QAAQ,CAAC;wBAAE,IAAI,CAAC,IAAI,CAAC,CAAC;gBACvD,CAAC;gBACD,UAAU,CAAC,KAAK,EAAE,CAAC;YACrB,CAAC,EACD,CAAC,KAAc,EAAE,EAAE;gBACjB,IAAI,CAAC,SAAS;oBAAE,UAAU,CAAC,KAAK,CAAC,IAAA,oCAAyB,EAAC,KAAK,CAAC,CAAC,CAAC;YACrE,CAAC,CACF;iBACA,OAAO,CAAC,YAAY,CAAC,CAAC;YAEzB,OAAO,EAAE,MAAM,EAAE,CAAC;QACpB,CAAC;KACF,CAAC;AACJ,CAAC;AAED,yHAAyH;AACzH,SAAgB,qBAAqB,CAAC,UAAsB,EAAE;IAC5D,MAAM,UAAU,GAAG,IAAA,kCAAuB,EAAC,OAAO,CAAC,CAAC;IACpD,OAAO;QACL,oBAAoB,EAAE,IAAI;QAC1B,SAAS,EAAE,KAAK,EAAE,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE;YACtC,IAAA,iCAAsB,EAAC,OAAO,CAAC,CAAC;YAChC,IAAI,CAAC,IAAA,+BAAoB,GAAE,EAAE,CAAC;gBAC5B,OAAO,MAAM,OAAO,EAAE,CAAC;YACzB,CAAC;YACD,MAAM,UAAU,GAAG,qBAAqB,CAAC,OAAO,CAAC,CAAC;YAClD,IAAI,CAAC;gBACH,OAAO,MAAM,eAAI,CAAC,OAAO,CAAC,KAAK,IAAI,EAAE,CAAC,OAAO,EAAE,EAAE;oBAC/C,GAAG,UAAU;oBACb,IAAI,EAAE,UAAU,CAAC,IAAI,IAAI,QAAQ,CAAC,KAAK,EAAE,OAAO,CAAC;iBAClD,CAAC,CAAC;YACL,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,MAAM,IAAA,oCAAyB,EAAC,KAAK,CAAC,CAAC;YACzC,CAAC;oBAAS,CAAC;gBACT,oBAAoB,CAAC,UAAU,CAAC,CAAC;YACnC,CAAC;QACH,CAAC;KACF,CAAC;AACJ,CAAC;AAED;;;;GAIG;AACH,SAAgB,iBAAiB,CAAC,UAAsB,EAAE;IACxD,MAAM,UAAU,GAAG,IAAA,kCAAuB,EAAC,OAAO,CAAC,CAAC;IACpD,OAAO;QACL,oBAAoB,EAAE,IAAI;QAC1B,YAAY,EAAE,KAAK,EAAE,EAAE,UAAU,EAAE,KAAK,EAAE,EAAE,EAAE;YAC5C,IAAA,iCAAsB,EAAC,eAAe,CAAC,CAAC;YACxC,IAAI,CAAC,IAAA,+BAAoB,GAAE,EAAE,CAAC;gBAC5B,OAAO,MAAM,UAAU,EAAE,CAAC;YAC5B,CAAC;YACD,IAAI,CAAC;gBACH,OAAO,MAAM,eAAI,CAAC,OAAO,CAAC,KAAK,IAAI,EAAE,CAAC,iBAAiB,CAAC,MAAM,UAAU,EAAE,CAAC,EAAE;oBAC3E,GAAG,UAAU;oBACb,IAAI,EAAE,UAAU,CAAC,IAAI,IAAI,QAAQ,CAAC,KAAK,EAAE,OAAO,CAAC;iBAClD,CAAC,CAAC;YACL,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,MAAM,IAAA,oCAAyB,EAAC,KAAK,CAAC,CAAC;YACzC,CAAC;QACH,CAAC;KACF,CAAC;AACJ,CAAC;AAED,uGAAuG;AACvG,SAAS,iBAAiB,CAAC,MAA0B;IACnD,MAAM,MAAM,GAAG,MAAM,CAAC,MAAiC,CAAC;IACxD,IAAI,MAAM,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,OAAO,KAAK,KAAK,QAAQ,CAAC,EAAE,CAAC;QACvD,OAAO,MAAM,CAAC;IAChB,CAAC;IACD,OAAO;QACL,GAAG,MAAM;QACT,MAAM,EAAE,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC;KAC3G,CAAC;AACJ,CAAC;AAED,SAAS,QAAQ,CAAC,KAAwD,EAAE,SAAiB;IAC3F,OAAO,GAAG,KAAK,CAAC,QAAQ,IAAI,KAAK,CAAC,OAAO,IAAI,SAAS,EAAE,CAAC;AAC3D,CAAC;AAED,8GAA8G;AAC9G,SAAS,WAAW,CAAC,KAAc;IACjC,IAAI,KAAK,YAAY,KAAK;QAAE,OAAO,KAAK,CAAC;IACzC,IAAI,OAAe,CAAC;IACpB,IAAI,CAAC;QACH,OAAO,GAAG,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;IACzF,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC;IAC1B,CAAC;IACD,MAAM,MAAM,GAAG,IAAI,KAAK,CAAC,OAAO,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC;IACpD,0HAA0H;IAC1H,MAAM,IAAI,GAAI,KAA+C,EAAE,IAAI,CAAC;IACpE,IAAI,OAAO,IAAI,KAAK,QAAQ;QAAE,MAAM,CAAC,IAAI,GAAG,IAAI,CAAC;IACjD,MAAM,WAAW,GAAI,KAAsD,EAAE,WAAW,CAAC;IACzF,OAAO,WAAW,KAAK,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE,EAAE,WAAW,EAAE,CAAC,CAAC;AACrF,CAAC;AAED,kHAAkH;AAClH,8GAA8G;AAC9G,SAAS,sBAAsB,CAAC,MAAqC;IACnE,MAAM,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC;IACjC,+GAA+G;IAC/G,IAAI,QAAQ,EAAE,EAAE,IAAI,IAAI,IAAI,QAAQ,EAAE,SAAS,IAAI,IAAI,EAAE,CAAC;QACxD,OAAO,MAAM,CAAC;IAChB,CAAC;IACD,OAAO;QACL,GAAG,MAAM;QACT,QAAQ,EAAE,EAAE,GAAG,QAAQ,EAAE,EAAE,EAAE,QAAQ,EAAE,EAAE,IAAI,IAAA,wBAAU,GAAE,EAAE,SAAS,EAAE,QAAQ,EAAE,SAAS,IAAI,IAAI,IAAI,EAAE,EAAE;KAC1G,CAAC;AACJ,CAAC;AAED,sGAAsG;AACtG,SAAS,mBAAmB,CAAC,MAAqC;IAChE,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;IACrD,OAAO,OAAO,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,CAAC,EAAE,EAAE,CAAC,IAAI,KAAK,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,GAAG,MAAM,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC;AACjG,CAAC;AAED,SAAS,gBAAgB,CAAC,IAA4B;IACpD,IACE,CAAC,IAAI,CAAC,IAAI,KAAK,MAAM,IAAI,IAAI,CAAC,IAAI,KAAK,gBAAgB,CAAC;QACxD,IAAI,CAAC,IAAI,CAAC,IAAI,KAAK,MAAM;QACzB,IAAI,CAAC,IAAI,CAAC,IAAI,YAAY,UAAU,EACpC,CAAC;QACD,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,sHAAsH;AACtH,MAAM,iBAAiB;IACJ,OAAO,GAA6B,EAAE,CAAC;IAChD,YAAY,GAAgC,EAAE,OAAO,EAAE,OAAO,EAAE,GAAG,EAAE,SAAS,EAAE,CAAC;IACjF,KAAK,GAAyB;QACpC,WAAW,EAAE,EAAE,KAAK,EAAE,SAAS,EAAE,OAAO,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE,UAAU,EAAE,SAAS,EAAE;QAClG,YAAY,EAAE,EAAE,KAAK,EAAE,SAAS,EAAE,IAAI,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE;KAC1E,CAAC;IACe,QAAQ,GAAsB,EAAE,CAAC;IAC1C,gBAAgB,CAA4B;IAC5C,gBAAgB,CAAmC;IAC3D,4HAA4H;IAC3G,UAAU,GAAG,IAAI,GAAG,EAA+B,CAAC;IACpD,eAAe,GAAG,IAAI,GAAG,EAAoC,CAAC;IAE/E,GAAG,CAAC,IAA+B;QACjC,QAAQ,IAAI,CAAC,IAAI,EAAE,CAAC;YAClB,KAAK,cAAc;gBACjB,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,QAAQ,CAAC,CAAC;gBACrC,MAAM;YACR,KAAK,YAAY,CAAC,CAAC,CAAC;gBAClB,MAAM,KAAK,GAAwB,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,EAAE,gBAAgB,EAAE,IAAI,CAAC,gBAAgB,EAAE,CAAC;gBACvG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC;gBACpC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;gBACzB,MAAM;YACR,CAAC;YACD,KAAK,YAAY,CAAC,CAAC,CAAC;gBAClB,MAAM,KAAK,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;gBAC5C,KAAK,CAAC,IAAI,IAAI,IAAI,CAAC,KAAK,CAAC;gBACzB,IAAI,IAAI,CAAC,gBAAgB;oBAAE,KAAK,CAAC,gBAAgB,GAAG,IAAI,CAAC,gBAAgB,CAAC;gBAC1E,MAAM;YACR,CAAC;YACD,KAAK,UAAU,CAAC,CAAC,CAAC;gBAChB,MAAM,KAAK,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;gBAC3C,IAAI,KAAK,IAAI,IAAI,CAAC,gBAAgB;oBAAE,KAAK,CAAC,gBAAgB,GAAG,IAAI,CAAC,gBAAgB,CAAC;gBACnF,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;gBAChC,MAAM;YACR,CAAC;YACD,KAAK,iBAAiB,CAAC,CAAC,CAAC;gBACvB,MAAM,KAAK,GAA6B;oBACtC,IAAI,EAAE,WAAW;oBACjB,IAAI,EAAE,EAAE;oBACR,gBAAgB,EAAE,IAAI,CAAC,gBAAgB;iBACxC,CAAC;gBACF,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC;gBACzC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;gBACzB,MAAM;YACR,CAAC;YACD,KAAK,iBAAiB,CAAC,CAAC,CAAC;gBACvB,MAAM,KAAK,GAAG,IAAI,CAAC,oBAAoB,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;gBACjD,KAAK,CAAC,IAAI,IAAI,IAAI,CAAC,KAAK,CAAC;gBACzB,IAAI,IAAI,CAAC,gBAAgB;oBAAE,KAAK,CAAC,gBAAgB,GAAG,IAAI,CAAC,gBAAgB,CAAC;gBAC1E,MAAM;YACR,CAAC;YACD,KAAK,eAAe,CAAC,CAAC,CAAC;gBACrB,MAAM,KAAK,GAAG,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;gBAChD,IAAI,KAAK,IAAI,IAAI,CAAC,gBAAgB;oBAAE,KAAK,CAAC,gBAAgB,GAAG,IAAI,CAAC,gBAAgB,CAAC;gBACnF,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;gBACrC,MAAM;YACR,CAAC;YACD,KAAK,mBAAmB;gBACtB,oFAAoF;gBACpF,IAAI,CAAC,gBAAgB,GAAG;oBACtB,EAAE,EAAE,IAAI,CAAC,EAAE,IAAI,IAAI,CAAC,gBAAgB,EAAE,EAAE;oBACxC,SAAS,EAAE,IAAI,CAAC,SAAS,IAAI,IAAI,CAAC,gBAAgB,EAAE,SAAS;oBAC7D,OAAO,EAAE,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,gBAAgB,EAAE,OAAO;iBACxD,CAAC;gBACF,MAAM;YACR,KAAK,QAAQ;gBACX,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,YAAY,CAAC;gBACtC,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC;gBACxB,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC,gBAAgB,CAAC;gBAC9C,MAAM;YACR,KAAK,kBAAkB,CAAC;YACxB,KAAK,kBAAkB,CAAC;YACxB,KAAK,gBAAgB,CAAC;YACtB,KAAK,KAAK,CAAC;YACX,KAAK,OAAO;gBACV,kEAAkE;gBAClE,MAAM;YACR;gBACE,+GAA+G;gBAC/G,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBACxB,MAAM;QACV,CAAC;IACH,CAAC;IAED,IAAI,UAAU;QACZ,OAAO,IAAI,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC;IACjC,CAAC;IAED,+GAA+G;IAC/G,+GAA+G;IAC/G,oBAAoB,CAAC,EAAU,EAAE,SAAe;QAC9C,6GAA6G;QAC7G,IAAI,IAAI,CAAC,gBAAgB,EAAE,EAAE,IAAI,IAAI,IAAI,IAAI,CAAC,gBAAgB,EAAE,SAAS,IAAI,IAAI,EAAE,CAAC;YAClF,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,IAAI,CAAC,gBAAgB,GAAG;YACtB,EAAE,EAAE,IAAI,CAAC,gBAAgB,EAAE,EAAE,IAAI,EAAE;YACnC,SAAS,EAAE,IAAI,CAAC,gBAAgB,EAAE,SAAS,IAAI,SAAS;YACxD,OAAO,EAAE,IAAI,CAAC,gBAAgB,EAAE,OAAO;SACxC,CAAC;QACF,OAAO,EAAE,IAAI,EAAE,mBAAmB,EAAE,GAAG,IAAI,CAAC,gBAAgB,EAAE,CAAC;IACjE,CAAC;IAED,kHAAkH;IAC1G,eAAe,CAAC,EAAU;QAChC,IAAI,KAAK,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QACpC,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,KAAK,GAAG,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC;YACnC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC;YAC/B,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC3B,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAEO,oBAAoB,CAAC,EAAU;QACrC,IAAI,KAAK,GAAG,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QACzC,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,KAAK,GAAG,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC;YACxC,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC;YACpC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC3B,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED,MAAM,CACJ,OAA4B,EAC5B,QAA+C;QAE/C,OAAO;YACL,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,YAAY,EAAE,IAAI,CAAC,YAAY;YAC/B,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,gBAAgB,EAAE,IAAI,CAAC,gBAAgB;YACvC,OAAO;YACP,QAAQ,EAAE,IAAI,CAAC,gBAAgB,CAAC,CAAC,CAAC,EAAE,GAAG,IAAI,CAAC,gBAAgB,EAAE,GAAG,QAAQ,EAAE,CAAC,CAAC,CAAC,QAAQ;SACvF,CAAC;IACJ,CAAC;CACF;AAED,oHAAoH;AACpH,QAAQ,CAAC,CAAC,WAAW,CAAC,MAAqC;IACzD,MAAM,EAAE,IAAI,EAAE,cAAc,EAAE,QAAQ,EAAE,MAAM,CAAC,QAAQ,IAAI,EAAE,EAAE,CAAC;IAChE,IAAI,MAAM,CAAC,QAAQ,EAAE,EAAE,KAAK,SAAS,IAAI,MAAM,CAAC,QAAQ,EAAE,SAAS,KAAK,SAAS,IAAI,MAAM,CAAC,QAAQ,EAAE,OAAO,KAAK,SAAS,EAAE,CAAC;QAC5H,MAAM;YACJ,IAAI,EAAE,mBAAmB;YACzB,EAAE,EAAE,MAAM,CAAC,QAAQ,CAAC,EAAE;YACtB,SAAS,EAAE,MAAM,CAAC,QAAQ,CAAC,SAAS;YACpC,OAAO,EAAE,MAAM,CAAC,QAAQ,CAAC,OAAO;SACjC,CAAC;IACJ,CAAC;IACD,IAAI,UAAU,GAAG,CAAC,CAAC;IACnB,KAAK,MAAM,IAAI,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;QAClC,MAAM,EAAE,GAAG,UAAU,UAAU,EAAE,EAAE,CAAC;QACpC,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YACzB,MAAM,EAAE,IAAI,EAAE,YAAY,EAAE,EAAE,EAAE,gBAAgB,EAAE,IAAI,CAAC,gBAAgB,EAAE,CAAC;YAC1E,IAAI,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC;gBAAE,MAAM,EAAE,IAAI,EAAE,YAAY,EAAE,EAAE,EAAE,KAAK,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC;YAC7E,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE,EAAE,EAAE,gBAAgB,EAAE,IAAI,CAAC,gBAAgB,EAAE,CAAC;QAC1E,CAAC;aAAM,IAAI,IAAI,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;YACrC,MAAM,EAAE,IAAI,EAAE,iBAAiB,EAAE,EAAE,EAAE,gBAAgB,EAAE,IAAI,CAAC,gBAAgB,EAAE,CAAC;YAC/E,IAAI,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC;gBAAE,MAAM,EAAE,IAAI,EAAE,iBAAiB,EAAE,EAAE,EAAE,KAAK,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC;YAClF,MAAM,EAAE,IAAI,EAAE,eAAe,EAAE,EAAE,EAAE,gBAAgB,EAAE,IAAI,CAAC,gBAAgB,EAAE,CAAC;QAC/E,CAAC;aAAM,IAAI,IAAI,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;YACrC,gIAAgI;YAChI,MAAM,EAAE,UAAU,EAAE,QAAQ,EAAE,gBAAgB,EAAE,OAAO,EAAE,GAAG,IAAI,CAAC;YACjE,MAAM,EAAE,IAAI,EAAE,kBAAkB,EAAE,EAAE,EAAE,UAAU,EAAE,QAAQ,EAAE,gBAAgB,EAAE,OAAO,EAAE,CAAC;YACxF,IAAI,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC;gBAAE,MAAM,EAAE,IAAI,EAAE,kBAAkB,EAAE,EAAE,EAAE,UAAU,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,CAAC;YACjG,MAAM,EAAE,IAAI,EAAE,gBAAgB,EAAE,EAAE,EAAE,UAAU,EAAE,CAAC;YACjD,MAAM,IAAI,CAAC;QACb,CAAC;aAAM,CAAC;YACN,MAAM,IAAI,CAAC;QACb,CAAC;IACH,CAAC;IACD,MAAM;QACJ,IAAI,EAAE,QAAQ;QACd,YAAY,EAAE,MAAM,CAAC,YAAY;QACjC,KAAK,EAAE,MAAM,CAAC,KAAK;QACnB,gBAAgB,EAAE,MAAM,CAAC,gBAAgB;KAC1C,CAAC;AACJ,CAAC"}

package/package.json

@@ -0,0 +1,64 @@
+{
+  "name": "@dbos-inc/vercel-ai",
+  "version": "0.1.5",
+  "description": "DBOS durable execution integration for the Vercel AI SDK",
+  "license": "MIT",
+  "type": "commonjs",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/dbos-inc/dbos-vercel-ai.git"
+  },
+  "bugs": {
+    "url": "https://github.com/dbos-inc/dbos-vercel-ai/issues"
+  },
+  "homepage": "https://docs.dbos.dev/",
+  "main": "dist/index.js",
+  "types": "dist/index.d.ts",
+  "engines": {
+    "node": ">=22"
+  },
+  "files": [
+    "dist",
+    "src"
+  ],
+  "publishConfig": {
+    "access": "public"
+  },
+  "scripts": {
+    "build": "tsc --project tsconfig.build.json",
+    "clean": "rm -rf dist",
+    "lint": "tsc --noEmit",
+    "pretest": "npm run build",
+    "test": "tsx --test tests/*.test.mts",
+    "check:package": "npm run build && publint && attw --pack .",
+    "prepublishOnly": "npm run clean && npm run build"
+  },
+  "keywords": [
+    "dbos",
+    "ai",
+    "vercel",
+    "ai-sdk",
+    "durable-execution",
+    "workflow",
+    "llm",
+    "agent"
+  ],
+  "peerDependencies": {
+    "@dbos-inc/dbos-sdk": "^4.21.0",
+    "ai": "^7.0.0"
+  },
+  "devDependencies": {
+    "@ai-sdk/gateway": "^4.0.0",
+    "@ai-sdk/provider": "^4.0.0",
+    "@arethetypeswrong/cli": "^0.18.0",
+    "@dbos-inc/dbos-sdk": "^4.23.6",
+    "@types/node": "^24.0.0",
+    "@types/pg": "^8.20.0",
+    "ai": "^7.0.0",
+    "pg": "^8.22.0",
+    "publint": "^0.3.0",
+    "tsx": "^4.20.0",
+    "typescript": "^5.8.0",
+    "zod": "^4.0.0"
+  }
+}

package/src/index.ts

@@ -0,0 +1,2 @@
+export { durableCalls, durableEmbeddingCalls, durableImageCalls } from './middleware';
+export { durableMCPTools, DurableMCPToolsOptions, MCPClientLike } from './mcp';

package/src/internal.ts

@@ -0,0 +1,66 @@
+import { DBOS, StepConfig } from '@dbos-inc/dbos-sdk';
+
+export function isInWorkflowFunction(): boolean {
+  // True only in workflow code proper: not in a step, transaction, or outside DBOS.
+  return DBOS.isInWorkflow();
+}
+
+export function assertNotInTransaction(operation: string): void {
+  if (DBOS.isInTransaction()) {
+    throw new Error(`Cannot call ${operation} inside a DBOS transaction; run it in workflow or step code.`);
+  }
+}
+
+// Aborts/timeouts are deliberate cancellations, never transient; retrying just re-runs an already-cancelled call.
+function isAbortError(error: unknown): boolean {
+  const name = (error as { name?: unknown } | null)?.name;
+  return name === 'AbortError' || name === 'TimeoutError';
+}
+
+// AI SDK errors (APICallError, GatewayError) expose an isRetryable flag; treat an explicit false, and any abort, as terminal.
+function isNonRetryable(error: unknown): boolean {
+  try {
+    return (
+      isAbortError(error) ||
+      (typeof error === 'object' &&
+        error !== null &&
+        'isRetryable' in error &&
+        (error as { isRetryable?: unknown }).isRetryable === false)
+    );
+  } catch {
+    // A throwing accessor must not replace the step's real error; treat as retryable.
+    return false;
+  }
+}
+
+// serialize-error drops the Symbol-keyed AI SDK error markers on replay, so isInstance() fails on the revived error.
+// The markers derive from error.name (which survives), so re-attach them; no @ai-sdk/provider/@ai-sdk/gateway import.
+// AISDKError (APICallError, ...) and GatewayError use distinct marker namespaces — ai's retry predicate checks both.
+export function restoreAISDKErrorIdentity(error: unknown): unknown {
+  // Wrap the whole body: identity restoration must never replace or mask the real error (a frozen/non-extensible
+  // error throws on assignment, a throwing `name` getter throws on read) — on failure, return the error untouched.
+  try {
+    const name = (error as { name?: unknown } | null)?.name;
+    if (error === null || typeof error !== 'object' || typeof name !== 'string') return error;
+    const base = name.startsWith('AI_') ? 'vercel.ai.error' : name.startsWith('Gateway') ? 'vercel.ai.gateway.error' : undefined;
+    if (base !== undefined) {
+      (error as Record<symbol, unknown>)[Symbol.for(base)] = true;
+      (error as Record<symbol, unknown>)[Symbol.for(`${base}.${name}`)] = true;
+    }
+  } catch {
+    // Fall through and return the error as-is.
+  }
+  return error;
+}
+
+// Default to DBOS-owned retries so a transient provider error is absorbed inside one step (never checkpointed as an
+// error that replay would re-run); the default shouldRetry skips provider-declared non-retryable errors and aborts.
+// A caller's retriesAllowed/shouldRetry wins, but an explicit `undefined` falls back to the default.
+export function withErrorClassification(options: StepConfig): StepConfig {
+  return {
+    ...options,
+    retriesAllowed: options.retriesAllowed ?? true,
+    shouldRetry: options.shouldRetry ?? ((error: unknown) => !isNonRetryable(error)),
+  };
+}
+