@mastra/ai-sdk 0.0.0-extract-tool-ui-inp-playground-ui-20251024041825 → 0.0.0-feat-improve-processors-20251205191721

package/dist/index.cjs

@@ -10,6 +10,51 @@ var stream = require('@mastra/core/stream');
 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);
+  }
+  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({
@@ -120,6 +165,28 @@ function convertMastraChunkToAISDKv5({
         toolName: chunk.payload.toolName,
         input: chunk.payload.args
       };
+    case "tool-call-approval":
+      return {
+        type: "data-tool-call-approval",
+        id: chunk.payload.toolCallId,
+        data: {
+          runId: chunk.runId,
+          toolCallId: chunk.payload.toolCallId,
+          toolName: chunk.payload.toolName,
+          args: chunk.payload.args
+        }
+      };
+    case "tool-call-suspended":
+      return {
+        type: "data-tool-call-suspended",
+        id: chunk.payload.toolCallId,
+        data: {
+          runId: chunk.runId,
+          toolCallId: chunk.payload.toolCallId,
+          toolName: chunk.payload.toolName,
+          suspendPayload: chunk.payload.suspendPayload
+        }
+      };
     case "tool-call-input-streaming-start":
       return {
         type: "tool-input-start",
@@ -210,6 +277,13 @@ function convertMastraChunkToAISDKv5({
         type: "object",
         object: chunk.object
       };
+    case "tripwire":
+      return {
+        type: "data-tripwire",
+        data: {
+          tripwireReason: chunk.payload.tripwireReason
+        }
+      };
     default:
       if (chunk.type && "payload" in chunk && chunk.payload) {
         return {
@@ -265,6 +339,14 @@ function convertFullStreamChunkToUIMessageStream({
       };
     }
     case "reasoning-delta": {
+      if (sendReasoning) {
+        return {
+          type: "reasoning-delta",
+          id: part.id,
+          delta: part.text,
+          ...part.providerMetadata != null ? { providerMetadata: part.providerMetadata } : {}
+        };
+      }
       return;
     }
     case "reasoning-end": {
@@ -282,6 +364,25 @@ function convertFullStreamChunkToUIMessageStream({
       };
     }
     case "source": {
+      if (sendSources && part.sourceType === "url") {
+        return {
+          type: "source-url",
+          sourceId: part.id,
+          url: part.url,
+          title: part.title,
+          ...part.providerMetadata != null ? { providerMetadata: part.providerMetadata } : {}
+        };
+      }
+      if (sendSources && part.sourceType === "document") {
+        return {
+          type: "source-document",
+          sourceId: part.id,
+          mediaType: part.mediaType,
+          title: part.title,
+          filename: part.filename,
+          ...part.providerMetadata != null ? { providerMetadata: part.providerMetadata } : {}
+        };
+      }
       return;
     }
     case "tool-input-start": {
@@ -339,6 +440,14 @@ function convertFullStreamChunkToUIMessageStream({
           toolCallId: part.toolCallId,
           payload: part.output
         };
+      } else if (isDataChunkType(part.output)) {
+        if (!("data" in part.output)) {
+          throw new Error(
+            `UI Messages require a data property when using data- prefixed chunks 
+ ${JSON.stringify(part)}`
+          );
+        }
+        return part.output;
       }
       return;
     }
@@ -364,21 +473,23 @@ function convertFullStreamChunkToUIMessageStream({
       return { type: "finish-step" };
     }
     case "start": {
-      {
+      if (sendStart) {
         return {
           type: "start",
           ...messageMetadataValue != null ? { messageMetadata: messageMetadataValue } : {},
           ...responseMessageId != null ? { messageId: responseMessageId } : {}
         };
       }
+      return;
     }
     case "finish": {
-      {
+      if (sendFinish) {
         return {
           type: "finish",
           ...messageMetadataValue != null ? { messageMetadata: messageMetadataValue } : {}
         };
       }
+      return;
     }
     case "abort": {
       return part;
@@ -405,7 +516,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) {
@@ -419,7 +533,7 @@ function WorkflowStreamToAISDKTransformer() {
       });
     },
     transform(chunk, controller) {
-      const transformed = transformWorkflow(chunk, bufferedWorkflows);
+      const transformed = transformWorkflow(chunk, bufferedWorkflows, false, includeTextStreamParts);
       if (transformed) controller.enqueue(transformed);
     }
   });
@@ -443,20 +557,37 @@ function AgentNetworkToAISDKTransformer() {
     }
   });
 }
-function AgentStreamToAISDKTransformer() {
+function AgentStreamToAISDKTransformer({
+  lastMessageId,
+  sendStart,
+  sendFinish,
+  sendReasoning,
+  sendSources,
+  messageMetadata,
+  onError
+}) {
   let bufferedSteps = /* @__PURE__ */ new Map();
+  let tripwireOccurred = false;
+  let finishEventSent = false;
   return new TransformStream({
     transform(chunk, controller) {
+      if (chunk.type === "tripwire") {
+        tripwireOccurred = true;
+      }
+      if (chunk.type === "finish") {
+        finishEventSent = true;
+      }
       const part = convertMastraChunkToAISDKv5({ chunk, mode: "stream" });
       const transformedChunk = convertFullStreamChunkToUIMessageStream({
         part,
-        sendReasoning: false,
-        sendSources: false,
-        sendStart: true,
-        sendFinish: true,
-        responseMessageId: chunk.runId,
-        onError() {
-          return "Error";
+        sendReasoning,
+        sendSources,
+        messageMetadataValue: messageMetadata?.({ part }),
+        sendStart,
+        sendFinish,
+        responseMessageId: lastMessageId,
+        onError(error) {
+          return onError ? onError(error) : safeParseErrorObject(error);
         }
       });
       if (transformedChunk) {
@@ -476,6 +607,14 @@ function AgentStreamToAISDKTransformer() {
           controller.enqueue(transformedChunk);
         }
       }
+    },
+    flush(controller) {
+      if (tripwireOccurred && !finishEventSent && sendFinish) {
+        controller.enqueue({
+          type: "finish",
+          finishReason: "other"
+        });
+      }
     }
   });
 }
@@ -615,7 +754,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, {
@@ -638,7 +777,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 +812,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;
@@ -685,6 +847,29 @@ 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(
+            `UI Messages require a data property when using data- prefixed chunks 
+ ${JSON.stringify(output)}`
+          );
+        }
+        return output;
+      }
+      return null;
+    }
     default: {
       if (isDataChunkType(payload)) {
         if (!("data" in payload)) {
@@ -702,19 +887,39 @@ function transformWorkflow(payload, bufferedWorkflows, isNested) {
 function transformNetwork(payload, bufferedNetworks, isNested) {
   switch (payload.type) {
     case "routing-agent-start": {
-      if (!bufferedNetworks.has(payload.payload.runId)) {
-        bufferedNetworks.set(payload.payload.runId, {
-          name: payload.payload.agentId,
-          steps: []
+      if (!bufferedNetworks.has(payload.runId)) {
+        bufferedNetworks.set(payload.runId, {
+          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.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
         }
       };
@@ -737,150 +942,180 @@ function transformNetwork(payload, bufferedNetworks, isNested) {
       };
     }
     case "agent-execution-start": {
-      const current = bufferedNetworks.get(payload.payload.runId) || { name: "", steps: [] };
+      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,
-        output: null
+        iteration: payload.payload.args?.iteration ?? 0,
+        input: { prompt: payload.payload.args?.prompt ?? "" },
+        output: null,
+        task: null,
+        suspendPayload: null,
+        resumePayload: null,
+        [PRIMITIVE_CACHE_SYMBOL]: /* @__PURE__ */ new Map()
       });
-      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);
+      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,
-        output: null
+        iteration: payload.payload.args?.iteration ?? 0,
+        input: { prompt: payload.payload.args?.prompt ?? "" },
+        output: null,
+        task: null,
+        suspendPayload: null,
+        resumePayload: null,
+        [PRIMITIVE_CACHE_SYMBOL]: /* @__PURE__ */ new Map()
       });
-      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);
+      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
+        output: null,
+        suspendPayload: null,
+        resumePayload: null,
+        [PRIMITIVE_CACHE_SYMBOL]: /* @__PURE__ */ new Map()
       });
-      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 "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
-      });
+      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,
         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 "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
-      });
+      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,
         data: {
-          name: current.name,
+          ...current,
           status: "running",
-          steps: current.steps,
-          output: payload.payload.result ?? null
+          output: payload.payload.result ?? current.output
         }
       };
     }
     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
-      });
+      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,
         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;
