@mastra/ai-sdk 0.0.0-cloud-storage-adapter-20251106204059 → 0.0.0-cloud-604-map-nested-flow-details-to-side-panel-20251212192149

package/dist/index.js

@@ -1,6 +1,9 @@
 import { registerApiRoute } from '@mastra/core/server';
-import { createUIMessageStream, createUIMessageStreamResponse } from 'ai';
-import { DefaultGeneratedFile, DefaultGeneratedFileWithType } from '@mastra/core/stream';
+import { createUIMessageStream, createUIMessageStreamResponse, wrapLanguageModel } from 'ai';
+import { convertFullStreamChunkToMastra, DefaultGeneratedFile, DefaultGeneratedFileWithType } from '@mastra/core/stream';
+import { TripWire, MessageList } from '@mastra/core/agent';
+import { RequestContext } from '@mastra/core/di';
+import { WorkingMemory, MessageHistory, SemanticRecall } from '@mastra/core/processors';
 
 // src/chat-route.ts
 
@@ -8,6 +11,35 @@ 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-approval",
+    "tool-call-suspended",
+    "tool-call",
+    "tool-result",
+    "tool-error",
+    "error",
+    "start-step",
+    "finish-step",
+    "start",
+    "finish",
+    "abort",
+    "tool-input-end",
+    "object",
+    "tripwire",
+    "raw"
+  ].includes(chunk.type);
+};
 function safeParseErrorObject(obj) {
   if (typeof obj !== "object" || obj === null) {
     return String(obj);
@@ -30,6 +62,12 @@ var isWorkflowExecutionDataChunkType = (chunk) => {
 };
 
 // src/helpers.ts
+function toAISDKFinishReason(reason) {
+  if (reason === "tripwire" || reason === "retry") {
+    return "other";
+  }
+  return reason;
+}
 function convertMastraChunkToAISDKv5({
   chunk,
   mode = "stream"
@@ -54,7 +92,7 @@ function convertMastraChunkToAISDKv5({
     case "finish": {
       return {
         type: "finish",
-        finishReason: chunk.payload.stepResult.reason,
+        finishReason: toAISDKFinishReason(chunk.payload.stepResult.reason),
         totalUsage: chunk.payload.output.usage
       };
     }
@@ -138,6 +176,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",
@@ -171,7 +231,7 @@ function convertMastraChunkToAISDKv5({
           ...rest2
         },
         usage: chunk.payload.output.usage,
-        finishReason: chunk.payload.stepResult.reason,
+        finishReason: toAISDKFinishReason(chunk.payload.stepResult.reason),
         providerMetadata
       };
     }
@@ -228,6 +288,16 @@ function convertMastraChunkToAISDKv5({
         type: "object",
         object: chunk.object
       };
+    case "tripwire":
+      return {
+        type: "data-tripwire",
+        data: {
+          reason: chunk.payload.reason,
+          retry: chunk.payload.retry,
+          metadata: chunk.payload.metadata,
+          processorId: chunk.payload.processorId
+        }
+      };
     default:
       if (chunk.type && "payload" in chunk && chunk.payload) {
         return {
@@ -283,6 +353,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": {
@@ -300,6 +378,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": {
@@ -357,6 +454,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;
     }
@@ -382,21 +487,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;
@@ -423,7 +530,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) {
@@ -437,7 +547,7 @@ function WorkflowStreamToAISDKTransformer() {
       });
     },
     transform(chunk, controller) {
-      const transformed = transformWorkflow(chunk, bufferedWorkflows);
+      const transformed = transformWorkflow(chunk, bufferedWorkflows, false, includeTextStreamParts);
       if (transformed) controller.enqueue(transformed);
     }
   });
@@ -461,20 +571,37 @@ function AgentNetworkToAISDKTransformer() {
     }
   });
 }
-function AgentStreamToAISDKTransformer(lastMessageId) {
+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,
+        sendReasoning,
+        sendSources,
+        messageMetadataValue: messageMetadata?.({ part }),
+        sendStart,
+        sendFinish,
         responseMessageId: lastMessageId,
         onError(error) {
-          return safeParseErrorObject(error);
+          return onError ? onError(error) : safeParseErrorObject(error);
         }
       });
       if (transformedChunk) {
@@ -494,6 +621,14 @@ function AgentStreamToAISDKTransformer(lastMessageId) {
           controller.enqueue(transformedChunk);
         }
       }
