@convex-dev/agent 0.5.0-alpha.1 → 0.6.0-alpha.0

package/dist/deltas.js

@@ -0,0 +1,447 @@
+import { readUIMessageStream, } from "ai";
+import { assert, pick } from "convex-helpers";
+import {} from "./UIMessages.js";
+import { joinText, sorted } from "./shared.js";
+import {} from "./validators.js";
+import { getErrorMessage } from "@ai-sdk/provider-utils";
+export function blankUIMessage(streamMessage, threadId) {
+    return {
+        id: `stream:${streamMessage.streamId}`,
+        key: `${threadId}-${streamMessage.order}-${streamMessage.stepOrder}`,
+        order: streamMessage.order,
+        stepOrder: streamMessage.stepOrder,
+        status: statusFromStreamStatus(streamMessage.status),
+        agentName: streamMessage.agentName,
+        text: "",
+        _creationTime: Date.now(),
+        role: "assistant",
+        parts: [],
+        ...(streamMessage.metadata ? { metadata: streamMessage.metadata } : {}),
+    };
+}
+export function statusFromStreamStatus(status) {
+    switch (status) {
+        case "streaming":
+            return "streaming";
+        case "finished":
+            return "success";
+        case "aborted":
+            return "failed";
+        default:
+            return "pending";
+    }
+}
+export async function updateFromUIMessageChunks(uiMessage, parts) {
+    const partsStream = new ReadableStream({
+        start(controller) {
+            for (const part of parts) {
+                controller.enqueue(part);
+            }
+            controller.close();
+        },
+    });
+    let failed = false;
+    let suppressError = false;
+    const messageStream = readUIMessageStream({
+        message: uiMessage,
+        stream: partsStream,
+        onError: (e) => {
+            const errorMessage = e instanceof Error ? e.message : String(e);
+            // Tool invocation errors can be safely ignored when streaming continuation
+            // after tool approval - the stored messages have the complete tool context
+            if (errorMessage.toLowerCase().includes("no tool invocation found")) {
+                // Silently suppress - this is expected after tool approval when the
+                // continuation stream has tool-result without the original tool-call
+                suppressError = true;
+                return;
+            }
+            failed = true;
+            console.error("Error in stream", e);
+        },
+        terminateOnError: true,
+    });
+    let message = uiMessage;
+    try {
+        for await (const messagePart of messageStream) {
+            assert(messagePart.id === message.id, `Expecting to only make one UIMessage in a stream`);
+            message = messagePart;
+        }
+    }
+    catch (e) {
+        // If we've already handled this error in onError and marked it as suppressed,
+        // don't rethrow - the stored messages provide the fallback
+        if (!suppressError) {
+            throw e;
+        }
+    }
+    if (failed) {
+        message.status = "failed";
+    }
+    message.text = joinText(message.parts);
+    return message;
+}
+export async function deriveUIMessagesFromDeltas(threadId, streamMessages, allDeltas) {
+    const messages = [];
+    for (const streamMessage of streamMessages) {
+        if (streamMessage.format === "UIMessageChunk") {
+            const { parts } = getParts(allDeltas.filter((d) => d.streamId === streamMessage.streamId), 0);
+            const uiMessage = await updateFromUIMessageChunks(blankUIMessage(streamMessage, threadId), parts);
+            // TODO: this fails on partial tool calls
+            messages.push(uiMessage);
+        }
+        else {
+            const [uiMessages] = deriveUIMessagesFromTextStreamParts(threadId, [streamMessage], [], allDeltas);
+            messages.push(...uiMessages);
+        }
+    }
+    return sorted(messages);
+}
+/**
+ *
+ */
+export function deriveUIMessagesFromTextStreamParts(threadId, streamMessages, existingStreams, allDeltas) {
+    const newStreams = [];
+    // Seed the existing chunks
+    let changed = false;
+    for (const streamMessage of streamMessages) {
+        const deltas = allDeltas.filter((d) => d.streamId === streamMessage.streamId);
+        const existing = existingStreams.find((s) => s.streamId === streamMessage.streamId);
+        const [newStream, messageChanged] = updateFromTextStreamParts(threadId, streamMessage, existing, deltas);
+        newStreams.push(newStream);
+        if (messageChanged)
+            changed = true;
+    }
+    for (const { streamId } of existingStreams) {
+        if (!newStreams.find((s) => s.streamId === streamId)) {
+            // There's a stream that's no longer active.
+            changed = true;
+        }
+    }
+    const messages = sorted(newStreams.map((s) => s.message));
+    return [messages, newStreams, changed];
+}
+export function getParts(deltas, fromCursor) {
+    const parts = [];
+    let cursor = fromCursor ?? 0;
+    for (const delta of deltas.sort((a, b) => a.start - b.start)) {
+        if (delta.parts.length === 0) {
+            console.debug(`Got delta with no parts: ${JSON.stringify(delta)}`);
+            continue;
+        }
+        if (cursor !== delta.start) {
+            if (cursor >= delta.end) {
+                continue;
+            }
+            else if (cursor < delta.start) {
+                console.warn(`Got delta for stream ${delta.streamId} that has a gap ${cursor} -> ${delta.start}`);
+                break;
+            }
+            else {
+                throw new Error(`Got unexpected delta for stream ${delta.streamId}: delta: ${delta.start} -> ${delta.end} existing cursor: ${cursor}`);
+            }
+        }
+        parts.push(...delta.parts);
+        cursor = delta.end;
+    }
+    return { parts, cursor };
+}
+/**
+ * This is historically from when we would use the onChunk callback instead of
+ * consuming the full UIMessageStream.
+ */
+// exported for testing
+export function updateFromTextStreamParts(threadId, streamMessage, existing, deltas) {
+    const { cursor, parts } = getParts(deltas, existing?.cursor);
+    const changed = parts.length > 0 ||
+        (existing &&
+            statusFromStreamStatus(streamMessage.status) !== existing.message.status);
+    const existingMessage = existing?.message ?? blankUIMessage(streamMessage, threadId);
+    if (!changed) {
+        return [
+            existing ?? {
+                streamId: streamMessage.streamId,
+                cursor,
+                message: existingMessage,
+            },
+            false,
+        ];
+    }
+    const message = structuredClone(existingMessage);
+    message.status = statusFromStreamStatus(streamMessage.status);
+    const textPartsById = new Map();
+    const toolPartsById = new Map(message.parts
+        .filter((p) => p.type.startsWith("tool-") || p.type === "dynamic-tool")
+        .map((p) => [p.toolCallId, p]));
+    const reasoningPartsById = new Map();
+    for (const part of parts) {
+        switch (part.type) {
+            case "text-start":
+            case "text-delta": {
+                if (!textPartsById.has(part.id)) {
+                    const lastPart = message.parts.at(-1);
+                    if (lastPart?.type === "text") {
+                        textPartsById.set(part.id, lastPart);
+                    }
+                    else {
+                        const newPart = {
+                            type: "text",
+                            text: "",
+                            providerMetadata: part.providerMetadata,
+                        };
+                        textPartsById.set(part.id, newPart);
+                        message.parts.push(newPart);
+                    }
+                }
+                if (part.type === "text-delta") {
+                    const textPart = textPartsById.get(part.id);
+                    textPart.text += part.text;
+                    textPart.providerMetadata = mergeProviderMetadata(textPart.providerMetadata, part.providerMetadata);
+                }
+                break;
+            }
+            case "tool-input-start": {
+                let newPart;
+                if (part.dynamic) {
+                    newPart = {
+                        type: "dynamic-tool",
+                        toolCallId: part.id,
+                        toolName: part.toolName,
+                        state: "input-streaming",
+                        input: "",
+                    };
+                }
+                else {
+                    newPart = {
+                        type: `tool-${part.toolName}`,
+                        toolCallId: part.id,
+                        state: "input-streaming",
+                        input: "",
+                        providerExecuted: part.providerExecuted,
+                    };
+                }
+                toolPartsById.set(part.id, newPart);
+                message.parts.push(newPart);
+                break;
+            }
+            case "tool-input-delta":
+                {
+                    const toUpdate = toolPartsById.get(part.id);
+                    assert(toUpdate, `Expected to find tool call part ${part.id} to update`);
+                    toUpdate.input = (toUpdate.input ?? "") + part.delta;
+                }
+                break;
+            case "tool-input-end":
+                {
+                    const toUpdate = toolPartsById.get(part.id);
+                    assert(toUpdate, `Expected to find tool call part ${part.id} to update`);
+                    toUpdate.state = "input-available";
+                    if (part.providerMetadata) {
+                        const updatable = toUpdate;
+                        updatable.callProviderMetadata = mergeProviderMetadata(updatable.callProviderMetadata, part.providerMetadata);
+                    }
+                }
+                break;
+            case "tool-call": {
+                let newPart;
+                if (part.dynamic) {
+                    newPart = {
+                        type: "dynamic-tool",
+                        toolCallId: part.toolCallId,
+                        toolName: part.toolName,
+                        input: part.input,
+                        state: "input-available",
+                    };
+                }
+                else {
+                    newPart = {
+                        type: `tool-${part.toolName}`,
+                        toolCallId: part.toolCallId,
+                        input: part.input,
+                        state: "input-available",
+                    };
+                    if (part.providerExecuted) {
+                        newPart.providerExecuted = part.providerExecuted;
+                    }
+                }
+                if (part.providerMetadata) {
+                    newPart.callProviderMetadata = part.providerMetadata;
+                }
+                if (toolPartsById.has(part.toolCallId)) {
+                    const toUpdate = toolPartsById.get(part.toolCallId);
+                    Object.assign(toUpdate, newPart);
+                }
+                else {
+                    toolPartsById.set(part.toolCallId, newPart);
+                    message.parts.push(newPart);
+                }
+                break;
+            }
+            case "tool-result": {
+                const toolCall = toolPartsById.get(part.toolCallId);
+                assert(toolCall, `Expected to find tool call part ${part.toolCallId} to update with result`);
+                let newPart;
+                if (toolCall.type === "dynamic-tool") {
+                    newPart = {
+                        ...toolCall,
+                        state: "output-available",
+                        input: part.input ?? toolCall.input,
+                        output: part.output ?? toolCall.output,
+                        ...pick(part, ["preliminary"]),
+                    };
+                }
+                else {
+                    newPart = {
+                        ...toolCall,
+                        state: "output-available",
+                        input: part.input ?? toolCall.input,
+                        output: part.output ?? toolCall.output,
+                        preliminary: part.preliminary,
+                    };
+                }
+                Object.assign(toolCall, newPart);
+                break;
+            }
+            case "reasoning-start":
+            case "reasoning-delta": {
+                if (!reasoningPartsById.has(part.id)) {
+                    const lastPart = message.parts.at(-1);
+                    if (lastPart?.type === "reasoning") {
+                        reasoningPartsById.set(part.id, lastPart);
+                    }
+                    else {
+                        const newPart = {
+                            type: "reasoning",
+                            state: "streaming",
+                            text: "",
+                            providerMetadata: part.providerMetadata,
+                        };
+                        reasoningPartsById.set(part.id, newPart);
+                        message.parts.push(newPart);
+                    }
+                }
+                const reasoningPart = reasoningPartsById.get(part.id);
+                if (part.type === "reasoning-delta") {
+                    reasoningPart.text += part.text;
+                    reasoningPart.providerMetadata = mergeProviderMetadata(reasoningPart.providerMetadata, part.providerMetadata);
+                }
+                break;
+            }
+            case "reasoning-end": {
+                const reasoningPart = reasoningPartsById.get(part.id) ??
+                    message.parts.find((p) => p.type === "reasoning" && p.state === "streaming");
+                if (reasoningPart) {
+                    reasoningPart.state = "done";
+                }
+                else {
+                    console.warn(`Expected to find reasoning part ${part.id} to finish, but found none`);
+                }
+                break;
+            }
+            case "source":
+                if (part.sourceType === "url") {
+                    message.parts.push({
+                        type: "source-url",
+                        url: part.url,
+                        sourceId: part.id,
+                        providerMetadata: part.providerMetadata,
+                        title: part.title,
+                    });
+                }
+                else if (part.sourceType === "document") {
+                    message.parts.push({
+                        type: "source-document",
+                        mediaType: part.mediaType,
+                        sourceId: part.id,
+                        title: part.title,
+                        filename: part.filename,
+                        providerMetadata: part.providerMetadata,
+                    });
+                }
+                else {
+                    console.warn("Got source part with unknown source type", part);
+                }
+                break;
+            case "abort":
+                message.status = "failed";
+                break;
+            case "error":
+                message.status = "failed";
+                console.warn("Generation failed with error", part.error);
+                break;
+            case "tool-error": {
+                const toolPart = toolPartsById.get(part.toolCallId);
+                if (toolPart) {
+                    toolPart.errorText = getErrorMessage(part.error);
+                }
+                break;
+            }
+            case "tool-approval-request": {
+                const typedPart = part;
+                const toolPart = toolPartsById.get(typedPart.toolCallId);
+                if (toolPart) {
+                    toolPart.state = "approval-requested";
+                    toolPart.approval = {
+                        id: typedPart.approvalId,
+                    };
+                }
+                else {
+                    console.warn(`Expected tool call part ${typedPart.toolCallId} for approval request`);
+                }
+                break;
+            }
+            case "file":
+            case "text-end":
+            case "finish-step":
+            case "finish":
+            case "raw":
+            case "start-step":
+            case "start":
+                // ignore
+                break;
+            default: {
+                // Exhaustiveness check disabled intentionally for forwards compatibility.
+                // New TextStreamPart types from future AI SDK versions will trigger a
+                // runtime warning rather than a compile error, allowing graceful degradation.
+                // const _: never = part;
+                console.warn(`Received unexpected part: ${JSON.stringify(part)}`);
+                break;
+            }
+        }
+    }
+    // Consider reasoning done once something else happens
+    for (let i = 0; i < message.parts.length - 1; i++) {
+        const part = message.parts[i];
+        if (part.type === "reasoning") {
+            part.state = "done";
+        }
+    }
+    message.text = joinText(message.parts);
+    return [
+        {
+            streamId: streamMessage.streamId,
+            cursor,
+            message,
+        },
+        true,
+    ];
+}
+function mergeProviderMetadata(existing, part) {
+    if (!existing && !part) {
+        return undefined;
+    }
+    if (!existing) {
+        return part;
+    }
+    if (!part) {
+        return existing;
+    }
+    const merged = existing;
+    for (const [provider, metadata] of Object.entries(part)) {
+        merged[provider] = {
+            ...merged[provider],
+            ...metadata,
+        };
+    }
+    return merged;
+}
+//# sourceMappingURL=deltas.js.map

