@mastra/ai-sdk 0.0.0-fix-11640-persist-workflow-events-to-memory-20260106220027 → 0.0.0-fix-agent-network-schema-test-2-20260119190919

package/dist/index.js

@@ -3,7 +3,7 @@ import * as z4 from 'zod/v4';
 import { z } from 'zod/v4';
 import { ZodFirstPartyTypeKind } from 'zod/v3';
 import { convertFullStreamChunkToMastra, DefaultGeneratedFile, DefaultGeneratedFileWithType } from '@mastra/core/stream';
-import { TripWire, MessageList } from '@mastra/core/agent';
+import { TripWire, MessageList, aiV5ModelMessageToV2PromptMessage } from '@mastra/core/agent';
 import { RequestContext } from '@mastra/core/di';
 import { WorkingMemory, MessageHistory, SemanticRecall } from '@mastra/core/processors';
 
@@ -3973,6 +3973,10 @@ function transformNetwork(payload, bufferedNetworks, isNested) {
         }
       };
     }
+    case "network-object":
+    case "network-object-result": {
+      return null;
+    }
     default: {
       if (isAgentExecutionDataChunkType(payload)) {
         if (!("data" in payload.payload)) {
@@ -4101,7 +4105,7 @@ async function handleChatStream({
   sendReasoning = false,
   sendSources = false
 }) {
-  const { messages, resumeData, runId, requestContext, ...rest } = params;
+  const { messages, resumeData, runId, requestContext, trigger, ...rest } = params;
   if (resumeData && !runId) {
     throw new Error("runId is required when resumeData is provided");
   }
@@ -4112,20 +4116,24 @@ async function handleChatStream({
   if (!Array.isArray(messages)) {
     throw new Error("Messages must be an array of UIMessage objects");
   }
-  const mergedOptions = {
-    ...defaultOptions2,
-    ...rest,
-    ...runId && { runId },
-    requestContext: requestContext || defaultOptions2?.requestContext
-  };
-  const result = resumeData ? await agentObj.resumeStream(resumeData, mergedOptions) : await agentObj.stream(messages, mergedOptions);
   let lastMessageId;
-  if (messages.length) {
+  let messagesToSend = messages;
+  if (messages.length > 0) {
     const lastMessage = messages[messages.length - 1];
     if (lastMessage?.role === "assistant") {
       lastMessageId = lastMessage.id;
+      if (trigger === "regenerate-message") {
+        messagesToSend = messages.slice(0, -1);
+      }
     }
   }
+  const mergedOptions = {
+    ...defaultOptions2,
+    ...rest,
+    ...runId && { runId },
+    requestContext: requestContext || defaultOptions2?.requestContext
+  };
+  const result = resumeData ? await agentObj.resumeStream(resumeData, mergedOptions) : await agentObj.stream(messagesToSend, mergedOptions);
   return createUIMessageStream({
     originalMessages: messages,
     execute: async ({ writer }) => {
@@ -4632,7 +4640,7 @@ function createProcessorMiddleware(options) {
           }
         }
       }
-      const newPrompt = messageList.get.all.aiV5.prompt().map(MessageList.aiV5ModelMessageToV2PromptMessage);
+      const newPrompt = messageList.get.all.aiV5.prompt().map(aiV5ModelMessageToV2PromptMessage);
       return {
         ...params,
         prompt: newPrompt