foliko 2.0.8 → 2.0.10

package/src/utils/message-validator.js

@@ -175,6 +175,63 @@ function validateAll(messages) {
   return validateMessagesPairing(messages);
 }
 
+// AI SDK v6 ToolResultOutput 允许的类型（用于 ModelMessage[] 输入 schema）
+const VALID_OUTPUT_TYPES = new Set(['text', 'json', 'execution-denied', 'content']);
+
+/**
+ * 规范化 tool-result 的 output 字段。
+ * AI SDK v6 响应中允许 'error-text' / 'error-json'（表示工具执行失败），
+ * 但 ModelMessage 输入 schema 不接受这两类。需要在发送前 / 持久化前转换。
+ *
+ * 转换规则：
+ * - error-text → text
+ * - error-json → json
+ * - 任何未知 type → text（保留 value 字段）
+ * - output 缺失或非对象 → 包装为 text 'null'
+ *
+ * @param {Array} messages - 消息数组（会被修改）
+ * @returns {number} 规范化次数
+ */
+function normalizeToolOutputs(messages) {
+  let fixed = 0;
+  for (const msg of messages) {
+    if (msg.role !== 'tool' || !Array.isArray(msg.content)) continue;
+    for (const part of msg.content) {
+      if (!part || part.type !== 'tool-result') continue;
+      const out = part.output;
+      // 缺 output 或 output 非对象：包装为 text
+      if (out === undefined || out === null || typeof out !== 'object') {
+        const v = out === undefined ? 'undefined' : (typeof out === 'string' ? out : JSON.stringify(out));
+        part.output = { type: 'text', value: String(v) };
+        fixed++;
+        continue;
+      }
+      // 已知合法 type：跳过
+      if (VALID_OUTPUT_TYPES.has(out.type)) continue;
+      // error-text → text（保留原 value 字符串）
+      if (out.type === 'error-text' || out.type === 'error_text') {
+        part.output = { type: 'text', value: typeof out.value === 'string' ? out.value : String(out.value ?? '') };
+        fixed++;
+        continue;
+      }
+      // error-json → json
+      if (out.type === 'error-json' || out.type === 'error_json') {
+        part.output = { type: 'json', value: out.value ?? null };
+        fixed++;
+        continue;
+      }
+      // 其它未知 type：降级为 text（保留 value 字符串或 JSON 序列化）
+      try {
+        part.output = { type: 'text', value: JSON.stringify(out) };
+      } catch {
+        part.output = { type: 'text', value: String(out) };
+      }
+      fixed++;
+    }
+  }
+  return fixed;
+}
+
 /**
  * 过滤消息，保留配对的 tool-call → tool-result 链条
  * 用于上下文压缩后保留完整的工具调用链路
@@ -283,5 +340,6 @@ module.exports = {
   validateMessagesPairing,
   validateToolCalls,
   validateAll,
+  normalizeToolOutputs,
   filterPairedMessages,
 };