@voltagent/core 1.1.18 → 1.1.19

package/dist/index.js

@@ -6580,184 +6580,6 @@ var BackgroundQueue = class {
   }
 };
 
-// src/agent/message-normalizer.ts
-var WORKING_MEMORY_TOOL_NAMES = /* @__PURE__ */ new Set([
-  "update_working_memory",
-  "get_working_memory",
-  "clear_working_memory"
-]);
-var isObject = /* @__PURE__ */ __name((value) => typeof value === "object" && value !== null, "isObject");
-var safeClone = /* @__PURE__ */ __name((value) => {
-  if (!isObject(value) && !Array.isArray(value)) {
-    return value;
-  }
-  const structuredCloneImpl = globalThis.structuredClone;
-  if (typeof structuredCloneImpl === "function") {
-    return structuredCloneImpl(value);
-  }
-  try {
-    return JSON.parse(JSON.stringify(value));
-  } catch (_error) {
-    if (Array.isArray(value)) {
-      return value.slice();
-    }
-    return { ...value };
-  }
-}, "safeClone");
-var normalizeText = /* @__PURE__ */ __name((part) => {
-  const text = typeof part.text === "string" ? part.text : "";
-  if (!text.trim()) {
-    return null;
-  }
-  const normalized = {
-    type: "text",
-    text
-  };
-  if (isObject(part.providerMetadata)) {
-    normalized.providerMetadata = safeClone(part.providerMetadata);
-  }
-  return normalized;
-}, "normalizeText");
-var normalizeReasoning = /* @__PURE__ */ __name((part) => {
-  const text = typeof part.text === "string" ? part.text : "";
-  if (!text.trim()) {
-    return null;
-  }
-  const normalized = {
-    type: "reasoning",
-    text
-  };
-  if (part.reasoningId) {
-    normalized.reasoningId = part.reasoningId;
-  }
-  if (part.reasoningConfidence !== void 0) {
-    normalized.reasoningConfidence = part.reasoningConfidence;
-  }
-  if (isObject(part.providerMetadata)) {
-    normalized.providerMetadata = safeClone(part.providerMetadata);
-  }
-  return normalized;
-}, "normalizeReasoning");
-var toolNameFromType = /* @__PURE__ */ __name((type) => {
-  if (typeof type !== "string") return void 0;
-  if (!type.startsWith("tool-")) return void 0;
-  return type.slice("tool-".length);
-}, "toolNameFromType");
-var isWorkingMemoryTool = /* @__PURE__ */ __name((part) => {
-  const toolName = toolNameFromType(part.type);
-  if (!toolName) return false;
-  return WORKING_MEMORY_TOOL_NAMES.has(toolName);
-}, "isWorkingMemoryTool");
-var normalizeToolPart = /* @__PURE__ */ __name((part) => {
-  if (isWorkingMemoryTool(part)) {
-    return null;
-  }
-  const toolName = toolNameFromType(part.type);
-  if (!toolName) {
-    return safeClone(part);
-  }
-  const normalized = {
-    type: `tool-${toolName}`
-  };
-  if (part.toolCallId) normalized.toolCallId = part.toolCallId;
-  if (part.state) normalized.state = part.state;
-  if (part.input !== void 0) normalized.input = safeClone(part.input);
-  if (part.output !== void 0) normalized.output = safeClone(part.output);
-  if (part.providerExecuted !== void 0) normalized.providerExecuted = part.providerExecuted;
-  if (part.isError !== void 0) normalized.isError = part.isError;
-  if (part.errorText !== void 0) normalized.errorText = part.errorText;
-  return normalized;
-}, "normalizeToolPart");
-var normalizeGenericPart = /* @__PURE__ */ __name((part) => {
-  switch (part.type) {
-    case "text":
-      return normalizeText(part);
-    case "reasoning":
-      return normalizeReasoning(part);
-    case "step-start":
-      return { type: "step-start" };
-    case "file": {
-      if (!isObject(part) || !part.url) {
-        return null;
-      }
-      const cloned = safeClone(part);
-      if (cloned.providerMetadata) {
-        cloned.providerMetadata = safeClone(cloned.providerMetadata);
-      }
-      return cloned;
-    }
-    default:
-      if (typeof part.type === "string" && part.type.startsWith("tool-")) {
-        return normalizeToolPart(part);
-      }
-      return safeClone(part);
-  }
-}, "normalizeGenericPart");
-var pruneEmptyToolRuns = /* @__PURE__ */ __name((parts) => {
-  const cleaned = [];
-  for (const part of parts) {
-    if (typeof part.type === "string" && part.type.startsWith("tool-")) {
-      const hasPendingState = part.state === "input-available";
-      const hasResult = part.state === "output-available" || part.output !== void 0;
-      if (!hasPendingState && !hasResult && part.input == null) {
-        continue;
-      }
-    }
-    cleaned.push(part);
-  }
-  return cleaned;
-}, "pruneEmptyToolRuns");
-var collapseRedundantStepStarts = /* @__PURE__ */ __name((parts) => {
-  const result = [];
-  for (const part of parts) {
-    if (part.type === "step-start") {
-      const prev = result.at(-1);
-      if (!prev || prev.type === "step-start") {
-        continue;
-      }
-    }
-    result.push(part);
-  }
-  return result;
-}, "collapseRedundantStepStarts");
-var sanitizeMessage = /* @__PURE__ */ __name((message) => {
-  const sanitizedParts = [];
-  for (const part of message.parts) {
-    const normalized = normalizeGenericPart(part);
-    if (!normalized) {
-      continue;
-    }
-    sanitizedParts.push(normalized);
-  }
-  const pruned = collapseRedundantStepStarts(pruneEmptyToolRuns(sanitizedParts));
-  const effectiveParts = pruned.filter((part) => {
-    if (part.type === "text") {
-      return typeof part.text === "string" && part.text.trim().length > 0;
-    }
-    if (part.type === "reasoning") {
-      return typeof part.text === "string" && part.text.trim().length > 0;
-    }
-    if (typeof part.type === "string" && part.type.startsWith("tool-")) {
-      return Boolean(part.toolCallId);
-    }
-    if (part.type === "file") {
-      return Boolean(part.url);
-    }
-    return true;
-  });
-  if (!effectiveParts.length) {
-    return null;
-  }
-  const clonedMetadata = isObject(message.metadata) ? safeClone(message.metadata) : message.metadata;
-  return {
-    ...message,
-    parts: effectiveParts,
-    ...clonedMetadata ? { metadata: clonedMetadata } : {}
-  };
-}, "sanitizeMessage");
-var sanitizeMessagesForPersistence = /* @__PURE__ */ __name((messages) => messages.map((message) => sanitizeMessage(message)).filter((message) => Boolean(message)), "sanitizeMessagesForPersistence");
-var sanitizeMessageForPersistence = /* @__PURE__ */ __name((message) => sanitizeMessage(message), "sanitizeMessageForPersistence");
-
 // src/memory/manager/memory-manager.ts
 var MemoryManager = class {
   static {
@@ -6820,19 +6642,10 @@ var MemoryManager = class {
     const memoryLogger = context5.logger.child({
       operation: "write"
     });
-    const sanitizedMessage = sanitizeMessageForPersistence(message);
-    if (!sanitizedMessage) {
-      memoryLogger.debug("[Memory] Skipped persisting empty message after sanitization", {
-        event: LogEvents.MEMORY_OPERATION_COMPLETED,
-        operation: "write",
-        reason: "sanitized-empty"
-      });
-      return;
-    }
     const trace8 = context5.traceContext;
-    const spanInput = { userId, conversationId, message: sanitizedMessage };
+    const spanInput = { userId, conversationId, message };
     const writeSpan = trace8.createChildSpan("memory.write", "memory", {
-      label: sanitizedMessage.role === "user" ? "Persist User Message" : "Persist Assistant Message",
+      label: message.role === "user" ? "Persist User Message" : "Persist Assistant Message",
       attributes: {
         "memory.operation": "write",
         input: (0, import_internal4.safeStringify)(spanInput)
@@ -6851,14 +6664,9 @@ var MemoryManager = class {
               metadata: {}
             });
           }
-          await this.conversationMemory?.saveMessageWithContext(
-            sanitizedMessage,
-            userId,
-            conversationId,
-            {
-              logger: memoryLogger
-            }
-          );
+          await this.conversationMemory?.saveMessageWithContext(message, userId, conversationId, {
+            logger: memoryLogger
+          });
         }
       });
       trace8.endChildSpan(writeSpan, "completed", {
@@ -9334,24 +9142,14 @@ var MemoryPersistQueue = class {
       this.logger?.debug?.("[MemoryPersistQueue] nothing-to-persist", payload2);
       return;
     }
-    const sanitized = sanitizeMessagesForPersistence(pending);
-    if (sanitized.length === 0) {
-      this.logger?.debug?.("[MemoryPersistQueue] sanitized-all", {
-        conversationId: oc.conversationId,
-        userId: oc.userId,
-        dropped: pending.length
-      });
-      return;
-    }
     const payload = {
       conversationId: oc.conversationId,
       userId: oc.userId,
-      count: sanitized.length,
-      dropped: pending.length - sanitized.length,
-      ids: sanitized.map((msg) => msg.id)
+      count: pending.length,
+      ids: pending.map((msg) => msg.id)
     };
     this.logger?.debug?.("[MemoryPersistQueue] persisting", payload);
-    for (const message of sanitized) {
+    for (const message of pending) {
       try {
         await this.memoryManager.saveMessage(oc, message, oc.userId, oc.conversationId);
       } catch (error) {
@@ -9393,6 +9191,172 @@ var MemoryPersistQueue = class {
   }
 };
 
+// src/agent/message-normalizer.ts
+var WORKING_MEMORY_TOOL_NAMES = /* @__PURE__ */ new Set([
+  "update_working_memory",
+  "get_working_memory",
+  "clear_working_memory"
+]);
+var isObject = /* @__PURE__ */ __name((value) => typeof value === "object" && value !== null, "isObject");
+var safeClone = /* @__PURE__ */ __name((value) => {
+  if (!isObject(value) && !Array.isArray(value)) {
+    return value;
+  }
+  const structuredCloneImpl = globalThis.structuredClone;
+  if (typeof structuredCloneImpl === "function") {
+    return structuredCloneImpl(value);
+  }
+  try {
+    return JSON.parse(JSON.stringify(value));
+  } catch (_error) {
+    if (Array.isArray(value)) {
+      return value.slice();
+    }
+    return { ...value };
+  }
+}, "safeClone");
+var normalizeText = /* @__PURE__ */ __name((part) => {
+  const text = typeof part.text === "string" ? part.text : "";
+  if (!text.trim()) {
+    return null;
+  }
+  return {
+    type: "text",
+    text
+  };
+}, "normalizeText");
+var normalizeReasoning = /* @__PURE__ */ __name((part) => {
+  const text = typeof part.text === "string" ? part.text : "";
+  if (!text.trim()) {
+    return null;
+  }
+  const normalized = {
+    type: "reasoning",
+    text
+  };
+  if (part.reasoningId) {
+    normalized.reasoningId = part.reasoningId;
+  }
+  if (part.reasoningConfidence !== void 0) {
+    normalized.reasoningConfidence = part.reasoningConfidence;
+  }
+  return normalized;
+}, "normalizeReasoning");
+var toolNameFromType = /* @__PURE__ */ __name((type) => {
+  if (typeof type !== "string") return void 0;
+  if (!type.startsWith("tool-")) return void 0;
+  return type.slice("tool-".length);
+}, "toolNameFromType");
+var isWorkingMemoryTool = /* @__PURE__ */ __name((part) => {
+  const toolName = toolNameFromType(part.type);
+  if (!toolName) return false;
+  return WORKING_MEMORY_TOOL_NAMES.has(toolName);
+}, "isWorkingMemoryTool");
+var normalizeToolPart = /* @__PURE__ */ __name((part) => {
+  if (isWorkingMemoryTool(part)) {
+    return null;
+  }
+  const toolName = toolNameFromType(part.type);
+  if (!toolName) {
+    return safeClone(part);
+  }
+  const normalized = {
+    type: `tool-${toolName}`
+  };
+  if (part.toolCallId) normalized.toolCallId = part.toolCallId;
+  if (part.state) normalized.state = part.state;
+  if (part.input !== void 0) normalized.input = safeClone(part.input);
+  if (part.output !== void 0) normalized.output = safeClone(part.output);
+  if (part.providerExecuted !== void 0) normalized.providerExecuted = part.providerExecuted;
+  if (part.isError !== void 0) normalized.isError = part.isError;
+  if (part.errorText !== void 0) normalized.errorText = part.errorText;
+  return normalized;
+}, "normalizeToolPart");
+var sanitizeMessagesForModel = /* @__PURE__ */ __name((messages) => messages.map((message) => sanitizeMessageForModel(message)).filter((message) => Boolean(message)), "sanitizeMessagesForModel");
+var sanitizeMessageForModel = /* @__PURE__ */ __name((message) => {
+  const sanitizedParts = [];
+  for (const part of message.parts) {
+    const normalized = normalizeGenericPart(part);
+    if (!normalized) {
+      continue;
+    }
+    sanitizedParts.push(normalized);
+  }
+  const pruned = collapseRedundantStepStarts(pruneEmptyToolRuns(sanitizedParts));
+  const effectiveParts = pruned.filter((part) => {
+    if (part.type === "text") {
+      return typeof part.text === "string" && part.text.trim().length > 0;
+    }
+    if (part.type === "reasoning") {
+      return typeof part.text === "string" && part.text.trim().length > 0;
+    }
+    if (typeof part.type === "string" && part.type.startsWith("tool-")) {
+      return Boolean(part.toolCallId);
+    }
+    if (part.type === "file") {
+      return Boolean(part.url);
+    }
+    return true;
+  });
+  if (!effectiveParts.length) {
+    return null;
+  }
+  return {
+    ...message,
+    parts: effectiveParts,
+    ...message.metadata ? { metadata: safeClone(message.metadata) } : {}
+  };
+}, "sanitizeMessageForModel");
+var normalizeGenericPart = /* @__PURE__ */ __name((part) => {
+  switch (part.type) {
+    case "text":
+      return normalizeText(part);
+    case "reasoning":
+      return normalizeReasoning(part);
+    case "step-start":
+      return { type: "step-start" };
+    case "file": {
+      if (!isObject(part) || !part.url) {
+        return null;
+      }
+      const cloned = safeClone(part);
+      return cloned;
+    }
+    default:
+      if (typeof part.type === "string" && part.type.startsWith("tool-")) {
+        return normalizeToolPart(part);
+      }
+      return safeClone(part);
+  }
+}, "normalizeGenericPart");
+var pruneEmptyToolRuns = /* @__PURE__ */ __name((parts) => {
+  const cleaned = [];
+  for (const part of parts) {
+    if (typeof part.type === "string" && part.type.startsWith("tool-")) {
+      const hasPendingState = part.state === "input-available";
+      const hasResult = part.state === "output-available" || part.output !== void 0;
+      if (!hasPendingState && !hasResult && part.input == null) {
+        continue;
+      }
+    }
+    cleaned.push(part);
+  }
+  return cleaned;
+}, "pruneEmptyToolRuns");
+var collapseRedundantStepStarts = /* @__PURE__ */ __name((parts) => {
+  const result = [];
+  for (const part of parts) {
+    if (part.type === "step-start") {
+      const prev = result.at(-1);
+      if (!prev || prev.type === "step-start") {
+        continue;
+      }
+    }
+    result.push(part);
+  }
+  return result;
+}, "collapseRedundantStepStarts");
+
 // src/agent/subagent/index.ts
 var import_utils11 = require("@voltagent/internal/utils");
 var import_zod2 = require("zod");
@@ -10146,7 +10110,11 @@ var Agent = class {
     return await oc.traceContext.withSpan(rootSpan, async () => {
       const buffer = this.getConversationBuffer(oc);
       const persistQueue = this.getMemoryPersistQueue(oc);
-      const { messages, model, tools, maxSteps } = await this.prepareExecution(input, oc, options);
+      const { messages, uiMessages, model, tools, maxSteps } = await this.prepareExecution(
+        input,
+        oc,
+        options
+      );
       const modelName = this.getModelName();
       const contextLimit = options?.contextLimit;
       addModelAttributesToSpan(
@@ -10161,6 +10129,7 @@ var Agent = class {
         rootSpan.setAttribute("agent.context", (0, import_utils12.safeStringify)(contextMap));
       }
       rootSpan.setAttribute("agent.messages", (0, import_utils12.safeStringify)(messages));
+      rootSpan.setAttribute("agent.messages.ui", (0, import_utils12.safeStringify)(uiMessages));
       const agentState = this.getFullState();
       rootSpan.setAttribute("agent.stateSnapshot", (0, import_utils12.safeStringify)(agentState));
       methodLogger.debug(
@@ -10285,7 +10254,11 @@ var Agent = class {
       const methodLogger = oc.logger;
       const buffer = this.getConversationBuffer(oc);
       const persistQueue = this.getMemoryPersistQueue(oc);
-      const { messages, model, tools, maxSteps } = await this.prepareExecution(input, oc, options);
+      const { messages, uiMessages, model, tools, maxSteps } = await this.prepareExecution(
+        input,
+        oc,
+        options
+      );
       const modelName = this.getModelName();
       const contextLimit = options?.contextLimit;
       if (oc.traceContext) {
@@ -10302,6 +10275,7 @@ var Agent = class {
           rootSpan2.setAttribute("agent.context", (0, import_utils12.safeStringify)(contextMap));
         }
         rootSpan2.setAttribute("agent.messages", (0, import_utils12.safeStringify)(messages));
+        rootSpan2.setAttribute("agent.messages.ui", (0, import_utils12.safeStringify)(uiMessages));
         const agentState = this.getFullState();
         rootSpan2.setAttribute("agent.stateSnapshot", (0, import_utils12.safeStringify)(agentState));
       }
@@ -10500,7 +10474,7 @@ var Agent = class {
     const methodLogger = oc.logger;
     const rootSpan = oc.traceContext.getRootSpan();
     return await oc.traceContext.withSpan(rootSpan, async () => {
-      const { messages, model } = await this.prepareExecution(input, oc, options);
+      const { messages, uiMessages, model } = await this.prepareExecution(input, oc, options);
       const modelName = this.getModelName();
       const schemaName = schema.description || "unknown";
       addModelAttributesToSpan(
@@ -10515,6 +10489,7 @@ var Agent = class {
         rootSpan.setAttribute("agent.context", (0, import_utils12.safeStringify)(contextMap));
       }
       rootSpan.setAttribute("agent.messages", (0, import_utils12.safeStringify)(messages));
+      rootSpan.setAttribute("agent.messages.ui", (0, import_utils12.safeStringify)(uiMessages));
       const agentState = this.getFullState();
       rootSpan.setAttribute("agent.stateSnapshot", (0, import_utils12.safeStringify)(agentState));
       methodLogger.debug(
@@ -10636,7 +10611,7 @@ var Agent = class {
     const rootSpan = oc.traceContext.getRootSpan();
     return await oc.traceContext.withSpan(rootSpan, async () => {
       const methodLogger = oc.logger;
-      const { messages, model } = await this.prepareExecution(input, oc, options);
+      const { messages, uiMessages, model } = await this.prepareExecution(input, oc, options);
       const modelName = this.getModelName();
       const schemaName = schema.description || "unknown";
       addModelAttributesToSpan(
@@ -10651,6 +10626,7 @@ var Agent = class {
         rootSpan.setAttribute("agent.context", (0, import_utils12.safeStringify)(contextMap));
       }
       rootSpan.setAttribute("agent.messages", (0, import_utils12.safeStringify)(messages));
+      rootSpan.setAttribute("agent.messages.ui", (0, import_utils12.safeStringify)(uiMessages));
       const agentState = this.getFullState();
       rootSpan.setAttribute("agent.stateSnapshot", (0, import_utils12.safeStringify)(agentState));
       methodLogger.debug(
@@ -10811,13 +10787,20 @@ var Agent = class {
     const agentToolList = await this.resolveValue(this.tools, oc);
     const buffer = this.getConversationBuffer(oc);
     const uiMessages = await this.prepareMessages(input, oc, options, buffer);
-    const messages = (0, import_ai.convertToModelMessages)(uiMessages);
+    const sanitizedUIMessages = sanitizeMessagesForModel(uiMessages);
+    const messages = (0, import_ai.convertToModelMessages)(sanitizedUIMessages);
     const maxSteps = options?.maxSteps ?? this.calculateMaxSteps();
     const agentToolsArray = Array.isArray(agentToolList) ? agentToolList : [];
     const optionToolsArray = options?.tools || [];
     const mergedTools = [...agentToolsArray, ...optionToolsArray];
     const tools = await this.prepareTools(mergedTools, oc, maxSteps, options);
-    return { messages, model, tools, maxSteps };
+    return {
+      messages,
+      uiMessages: sanitizedUIMessages,
+      model,
+      tools,
+      maxSteps
+    };
   }
   /**
    * Create execution context
@@ -11122,10 +11105,11 @@ var Agent = class {
             buffer.ingestUIMessages(result.messages, true);
             return result.messages;
           });
+          const retrievedMessagesCount = Array.isArray(memoryResult) ? memoryResult.length : 0;
           traceContext.endChildSpan(memoryReadSpan, "completed", {
             output: memoryResult,
             attributes: {
-              "memory.message_count": Array.isArray(memoryResult) ? memoryResult.length : 0
+              "memory.message_count": retrievedMessagesCount
             }
           });
           if (isSemanticSearch && !oc.conversationId) {