@rivus/runtime 0.16.6 → 0.18.0

package/dist/chunks/user-input-xml.js

@@ -0,0 +1,224 @@
+import { a as isValidUserInput } from "./agent-user-input.js";
+import { Effect, Result } from "effect";
+//#region src/core/application/agent-execution/loop/agent-loop.ts
+function createAgentLoopTextDelta(delta) {
+	return {
+		delta,
+		type: "assistant_text_delta"
+	};
+}
+function createAgentLoopTurnStart() {
+	return { type: "turn_start" };
+}
+function createAgentLoopThinkingDelta(delta) {
+	return {
+		delta,
+		type: "assistant_thinking_delta"
+	};
+}
+function createAgentLoopModelExecutionStart(options) {
+	return {
+		...options,
+		type: "model_execution_start"
+	};
+}
+function createAgentLoopModelExecutionEnd(options) {
+	return {
+		...options,
+		type: "model_execution_end"
+	};
+}
+function createAgentLoopSkillExecutionStart(options) {
+	return {
+		...options,
+		type: "skill_execution_start"
+	};
+}
+function createAgentLoopSkillExecutionEnd(options) {
+	return {
+		...options,
+		type: "skill_execution_end"
+	};
+}
+function createAgentLoopToolExecutionStart(options) {
+	return {
+		input: options.input,
+		toolCallId: options.toolCallId,
+		toolName: options.toolName,
+		type: "tool_execution_start"
+	};
+}
+function createAgentLoopToolExecutionUpdate(options) {
+	return {
+		input: options.input,
+		partialResult: options.partialResult,
+		toolCallId: options.toolCallId,
+		toolName: options.toolName,
+		type: "tool_execution_update"
+	};
+}
+function createAgentLoopToolExecutionEnd(options) {
+	return {
+		isError: options.isError,
+		result: options.result,
+		toolCallId: options.toolCallId,
+		toolName: options.toolName,
+		type: "tool_execution_end"
+	};
+}
+//#endregion
+//#region src/adapters/model-input/user-input-xml.ts
+/**
+* Prepare one run's structured input at the provider edge.  UserInput never
+* stores paths.  Paths and bytes are resolved through the run-scoped access
+* binding immediately before the provider prompt is sent.
+*/
+function prepareAgentModelInput(input, options = {}) {
+	return Effect.gen(function* () {
+		const images = [];
+		const unavailable = [];
+		let hasUsableContent = false;
+		if (!input.userInput) return {
+			hasUsableContent: input.text.trim().length > 0,
+			images,
+			text: input.text
+		};
+		if (!isValidUserInput(input.userInput)) return {
+			hasUsableContent: false,
+			images,
+			fallbackText: "收到的结构化消息格式无效，无法安全处理。",
+			text: input.text
+		};
+		const rendered = yield* renderMessage(input.userInput.message, {
+			access: input.attachmentAccess,
+			images,
+			markUsable: () => {
+				hasUsableContent = true;
+			},
+			options,
+			unavailable
+		});
+		const fallbackText = [...collectUnavailableText(input.userInput.message), ...unavailable].join("\n");
+		return {
+			hasUsableContent,
+			images,
+			...hasUsableContent || !fallbackText ? {} : { fallbackText },
+			text: `<user_message version="1" id="${escapeAttribute(input.userInput.deliveryId)}">${rendered}</user_message>`
+		};
+	});
+}
+function collectUnavailableText(message) {
+	const notices = [];
+	const visit = (current) => {
+		for (const part of current.parts) if (part.type === "notice") notices.push(part.text);
+		else if (part.type === "quote") for (const nested of part.messages) visit(nested);
+	};
+	visit(message);
+	return notices;
+}
+function renderMessage(message, context) {
+	return Effect.gen(function* () {
+		const parts = [];
+		for (const part of message.parts) parts.push(yield* renderPart(part, context));
+		return `<message ${[
+			`id="${escapeAttribute(message.id)}"`,
+			`source="${message.source}"`,
+			...message.author?.id ? [`author_id="${escapeAttribute(message.author.id)}"`] : [],
+			...message.author?.name ? [`author_name="${escapeAttribute(message.author.name)}"`] : [],
+			...message.sentAt ? [`sent_at="${escapeAttribute(message.sentAt)}"`] : []
+		].join(" ")}>${parts.join("")}</message>`;
+	});
+}
+function renderPart(part, context) {
+	const { access, markUsable, unavailable } = context;
+	if (part.type === "text") {
+		if (part.text.trim().length > 0) markUsable();
+		return Effect.succeed(`<text>${escapeXml(part.text)}</text>`);
+	}
+	if (part.type === "notice") return Effect.sync(() => {
+		const recovery = access?.trustedRecovery?.(part);
+		if (recovery) markUsable();
+		return renderNotice(part, recovery);
+	});
+	if (part.type === "quote") return Effect.gen(function* () {
+		const messages = [];
+		for (const message of part.messages) messages.push(yield* renderMessage(message, context));
+		return `<quote kind="${part.kind}">${messages.join("")}</quote>`;
+	});
+	if (part.type === "image") return renderImage(part, context);
+	return renderAttachment(part, access, markUsable, unavailable);
+}
+function renderImage(part, context) {
+	const { access, images, markUsable, options, unavailable } = context;
+	const imageBytes = images.reduce((total, image) => total + Buffer.byteLength(image.data, "base64"), 0);
+	const withinImageBudget = (size) => images.length < 4 && size <= 5242880 && imageBytes + size <= 10485760;
+	if (options.allowImages === false) return Effect.succeed(renderUnavailableNoticeAndRecord(part.attachmentId, "当前模型不支持原生图片输入", unavailable));
+	if (!access) return Effect.succeed(renderUnavailableNoticeAndRecord(part.attachmentId, "执行环境未提供图片读取能力", unavailable));
+	if (!/^(image\/(png|jpeg|webp|gif))$/.test(part.mimeType)) return Effect.succeed(renderUnavailableNoticeAndRecord(part.attachmentId, "当前模型不支持此图片格式", unavailable));
+	if (part.size !== void 0 && !withinImageBudget(part.size)) return Effect.succeed(renderUnavailableNoticeAndRecord(part.attachmentId, "超出本次图片输入预算", unavailable));
+	if (part.size !== void 0 && access.canProvideImage && !access.canProvideImage({
+		mimeType: part.mimeType,
+		size: part.size,
+		index: images.length + 1
+	})) return Effect.succeed(renderUnavailableNoticeAndRecord(part.attachmentId, "超出本次图片输入预算或当前模型不支持视觉", unavailable));
+	return access.readBytes(part.attachmentId).pipe(Effect.result, Effect.map((result) => {
+		if (Result.isFailure(result)) return renderUnavailableNoticeAndRecord(part.attachmentId, "图片在执行环境中不可读", unavailable);
+		if (!withinImageBudget(result.success.byteLength) || result.success.byteLength === 0) return renderUnavailableNoticeAndRecord(part.attachmentId, "超出本次图片输入预算或图片为空", unavailable);
+		if (access.canProvideImage && !access.canProvideImage({
+			mimeType: part.mimeType,
+			size: result.success.byteLength,
+			index: images.length + 1
+		})) return renderUnavailableNoticeAndRecord(part.attachmentId, "超出本次图片输入预算或当前模型不支持视觉", unavailable);
+		const imageIndex = images.length + 1;
+		images.push({
+			type: "image",
+			attachmentId: part.attachmentId,
+			data: Buffer.from(result.success).toString("base64"),
+			imageIndex,
+			mimeType: part.mimeType
+		});
+		markUsable();
+		return `<image attachment_id="${escapeAttribute(part.attachmentId)}" image_index="${imageIndex}" mime_type="${escapeAttribute(part.mimeType)}" state="provided" />`;
+	}));
+}
+function renderAttachment(part, access, markUsable, unavailable) {
+	if (!access) return Effect.succeed(renderUnavailableNoticeAndRecord(part.fileName ?? part.attachmentId, "执行环境未提供文件读取能力", unavailable));
+	return access.materialize(part.attachmentId).pipe(Effect.result, Effect.map((result) => {
+		if (Result.isFailure(result)) return renderUnavailableNoticeAndRecord(part.fileName ?? part.attachmentId, "附件无法物化到执行环境", unavailable);
+		const value = result.success;
+		const canUnderstand = access.canUnderstand?.({
+			mediaKind: part.mediaKind,
+			mimeType: value.mimeType
+		}) ?? false;
+		if (canUnderstand) markUsable();
+		else unavailable.push(`${part.fileName ?? part.attachmentId}：附件已下载，但当前执行环境没有适用的${part.mediaKind === "audio" ? "音频理解工具，请改发文字" : part.mediaKind === "video" ? "视频解析工具" : "文件读取或解析工具"}`);
+		const durationMs = value.durationMs ?? part.durationMs;
+		return `<file attachment_id="${escapeAttribute(value.attachmentId)}"${value.fileName || part.fileName ? ` name="${escapeAttribute(value.fileName ?? part.fileName ?? "attachment")}"` : ""} path="${escapeAttribute(value.path)}" mime_type="${escapeAttribute(value.mimeType)}" size_bytes="${value.size}" media_kind="${value.mediaKind}" state="available"${durationMs === void 0 ? "" : ` duration_ms="${durationMs}"`}${canUnderstand ? "" : " understood=\"false\""} />`;
+	}));
+}
+function renderNotice(part, recovery) {
+	const retryable = recovery ? "true" : "false";
+	if (!recovery) return `<notice reason="${part.reason}" retryable="${retryable}">${escapeXml(part.text)}</notice>`;
+	const args = Object.entries(recovery.arguments).map(([key, value]) => `<arg name="${escapeAttribute(key)}" value="${escapeAttribute(String(value))}" />`).join("");
+	return `<notice reason="${part.reason}" retryable="true">${escapeXml(part.text)}<recovery tool="${escapeAttribute(recovery.toolName)}">${args}</recovery></notice>`;
+}
+function renderUnavailableNotice(name, reason) {
+	return `<notice reason="unavailable" retryable="false">${escapeXml(`${name}：${reason}`)}</notice>`;
+}
+function renderUnavailableNoticeAndRecord(name, reason, unavailable) {
+	unavailable.push(`${name}：${reason}`);
+	return renderUnavailableNotice(name, reason);
+}
+function escapeAttribute(value) {
+	return escapeXml(value).replaceAll("'", "&apos;").replaceAll("\"", "&quot;");
+}
+function escapeXml(value) {
+	return Array.from(value).map((character) => {
+		const code = character.codePointAt(0) ?? 0;
+		if (code >= 0 && code < 32 && code !== 9 && code !== 10 && code !== 13 || code === 65534 || code === 65535) return "�";
+		return character;
+	}).join("").replaceAll("&", "&amp;").replaceAll("<", "&lt;").replaceAll(">", "&gt;");
+}
+//#endregion
+export { createAgentLoopSkillExecutionStart as a, createAgentLoopToolExecutionEnd as c, createAgentLoopTurnStart as d, createAgentLoopSkillExecutionEnd as i, createAgentLoopToolExecutionStart as l, createAgentLoopModelExecutionEnd as n, createAgentLoopTextDelta as o, createAgentLoopModelExecutionStart as r, createAgentLoopThinkingDelta as s, prepareAgentModelInput as t, createAgentLoopToolExecutionUpdate as u };
+