@mastra/core 0.16.4-alpha.2 → 0.17.0-alpha.4

package/dist/{chunk-DUUXEPE5.cjs → chunk-MRZHPP3G.cjs}

@@ -1,6 +1,6 @@
 'use strict';
 
-var chunkERRIEUC2_cjs = require('./chunk-ERRIEUC2.cjs');
+var chunkCP4TZ4HO_cjs = require('./chunk-CP4TZ4HO.cjs');
 var chunkC73WLCY3_cjs = require('./chunk-C73WLCY3.cjs');
 var crypto = require('crypto');
 var AIV4 = require('ai');
@@ -613,7 +613,7 @@ function convertDataContentToBase64String(content) {
 var downloadFromUrl = async ({ url, downloadRetries }) => {
   const urlText = url.toString();
   try {
-    const response = await chunkERRIEUC2_cjs.fetchWithRetry(
+    const response = await chunkCP4TZ4HO_cjs.fetchWithRetry(
       urlText,
       {
         method: "GET"
@@ -1082,7 +1082,105 @@ var MessageList = class _MessageList {
       model: () => this.aiV5UIMessagesToAIV5ModelMessages(this.response.aiV5.ui()).filter(
         (m) => m.role === `tool` || m.role === `assistant`
       ),
-      modelContent: () => {
+      modelContent: (stepNumber) => {
+        if (typeof stepNumber === "number") {
+          const uiMessages = this.response.aiV5.ui();
+          const uiMessagesParts = uiMessages.flatMap((item) => item.parts);
+          const stepBoundaries = [];
+          uiMessagesParts.forEach((part, index) => {
+            if (part.type === "step-start") {
+              stepBoundaries.push(index);
+            }
+          });
+          if (stepNumber === -1) {
+            const toolParts = uiMessagesParts.filter((p) => p.type?.startsWith("tool-"));
+            const hasStepStart = stepBoundaries.length > 0;
+            if (!hasStepStart && toolParts.length > 0) {
+              const lastToolPart = toolParts[toolParts.length - 1];
+              if (!lastToolPart) {
+                return [];
+              }
+              const lastToolIndex = uiMessagesParts.indexOf(lastToolPart);
+              const previousToolPart = toolParts[toolParts.length - 2];
+              const previousToolIndex = previousToolPart ? uiMessagesParts.indexOf(previousToolPart) : -1;
+              const startIndex2 = previousToolIndex + 1;
+              const stepParts3 = uiMessagesParts.slice(startIndex2, lastToolIndex + 1);
+              const stepUiMessages3 = [
+                {
+                  id: "last-step",
+                  role: "assistant",
+                  parts: stepParts3
+                }
+              ];
+              const modelMessages3 = AIV5__namespace.convertToModelMessages(this.sanitizeV5UIMessages(stepUiMessages3));
+              return modelMessages3.flatMap(this.response.aiV5.stepContent);
+            }
+            const totalSteps = stepBoundaries.length + 1;
+            if (totalSteps === 1 && !hasStepStart) {
+              const stepUiMessages3 = [
+                {
+                  id: "last-step",
+                  role: "assistant",
+                  parts: uiMessagesParts
+                }
+              ];
+              const modelMessages3 = AIV5__namespace.convertToModelMessages(this.sanitizeV5UIMessages(stepUiMessages3));
+              return modelMessages3.flatMap(this.response.aiV5.stepContent);
+            }
+            const lastStepStart = stepBoundaries[stepBoundaries.length - 1];
+            if (lastStepStart === void 0) {
+              return [];
+            }
+            const stepParts2 = uiMessagesParts.slice(lastStepStart + 1);
+            if (stepParts2.length === 0) {
+              return [];
+            }
+            const stepUiMessages2 = [
+              {
+                id: "last-step",
+                role: "assistant",
+                parts: stepParts2
+              }
+            ];
+            const modelMessages2 = AIV5__namespace.convertToModelMessages(this.sanitizeV5UIMessages(stepUiMessages2));
+            return modelMessages2.flatMap(this.response.aiV5.stepContent);
+          }
+          if (stepNumber === 1) {
+            const firstStepStart = stepBoundaries[0] ?? uiMessagesParts.length;
+            if (firstStepStart === 0) {
+              return [];
+            }
+            const stepParts2 = uiMessagesParts.slice(0, firstStepStart);
+            const stepUiMessages2 = [
+              {
+                id: "step-1",
+                role: "assistant",
+                parts: stepParts2
+              }
+            ];
+            const modelMessages2 = AIV5__namespace.convertToModelMessages(this.sanitizeV5UIMessages(stepUiMessages2));
+            return modelMessages2.flatMap(this.response.aiV5.stepContent);
+          }
+          const stepIndex = stepNumber - 2;
+          if (stepIndex < 0 || stepIndex >= stepBoundaries.length) {
+            return [];
+          }
+          const startIndex = (stepBoundaries[stepIndex] ?? 0) + 1;
+          const endIndex = stepBoundaries[stepIndex + 1] ?? uiMessagesParts.length;
+          if (startIndex >= endIndex) {
+            return [];
+          }
+          const stepParts = uiMessagesParts.slice(startIndex, endIndex);
+          const stepUiMessages = [
+            {
+              id: `step-${stepNumber}`,
+              role: "assistant",
+              parts: stepParts
+            }
+          ];
+          const modelMessages = AIV5__namespace.convertToModelMessages(this.sanitizeV5UIMessages(stepUiMessages));
+          return modelMessages.flatMap(this.response.aiV5.stepContent);
+        }
         return this.response.aiV5.model().map(this.response.aiV5.stepContent).flat();
       },
       stepContent: (message) => {
@@ -2884,5 +2982,5 @@ exports.DefaultGeneratedFile = DefaultGeneratedFile;
 exports.DefaultGeneratedFileWithType = DefaultGeneratedFileWithType;
 exports.MessageList = MessageList;
 exports.convertMessages = convertMessages;
-//# sourceMappingURL=chunk-DUUXEPE5.cjs.map
-//# sourceMappingURL=chunk-DUUXEPE5.cjs.map
+//# sourceMappingURL=chunk-MRZHPP3G.cjs.map
+//# sourceMappingURL=chunk-MRZHPP3G.cjs.map