@mastra/ai-sdk 1.6.0 → 1.6.1-alpha.0

package/dist/index.js

@@ -10,7 +10,7 @@ import { TripWire, MessageList, aiV5ModelMessageToV2PromptMessage } from '@mastr
 import { RequestContext } from '@mastra/core/di';
 import { WorkingMemory, MessageHistory, SemanticRecall } from '@mastra/core/processors';
 
-// ../../packages/_vendored/ai_v5/dist/chunk-I2JBSJEA.js
+// ../../packages/_vendored/ai_v5/dist/chunk-RGIGIB4E.js
 var marker = "vercel.ai.error";
 var symbol = Symbol.for(marker);
 var _a;
@@ -13900,10 +13900,21 @@ async function handleChatStream({
   if (resumeData && !runId) {
     throw new Error("runId is required when resumeData is provided");
   }
-  const agentObj = agentVersion ? await mastra.getAgentById(agentId, agentVersion) : mastra.getAgentById(agentId);
-  if (!agentObj) {
+  const baseAgent = mastra.getAgentById(agentId);
+  if (!baseAgent) {
     throw new Error(`Agent ${agentId} not found`);
   }
+  let agentObj = baseAgent;
+  const editorAgent = mastra.getEditor?.()?.agent;
+  if (editorAgent) {
+    agentObj = await editorAgent.applyStoredOverrides(
+      baseAgent,
+      agentVersion ?? { status: "published" },
+      requestContext
+    );
+  } else if (agentVersion) {
+    agentObj = await mastra.getAgentById(agentId, agentVersion);
+  }
   if (!Array.isArray(messages)) {
     throw new Error("Messages must be an array of UIMessage objects");
   }