package/dist/deltas.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"deltas.js","sourceRoot":"","sources":["../src/deltas.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,mBAAmB,GASpB,MAAM,IAAI,CAAC;AACZ,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,gBAAgB,CAAC;AAC9C,OAAO,EAAkB,MAAM,iBAAiB,CAAC;AACjD,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AAC/C,OAAO,EAIN,MAAM,iBAAiB,CAAC;AACzB,OAAO,EAAE,eAAe,EAAE,MAAM,wBAAwB,CAAC;AAEzD,MAAM,UAAU,cAAc,CAC5B,aAAsD,EACtD,QAAgB;IAEhB,OAAO;QACL,EAAE,EAAE,UAAU,aAAa,CAAC,QAAQ,EAAE;QACtC,GAAG,EAAE,GAAG,QAAQ,IAAI,aAAa,CAAC,KAAK,IAAI,aAAa,CAAC,SAAS,EAAE;QACpE,KAAK,EAAE,aAAa,CAAC,KAAK;QAC1B,SAAS,EAAE,aAAa,CAAC,SAAS;QAClC,MAAM,EAAE,sBAAsB,CAAC,aAAa,CAAC,MAAM,CAAC;QACpD,SAAS,EAAE,aAAa,CAAC,SAAS;QAClC,IAAI,EAAE,EAAE;QACR,aAAa,EAAE,IAAI,CAAC,GAAG,EAAE;QACzB,IAAI,EAAE,WAAW;QACjB,KAAK,EAAE,EAAE;QACT,GAAG,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,aAAa,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KACxE,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,sBAAsB,CACpC,MAA+B;IAE/B,QAAQ,MAAM,EAAE,CAAC;QACf,KAAK,WAAW;YACd,OAAO,WAAW,CAAC;QACrB,KAAK,UAAU;YACb,OAAO,SAAS,CAAC;QACnB,KAAK,SAAS;YACZ,OAAO,QAAQ,CAAC;QAClB;YACE,OAAO,SAAS,CAAC;IACrB,CAAC;AACH,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,yBAAyB,CAC7C,SAAoB,EACpB,KAAuB;IAEvB,MAAM,WAAW,GAAG,IAAI,cAAc,CAAiB;QACrD,KAAK,CAAC,UAAU;YACd,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;gBACzB,UAAU,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;YAC3B,CAAC;YACD,UAAU,CAAC,KAAK,EAAE,CAAC;QACrB,CAAC;KACF,CAAC,CAAC;IACH,IAAI,MAAM,GAAG,KAAK,CAAC;IACnB,IAAI,aAAa,GAAG,KAAK,CAAC;IAC1B,MAAM,aAAa,GAAG,mBAAmB,CAAC;QACxC,OAAO,EAAE,SAAS;QAClB,MAAM,EAAE,WAAW;QACnB,OAAO,EAAE,CAAC,CAAC,EAAE,EAAE;YACb,MAAM,YAAY,GAAG,CAAC,YAAY,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;YAChE,2EAA2E;YAC3E,2EAA2E;YAC3E,IAAI,YAAY,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,0BAA0B,CAAC,EAAE,CAAC;gBACpE,oEAAoE;gBACpE,qEAAqE;gBACrE,aAAa,GAAG,IAAI,CAAC;gBACrB,OAAO;YACT,CAAC;YACD,MAAM,GAAG,IAAI,CAAC;YACd,OAAO,CAAC,KAAK,CAAC,iBAAiB,EAAE,CAAC,CAAC,CAAC;QACtC,CAAC;QACD,gBAAgB,EAAE,IAAI;KACvB,CAAC,CAAC;IACH,IAAI,OAAO,GAAG,SAAS,CAAC;IACxB,IAAI,CAAC;QACH,IAAI,KAAK,EAAE,MAAM,WAAW,IAAI,aAAa,EAAE,CAAC;YAC9C,MAAM,CACJ,WAAW,CAAC,EAAE,KAAK,OAAO,CAAC,EAAE,EAC7B,kDAAkD,CACnD,CAAC;YACF,OAAO,GAAG,WAAW,CAAC;QACxB,CAAC;IACH,CAAC;IAAC,OAAO,CAAC,EAAE,CAAC;QACX,8EAA8E;QAC9E,2DAA2D;QAC3D,IAAI,CAAC,aAAa,EAAE,CAAC;YACnB,MAAM,CAAC,CAAC;QACV,CAAC;IACH,CAAC;IACD,IAAI,MAAM,EAAE,CAAC;QACX,OAAO,CAAC,MAAM,GAAG,QAAQ,CAAC;IAC5B,CAAC;IACD,OAAO,CAAC,IAAI,GAAG,QAAQ,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;IACvC,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,0BAA0B,CAC9C,QAAgB,EAChB,cAA+B,EAC/B,SAAwB;IAExB,MAAM,QAAQ,GAAgB,EAAE,CAAC;IACjC,KAAK,MAAM,aAAa,IAAI,cAAc,EAAE,CAAC;QAC3C,IAAI,aAAa,CAAC,MAAM,KAAK,gBAAgB,EAAE,CAAC;YAC9C,MAAM,EAAE,KAAK,EAAE,GAAG,QAAQ,CACxB,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,KAAK,aAAa,CAAC,QAAQ,CAAC,EAC9D,CAAC,CACF,CAAC;YACF,MAAM,SAAS,GAAG,MAAM,yBAAyB,CAC/C,cAAc,CAAC,aAAa,EAAE,QAAQ,CAAC,EACvC,KAAK,CACN,CAAC;YACF,yCAAyC;YACzC,QAAQ,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAC3B,CAAC;aAAM,CAAC;YACN,MAAM,CAAC,UAAU,CAAC,GAAG,mCAAmC,CACtD,QAAQ,EACR,CAAC,aAAa,CAAC,EACf,EAAE,EACF,SAAS,CACV,CAAC;YACF,QAAQ,CAAC,IAAI,CAAC,GAAG,UAAU,CAAC,CAAC;QAC/B,CAAC;IACH,CAAC;IACD,OAAO,MAAM,CAAC,QAAQ,CAAC,CAAC;AAC1B,CAAC;AAED;;GAEG;AAEH,MAAM,UAAU,mCAAmC,CACjD,QAAgB,EAChB,cAA+B,EAC/B,eAIE,EACF,SAAwB;IAMxB,MAAM,UAAU,GAIX,EAAE,CAAC;IACR,2BAA2B;IAC3B,IAAI,OAAO,GAAG,KAAK,CAAC;IACpB,KAAK,MAAM,aAAa,IAAI,cAAc,EAAE,CAAC;QAC3C,MAAM,MAAM,GAAG,SAAS,CAAC,MAAM,CAC7B,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,KAAK,aAAa,CAAC,QAAQ,CAC7C,CAAC;QACF,MAAM,QAAQ,GAAG,eAAe,CAAC,IAAI,CACnC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,KAAK,aAAa,CAAC,QAAQ,CAC7C,CAAC;QACF,MAAM,CAAC,SAAS,EAAE,cAAc,CAAC,GAAG,yBAAyB,CAC3D,QAAQ,EACR,aAAa,EACb,QAAQ,EACR,MAAM,CACP,CAAC;QACF,UAAU,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAC3B,IAAI,cAAc;YAAE,OAAO,GAAG,IAAI,CAAC;IACrC,CAAC;IACD,KAAK,MAAM,EAAE,QAAQ,EAAE,IAAI,eAAe,EAAE,CAAC;QAC3C,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,KAAK,QAAQ,CAAC,EAAE,CAAC;YACrD,4CAA4C;YAC5C,OAAO,GAAG,IAAI,CAAC;QACjB,CAAC;IACH,CAAC;IACD,MAAM,QAAQ,GAAG,MAAM,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC;IAC1D,OAAO,CAAC,QAAQ,EAAE,UAAU,EAAE,OAAO,CAAC,CAAC;AACzC,CAAC;AAED,MAAM,UAAU,QAAQ,CACtB,MAAqB,EACrB,UAAmB;IAEnB,MAAM,KAAK,GAAQ,EAAE,CAAC;IACtB,IAAI,MAAM,GAAG,UAAU,IAAI,CAAC,CAAC;IAC7B,KAAK,MAAM,KAAK,IAAI,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC,EAAE,CAAC;QAC7D,IAAI,KAAK,CAAC,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC7B,OAAO,CAAC,KAAK,CAAC,4BAA4B,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;YACnE,SAAS;QACX,CAAC;QACD,IAAI,MAAM,KAAK,KAAK,CAAC,KAAK,EAAE,CAAC;YAC3B,IAAI,MAAM,IAAI,KAAK,CAAC,GAAG,EAAE,CAAC;gBACxB,SAAS;YACX,CAAC;iBAAM,IAAI,MAAM,GAAG,KAAK,CAAC,KAAK,EAAE,CAAC;gBAChC,OAAO,CAAC,IAAI,CACV,wBAAwB,KAAK,CAAC,QAAQ,mBAAmB,MAAM,OAAO,KAAK,CAAC,KAAK,EAAE,CACpF,CAAC;gBACF,MAAM;YACR,CAAC;iBAAM,CAAC;gBACN,MAAM,IAAI,KAAK,CACb,mCAAmC,KAAK,CAAC,QAAQ,YAAY,KAAK,CAAC,KAAK,OAAO,KAAK,CAAC,GAAG,qBAAqB,MAAM,EAAE,CACtH,CAAC;YACJ,CAAC;QACH,CAAC;QACD,KAAK,CAAC,IAAI,CAAC,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC;QAC3B,MAAM,GAAG,KAAK,CAAC,GAAG,CAAC;IACrB,CAAC;IACD,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC;AAC3B,CAAC;AAED;;;GAGG;AAEH,uBAAuB;AACvB,MAAM,UAAU,yBAAyB,CACvC,QAAgB,EAChB,aAA4B,EAC5B,QAEa,EACb,MAAqB;IAErB,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,QAAQ,CAChC,MAAM,EACN,QAAQ,EAAE,MAAM,CACjB,CAAC;IACF,MAAM,OAAO,GACX,KAAK,CAAC,MAAM,GAAG,CAAC;QAChB,CAAC,QAAQ;YACP,sBAAsB,CAAC,aAAa,CAAC,MAAM,CAAC,KAAK,QAAQ,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;IAC9E,MAAM,eAAe,GACnB,QAAQ,EAAE,OAAO,IAAI,cAAc,CAAC,aAAa,EAAE,QAAQ,CAAC,CAAC;IAC/D,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,OAAO;YACL,QAAQ,IAAI;gBACV,QAAQ,EAAE,aAAa,CAAC,QAAQ;gBAChC,MAAM;gBACN,OAAO,EAAE,eAAe;aACzB;YACD,KAAK;SACN,CAAC;IACJ,CAAC;IAED,MAAM,OAAO,GAAc,eAAe,CAAC,eAAe,CAAC,CAAC;IAC5D,OAAO,CAAC,MAAM,GAAG,sBAAsB,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC;IAE9D,MAAM,aAAa,GAAG,IAAI,GAAG,EAAsB,CAAC;IACpD,MAAM,aAAa,GAAG,IAAI,GAAG,CAC3B,OAAO,CAAC,KAAK;SACV,MAAM,CACL,CAAC,CAAC,EAAuC,EAAE,CACzC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,IAAI,KAAK,cAAc,CAC1D;SACA,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,CACjC,CAAC;IACF,MAAM,kBAAkB,GAAG,IAAI,GAAG,EAA2B,CAAC;IAE9D,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,QAAQ,IAAI,CAAC,IAAI,EAAE,CAAC;YAClB,KAAK,YAAY,CAAC;YAClB,KAAK,YAAY,CAAC,CAAC,CAAC;gBAClB,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC;oBAChC,MAAM,QAAQ,GAAG,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;oBACtC,IAAI,QAAQ,EAAE,IAAI,KAAK,MAAM,EAAE,CAAC;wBAC9B,aAAa,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,QAAQ,CAAC,CAAC;oBACvC,CAAC;yBAAM,CAAC;wBACN,MAAM,OAAO,GAAG;4BACd,IAAI,EAAE,MAAM;4BACZ,IAAI,EAAE,EAAE;4BACR,gBAAgB,EAAE,IAAI,CAAC,gBAAgB;yBACnB,CAAC;wBACvB,aAAa,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;wBACpC,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;oBAC9B,CAAC;gBACH,CAAC;gBACD,IAAI,IAAI,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;oBAC/B,MAAM,QAAQ,GAAG,aAAa,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAE,CAAC;oBAC7C,QAAQ,CAAC,IAAI,IAAI,IAAI,CAAC,IAAI,CAAC;oBAC3B,QAAQ,CAAC,gBAAgB,GAAG,qBAAqB,CAC/C,QAAQ,CAAC,gBAAgB,EACzB,IAAI,CAAC,gBAAgB,CACtB,CAAC;gBACJ,CAAC;gBACD,MAAM;YACR,CAAC;YACD,KAAK,kBAAkB,CAAC,CAAC,CAAC;gBACxB,IAAI,OAAuC,CAAC;gBAC5C,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;oBACjB,OAAO,GAAG;wBACR,IAAI,EAAE,cAAc;wBACpB,UAAU,EAAE,IAAI,CAAC,EAAE;wBACnB,QAAQ,EAAE,IAAI,CAAC,QAAQ;wBACvB,KAAK,EAAE,iBAAiB;wBACxB,KAAK,EAAE,EAAE;qBACkB,CAAC;gBAChC,CAAC;qBAAM,CAAC;oBACN,OAAO,GAAG;wBACR,IAAI,EAAE,QAAQ,IAAI,CAAC,QAAQ,EAAE;wBAC7B,UAAU,EAAE,IAAI,CAAC,EAAE;wBACnB,KAAK,EAAE,iBAAiB;wBACxB,KAAK,EAAE,EAAE;wBACT,gBAAgB,EAAE,IAAI,CAAC,gBAAgB;qBACnB,CAAC;gBACzB,CAAC;gBACD,aAAa,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;gBACpC,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;gBAC5B,MAAM;YACR,CAAC;YACD,KAAK,kBAAkB;gBACrB,CAAC;oBACC,MAAM,QAAQ,GAAG,aAAa,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;oBAC5C,MAAM,CACJ,QAAQ,EACR,mCAAmC,IAAI,CAAC,EAAE,YAAY,CACvD,CAAC;oBACF,QAAQ,CAAC,KAAK,GAAG,CAAC,QAAQ,CAAC,KAAK,IAAI,EAAE,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC;gBACvD,CAAC;gBACD,MAAM;YACR,KAAK,gBAAgB;gBACnB,CAAC;oBACC,MAAM,QAAQ,GAAG,aAAa,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;oBAC5C,MAAM,CACJ,QAAQ,EACR,mCAAmC,IAAI,CAAC,EAAE,YAAY,CACvD,CAAC;oBACF,QAAQ,CAAC,KAAK,GAAG,iBAAiB,CAAC;oBACnC,IAAI,IAAI,CAAC,gBAAgB,EAAE,CAAC;wBAC1B,MAAM,SAAS,GAAG,QAGjB,CAAC;wBACF,SAAS,CAAC,oBAAoB,GAAG,qBAAqB,CACpD,SAAS,CAAC,oBAAoB,EAC9B,IAAI,CAAC,gBAAgB,CACtB,CAAC;oBACJ,CAAC;gBACH,CAAC;gBACD,MAAM;YACR,KAAK,WAAW,CAAC,CAAC,CAAC;gBACjB,IAAI,OAAuC,CAAC;gBAC5C,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;oBACjB,OAAO,GAAG;wBACR,IAAI,EAAE,cAAc;wBACpB,UAAU,EAAE,IAAI,CAAC,UAAU;wBAC3B,QAAQ,EAAE,IAAI,CAAC,QAAQ;wBACvB,KAAK,EAAE,IAAI,CAAC,KAAK;wBACjB,KAAK,EAAE,iBAAiB;qBACzB,CAAC;gBACJ,CAAC;qBAAM,CAAC;oBACN,OAAO,GAAG;wBACR,IAAI,EAAE,QAAQ,IAAI,CAAC,QAAQ,EAAE;wBAC7B,UAAU,EAAE,IAAI,CAAC,UAAU;wBAC3B,KAAK,EAAE,IAAI,CAAC,KAAK;wBACjB,KAAK,EAAE,iBAAiB;qBACzB,CAAC;oBACF,IAAI,IAAI,CAAC,gBAAgB,EAAE,CAAC;wBAC1B,OAAO,CAAC,gBAAgB,GAAG,IAAI,CAAC,gBAAgB,CAAC;oBACnD,CAAC;gBACH,CAAC;gBACD,IAAI,IAAI,CAAC,gBAAgB,EAAE,CAAC;oBAC1B,OAAO,CAAC,oBAAoB,GAAG,IAAI,CAAC,gBAAgB,CAAC;gBACvD,CAAC;gBACD,IAAI,aAAa,CAAC,GAAG,CAAC,IAAI,CAAC,UAAU,CAAC,EAAE,CAAC;oBACvC,MAAM,QAAQ,GAAG,aAAa,CAAC,GAAG,CAAC,IAAI,CAAC,UAAU,CAAE,CAAC;oBACrD,MAAM,CAAC,MAAM,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;gBACnC,CAAC;qBAAM,CAAC;oBACN,aAAa,CAAC,GAAG,CAAC,IAAI,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;oBAC5C,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;gBAC9B,CAAC;gBACD,MAAM;YACR,CAAC;YACD,KAAK,aAAa,CAAC,CAAC,CAAC;gBACnB,MAAM,QAAQ,GAAG,aAAa,CAAC,GAAG,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;gBACpD,MAAM,CACJ,QAAQ,EACR,mCAAmC,IAAI,CAAC,UAAU,wBAAwB,CAC3E,CAAC;gBACF,IAAI,OAAuC,CAAC;gBAC5C,IAAI,QAAQ,CAAC,IAAI,KAAK,cAAc,EAAE,CAAC;oBACrC,OAAO,GAAG;wBACR,GAAG,QAAQ;wBACX,KAAK,EAAE,kBAAkB;wBACzB,KAAK,EAAE,IAAI,CAAC,KAAK,IAAI,QAAQ,CAAC,KAAK;wBACnC,MAAM,EAAE,IAAI,CAAC,MAAM,IAAI,QAAQ,CAAC,MAAM;wBACtC,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC,aAAa,CAAC,CAAC;qBACV,CAAC;gBACzB,CAAC;qBAAM,CAAC;oBACN,OAAO,GAAG;wBACR,GAAG,QAAQ;wBACX,KAAK,EAAE,kBAAkB;wBACzB,KAAK,EAAE,IAAI,CAAC,KAAK,IAAI,QAAQ,CAAC,KAAK;wBACnC,MAAM,EAAE,IAAI,CAAC,MAAM,IAAI,QAAQ,CAAC,MAAM;wBACtC,WAAW,EAAE,IAAI,CAAC,WAAW;qBAChB,CAAC;gBAClB,CAAC;gBACD,MAAM,CAAC,MAAM,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;gBACjC,MAAM;YACR,CAAC;YACD,KAAK,iBAAiB,CAAC;YACvB,KAAK,iBAAiB,CAAC,CAAC,CAAC;gBACvB,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC;oBACrC,MAAM,QAAQ,GAAG,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;oBACtC,IAAI,QAAQ,EAAE,IAAI,KAAK,WAAW,EAAE,CAAC;wBACnC,kBAAkB,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,QAAQ,CAAC,CAAC;oBAC5C,CAAC;yBAAM,CAAC;wBACN,MAAM,OAAO,GAAG;4BACd,IAAI,EAAE,WAAW;4BACjB,KAAK,EAAE,WAAW;4BAClB,IAAI,EAAE,EAAE;4BACR,gBAAgB,EAAE,IAAI,CAAC,gBAAgB;yBACd,CAAC;wBAC5B,kBAAkB,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;wBACzC,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;oBAC9B,CAAC;gBACH,CAAC;gBACD,MAAM,aAAa,GAAG,kBAAkB,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAE,CAAC;gBACvD,IAAI,IAAI,CAAC,IAAI,KAAK,iBAAiB,EAAE,CAAC;oBACpC,aAAa,CAAC,IAAI,IAAI,IAAI,CAAC,IAAI,CAAC;oBAChC,aAAa,CAAC,gBAAgB,GAAG,qBAAqB,CACpD,aAAa,CAAC,gBAAgB,EAC9B,IAAI,CAAC,gBAAgB,CACtB,CAAC;gBACJ,CAAC;gBACD,MAAM;YACR,CAAC;YACD,KAAK,eAAe,CAAC,CAAC,CAAC;gBACrB,MAAM,aAAa,GACjB,kBAAkB,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;oBAC/B,OAAO,CAAC,KAAK,CAAC,IAAI,CAChB,CAAC,CAAC,EAAwB,EAAE,CAC1B,CAAC,CAAC,IAAI,KAAK,WAAW,IAAI,CAAC,CAAC,KAAK,KAAK,WAAW,CACnD,CAAC;gBACL,IAAI,aAAa,EAAE,CAAC;oBAClB,aAAa,CAAC,KAAK,GAAG,MAAM,CAAC;gBAC/B,CAAC;qBAAM,CAAC;oBACN,OAAO,CAAC,IAAI,CACV,mCAAmC,IAAI,CAAC,EAAE,4BAA4B,CACvE,CAAC;gBACJ,CAAC;gBACD,MAAM;YACR,CAAC;YACD,KAAK,QAAQ;gBACX,IAAI,IAAI,CAAC,UAAU,KAAK,KAAK,EAAE,CAAC;oBAC9B,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC;wBACjB,IAAI,EAAE,YAAY;wBAClB,GAAG,EAAE,IAAI,CAAC,GAAG;wBACb,QAAQ,EAAE,IAAI,CAAC,EAAE;wBACjB,gBAAgB,EAAE,IAAI,CAAC,gBAAgB;wBACvC,KAAK,EAAE,IAAI,CAAC,KAAK;qBAClB,CAAC,CAAC;gBACL,CAAC;qBAAM,IAAI,IAAI,CAAC,UAAU,KAAK,UAAU,EAAE,CAAC;oBAC1C,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC;wBACjB,IAAI,EAAE,iBAAiB;wBACvB,SAAS,EAAE,IAAI,CAAC,SAAS;wBACzB,QAAQ,EAAE,IAAI,CAAC,EAAE;wBACjB,KAAK,EAAE,IAAI,CAAC,KAAK;wBACjB,QAAQ,EAAE,IAAI,CAAC,QAAQ;wBACvB,gBAAgB,EAAE,IAAI,CAAC,gBAAgB;qBACxC,CAAC,CAAC;gBACL,CAAC;qBAAM,CAAC;oBACN,OAAO,CAAC,IAAI,CAAC,0CAA0C,EAAE,IAAI,CAAC,CAAC;gBACjE,CAAC;gBACD,MAAM;YACR,KAAK,OAAO;gBACV,OAAO,CAAC,MAAM,GAAG,QAAQ,CAAC;gBAC1B,MAAM;YACR,KAAK,OAAO;gBACV,OAAO,CAAC,MAAM,GAAG,QAAQ,CAAC;gBAC1B,OAAO,CAAC,IAAI,CAAC,8BAA8B,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC;gBACzD,MAAM;YACR,KAAK,YAAY,CAAC,CAAC,CAAC;gBAClB,MAAM,QAAQ,GAAG,aAAa,CAAC,GAAG,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;gBACpD,IAAI,QAAQ,EAAE,CAAC;oBACb,QAAQ,CAAC,SAAS,GAAG,eAAe,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;gBACnD,CAAC;gBACD,MAAM;YACR,CAAC;YACD,KAAK,uBAAuB,CAAC,CAAC,CAAC;gBAC7B,MAAM,SAAS,GAAG,IAIjB,CAAC;gBACF,MAAM,QAAQ,GAAG,aAAa,CAAC,GAAG,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;gBACzD,IAAI,QAAQ,EAAE,CAAC;oBACb,QAAQ,CAAC,KAAK,GAAG,oBAAoB,CAAC;oBACrC,QAA+C,CAAC,QAAQ,GAAG;wBAC1D,EAAE,EAAE,SAAS,CAAC,UAAU;qBACzB,CAAC;gBACJ,CAAC;qBAAM,CAAC;oBACN,OAAO,CAAC,IAAI,CACV,2BAA2B,SAAS,CAAC,UAAU,uBAAuB,CACvE,CAAC;gBACJ,CAAC;gBACD,MAAM;YACR,CAAC;YACD,KAAK,MAAM,CAAC;YACZ,KAAK,UAAU,CAAC;YAChB,KAAK,aAAa,CAAC;YACnB,KAAK,QAAQ,CAAC;YACd,KAAK,KAAK,CAAC;YACX,KAAK,YAAY,CAAC;YAClB,KAAK,OAAO;gBACV,SAAS;gBACT,MAAM;YACR,OAAO,CAAC,CAAC,CAAC;gBACR,0EAA0E;gBAC1E,sEAAsE;gBACtE,8EAA8E;gBAC9E,yBAAyB;gBACzB,OAAO,CAAC,IAAI,CAAC,6BAA6B,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;gBAClE,MAAM;YACR,CAAC;QACH,CAAC;IACH,CAAC;IACD,sDAAsD;IACtD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,OAAO,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QAClD,MAAM,IAAI,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAC9B,IAAI,IAAI,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;YAC9B,IAAI,CAAC,KAAK,GAAG,MAAM,CAAC;QACtB,CAAC;IACH,CAAC;IACD,OAAO,CAAC,IAAI,GAAG,QAAQ,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;IACvC,OAAO;QACL;YACE,QAAQ,EAAE,aAAa,CAAC,QAAQ;YAChC,MAAM;YACN,OAAO;SACR;QACD,IAAI;KACL,CAAC;AACJ,CAAC;AAED,SAAS,qBAAqB,CAC5B,QAAsC,EACtC,IAAkC;IAElC,IAAI,CAAC,QAAQ,IAAI,CAAC,IAAI,EAAE,CAAC;QACvB,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,IAAI,CAAC,QAAQ,EAAE,CAAC;QACd,OAAO,IAAI,CAAC;IACd,CAAC;IACD,IAAI,CAAC,IAAI,EAAE,CAAC;QACV,OAAO,QAAQ,CAAC;IAClB,CAAC;IACD,MAAM,MAAM,GAAqB,QAAQ,CAAC;IAC1C,KAAK,MAAM,CAAC,QAAQ,EAAE,QAAQ,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC;QACxD,MAAM,CAAC,QAAQ,CAAC,GAAG;YACjB,GAAG,MAAM,CAAC,QAAQ,CAAC;YACnB,GAAG,QAAQ;SACZ,CAAC;IACJ,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC"}

