@mastra/core 0.17.0-alpha.3 → 0.17.0-alpha.5

package/dist/{chunk-A6DPQCNR.js → chunk-XRJDA7BI.js}

@@ -1,4 +1,4 @@
-import { fetchWithRetry } from './chunk-UIOXAMCE.js';
+import { fetchWithRetry } from './chunk-PL3B3FZM.js';
 import { MastraError } from './chunk-MCOVMKIS.js';
 import { randomUUID } from 'crypto';
 import * as AIV4 from 'ai';
@@ -1059,7 +1059,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.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.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.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.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.convertToModelMessages(this.sanitizeV5UIMessages(stepUiMessages));
+          return modelMessages.flatMap(this.response.aiV5.stepContent);
+        }
         return this.response.aiV5.model().map(this.response.aiV5.stepContent).flat();
       },
       stepContent: (message) => {
@@ -2858,5 +2956,5 @@ function convertMessages(messages) {
 }
 
 export { DefaultGeneratedFile, DefaultGeneratedFileWithType, MessageList, convertMessages };
-//# sourceMappingURL=chunk-A6DPQCNR.js.map
-//# sourceMappingURL=chunk-A6DPQCNR.js.map
+//# sourceMappingURL=chunk-XRJDA7BI.js.map
+//# sourceMappingURL=chunk-XRJDA7BI.js.map