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

package/dist/mapping.js

@@ -1,677 +0,0 @@
-import {} from "ai";
-import { vMessageWithMetadata, vToolResultOutput, vToolApprovalRequest, vToolApprovalResponse, } from "./validators.js";
-import { MAX_FILE_SIZE, storeFile } from "./client/files.js";
-import { convertUint8ArrayToBase64, } from "@ai-sdk/provider-utils";
-import { parse, validate } from "convex-helpers/validators";
-import { getModelName, getProviderName, } from "./shared.js";
-export async function serializeMessage(ctx, component, message) {
-    const { content, fileIds } = await serializeContent(ctx, component, message.content);
-    return {
-        message: {
-            role: message.role,
-            content,
-            ...(message.providerOptions
-                ? { providerOptions: message.providerOptions }
-                : {}),
-        },
-        fileIds,
-    };
-}
-// Similar to serializeMessage, but doesn't save any files and is looser
-// For use on the frontend / in synchronous environments.
-export function fromModelMessage(message) {
-    const content = fromModelMessageContent(message.content);
-    return {
-        role: message.role,
-        content,
-        ...(message.providerOptions
-            ? { providerOptions: message.providerOptions }
-            : {}),
-    };
-}
-export async function serializeOrThrow(message) {
-    const { content } = await serializeContent({}, {}, message.content);
-    return {
-        role: message.role,
-        content,
-        ...(message.providerOptions
-            ? { providerOptions: message.providerOptions }
-            : {}),
-    };
-}
-export function toModelMessage(message) {
-    return {
-        ...message,
-        content: toModelMessageContent(message.content),
-    };
-}
-export function docsToModelMessages(messages) {
-    return messages
-        .map((m) => m.message)
-        .filter((m) => !!m)
-        .filter((m) => !!m.content.length)
-        .map(toModelMessage);
-}
-/**
- * Merge consecutive tool messages that contain `tool-approval-response` parts
- * into a single tool message. The AI SDK's `collectToolApprovals` only examines
- * the last tool message, so when multiple approvals are saved as separate
- * messages (e.g. approve tool A, then deny tool B), they must be combined
- * for the SDK to process them all.
- */
-export function mergeApprovalResponseMessages(messages) {
-    const result = [];
-    for (const msg of messages) {
-        const prev = result.at(-1);
-        if (msg.role === "tool" &&
-            prev?.role === "tool" &&
-            Array.isArray(msg.content) &&
-            Array.isArray(prev.content) &&
-            hasApprovalResponse(msg.content) &&
-            hasApprovalResponse(prev.content)) {
-            // Clone before merging to avoid mutating the original message's content array
-            const cloned = { ...prev, content: [...prev.content] };
-            result[result.length - 1] = cloned;
-            cloned.content.push(...msg.content);
-        }
-        else {
-            result.push(msg);
-        }
-    }
-    return result;
-}
-function hasApprovalResponse(content) {
-    return content.some((p) => p.type === "tool-approval-response");
-}
-export function serializeUsage(usage) {
-    return {
-        promptTokens: usage.inputTokens ?? 0,
-        completionTokens: usage.outputTokens ?? 0,
-        totalTokens: usage.totalTokens ?? 0,
-        reasoningTokens: usage.reasoningTokens,
-        cachedInputTokens: usage.cachedInputTokens,
-    };
-}
-export function toModelMessageUsage(usage) {
-    return {
-        inputTokens: usage.promptTokens,
-        outputTokens: usage.completionTokens,
-        totalTokens: usage.totalTokens,
-        reasoningTokens: usage.reasoningTokens,
-        cachedInputTokens: usage.cachedInputTokens,
-        // These detail fields are required by LanguageModelUsage type but we don't
-        // have the granular data, so we provide empty objects with undefined values.
-        inputTokenDetails: {
-            cacheReadTokens: undefined,
-            cacheWriteTokens: undefined,
-            noCacheTokens: undefined,
-        },
-        outputTokenDetails: {
-            textTokens: undefined,
-            reasoningTokens: undefined,
-        },
-    };
-}
-export function serializeWarnings(warnings) {
-    if (!warnings) {
-        return undefined;
-    }
-    return warnings.map((warning) => {
-        if (warning.type === "compatibility") {
-            return {
-                type: "unsupported-setting",
-                setting: warning.feature,
-                details: warning.details,
-            };
-        }
-        return warning;
-    });
-}
-export function toModelMessageWarnings(warnings) {
-    if (!warnings) {
-        return undefined;
-    }
-    return warnings.map((warning) => {
-        if (warning.type === "unsupported-setting") {
-            return {
-                type: "compatibility",
-                feature: warning.setting,
-                details: warning.details,
-            };
-        }
-        return warning;
-    });
-}
-export async function serializeNewMessagesInStep(ctx, component, step, model, 
-/**
- * If provided, these are the new response messages for this step
- * (pre-sliced by the caller). When not provided, falls back to the
- * existing heuristic of slicing the last 1-2 messages.
- *
- * This is needed for tool approval flows where the SDK adds extra
- * messages (e.g. approval tool-results) at the beginning of
- * responseMessages that the old slice(-1/-2) logic would miss.
- */
-newResponseMessages) {
-    // If there are tool results, there's another message with the tool results
-    // ref: https://github.com/vercel/ai/blob/main/packages/ai/src/generate-text/to-response-messages.ts#L120
-    const hasToolMessage = step.response.messages.at(-1)?.role === "tool";
-    const assistantFields = {
-        model: model ? getModelName(model) : undefined,
-        provider: model ? getProviderName(model) : undefined,
-        providerMetadata: step.providerMetadata,
-        reasoning: step.reasoningText,
-        reasoningDetails: step.reasoning,
-        usage: serializeUsage(step.usage),
-        warnings: serializeWarnings(step.warnings),
-        finishReason: step.finishReason,
-        // Only store the sources on one message
-        sources: hasToolMessage ? undefined : step.sources,
-    };
-    const toolFields = { sources: step.sources };
-    // Determine which messages to serialize for this step
-    let messagesToSerialize;
-    if (newResponseMessages) {
-        messagesToSerialize = newResponseMessages;
-    }
-    else if (hasToolMessage) {
-        messagesToSerialize = step.response.messages.slice(-2);
-    }
-    else if (step.content.length) {
-        messagesToSerialize = step.response.messages.slice(-1);
-    }
-    else {
-        messagesToSerialize = [{ role: "assistant", content: [] }];
-    }
-    const messages = await Promise.all(messagesToSerialize.map(async (msg) => {
-        const { message, fileIds } = await serializeMessage(ctx, component, msg);
-        return parse(vMessageWithMetadata, {
-            message,
-            ...(message.role === "tool" ? toolFields : assistantFields),
-            text: step.text,
-            fileIds,
-        });
-    }));
-    // TODO: capture step.files separately?
-    return { messages };
-}
-export async function serializeObjectResult(ctx, component, result, model) {
-    const text = JSON.stringify(result.object);
-    const { message, fileIds } = await serializeMessage(ctx, component, {
-        role: "assistant",
-        content: text,
-    });
-    return {
-        messages: [
-            {
-                message,
-                model: model ? getModelName(model) : undefined,
-                provider: model ? getProviderName(model) : undefined,
-                providerMetadata: result.providerMetadata,
-                finishReason: result.finishReason,
-                text,
-                usage: serializeUsage(result.usage),
-                warnings: serializeWarnings(result.warnings),
-                fileIds,
-            },
-        ],
-    };
-}
-function getMimeOrMediaType(part) {
-    if ("mediaType" in part) {
-        return part.mediaType;
-    }
-    if ("mimeType" in part) {
-        return part.mimeType;
-    }
-    return undefined;
-}
-export async function serializeContent(ctx, component, content) {
-    if (typeof content === "string") {
-        return { content };
-    }
-    const fileIds = [];
-    const serialized = await Promise.all(content.map(async (part) => {
-        const metadata = {};
-        if ("providerOptions" in part) {
-            metadata.providerOptions = part.providerOptions;
-        }
-        if ("providerMetadata" in part) {
-            metadata.providerMetadata = part.providerMetadata;
-        }
-        switch (part.type) {
-            case "text": {
-                return {
-                    type: part.type,
-                    text: part.text,
-                    ...metadata,
-                };
-            }
-            case "image": {
-                let image = serializeDataOrUrl(part.image);
-                if (image instanceof ArrayBuffer &&
-                    image.byteLength > MAX_FILE_SIZE) {
-                    const { file } = await storeFile(ctx, component, new Blob([image], {
-                        type: getMimeOrMediaType(part) || guessMimeType(image),
-                    }));
-                    image = file.url;
-                    fileIds.push(file.fileId);
-                }
-                return {
-                    type: part.type,
-                    mediaType: getMimeOrMediaType(part),
-                    ...metadata,
-                    image,
-                };
-            }
-            case "file": {
-                let data = serializeDataOrUrl(part.data);
-                if (data instanceof ArrayBuffer && data.byteLength > MAX_FILE_SIZE) {
-                    const { file } = await storeFile(ctx, component, new Blob([data], { type: getMimeOrMediaType(part) }));
-                    data = file.url;
-                    fileIds.push(file.fileId);
-                }
-                return {
-                    type: part.type,
-                    data,
-                    filename: part.filename,
-                    mediaType: getMimeOrMediaType(part),
-                    ...metadata,
-                };
-            }
-            case "tool-call": {
-                // Handle legacy data where only args field exists
-                const input = part.input ?? part?.args ?? {};
-                return {
-                    type: part.type,
-                    input,
-                    /** @deprecated Use `input` instead. */
-                    args: input,
-                    toolCallId: part.toolCallId,
-                    toolName: part.toolName,
-                    providerExecuted: part.providerExecuted,
-                    ...metadata,
-                };
-            }
-            case "tool-result": {
-                return normalizeToolResult(part, metadata);
-            }
-            case "reasoning": {
-                return {
-                    type: part.type,
-                    text: part.text,
-                    ...metadata,
-                };
-            }
-            // Not in current generation output, but could be in historical messages
-            case "redacted-reasoning": {
-                return {
-                    type: part.type,
-                    data: part.data,
-                    ...metadata,
-                };
-            }
-            case "source": {
-                return part;
-            }
-            case "tool-approval-request": {
-                return {
-                    type: part.type,
-                    approvalId: part.approvalId,
-                    toolCallId: part.toolCallId,
-                    ...metadata,
-                };
-            }
-            case "tool-approval-response": {
-                return {
-                    type: part.type,
-                    approvalId: part.approvalId,
-                    approved: part.approved,
-                    reason: part.reason,
-                    providerExecuted: part.providerExecuted,
-                    ...metadata,
-                };
-            }
-            default:
-                return null;
-        }
-    }));
-    return {
-        content: serialized.filter((p) => p !== null),
-        fileIds: fileIds.length > 0 ? fileIds : undefined,
-    };
-}
-export function fromModelMessageContent(content) {
-    if (typeof content === "string") {
-        return content;
-    }
-    return content
-        .map((part) => {
-        const metadata = {};
-        if ("providerOptions" in part) {
-            metadata.providerOptions = part.providerOptions;
-        }
-        if ("providerMetadata" in part) {
-            metadata.providerMetadata = part.providerMetadata;
-        }
-        switch (part.type) {
-            case "text":
-                return part;
-            case "image":
-                return {
-                    type: part.type,
-                    mediaType: getMimeOrMediaType(part),
-                    ...metadata,
-                    image: serializeDataOrUrl(part.image),
-                };
-            case "file":
-                return {
-                    type: part.type,
-                    data: serializeDataOrUrl(part.data),
-                    filename: part.filename,
-                    mediaType: getMimeOrMediaType(part),
-                    ...metadata,
-                };
-            case "tool-call":
-                // Handle legacy data where only args field exists
-                return {
-                    type: part.type,
-                    input: part.input ?? part?.args ?? {},
-                    /** @deprecated Use `input` instead. */
-                    args: part.input ?? part?.args ?? {},
-                    toolCallId: part.toolCallId,
-                    toolName: part.toolName,
-                    providerExecuted: part.providerExecuted,
-                    ...metadata,
-                };
-            case "tool-result":
-                return normalizeToolResult(part, metadata);
-            case "reasoning":
-                return {
-                    type: part.type,
-                    text: part.text,
-                    ...metadata,
-                };
-            case "tool-approval-request":
-                return {
-                    type: part.type,
-                    approvalId: part.approvalId,
-                    toolCallId: part.toolCallId,
-                    ...metadata,
-                };
-            case "tool-approval-response":
-                return {
-                    type: part.type,
-                    approvalId: part.approvalId,
-                    approved: part.approved,
-                    reason: part.reason,
-                    providerExecuted: part.providerExecuted,
-                    ...metadata,
-                };
-            // Not in current generation output, but could be in historical messages
-            default:
-                return null;
-        }
-    })
-        .filter((p) => p !== null);
-}
-export function toModelMessageContent(content) {
-    if (typeof content === "string") {
-        return content;
-    }
-    return content
-        .map((part) => {
-        const metadata = {};
-        if ("providerOptions" in part) {
-            metadata.providerOptions = part.providerOptions;
-        }
-        if ("providerMetadata" in part) {
-            metadata.providerMetadata = part.providerMetadata;
-        }
-        switch (part.type) {
-            case "text":
-                return {
-                    type: part.type,
-                    text: part.text,
-                    ...metadata,
-                };
-            case "image":
-                return {
-                    type: part.type,
-                    image: toModelMessageDataOrUrl(part.image),
-                    mediaType: getMimeOrMediaType(part),
-                    ...metadata,
-                };
-            case "file":
-                return {
-                    type: part.type,
-                    data: toModelMessageDataOrUrl(part.data),
-                    filename: part.filename,
-                    mediaType: getMimeOrMediaType(part),
-                    ...metadata,
-                };
-            case "tool-call": {
-                // Handle legacy data where only args field exists
-                const input = part.input ?? part?.args ?? {};
-                return {
-                    type: part.type,
-                    input,
-                    toolCallId: part.toolCallId,
-                    toolName: part.toolName,
-                    providerExecuted: part.providerExecuted,
-                    ...metadata,
-                };
-            }
-            case "tool-result": {
-                return normalizeToolResult(part, metadata);
-            }
-            case "reasoning":
-                return {
-                    type: part.type,
-                    text: part.text,
-                    ...metadata,
-                };
-            case "redacted-reasoning":
-                // TODO: should we just drop this?
-                return {
-                    type: "reasoning",
-                    text: "",
-                    ...metadata,
-                    providerOptions: metadata.providerOptions
-                        ? {
-                            ...Object.fromEntries(Object.entries(metadata.providerOptions ?? {}).map(([key, value]) => [
-                                key,
-                                { ...value, redactedData: part.data },
-                            ])),
-                        }
-                        : undefined,
-                };
-            case "source":
-                return part;
-            case "tool-approval-request":
-                return {
-                    type: part.type,
-                    approvalId: part.approvalId,
-                    toolCallId: part.toolCallId,
-                    ...metadata,
-                };
-            case "tool-approval-response":
-                return {
-                    type: part.type,
-                    approvalId: part.approvalId,
-                    approved: part.approved,
-                    reason: part.reason,
-                    providerExecuted: part.providerExecuted,
-                    ...metadata,
-                };
-            default:
-                return null;
-        }
-    })
-        .filter((p) => p !== null);
-}
-export function normalizeToolOutput(result) {
-    if (typeof result === "string") {
-        return {
-            type: "text",
-            value: result,
-        };
-    }
-    if (validate(vToolResultOutput, result)) {
-        return result;
-    }
-    return {
-        type: "json",
-        value: result ?? null,
-    };
-}
-function normalizeToolResult(part, metadata) {
-    return {
-        type: part.type,
-        output: part.output
-            ? validate(vToolResultOutput, part.output)
-                ? part.output
-                : normalizeToolOutput(JSON.stringify(part.output))
-            : normalizeToolOutput("result" in part ? part.result : undefined),
-        toolCallId: part.toolCallId,
-        toolName: part.toolName,
-        // Preserve isError flag for error reporting
-        ...("isError" in part && part.isError ? { isError: true } : {}),
-        ...metadata,
-    };
-}
-/**
- * Return a best-guess MIME type based on the magic-number signature
- * found at the start of an ArrayBuffer.
- *
- * @param buf – the source ArrayBuffer
- * @returns the detected MIME type, or `"application/octet-stream"` if unknown
- */
-export function guessMimeType(buf) {
-    if (typeof buf === "string") {
-        if (buf.match(/^data:\w+\/\w+;base64/)) {
-            return buf.split(";")[0].split(":")[1];
-        }
-        return "text/plain";
-    }
-    if (buf.byteLength < 4)
-        return "application/octet-stream";
-    // Read the first 12 bytes (enough for all signatures below)
-    const bytes = new Uint8Array(buf.slice(0, 12));
-    const hex = [...bytes].map((b) => b.toString(16).padStart(2, "0")).join("");
-    // Helper so we can look at only the needed prefix
-    const startsWith = (sig) => hex.startsWith(sig.toLowerCase());
-    // --- image formats ---
-    if (startsWith("89504e47"))
-        return "image/png"; // PNG  - 89 50 4E 47
-    if (startsWith("ffd8ffdb") ||
-        startsWith("ffd8ffe0") ||
-        startsWith("ffd8ffee") ||
-        startsWith("ffd8ffe1"))
-        return "image/jpeg"; // JPEG
-    if (startsWith("47494638"))
-        return "image/gif"; // GIF
-    if (startsWith("424d"))
-        return "image/bmp"; // BMP
-    if (startsWith("52494646") && hex.substr(16, 8) === "57454250")
-        return "image/webp"; // WEBP (RIFF....WEBP)
-    if (startsWith("49492a00"))
-        return "image/tiff"; // TIFF
-    // <svg in hex is 3c 3f 78 6d 6c
-    if (startsWith("3c737667"))
-        return "image/svg+xml"; // <svg
-    if (startsWith("3c3f786d"))
-        return "image/svg+xml"; // <?xm
-    // --- audio/video ---
-    if (startsWith("494433"))
-        return "audio/mpeg"; // MP3 (ID3)
-    if (startsWith("000001ba") || startsWith("000001b3"))
-        return "video/mpeg"; // MPEG container
-    if (startsWith("1a45dfa3"))
-        return "video/webm"; // WEBM / Matroska
-    if (startsWith("00000018") && hex.substr(16, 8) === "66747970")
-        return "video/mp4"; // MP4
-    if (startsWith("4f676753"))
-        return "audio/ogg"; // OGG / Opus
-    // --- documents & archives ---
-    if (startsWith("25504446"))
-        return "application/pdf"; // PDF
-    if (startsWith("504b0304") ||
-        startsWith("504b0506") ||
-        startsWith("504b0708"))
-        return "application/zip"; // ZIP / DOCX / PPTX / XLSX / EPUB
-    if (startsWith("52617221"))
-        return "application/x-rar-compressed"; // RAR
-    if (startsWith("7f454c46"))
-        return "application/x-elf"; // ELF binaries
-    if (startsWith("1f8b08"))
-        return "application/gzip"; // GZIP
-    if (startsWith("425a68"))
-        return "application/x-bzip2"; // BZIP2
-    if (startsWith("3c3f786d6c"))
-        return "application/xml"; // XML
-    // Plain text, JSON and others are trickier—fallback:
-    return "application/octet-stream";
-}
-/**
- * Serialize an AI SDK `DataContent` or `URL` to a Convex-serializable format.
- * @param dataOrUrl - The data or URL to serialize.
- * @returns The serialized data as an ArrayBuffer or the URL as a string.
- */
-export function serializeDataOrUrl(dataOrUrl) {
-    if (typeof dataOrUrl === "string") {
-        return dataOrUrl;
-    }
-    if (dataOrUrl instanceof ArrayBuffer) {
-        return dataOrUrl; // Already an ArrayBuffer
-    }
-    if (dataOrUrl instanceof URL) {
-        return dataOrUrl.toString();
-    }
-    return dataOrUrl.buffer.slice(dataOrUrl.byteOffset, dataOrUrl.byteOffset + dataOrUrl.byteLength);
-}
-export function toModelMessageDataOrUrl(urlOrString) {
-    if (urlOrString instanceof URL) {
-        return urlOrString;
-    }
-    if (typeof urlOrString === "string") {
-        if (urlOrString.startsWith("http://") ||
-            urlOrString.startsWith("https://")) {
-            return new URL(urlOrString);
-        }
-        return urlOrString;
-    }
-    return urlOrString;
-}
-export function toUIFilePart(part) {
-    const dataOrUrl = part.type === "image" ? part.image : part.data;
-    const url = dataOrUrl instanceof ArrayBuffer
-        ? convertUint8ArrayToBase64(new Uint8Array(dataOrUrl))
-        : dataOrUrl.toString();
-    return {
-        type: "file",
-        mediaType: part.mediaType,
-        filename: part.type === "file" ? part.filename : undefined,
-        url,
-        providerMetadata: part.providerOptions,
-    };
-}
-// Currently unused
-// export function toModelMessages(args: {
-//   messages?: ModelMessage[] | AIMessageWithoutId[];
-// }): ModelMessage[] {
-//   const messages: ModelMessage[] = [];
-//   if (args.messages) {
-//     if (
-//       args.messages.every(
-//         (m) => typeof m === "object" && m !== null && "parts" in m,
-//       )
-//     ) {
-//       messages.push(...convertToModelMessages(args.messages));
-//     } else {
-//       messages.push(...modelMessageSchema.array().parse(args.messages));
-//     }
-//   }
-//   return messages;
-// }
-//# sourceMappingURL=mapping.js.map