@mastra/ai-sdk 0.0.0-just-snapshot-20251014192224 → 0.0.0-main-test-20251105183450

package/dist/index.js

@@ -3,6 +3,33 @@ import { createUIMessageStream, createUIMessageStreamResponse } from 'ai';
 import { DefaultGeneratedFile, DefaultGeneratedFileWithType } from '@mastra/core/stream';
 
 // src/chat-route.ts
+
+// src/utils.ts
+var isDataChunkType = (chunk) => {
+  return chunk && typeof chunk === "object" && "type" in chunk && chunk.type?.startsWith("data-");
+};
+function safeParseErrorObject(obj) {
+  if (typeof obj !== "object" || obj === null) {
+    return String(obj);
+  }
+  try {
+    const stringified = JSON.stringify(obj);
+    if (stringified === "{}") {
+      return String(obj);
+    }
+    return stringified;
+  } catch {
+    return String(obj);
+  }
+}
+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({
   chunk,
   mode = "stream"
@@ -208,6 +235,9 @@ function convertMastraChunkToAISDKv5({
           ...chunk.payload || {}
         };
       }
+      if ("type" in chunk && chunk.type?.startsWith("data-")) {
+        return chunk;
+      }
       return;
   }
 }
@@ -221,7 +251,7 @@ function convertFullStreamChunkToUIMessageStream({
   sendFinish,
   responseMessageId
 }) {
-  const partType = part.type;
+  const partType = part?.type;
   switch (partType) {
     case "text-start": {
       return {
@@ -378,8 +408,16 @@ function convertFullStreamChunkToUIMessageStream({
       return;
     }
     default: {
-      const exhaustiveCheck = partType;
-      throw new Error(`Unknown chunk type: ${exhaustiveCheck}`);
+      if (isDataChunkType(part)) {
+        if (!("data" in part)) {
+          throw new Error(
+            `UI Messages require a data property when using data- prefixed chunks 
+ ${JSON.stringify(part)}`
+          );
+        }
+        return part;
+      }
+      return;
     }
   }
 }
@@ -423,7 +461,7 @@ function AgentNetworkToAISDKTransformer() {
     }
   });
 }