package/dist/mapping.d.ts

@@ -1,7 +1,8 @@
-import { type UIMessage as AIMessage, type AssistantContent, type ModelMessage, type DataContent, type FilePart, type GenerateObjectResult, type ImagePart, type StepResult, type ToolContent, type ToolSet, type UserContent, type FileUIPart, type LanguageModelUsage, type CallWarning } from "ai";
-import type { Message, MessageWithMetadata, Usage } from "./validators.js";
+import { type UIMessage as AIMessage, type AssistantContent, type ModelMessage, type DataContent, type FilePart, type GenerateObjectResult, type ImagePart, type StepResult, type ToolContent, type ToolSet, type UserContent, type FileUIPart, type LanguageModelUsage, type CallWarning, type ToolResultPart, type JSONValue } from "ai";
+import { type Message, type MessageWithMetadata, type Usage, type MessageDoc } from "./validators.js";
 import type { ActionCtx, AgentComponent } from "./client/types.js";
-import type { RunMutationCtx } from "./client/types.js";
+import type { MutationCtx } from "./client/types.js";
+import { type ModelOrMetadata } from "./shared.js";
 export type AIMessageWithoutId = Omit<AIMessage, "id">;
 export type SerializeUrlsAndUint8Arrays<T> = T extends URL ? string : T extends Uint8Array | ArrayBufferLike ? ArrayBuffer : T extends Array<infer Inner> ? Array<SerializeUrlsAndUint8Arrays<Inner>> : T extends Record<string, any> ? {
     [K in keyof T]: SerializeUrlsAndUint8Arrays<T[K]>;
@@ -9,32 +10,31 @@ export type SerializeUrlsAndUint8Arrays<T> = T extends URL ? string : T extends
 export type Content = UserContent | AssistantContent | ToolContent;
 export type SerializedContent = Message["content"];
 export type SerializedMessage = Message;
-export declare function serializeMessage(ctx: ActionCtx | RunMutationCtx, component: AgentComponent, messageWithId: (ModelMessage & {
-    id?: string;
-}) | Message): Promise<{
+export declare function serializeMessage(ctx: ActionCtx | MutationCtx, component: AgentComponent, message: ModelMessage | Message): Promise<{
     message: SerializedMessage;
     fileIds?: string[];
 }>;
-export declare function deserializeMessage(message: SerializedMessage): ModelMessage;
+export declare function fromModelMessage(message: ModelMessage): Message;
+export declare function serializeOrThrow(message: ModelMessage | Message): Promise<SerializedMessage>;
+export declare function toModelMessage(message: SerializedMessage | ModelMessage): ModelMessage;
+export declare function docsToModelMessages(messages: MessageDoc[]): ModelMessage[];
 export declare function serializeUsage(usage: LanguageModelUsage): Usage;
-export declare function deserializeUsage(usage: Usage): LanguageModelUsage;
+export declare function toModelMessageUsage(usage: Usage): LanguageModelUsage;
 export declare function serializeWarnings(warnings: CallWarning[] | undefined): MessageWithMetadata["warnings"];
-export declare function deserializeWarnings(warnings: MessageWithMetadata["warnings"]): CallWarning[] | undefined;
-export declare function serializeNewMessagesInStep<TOOLS extends ToolSet>(ctx: ActionCtx, component: AgentComponent, step: StepResult<TOOLS>, metadata: {
-    model: string;
-    provider: string;
-}): Promise<MessageWithMetadata[]>;
-export declare function serializeObjectResult(ctx: ActionCtx, component: AgentComponent, result: GenerateObjectResult<unknown>, metadata: {
-    model: string;
-    provider: string;
-}): Promise<{
+export declare function toModelMessageWarnings(warnings: MessageWithMetadata["warnings"]): CallWarning[] | undefined;
+export declare function serializeNewMessagesInStep<TOOLS extends ToolSet>(ctx: ActionCtx, component: AgentComponent, step: StepResult<TOOLS>, model: ModelOrMetadata | undefined): Promise<{
     messages: MessageWithMetadata[];
 }>;
-export declare function serializeContent(ctx: ActionCtx | RunMutationCtx, component: AgentComponent, content: Content | Message["content"]): Promise<{
+export declare function serializeObjectResult(ctx: ActionCtx, component: AgentComponent, result: GenerateObjectResult<unknown>, model: ModelOrMetadata | undefined): Promise<{
+    messages: MessageWithMetadata[];
+}>;
+export declare function serializeContent(ctx: ActionCtx | MutationCtx, component: AgentComponent, content: Content | Message["content"]): Promise<{
     content: SerializedContent;
     fileIds?: string[];
 }>;
-export declare function deserializeContent(content: SerializedContent): Content;
+export declare function fromModelMessageContent(content: Content): Message["content"];
+export declare function toModelMessageContent(content: SerializedContent | ModelMessage["content"]): Content;
+export declare function normalizeToolOutput(result: string | JSONValue | undefined): ToolResultPart["output"];
 /**
  * Return a best-guess MIME type based on the magic-number signature
  * found at the start of an ArrayBuffer.
@@ -49,6 +49,6 @@ export declare function guessMimeType(buf: ArrayBuffer | string): string;
  * @returns The serialized data as an ArrayBuffer or the URL as a string.
  */
 export declare function serializeDataOrUrl(dataOrUrl: DataContent | URL): ArrayBuffer | string;
-export declare function deserializeUrl(urlOrString: string | ArrayBuffer): URL | DataContent;
+export declare function toModelMessageDataOrUrl(urlOrString: string | ArrayBuffer | URL | DataContent): URL | DataContent;
 export declare function toUIFilePart(part: ImagePart | FilePart): FileUIPart;
 //# sourceMappingURL=mapping.d.ts.map

package/dist/mapping.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"mapping.d.ts","sourceRoot":"","sources":["../src/mapping.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,KAAK,SAAS,IAAI,SAAS,EAC3B,KAAK,gBAAgB,EACrB,KAAK,YAAY,EACjB,KAAK,WAAW,EAChB,KAAK,QAAQ,EACb,KAAK,oBAAoB,EACzB,KAAK,SAAS,EACd,KAAK,UAAU,EACf,KAAK,WAAW,EAChB,KAAK,OAAO,EACZ,KAAK,WAAW,EAChB,KAAK,UAAU,EACf,KAAK,kBAAkB,EACvB,KAAK,WAAW,EAKjB,MAAM,IAAI,CAAC;AACZ,OAAO,KAAK,EACV,OAAO,EACP,mBAAmB,EACnB,KAAK,EASN,MAAM,iBAAiB,CAAC;AACzB,OAAO,KAAK,EAAE,SAAS,EAAE,cAAc,EAAE,MAAM,mBAAmB,CAAC;AACnE,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,mBAAmB,CAAC;AAKxD,MAAM,MAAM,kBAAkB,GAAG,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;AAEvD,MAAM,MAAM,2BAA2B,CAAC,CAAC,IAAI,CAAC,SAAS,GAAG,GACtD,MAAM,GACN,CAAC,SAAS,UAAU,GAAG,eAAe,GACpC,WAAW,GACX,CAAC,SAAS,KAAK,CAAC,MAAM,KAAK,CAAC,GAC1B,KAAK,CAAC,2BAA2B,CAAC,KAAK,CAAC,CAAC,GAEzC,CAAC,SAAS,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,GAC3B;KAAG,CAAC,IAAI,MAAM,CAAC,GAAG,2BAA2B,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;CAAE,GACrD,CAAC,CAAC;AAEZ,MAAM,MAAM,OAAO,GAAG,WAAW,GAAG,gBAAgB,GAAG,WAAW,CAAC;AACnE,MAAM,MAAM,iBAAiB,GAAG,OAAO,CAAC,SAAS,CAAC,CAAC;AAEnD,MAAM,MAAM,iBAAiB,GAAG,OAAO,CAAC;AAExC,wBAAsB,gBAAgB,CACpC,GAAG,EAAE,SAAS,GAAG,cAAc,EAC/B,SAAS,EAAE,cAAc,EACzB,aAAa,EAAE,CAAC,YAAY,GAAG;IAAE,EAAE,CAAC,EAAE,MAAM,CAAA;CAAE,CAAC,GAAG,OAAO,GACxD,OAAO,CAAC;IAAE,OAAO,EAAE,iBAAiB,CAAC;IAAC,OAAO,CAAC,EAAE,MAAM,EAAE,CAAA;CAAE,CAAC,CAe7D;AAED,wBAAgB,kBAAkB,CAAC,OAAO,EAAE,iBAAiB,GAAG,YAAY,CAK3E;AAED,wBAAgB,cAAc,CAAC,KAAK,EAAE,kBAAkB,GAAG,KAAK,CAO/D;AAED,wBAAgB,gBAAgB,CAAC,KAAK,EAAE,KAAK,GAAG,kBAAkB,CAQjE;AAED,wBAAgB,iBAAiB,CAC/B,QAAQ,EAAE,WAAW,EAAE,GAAG,SAAS,GAClC,mBAAmB,CAAC,UAAU,CAAC,CAajC;AAED,wBAAgB,mBAAmB,CACjC,QAAQ,EAAE,mBAAmB,CAAC,UAAU,CAAC,GACxC,WAAW,EAAE,GAAG,SAAS,CAG3B;AAED,wBAAsB,0BAA0B,CAAC,KAAK,SAAS,OAAO,EACpE,GAAG,EAAE,SAAS,EACd,SAAS,EAAE,cAAc,EACzB,IAAI,EAAE,UAAU,CAAC,KAAK,CAAC,EACvB,QAAQ,EAAE;IAAE,KAAK,EAAE,MAAM,CAAC;IAAC,QAAQ,EAAE,MAAM,CAAA;CAAE,GAC5C,OAAO,CAAC,mBAAmB,EAAE,CAAC,CAwChC;AAED,wBAAsB,qBAAqB,CACzC,GAAG,EAAE,SAAS,EACd,SAAS,EAAE,cAAc,EACzB,MAAM,EAAE,oBAAoB,CAAC,OAAO,CAAC,EACrC,QAAQ,EAAE;IAAE,KAAK,EAAE,MAAM,CAAC;IAAC,QAAQ,EAAE,MAAM,CAAA;CAAE,GAC5C,OAAO,CAAC;IAAE,QAAQ,EAAE,mBAAmB,EAAE,CAAA;CAAE,CAAC,CAuB9C;AAED,wBAAsB,gBAAgB,CACpC,GAAG,EAAE,SAAS,GAAG,cAAc,EAC/B,SAAS,EAAE,cAAc,EACzB,OAAO,EAAE,OAAO,GAAG,OAAO,CAAC,SAAS,CAAC,GACpC,OAAO,CAAC;IAAE,OAAO,EAAE,iBAAiB,CAAC;IAAC,OAAO,CAAC,EAAE,MAAM,EAAE,CAAA;CAAE,CAAC,CA8G7D;AAED,wBAAgB,kBAAkB,CAAC,OAAO,EAAE,iBAAiB,GAAG,OAAO,CA8DtE;AAED;;;;;;GAMG;AACH,wBAAgB,aAAa,CAAC,GAAG,EAAE,WAAW,GAAG,MAAM,GAAG,MAAM,CA0D/D;AAED;;;;GAIG;AACH,wBAAgB,kBAAkB,CAChC,SAAS,EAAE,WAAW,GAAG,GAAG,GAC3B,WAAW,GAAG,MAAM,CActB;AAED,wBAAgB,cAAc,CAC5B,WAAW,EAAE,MAAM,GAAG,WAAW,GAChC,GAAG,GAAG,WAAW,CAWnB;AAED,wBAAgB,YAAY,CAAC,IAAI,EAAE,SAAS,GAAG,QAAQ,GAAG,UAAU,CAcnE"}
+{"version":3,"file":"mapping.d.ts","sourceRoot":"","sources":["../src/mapping.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,KAAK,SAAS,IAAI,SAAS,EAC3B,KAAK,gBAAgB,EACrB,KAAK,YAAY,EACjB,KAAK,WAAW,EAChB,KAAK,QAAQ,EACb,KAAK,oBAAoB,EACzB,KAAK,SAAS,EACd,KAAK,UAAU,EACf,KAAK,WAAW,EAChB,KAAK,OAAO,EACZ,KAAK,WAAW,EAChB,KAAK,UAAU,EACf,KAAK,kBAAkB,EACvB,KAAK,WAAW,EAGhB,KAAK,cAAc,EAEnB,KAAK,SAAS,EACf,MAAM,IAAI,CAAC;AACZ,OAAO,EAGL,KAAK,OAAO,EACZ,KAAK,mBAAmB,EACxB,KAAK,KAAK,EAUV,KAAK,UAAU,EAGhB,MAAM,iBAAiB,CAAC;AACzB,OAAO,KAAK,EAAE,SAAS,EAAE,cAAc,EAAE,MAAM,mBAAmB,CAAC;AACnE,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AASrD,OAAO,EAGL,KAAK,eAAe,EACrB,MAAM,aAAa,CAAC;AACrB,MAAM,MAAM,kBAAkB,GAAG,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;AAEvD,MAAM,MAAM,2BAA2B,CAAC,CAAC,IAAI,CAAC,SAAS,GAAG,GACtD,MAAM,GACN,CAAC,SAAS,UAAU,GAAG,eAAe,GACpC,WAAW,GACX,CAAC,SAAS,KAAK,CAAC,MAAM,KAAK,CAAC,GAC1B,KAAK,CAAC,2BAA2B,CAAC,KAAK,CAAC,CAAC,GACzC,CAAC,SAAS,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,GAC3B;KAAG,CAAC,IAAI,MAAM,CAAC,GAAG,2BAA2B,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;CAAE,GACrD,CAAC,CAAC;AAEZ,MAAM,MAAM,OAAO,GAAG,WAAW,GAAG,gBAAgB,GAAG,WAAW,CAAC;AACnE,MAAM,MAAM,iBAAiB,GAAG,OAAO,CAAC,SAAS,CAAC,CAAC;AAEnD,MAAM,MAAM,iBAAiB,GAAG,OAAO,CAAC;AAExC,wBAAsB,gBAAgB,CACpC,GAAG,EAAE,SAAS,GAAG,WAAW,EAC5B,SAAS,EAAE,cAAc,EACzB,OAAO,EAAE,YAAY,GAAG,OAAO,GAC9B,OAAO,CAAC;IAAE,OAAO,EAAE,iBAAiB,CAAC;IAAC,OAAO,CAAC,EAAE,MAAM,EAAE,CAAA;CAAE,CAAC,CAgB7D;AAID,wBAAgB,gBAAgB,CAAC,OAAO,EAAE,YAAY,GAAG,OAAO,CAS/D;AAED,wBAAsB,gBAAgB,CACpC,OAAO,EAAE,YAAY,GAAG,OAAO,GAC9B,OAAO,CAAC,iBAAiB,CAAC,CAa5B;AAED,wBAAgB,cAAc,CAC5B,OAAO,EAAE,iBAAiB,GAAG,YAAY,GACxC,YAAY,CAKd;AAED,wBAAgB,mBAAmB,CAAC,QAAQ,EAAE,UAAU,EAAE,GAAG,YAAY,EAAE,CAM1E;AAED,wBAAgB,cAAc,CAAC,KAAK,EAAE,kBAAkB,GAAG,KAAK,CAQ/D;AAED,wBAAgB,mBAAmB,CAAC,KAAK,EAAE,KAAK,GAAG,kBAAkB,CAmBpE;AAED,wBAAgB,iBAAiB,CAC/B,QAAQ,EAAE,WAAW,EAAE,GAAG,SAAS,GAClC,mBAAmB,CAAC,UAAU,CAAC,CAcjC;AAED,wBAAgB,sBAAsB,CACpC,QAAQ,EAAE,mBAAmB,CAAC,UAAU,CAAC,GACxC,WAAW,EAAE,GAAG,SAAS,CAc3B;AAED,wBAAsB,0BAA0B,CAAC,KAAK,SAAS,OAAO,EACpE,GAAG,EAAE,SAAS,EACd,SAAS,EAAE,cAAc,EACzB,IAAI,EAAE,UAAU,CAAC,KAAK,CAAC,EACvB,KAAK,EAAE,eAAe,GAAG,SAAS,GACjC,OAAO,CAAC;IAAE,QAAQ,EAAE,mBAAmB,EAAE,CAAA;CAAE,CAAC,CAmC9C;AAED,wBAAsB,qBAAqB,CACzC,GAAG,EAAE,SAAS,EACd,SAAS,EAAE,cAAc,EACzB,MAAM,EAAE,oBAAoB,CAAC,OAAO,CAAC,EACrC,KAAK,EAAE,eAAe,GAAG,SAAS,GACjC,OAAO,CAAC;IAAE,QAAQ,EAAE,mBAAmB,EAAE,CAAA;CAAE,CAAC,CAsB9C;AAYD,wBAAsB,gBAAgB,CACpC,GAAG,EAAE,SAAS,GAAG,WAAW,EAC5B,SAAS,EAAE,cAAc,EACzB,OAAO,EAAE,OAAO,GAAG,OAAO,CAAC,SAAS,CAAC,GACpC,OAAO,CAAC;IAAE,OAAO,EAAE,iBAAiB,CAAC;IAAC,OAAO,CAAC,EAAE,MAAM,EAAE,CAAA;CAAE,CAAC,CAiI7D;AAED,wBAAgB,uBAAuB,CAAC,OAAO,EAAE,OAAO,GAAG,OAAO,CAAC,SAAS,CAAC,CA4E5E;AAED,wBAAgB,qBAAqB,CACnC,OAAO,EAAE,iBAAiB,GAAG,YAAY,CAAC,SAAS,CAAC,GACnD,OAAO,CAqGT;AAED,wBAAgB,mBAAmB,CACjC,MAAM,EAAE,MAAM,GAAG,SAAS,GAAG,SAAS,GACrC,cAAc,CAAC,QAAQ,CAAC,CAc1B;AAwBD;;;;;;GAMG;AACH,wBAAgB,aAAa,CAAC,GAAG,EAAE,WAAW,GAAG,MAAM,GAAG,MAAM,CA0D/D;AAED;;;;GAIG;AACH,wBAAgB,kBAAkB,CAChC,SAAS,EAAE,WAAW,GAAG,GAAG,GAC3B,WAAW,GAAG,MAAM,CActB;AAED,wBAAgB,uBAAuB,CACrC,WAAW,EAAE,MAAM,GAAG,WAAW,GAAG,GAAG,GAAG,WAAW,GACpD,GAAG,GAAG,WAAW,CAcnB;AAED,wBAAgB,YAAY,CAAC,IAAI,EAAE,SAAS,GAAG,QAAQ,GAAG,UAAU,CAcnE"}