ai 7.0.123 → 7.0.126

package/docs/03-ai-sdk-core/16-mcp-tools.mdx

@@ -178,6 +178,31 @@ transports by providing an `authProvider`.
 
 </Note>
 
+### Shared OAuth Credential Storage
+
+When multiple MCP clients or replicas share OAuth credentials, a rotating refresh
+token can be used concurrently. One refresh may save new tokens while another
+receives `invalid_grant` for the old refresh token.
+
+The optional `OAuthClientProvider.invalidateCredentials` method receives a
+second argument identifying the tokens used by a failed refresh:
+
+```typescript
+invalidateCredentials(
+  scope: 'all' | 'client' | 'tokens' | 'verifier',
+  context?: { tokens: OAuthTokens },
+): void | Promise<void>;
+```
+
+For `scope === 'tokens'` with a context, delete the credential only if the stored
+access and refresh tokens still match `context.tokens`. The comparison and
+deletion must be atomic, using a conditional database operation or a transaction.
+If they no longer match, leave the newer tokens in storage. The SDK retries
+authorization and reads the stored tokens again.
+
+The SDK also provides this context when invalidating stored tokens that lack an
+authorization server pin. Other invalidations can omit the context.
+
 ### OAuth Authorization Server Validation
 
 When using MCP OAuth in server-side applications, the MCP server can advertise

package/docs/03-ai-sdk-core/36-transcription.mdx

@@ -30,6 +30,42 @@ const language = transcript.language; // language of the transcript e.g. "en", i
 const durationInSeconds = transcript.durationInSeconds; // duration of the transcript in seconds, if available
 ```
 
+## Telemetry
+
+`transcribe` and `experimental_streamTranscribe` accept the standard
+`telemetry` option. Registered integrations receive start, end, and error
+events with the provider and model, audio byte length and media type, final
+transcript, warnings, response metadata, provider metadata, and usage when the
+provider reports it. Because streaming transcription is experimental, custom
+integrations receive its lifecycle through
+`experimental_onStreamTranscriptionStart` and
+`experimental_onStreamTranscriptionEnd`; errors continue to use `onError`.
+Integrations that wrap or delegate to another telemetry integration must
+forward these experimental callbacks explicitly for streaming transcription
+to be observed.
+
+```ts highlight="5-7"
+const transcript = await transcribe({
+  model: openai.transcription('gpt-4o-transcribe'),
+  audio: await readFile('audio.mp3'),
+  telemetry: {
+    functionId: 'transcribe-customer-call',
+  },
+});
+```
+
+Raw audio is not included in telemetry events. For streaming transcription,
+the final byte length is emitted only after the input stream is consumed and
+the transcription finishes. Use `recordInputs: false` or
+`recordOutputs: false` to prevent integrations from recording sensitive input
+or output fields.
+
+For URL audio, the downloaded media type is passed to the provider and reported
+on successful telemetry events. If downloading or decoding input audio fails,
+the start event is still emitted before the correlated error event, but its
+audio byte length and media type can be undefined because preparation did not
+complete.
+
 ## Streaming Transcription
 
 <Note type="warning">Streaming transcription is an experimental feature.</Note>
@@ -307,7 +343,8 @@ try {
 | [Groq](/providers/ai-sdk-providers/groq#transcription-models)                       | `whisper-large-v3`          |
 | [Mistral](/providers/ai-sdk-providers/mistral#transcription-models)                 | `voxtral-mini-latest`       |
 | [Azure OpenAI](/providers/ai-sdk-providers/azure#transcription-models)              | `whisper-1`                 |
-| [Azure Speech](/providers/ai-sdk-providers/azure#mai-transcribe-2)                  | `mai-transcribe-2`          |
+| [Azure Speech](/providers/ai-sdk-providers/azure#mai-transcribe)                    | `mai-transcribe-2`          |
+| [Azure Speech](/providers/ai-sdk-providers/azure#mai-transcribe)                    | `mai-transcribe-1.5`        |
 | [Azure OpenAI](/providers/ai-sdk-providers/azure#transcription-models)              | `gpt-4o-transcribe`         |
 | [Azure OpenAI](/providers/ai-sdk-providers/azure#transcription-models)              | `gpt-4o-mini-transcribe`    |
 | [Rev.ai](/providers/ai-sdk-providers/revai#transcription-models)                    | `machine`                   |

package/docs/03-ai-sdk-core/37-speech.mdx

@@ -19,6 +19,27 @@ const result = await generateSpeech({
 });
 ```
 
