@assistant-ui/react-ag-ui 0.0.31 → 0.0.33

package/dist/runtime/adapter/conversions.js

@@ -1,427 +1,331 @@
 "use client";
 import { toToolsJSONSchema } from "assistant-stream";
+//#region src/runtime/adapter/conversions.ts
 const isObject = (value) => typeof value === "object" && value !== null;
 const getString = (record, key) => {
-    const value = record[key];
-    return typeof value === "string" ? value : undefined;
+	const value = record[key];
+	return typeof value === "string" ? value : void 0;
 };
 const getToolCallId = (record) => getString(record, "toolCallId") ?? getString(record, "tool_call_id");
 function parseJSONText(value) {
-    if (!value)
-        return value;
-    try {
-        return JSON.parse(value);
-    }
-    catch {
-        return value;
-    }
+	if (!value) return value;
+	try {
+		return JSON.parse(value);
+	} catch {
+		return value;
+	}
 }
 function generateId() {
-    return (globalThis.crypto?.randomUUID?.() ??
-        Math.random().toString(36).slice(2));
+	return globalThis.crypto?.randomUUID?.() ?? Math.random().toString(36).slice(2);
 }
 function normalizeToolCall(part) {
-    const id = part.toolCallId ?? generateId();
-    const argsText = typeof part.argsText === "string"
-        ? part.argsText
-        : JSON.stringify(part.args ?? {});
-    return {
-        id,
-        call: {
-            id,
-            type: "function",
-            function: {
-                name: part.toolName ?? "tool",
-                arguments: argsText,
-            },
-        },
-    };
+	const id = part.toolCallId ?? generateId();
+	const argsText = typeof part.argsText === "string" ? part.argsText : JSON.stringify(part.args ?? {});
+	return {
+		id,
+		call: {
+			id,
+			type: "function",
+			function: {
+				name: part.toolName ?? "tool",
+				arguments: argsText
+			}
+		}
+	};
 }
 function extractText(content) {
-    if (typeof content === "string")
-        return content;
-    if (!Array.isArray(content))
-        return "";
-    return content
-        .filter((part) => part?.type === "text" && typeof part?.text === "string")
-        .map((part) => part.text)
-        .join("\n");
+	if (typeof content === "string") return content;
+	if (!Array.isArray(content)) return "";
+	return content.filter((part) => part?.type === "text" && typeof part?.text === "string").map((part) => part.text).join("\n");
 }
 function parseDataUrl(value) {
-    const match = value.match(/^data:([^;,]+)(?:;[^;,]+)*;base64,(.+)$/);
-    if (!match)
-        return null;
-    return { mimeType: match[1], data: match[2] };
+	const match = value.match(/^data:([^;,]+)(?:;[^;,]+)*;base64,(.+)$/);
+	if (!match) return null;
+	return {
+		mimeType: match[1],
+		data: match[2]
+	};
 }
 const httpUrlPattern = /^https?:\/\//i;
 function toInputContent(part, fallbackMimeType) {
-    if (!isObject(part))
-        return null;
-    const type = getString(part, "type");
-    if (type === "text") {
-        const text = getString(part, "text");
-        if (text === undefined)
-            return null;
-        return { type: "text", text };
-    }
-    if (type === "image") {
-        const image = getString(part, "image");
-        if (image === undefined)
-            return null;
-        const parsed = parseDataUrl(image);
-        if (parsed) {
-            return {
-                type: "image",
-                source: {
-                    type: "data",
-                    value: parsed.data,
-                    mimeType: parsed.mimeType,
-                },
-            };
-        }
-        return {
-            type: "image",
-            source: {
-                type: "url",
-                value: image,
-                ...(fallbackMimeType !== undefined
-                    ? { mimeType: fallbackMimeType }
-                    : {}),
-            },
-        };
-    }
-    if (type === "file") {
-        const data = getString(part, "data");
-        if (data === undefined)
-            return null;
-        const partMimeType = getString(part, "mimeType");
-        const filename = getString(part, "filename");
-        const mimeType = partMimeType || fallbackMimeType || "application/octet-stream";
-        if (httpUrlPattern.test(data)) {
-            return {
-                type: "binary",
-                mimeType,
-                url: data,
-                ...(filename !== undefined ? { filename } : {}),
-            };
-        }
-        const parsed = parseDataUrl(data);
-        return {
-            type: "binary",
-            mimeType: parsed?.mimeType ?? mimeType,
-            data: parsed?.data ?? data,
-            ...(filename !== undefined ? { filename } : {}),
-        };
-    }
-    return null;
+	if (!isObject(part)) return null;
+	const type = getString(part, "type");
+	if (type === "text") {
+		const text = getString(part, "text");
+		if (text === void 0) return null;
+		return {
+			type: "text",
+			text
+		};
+	}
+	if (type === "image") {
+		const image = getString(part, "image");
+		if (image === void 0) return null;
+		const parsed = parseDataUrl(image);
+		if (parsed) return {
+			type: "image",
+			source: {
+				type: "data",
+				value: parsed.data,
+				mimeType: parsed.mimeType
+			}
+		};
+		return {
+			type: "image",
+			source: {
+				type: "url",
+				value: image,
+				...fallbackMimeType !== void 0 ? { mimeType: fallbackMimeType } : {}
+			}
+		};
+	}
+	if (type === "file") {
+		const data = getString(part, "data");
+		if (data === void 0) return null;
+		const partMimeType = getString(part, "mimeType");
+		const filename = getString(part, "filename");
+		const mimeType = partMimeType || fallbackMimeType || "application/octet-stream";
+		if (httpUrlPattern.test(data)) return {
+			type: "binary",
+			mimeType,
+			url: data,
+			...filename !== void 0 ? { filename } : {}
+		};
+		const parsed = parseDataUrl(data);
+		return {
+			type: "binary",
+			mimeType: parsed?.mimeType ?? mimeType,
+			data: parsed?.data ?? data,
+			...filename !== void 0 ? { filename } : {}
+		};
+	}
+	return null;
 }
 function buildUserContent(message) {
-    // File parts in message.content are intentionally skipped: the canonical
-    // binary payload for files always flows through message.attachments.
-    const contentParts = Array.isArray(message.content)
-        ? message.content.filter((part) => !(isObject(part) && part.type === "file"))
-        : [];
-    const attachments = message.attachments ?? [];
-    const converted = [];
-    // Promote string-form content to a leading text part so it survives when
-    // non-text attachments are present (fromAgUiMessages emits string content).
-    if (typeof message.content === "string" && message.content.length > 0) {
-        converted.push({ type: "text", text: message.content });
-    }
-    for (const part of contentParts) {
-        const input = toInputContent(part, undefined);
-        if (input)
-            converted.push(input);
-    }
-    for (const attachment of attachments) {
-        if (!isObject(attachment))
-            continue;
-        const attachmentContent = attachment.content;
-        if (!Array.isArray(attachmentContent))
-            continue;
-        const fallbackMime = getString(attachment, "contentType");
-        for (const part of attachmentContent) {
-            const input = toInputContent(part, fallbackMime);
-            if (input)
-                converted.push(input);
-        }
-    }
-    const hasNonText = converted.some((part) => part.type !== "text");
-    if (hasNonText)
-        return converted;
-    // All-text path: collapse to plain string. Join text parts collected from
-    // both content and attachments so attachment-sourced text is not dropped.
-    if (converted.length === 0)
-        return extractText(message.content);
-    return converted
-        .filter((part) => part.type === "text")
-        .map((part) => part.text)
-        .join("\n");
+	const contentParts = Array.isArray(message.content) ? message.content.filter((part) => !(isObject(part) && part.type === "file")) : [];
+	const attachments = message.attachments ?? [];
+	const converted = [];
+	if (typeof message.content === "string" && message.content.length > 0) converted.push({
+		type: "text",
+		text: message.content
+	});
+	for (const part of contentParts) {
+		const input = toInputContent(part, void 0);
+		if (input) converted.push(input);
+	}
+	for (const attachment of attachments) {
+		if (!isObject(attachment)) continue;
+		const attachmentContent = attachment.content;
+		if (!Array.isArray(attachmentContent)) continue;
+		const fallbackMime = getString(attachment, "contentType");
+		for (const part of attachmentContent) {
+			const input = toInputContent(part, fallbackMime);
+			if (input) converted.push(input);
+		}
+	}
+	if (converted.some((part) => part.type !== "text")) return converted;
+	if (converted.length === 0) return extractText(message.content);
+	return converted.filter((part) => part.type === "text").map((part) => part.text).join("\n");
 }
 function toToolCallPart(value) {
-    if (!isObject(value))
-        return null;
-    const rawFunction = isObject(value.function) ? value.function : null;
-    const toolCallId = getString(value, "toolCallId") ?? getString(value, "id");
-    const toolName = getString(value, "toolName") ??
-        getString(value, "name") ??
-        (rawFunction ? getString(rawFunction, "name") : undefined) ??
-        "tool";
-    const argsText = getString(value, "argsText") ??
-        getString(value, "arguments") ??
-        (rawFunction ? getString(rawFunction, "arguments") : undefined);
-    const parsedArgs = typeof argsText === "string" ? parseJSONText(argsText) : undefined;
-    const args = isObject(parsedArgs) && !Array.isArray(parsedArgs)
-        ? parsedArgs
-        : isObject(value.args) && !Array.isArray(value.args)
-            ? value.args
-            : undefined;
-    const part = {
-        type: "tool-call",
-        ...(toolCallId !== undefined ? { toolCallId } : {}),
-        toolName,
-        argsText: argsText ?? JSON.stringify(args ?? {}),
-        ...(args !== undefined ? { args } : {}),
-    };
-    if (value.type === "tool-call") {
-        const result = value.result;
-        const isError = value.isError;
-        if (result !== undefined)
-            part.result = result;
-        if (typeof isError === "boolean")
-            part.isError = isError;
-    }
-    return part;
+	if (!isObject(value)) return null;
+	const rawFunction = isObject(value.function) ? value.function : null;
+	const toolCallId = getString(value, "toolCallId") ?? getString(value, "id");
+	const toolName = getString(value, "toolName") ?? getString(value, "name") ?? (rawFunction ? getString(rawFunction, "name") : void 0) ?? "tool";
+	const argsText = getString(value, "argsText") ?? getString(value, "arguments") ?? (rawFunction ? getString(rawFunction, "arguments") : void 0);
+	const parsedArgs = typeof argsText === "string" ? parseJSONText(argsText) : void 0;
+	const args = isObject(parsedArgs) && !Array.isArray(parsedArgs) ? parsedArgs : isObject(value.args) && !Array.isArray(value.args) ? value.args : void 0;
+	const part = {
+		type: "tool-call",
+		...toolCallId !== void 0 ? { toolCallId } : {},
+		toolName,
+		argsText: argsText ?? JSON.stringify(args ?? {}),
+		...args !== void 0 ? { args } : {}
+	};
+	if (value.type === "tool-call") {
+		const result = value.result;
+		const isError = value.isError;
+		if (result !== void 0) part.result = result;
+		if (typeof isError === "boolean") part.isError = isError;
+	}
+	return part;
 }
 function extractAssistantToolCalls(message) {
-    const parts = [];
-    const seenToolCallIds = new Set();
-    const pushPart = (part) => {
-        if (!part)
-            return;
-        const id = part.toolCallId ?? generateId();
-        if (seenToolCallIds.has(id))
-            return;
-        seenToolCallIds.add(id);
-        parts.push({
-            ...part,
-            toolCallId: id,
-        });
-    };
-    const content = message.content;
-    if (Array.isArray(content)) {
-        for (const part of content) {
-            if (isObject(part) && part.type === "tool-call") {
-                pushPart(toToolCallPart(part));
-            }
-        }
-    }
-    const toolCalls = Array.isArray(message.toolCalls)
-        ? message.toolCalls
-        : Array.isArray(message.tool_calls)
-            ? message.tool_calls
-            : [];
-    for (const call of toolCalls) {
-        pushPart(toToolCallPart(call));
-    }
-    return parts;
+	const parts = [];
+	const seenToolCallIds = /* @__PURE__ */ new Set();
+	const pushPart = (part) => {
+		if (!part) return;
+		const id = part.toolCallId ?? generateId();
+		if (seenToolCallIds.has(id)) return;
+		seenToolCallIds.add(id);
+		parts.push({
+			...part,
+			toolCallId: id
+		});
+	};
+	const content = message.content;
+	if (Array.isArray(content)) {
+		for (const part of content) if (isObject(part) && part.type === "tool-call") pushPart(toToolCallPart(part));
+	}
+	const toolCalls = Array.isArray(message.toolCalls) ? message.toolCalls : Array.isArray(message.tool_calls) ? message.tool_calls : [];
+	for (const call of toolCalls) pushPart(toToolCallPart(call));
+	return parts;
 }
 function toAssistantSnapshotMessage(rawMessage) {
-    const text = extractText(rawMessage.content);
-    const toolCallParts = extractAssistantToolCalls(rawMessage);
-    const assistantContent = [
-        ...(text.length > 0 ? [{ type: "text", text }] : []),
-        ...toolCallParts,
-    ];
-    const messageName = getString(rawMessage, "name");
-    return {
-        id: getString(rawMessage, "id") ?? generateId(),
-        role: "assistant",
-        content: assistantContent.length > 0 ? assistantContent : "",
-        ...(messageName !== undefined ? { name: messageName } : {}),
-    };
+	const text = extractText(rawMessage.content);
+	const toolCallParts = extractAssistantToolCalls(rawMessage);
+	const assistantContent = [...text.length > 0 ? [{
+		type: "text",
+		text
+	}] : [], ...toolCallParts];
+	const messageName = getString(rawMessage, "name");
+	return {
+		id: getString(rawMessage, "id") ?? generateId(),
+		role: "assistant",
+		content: assistantContent.length > 0 ? assistantContent : "",
+		...messageName !== void 0 ? { name: messageName } : {}
+	};
 }
 function toUserOrSystemSnapshotMessage(role, rawMessage) {
-    const messageName = getString(rawMessage, "name");
-    return {
-        id: getString(rawMessage, "id") ?? generateId(),
-        role,
-        content: extractText(rawMessage.content),
-        ...(messageName !== undefined ? { name: messageName } : {}),
-    };
+	const messageName = getString(rawMessage, "name");
+	return {
+		id: getString(rawMessage, "id") ?? generateId(),
+		role,
+		content: extractText(rawMessage.content),
+		...messageName !== void 0 ? { name: messageName } : {}
+	};
 }
