@mastra/ai-sdk 0.2.7 → 0.3.1

package/CHANGELOG.md

@@ -1,5 +1,80 @@
 # @mastra/ai-sdk
 
+## 0.3.1
+
+### Patch Changes
+
+- Added support for tripwire data chunks in streaming responses. ([#10291](https://github.com/mastra-ai/mastra/pull/10291))
+
+  Tripwire chunks allow the AI SDK to emit special data events when certain conditions are triggered during stream processing. These chunks include a `tripwireReason` field explaining why the tripwire was activated.
+
+  #### Usage
+
+  When converting Mastra chunks to AI SDK v5 format, tripwire chunks are now automatically handled:
+
+  ```typescript
+  // Tripwire chunks are converted to data-tripwire format
+  const chunk = {
+    type: 'tripwire',
+    payload: { tripwireReason: 'Rate limit approaching' }
+  };
+
+  // Converts to:
+  {
+    type: 'data-tripwire',
+    data: { tripwireReason: 'Rate limit approaching' }
+  }
+  ```
+
+- Updated dependencies [[`7491cc0`](https://github.com/mastra-ai/mastra/commit/7491cc0350b2ba067f98c4915bf607119bd0150f), [`0d10ac7`](https://github.com/mastra-ai/mastra/commit/0d10ac7b8efa03c2f0c330eb2520148bfa6091e9), [`e3e899c`](https://github.com/mastra-ai/mastra/commit/e3e899c650f4c435445303bd97a66f5840a52a1e)]:
+  - @mastra/core@0.24.3
+
+## 0.3.1-alpha.0
+
+### Patch Changes
+
+- Added support for tripwire data chunks in streaming responses. ([#10291](https://github.com/mastra-ai/mastra/pull/10291))
+
+  Tripwire chunks allow the AI SDK to emit special data events when certain conditions are triggered during stream processing. These chunks include a `tripwireReason` field explaining why the tripwire was activated.
+
+  #### Usage
+
+  When converting Mastra chunks to AI SDK v5 format, tripwire chunks are now automatically handled:
+
+  ```typescript
+  // Tripwire chunks are converted to data-tripwire format
+  const chunk = {
+    type: 'tripwire',
+    payload: { tripwireReason: 'Rate limit approaching' }
+  };
+
+  // Converts to:
+  {
+    type: 'data-tripwire',
+    data: { tripwireReason: 'Rate limit approaching' }
+  }
+  ```
+
+- Updated dependencies [[`7491cc0`](https://github.com/mastra-ai/mastra/commit/7491cc0350b2ba067f98c4915bf607119bd0150f), [`0d10ac7`](https://github.com/mastra-ai/mastra/commit/0d10ac7b8efa03c2f0c330eb2520148bfa6091e9), [`e3e899c`](https://github.com/mastra-ai/mastra/commit/e3e899c650f4c435445303bd97a66f5840a52a1e)]:
+  - @mastra/core@0.24.3-alpha.0
+
+## 0.3.0
+
+### Minor Changes
+
+- Add sendStart, sendFinish, sendReasoning, and sendSources options to toAISdkV5Stream function, allowing fine-grained control over which message chunks are included in the converted stream. Previously, these values were hardcoded in the transformer. ([#10250](https://github.com/mastra-ai/mastra/pull/10250))
+
+  BREAKING CHANGE: AgentStreamToAISDKTransformer now accepts an options object instead of a single lastMessageId parameter
+
+  Also, add sendStart, sendFinish, sendReasoning, and sendSources parameters to
+  chatRoute function, enabling fine-grained control over which chunks are
+  included in the AI SDK stream output.
+
+### Patch Changes
+
+- Updated dependencies [[`eebb7bb`](https://github.com/mastra-ai/mastra/commit/eebb7bb407c57342a3be3a2efbe68c696589feeb), [`c6e6d07`](https://github.com/mastra-ai/mastra/commit/c6e6d071ecf346a80dceab410af2c567c7e66a57), [`0e6df8f`](https://github.com/mastra-ai/mastra/commit/0e6df8f66340992cb1b319834657deb17368de52), [`6295fd7`](https://github.com/mastra-ai/mastra/commit/6295fd783d49075d5bf2c18ff9486e94d36aaa56), [`d813cf7`](https://github.com/mastra-ai/mastra/commit/d813cf7251695d85cc60469058384ffa74974484)]:
+  - @mastra/core@0.24.2
+
 ## 0.2.7
 
 ### Patch Changes

package/dist/chat-route.d.ts

@@ -9,6 +9,56 @@ export type chatRouteOptions<OUTPUT extends OutputSchema = undefined> = {
 } | {
     path: string;
     agent: string;
-});
-export declare function chatRoute<OUTPUT extends OutputSchema = undefined>({ path, agent, defaultOptions, }: chatRouteOptions<OUTPUT>): ReturnType<typeof registerApiRoute>;
+}) & {
+    sendStart?: boolean;
+    sendFinish?: boolean;
+    sendReasoning?: boolean;
+    sendSources?: boolean;
+};
+/**
+ * Creates a chat route handler for streaming agent conversations using the AI SDK format.
+ *
+ * This function registers an HTTP POST endpoint that accepts messages, executes an agent,
+ * and streams the response back to the client in AI SDK v5 compatible format.
+ * *
+ * @param {chatRouteOptions} options - Configuration options for the chat route
+ * @param {string} [options.path='/chat/:agentId'] - The route path. Include `:agentId` for dynamic routing
+ * @param {string} [options.agent] - Fixed agent ID when not using dynamic routing
+ * @param {AgentExecutionOptions} [options.defaultOptions] - Default options passed to agent execution
+ * @param {boolean} [options.sendStart=true] - Whether to send start events in the stream
+ * @param {boolean} [options.sendFinish=true] - Whether to send finish events in the stream
+ * @param {boolean} [options.sendReasoning=false] - Whether to include reasoning steps in the stream
+ * @param {boolean} [options.sendSources=false] - Whether to include source citations in the stream
+ *
+ * @returns {ReturnType<typeof registerApiRoute>} A registered API route handler
+ *
+ * @throws {Error} When path doesn't include `:agentId` and no fixed agent is specified
+ * @throws {Error} When agent ID is missing at runtime
+ * @throws {Error} When specified agent is not found in Mastra instance
+ *
+ * @example
+ * // Dynamic agent routing
+ * chatRoute({
+ *   path: '/chat/:agentId',
+ *   sendReasoning: true,
+ * });
+ *
+ * @example
+ * // Fixed agent with custom path
+ * chatRoute({
+ *   path: '/api/support-chat',
+ *   agent: 'support-agent',
+ *   defaultOptions: {
+ *     maxSteps: 5,
+ *   },
+ * });
+ *
+ * @remarks
+ * - The route handler expects a JSON body with a `messages` array
+ * - Messages should follow the format: `{ role: 'user' | 'assistant' | 'system', content: string }`
+ * - The response is a Server-Sent Events (SSE) stream compatible with AI SDK v5
+ * - If both `agent` and `:agentId` are present, a warning is logged and the fixed `agent` takes precedence
+ * - Request context from the incoming request overrides `defaultOptions.requestContext` if both are present
+ */
+export declare function chatRoute<OUTPUT extends OutputSchema = undefined>({ path, agent, defaultOptions, sendStart, sendFinish, sendReasoning, sendSources, }: chatRouteOptions<OUTPUT>): ReturnType<typeof registerApiRoute>;
 //# sourceMappingURL=chat-route.d.ts.map

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

@@ -1 +1 @@
-{"version":3,"file":"chat-route.d.ts","sourceRoot":"","sources":["../src/chat-route.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,qBAAqB,EAAE,MAAM,oBAAoB,CAAC;AAEhE,OAAO,EAAE,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AACvD,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AAIxD,MAAM,MAAM,gBAAgB,CAAC,MAAM,SAAS,YAAY,GAAG,SAAS,IAAI;IACtE,cAAc,CAAC,EAAE,qBAAqB,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACzD,GAAG,CACA;IACE,IAAI,EAAE,GAAG,MAAM,WAAW,MAAM,EAAE,CAAC;IACnC,KAAK,CAAC,EAAE,KAAK,CAAC;CACf,GACD;IACE,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,MAAM,CAAC;CACf,CACJ,CAAC;AAEF,wBAAgB,SAAS,CAAC,MAAM,SAAS,YAAY,GAAG,SAAS,EAAE,EACjE,IAAuB,EACvB,KAAK,EACL,cAAc,GACf,EAAE,gBAAgB,CAAC,MAAM,CAAC,GAAG,UAAU,CAAC,OAAO,gBAAgB,CAAC,CA6JhE"}
+{"version":3,"file":"chat-route.d.ts","sourceRoot":"","sources":["../src/chat-route.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,qBAAqB,EAAE,MAAM,oBAAoB,CAAC;AAEhE,OAAO,EAAE,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AACvD,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AAIxD,MAAM,MAAM,gBAAgB,CAAC,MAAM,SAAS,YAAY,GAAG,SAAS,IAAI;IACtE,cAAc,CAAC,EAAE,qBAAqB,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACzD,GAAG,CACA;IACE,IAAI,EAAE,GAAG,MAAM,WAAW,MAAM,EAAE,CAAC;IACnC,KAAK,CAAC,EAAE,KAAK,CAAC;CACf,GACD;IACE,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,MAAM,CAAC;CACf,CACJ,GAAG;IACA,SAAS,CAAC,EAAE,OAAO,CAAC;IACpB,UAAU,CAAC,EAAE,OAAO,CAAC;IACrB,aAAa,CAAC,EAAE,OAAO,CAAC;IACxB,WAAW,CAAC,EAAE,OAAO,CAAC;CACvB,CAAC;AAEJ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA4CG;AACH,wBAAgB,SAAS,CAAC,MAAM,SAAS,YAAY,GAAG,SAAS,EAAE,EACjE,IAAuB,EACvB,KAAK,EACL,cAAc,EACd,SAAgB,EAChB,UAAiB,EACjB,aAAqB,EACrB,WAAmB,GACpB,EAAE,gBAAgB,CAAC,MAAM,CAAC,GAAG,UAAU,CAAC,OAAO,gBAAgB,CAAC,CAoKhE"}

package/dist/helpers.d.ts

@@ -1,7 +1,7 @@
 import type { ChunkType } from '@mastra/core';
 import type { PartialSchemaOutput, OutputSchema, DataChunkType } from '@mastra/core/stream';
 import type { InferUIMessageChunk, ObjectStreamPart, TextStreamPart, ToolSet, UIMessage } from 'ai';
-export type OutputChunkType<OUTPUT extends OutputSchema = undefined> = TextStreamPart<ToolSet> | ObjectStreamPart<PartialSchemaOutput<OUTPUT>> | undefined;
+export type OutputChunkType<OUTPUT extends OutputSchema = undefined> = TextStreamPart<ToolSet> | ObjectStreamPart<PartialSchemaOutput<OUTPUT>> | DataChunkType | undefined;
 export type ToolAgentChunkType = {
     type: 'tool-agent';
     toolCallId: string;

package/dist/helpers.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"helpers.d.ts","sourceRoot":"","sources":["../src/helpers.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,cAAc,CAAC;AAE9C,OAAO,KAAK,EAAE,mBAAmB,EAAE,YAAY,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AAE5F,OAAO,KAAK,EAAE,mBAAmB,EAAE,gBAAgB,EAAE,cAAc,EAAE,OAAO,EAAE,SAAS,EAAE,MAAM,IAAI,CAAC;AAGpG,MAAM,MAAM,eAAe,CAAC,MAAM,SAAS,YAAY,GAAG,SAAS,IAC/D,cAAc,CAAC,OAAO,CAAC,GACvB,gBAAgB,CAAC,mBAAmB,CAAC,MAAM,CAAC,CAAC,GAC7C,SAAS,CAAC;AAEd,MAAM,MAAM,kBAAkB,GAAG;IAAE,IAAI,EAAE,YAAY,CAAC;IAAC,UAAU,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,GAAG,CAAA;CAAE,CAAC;AAC1F,MAAM,MAAM,qBAAqB,GAAG;IAAE,IAAI,EAAE,eAAe,CAAC;IAAC,UAAU,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,GAAG,CAAA;CAAE,CAAC;AAChG,MAAM,MAAM,oBAAoB,GAAG;IAAE,IAAI,EAAE,cAAc,CAAC;IAAC,UAAU,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,GAAG,CAAA;CAAE,CAAC;AAE9F,wBAAgB,2BAA2B,CAAC,MAAM,SAAS,YAAY,GAAG,SAAS,EAAE,EACnF,KAAK,EACL,IAAe,GAChB,EAAE;IACD,KAAK,EAAE,SAAS,CAAC,MAAM,CAAC,CAAC;IACzB,IAAI,CAAC,EAAE,UAAU,GAAG,QAAQ,CAAC;CAC9B,GAAG,eAAe,CAAC,MAAM,CAAC,CAqN1B;AAED,wBAAgB,uCAAuC,CAAC,UAAU,SAAS,SAAS,EAAE,EACpF,IAAI,EACJ,oBAAoB,EACpB,aAAa,EACb,WAAW,EACX,OAAO,EACP,SAAS,EACT,UAAU,EACV,iBAAiB,GAClB,EAAE;IAED,IAAI,EAAE,cAAc,CAAC,OAAO,CAAC,GAAG,aAAa,GAAG;QAAE,IAAI,EAAE,aAAa,CAAC;QAAC,UAAU,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,GAAG,CAAA;KAAE,CAAC;IACzG,oBAAoB,CAAC,EAAE,OAAO,CAAC;IAC/B,aAAa,CAAC,EAAE,OAAO,CAAC;IACxB,WAAW,CAAC,EAAE,OAAO,CAAC;IACtB,OAAO,EAAE,CAAC,KAAK,EAAE,OAAO,KAAK,MAAM,CAAC;IACpC,SAAS,CAAC,EAAE,OAAO,CAAC;IACpB,UAAU,CAAC,EAAE,OAAO,CAAC;IACrB,iBAAiB,CAAC,EAAE,MAAM,CAAC;CAC5B,GAAG,mBAAmB,CAAC,UAAU,CAAC,GAAG,kBAAkB,GAAG,qBAAqB,GAAG,oBAAoB,GAAG,SAAS,CAwOlH"}
+{"version":3,"file":"helpers.d.ts","sourceRoot":"","sources":["../src/helpers.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,cAAc,CAAC;AAE9C,OAAO,KAAK,EAAE,mBAAmB,EAAE,YAAY,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AAE5F,OAAO,KAAK,EAAE,mBAAmB,EAAE,gBAAgB,EAAE,cAAc,EAAE,OAAO,EAAE,SAAS,EAAE,MAAM,IAAI,CAAC;AAGpG,MAAM,MAAM,eAAe,CAAC,MAAM,SAAS,YAAY,GAAG,SAAS,IAC/D,cAAc,CAAC,OAAO,CAAC,GACvB,gBAAgB,CAAC,mBAAmB,CAAC,MAAM,CAAC,CAAC,GAC7C,aAAa,GACb,SAAS,CAAC;AAEd,MAAM,MAAM,kBAAkB,GAAG;IAAE,IAAI,EAAE,YAAY,CAAC;IAAC,UAAU,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,GAAG,CAAA;CAAE,CAAC;AAC1F,MAAM,MAAM,qBAAqB,GAAG;IAAE,IAAI,EAAE,eAAe,CAAC;IAAC,UAAU,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,GAAG,CAAA;CAAE,CAAC;AAChG,MAAM,MAAM,oBAAoB,GAAG;IAAE,IAAI,EAAE,cAAc,CAAC;IAAC,UAAU,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,GAAG,CAAA;CAAE,CAAC;AAE9F,wBAAgB,2BAA2B,CAAC,MAAM,SAAS,YAAY,GAAG,SAAS,EAAE,EACnF,KAAK,EACL,IAAe,GAChB,EAAE;IACD,KAAK,EAAE,SAAS,CAAC,MAAM,CAAC,CAAC;IACzB,IAAI,CAAC,EAAE,UAAU,GAAG,QAAQ,CAAC;CAC9B,GAAG,eAAe,CAAC,MAAM,CAAC,CAiP1B;AAED,wBAAgB,uCAAuC,CAAC,UAAU,SAAS,SAAS,EAAE,EACpF,IAAI,EACJ,oBAAoB,EACpB,aAAa,EACb,WAAW,EACX,OAAO,EACP,SAAS,EACT,UAAU,EACV,iBAAiB,GAClB,EAAE;IAED,IAAI,EAAE,cAAc,CAAC,OAAO,CAAC,GAAG,aAAa,GAAG;QAAE,IAAI,EAAE,aAAa,CAAC;QAAC,UAAU,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,GAAG,CAAA;KAAE,CAAC;IACzG,oBAAoB,CAAC,EAAE,OAAO,CAAC;IAC/B,aAAa,CAAC,EAAE,OAAO,CAAC;IACxB,WAAW,CAAC,EAAE,OAAO,CAAC;IACtB,OAAO,EAAE,CAAC,KAAK,EAAE,OAAO,KAAK,MAAM,CAAC;IACpC,SAAS,CAAC,EAAE,OAAO,CAAC;IACpB,UAAU,CAAC,EAAE,OAAO,CAAC;IACrB,iBAAiB,CAAC,EAAE,MAAM,CAAC;CAC5B,GAAG,mBAAmB,CAAC,UAAU,CAAC,GAAG,kBAAkB,GAAG,qBAAqB,GAAG,oBAAoB,GAAG,SAAS,CAwOlH"}

package/dist/index.cjs

@@ -126,6 +126,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",
@@ -216,6 +238,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 {
@@ -271,6 +300,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": {
@@ -288,6 +325,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": {
@@ -378,21 +434,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;
@@ -457,17 +515,31 @@ function AgentNetworkToAISDKTransformer() {
     }
   });
 }
-function AgentStreamToAISDKTransformer(lastMessageId) {
+function AgentStreamToAISDKTransformer({
+  lastMessageId,
+  sendStart,
+  sendFinish,
+  sendReasoning,
+  sendSources
+}) {
   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,
+        sendStart,
+        sendFinish,
         responseMessageId: lastMessageId,
         onError() {
           return "Error";
@@ -490,6 +562,14 @@ function AgentStreamToAISDKTransformer(lastMessageId) {
           controller.enqueue(transformedChunk);
         }
       }
+    },
+    flush(controller) {
+      if (tripwireOccurred && !finishEventSent && sendFinish) {
+        controller.enqueue({
+          type: "finish",
+          finishReason: "other"
+        });
+      }
     }
   });
 }
@@ -976,7 +1056,11 @@ function transformNetwork(payload, bufferedNetworks, isNested) {
 }
 
 // src/to-ai-sdk-format.ts
-function toAISdkFormat(stream, options = { from: "agent" }) {
+function toAISdkFormat(stream, options = {
+  from: "agent",
+  sendStart: true,
+  sendFinish: true
+}) {
   const from = options?.from;
   if (from === "workflow") {
     return stream.pipeThrough(WorkflowStreamToAISDKTransformer());
@@ -985,14 +1069,26 @@ function toAISdkFormat(stream, options = { from: "agent" }) {
     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
+    })
+  );
 }
 
 // src/chat-route.ts
 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");
@@ -1126,7 +1222,14 @@ function chatRoute({
       const uiMessageStream = ai.createUIMessageStream({
         originalMessages: messages,
         execute: async ({ writer }) => {
-          for await (const part of toAISdkFormat(result, { from: "agent", lastMessageId })) {
+          for await (const part of toAISdkFormat(result, {
+            from: "agent",
+            lastMessageId,
+            sendStart,
+            sendFinish,
+            sendReasoning,
+            sendSources
+          })) {
             writer.write(part);
           }
         }
@@ -1170,7 +1273,7 @@ function workflowRoute({
                 resourceId: { type: "string" },
                 inputData: { type: "object", additionalProperties: true },
                 resumeData: { type: "object", additionalProperties: true },
-                requestContext: { type: "object", additionalProperties: true },
+                runtimeContext: { type: "object", additionalProperties: true },
                 tracingOptions: { type: "object", additionalProperties: true },
                 step: { type: "string" }
               }