@yeaft/webchat-agent 1.0.94 → 1.0.95

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@yeaft/webchat-agent",
-  "version": "1.0.94",
+  "version": "1.0.95",
   "description": "Remote worker agent for Yeaft Web Code Agent — connects the native Yeaft engine, CLI providers, and workbench tools",
   "main": "index.js",
   "type": "module",

package/yeaft/dream/segment-extract.js

@@ -168,9 +168,7 @@ function normalizeExtractedSegment({ item, scope, now }) {
   if (!body) return null;
   const kind = VALID_KINDS.has(String(item.kind || '')) ? String(item.kind) : 'context';
   const tags = Array.isArray(item.tags) ? item.tags.map(t => String(t).trim()).filter(Boolean) : [];
-  const sourceMessages = Array.isArray(item.sourceMessages)
-    ? item.sourceMessages.map(id => String(id).trim()).filter(Boolean)
-    : [];
+  const sourceMessages = normalizeSourceMessageIds(item.sourceMessages);
   return makeSegment({
     scope,
     kind,
@@ -182,6 +180,23 @@ function normalizeExtractedSegment({ item, scope, now }) {
   });
 }
 
+function normalizeSourceMessageIds(value) {
+  if (!Array.isArray(value)) return [];
+  const ids = [];
+  for (const item of value) {
+    if (typeof item === 'string' || typeof item === 'number') {
+      const id = String(item).trim();
+      if (id) ids.push(id);
+      continue;
+    }
+    if (item && typeof item === 'object') {
+      const id = String(item.id || item.messageId || item.uuid || '').trim();
+      if (id) ids.push(id);
+    }
+  }
+  return ids;
+}
+
 function buildRecentSegment({ scope, messages, now }) {
   const recentMessages = messages.slice(-RECENT_MESSAGE_COUNT);
   const body = [

package/yeaft/dream/session-wiring.js

@@ -313,6 +313,24 @@ function normalizeSessionJsonlMessage(row, sessionId) {
 function stringifyMessageBody(value) {
   if (typeof value === 'string') return value;
   if (value == null) return '';
+  if (Array.isArray(value)) return value.map(stringifyMessageBody).filter(Boolean).join('\n');
+  if (typeof value === 'object') {
+    if (typeof value.text === 'string') return value.text;
+    if (typeof value.content === 'string') return value.content;
+    if (typeof value.body === 'string') return value.body;
+    if (typeof value.value === 'string') return value.value;
+    if (Array.isArray(value.content)) return stringifyMessageBody(value.content);
+    if (Array.isArray(value.parts)) return stringifyMessageBody(value.parts);
+    if (Array.isArray(value.blocks)) return stringifyMessageBody(value.blocks);
+    if (value.type && (value.type === 'image' || value.type === 'image_url')) return '[image]';
+    if (value.type && (value.type === 'tool_use' || value.type === 'tool_call')) {
+      const name = value.name || value.toolName || value.function?.name || 'tool';
+      return `[tool call: ${name}]`;
+    }
+    if (value.type && (value.type === 'tool_result' || value.type === 'function_result')) {
+      return stringifyMessageBody(value.result ?? value.output ?? value.content ?? '[tool result]');
+    }
+  }
   try { return JSON.stringify(value); } catch { return String(value); }
 }
 

package/yeaft/memory/segment.js

@@ -93,9 +93,7 @@ export function makeSegment(raw) {
   const tags = Array.isArray(raw.tags)
     ? raw.tags.map(t => String(t).trim()).filter(Boolean)
     : [];
-  const sourceMessages = Array.isArray(raw.sourceMessages)
-    ? raw.sourceMessages.map(t => String(t).trim()).filter(Boolean)
-    : [];
+  const sourceMessages = normalizeSourceMessages(raw.sourceMessages);
 
   const now = new Date().toISOString();
   const createdAt = raw.createdAt || now;
@@ -105,6 +103,23 @@ export function makeSegment(raw) {
   return { id, scope, kind, tags, sourceMessages, createdAt, updatedAt, body };
 }
 
+function normalizeSourceMessages(value) {
+  if (!Array.isArray(value)) return [];
+  const out = [];
+  for (const item of value) {
+    if (typeof item === 'string' || typeof item === 'number') {
+      const id = String(item).trim();
+      if (id && id !== '[object Object]') out.push(id);
+      continue;
+    }
+    if (item && typeof item === 'object') {
+      const id = String(item.id || item.messageId || item.uuid || '').trim();
+      if (id) out.push(id);
+    }
+  }
+  return out;
+}
+
 /**
  * Parse a memory.md text into segments. Tolerant by design:
  *   - Empty / whitespace input → [].