@mastra/ai-sdk 0.0.0-fix-issue-10434-concurrent-write-corruption-20251124213939 → 0.0.0-fix-backport-setserver-20251201151948

package/dist/index.js

@@ -8,6 +8,31 @@ import { DefaultGeneratedFile, DefaultGeneratedFileWithType } from '@mastra/core
 var isDataChunkType = (chunk) => {
   return chunk && typeof chunk === "object" && "type" in chunk && chunk.type?.startsWith("data-");
 };
+var isMastraTextStreamChunk = (chunk) => {
+  return chunk && typeof chunk === "object" && "type" in chunk && typeof chunk.type === "string" && [
+    "text-start",
+    "text-delta",
+    "text-end",
+    "reasoning-start",
+    "reasoning-delta",
+    "reasoning-end",
+    "file",
+    "source",
+    "tool-input-start",
+    "tool-input-delta",
+    "tool-call",
+    "tool-result",
+    "tool-error",
+    "error",
+    "start-step",
+    "finish-step",
+    "start",
+    "finish",
+    "abort",
+    "tool-input-end",
+    "raw"
+  ].includes(chunk.type);
+};
 function safeParseErrorObject(obj) {
   if (typeof obj !== "object" || obj === null) {
     return String(obj);
@@ -489,7 +514,10 @@ function convertFullStreamChunkToUIMessageStream({
 }
 
 // src/transformers.ts
-function WorkflowStreamToAISDKTransformer() {
+var PRIMITIVE_CACHE_SYMBOL = Symbol("primitive-cache");
+function WorkflowStreamToAISDKTransformer({
+  includeTextStreamParts
+} = {}) {
   const bufferedWorkflows = /* @__PURE__ */ new Map();
   return new TransformStream({
     start(controller) {
@@ -503,7 +531,7 @@ function WorkflowStreamToAISDKTransformer() {
       });
     },
     transform(chunk, controller) {
-      const transformed = transformWorkflow(chunk, bufferedWorkflows);
+      const transformed = transformWorkflow(chunk, bufferedWorkflows, false, includeTextStreamParts);
       if (transformed) controller.enqueue(transformed);
     }
   });
@@ -724,7 +752,7 @@ function transformAgent(payload, bufferedSteps) {
   }
   return null;
 }
-function transformWorkflow(payload, bufferedWorkflows, isNested) {
+function transformWorkflow(payload, bufferedWorkflows, isNested, includeTextStreamParts) {
   switch (payload.type) {
     case "workflow-start":
       bufferedWorkflows.set(payload.runId, {
@@ -819,6 +847,16 @@ function transformWorkflow(payload, bufferedWorkflows, isNested) {
     }
     case "workflow-step-output": {
       const output = payload.payload.output;
+      if (includeTextStreamParts && output && isMastraTextStreamChunk(output)) {
+        const part = convertMastraChunkToAISDKv5({ chunk: output, mode: "stream" });
+        const transformedChunk = convertFullStreamChunkToUIMessageStream({
+          part,
+          onError(error) {
+            return safeParseErrorObject(error);
+          }
+        });
+        return transformedChunk;
+      }
       if (output && isDataChunkType(output)) {
         if (!("data" in output)) {
           throw new Error(
@@ -849,12 +887,29 @@ function transformNetwork(payload, bufferedNetworks, isNested) {
     case "routing-agent-start": {
       if (!bufferedNetworks.has(payload.runId)) {
         bufferedNetworks.set(payload.runId, {
-          name: payload.payload.agentId,
+          name: payload.payload.networkId,
           steps: [],
           usage: null,
           output: null
         });
       }
+      const current = bufferedNetworks.get(payload.runId);
+      current.steps.push({
+        id: payload.payload.runId,
+        name: payload.payload.agentId,
+        status: "running",
+        iteration: payload.payload.inputData.iteration,
+        input: {
+          task: payload.payload.inputData.task,
+          threadId: payload.payload.inputData.threadId,
+          threadResourceId: payload.payload.inputData.threadResourceId
+        },
+        output: "",
+        task: null,
+        suspendPayload: null,
+        resumePayload: null,
+        [PRIMITIVE_CACHE_SYMBOL]: /* @__PURE__ */ new Map()
+      });
       return {
         type: isNested ? "data-tool-network" : "data-network",
         id: payload.runId,
@@ -885,14 +940,19 @@ function transformNetwork(payload, bufferedNetworks, isNested) {
       };
     }
     case "agent-execution-start": {
-      const current = bufferedNetworks.get(payload.runId) || { name: "", steps: [], usage: null, output: null };
+      const current = bufferedNetworks.get(payload.runId);
+      if (!current) return null;
       current.steps.push({
+        id: payload.payload.runId,
         name: payload.payload.agentId,
         status: "running",
-        input: payload.payload.args || null,
+        iteration: payload.payload.args?.iteration ?? 0,
+        input: { prompt: payload.payload.args?.prompt ?? "" },
         output: null,
+        task: null,
         suspendPayload: null,
-        resumePayload: null
+        resumePayload: null,
+        [PRIMITIVE_CACHE_SYMBOL]: /* @__PURE__ */ new Map()
       });
       bufferedNetworks.set(payload.runId, current);
       return {
@@ -905,14 +965,19 @@ function transformNetwork(payload, bufferedNetworks, isNested) {
       };
     }
     case "workflow-execution-start": {
-      const current = bufferedNetworks.get(payload.runId) || { name: "", steps: [], usage: null, output: null };
+      const current = bufferedNetworks.get(payload.runId);
+      if (!current) return null;
       current.steps.push({
-        name: payload.payload.name,
+        id: payload.payload.runId,
+        name: payload.payload.workflowId,
         status: "running",
-        input: payload.payload.args || null,
+        iteration: payload.payload.args?.iteration ?? 0,
+        input: { prompt: payload.payload.args?.prompt ?? "" },
         output: null,
+        task: null,
         suspendPayload: null,
-        resumePayload: null
+        resumePayload: null,
+        [PRIMITIVE_CACHE_SYMBOL]: /* @__PURE__ */ new Map()
       });
       bufferedNetworks.set(payload.runId, current);
       return {
@@ -925,14 +990,21 @@ function transformNetwork(payload, bufferedNetworks, isNested) {
       };
     }
     case "tool-execution-start": {
-      const current = bufferedNetworks.get(payload.runId) || { name: "", steps: [], usage: null, output: null };
+      const current = bufferedNetworks.get(payload.runId);
+      if (!current) return null;
       current.steps.push({
+        id: payload.payload.args.toolCallId,
         name: payload.payload.args?.toolName,
         status: "running",
+        iteration: payload.payload.args?.iteration ? Number(payload.payload.args.iteration) : 0,
+        task: {
+          id: payload.payload.args?.toolName
+        },
         input: payload.payload.args?.args || null,
         output: null,
         suspendPayload: null,
-        resumePayload: null
+        resumePayload: null,
+        [PRIMITIVE_CACHE_SYMBOL]: /* @__PURE__ */ new Map()
       });
       bufferedNetworks.set(payload.runId, current);
       return {
@@ -947,14 +1019,13 @@ function transformNetwork(payload, bufferedNetworks, isNested) {
     case "agent-execution-end": {
       const current = bufferedNetworks.get(payload.runId);
       if (!current) return null;
-      current.steps.push({
-        name: payload.payload.agentId,
-        status: "success",
-        input: null,
-        output: payload.payload.result,
-        suspendPayload: null,
-        resumePayload: null
-      });
+      const stepId = payload.payload.runId;
+      const step = current.steps.find((step2) => step2.id === stepId);
+      if (!step) {
+        return null;
+      }
+      step.status = "success";
+      step.output = payload.payload.result;
       return {
         type: isNested ? "data-tool-network" : "data-network",
         id: payload.runId,
@@ -969,14 +1040,13 @@ function transformNetwork(payload, bufferedNetworks, isNested) {
     case "tool-execution-end": {
       const current = bufferedNetworks.get(payload.runId);
       if (!current) return null;
-      current.steps.push({
-        name: payload.payload.toolName,
-        status: "success",
-        input: null,
-        output: payload.payload.result,
-        suspendPayload: null,
-        resumePayload: null
-      });
+      const stepId = payload.payload.toolCallId;
+      const step = current.steps.find((step2) => step2.id === stepId);
+      if (!step) {
+        return null;
+      }
+      step.status = "success";
+      step.output = payload.payload.result;
       return {
         type: isNested ? "data-tool-network" : "data-network",
         id: payload.runId,
@@ -990,14 +1060,13 @@ function transformNetwork(payload, bufferedNetworks, isNested) {
     case "workflow-execution-end": {
       const current = bufferedNetworks.get(payload.runId);
       if (!current) return null;
-      current.steps.push({
-        name: payload.payload.name,
-        status: "success",
-        input: null,
-        output: payload.payload.result,
-        suspendPayload: null,
-        resumePayload: null
-      });
+      const stepId = payload.payload.runId;
+      const step = current.steps.find((step2) => step2.id === stepId);
+      if (!step) {
+        return null;
+      }
+      step.status = "success";
+      step.output = payload.payload.result;
       return {
         type: isNested ? "data-tool-network" : "data-network",
         id: payload.runId,
@@ -1012,12 +1081,24 @@ function transformNetwork(payload, bufferedNetworks, isNested) {
     case "routing-agent-end": {
       const current = bufferedNetworks.get(payload.runId);
       if (!current) return null;
+      const stepId = payload.payload.runId;
+      const step = current.steps.find((step2) => step2.id === stepId);
+      if (!step) {
+        return null;
+      }
+      step.status = "success";
+      step.task = {
+        id: payload.payload.primitiveId,
+        type: payload.payload.primitiveType,
+        name: payload.payload.task,
+        reason: payload.payload.selectionReason
+      };
+      step.output = payload.payload.result;
       return {
         type: isNested ? "data-tool-network" : "data-network",
         id: payload.runId,
         data: {
           ...current,
-          status: "finished",
           usage: payload.payload?.usage ?? current.usage,
           output: payload.payload?.result ?? current.output
         }
@@ -1051,6 +1132,39 @@ function transformNetwork(payload, bufferedNetworks, isNested) {
       };
     }
     default: {
+      if (payload.type.startsWith("agent-execution-event-")) {
+        const stepId = payload.payload.runId;
+        const current = bufferedNetworks.get(payload.runId);
+        if (!current) return null;
+        const step = current.steps.find((step2) => step2.id === stepId);
+        if (!step) {
+          return null;
+        }
+        step[PRIMITIVE_CACHE_SYMBOL] = step[PRIMITIVE_CACHE_SYMBOL] || /* @__PURE__ */ new Map();
+        const result = transformAgent(payload.payload, step[PRIMITIVE_CACHE_SYMBOL]);
+        if (result) {
+          const { request, response, ...data } = result.data;
+          step.task = data;
+        }
+      }
+      if (payload.type.startsWith("workflow-execution-event-")) {
+        const stepId = payload.payload.runId;
+        const current = bufferedNetworks.get(payload.runId);
+        if (!current) return null;
+        const step = current.steps.find((step2) => step2.id === stepId);
+        if (!step) {
+          return null;
+        }
+        step[PRIMITIVE_CACHE_SYMBOL] = step[PRIMITIVE_CACHE_SYMBOL] || /* @__PURE__ */ new Map();
+        const result = transformWorkflow(payload.payload, step[PRIMITIVE_CACHE_SYMBOL]);
+        if (result && "data" in result) {
+          const data = result.data;
+          step.task = data;
+          if (data.name && step.task) {
+            step.task.id = data.name;
+          }
+        }
+      }
       if (isDataChunkType(payload)) {
         if (!("data" in payload)) {
           throw new Error(
@@ -1058,7 +1172,8 @@ function transformNetwork(payload, bufferedNetworks, isNested) {
  ${JSON.stringify(payload)}`
           );
         }
-        return payload;
+        const { type, data } = payload;
+        return { type, data };
       }
       if (isAgentExecutionDataChunkType(payload)) {
         if (!("data" in payload.payload)) {
@@ -1067,7 +1182,8 @@ function transformNetwork(payload, bufferedNetworks, isNested) {
  ${JSON.stringify(payload)}`
           );
         }
-        return payload.payload;
+        const { type, data } = payload.payload;
+        return { type, data };
       }
       if (isWorkflowExecutionDataChunkType(payload)) {
         if (!("data" in payload.payload)) {
@@ -1076,22 +1192,26 @@ function transformNetwork(payload, bufferedNetworks, isNested) {
  ${JSON.stringify(payload)}`
           );
         }
-        return payload.payload;
+        const { type, data } = payload.payload;
+        return { type, data };
       }
       return null;
     }
   }
 }
 
-// src/convert-streams.ts
-function toAISdkV5Stream(stream, options = {
+// src/to-ai-sdk-format.ts
+function toAISdkFormat(stream, options = {
   from: "agent",
   sendStart: true,
   sendFinish: true
 }) {
   const from = options?.from;
   if (from === "workflow") {
-    return stream.pipeThrough(WorkflowStreamToAISDKTransformer());
+    const includeTextStreamParts = options?.includeTextStreamParts ?? false;
+    return stream.pipeThrough(
+      WorkflowStreamToAISDKTransformer({ includeTextStreamParts })
+    );
   }
   if (from === "network") {
     return stream.pipeThrough(AgentNetworkToAISDKTransformer());
@@ -1219,7 +1339,7 @@ function chatRoute({
     handler: async (c) => {
       const { messages, ...rest } = await c.req.json();
       const mastra = c.get("mastra");
-      const requestContext = c.get("requestContext");
+      const runtimeContext = c.get("runtimeContext");
       let agentToUse = agent;
       if (!agent) {
         const agentId = c.req.param("agentId");
@@ -1230,20 +1350,20 @@ function chatRoute({
           `Fixed agent ID was set together with an agentId path parameter. This can lead to unexpected behavior.`
         );
       }
-      if (requestContext && defaultOptions?.requestContext) {
-        mastra.getLogger()?.warn(`"requestContext" set in the route options will be overridden by the request's "requestContext".`);
+      if (runtimeContext && defaultOptions?.runtimeContext) {
+        mastra.getLogger()?.warn(`"runtimeContext" set in the route options will be overridden by the request's "runtimeContext".`);
       }
       if (!agentToUse) {
         throw new Error("Agent ID is required");
       }
-      const agentObj = mastra.getAgent(agentToUse);
+      const agentObj = mastra.getAgentById(agentToUse);
       if (!agentObj) {
         throw new Error(`Agent ${agentToUse} not found`);
       }
       const result = await agentObj.stream(messages, {
         ...defaultOptions,
         ...rest,
-        requestContext: requestContext || defaultOptions?.requestContext
+        runtimeContext: runtimeContext || defaultOptions?.runtimeContext
       });
       let lastMessageId;
       if (messages.length > 0 && messages[messages.length - 1].role === "assistant") {
@@ -1252,7 +1372,7 @@ function chatRoute({
       const uiMessageStream = createUIMessageStream({
         originalMessages: messages,
         execute: async ({ writer }) => {
-          for await (const part of toAISdkV5Stream(result, {
+          for await (const part of toAISdkFormat(result, {
             from: "agent",
             lastMessageId,
             sendStart,
@@ -1272,7 +1392,8 @@ function chatRoute({
 }
 function workflowRoute({
   path = "/api/workflows/:workflowId/stream",
-  workflow
+  workflow,
+  includeTextStreamParts = false
 }) {
   if (!workflow && !path.includes("/:workflowId")) {
     throw new Error("Path must include :workflowId to route to the correct workflow or pass the workflow explicitly");
@@ -1303,7 +1424,7 @@ function workflowRoute({
                 resourceId: { type: "string" },
                 inputData: { type: "object", additionalProperties: true },
                 resumeData: { type: "object", additionalProperties: true },
-                requestContext: { type: "object", additionalProperties: true },
+                runtimeContext: { type: "object", additionalProperties: true },
                 tracingOptions: { type: "object", additionalProperties: true },
                 step: { type: "string" }
               }
@@ -1325,7 +1446,7 @@ function workflowRoute({
     handler: async (c) => {
       const { runId, resourceId, inputData, resumeData, ...rest } = await c.req.json();
       const mastra = c.get("mastra");
-      const requestContext = c.get("requestContext");
+      const runtimeContext = c.get("runtimeContext");
       let workflowToUse = workflow;
       if (!workflow) {
         const workflowId = c.req.param("workflowId");
@@ -1339,20 +1460,20 @@ function workflowRoute({
       if (!workflowToUse) {
         throw new Error("Workflow ID is required");
       }
-      const workflowObj = mastra.getWorkflow(workflowToUse);
+      const workflowObj = mastra.getWorkflowById(workflowToUse);
       if (!workflowObj) {
         throw new Error(`Workflow ${workflowToUse} not found`);
       }
-      if (requestContext && rest.requestContext) {
+      if (runtimeContext && rest.runtimeContext) {
         mastra.getLogger()?.warn(
-          `"requestContext" from the request body will be ignored because "requestContext" is already set in the route options.`
+          `"runtimeContext" from the request body will be ignored because "runtimeContext" is already set in the route options.`
         );
       }
-      const run = await workflowObj.createRun({ runId, resourceId, ...rest });
-      const stream = resumeData ? run.resumeStream({ resumeData, ...rest, requestContext: requestContext || rest.requestContext }) : run.stream({ inputData, ...rest, requestContext: requestContext || rest.requestContext });
+      const run = await workflowObj.createRunAsync({ runId, resourceId, ...rest });
+      const stream = resumeData ? run.resumeStream({ resumeData, ...rest, runtimeContext: runtimeContext || rest.runtimeContext }) : run.stream({ inputData, ...rest, runtimeContext: runtimeContext || rest.runtimeContext });
       const uiMessageStream = createUIMessageStream({
         execute: async ({ writer }) => {
-          for await (const part of toAISdkV5Stream(stream, { from: "workflow" })) {
+          for await (const part of toAISdkFormat(stream, { from: "workflow", includeTextStreamParts })) {
             writer.write(part);
           }
         }
@@ -1392,12 +1513,13 @@ function networkRoute({
               type: "object",
               properties: {
                 messages: { type: "array", items: { type: "object" } },
-                requestContext: { type: "object", additionalProperties: true },
+                runtimeContext: { type: "object", additionalProperties: true },
                 runId: { type: "string" },
                 maxSteps: { type: "number" },
                 threadId: { type: "string" },
                 resourceId: { type: "string" },
                 modelSettings: { type: "object", additionalProperties: true },
+                telemetry: { type: "object", additionalProperties: true },
                 tools: { type: "array", items: { type: "object" } }
               },
               required: ["messages"]
@@ -1436,7 +1558,7 @@ function networkRoute({
       if (!agentToUse) {
         throw new Error("Agent ID is required");
       }
-      const agentObj = mastra.getAgent(agentToUse);
+      const agentObj = mastra.getAgentById(agentToUse);
       if (!agentObj) {
         throw new Error(`Agent ${agentToUse} not found`);
       }
@@ -1446,7 +1568,7 @@ function networkRoute({
       });
       const uiMessageStream = createUIMessageStream({
         execute: async ({ writer }) => {
-          for await (const part of toAISdkV5Stream(result, { from: "network" })) {
+          for await (const part of toAISdkFormat(result, { from: "network" })) {
             writer.write(part);
           }
         }
@@ -1456,13 +1578,6 @@ function networkRoute({
   });
 }
 
-// src/to-ai-sdk-format.ts
-function toAISdkFormat() {
-  throw new Error(
-    'toAISdkFormat() has been deprecated. Please use toAISdkStream() instead.\n\nMigration:\n  import { toAISdkFormat } from "@mastra/ai-sdk";\n  // Change to:\n  import { toAISdkStream } from "@mastra/ai-sdk";\n\nThe function signature remains the same.'
-  );
-}
-
-export { chatRoute, networkRoute, toAISdkFormat, toAISdkV5Stream as toAISdkStream, workflowRoute };
+export { chatRoute, networkRoute, toAISdkFormat, workflowRoute };
 //# sourceMappingURL=index.js.map
 //# sourceMappingURL=index.js.map