+    },
+    flush(controller) {
+      if (tripwireOccurred && !finishEventSent && sendFinish) {
+        controller.enqueue({
+          type: "finish",
+          finishReason: "other"
+        });
+      }
     }
   });
 }
@@ -633,7 +768,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, {
@@ -726,6 +861,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)) {
@@ -745,12 +903,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,
@@ -781,14 +956,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 {
@@ -801,14 +981,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 {
@@ -821,14 +1006,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 {
@@ -843,14 +1035,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,
@@ -865,14 +1056,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,
@@ -886,14 +1076,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,
@@ -908,12 +1097,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
         }
@@ -947,32 +1148,86 @@ function transformNetwork(payload, bufferedNetworks, isNested) {
       };
     }
     default: {
-      if (isDataChunkType(payload)) {
-        if (!("data" in 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;
+        const { type, data } = payload.payload;
+        return { type, data };
       }
-      if (isAgentExecutionDataChunkType(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;
+        const { type, data } = payload.payload;
+        return { type, data };
       }
-      if (isWorkflowExecutionDataChunkType(payload)) {
-        if (!("data" in payload.payload)) {
+      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;
+        }
+        bufferedNetworks.set(payload.runId, current);
+        return {
+          type: isNested ? "data-tool-network" : "data-network",
+          id: payload.runId,
+          data: {
+            ...current,
+            status: "running"
+          }
+        };
+      }
+      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;
+          }
+        }
+        bufferedNetworks.set(payload.runId, current);
+        return {
+          type: isNested ? "data-tool-network" : "data-network",
+          id: payload.runId,
+          data: {
+            ...current,
+            status: "running"
+          }
+        };
+      }
+      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.payload;
+        const { type, data } = payload;
+        return { type, data };
       }
       return null;
     }
@@ -980,23 +1235,95 @@ function transformNetwork(payload, bufferedNetworks, isNested) {
 }
 
 // src/convert-streams.ts
-function toAISdkV5Stream(stream, options = { from: "agent" }) {
+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(options?.lastMessageId));
+  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 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");
@@ -1025,6 +1352,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",
@@ -1095,9 +1430,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 requestContext = c.get("requestContext");
+      const contextRequestContext = c.get("requestContext");
       let agentToUse = agent;
       if (!agent) {
         const agentId = c.req.param("agentId");
@@ -1108,32 +1443,24 @@ function chatRoute({
           `Fixed agent ID was set together with an agentId path parameter. This can lead to unexpected behavior.`
         );
       }
-      if (requestContext && defaultOptions?.requestContext) {
+      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,
-        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 toAISdkV5Stream(result, { from: "agent", lastMessageId })) {
-            writer.write(part);
-          }
-        }
+      const uiMessageStream = await handleChatStream({
+        mastra,
+        agentId: agentToUse,
+        params: {
+          ...params,
+          requestContext: contextRequestContext || params.requestContext
+        },
+        defaultOptions,
+        sendStart,
+        sendFinish,
+        sendReasoning,
+        sendSources
       });
       return createUIMessageStreamResponse({
         stream: uiMessageStream
@@ -1141,9 +1468,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 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");
@@ -1170,9 +1519,13 @@ function workflowRoute({
             schema: {
               type: "object",
               properties: {
+                runId: { type: "string" },
+                resourceId: { type: "string" },
                 inputData: { type: "object", additionalProperties: true },
+                resumeData: { type: "object", additionalProperties: true },
                 requestContext: { type: "object", additionalProperties: true },
-                tracingOptions: { type: "object", additionalProperties: true }
+                tracingOptions: { type: "object", additionalProperties: true },
+                step: { type: "string" }
               }
             }
           }
@@ -1190,8 +1543,9 @@ function workflowRoute({
       }
     },
     handler: async (c) => {
-      const { inputData, resumeData, ...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");
@@ -1205,23 +1559,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.createRun();
-      const stream = resumeData ? run.resumeStream({ resumeData, ...rest }) : run.stream({ inputData, ...rest });
-      const uiMessageStream = createUIMessageStream({
-        execute: async ({ writer }) => {
-          for await (const part of toAISdkV5Stream(stream, { from: "workflow" })) {
-            writer.write(part);
-          }
-        }
+      const uiMessageStream = await handleWorkflowStream({
+        mastra,
+        workflowId: workflowToUse,
+        params: {
+          ...params,
+          requestContext: contextRequestContext || params.requestContext
+        },
+        includeTextStreamParts
       });
       return 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 createUIMessageStream({
+    execute: async ({ writer }) => {
+      for await (const part of toAISdkV5Stream(result, { from: "network" })) {
+        writer.write(part);
+      }
+    }
+  });
+}
 function networkRoute({
   path = "/network/:agentId",
   agent,
@@ -1282,7 +1660,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) {
@@ -1297,25 +1675,465 @@ 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 uiMessageStream = await handleNetworkStream({
+        mastra,
+        agentId: agentToUse,
+        params,
+        defaultOptions
+      });
+      return createUIMessageStreamResponse({ stream: uiMessageStream });
+    }
+  });
+}
+function withMastra(model, options = {}) {
+  const { memory, inputProcessors = [], outputProcessors = [] } = options;
+  const allInputProcessors = [...inputProcessors];
+  const allOutputProcessors = [...outputProcessors];
+  if (memory) {
+    const { storage, lastMessages, semanticRecall, workingMemory } = memory;
+    const isWorkingMemoryEnabled = typeof workingMemory === "object" && workingMemory.enabled !== false;
+    if (isWorkingMemoryEnabled && typeof workingMemory === "object") {
+      let template;
+      if (workingMemory.template) {
+        template = {
+          format: "markdown",
+          content: workingMemory.template
+        };
+      }
+      const workingMemoryProcessor = new WorkingMemory({
+        storage,
+        template,
+        scope: workingMemory.scope,
+        useVNext: "version" in workingMemory && workingMemory.version === "vnext"
+      });
+      allInputProcessors.push(workingMemoryProcessor);
+    }
+    if (lastMessages !== false && lastMessages !== void 0) {
+      const messageHistory = new MessageHistory({
+        storage,
+        lastMessages: typeof lastMessages === "number" ? lastMessages : void 0
+      });
+      allInputProcessors.push(messageHistory);
+      allOutputProcessors.push(messageHistory);
+    }
+    if (semanticRecall) {
+      const { vector, embedder, indexName, ...semanticConfig } = semanticRecall;
+      const semanticRecallProcessor = new SemanticRecall({
+        storage,
+        vector,
+        embedder,
+        indexName: indexName || "memory_messages",
+        ...semanticConfig
+      });
+      allInputProcessors.push(semanticRecallProcessor);
+      allOutputProcessors.push(semanticRecallProcessor);
+    }
+  }
+  return wrapLanguageModel({
+    model,
+    middleware: createProcessorMiddleware({
+      inputProcessors: allInputProcessors,
+      outputProcessors: allOutputProcessors,
+      memory: memory ? {
+        threadId: memory.threadId,
+        resourceId: memory.resourceId
+      } : void 0
+    })
+  });
+}
+function createProcessorMiddleware(options) {
+  const { inputProcessors = [], outputProcessors = [], memory } = options;
+  const requestContext = new RequestContext();
+  if (memory) {
+    requestContext.set("MastraMemory", {
+      thread: memory.threadId ? { id: memory.threadId } : void 0,
+      resourceId: memory.resourceId,
+      memoryConfig: memory.config
+    });
+  }
+  return {
+    middlewareVersion: "v2",
+    /**
+     * Transform params runs input processors (processInput)
+     */
+    async transformParams({ params }) {
+      const messageList = new MessageList({
+        threadId: memory?.threadId,
+        resourceId: memory?.resourceId
+      });
+      for (const msg of params.prompt) {
+        if (msg.role === "system") {
+          messageList.addSystem(msg.content);
+        } else {
+          messageList.add(msg, "input");
+        }
+      }
+      for (const processor of inputProcessors) {
+        if (processor.processInput) {
+          try {
+            await processor.processInput({
+              messages: messageList.get.input.db(),
+              systemMessages: messageList.getAllSystemMessages(),
+              messageList,
+              requestContext,
+              abort: (reason) => {
+                throw new TripWire(reason || "Aborted by processor");
+              }
+            });
+          } catch (error) {
+            if (error instanceof TripWire) {
+              return {
+                ...params,
+                providerOptions: {
+                  ...params.providerOptions,
+                  mastraProcessors: {
+                    tripwire: true,
+                    reason: error.message
+                  }
+                }
+              };
+            }
+            throw error;
+          }
+        }
+      }
+      const newPrompt = messageList.get.all.aiV5.prompt().map(MessageList.aiV5ModelMessageToV2PromptMessage);
+      return {
+        ...params,
+        prompt: newPrompt
+      };
+    },
+    /**
+     * Wrap generate for non-streaming output processing
+     */
+    async wrapGenerate({ doGenerate, params }) {
+      const processorState = params.providerOptions?.mastraProcessors;
+      if (processorState?.tripwire) {
+        const reason = processorState.reason || "Blocked by processor";
+        return {
+          content: [{ type: "text", text: reason }],
+          finishReason: "stop",
+          usage: { inputTokens: 0, outputTokens: 0, totalTokens: 0 },
+          warnings: [{ type: "other", message: `Tripwire: ${reason}` }]
+        };
       }
-      const result = await agentObj.network(messages, {
-        ...defaultOptions,
-        ...rest
+      const result = await doGenerate();
+      if (!outputProcessors.length) return result;
+      const messageList = new MessageList({
+        threadId: memory?.threadId,
+        resourceId: memory?.resourceId
       });
-      const uiMessageStream = createUIMessageStream({
-        execute: async ({ writer }) => {
-          for await (const part of toAISdkV5Stream(result, { from: "network" })) {
-            writer.write(part);
+      for (const msg of params.prompt) {
+        if (msg.role === "system") {
+          messageList.addSystem(msg.content);
+        } else {
+          messageList.add(msg, "input");
+        }
+      }
+      const textContent = result.content.filter((c) => c.type === "text").map((c) => c.text).join("");
+      const responseMessage = {
+        id: crypto.randomUUID(),
+        role: "assistant",
+        content: {
+          format: 2,
+          parts: [{ type: "text", text: textContent }]
+        },
+        createdAt: /* @__PURE__ */ new Date(),
+        ...memory?.threadId && { threadId: memory.threadId },
+        ...memory?.resourceId && { resourceId: memory.resourceId }
+      };
+      messageList.add(responseMessage, "response");
+      for (const processor of outputProcessors) {
+        if (processor.processOutputResult) {
+          try {
+            await processor.processOutputResult({
+              messages: messageList.get.all.db(),
+              messageList,
+              requestContext,
+              abort: (reason) => {
+                throw new TripWire(reason || "Aborted by processor");
+              }
+            });
+          } catch (error) {
+            if (error instanceof TripWire) {
+              return {
+                content: [{ type: "text", text: error.message }],
+                finishReason: "stop",
+                usage: result.usage,
+                warnings: [{ type: "other", message: `Output blocked: ${error.message}` }]
+              };
+            }
+            throw error;
           }
         }
+      }
+      const processedText = messageList.get.response.db().map((m) => extractTextFromMastraMessage(m)).join("");
+      return {
+        ...result,
+        content: [{ type: "text", text: processedText }]
+      };
+    },
+    /**
+     * Wrap stream for streaming output processing
+     */
+    async wrapStream({ doStream, params }) {
+      const processorState = params.providerOptions?.mastraProcessors;
+      if (processorState?.tripwire) {
+        const reason = processorState.reason || "Blocked by processor";
+        return {
+          stream: createBlockedStream(reason)
+        };
+      }
+      const { stream, ...rest } = await doStream();
+      if (!outputProcessors.length) return { stream, ...rest };
+      const processorStates = /* @__PURE__ */ new Map();
+      const runId = crypto.randomUUID();
+      const transformedStream = stream.pipeThrough(
+        new TransformStream({
+          async transform(chunk, controller) {
+            let mastraChunk = convertFullStreamChunkToMastra(
+              chunk,
+              { runId }
+            );
+            if (!mastraChunk) {
+              controller.enqueue(chunk);
+              return;
+            }
+            for (const processor of outputProcessors) {
+              if (processor.processOutputStream && mastraChunk) {
+                let state = processorStates.get(processor.id);
+                if (!state) {
+                  state = { streamParts: [], customState: {} };
+                  processorStates.set(processor.id, state);
+                }
+                state.streamParts.push(mastraChunk);
+                try {
+                  const result = await processor.processOutputStream({
+                    part: mastraChunk,
+                    streamParts: state.streamParts,
+                    state: state.customState,
+                    requestContext,
+                    abort: (reason) => {
+                      throw new TripWire(reason || "Aborted by processor");
+                    }
+                  });
+                  if (result === null || result === void 0) {
+                    mastraChunk = void 0;
+                  } else {
+                    mastraChunk = result;
+                  }
+                } catch (error) {
+                  if (error instanceof TripWire) {
+                    controller.enqueue({
+                      type: "error",
+                      error: new Error(error.message)
+                    });
+                    controller.terminate();
+                    return;
+                  }
+                  throw error;
+                }
+              }
+            }
+            if (mastraChunk) {
+              const aiChunk = convertMastraChunkToAISDKStreamPart(mastraChunk);
+              if (aiChunk) {
+                controller.enqueue(aiChunk);
+              }
+            }
+          }
+        })
+      );
+      return { stream: transformedStream, ...rest };
+    }
+  };
+}
+function createBlockedStream(reason) {
+  return new ReadableStream({
+    start(controller) {
+      const id = crypto.randomUUID();
+      controller.enqueue({
+        type: "text-start",
+        id
       });
-      return createUIMessageStreamResponse({ stream: uiMessageStream });
+      controller.enqueue({
+        type: "text-delta",
+        id,
+        delta: reason
+      });
+      controller.enqueue({
+        type: "text-end",
+        id
+      });
+      controller.enqueue({
+        type: "finish",
+        finishReason: "stop",
+        usage: { inputTokens: 0, outputTokens: 0, totalTokens: 0 }
+      });
+      controller.close();
     }
   });
 }
+function extractTextFromMastraMessage(msg) {
+  const content = msg.content;
+  if (typeof content === "string") {
+    return content;
+  }
+  if (content?.parts) {
+    return content.parts.filter((p) => p.type === "text" && "text" in p).map((p) => p.text).join("");
+  }
+  return "";
+}
+function convertMastraChunkToAISDKStreamPart(chunk) {
+  switch (chunk.type) {
+    // Text streaming
+    case "text-start":
+      return {
+        type: "text-start",
+        id: chunk.payload.id || crypto.randomUUID(),
+        providerMetadata: chunk.payload.providerMetadata
+      };
+    case "text-delta":
+      return {
+        type: "text-delta",
+        id: chunk.payload.id || crypto.randomUUID(),
+        delta: chunk.payload.text,
+        providerMetadata: chunk.payload.providerMetadata
+      };
+    case "text-end":
+      return {
+        type: "text-end",
+        id: chunk.payload.id || crypto.randomUUID(),
+        providerMetadata: chunk.payload.providerMetadata
+      };
+    // Reasoning streaming
+    case "reasoning-start":
+      return {
+        type: "reasoning-start",
+        id: chunk.payload.id || crypto.randomUUID(),
+        providerMetadata: chunk.payload.providerMetadata
+      };
+    case "reasoning-delta":
+      return {
+        type: "reasoning-delta",
+        id: chunk.payload.id || crypto.randomUUID(),
+        delta: chunk.payload.text,
+        providerMetadata: chunk.payload.providerMetadata
+      };
+    case "reasoning-end":
+      return {
+        type: "reasoning-end",
+        id: chunk.payload.id || crypto.randomUUID(),
+        providerMetadata: chunk.payload.providerMetadata
+      };
+    // Tool call (complete)
+    case "tool-call":
+      return {
+        type: "tool-call",
+        toolCallId: chunk.payload.toolCallId,
+        toolName: chunk.payload.toolName,
+        input: JSON.stringify(chunk.payload.args),
+        providerExecuted: chunk.payload.providerExecuted,
+        providerMetadata: chunk.payload.providerMetadata
+      };
+    // Tool call input streaming
+    case "tool-call-input-streaming-start":
+      return {
+        type: "tool-input-start",
+        id: chunk.payload.toolCallId,
+        toolName: chunk.payload.toolName,
+        providerExecuted: chunk.payload.providerExecuted,
+        providerMetadata: chunk.payload.providerMetadata
+      };
+    case "tool-call-delta":
+      return {
+        type: "tool-input-delta",
+        id: chunk.payload.toolCallId,
+        delta: chunk.payload.argsTextDelta,
+        providerMetadata: chunk.payload.providerMetadata
+      };
+    case "tool-call-input-streaming-end":
+      return {
+        type: "tool-input-end",
+        id: chunk.payload.toolCallId,
+        providerMetadata: chunk.payload.providerMetadata
+      };
+    // Tool result
+    case "tool-result":
+      return {
+        type: "tool-result",
+        toolCallId: chunk.payload.toolCallId,
+        toolName: chunk.payload.toolName,
+        result: { type: "json", value: chunk.payload.result },
+        isError: chunk.payload.isError,
+        providerExecuted: chunk.payload.providerExecuted,
+        providerMetadata: chunk.payload.providerMetadata
+      };
+    // Source (citations)
+    case "source":
+      if (chunk.payload.sourceType === "url") {
+        return {
+          type: "source",
+          sourceType: "url",
+          id: chunk.payload.id,
+          url: chunk.payload.url,
+          title: chunk.payload.title,
+          providerMetadata: chunk.payload.providerMetadata
+        };
+      } else {
+        return {
+          type: "source",
+          sourceType: "document",
+          id: chunk.payload.id,
+          mediaType: chunk.payload.mimeType,
+          title: chunk.payload.title,
+          filename: chunk.payload.filename,
+          providerMetadata: chunk.payload.providerMetadata
+        };
+      }
+    // File output
+    case "file":
+      return {
+        type: "file",
+        data: chunk.payload.data || chunk.payload.base64,
+        mediaType: chunk.payload.mimeType
+      };
+    // Response metadata
+    case "response-metadata":
+      return {
+        type: "response-metadata",
+        ...chunk.payload
+      };
+    // Raw provider data
+    case "raw":
+      return {
+        type: "raw",
+        rawValue: chunk.payload
+      };
+    // Finish
+    case "finish": {
+      const usage = chunk.payload.output?.usage;
+      return {
+        type: "finish",
+        finishReason: toAISDKFinishReason(chunk.payload.stepResult?.reason || "stop"),
+        usage: usage ? {
+          inputTokens: usage.inputTokens || 0,
+          outputTokens: usage.outputTokens || 0,
+          totalTokens: usage.totalTokens || 0
+        } : { inputTokens: 0, outputTokens: 0, totalTokens: 0 },
+        providerMetadata: chunk.payload.metadata?.providerMetadata
+      };
+    }
+    // Error
+    case "error":
+      return {
+        type: "error",
+        error: chunk.payload.error || chunk.payload
+      };
+    default:
+      return null;
+  }
+}
 
 // src/to-ai-sdk-format.ts
 function toAISdkFormat() {
@@ -1324,6 +2142,6 @@ function toAISdkFormat() {
   );
 }
 
-export { chatRoute, networkRoute, toAISdkFormat, toAISdkV5Stream as toAISdkStream, workflowRoute };
+export { chatRoute, handleChatStream, handleNetworkStream, handleWorkflowStream, networkRoute, toAISdkFormat, toAISdkV5Stream as toAISdkStream, withMastra, workflowRoute };
 //# sourceMappingURL=index.js.map
 //# sourceMappingURL=index.js.map