@mastra/core 1.13.3-alpha.1 → 1.14.0-alpha.2

package/dist/{chunk-E2Y2HLCU.cjs → chunk-W2LUCUUP.cjs}

@@ -1578,7 +1578,7 @@ function aiV5ModelMessageToV2PromptMessage(modelMessage) {
         break;
       }
       case "tool-result": {
-        if (role === `assistant` || role === `user`) {
+        if (role === `user`) {
           throw new Error(incompatibleMessage);
         }
         roleContent[role].push({
@@ -1823,10 +1823,7 @@ function sanitizeV5UIMessages(messages, filterIncompleteToolCalls = false) {
       }
       if (!chunkVBVOKSJK_cjs.isToolUIPart(p)) return true;
       if (filterIncompleteToolCalls) {
-        if (p.state === "output-available" || p.state === "output-error") {
-          if (p.providerExecuted) return false;
-          return true;
-        }
+        if (p.state === "output-available" || p.state === "output-error") return true;
         if (p.state === "input-available" && p.providerExecuted) return true;
         return false;
       }
@@ -3709,6 +3706,43 @@ var MessageList = class {
   isNewMessage(messageOrId) {
     return this.stateManager.isNewMessage(messageOrId);
   }
+  /**
+   * Replace a tool-invocation part matching the given toolCallId with the
+   * provided result part. Walks backwards through messages to find the match.
+   * If the message was already persisted (e.g. as a memory message), it is
+   * moved to the response source so it will be re-saved.
+   *
+   * @returns true if the tool call was found and updated, false otherwise.
+   */
+  updateToolInvocation(inputPart) {
+    if (!inputPart.toolInvocation?.toolCallId) {
+      return false;
+    }
+    const toolCallId = inputPart.toolInvocation.toolCallId;
+    for (let m = this.messages.length - 1; m >= 0; m--) {
+      const msg = this.messages[m];
+      if (msg.role !== "assistant" || !msg.content?.parts) continue;
+      for (let i = 0; i < msg.content.parts.length; i++) {
+        const part = msg.content.parts[i];
+        if (part?.type === "tool-invocation" && part.toolInvocation?.toolCallId === toolCallId) {
+          msg.content.parts[i] = {
+            ...inputPart,
+            toolInvocation: {
+              ...inputPart.toolInvocation,
+              args: part.toolInvocation.args
+            }
+          };
+          if (!this.stateManager.isResponseMessage(msg)) {
+            this.stateManager.removeMessage(msg);
+            this.stateManager.addToSource(msg, "response");
+          }
+          return true;
+        }
+      }
+    }
+    this.logger?.warn(`updateToolInvocation: no matching tool call found for toolCallId=${toolCallId}`);
+    return false;
+  }
   getSystemMessages(tag) {
     if (tag) {
       return this.taggedSystemMessages[tag] || [];
@@ -4022,5 +4056,5 @@ exports.getOpenAIReasoningItemId = getOpenAIReasoningItemId;
 exports.hasOpenAIReasoningItemId = hasOpenAIReasoningItemId;
 exports.messagesAreEqual = messagesAreEqual;
 exports.sanitizeToolName = sanitizeToolName;
-//# sourceMappingURL=chunk-E2Y2HLCU.cjs.map
-//# sourceMappingURL=chunk-E2Y2HLCU.cjs.map
+//# sourceMappingURL=chunk-W2LUCUUP.cjs.map
+//# sourceMappingURL=chunk-W2LUCUUP.cjs.map