@mastra/ai-sdk 0.2.5 → 0.2.6

package/CHANGELOG.md

@@ -1,5 +1,35 @@
 # @mastra/ai-sdk
 
+## 0.2.6
+
+### Patch Changes
+
+- update peerdeps ([`5ca1cca`](https://github.com/mastra-ai/mastra/commit/5ca1ccac61ffa7141e6d9fa8f22d3ad4d03bf5dc))
+
+- Preserve lastMessageId in chatRoute ([#9728](https://github.com/mastra-ai/mastra/pull/9728))
+
+- Handle custom data writes in agent network execution events in ai sdk transformers ([#9734](https://github.com/mastra-ai/mastra/pull/9734))
+
+- Add support for suspend/resume in AI SDK workflowRoute ([#9732](https://github.com/mastra-ai/mastra/pull/9732))
+
+- Updated dependencies [[`5ca1cca`](https://github.com/mastra-ai/mastra/commit/5ca1ccac61ffa7141e6d9fa8f22d3ad4d03bf5dc), [`6d7e90d`](https://github.com/mastra-ai/mastra/commit/6d7e90db09713e6250f4d6c3d3cff1b4740e50f9), [`f78b908`](https://github.com/mastra-ai/mastra/commit/f78b9080e11af765969b36b4a619761056030840), [`23c2614`](https://github.com/mastra-ai/mastra/commit/23c26140fdbf04b8c59e8d7d52106d67dad962ec), [`e365eda`](https://github.com/mastra-ai/mastra/commit/e365eda45795b43707310531cac1e2ce4e5a0712)]:
+  - @mastra/core@0.24.0
+
+## 0.2.6-alpha.0
+
+### Patch Changes
+
+- update peerdeps ([`5ca1cca`](https://github.com/mastra-ai/mastra/commit/5ca1ccac61ffa7141e6d9fa8f22d3ad4d03bf5dc))
+
+- Preserve lastMessageId in chatRoute ([#9728](https://github.com/mastra-ai/mastra/pull/9728))
+
+- Handle custom data writes in agent network execution events in ai sdk transformers ([#9734](https://github.com/mastra-ai/mastra/pull/9734))
+
+- Add support for suspend/resume in AI SDK workflowRoute ([#9732](https://github.com/mastra-ai/mastra/pull/9732))
+
+- Updated dependencies [[`5ca1cca`](https://github.com/mastra-ai/mastra/commit/5ca1ccac61ffa7141e6d9fa8f22d3ad4d03bf5dc), [`6d7e90d`](https://github.com/mastra-ai/mastra/commit/6d7e90db09713e6250f4d6c3d3cff1b4740e50f9), [`f78b908`](https://github.com/mastra-ai/mastra/commit/f78b9080e11af765969b36b4a619761056030840), [`23c2614`](https://github.com/mastra-ai/mastra/commit/23c26140fdbf04b8c59e8d7d52106d67dad962ec), [`e365eda`](https://github.com/mastra-ai/mastra/commit/e365eda45795b43707310531cac1e2ce4e5a0712)]:
+  - @mastra/core@0.24.0-alpha.0
+
 ## 0.2.5
 
 ### Patch Changes

package/dist/chat-route.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"chat-route.d.ts","sourceRoot":"","sources":["../src/chat-route.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,qBAAqB,EAAE,MAAM,oBAAoB,CAAC;AAEhE,OAAO,EAAE,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AACvD,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AAIxD,MAAM,MAAM,gBAAgB,CAAC,MAAM,SAAS,YAAY,GAAG,SAAS,IAAI;IACtE,cAAc,CAAC,EAAE,qBAAqB,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACzD,GAAG,CACA;IACE,IAAI,EAAE,GAAG,MAAM,WAAW,MAAM,EAAE,CAAC;IACnC,KAAK,CAAC,EAAE,KAAK,CAAC;CACf,GACD;IACE,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,MAAM,CAAC;CACf,CACJ,CAAC;AAEF,wBAAgB,SAAS,CAAC,MAAM,SAAS,YAAY,GAAG,SAAS,EAAE,EACjE,IAAuB,EACvB,KAAK,EACL,cAAc,GACf,EAAE,gBAAgB,CAAC,MAAM,CAAC,GAAG,UAAU,CAAC,OAAO,gBAAgB,CAAC,CAwJhE"}
+{"version":3,"file":"chat-route.d.ts","sourceRoot":"","sources":["../src/chat-route.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,qBAAqB,EAAE,MAAM,oBAAoB,CAAC;AAEhE,OAAO,EAAE,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AACvD,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AAIxD,MAAM,MAAM,gBAAgB,CAAC,MAAM,SAAS,YAAY,GAAG,SAAS,IAAI;IACtE,cAAc,CAAC,EAAE,qBAAqB,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACzD,GAAG,CACA;IACE,IAAI,EAAE,GAAG,MAAM,WAAW,MAAM,EAAE,CAAC;IACnC,KAAK,CAAC,EAAE,KAAK,CAAC;CACf,GACD;IACE,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,MAAM,CAAC;CACf,CACJ,CAAC;AAEF,wBAAgB,SAAS,CAAC,MAAM,SAAS,YAAY,GAAG,SAAS,EAAE,EACjE,IAAuB,EACvB,KAAK,EACL,cAAc,GACf,EAAE,gBAAgB,CAAC,MAAM,CAAC,GAAG,UAAU,CAAC,OAAO,gBAAgB,CAAC,CA6JhE"}

package/dist/index.cjs

@@ -10,6 +10,12 @@ var stream = require('@mastra/core/stream');
 var isDataChunkType = (chunk) => {
   return chunk && typeof chunk === "object" && "type" in chunk && chunk.type?.startsWith("data-");
 };
+var isAgentExecutionDataChunkType = (chunk) => {
+  return chunk && typeof chunk === "object" && "type" in chunk && chunk.type?.startsWith("agent-execution-event-") && "payload" in chunk && typeof chunk.payload === "object" && "type" in chunk.payload && chunk.payload.type?.startsWith("data-");
+};
+var isWorkflowExecutionDataChunkType = (chunk) => {
+  return chunk && typeof chunk === "object" && "type" in chunk && chunk.type?.startsWith("workflow-execution-event-") && "payload" in chunk && typeof chunk.payload === "object" && "type" in chunk.payload && chunk.payload.type?.startsWith("data-");
+};
 
 // src/helpers.ts
 function convertMastraChunkToAISDKv5({
@@ -443,7 +449,7 @@ function AgentNetworkToAISDKTransformer() {
     }
   });
 }
-function AgentStreamToAISDKTransformer() {
+function AgentStreamToAISDKTransformer(lastMessageId) {
   let bufferedSteps = /* @__PURE__ */ new Map();
   return new TransformStream({
     transform(chunk, controller) {
@@ -454,7 +460,7 @@ function AgentStreamToAISDKTransformer() {
         sendSources: false,
         sendStart: true,
         sendFinish: true,
-        responseMessageId: chunk.runId,
+        responseMessageId: lastMessageId,
         onError() {
           return "Error";
         }
@@ -638,7 +644,9 @@ function transformWorkflow(payload, bufferedWorkflows, isNested) {
         name: payload.payload.id,
         status: payload.payload.status,
         input: payload.payload.payload ?? null,
-        output: null
+        output: null,
+        suspendPayload: null,
+        resumePayload: null
       };
       bufferedWorkflows.set(payload.runId, current);
       return {
@@ -671,6 +679,27 @@ function transformWorkflow(payload, bufferedWorkflows, isNested) {
         }
       };
     }
+    case "workflow-step-suspended": {
+      const current = bufferedWorkflows.get(payload.runId);
+      if (!current) return null;
+      current.steps[payload.payload.id] = {
+        ...current.steps[payload.payload.id],
+        status: payload.payload.status,
+        suspendPayload: payload.payload.suspendPayload ?? null,
+        resumePayload: payload.payload.resumePayload ?? null,
+        output: null
+      };
+      return {
+        type: isNested ? "data-tool-workflow" : "data-workflow",
+        id: payload.runId,
+        data: {
+          name: current.name,
+          status: "suspended",
+          steps: current.steps,
+          output: null
+        }
+      };
+    }
     case "workflow-finish": {
       const current = bufferedWorkflows.get(payload.runId);
       if (!current) return null;
@@ -745,7 +774,9 @@ function transformNetwork(payload, bufferedNetworks, isNested) {
         name: payload.payload.agentId,
         status: "running",
         input: payload.payload.args || null,
-        output: null
+        output: null,
+        suspendPayload: null,
+        resumePayload: null
       });
       bufferedNetworks.set(payload.runId, current);
       return {
@@ -763,7 +794,9 @@ function transformNetwork(payload, bufferedNetworks, isNested) {
         name: payload.payload.name,
         status: "running",
         input: payload.payload.args || null,
-        output: null
+        output: null,
+        suspendPayload: null,
+        resumePayload: null
       });
       bufferedNetworks.set(payload.runId, current);
       return {
@@ -781,7 +814,9 @@ function transformNetwork(payload, bufferedNetworks, isNested) {
         name: payload.payload.args?.toolName,
         status: "running",
         input: payload.payload.args?.args || null,
-        output: null
+        output: null,
+        suspendPayload: null,
+        resumePayload: null
       });
       bufferedNetworks.set(payload.runId, current);
       return {
@@ -800,7 +835,9 @@ function transformNetwork(payload, bufferedNetworks, isNested) {
         name: payload.payload.agentId,
         status: "success",
         input: null,
-        output: payload.payload.result
+        output: payload.payload.result,
+        suspendPayload: null,
+        resumePayload: null
       });
       return {
         type: isNested ? "data-tool-network" : "data-network",
@@ -820,7 +857,9 @@ function transformNetwork(payload, bufferedNetworks, isNested) {
         name: payload.payload.toolName,
         status: "success",
         input: null,
-        output: payload.payload.result
+        output: payload.payload.result,
+        suspendPayload: null,
+        resumePayload: null
       });
       return {
         type: isNested ? "data-tool-network" : "data-network",
@@ -839,7 +878,9 @@ function transformNetwork(payload, bufferedNetworks, isNested) {
         name: payload.payload.name,
         status: "success",
         input: null,
-        output: payload.payload.result
+        output: payload.payload.result,
+        suspendPayload: null,
+        resumePayload: null
       });
       return {
         type: isNested ? "data-tool-network" : "data-network",
@@ -903,6 +944,24 @@ function transformNetwork(payload, bufferedNetworks, isNested) {
         }
         return payload;
       }
+      if (isAgentExecutionDataChunkType(payload)) {
+        if (!("data" in payload.payload)) {
+          throw new Error(
+            `UI Messages require a data property when using data- prefixed chunks 
+ ${JSON.stringify(payload)}`
+          );
+        }
+        return payload.payload;
+      }
+      if (isWorkflowExecutionDataChunkType(payload)) {
+        if (!("data" in payload.payload)) {
+          throw new Error(
+            `UI Messages require a data property when using data- prefixed chunks 
+ ${JSON.stringify(payload)}`
+          );
+        }
+        return payload.payload;
+      }
       return null;
     }
   }
@@ -918,7 +977,7 @@ function toAISdkFormat(stream, options = { from: "agent" }) {
     return stream.pipeThrough(AgentNetworkToAISDKTransformer());
   }
   const agentReadable = "fullStream" in stream ? stream.fullStream : stream;
-  return agentReadable.pipeThrough(AgentStreamToAISDKTransformer());
+  return agentReadable.pipeThrough(AgentStreamToAISDKTransformer(options?.lastMessageId));
 }
 
 // src/chat-route.ts
@@ -1052,10 +1111,14 @@ function chatRoute({
         ...rest,
         runtimeContext: runtimeContext || defaultOptions?.runtimeContext
       });
+      let lastMessageId;
+      if (messages.length > 0 && messages[messages.length - 1].role === "assistant") {
+        lastMessageId = messages[messages.length - 1].id;
+      }
       const uiMessageStream = ai.createUIMessageStream({
         originalMessages: messages,
         execute: async ({ writer }) => {
-          for await (const part of toAISdkFormat(result, { from: "agent" })) {
+          for await (const part of toAISdkFormat(result, { from: "agent", lastMessageId })) {
             writer.write(part);
           }
         }
@@ -1115,7 +1178,7 @@ function workflowRoute({
       }
     },
     handler: async (c) => {
-      const { inputData, ...rest } = await c.req.json();
+      const { inputData, resumeData, ...rest } = await c.req.json();
       const mastra = c.get("mastra");
       let workflowToUse = workflow;
       if (!workflow) {
@@ -1135,7 +1198,7 @@ function workflowRoute({
         throw new Error(`Workflow ${workflowToUse} not found`);
       }
       const run = await workflowObj.createRunAsync();
-      const stream = run.streamVNext({ inputData, ...rest });
+      const stream = resumeData ? run.resumeStream({ resumeData, ...rest }) : run.stream({ inputData, ...rest });
       const uiMessageStream = ai.createUIMessageStream({
         execute: async ({ writer }) => {
           for await (const part of toAISdkFormat(stream, { from: "workflow" })) {