@mastra/ai-sdk 1.0.0-beta.11 → 1.0.0-beta.13

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';
 
@@ -218,7 +218,7 @@ function secureJsonParse(text2) {
     Error.stackTraceLimit = stackTraceLimit;
   }
 }
-var validatorSymbol = Symbol.for("vercel.ai.validator");
+var validatorSymbol = /* @__PURE__ */ Symbol.for("vercel.ai.validator");
 function validator(validate) {
   return { [validatorSymbol]: true, validate };
 }
@@ -301,7 +301,7 @@ var getRelativePath = (pathA, pathB) => {
   }
   return [(pathA.length - i).toString(), ...pathB.slice(i)].join("/");
 };
-var ignoreOverride = Symbol(
+var ignoreOverride = /* @__PURE__ */ Symbol(
   "Let zodToJsonSchema decide on which parser to use"
 );
 var defaultOptions = {
@@ -1421,7 +1421,7 @@ function zodSchema(zodSchema2, options) {
     return zod3Schema(zodSchema2);
   }
 }
-var schemaSymbol = Symbol.for("vercel.ai.schema");
+var schemaSymbol = /* @__PURE__ */ Symbol.for("vercel.ai.schema");
 function jsonSchema(jsonSchema2, {
   validate
 } = {}) {
@@ -2978,7 +2978,8 @@ function convertMastraChunkToAISDKv5({
           runId: chunk.runId,
           toolCallId: chunk.payload.toolCallId,
           toolName: chunk.payload.toolName,
-          args: chunk.payload.args
+          args: chunk.payload.args,
+          resumeSchema: chunk.payload.resumeSchema
         }
       };
     case "tool-call-suspended":
@@ -2989,7 +2990,8 @@ function convertMastraChunkToAISDKv5({
           runId: chunk.runId,
           toolCallId: chunk.payload.toolCallId,
           toolName: chunk.payload.toolName,
-          suspendPayload: chunk.payload.suspendPayload
+          suspendPayload: chunk.payload.suspendPayload,
+          resumeSchema: chunk.payload.resumeSchema
         }
       };
     case "tool-call-input-streaming-start":
@@ -3255,7 +3257,8 @@ function convertFullStreamChunkToUIMessageStream({
  ${JSON.stringify(part)}`
           );
         }
-        return part.output;
+        const { type, data, id } = part.output;
+        return { type, data, ...id !== void 0 && { id } };
       }
       return;
     }
@@ -3317,7 +3320,8 @@ function convertFullStreamChunkToUIMessageStream({
  ${JSON.stringify(part)}`
           );
         }
-        return part;
+        const { type, data, id } = part;
+        return { type, data, ...id !== void 0 && { id } };
       }
       return;
     }
@@ -3325,7 +3329,7 @@ function convertFullStreamChunkToUIMessageStream({
 }
 
 // src/transformers.ts
-var PRIMITIVE_CACHE_SYMBOL = Symbol("primitive-cache");
+var PRIMITIVE_CACHE_SYMBOL = /* @__PURE__ */ Symbol("primitive-cache");
 function WorkflowStreamToAISDKTransformer({
   includeTextStreamParts
 } = {}) {
@@ -3362,7 +3366,15 @@ function AgentNetworkToAISDKTransformer() {
     },
     transform(chunk, controller) {
       const transformed = transformNetwork(chunk, bufferedNetworks);
-      if (transformed) controller.enqueue(transformed);
+      if (transformed) {
+        if (Array.isArray(transformed)) {
+          for (const item of transformed) {
+            controller.enqueue(item);
+          }
+        } else {
+          controller.enqueue(transformed);
+        }
+      }
     }
   });
 }
@@ -3675,7 +3687,8 @@ function transformWorkflow(payload, bufferedWorkflows, isNested, includeTextStre
  ${JSON.stringify(output)}`
           );
         }
-        return output;
+        const { type, data, id } = output;
+        return { type, data, ...id !== void 0 && { id } };
       }
       return null;
     }
@@ -3687,7 +3700,12 @@ function transformWorkflow(payload, bufferedWorkflows, isNested, includeTextStre
  ${JSON.stringify(payload)}`
           );
         }
-        return payload;
+        const { type, data, id } = payload;
+        return {
+          type,
+          data,
+          ...id !== void 0 && { id }
+        };
       }
       return null;
     }
@@ -3701,7 +3719,8 @@ function transformNetwork(payload, bufferedNetworks, isNested) {
           name: payload.payload.networkId,
           steps: [],
           usage: null,
-          output: null
+          output: null,
+          hasEmittedText: false
         });
       }
       const current = bufferedNetworks.get(payload.runId);
@@ -3736,6 +3755,7 @@ function transformNetwork(payload, bufferedNetworks, isNested) {
     case "routing-agent-text-start": {
       const current = bufferedNetworks.get(payload.runId);
       if (!current) return null;
+      current.hasEmittedText = true;
       return {
         type: "text-start",
         id: payload.runId
@@ -3744,6 +3764,7 @@ function transformNetwork(payload, bufferedNetworks, isNested) {
     case "routing-agent-text-delta": {
       const current = bufferedNetworks.get(payload.runId);
       if (!current) return null;
+      current.hasEmittedText = true;
       return {
         type: "text-delta",
         id: payload.runId,
@@ -3918,15 +3939,25 @@ function transformNetwork(payload, bufferedNetworks, isNested) {
     case "network-execution-event-step-finish": {
       const current = bufferedNetworks.get(payload.runId);
       if (!current) return null;
-      return {
+      const resultText = payload.payload?.result;
+      const dataNetworkChunk = {
         type: isNested ? "data-tool-network" : "data-network",
         id: payload.runId,
         data: {
           ...current,
           status: "finished",
-          output: payload.payload?.result ?? current.output
+          output: resultText ?? current.output
         }
       };
+      if (!current.hasEmittedText && resultText && typeof resultText === "string" && resultText.length > 0) {
+        current.hasEmittedText = true;
+        return [
+          { type: "text-start", id: payload.runId },
+          { type: "text-delta", id: payload.runId, delta: resultText },
+          dataNetworkChunk
+        ];
+      }
+      return dataNetworkChunk;
     }
     case "network-execution-event-finish": {
       const current = bufferedNetworks.get(payload.runId);
@@ -3950,8 +3981,8 @@ function transformNetwork(payload, bufferedNetworks, isNested) {
  ${JSON.stringify(payload)}`
           );
         }
-        const { type, data } = payload.payload;
-        return { type, data };
+        const { type, data, id } = payload.payload;
+        return { type, data, ...id !== void 0 && { id } };
       }
       if (isWorkflowExecutionDataChunkType(payload)) {
         if (!("data" in payload.payload)) {
@@ -3960,8 +3991,8 @@ function transformNetwork(payload, bufferedNetworks, isNested) {
  ${JSON.stringify(payload)}`
           );
         }
-        const { type, data } = payload.payload;
-        return { type, data };
+        const { type, data, id } = payload.payload;
+        return { type, data, ...id !== void 0 && { id } };
       }
       if (payload.type.startsWith("agent-execution-event-")) {
         const stepId = payload.payload.runId;
@@ -4021,8 +4052,8 @@ function transformNetwork(payload, bufferedNetworks, isNested) {
  ${JSON.stringify(payload)}`
           );
         }
-        const { type, data } = payload;
-        return { type, data };
+        const { type, data, id } = payload;
+        return { type, data, ...id !== void 0 && { id } };
       }
       return null;
     }
@@ -4070,7 +4101,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");
   }
@@ -4081,20 +4112,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 }) => {
@@ -4601,7 +4636,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