@mastra/ai-sdk 0.2.0-alpha.0 → 0.2.0-alpha.2

package/CHANGELOG.md

@@ -1,5 +1,29 @@
 # @mastra/ai-sdk
 
+## 0.2.0-alpha.2
+
+### Patch Changes
+
+- Improve types for networkRoute and workflowRoute functions ([#8844](https://github.com/mastra-ai/mastra/pull/8844))
+
+- Updated dependencies [[`a1799bc`](https://github.com/mastra-ai/mastra/commit/a1799bcc1b5a1cdc188f2ac0165f17a1c4ac6f7b), [`6ff6094`](https://github.com/mastra-ai/mastra/commit/6ff60946f4ecfebdeef6e21d2b230c2204f2c9b8)]:
+  - @mastra/core@0.21.0-alpha.3
+
+## 0.2.0-alpha.1
+
+### Minor Changes
+
+- Update peer dependencies to match core package version bump (0.20.3) ([#8672](https://github.com/mastra-ai/mastra/pull/8672))
+
+### Patch Changes
+
+- pass runtimeContext to agent stream options in chatRoute ([#8641](https://github.com/mastra-ai/mastra/pull/8641))
+
+- ai-sdk workflow route, agent network route ([#8672](https://github.com/mastra-ai/mastra/pull/8672))
+
+- Updated dependencies [[`1ed9670`](https://github.com/mastra-ai/mastra/commit/1ed9670d3ca50cb60dc2e517738c5eef3968ed27), [`158381d`](https://github.com/mastra-ai/mastra/commit/158381d39335be934b81ef8a1947bccace492c25), [`fb703b9`](https://github.com/mastra-ai/mastra/commit/fb703b9634eeaff1a6eb2b5531ce0f9e8fb04727), [`37a2314`](https://github.com/mastra-ai/mastra/commit/37a23148e0e5a3b40d4f9f098b194671a8a49faf), [`05a9dee`](https://github.com/mastra-ai/mastra/commit/05a9dee3d355694d28847bfffb6289657fcf7dfa), [`e3c1077`](https://github.com/mastra-ai/mastra/commit/e3c107763aedd1643d3def5df450c235da9ff76c), [`1bccdb3`](https://github.com/mastra-ai/mastra/commit/1bccdb33eb90cbeba2dc5ece1c2561fb774b26b6), [`5ef944a`](https://github.com/mastra-ai/mastra/commit/5ef944a3721d93105675cac2b2311432ff8cc393), [`d6b186f`](https://github.com/mastra-ai/mastra/commit/d6b186fb08f1caf1b86f73d3a5ee88fb999ca3be), [`65493b3`](https://github.com/mastra-ai/mastra/commit/65493b31c36f6fdb78f9679f7e1ecf0c250aa5ee), [`a998b8f`](https://github.com/mastra-ai/mastra/commit/a998b8f858091c2ec47683e60766cf12d03001e4), [`8a37bdd`](https://github.com/mastra-ai/mastra/commit/8a37bddb6d8614a32c5b70303d583d80c620ea61)]:
+  - @mastra/core@0.21.0-alpha.1
+
 ## 0.2.0-alpha.0
 
 ### Minor 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;AAChE,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,CA+IhE"}
+{"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,CAuJhE"}

package/dist/index.cjs

@@ -392,24 +392,35 @@ function WorkflowStreamToAISDKTransformer() {
   return new TransformStream({
     start(controller) {
       controller.enqueue({
-        data: JSON.stringify({
-          type: "start",
-          messageId: "1"
-        })
+        type: "start"
       });
     },
     flush(controller) {
       controller.enqueue({
-        data: JSON.stringify({
-          type: "finish"
-        })
+        type: "finish"
+      });
+    },
+    transform(chunk, controller) {
+      const transformed = transformWorkflow(chunk, bufferedWorkflows);
+      if (transformed) controller.enqueue(transformed);
+    }
+  });
+}
+function AgentNetworkToAISDKTransformer() {
+  const bufferedNetworks = /* @__PURE__ */ new Map();
+  return new TransformStream({
+    start(controller) {
+      controller.enqueue({
+        type: "start"
       });
+    },
+    flush(controller) {
       controller.enqueue({
-        data: "[DONE]"
+        type: "finish"
       });
     },
     transform(chunk, controller) {
-      const transformed = transformWorkflow(chunk, bufferedWorkflows);
+      const transformed = transformNetwork(chunk, bufferedNetworks);
       if (transformed) controller.enqueue(transformed);
     }
   });
@@ -437,11 +448,11 @@ function AgentStreamToAISDKTransformer() {
           if (agentTransformed) controller.enqueue(agentTransformed);
         } else if (transformedChunk.type === "tool-workflow") {
           const payload = transformedChunk.payload;
-          const workflowChunk = transformWorkflow(payload, bufferedSteps);
+          const workflowChunk = transformWorkflow(payload, bufferedSteps, true);
           if (workflowChunk) controller.enqueue(workflowChunk);
         } else if (transformedChunk.type === "tool-network") {
           const payload = transformedChunk.payload;
-          const networkChunk = transformNetwork(payload, bufferedSteps);
+          const networkChunk = transformNetwork(payload, bufferedSteps, true);
           if (networkChunk) controller.enqueue(networkChunk);
         } else {
           controller.enqueue(transformedChunk);
@@ -586,7 +597,7 @@ function transformAgent(payload, bufferedSteps) {
   }
   return null;
 }
-function transformWorkflow(payload, bufferedWorkflows) {
+function transformWorkflow(payload, bufferedWorkflows, isNested) {
   switch (payload.type) {
     case "workflow-start":
       bufferedWorkflows.set(payload.runId, {
@@ -594,7 +605,7 @@ function transformWorkflow(payload, bufferedWorkflows) {
         steps: {}
       });
       return {
-        type: "data-workflow",
+        type: isNested ? "data-tool-workflow" : "data-workflow",
         id: payload.runId,
         data: {
           name: bufferedWorkflows.get(payload.runId).name,
@@ -613,7 +624,7 @@ function transformWorkflow(payload, bufferedWorkflows) {
       };
       bufferedWorkflows.set(payload.runId, current);
       return {
-        type: "data-workflow",
+        type: isNested ? "data-tool-workflow" : "data-workflow",
         id: payload.runId,
         data: {
           name: current.name,
@@ -632,7 +643,7 @@ function transformWorkflow(payload, bufferedWorkflows) {
         output: payload.payload.output ?? null
       };
       return {
-        type: "data-workflow",
+        type: isNested ? "data-tool-workflow" : "data-workflow",
         id: payload.runId,
         data: {
           name: current.name,
@@ -646,7 +657,7 @@ function transformWorkflow(payload, bufferedWorkflows) {
       const current = bufferedWorkflows.get(payload.runId);
       if (!current) return null;
       return {
-        type: "data-workflow",
+        type: isNested ? "data-tool-workflow" : "data-workflow",
         id: payload.runId,
         data: {
           name: current.name,
@@ -660,15 +671,17 @@ function transformWorkflow(payload, bufferedWorkflows) {
       return null;
   }
 }
-function transformNetwork(payload, bufferedNetworks) {
+function transformNetwork(payload, bufferedNetworks, isNested) {
   switch (payload.type) {
     case "routing-agent-start": {
-      bufferedNetworks.set(payload.payload.runId, {
-        name: payload.payload.agentId,
-        steps: []
-      });
+      if (!bufferedNetworks.has(payload.payload.runId)) {
+        bufferedNetworks.set(payload.payload.runId, {
+          name: payload.payload.agentId,
+          steps: []
+        });
+      }
       return {
-        type: "data-network",
+        type: isNested ? "data-tool-network" : "data-network",
         id: payload.payload.runId,
         data: {
           name: bufferedNetworks.get(payload.payload.runId).name,
@@ -688,7 +701,7 @@ function transformNetwork(payload, bufferedNetworks) {
       });
       bufferedNetworks.set(payload.payload.runId, current);
       return {
-        type: "data-network",
+        type: isNested ? "data-tool-network" : "data-network",
         id: payload.payload.runId,
         data: {
           name: current.name,
@@ -708,7 +721,7 @@ function transformNetwork(payload, bufferedNetworks) {
       });
       bufferedNetworks.set(payload.payload.runId, current);
       return {
-        type: "data-network",
+        type: isNested ? "data-tool-network" : "data-network",
         id: payload.payload.runId,
         data: {
           name: current.name,
@@ -728,7 +741,7 @@ function transformNetwork(payload, bufferedNetworks) {
       });
       bufferedNetworks.set(payload.payload.runId, current);
       return {
-        type: "data-network",
+        type: isNested ? "data-tool-network" : "data-network",
         id: payload.payload.runId,
         data: {
           name: current.name,
@@ -738,32 +751,63 @@ function transformNetwork(payload, bufferedNetworks) {
         }
       };
     }
-    case "agent-execution-end":
+    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
+      });
+      return {
+        type: isNested ? "data-tool-network" : "data-network",
+        id: payload.runId,
+        data: {
+          name: current.name,
+          status: "running",
+          steps: current.steps,
+          output: payload.payload.result ?? null
+        }
+      };
+    }
     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
+      });
       return {
-        type: "data-network",
+        type: isNested ? "data-tool-network" : "data-network",
         id: payload.runId,
         data: {
           name: current.name,
           status: "running",
           steps: current.steps,
-          output: null
+          output: payload.payload.result ?? null
         }
       };
     }
     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
+      });
       return {
-        type: "data-network",
+        type: isNested ? "data-tool-network" : "data-network",
         id: payload.runId,
         data: {
           name: current.name,
           status: "running",
           steps: current.steps,
-          output: null
+          output: payload.payload.result ?? null
         }
       };
     }
@@ -771,7 +815,7 @@ function transformNetwork(payload, bufferedNetworks) {
       const current = bufferedNetworks.get(payload.payload.runId);
       if (!current) return null;
       return {
-        type: "data-network",
+        type: isNested ? "data-tool-network" : "data-network",
         id: payload.payload.runId,
         data: {
           name: current.name,
@@ -785,7 +829,7 @@ function transformNetwork(payload, bufferedNetworks) {
       const current = bufferedNetworks.get(payload.payload.runId);
       if (!current) return null;
       return {
-        type: "data-network",
+        type: isNested ? "data-tool-network" : "data-network",
         id: payload.payload.runId,
         data: {
           name: current.name,
@@ -799,7 +843,7 @@ function transformNetwork(payload, bufferedNetworks) {
       const current = bufferedNetworks.get(payload.runId);
       if (!current) return null;
       return {
-        type: "data-network",
+        type: isNested ? "data-tool-network" : "data-network",
         id: payload.runId,
         data: {
           name: current.name,
@@ -815,8 +859,16 @@ function transformNetwork(payload, bufferedNetworks) {
 }
 
 // src/to-ai-sdk-format.ts
-function toAISdkFormat(stream) {
-  return stream.fullStream.pipeThrough(AgentStreamToAISDKTransformer());
+function toAISdkFormat(stream, options = { from: "agent" }) {
+  const from = options?.from;
+  if (from === "workflow") {
+    return stream.pipeThrough(WorkflowStreamToAISDKTransformer());
+  }
+  if (from === "network") {
+    return stream.pipeThrough(AgentNetworkToAISDKTransformer());
+  }
+  const agentReadable = "fullStream" in stream ? stream.fullStream : stream;
+  return agentReadable.pipeThrough(AgentStreamToAISDKTransformer());
 }
 
 // src/chat-route.ts
@@ -924,6 +976,7 @@ function chatRoute({
     handler: async (c) => {
       const { messages, ...rest } = await c.req.json();
       const mastra = c.get("mastra");
+      const runtimeContext = c.get("runtimeContext");
       let agentToUse = agent;
       if (!agent) {
         const agentId = c.req.param("agentId");
@@ -934,6 +987,9 @@ function chatRoute({
           `Fixed agent ID was set together with an agentId path parameter. This can lead to unexpected behavior.`
         );
       }
+      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");
       }
@@ -943,11 +999,12 @@ function chatRoute({
       }
       const result = await agentObj.stream(messages, {
         ...defaultOptions,
-        ...rest
+        ...rest,
+        runtimeContext: runtimeContext || defaultOptions?.runtimeContext
       });
       const uiMessageStream = ai.createUIMessageStream({
         execute: async ({ writer }) => {
-          for await (const part of toAISdkFormat(result)) {
+          for await (const part of toAISdkFormat(result, { from: "agent" })) {
             writer.write(part);
           }
         }
@@ -958,9 +1015,186 @@ function chatRoute({
     }
   });
 }
+function workflowRoute({
+  path = "/api/workflows/:workflowId/stream",
+  workflow
+}) {
+  if (!workflow && !path.includes("/:workflowId")) {
+    throw new Error("Path must include :workflowId to route to the correct workflow or pass the workflow explicitly");
+  }
+  return server.registerApiRoute(path, {
+    method: "POST",
+    openapi: {
+      summary: "Stream a workflow in AI SDK format",
+      description: "Starts a workflow run and streams events as AI SDK UIMessage chunks",
+      tags: ["ai-sdk"],
+      parameters: [
+        {
+          name: "workflowId",
+          in: "path",
+          required: true,
+          description: "The ID of the workflow to stream",
+          schema: { type: "string" }
+        }
+      ],
+      requestBody: {
+        required: true,
+        content: {
+          "application/json": {
+            schema: {
+              type: "object",
+              properties: {
+                inputData: { type: "object", additionalProperties: true },
+                runtimeContext: { type: "object", additionalProperties: true },
+                tracingOptions: { type: "object", additionalProperties: true }
+              }
+            }
+          }
+        }
+      },
+      responses: {
+        "200": {
+          description: "Workflow UIMessage event stream",
+          content: {
+            "text/plain": {
+              schema: { type: "string", description: "SSE stream" }
+            }
+          }
+        }
+      }
+    },
+    handler: async (c) => {
+      const { inputData, ...rest } = await c.req.json();
+      const mastra = c.get("mastra");
+      let workflowToUse = workflow;
+      if (!workflow) {
+        const workflowId = c.req.param("workflowId");
+        workflowToUse = workflowId;
+      }
+      if (c.req.param("workflowId") && workflow) {
+        mastra.getLogger()?.warn(
+          `Fixed workflow ID was set together with a workflowId path parameter. This can lead to unexpected behavior.`
+        );
+      }
+      if (!workflowToUse) {
+        throw new Error("Workflow ID is required");
+      }
+      const workflowObj = mastra.getWorkflow(workflowToUse);
+      if (!workflowObj) {
+        throw new Error(`Workflow ${workflowToUse} not found`);
+      }
+      const run = await workflowObj.createRunAsync();
+      const stream = run.streamVNext({ inputData, ...rest });
+      const uiMessageStream = ai.createUIMessageStream({
+        execute: async ({ writer }) => {
+          for await (const part of toAISdkFormat(stream, { from: "workflow" })) {
+            writer.write(part);
+          }
+        }
+      });
+      return ai.createUIMessageStreamResponse({ stream: uiMessageStream });
+    }
+  });
+}
+function networkRoute({
+  path = "/network/:agentId",
+  agent,
+  defaultOptions
+}) {
+  if (!agent && !path.includes("/:agentId")) {
+    throw new Error("Path must include :agentId to route to the correct agent or pass the agent explicitly");
+  }
+  return server.registerApiRoute(path, {
+    method: "POST",
+    openapi: {
+      summary: "Execute an agent network and stream AI SDK events",
+      description: "Routes a request to an agent network and streams UIMessage chunks in AI SDK format",
+      tags: ["ai-sdk"],
+      parameters: [
+        {
+          name: "agentId",
+          in: "path",
+          required: true,
+          description: "The ID of the routing agent to execute as a network",
+          schema: { type: "string" }
+        }
+      ],
+      requestBody: {
+        required: true,
+        content: {
+          "application/json": {
+            schema: {
+              type: "object",
+              properties: {
+                messages: { type: "array", items: { type: "object" } },
+                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"]
+            }
+          }
+        }
+      },
+      responses: {
+        "200": {
+          description: "Streaming AI SDK UIMessage event stream for the agent network",
+          content: { "text/plain": { schema: { type: "string", description: "SSE stream" } } }
+        },
+        "404": {
+          description: "Agent not found",
+          content: {
+            "application/json": {
+              schema: { type: "object", properties: { error: { type: "string" } } }
+            }
+          }
+        }
+      }
+    },
+    handler: async (c) => {
+      const { messages, ...rest } = await c.req.json();
+      const mastra = c.get("mastra");
+      let agentToUse = agent;
+      if (!agent) {
+        const agentId = c.req.param("agentId");
+        agentToUse = agentId;
+      }
+      if (c.req.param("agentId") && agent) {
+        mastra.getLogger()?.warn(
+          `Fixed agent ID was set together with an agentId path parameter. This can lead to unexpected behavior.`
+        );
+      }
+      if (!agentToUse) {
+        throw new Error("Agent ID is required");
+      }
+      const agentObj = mastra.getAgent(agentToUse);
+      if (!agentObj) {
+        throw new Error(`Agent ${agentToUse} not found`);
+      }
+      const result = await agentObj.network(messages, {
+        ...defaultOptions,
+        ...rest
+      });
+      const uiMessageStream = ai.createUIMessageStream({
+        execute: async ({ writer }) => {
+          for await (const part of toAISdkFormat(result, { from: "network" })) {
+            writer.write(part);
+          }
+        }
+      });
+      return ai.createUIMessageStreamResponse({ stream: uiMessageStream });
+    }
+  });
+}
 
-exports.WorkflowStreamToAISDKTransformer = WorkflowStreamToAISDKTransformer;
 exports.chatRoute = chatRoute;
+exports.networkRoute = networkRoute;
 exports.toAISdkFormat = toAISdkFormat;
+exports.workflowRoute = workflowRoute;
 //# sourceMappingURL=index.cjs.map
 //# sourceMappingURL=index.cjs.map