ai 7.0.130 → 7.0.132

package/docs/07-reference/01-ai-sdk-core/index.mdx

@@ -147,6 +147,12 @@ It also contains the following helper functions:
       description: 'Creates a client for connecting to MCP servers.',
       href: '/docs/reference/ai-sdk-core/create-mcp-client',
     },
+    {
+      title: 'MCP Events',
+      description:
+        'Experimental event methods, webhook delivery, and subscription types.',
+      href: '/docs/reference/ai-sdk-core/mcp-events',
+    },
     {
       title: 'MCP Apps',
       description:

package/docs/07-reference/02-ai-sdk-ui/50-direct-chat-transport.mdx

@@ -77,6 +77,19 @@ export default function Chat() {
       description:
         'Extracts message metadata that will be sent to the client. Called on `start` and `finish` events.',
     },
+    {
+      name: 'onStepEnd',
+      type: 'UIMessageStreamOnStepEndCallback',
+      isOptional: true,
+      description:
+        'Called after each step is converted to UI message parts. Receives messages, isContinuation, and the accumulated responseMessage, including step metadata, for inspection or persistence.',
+    },
+    {
+      name: 'onStepFinish',
+      type: 'UIMessageStreamOnStepFinishCallback',
+      isOptional: true,
+      description: 'Deprecated alias for onStepEnd.',
+    },
     {
       name: 'sendReasoning',
       type: 'boolean',

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ai",
-  "version": "7.0.130",
+  "version": "7.0.132",
   "type": "module",
   "description": "AI SDK by Vercel - build apps like ChatGPT, Claude, Gemini, and more with a single interface for any model using the Vercel AI Gateway or go direct to OpenAI, Anthropic, Google, or any other model provider.",
   "license": "Apache-2.0",
@@ -42,20 +42,20 @@
     }
   },
   "dependencies": {
-    "@ai-sdk/gateway": "4.0.106",
+    "@ai-sdk/gateway": "4.0.107",
     "@ai-sdk/provider": "4.0.24",
     "@ai-sdk/provider-utils": "5.0.56"
   },
   "devDependencies": {
-    "@ai-sdk/amazon-bedrock": "5.0.108",
+    "@ai-sdk/amazon-bedrock": "5.0.110",
     "@ai-sdk/deepseek": "3.0.61",
-    "@ai-sdk/google": "4.0.90",
+    "@ai-sdk/google": "4.0.91",
     "@ai-sdk/groq": "4.0.57",
     "@ai-sdk/huggingface": "2.0.65",
     "@ai-sdk/moonshotai": "3.0.65",
-    "@ai-sdk/openai": "4.0.86",
+    "@ai-sdk/openai": "4.0.88",
     "@ai-sdk/test-server": "2.0.4",
-    "@ai-sdk/xai": "5.0.17",
+    "@ai-sdk/xai": "5.0.18",
     "@edge-runtime/vm": "^5.0.0",
     "@smithy/eventstream-codec": "^4.3.3",
     "@smithy/util-utf8": "^4.3.3",

package/src/agent/create-agent-ui-stream-response.ts

@@ -4,20 +4,18 @@ import type {
   Experimental_SandboxSession as SandboxSession,
   ToolSet,
 } from '@ai-sdk/provider-utils';
-import type {
-  GenerateTextOnStepEndCallback,
-  GenerateTextOnStepFinishCallback,
-} from '../generate-text/generate-text-events';
 import type { Output } from '../generate-text/output';
 import type { StreamTextTransform } from '../generate-text/stream-text';
-import type { UIMessageStreamOptions } from '../generate-text/stream-text-result';
 import type { TimeoutConfiguration } from '../prompt/request-options';
 import { createUIMessageStreamResponse } from '../ui-message-stream';
 import type { UIMessageStreamResponseInit } from '../ui-message-stream/ui-message-stream-response-init';
 import type { InferUITools, UIMessage, UIDataTypes } from '../ui/ui-messages';
 import type { Agent } from './agent';
 import type { convertToModelMessages } from '../ui/convert-to-model-messages';
-import { createAgentUIStream } from './create-agent-ui-stream';
+import {
+  createAgentUIStream,
+  type AgentUIMessageStreamOptions,
+} from './create-agent-ui-stream';
 
 /**
  * Runs the agent and returns a response object with a UI message stream.
@@ -32,6 +30,7 @@ import { createAgentUIStream } from './create-agent-ui-stream';
  * @param experimental_transform - Stream transformations. Optional.
  * @param onStepEnd - Callback that is called when each step ends. Optional.
  * @param onStepFinish - Deprecated alias for `onStepEnd`. Optional.
+ * @param onUIMessageStepEnd - Callback with the accumulated UI message after each step. Optional.
  * @param headers - Additional headers for the response. Optional.
  * @param status - The status code for the response. Optional.
  * @param statusText - The status text for the response. Optional.
@@ -73,11 +72,8 @@ export async function createAgentUIStreamResponse<
   experimental_sandbox?: SandboxSession;
   options?: CALL_OPTIONS;
   experimental_transform?: Arrayable<StreamTextTransform<TOOLS>>;
-  onStepEnd?: GenerateTextOnStepEndCallback<TOOLS>;
-  /** @deprecated Use `onStepEnd` instead. */
-  onStepFinish?: GenerateTextOnStepFinishCallback<TOOLS>;
 } & UIMessageStreamResponseInit &