+## Telemetry
+
+`generateSpeech` accepts the standard `telemetry` option. Registered
+integrations receive start, end, and error events with the provider and model,
+source text, generated audio byte length, media type and format, warnings,
+response metadata, provider metadata, and usage when the provider reports it.
+Raw generated audio is not included in telemetry events.
+
+```ts highlight="5-7"
+const result = await generateSpeech({
+  model: openai.speech('gpt-4o-mini-tts'),
+  text: 'Hello from the AI SDK!',
+  telemetry: {
+    functionId: 'speak-greeting',
+  },
+});
+```
+
+Use `recordInputs: false` or `recordOutputs: false` when the text or output
+metadata should not be recorded by telemetry integrations.
+
 To access the generated audio:
 
 ```ts
@@ -137,6 +158,8 @@ try {
 | [OpenAI](/providers/ai-sdk-providers/openai#speech-models)               | `tts-1-hd`                          |
 | [OpenAI](/providers/ai-sdk-providers/openai#speech-models)               | `gpt-4o-mini-tts`                   |
 | [Mistral](/providers/ai-sdk-providers/mistral#speech-models)             | `voxtral-mini-tts-2603`             |
+| [Azure Speech](/providers/ai-sdk-providers/azure#mai-voice)              | `mai-voice-2-flash`                 |
+| [Azure Speech](/providers/ai-sdk-providers/azure#mai-voice)              | `mai-voice-2`                       |
 | [ElevenLabs](/providers/ai-sdk-providers/elevenlabs#speech-models)       | `eleven_v3`                         |
 | [ElevenLabs](/providers/ai-sdk-providers/elevenlabs#speech-models)       | `eleven_multilingual_v2`            |
 | [ElevenLabs](/providers/ai-sdk-providers/elevenlabs#speech-models)       | `eleven_flash_v2_5`                 |

package/docs/03-ai-sdk-core/60-telemetry.mdx

@@ -53,7 +53,10 @@ For Node.js applications (without Next.js), register the integration at the top
 
 ### Step 2: Enabling Telemetry
 
-Once a telemetry integration is registered, all AI SDK calls emit telemetry events by default.
+Once a telemetry integration is registered, supported AI SDK calls emit
+telemetry events by default. This includes text and object generation,
+embeddings, reranking, speech generation, and both non-streaming and streaming
+transcription.
 You can still pass `telemetry` to attach metadata (like `functionId`) or to opt out of a specific call:
 
 ```ts highlight="4-6"
@@ -71,6 +74,34 @@ By default, both inputs and outputs are recorded. You can disable them by settin
 Disabling the recording of inputs and outputs can be useful for privacy, data transfer, and performance reasons.
 You might for example want to disable recording inputs if they contain sensitive information.
 
+For `generateSpeech`, input telemetry contains the source text and output
+telemetry contains the generated audio byte length, media type, and format.
+For `transcribe` and `experimental_streamTranscribe`, telemetry contains the
+input audio byte length and media type plus the output transcript. Raw audio is
+never included in telemetry events. Streaming transcription reports its final
+byte length when the audio stream has been consumed and the transcription
+finishes.
+
+Custom integrations receive stable `generateSpeech` and `transcribe`
+operations through `onStart` and `onEnd`. To keep the experimental API isolated,
+`experimental_streamTranscribe` uses
+`experimental_onStreamTranscriptionStart` and
+`experimental_onStreamTranscriptionEnd`; all three operations report failures
+through `onError`.
+
+An integration that wraps or delegates to another integration must explicitly
+forward the experimental streaming transcription callbacks. Forwarding only
+the stable `onStart` and `onEnd` callbacks does not observe
+`experimental_streamTranscribe`.
+
+The built-in OpenTelemetry integrations record numeric provider usage by
+default, preserving each reported unit. The modern integration uses attributes
+such as `gen_ai.usage.characters`, `gen_ai.usage.seconds`, and
+`gen_ai.usage.input_tokens`; `LegacyOpenTelemetry` uses the corresponding
+`ai.usage.*` attributes. The provider-native usage object remains available to
+callbacks, and can also be included as the supplemental `ai.response.usage`
+attribute by enabling `usage: true`.
+
 ### Opting out
 
 Telemetry is opt-out. To disable telemetry for a specific call, set `isEnabled: false`:
@@ -550,6 +581,23 @@ For `rerank`, the integration records spans with `CLIENT` kind:
   - `gen_ai.provider.name`: the provider
   - `gen_ai.request.model`: the model ID
 
+#### generateSpeech / transcribe / experimental_streamTranscribe
+
+Each call records a single `CLIENT` span named `{operationId} {modelId}`, where
+`operationId` is `ai.generateSpeech`, `ai.transcribe`, or `ai.streamTranscribe`.
+
+- `gen_ai.output.type`: `"speech"` for speech generation and `"text"` for transcription.
+- `gen_ai.request.stream`: `true` for streaming transcription and `false` otherwise.
+- `gen_ai.input.messages`: speech generation input text as a user message with a text part, when `recordInputs` is enabled.
+- `gen_ai.output.messages`: the completed transcript as an assistant message with a text part, when `recordOutputs` is enabled. Streaming transcription records this once on completion.
+
+The existing `ai.request.text` and `ai.response.text` attributes are also retained.
+Audio byte length, media type, and output format remain in the
+`ai.request.audio.*` and `ai.response.audio.*` attributes, subject to the
+corresponding input/output recording controls. Raw audio is not captured, so
+transcription input and speech output do not populate message attributes with
+audio content or placeholder audio parts.
+
 #### experimental_evaluate
 
 For `experimental_evaluate`, the integration records two spans with `CLIENT`

package/docs/07-reference/01-ai-sdk-core/11-stream-transcribe.mdx

@@ -88,6 +88,13 @@ console.log(await result.text);
       description:
         'When true, the provider includes raw provider chunks in the stream as `raw` parts.',
     },
+    {
+      name: 'telemetry',
+      type: 'TelemetryOptions',
+      isOptional: true,
+      description:
+        'Telemetry configuration. Experimental lifecycle callbacks use experimental_onStreamTranscriptionStart and experimental_onStreamTranscriptionEnd. The end event is emitted after the audio stream is consumed and includes the final input byte length.',
+    },
   ]}
 />
 

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

@@ -64,6 +64,13 @@ console.log(transcript);
       isOptional: true,
       description: 'Additional HTTP headers for the request.',
     },
+    {
+      name: 'telemetry',
+      type: 'TelemetryOptions',
+      isOptional: true,
+      description:
+        'Telemetry configuration. Supports functionId, recordInputs, recordOutputs, isEnabled, and per-call integrations.',
+    },
     {
       name: 'download',
       type: '(options: { url: URL; abortSignal?: AbortSignal }) => Promise<{ data: Uint8Array; mediaType: string | undefined }>',

package/docs/07-reference/01-ai-sdk-core/12-generate-speech.mdx

@@ -124,6 +124,13 @@ const { audio } = await generateSpeech({
       isOptional: true,
       description: 'Additional HTTP headers for the request.',
     },
+    {
+      name: 'telemetry',
+      type: 'TelemetryOptions',
+      isOptional: true,
+      description:
+        'Telemetry configuration. Supports functionId, recordInputs, recordOutputs, isEnabled, and per-call integrations.',
+    },
   ]}
 />
 

package/docs/07-reference/01-ai-sdk-core/17-create-agent-ui-stream.mdx

@@ -59,6 +59,13 @@ export async function* streamAgent(
       description:
         'Array of input UI message objects (e.g., user/assistant/chat history). These will be validated and converted for the agent.',
     },