+      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.payload.runId,
+        id: payload.runId,
         data: {
-          name: current.name,
-          status: "finished",
-          steps: current.steps,
-          output: payload.payload?.result ?? null
+          ...current,
+          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
         }
       };
     }
@@ -891,14 +1126,47 @@ 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: {
+      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(
@@ -906,31 +1174,124 @@ 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)) {
+          throw new Error(
+            `UI Messages require a data property when using data- prefixed chunks 
+ ${JSON.stringify(payload)}`
+          );
+        }
+        const { type, data } = payload.payload;
+        return { type, data };
+      }
+      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)}`
+          );
+        }
+        const { type, data } = payload.payload;
+        return { type, data };
       }
       return null;
     }
   }
 }
 
-// src/to-ai-sdk-format.ts
-function toAISdkFormat(stream, options = { from: "agent" }) {
+// src/convert-streams.ts
+function toAISdkV5Stream(stream, options = {
+  from: "agent",
+  sendStart: true,
+  sendFinish: true
+}) {
   const from = options?.from;
   if (from === "workflow") {
-    return stream.pipeThrough(WorkflowStreamToAISDKTransformer());
+    const includeTextStreamParts = options?.includeTextStreamParts ?? true;
+    return stream.pipeThrough(
+      WorkflowStreamToAISDKTransformer({ includeTextStreamParts })
+    );
   }
   if (from === "network") {
     return stream.pipeThrough(AgentNetworkToAISDKTransformer());
   }
   const agentReadable = "fullStream" in stream ? stream.fullStream : stream;
-  return agentReadable.pipeThrough(AgentStreamToAISDKTransformer());
+  return agentReadable.pipeThrough(
+    AgentStreamToAISDKTransformer({
+      lastMessageId: options?.lastMessageId,
+      sendStart: options?.sendStart,
+      sendFinish: options?.sendFinish,
+      sendReasoning: options?.sendReasoning,
+      sendSources: options?.sendSources,
+      messageMetadata: options?.messageMetadata,
+      onError: options?.onError
+    })
+  );
 }
 
 // src/chat-route.ts
+async function handleChatStream({
+  mastra,
+  agentId,
+  params,
+  defaultOptions,
+  sendStart = true,
+  sendFinish = true,
+  sendReasoning = false,
+  sendSources = false
+}) {
+  const { messages, resumeData, runId, requestContext, ...rest } = params;
+  if (resumeData && !runId) {
+    throw new Error("runId is required when resumeData is provided");
+  }
+  const agentObj = mastra.getAgentById(agentId);
+  if (!agentObj) {
+    throw new Error(`Agent ${agentId} not found`);
+  }
+  if (!Array.isArray(messages)) {
+    throw new Error("Messages must be an array of UIMessage objects");
+  }
+  const mergedOptions = {
+    ...defaultOptions,
+    ...rest,
+    ...runId && { runId },
+    requestContext: requestContext || defaultOptions?.requestContext
+  };
+  const result = resumeData ? await agentObj.resumeStream(resumeData, mergedOptions) : await agentObj.stream(messages, mergedOptions);
+  let lastMessageId;
+  if (messages.length) {
+    const lastMessage = messages[messages.length - 1];
+    if (lastMessage?.role === "assistant") {
+      lastMessageId = lastMessage.id;
+    }
+  }
+  return ai.createUIMessageStream({
+    originalMessages: messages,
+    execute: async ({ writer }) => {
+      for await (const part of toAISdkV5Stream(result, {
+        from: "agent",
+        lastMessageId,
+        sendStart,
+        sendFinish,
+        sendReasoning,
+        sendSources
+      })) {
+        writer.write(part);
+      }
+    }
+  });
+}
 function chatRoute({
   path = "/chat/:agentId",
   agent,
-  defaultOptions
+  defaultOptions,
+  sendStart = true,
+  sendFinish = true,
+  sendReasoning = false,
+  sendSources = false
 }) {
   if (!agent && !path.includes("/:agentId")) {
     throw new Error("Path must include :agentId to route to the correct agent or pass the agent explicitly");
@@ -959,6 +1320,14 @@ function chatRoute({
             schema: {
               type: "object",
               properties: {
+                resumeData: {
+                  type: "object",
+                  description: "Resume data for the agent"
+                },
+                runId: {
+                  type: "string",
+                  description: "The run ID required when resuming an agent execution"
+                },
                 messages: {
                   type: "array",
                   description: "Array of messages in the conversation",
@@ -1029,9 +1398,9 @@ function chatRoute({
       }
     },
     handler: async (c) => {
-      const { messages, ...rest } = await c.req.json();
+      const params = await c.req.json();
       const mastra = c.get("mastra");
-      const runtimeContext = c.get("runtimeContext");
+      const contextRequestContext = c.get("requestContext");
       let agentToUse = agent;
       if (!agent) {
         const agentId = c.req.param("agentId");
@@ -1042,28 +1411,24 @@ 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 (contextRequestContext && 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");
       }
-      const agentObj = mastra.getAgent(agentToUse);
-      if (!agentObj) {
-        throw new Error(`Agent ${agentToUse} not found`);
-      }
-      const result = await agentObj.stream(messages, {
-        ...defaultOptions,
-        ...rest,
-        runtimeContext: runtimeContext || defaultOptions?.runtimeContext
-      });
-      const uiMessageStream = ai.createUIMessageStream({
-        originalMessages: messages,
-        execute: async ({ writer }) => {
-          for await (const part of toAISdkFormat(result, { from: "agent" })) {
-            writer.write(part);
-          }
-        }
+      const uiMessageStream = await handleChatStream({
+        mastra,
+        agentId: agentToUse,
+        params: {
+          ...params,
+          requestContext: contextRequestContext || params.requestContext
+        },
+        defaultOptions,
+        sendStart,
+        sendFinish,
+        sendReasoning,
+        sendSources
       });
       return ai.createUIMessageStreamResponse({
         stream: uiMessageStream
@@ -1071,9 +1436,31 @@ function chatRoute({
     }
   });
 }
+async function handleWorkflowStream({
+  mastra,
+  workflowId,
+  params,
+  includeTextStreamParts = true
+}) {
+  const { runId, resourceId, inputData, resumeData, requestContext, ...rest } = params;
+  const workflowObj = mastra.getWorkflowById(workflowId);
+  if (!workflowObj) {
+    throw new Error(`Workflow ${workflowId} not found`);
+  }
+  const run = await workflowObj.createRun({ runId, resourceId, ...rest });
+  const stream = resumeData ? run.resumeStream({ resumeData, ...rest, requestContext }) : run.stream({ inputData, ...rest, requestContext });
+  return ai.createUIMessageStream({
+    execute: async ({ writer }) => {
+      for await (const part of toAISdkV5Stream(stream, { from: "workflow", includeTextStreamParts })) {
+        writer.write(part);
+      }
+    }
+  });
+}
 function workflowRoute({
   path = "/api/workflows/:workflowId/stream",
-  workflow
+  workflow,
+  includeTextStreamParts = true
 }) {
   if (!workflow && !path.includes("/:workflowId")) {
     throw new Error("Path must include :workflowId to route to the correct workflow or pass the workflow explicitly");
@@ -1100,9 +1487,13 @@ function workflowRoute({
             schema: {
               type: "object",
               properties: {
+                runId: { type: "string" },
+                resourceId: { type: "string" },
                 inputData: { type: "object", additionalProperties: true },
-                runtimeContext: { type: "object", additionalProperties: true },
-                tracingOptions: { type: "object", additionalProperties: true }
+                resumeData: { type: "object", additionalProperties: true },
+                requestContext: { type: "object", additionalProperties: true },
+                tracingOptions: { type: "object", additionalProperties: true },
+                step: { type: "string" }
               }
             }
           }
@@ -1120,8 +1511,9 @@ function workflowRoute({
       }
     },
     handler: async (c) => {
-      const { inputData, ...rest } = await c.req.json();
+      const params = await c.req.json();
       const mastra = c.get("mastra");
+      const contextRequestContext = c.get("requestContext");
       let workflowToUse = workflow;
       if (!workflow) {
         const workflowId = c.req.param("workflowId");
@@ -1135,23 +1527,47 @@ function workflowRoute({
       if (!workflowToUse) {
         throw new Error("Workflow ID is required");
       }
-      const workflowObj = mastra.getWorkflow(workflowToUse);
-      if (!workflowObj) {
-        throw new Error(`Workflow ${workflowToUse} not found`);
+      if (contextRequestContext && params.requestContext) {
+        mastra.getLogger()?.warn(
+          `"requestContext" from the request body will be ignored because "requestContext" is already set in the route options.`
+        );
       }
-      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);
-          }
-        }
+      const uiMessageStream = await handleWorkflowStream({
+        mastra,
+        workflowId: workflowToUse,
+        params: {
+          ...params,
+          requestContext: contextRequestContext || params.requestContext
+        },
+        includeTextStreamParts
       });
       return ai.createUIMessageStreamResponse({ stream: uiMessageStream });
     }
   });
 }
+async function handleNetworkStream({
+  mastra,
+  agentId,
+  params,
+  defaultOptions
+}) {
+  const { messages, ...rest } = params;
+  const agentObj = mastra.getAgentById(agentId);
+  if (!agentObj) {
+    throw new Error(`Agent ${agentId} not found`);
+  }
+  const result = await agentObj.network(messages, {
+    ...defaultOptions,
+    ...rest
+  });
+  return ai.createUIMessageStream({
+    execute: async ({ writer }) => {
+      for await (const part of toAISdkV5Stream(result, { from: "network" })) {
+        writer.write(part);
+      }
+    }
+  });
+}
 function networkRoute({
   path = "/network/:agentId",
   agent,
@@ -1183,13 +1599,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"]
@@ -1213,7 +1628,7 @@ function networkRoute({
       }
     },
     handler: async (c) => {
-      const { messages, ...rest } = await c.req.json();
+      const params = await c.req.json();
       const mastra = c.get("mastra");
       let agentToUse = agent;
       if (!agent) {
@@ -1228,29 +1643,31 @@ function networkRoute({
       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);
-          }
-        }
+      const uiMessageStream = await handleNetworkStream({
+        mastra,
+        agentId: agentToUse,
+        params,
+        defaultOptions
       });
       return ai.createUIMessageStreamResponse({ stream: uiMessageStream });
     }
   });
 }
 
+// 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.'
+  );
+}
+
 exports.chatRoute = chatRoute;
+exports.handleChatStream = handleChatStream;
+exports.handleNetworkStream = handleNetworkStream;
+exports.handleWorkflowStream = handleWorkflowStream;
 exports.networkRoute = networkRoute;
 exports.toAISdkFormat = toAISdkFormat;
+exports.toAISdkStream = toAISdkV5Stream;
 exports.workflowRoute = workflowRoute;
 //# sourceMappingURL=index.cjs.map
 //# sourceMappingURL=index.cjs.map