-  UIMessageStreamOptions<UI_MESSAGE>): Promise<Response> {
+  AgentUIMessageStreamOptions<TOOLS, UI_MESSAGE>): Promise<Response> {
   return createUIMessageStreamResponse({
     headers,
     status,

package/src/agent/create-agent-ui-stream.ts

@@ -14,6 +14,7 @@ import type { StreamTextTransform } from '../generate-text/stream-text';
 import type { UIMessageStreamOptions } from '../generate-text/stream-text-result';
 import type { TimeoutConfiguration } from '../prompt/request-options';
 import type { InferUIMessageChunk } from '../ui-message-stream';
+import type { UIMessageStreamOnStepEndCallback } from '../ui-message-stream/ui-message-stream-on-step-end-callback';
 import { toUIMessageStream } from '../ui-message-stream/to-ui-message-stream';
 import { convertToModelMessages } from '../ui/convert-to-model-messages';
 import type {
@@ -29,6 +30,23 @@ import {
 } from '../util/async-iterable-stream';
 import type { Agent } from './agent';
 
+/**
+ * UI stream options with separate generation and UI message step callbacks.
+ */
+export type AgentUIMessageStreamOptions<
+  TOOLS extends ToolSet,
+  UI_MESSAGE extends UIMessage,
+> = Omit<UIMessageStreamOptions<UI_MESSAGE>, 'onStepEnd' | 'onStepFinish'> & {
+  onStepEnd?: GenerateTextOnStepEndCallback<TOOLS>;
+  /** @deprecated Use `onStepEnd` instead. */
+  onStepFinish?: GenerateTextOnStepFinishCallback<TOOLS>;
+  /**
+   * Called after each step is converted to UI message parts.
+   * Receives the accumulated UI message for inspection or persistence.
+   */
+  onUIMessageStepEnd?: UIMessageStreamOnStepEndCallback<UI_MESSAGE>;
+};
+
 /**
  * Runs the agent and stream the output as a UI message stream.
  *
@@ -42,6 +60,7 @@ import type { Agent } from './agent';
  * @param experimental_transform - The stream transformations. Optional.
  * @param onStepEnd - Callback that is called when each step ends. Optional.
  * @param onStepFinish - Deprecated alias for `onStepEnd`. Optional.
+ * @param onUIMessageStepEnd - Callback with the accumulated UI message after each step. Optional.
  *
  * @returns The UI message stream.
  */
@@ -67,6 +86,7 @@ export async function createAgentUIStream<
   experimental_transform,
   onStepEnd,
   onStepFinish,
+  onUIMessageStepEnd,
   ...uiMessageStreamOptions
 }: {
   agent: Agent<CALL_OPTIONS, TOOLS, RUNTIME_CONTEXT, OUTPUT>;
@@ -83,11 +103,8 @@ export async function createAgentUIStream<
   experimental_sandbox?: SandboxSession;
   options?: CALL_OPTIONS;
   experimental_transform?: Arrayable<StreamTextTransform<TOOLS>>;
-  onStepEnd?: GenerateTextOnStepEndCallback<TOOLS>;
-  /** @deprecated Use `onStepEnd` instead. */
-  onStepFinish?: GenerateTextOnStepFinishCallback<TOOLS>;
   // TODO `originalMessages` is part of this for bc, omit in v7
-} & UIMessageStreamOptions<UI_MESSAGE>): Promise<
+} & AgentUIMessageStreamOptions<TOOLS, UI_MESSAGE>): Promise<
   AsyncIterableStream<InferUIMessageChunk<UI_MESSAGE>>
 > {
   const validatedMessages = await validateUIMessagesForAgent<UI_MESSAGE>({
@@ -125,6 +142,7 @@ export async function createAgentUIStream<
     toUIMessageStream({
       ...uiMessageStreamOptions,
       originalMessages,
+      onStepEnd: onUIMessageStepEnd,
       stream: result.stream,
       tools: agent.tools,
     }),

package/src/agent/pipe-agent-ui-stream-to-response.ts

@@ -5,20 +5,18 @@ import type {
   ToolSet,
 } from '@ai-sdk/provider-utils';
 import type { ServerResponse } from 'node:http';
-import type {
-  GenerateTextOnStepEndCallback,
-  GenerateTextOnStepFinishCallback,
-} from '../generate-text/generate-text-events';
 import type { Output } from '../generate-text/output';
 import type { StreamTextTransform } from '../generate-text/stream-text';
-import type { UIMessageStreamOptions } from '../generate-text/stream-text-result';
 import type { TimeoutConfiguration } from '../prompt/request-options';
 import { pipeUIMessageStreamToResponse } from '../ui-message-stream';
 import type { UIMessageStreamResponseInit } from '../ui-message-stream/ui-message-stream-response-init';
 import type { InferUITools, UIMessage, UIDataTypes } from '../ui/ui-messages';
 import type { Agent } from './agent';
 import type { convertToModelMessages } from '../ui/convert-to-model-messages';
-import { createAgentUIStream } from './create-agent-ui-stream';
+import {
+  createAgentUIStream,
+  type AgentUIMessageStreamOptions,
+} from './create-agent-ui-stream';
 
 /**
  * Pipes the agent UI message stream to a Node.js ServerResponse object.
@@ -34,6 +32,7 @@ import { createAgentUIStream } from './create-agent-ui-stream';
  * @param experimental_transform - Stream transformations. Optional.
  * @param onStepEnd - Callback that is called when each step ends. Optional.
  * @param onStepFinish - Deprecated alias for `onStepEnd`. Optional.
+ * @param onUIMessageStepEnd - Callback with the accumulated UI message after each step. Optional.
  * @param headers - Additional headers for the response. Optional.
  * @param status - The status code for the response. Optional.
  * @param statusText - The status text for the response. Optional.
@@ -75,11 +74,8 @@ export async function pipeAgentUIStreamToResponse<
   experimental_sandbox?: SandboxSession;
   options?: CALL_OPTIONS;
   experimental_transform?: Arrayable<StreamTextTransform<TOOLS>>;
-  onStepEnd?: GenerateTextOnStepEndCallback<TOOLS>;
-  /** @deprecated Use `onStepEnd` instead. */
-  onStepFinish?: GenerateTextOnStepFinishCallback<TOOLS>;
 } & UIMessageStreamResponseInit &
-  UIMessageStreamOptions<UI_MESSAGE>): Promise<void> {
+  AgentUIMessageStreamOptions<TOOLS, UI_MESSAGE>): Promise<void> {
   return pipeUIMessageStreamToResponse({
     response,
     headers,

package/src/generate-text/stream-text-result.ts

@@ -12,6 +12,8 @@ import type { LanguageModelResponseMetadata } from '../types/language-model-resp
 import type { LanguageModelUsage } from '../types/usage';
 import type { InferUIMessageChunk } from '../ui-message-stream/ui-message-chunks';
 import type { UIMessageStreamOnEndCallback } from '../ui-message-stream/ui-message-stream-on-end-callback';
+import type { UIMessageStreamOnStepEndCallback } from '../ui-message-stream/ui-message-stream-on-step-end-callback';
+import type { UIMessageStreamOnStepFinishCallback } from '../ui-message-stream/ui-message-stream-on-step-finish-callback';
 import type { UIMessageStreamResponseInit } from '../ui-message-stream/ui-message-stream-response-init';
 import type { InferUIMessageMetadata, UIMessage } from '../ui/ui-messages';
 import type { AsyncIterableStream } from '../util/async-iterable-stream';
@@ -57,6 +59,15 @@ export type UIMessageStreamOptions<UI_MESSAGE extends UIMessage> = {
    */
   generateMessageId?: IdGenerator;
 
+  /**
+   * Called after each step is converted to UI message parts.
+   * Receives the accumulated UI message for inspection or persistence.
+   */
+  onStepEnd?: UIMessageStreamOnStepEndCallback<UI_MESSAGE>;
+
+  /** @deprecated Use `onStepEnd` instead. */
+  onStepFinish?: UIMessageStreamOnStepFinishCallback<UI_MESSAGE>;
+
   onEnd?: UIMessageStreamOnEndCallback<UI_MESSAGE>;
 
   /**

package/src/generate-text/stream-text.ts

@@ -3458,6 +3458,8 @@ class DefaultStreamTextResult<
   toUIMessageStream<UI_MESSAGE extends UIMessage>({
     originalMessages,
     generateMessageId,
+    onStepEnd,
+    onStepFinish,
     onEnd,
     onFinish,
     messageMetadata,
@@ -3475,6 +3477,7 @@ class DefaultStreamTextResult<
         tools: this.tools,
         originalMessages,
         generateMessageId,
+        onStepEnd: onStepEnd ?? onStepFinish,
         onEnd: onEnd ?? onFinish,
         messageMetadata,
         sendReasoning,
@@ -3491,6 +3494,8 @@ class DefaultStreamTextResult<
     {
       originalMessages,
       generateMessageId,
+      onStepEnd,
+      onStepFinish,
       onEnd,
       onFinish,
       messageMetadata,
@@ -3507,6 +3512,7 @@ class DefaultStreamTextResult<
       stream: this.toUIMessageStream({
         originalMessages,
         generateMessageId,
+        onStepEnd: onStepEnd ?? onStepFinish,
         onEnd: onEnd ?? onFinish,
         messageMetadata,
         sendReasoning,
@@ -3530,6 +3536,8 @@ class DefaultStreamTextResult<
   toUIMessageStreamResponse<UI_MESSAGE extends UIMessage>({
     originalMessages,
     generateMessageId,
+    onStepEnd,
+    onStepFinish,
     onEnd,
     onFinish,
     messageMetadata,
@@ -3545,6 +3553,7 @@ class DefaultStreamTextResult<
       stream: this.toUIMessageStream({
         originalMessages,
         generateMessageId,
+        onStepEnd: onStepEnd ?? onStepFinish,
         onEnd: onEnd ?? onFinish,
         messageMetadata,
         sendReasoning,

package/src/ui-message-stream/to-ui-message-stream.ts

@@ -14,7 +14,7 @@ import { toUIMessageChunk } from './to-ui-message-chunk';
  * Converts a stream of `TextStreamPart<TOOLS>` chunks (as emitted by
  * `streamText`'s `stream`) into a stream of `UIMessageChunk`s suitable for
  * UI message streaming, including response message ID injection and
- * `onEnd` handling.
+ * step and stream end callbacks.
  */
 export function toUIMessageStream<
   TOOLS extends ToolSet = ToolSet,
@@ -30,6 +30,8 @@ export function toUIMessageStream<
   messageMetadata,
   originalMessages,
   generateMessageId,
+  onStepEnd,
+  onStepFinish,
   onEnd,
   onFinish,
 }: {
@@ -130,7 +132,7 @@ export function toUIMessageStream<
             responseMessageId,
           });
 
-          if (uiMessageChunk != null) {
+          if (uiMessageChunk != null && part.type !== 'finish-step') {
             controller.enqueue(uiMessageChunk);
           }
 
@@ -147,6 +149,11 @@ export function toUIMessageStream<
             });
           }
 
+          // Apply step metadata before the boundary triggers a UI snapshot.
+          if (uiMessageChunk != null && part.type === 'finish-step') {
+            controller.enqueue(uiMessageChunk);
+          }
+
           if (part.type === 'finish') {
             setSourceOutcome({ status: 'completed' });
           } else if (part.type === 'abort') {
@@ -166,6 +173,7 @@ export function toUIMessageStream<
     stream: uiMessageChunkStream,
     messageId: responseMessageId ?? generateMessageId?.(),
     originalMessages,
+    onStepEnd: onStepEnd ?? onStepFinish,
     onEnd: onEnd ?? onFinish,
     onError,
     getOutcome: () => outcome,