-function AgentStreamToAISDKTransformer() {
+function AgentStreamToAISDKTransformer(lastMessageId) {
   let bufferedSteps = /* @__PURE__ */ new Map();
   return new TransformStream({
     transform(chunk, controller) {
@@ -434,9 +472,9 @@ function AgentStreamToAISDKTransformer() {
         sendSources: false,
         sendStart: true,
         sendFinish: true,
-        responseMessageId: chunk.runId,
-        onError() {
-          return "Error";
+        responseMessageId: lastMessageId,
+        onError(error) {
+          return safeParseErrorObject(error);
         }
       });
       if (transformedChunk) {
@@ -618,7 +656,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 {
@@ -651,6 +691,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;
@@ -665,87 +726,117 @@ function transformWorkflow(payload, bufferedWorkflows, isNested) {
         }
       };
     }
-    default:
+    default: {
+      if (isDataChunkType(payload)) {
+        if (!("data" in payload)) {
+          throw new Error(
+            `UI Messages require a data property when using data- prefixed chunks 
+ ${JSON.stringify(payload)}`
+          );
+        }
+        return payload;
+      }
       return null;
+    }
   }
 }
 function transformNetwork(payload, bufferedNetworks, isNested) {
   switch (payload.type) {
     case "routing-agent-start": {
-      if (!bufferedNetworks.has(payload.payload.runId)) {
-        bufferedNetworks.set(payload.payload.runId, {
+      if (!bufferedNetworks.has(payload.runId)) {
+        bufferedNetworks.set(payload.runId, {
           name: payload.payload.agentId,
-          steps: []
+          steps: [],
+          usage: null,
+          output: null
         });
       }
       return {
         type: isNested ? "data-tool-network" : "data-network",
-        id: payload.payload.runId,
+        id: payload.runId,
         data: {
-          name: bufferedNetworks.get(payload.payload.runId).name,
+          name: bufferedNetworks.get(payload.runId).name,
           status: "running",
-          steps: bufferedNetworks.get(payload.payload.runId).steps,
+          usage: null,
+          steps: bufferedNetworks.get(payload.runId).steps,
           output: null
         }
       };
     }
+    case "routing-agent-text-start": {
+      const current = bufferedNetworks.get(payload.runId);
+      if (!current) return null;
+      return {
+        type: "text-start",
+        id: payload.runId
+      };
+    }
+    case "routing-agent-text-delta": {
+      const current = bufferedNetworks.get(payload.runId);
+      if (!current) return null;
+      return {
+        type: "text-delta",
+        id: payload.runId,
+        delta: payload.payload.text
+      };
+    }
     case "agent-execution-start": {
-      const current = bufferedNetworks.get(payload.payload.runId) || { name: "", steps: [] };
+      const current = bufferedNetworks.get(payload.runId) || { name: "", steps: [], usage: null, output: null };
       current.steps.push({
         name: payload.payload.agentId,
         status: "running",
         input: payload.payload.args || null,
-        output: null
+        output: null,
+        suspendPayload: null,
+        resumePayload: null
       });
-      bufferedNetworks.set(payload.payload.runId, current);
+      bufferedNetworks.set(payload.runId, current);
       return {
         type: isNested ? "data-tool-network" : "data-network",
-        id: payload.payload.runId,
+        id: payload.runId,
         data: {
-          name: current.name,
-          status: "running",
-          steps: current.steps,
-          output: null
+          ...current,
+          status: "running"
         }
       };
     }
     case "workflow-execution-start": {
-      const current = bufferedNetworks.get(payload.payload.runId) || { name: "", steps: [] };
+      const current = bufferedNetworks.get(payload.runId) || { name: "", steps: [], usage: null, output: null };
       current.steps.push({
         name: payload.payload.name,
         status: "running",
         input: payload.payload.args || null,
-        output: null
+        output: null,
+        suspendPayload: null,
+        resumePayload: null
       });
-      bufferedNetworks.set(payload.payload.runId, current);
+      bufferedNetworks.set(payload.runId, current);
       return {
         type: isNested ? "data-tool-network" : "data-network",
-        id: payload.payload.runId,
+        id: payload.runId,
         data: {
-          name: current.name,
-          status: "running",
-          steps: current.steps,
-          output: null
+          ...current,
+          status: "running"
         }
       };
     }
     case "tool-execution-start": {
-      const current = bufferedNetworks.get(payload.payload.runId) || { name: "", steps: [] };
+      const current = bufferedNetworks.get(payload.runId) || { name: "", steps: [], usage: null, output: null };
       current.steps.push({
         name: payload.payload.args?.toolName,
         status: "running",
         input: payload.payload.args?.args || null,
-        output: null
+        output: null,
+        suspendPayload: null,
+        resumePayload: null
       });
-      bufferedNetworks.set(payload.payload.runId, current);
+      bufferedNetworks.set(payload.runId, current);
       return {
         type: isNested ? "data-tool-network" : "data-network",
-        id: payload.payload.runId,
+        id: payload.runId,
         data: {
-          name: current.name,
-          status: "running",
-          steps: current.steps,
-          output: null
+          ...current,
+          status: "running"
         }
       };
     }
@@ -756,16 +847,18 @@ 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",
         id: payload.runId,
         data: {
-          name: current.name,
+          ...current,
+          usage: payload.payload?.usage ?? current.usage,
           status: "running",
-          steps: current.steps,
-          output: payload.payload.result ?? null
+          output: payload.payload.result ?? current.output
         }
       };
     }
@@ -776,16 +869,17 @@ 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",
         id: payload.runId,
         data: {
-          name: current.name,
+          ...current,
           status: "running",
-          steps: current.steps,
-          output: payload.payload.result ?? null
+          output: payload.payload.result ?? current.output
         }
       };
     }
@@ -796,44 +890,45 @@ 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",
         id: payload.runId,
         data: {
-          name: current.name,
+          ...current,
+          usage: payload.payload?.usage ?? current.usage,
           status: "running",
-          steps: current.steps,
-          output: payload.payload.result ?? null
+          output: payload.payload.result ?? current.output
         }
       };
     }
     case "routing-agent-end": {
-      const current = bufferedNetworks.get(payload.payload.runId);
+      const current = bufferedNetworks.get(payload.runId);
       if (!current) return null;
       return {
         type: isNested ? "data-tool-network" : "data-network",
-        id: payload.payload.runId,
+        id: payload.runId,
         data: {
-          name: current.name,
+          ...current,
           status: "finished",
-          steps: current.steps,
-          output: payload.payload?.result ?? null
+          usage: payload.payload?.usage ?? current.usage,
+          output: payload.payload?.result ?? current.output
         }
       };
     }
     case "network-execution-event-step-finish": {
-      const current = bufferedNetworks.get(payload.payload.runId);
+      const current = bufferedNetworks.get(payload.runId);
       if (!current) return null;
       return {
         type: isNested ? "data-tool-network" : "data-network",
-        id: payload.payload.runId,
+        id: payload.runId,
         data: {
-          name: current.name,
+          ...current,
           status: "finished",
-          steps: current.steps,
-          output: payload.payload?.result ?? null
+          output: payload.payload?.result ?? current.output
         }
       };
     }
@@ -844,20 +939,48 @@ function transformNetwork(payload, bufferedNetworks, isNested) {
         type: isNested ? "data-tool-network" : "data-network",
         id: payload.runId,
         data: {
-          name: current.name,
+          ...current,
+          usage: payload.payload?.usage ?? current.usage,
           status: "finished",
-          steps: current.steps,
-          output: payload.payload?.result ?? null
+          output: payload.payload?.result ?? current.output
         }
       };
     }
-    default:
+    default: {
+      if (isDataChunkType(payload)) {
+        if (!("data" in payload)) {
+          throw new Error(
+            `UI Messages require a data property when using data- prefixed chunks 
+ ${JSON.stringify(payload)}`
+          );
+        }
+        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;
+    }
   }
 }
 