-export function fromAgUiMessages(messages) {
-    const converted = [];
-    for (const rawMessage of messages) {
-        if (!isObject(rawMessage))
-            continue;
-        const role = getString(rawMessage, "role");
-        if (!role)
-            continue;
-        if (role === "tool") {
-            const toolCallId = getToolCallId(rawMessage) ?? `tool-${generateId()}`;
-            const toolMessageId = getString(rawMessage, "id");
-            const result = rawMessage.result !== undefined
-                ? rawMessage.result
-                : typeof rawMessage.content === "string"
-                    ? parseJSONText(rawMessage.content)
-                    : rawMessage.content;
-            const isError = typeof rawMessage.error === "string" ||
-                rawMessage.isError === true ||
-                rawMessage.status === "error"
-                ? true
-                : rawMessage.isError === false
-                    ? false
-                    : undefined;
-            let updated = false;
-            for (let messageIndex = converted.length - 1; messageIndex >= 0 && !updated; messageIndex--) {
-                const message = converted[messageIndex];
-                if (!message ||
-                    message.role !== "assistant" ||
-                    !Array.isArray(message.content))
-                    continue;
-                for (let partIndex = message.content.length - 1; partIndex >= 0; partIndex--) {
-                    const part = message.content[partIndex];
-                    if (!isObject(part) || part.type !== "tool-call")
-                        continue;
-                    if (getString(part, "toolCallId") !== toolCallId)
-                        continue;
-                    const updatedPart = {
-                        ...part,
-                        result,
-                        ...(isError !== undefined ? { isError } : {}),
-                        ...(toolMessageId !== undefined
-                            ? { unstable_toolMessageId: toolMessageId }
-                            : {}),
-                    };
-                    const updatedContent = message.content.map((contentPart, index) => index === partIndex ? updatedPart : contentPart);
-                    converted[messageIndex] = { ...message, content: updatedContent };
-                    updated = true;
-                    break;
-                }
-            }
-            if (updated) {
-                continue;
-            }
-            const id = toolMessageId ?? toolCallId;
-            const toolName = getString(rawMessage, "name") ??
-                getString(rawMessage, "toolName") ??
-                "tool";
-            converted.push({
-                id: `${id}:assistant`,
-                role: "assistant",
-                content: [
-                    {
-                        type: "tool-call",
-                        toolCallId,
-                        toolName,
-                        args: {},
-                        argsText: "{}",
-                        result,
-                        ...(isError !== undefined ? { isError } : {}),
-                        ...(toolMessageId !== undefined
-                            ? { unstable_toolMessageId: toolMessageId }
-                            : {}),
-                    },
-                ],
-            });
-            continue;
-        }
-        if (role === "assistant") {
-            converted.push(toAssistantSnapshotMessage(rawMessage));
-            continue;
-        }
-        if (role === "user" || role === "system") {
-            converted.push(toUserOrSystemSnapshotMessage(role, rawMessage));
-        }
-    }
-    return converted;
+function fromAgUiMessages(messages) {
+	const converted = [];
+	for (const rawMessage of messages) {
+		if (!isObject(rawMessage)) continue;
+		const role = getString(rawMessage, "role");
+		if (!role) continue;
+		if (role === "tool") {
+			const toolCallId = getToolCallId(rawMessage) ?? `tool-${generateId()}`;
+			const toolMessageId = getString(rawMessage, "id");
+			const result = rawMessage.result !== void 0 ? rawMessage.result : typeof rawMessage.content === "string" ? parseJSONText(rawMessage.content) : rawMessage.content;
+			const isError = typeof rawMessage.error === "string" || rawMessage.isError === true || rawMessage.status === "error" ? true : rawMessage.isError === false ? false : void 0;
+			let updated = false;
+			for (let messageIndex = converted.length - 1; messageIndex >= 0 && !updated; messageIndex--) {
+				const message = converted[messageIndex];
+				if (!message || message.role !== "assistant" || !Array.isArray(message.content)) continue;
+				for (let partIndex = message.content.length - 1; partIndex >= 0; partIndex--) {
+					const part = message.content[partIndex];
+					if (!isObject(part) || part.type !== "tool-call") continue;
+					if (getString(part, "toolCallId") !== toolCallId) continue;
+					const updatedPart = {
+						...part,
+						result,
+						...isError !== void 0 ? { isError } : {},
+						...toolMessageId !== void 0 ? { unstable_toolMessageId: toolMessageId } : {}
+					};
+					const updatedContent = message.content.map((contentPart, index) => index === partIndex ? updatedPart : contentPart);
+					converted[messageIndex] = {
+						...message,
+						content: updatedContent
+					};
+					updated = true;
+					break;
+				}
+			}
+			if (updated) continue;
+			const id = toolMessageId ?? toolCallId;
+			const toolName = getString(rawMessage, "name") ?? getString(rawMessage, "toolName") ?? "tool";
+			converted.push({
+				id: `${id}:assistant`,
+				role: "assistant",
+				content: [{
+					type: "tool-call",
+					toolCallId,
+					toolName,
+					args: {},
+					argsText: "{}",
+					result,
+					...isError !== void 0 ? { isError } : {},
+					...toolMessageId !== void 0 ? { unstable_toolMessageId: toolMessageId } : {}
+				}]
+			});
+			continue;
+		}
+		if (role === "assistant") {
+			converted.push(toAssistantSnapshotMessage(rawMessage));
+			continue;
+		}
+		if (role === "user" || role === "system") converted.push(toUserOrSystemSnapshotMessage(role, rawMessage));
+	}
+	return converted;
 }
 function convertAssistantMessage(message, converted) {
-    const content = extractText(message.content);
-    const contentArray = Array.isArray(message.content) ? message.content : [];
-    const toolCallParts = contentArray.filter((part) => part?.type === "tool-call");
-    const toolCalls = toolCallParts.map((part) => ({
-        ...normalizeToolCall(part),
-        part,
-    }));
-    const assistantMessage = {
-        id: message.id,
-        role: "assistant",
-        content,
-    };
-    if (message.name) {
-        assistantMessage.name = message.name;
-    }
-    if (toolCalls.length > 0) {
-        assistantMessage.toolCalls = toolCalls.map((entry) => entry.call);
-    }
-    converted.push(assistantMessage);
-    for (const { id: toolCallId, part } of toolCalls) {
-        if (part.result === undefined)
-            continue;
-        const resultContent = typeof part.result === "string"
-            ? part.result
-            : JSON.stringify(part.result);
-        const toolMessage = {
-            id: part.unstable_toolMessageId ?? `${toolCallId}:tool`,
-            role: "tool",
-            content: resultContent,
-            toolCallId,
-        };
-        if (part.isError) {
-            toolMessage.error = resultContent;
-        }
-        converted.push(toolMessage);
-    }
+	const content = extractText(message.content);
+	const toolCalls = (Array.isArray(message.content) ? message.content : []).filter((part) => part?.type === "tool-call").map((part) => ({
+		...normalizeToolCall(part),
+		part
+	}));
+	const assistantMessage = {
+		id: message.id,
+		role: "assistant",
+		content
+	};
+	if (message.name) assistantMessage.name = message.name;
+	if (toolCalls.length > 0) assistantMessage.toolCalls = toolCalls.map((entry) => entry.call);
+	converted.push(assistantMessage);
+	for (const { id: toolCallId, part } of toolCalls) {
+		if (part.result === void 0) continue;
+		const resultContent = typeof part.result === "string" ? part.result : JSON.stringify(part.result);
+		const toolMessage = {
+			id: part.unstable_toolMessageId ?? `${toolCallId}:tool`,
+			role: "tool",
+			content: resultContent,
+			toolCallId
+		};
+		if (part.isError) toolMessage.error = resultContent;
+		converted.push(toolMessage);
+	}
 }
 function convertToolMessage(message, converted) {
-    const content = extractText(message.content);
-    const toolCallId = message.toolCallId ?? generateId();
-    const toolMessage = {
-        id: message.id,
-        role: "tool",
-        content,
-        toolCallId,
-    };
-    if (typeof message.error === "string") {
-        toolMessage.error = message.error;
-    }
-    converted.push(toolMessage);
+	const content = extractText(message.content);
+	const toolCallId = message.toolCallId ?? generateId();
+	const toolMessage = {
+		id: message.id,
+		role: "tool",
+		content,
+		toolCallId
+	};
+	if (typeof message.error === "string") toolMessage.error = message.error;
+	converted.push(toolMessage);
 }
-export function toAgUiMessages(messages) {
-    const converted = [];
-    for (const message of messages) {
-        if (message.role === "assistant") {
-            convertAssistantMessage(message, converted);
-            continue;
-        }
-        if (message.role === "tool") {
-            convertToolMessage(message, converted);
-            continue;
-        }
-        const genericMessage = {
-            id: message.id,
-            role: message.role,
-            content: message.role === "user"
-                ? buildUserContent(message)
-                : extractText(message.content),
-        };
-        if (message.name) {
-            genericMessage.name = message.name;
-        }
-        converted.push(genericMessage);
-    }
-    return converted;
+function toAgUiMessages(messages) {
+	const converted = [];
+	for (const message of messages) {
+		if (message.role === "assistant") {
+			convertAssistantMessage(message, converted);
+			continue;
+		}
+		if (message.role === "tool") {
+			convertToolMessage(message, converted);
+			continue;
+		}
+		const genericMessage = {
+			id: message.id,
+			role: message.role,
+			content: message.role === "user" ? buildUserContent(message) : extractText(message.content)
+		};
+		if (message.name) genericMessage.name = message.name;
+		converted.push(genericMessage);
+	}
+	return converted;
 }
-export function toAgUiTools(tools) {
-    if (!tools)
-        return [];
-    const toolsSchema = toToolsJSONSchema(tools);
-    return Object.entries(toolsSchema).map(([name, tool]) => ({
-        name,
-        description: tool.description,
-        parameters: tool.parameters,
-    }));
+function toAgUiTools(tools) {
+	if (!tools) return [];
+	const toolsSchema = toToolsJSONSchema(tools);
+	return Object.entries(toolsSchema).map(([name, tool]) => ({
+		name,
+		description: tool.description,
+		parameters: tool.parameters
+	}));
 }
+//#endregion
+export { fromAgUiMessages, toAgUiMessages, toAgUiTools };
+
 //# sourceMappingURL=conversions.js.map