@mastra/ai-sdk 0.0.0-feat-mcp-embedded-docs-tools-clean-20260105080340 → 0.0.0-feat-mcp-embedded-docs-tools-clean-20260108085651

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
 } = {}) {
@@ -3329,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
 } = {}) {
@@ -4101,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");
   }
@@ -4112,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 }) => {
@@ -4632,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