-// src/to-ai-sdk-format.ts
-function toAISdkFormat(stream, options = { from: "agent" }) {
+// src/convert-streams.ts
+function toAISdkV5Stream(stream, options = { from: "agent" }) {
   const from = options?.from;
   if (from === "workflow") {
     return stream.pipeThrough(WorkflowStreamToAISDKTransformer());
@@ -866,7 +989,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
@@ -974,7 +1097,7 @@ function chatRoute({
     handler: async (c) => {
       const { messages, ...rest } = await c.req.json();
       const mastra = c.get("mastra");
-      const runtimeContext = c.get("runtimeContext");
+      const requestContext = c.get("requestContext");
       let agentToUse = agent;
       if (!agent) {
         const agentId = c.req.param("agentId");
@@ -985,8 +1108,8 @@ 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 (requestContext && defaultOptions?.requestContext) {
+        mastra.getLogger()?.warn(`"requestContext" set in the route options will be overridden by the request's "requestContext".`);
       }
       if (!agentToUse) {
         throw new Error("Agent ID is required");
@@ -998,11 +1121,16 @@ function chatRoute({
       const result = await agentObj.stream(messages, {
         ...defaultOptions,
         ...rest,
-        runtimeContext: runtimeContext || defaultOptions?.runtimeContext
+        requestContext: requestContext || defaultOptions?.requestContext
       });
+      let lastMessageId;
+      if (messages.length > 0 && messages[messages.length - 1].role === "assistant") {
+        lastMessageId = messages[messages.length - 1].id;
+      }
       const uiMessageStream = createUIMessageStream({
+        originalMessages: messages,
         execute: async ({ writer }) => {
-          for await (const part of toAISdkFormat(result, { from: "agent" })) {
+          for await (const part of toAISdkV5Stream(result, { from: "agent", lastMessageId })) {
             writer.write(part);
           }
         }
@@ -1043,7 +1171,7 @@ function workflowRoute({
               type: "object",
               properties: {
                 inputData: { type: "object", additionalProperties: true },
-                runtimeContext: { type: "object", additionalProperties: true },
+                requestContext: { type: "object", additionalProperties: true },
                 tracingOptions: { type: "object", additionalProperties: true }
               }
             }
@@ -1062,7 +1190,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) {
@@ -1081,11 +1209,11 @@ function workflowRoute({
       if (!workflowObj) {
         throw new Error(`Workflow ${workflowToUse} not found`);
       }
-      const run = await workflowObj.createRunAsync();
-      const stream = run.streamVNext({ inputData, ...rest });
+      const run = await workflowObj.createRun();
+      const stream = resumeData ? run.resumeStream({ resumeData, ...rest }) : run.stream({ inputData, ...rest });
       const uiMessageStream = createUIMessageStream({
         execute: async ({ writer }) => {
-          for await (const part of toAISdkFormat(stream, { from: "workflow" })) {
+          for await (const part of toAISdkV5Stream(stream, { from: "workflow" })) {
             writer.write(part);
           }
         }
@@ -1125,13 +1253,12 @@ function networkRoute({
               type: "object",
               properties: {
                 messages: { type: "array", items: { type: "object" } },
-                runtimeContext: { type: "object", additionalProperties: true },
+                requestContext: { 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"]
@@ -1180,7 +1307,7 @@ function networkRoute({
       });
       const uiMessageStream = createUIMessageStream({
         execute: async ({ writer }) => {
-          for await (const part of toAISdkFormat(result, { from: "network" })) {
+          for await (const part of toAISdkV5Stream(result, { from: "network" })) {
             writer.write(part);
           }
         }
@@ -1190,6 +1317,13 @@ function networkRoute({
   });
 }
 
-export { chatRoute, networkRoute, toAISdkFormat, workflowRoute };
+// 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 };
 //# sourceMappingURL=index.js.map
 //# sourceMappingURL=index.js.map