+    {
+      name: 'convertDataPart',
+      type: '(part: DataUIPart) => TextPart | FilePart | undefined',
+      isRequired: false,
+      description:
+        'Converts custom UI data parts into text or file parts for the model. Data parts are ignored when this callback is omitted or returns undefined.',
+    },
     {
       name: 'abortSignal',
       type: 'AbortSignal',

package/docs/07-reference/01-ai-sdk-core/18-create-agent-ui-stream-response.mdx

@@ -57,6 +57,13 @@ export async function POST(request: Request) {
       description:
         'Array of input UI messages provided to the agent (e.g., user and assistant messages).',
     },
+    {
+      name: 'convertDataPart',
+      type: '(part: DataUIPart) => TextPart | FilePart | undefined',
+      isRequired: false,
+      description:
+        'Converts custom UI data parts into text or file parts for the model. Data parts are ignored when this callback is omitted or returns undefined.',
+    },
     {
       name: 'abortSignal',
       type: 'AbortSignal',

package/docs/07-reference/01-ai-sdk-core/18-pipe-agent-ui-stream-to-response.mdx

@@ -61,6 +61,13 @@ export async function handler(req, res) {
       description:
         'Array of input UI messages sent to the agent (such as user/assistant message objects).',
     },
+    {
+      name: 'convertDataPart',
+      type: '(part: DataUIPart) => TextPart | FilePart | undefined',
+      isRequired: false,
+      description:
+        'Converts custom UI data parts into text or file parts for the model. Data parts are ignored when this callback is omitted or returns undefined.',
+    },
     {
       name: 'abortSignal',
       type: 'AbortSignal',

package/docs/07-reference/04-ai-sdk-workflow/01-workflow-agent.mdx

@@ -120,7 +120,7 @@ To see `WorkflowAgent` in action, check out [these examples](#examples).
       type: 'ToolCallRepairFunction',
       isOptional: true,
       description:
-        'Default function to repair tool calls that fail to parse. Per-stream values override this default.',
+        "Default function to repair tool calls that fail to parse. Mark the function with 'use step' when used in a workflow. Per-stream values override this default.",
     },
     {
       name: 'experimental_download',
@@ -573,13 +573,13 @@ const result = await agent.stream({
       name: 'repairToolCall',
       type: 'ToolCallRepairFunction',
       isOptional: true,
-      description: 'Callback to attempt automatic recovery when a tool call cannot be parsed.',
+      description: "Callback to attempt automatic recovery when a tool call cannot be parsed. Mark the function with 'use step' when used in a workflow.",
     },
     {
       name: 'experimental_transform',
       type: 'StreamTextTransform | Array<StreamTextTransform>',
       isOptional: true,
-      description: 'Stream transformations applied in order. Must maintain the stream structure.',
+      description: "Factories for stream transformations, applied in order to raw provider events before model processing and writable output. Each factory must return a TransformStream synchronously and maintain the stream structure. Mark each factory with 'use step' when used in a workflow.",
     },
     {
       name: 'experimental_download',
@@ -750,6 +750,63 @@ Returns a `Promise<WorkflowAgentStreamResult>` with the following properties:
   ]}
 />
 
+### `repairToolCall` in a workflow
+
+`repairToolCall` runs inside the model step. When you use it in a workflow,
+define the callback as a function with a `'use step'` directive. You can set
+it on the constructor or pass it to `stream()`.
+
+Workflow saves a reference to the registered function and loads its
+implementation inside the model step. Keep any values captured by the
+function serializable. An ordinary inline callback causes a serialization
+error when it crosses this step boundary.
+
+The callback can be asynchronous. Return the repaired tool call, or `null`
+if the repair cannot be completed. See [tool call repair](/docs/ai-sdk-core/tools-and-tool-calling#tool-call-repair)
+for repair strategies.
+
+### `experimental_transform` in a workflow
+
+Pass a transform factory, or an array of factories, to `stream()`.
+Mark each factory with `'use step'` so Workflow can load it inside the model
+step. The factory must return a `TransformStream` synchronously. Keep any
+captured values serializable.
+
+```ts
+import type { LanguageModelV4StreamPart } from '@ai-sdk/provider';
+import { WorkflowAgent } from '@ai-sdk/workflow';
+import { getWritable } from 'workflow';
+
+function uppercaseText() {
+  'use step';
+
+  return new TransformStream<
+    LanguageModelV4StreamPart,
+    LanguageModelV4StreamPart
+  >({
+    transform(part, controller) {
+      controller.enqueue(
+        part.type === 'text-delta'
+          ? { ...part, delta: part.delta.toUpperCase() }
+          : part,
+      );
+    },
+  });
+}
+
+export async function uppercaseWorkflow() {
+  'use workflow';
+
+  const agent = new WorkflowAgent({ model: 'openai/gpt-5' });
+
+  return agent.stream({
+    messages: [{ role: 'user', content: 'Write a short greeting.' }],
+    writable: getWritable(),
+    experimental_transform: uppercaseText,
+  });
+}
+```
+
 ## Utilities
 
 ### `createModelCallToUIChunkTransform(options?)`

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ai",
-  "version": "7.0.123",
+  "version": "7.0.126",
   "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.101",
-    "@ai-sdk/provider": "4.0.20",
-    "@ai-sdk/provider-utils": "5.0.52"
+    "@ai-sdk/gateway": "4.0.102",
+    "@ai-sdk/provider": "4.0.21",
+    "@ai-sdk/provider-utils": "5.0.53"
   },
   "devDependencies": {
-    "@ai-sdk/amazon-bedrock": "5.0.102",
-    "@ai-sdk/deepseek": "3.0.57",
-    "@ai-sdk/google": "4.0.86",
-    "@ai-sdk/groq": "4.0.53",
-    "@ai-sdk/huggingface": "2.0.60",
-    "@ai-sdk/moonshotai": "3.0.61",
-    "@ai-sdk/openai": "4.0.82",
+    "@ai-sdk/amazon-bedrock": "5.0.104",
+    "@ai-sdk/deepseek": "3.0.58",
+    "@ai-sdk/google": "4.0.87",
+    "@ai-sdk/groq": "4.0.54",
+    "@ai-sdk/huggingface": "2.0.62",
+    "@ai-sdk/moonshotai": "3.0.62",
+    "@ai-sdk/openai": "4.0.83",
     "@ai-sdk/test-server": "2.0.3",
-    "@ai-sdk/xai": "5.0.13",
+    "@ai-sdk/xai": "5.0.14",
     "@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

@@ -14,8 +14,9 @@ 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 } from '../ui/ui-messages';
+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';
 
 /**
@@ -23,6 +24,7 @@ import { createAgentUIStream } from './create-agent-ui-stream';
  *
  * @param agent - The agent to run.
  * @param uiMessages - The input UI messages.
+ * @param convertDataPart - Optional function to convert custom data parts to text or file model message parts.
  * @param abortSignal - Abort signal. Optional.
  * @param timeout - Timeout in milliseconds. Optional.
  * @param experimental_sandbox - The sandbox environment that is passed through to tool execution. Optional.
@@ -44,6 +46,11 @@ export async function createAgentUIStreamResponse<
   RUNTIME_CONTEXT extends Context = Context,
   OUTPUT extends Output = never,
   MESSAGE_METADATA = unknown,
+  UI_MESSAGE extends UIMessage<
+    MESSAGE_METADATA,
+    UIDataTypes,
+    InferUITools<TOOLS>
+  > = UIMessage<MESSAGE_METADATA, UIDataTypes, InferUITools<TOOLS>>,
 >({
   headers,
   status,
@@ -54,6 +61,13 @@ export async function createAgentUIStreamResponse<
 }: {
   agent: Agent<CALL_OPTIONS, TOOLS, RUNTIME_CONTEXT, OUTPUT>;
   uiMessages: unknown[];
+  /**
+   * Converts custom UI data parts to text or file model message parts.
+   * Data parts are ignored when omitted or when the callback returns undefined.
+   */
+  convertDataPart?: NonNullable<
+    Parameters<typeof convertToModelMessages<UI_MESSAGE>>[1]
+  >['convertDataPart'];
   abortSignal?: AbortSignal;
   timeout?: TimeoutConfiguration<TOOLS>;
   experimental_sandbox?: SandboxSession;
@@ -63,9 +77,7 @@ export async function createAgentUIStreamResponse<
   /** @deprecated Use `onStepEnd` instead. */
   onStepFinish?: GenerateTextOnStepFinishCallback<TOOLS>;
 } & UIMessageStreamResponseInit &
-  UIMessageStreamOptions<
-    UIMessage<MESSAGE_METADATA, never, InferUITools<TOOLS>>
-  >): Promise<Response> {
+  UIMessageStreamOptions<UI_MESSAGE>): Promise<Response> {
   return createUIMessageStreamResponse({
     headers,
     status,

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

@@ -20,6 +20,7 @@ import type {
   InferUIMessageTools,
   InferUITools,
   UIMessage,
+  UIDataTypes,
 } from '../ui/ui-messages';
 import { validateUIMessagesForAgent } from '../ui/validate-ui-messages';
 import {
@@ -33,6 +34,7 @@ import type { Agent } from './agent';
  *
  * @param agent - The agent to run.
  * @param uiMessages - The input UI messages.
+ * @param convertDataPart - Optional function to convert custom data parts to text or file model message parts.
  * @param abortSignal - The abort signal. Optional.
  * @param timeout - Timeout in milliseconds. Optional.
  * @param experimental_sandbox - The sandbox environment that is passed through to tool execution. Optional.
@@ -49,11 +51,15 @@ export async function createAgentUIStream<
   RUNTIME_CONTEXT extends Context = Context,
   OUTPUT extends Output = never,
   MESSAGE_METADATA = unknown,
-  UI_MESSAGE extends UIMessage<MESSAGE_METADATA, never, InferUITools<TOOLS>> =
-    UIMessage<MESSAGE_METADATA, never, InferUITools<TOOLS>>,
+  UI_MESSAGE extends UIMessage<
+    MESSAGE_METADATA,
+    UIDataTypes,
+    InferUITools<TOOLS>
+  > = UIMessage<MESSAGE_METADATA, UIDataTypes, InferUITools<TOOLS>>,
 >({
   agent,
   uiMessages,
+  convertDataPart,
   options,
   abortSignal,
   timeout,
@@ -65,6 +71,13 @@ export async function createAgentUIStream<
 }: {
   agent: Agent<CALL_OPTIONS, TOOLS, RUNTIME_CONTEXT, OUTPUT>;
   uiMessages: unknown[];
+  /**
+   * Converts custom UI data parts to text or file model message parts.
+   * Data parts are ignored when omitted or when the callback returns undefined.
+   */
+  convertDataPart?: NonNullable<
+    Parameters<typeof convertToModelMessages<UI_MESSAGE>>[1]
+  >['convertDataPart'];
   abortSignal?: AbortSignal;
   timeout?: TimeoutConfiguration<TOOLS>;
   experimental_sandbox?: SandboxSession;
@@ -91,6 +104,7 @@ export async function createAgentUIStream<
 
   const modelMessages = await convertToModelMessages(validatedMessages, {
     tools: agent.tools,
+    convertDataPart,
   });
 
   const result = await agent.stream({

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

@@ -15,8 +15,9 @@ 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 } from '../ui/ui-messages';
+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';
 
 /**
@@ -25,6 +26,7 @@ import { createAgentUIStream } from './create-agent-ui-stream';
  * @param response - The Node.js ServerResponse object to pipe to.
  * @param agent - The agent to run.
  * @param uiMessages - The input UI messages.
+ * @param convertDataPart - Optional function to convert custom data parts to text or file model message parts.
  * @param abortSignal - Abort signal. Optional.
  * @param timeout - Timeout in milliseconds. Optional.
  * @param experimental_sandbox - The sandbox environment that is passed through to tool execution. Optional.
@@ -44,6 +46,11 @@ export async function pipeAgentUIStreamToResponse<
   RUNTIME_CONTEXT extends Context = Context,
   OUTPUT extends Output = never,
   MESSAGE_METADATA = unknown,
+  UI_MESSAGE extends UIMessage<
+    MESSAGE_METADATA,
+    UIDataTypes,
+    InferUITools<TOOLS>
+  > = UIMessage<MESSAGE_METADATA, UIDataTypes, InferUITools<TOOLS>>,
 >({
   response,
   headers,
@@ -56,6 +63,13 @@ export async function pipeAgentUIStreamToResponse<
   response: ServerResponse;
   agent: Agent<CALL_OPTIONS, TOOLS, RUNTIME_CONTEXT, OUTPUT>;
   uiMessages: unknown[];
+  /**
+   * Converts custom UI data parts to text or file model message parts.
+   * Data parts are ignored when omitted or when the callback returns undefined.
+   */
+  convertDataPart?: NonNullable<
+    Parameters<typeof convertToModelMessages<UI_MESSAGE>>[1]
+  >['convertDataPart'];
   abortSignal?: AbortSignal;
   timeout?: TimeoutConfiguration<TOOLS>;
   experimental_sandbox?: SandboxSession;
@@ -65,9 +79,7 @@ export async function pipeAgentUIStreamToResponse<
   /** @deprecated Use `onStepEnd` instead. */
   onStepFinish?: GenerateTextOnStepFinishCallback<TOOLS>;
 } & UIMessageStreamResponseInit &
-  UIMessageStreamOptions<
-    UIMessage<MESSAGE_METADATA, never, InferUITools<TOOLS>>
-  >): Promise<void> {
+  UIMessageStreamOptions<UI_MESSAGE>): Promise<void> {
   return pipeUIMessageStreamToResponse({
     response,
     headers,