ai 6.0.296 → 6.0.298

package/dist/internal/index.js

@@ -164,7 +164,7 @@ function detectMediaType({
 var import_provider_utils2 = require("@ai-sdk/provider-utils");
 
 // src/version.ts
-var VERSION = true ? "6.0.296" : "0.0.0-test";
+var VERSION = true ? "6.0.298" : "0.0.0-test";
 
 // src/util/download/download.ts
 var download = async ({

package/dist/internal/index.mjs

@@ -144,7 +144,7 @@ import {
 } from "@ai-sdk/provider-utils";
 
 // src/version.ts
-var VERSION = true ? "6.0.296" : "0.0.0-test";
+var VERSION = true ? "6.0.298" : "0.0.0-test";
 
 // src/util/download/download.ts
 var download = async ({

package/docs/04-ai-sdk-ui/02-chatbot.mdx

@@ -728,7 +728,16 @@ With `streamText`, you can control how error messages and usage information are
 
 By default, the error message is masked for security reasons.
 The default error message is "An error occurred."
-You can forward error messages or send your own error message by providing a `getErrorMessage` function:
+
+Tool execution errors marked `providerExecuted: true` are an exception. They
+bypass the UI stream's `onError` callback: strings are forwarded unchanged, and
+other error values are serialized with `JSON.stringify`. This preserves provider
+error codes when chat history is converted back into model messages. Stream
+errors and invalid tool calls still use `onError`, even for provider-executed
+tools. Do not rely on this callback to redact provider-executed tool execution
+errors before they reach the client.
+
+You can forward error messages or send your own error message by providing an `onError` function:
 
 ```ts filename="app/api/chat/route.ts" highlight="13-27"
 import { convertToModelMessages, streamText, UIMessage } from 'ai';

package/docs/04-ai-sdk-ui/03-chatbot-message-persistence.mdx

@@ -92,6 +92,14 @@ export async function loadChat(id: string): Promise<UIMessage[]> {
 
 When processing messages on the server that contain tool calls, custom metadata, or data parts, you should validate them using `validateUIMessages` before sending them to the model.
 
+### Persisting streaming tool input
+
+An `input-streaming` tool part includes `rawInput` while tool arguments are
+still arriving. Persist the complete UI message, including this field, so a
+resumed stream can append later tool-input deltas to the exact text received
+before the interruption. AI SDK updates `rawInput` as each delta arrives and
+removes it when the tool input becomes available.
+
 ### Validation with tools
 
 When your messages include tool calls, validate them against your tool definitions:

package/docs/04-ai-sdk-ui/03-chatbot-resume-streams.mdx

@@ -90,6 +90,10 @@ export function Chat({
 
 When you enable `resume`, the `useChat` hook makes a `GET` request to `/api/chat/[id]/stream` on mount to check for and resume any active streams.
 
+The default transport URL-encodes the chat ID as a single path segment. IDs equal
+to `.` or `..` are rejected with an `InvalidArgumentError` because URL parsers
+interpret them as path traversal segments.
+
 Let's start by creating the POST handler to create the resumable stream.
 
 ### 2. Create the POST handler
@@ -215,6 +219,9 @@ The diagram above shows the complete lifecycle of a resumable stream:
 
 By default, the `useChat` hook makes a GET request to `/api/chat/[id]/stream` when resuming. Customize this endpoint, credentials, and headers, using the `prepareReconnectToStreamRequest` option in `DefaultChatTransport`:
 
+The callback receives the original, unencoded ID. Its returned `api` URL is used
+verbatim, so encode and validate any ID you include in a custom URL yourself.
+
 ```tsx filename="app/chat/[chatId]/chat.tsx"
 import { useChat } from '@ai-sdk/react';
 import { DefaultChatTransport } from 'ai';
@@ -228,10 +235,10 @@ export function Chat({ chatData, resume }) {
       // Customize reconnect settings (optional)
       prepareReconnectToStreamRequest: ({ id }) => {
         return {
-          api: `/api/chat/${id}/stream`, // Default pattern
+          api: `/api/chat/${encodeURIComponent(id)}/stream`,
           // Or use a different pattern:
-          // api: `/api/streams/${id}/resume`,
-          // api: `/api/resume-chat?id=${id}`,
+          // api: `/api/streams/${encodeURIComponent(id)}/resume`,
+          // api: `/api/resume-chat?id=${encodeURIComponent(id)}`,
           credentials: 'include', // Include cookies/auth
           headers: {
             Authorization: 'Bearer token',

package/docs/04-ai-sdk-ui/03-chatbot-tool-usage.mdx

@@ -682,7 +682,16 @@ export async function POST(req: Request) {
 Language models can make errors when calling tools.
 By default, these errors are masked for security reasons, and show up as "An error occurred" in the UI.
 
-To surface the errors, you can use the `onError` function when calling `toUIMessageResponse`.
+Tool execution errors marked `providerExecuted: true` bypass this masking and the
+UI stream's `onError` callback. String errors pass through unchanged; other error
+values are serialized with `JSON.stringify`. The original error data is needed
+when converting persisted UI messages back into model messages, for example to
+preserve Anthropic tool error codes. The callback still handles invalid tool
+calls and stream errors, regardless of whether a tool is provider-executed.
+
+To customize masked errors, pass `onError` to `result.toUIMessageStream()` or
+`result.toUIMessageStreamResponse()`. This callback does not redact
+provider-executed tool execution errors sent to the client.
 
 ```tsx
 export function errorHandler(error: unknown) {

package/docs/07-reference/01-ai-sdk-core/02-stream-text.mdx

@@ -3559,7 +3559,7 @@ To see `streamText` in action, check out [these examples](#examples).
               type: '(error: unknown) => string',
               isOptional: true,
               description:
-                'Process an error, e.g. to log it. Returns error message to include in the data stream. Defaults to () => "An error occurred."',
+                'Process an error, e.g. to log it. Returns the error message to include in the data stream. Defaults to () => "An error occurred." Tool execution errors marked providerExecuted: true bypass this callback: strings pass through unchanged and other values are JSON-stringified to preserve provider error data for model-message round trips. Stream errors and invalid tool calls still use this callback.',
             },
             {
               name: 'consumeSseStream',

package/docs/07-reference/02-ai-sdk-ui/41-create-ui-message-stream-response.mdx

@@ -102,6 +102,13 @@ const response = createUIMessageStreamResponse({
       isOptional: true,
       description: 'Additional headers for the response.',
     },
+    {
+      name: 'keepAliveMs',
+      type: 'number',
+      isOptional: true,
+      description:
+        'Interval in milliseconds for sending SSE keep-alive comments. When set, an opening comment is sent immediately and additional comments are sent after the stream has been idle for the configured interval.',
+    },
     {
       name: 'consumeSseStream',
       type: '(options: { stream: ReadableStream<string> }) => PromiseLike<void> | void',

package/docs/07-reference/02-ai-sdk-ui/42-pipe-ui-message-stream-to-response.mdx

@@ -66,6 +66,13 @@ await pipeUIMessageStreamToResponse({
       isOptional: true,
       description: 'Additional headers for the response.',
     },
+    {
+      name: 'keepAliveMs',
+      type: 'number',
+      isOptional: true,
+      description:
+        'Interval in milliseconds for sending SSE keep-alive comments. When set, an opening comment is sent immediately and additional comments are sent after the stream has been idle for the configured interval.',
+    },
     {
       name: 'consumeSseStream',
       type: '({ stream }: { stream: ReadableStream<string> }) => PromiseLike<void> | void',

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ai",
-  "version": "6.0.296",
+  "version": "6.0.298",
   "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",
   "sideEffects": false,
@@ -45,9 +45,9 @@
   },
   "dependencies": {
     "@opentelemetry/api": "^1.9.0",
-    "@ai-sdk/gateway": "3.0.205",
+    "@ai-sdk/gateway": "3.0.207",
     "@ai-sdk/provider": "3.0.18",
-    "@ai-sdk/provider-utils": "4.0.56"
+    "@ai-sdk/provider-utils": "4.0.57"
   },
   "devDependencies": {
     "@edge-runtime/vm": "^5.0.0",
@@ -58,7 +58,7 @@
     "tsx": "^4.19.2",
     "typescript": "5.8.3",
     "zod": "3.25.76",
-    "@ai-sdk/google": "3.0.129",
+    "@ai-sdk/google": "3.0.130",
     "@ai-sdk/test-server": "1.0.6",
     "@vercel/ai-tsconfig": "0.0.0"
   },

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

@@ -25,6 +25,7 @@ import type { ToolLoopAgentOnStepFinishCallback } from './tool-loop-agent-settin
  * @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.
+ * @param keepAliveMs - Optional interval for sending SSE keep-alive comments.
  * @param consumeSseStream - Whether to consume the SSE stream. Optional.
  *
  * @returns The response object.
@@ -38,6 +39,7 @@ export async function createAgentUIStreamResponse<
   headers,
   status,
   statusText,
+  keepAliveMs,
   consumeSseStream,
   ...options
 }: {
@@ -58,6 +60,7 @@ export async function createAgentUIStreamResponse<
     headers,
     status,
     statusText,
+    keepAliveMs,
     consumeSseStream,
     stream: await createAgentUIStream(options),
   });

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

@@ -27,6 +27,7 @@ import type { ToolLoopAgentOnStepFinishCallback } from './tool-loop-agent-settin
  * @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.
+ * @param keepAliveMs - Optional interval for sending SSE keep-alive comments.
  * @param consumeSseStream - Whether to consume the SSE stream. Optional.
  */
 export async function pipeAgentUIStreamToResponse<
@@ -39,6 +40,7 @@ export async function pipeAgentUIStreamToResponse<
   headers,
   status,
   statusText,
+  keepAliveMs,
   consumeSseStream,
   ...options
 }: {
@@ -61,6 +63,7 @@ export async function pipeAgentUIStreamToResponse<
     headers,
     status,
     statusText,
+    keepAliveMs,
     consumeSseStream,
     stream: await createAgentUIStream(options),
   });

package/src/embed/embed-many.ts

@@ -1,3 +1,5 @@
+import { getOwn } from '../util/get-own';
+import { setOwn } from '../util/set-own';
 import {
   withUserAgentSuffix,
   type ProviderOptions,
@@ -358,10 +360,10 @@ export async function embedMany({
               for (const [providerName, metadata] of Object.entries(
                 result.providerMetadata,
               )) {
-                providerMetadata[providerName] = {
-                  ...providerMetadata[providerName],
+                setOwn(providerMetadata, providerName, {
+                  ...getOwn(providerMetadata, providerName),
                   ...metadata,
-                };
+                });
               }
             }
           }

package/src/generate-image/generate-image.ts

@@ -1,3 +1,5 @@
+import { getOwn } from '../util/get-own';
+import { setOwn } from '../util/set-own';
 import type {
   ImageModelV3,
   ImageModelV3CallOptions,
@@ -284,10 +286,13 @@ export async function generateImage({
               .images;
           }
         } else {
-          providerMetadata[providerName] ??= { images: [] };
-          providerMetadata[providerName].images.push(
+          const currentEntry = getOwn(providerMetadata, providerName) ?? {
+            images: [],
+          };
+          currentEntry.images.push(
             ...result.providerMetadata[providerName].images,
           );
+          setOwn(providerMetadata, providerName, currentEntry);
         }
       }
     }

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

@@ -99,6 +99,11 @@ export type UIMessageStreamOptions<UI_MESSAGE extends UIMessage> = {
   /**
    * Process an error, e.g. to log it. Default to `() => 'An error occurred.'`.
    *
+   * Tool execution errors marked `providerExecuted: true` bypass this callback.
+   * Their strings pass through unchanged; other values are JSON-stringified to
+   * preserve provider error data for model-message round trips. Stream errors
+   * and invalid tool calls still use it.
+   *
    * @returns error message to include in the data stream.
    */
   onError?: (error: unknown) => string;

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

@@ -2698,7 +2698,7 @@ class DefaultStreamTextResult<
     sendSources = false,
     sendStart = true,
     sendFinish = true,
-    onError = () => 'An error occurred.', // prevent leaking server error details to the client by default
+    onError = () => 'An error occurred.', // masks errors except provider-executed tool execution errors
   }: UIMessageStreamOptions<UI_MESSAGE> = {}): AsyncIterableStream<
     InferUIMessageChunk<UI_MESSAGE>
   > {
@@ -3032,6 +3032,9 @@ class DefaultStreamTextResult<
             case 'tool-error': {
               const dynamic = isDynamic(part);
 
+              // Preserve provider error codes for model-message round trips.
+              // These execution errors intentionally bypass onError; invalid
+              // tool calls and stream errors still go through it.
               controller.enqueue({
                 type: 'tool-output-error',
                 toolCallId: part.toolCallId,

package/src/generate-video/generate-video.ts

@@ -1,3 +1,5 @@
+import { getOwn } from '../util/get-own';
+import { setOwn } from '../util/set-own';
 import type {
   Experimental_VideoModelV3,
   Experimental_VideoModelV3CallOptions,
@@ -385,12 +387,12 @@ export async function experimental_generateVideo({
       for (const [providerName, metadata] of Object.entries(
         result.providerMetadata,
       )) {
-        const existingMetadata = providerMetadata[providerName];
+        const existingMetadata = getOwn(providerMetadata, providerName);
         if (existingMetadata != null && typeof existingMetadata === 'object') {
-          providerMetadata[providerName] = {
+          setOwn(providerMetadata, providerName, {
             ...existingMetadata,
             ...metadata,
-          };
+          });
 
           // Merge videos arrays if both exist
           if (
@@ -405,7 +407,7 @@ export async function experimental_generateVideo({
             ];
           }
         } else {
-          providerMetadata[providerName] = metadata;
+          setOwn(providerMetadata, providerName, metadata);
         }
       }
     }

package/src/registry/provider-registry.ts

@@ -133,7 +133,7 @@ class DefaultProviderRegistry<
   PROVIDERS extends Record<string, ProviderV3>,
   SEPARATOR extends string,
 > implements ProviderRegistryProvider<PROVIDERS, SEPARATOR> {
-  private providers: PROVIDERS = {} as PROVIDERS;
+  private providers: PROVIDERS = Object.create(null);
   private separator: SEPARATOR;
   private languageModelMiddleware?:
     | LanguageModelMiddleware

package/src/ui/chat.ts

@@ -792,6 +792,14 @@ export abstract class AbstractChat<UI_MESSAGE extends UIMessage> {
       responseMessageIndex === -1
         ? lastMessage
         : this.state.messages[responseMessageIndex];
+    const resumableResponseMessage =
+      trigger === 'resume-stream' &&
+      responseMessage?.role === 'assistant' &&
+      responseMessage.parts.some(
+        part => isToolUIPart(part) && part.state === 'input-streaming',
+      )
+        ? this.state.snapshot(responseMessage)
+        : undefined;
     const usesEarlierAssistantMessage =
       responseMessageIndex !== -1 &&
       responseMessageIndex < this.state.messages.length - 1 &&
@@ -809,10 +817,11 @@ export abstract class AbstractChat<UI_MESSAGE extends UIMessage> {
             ? this.resumableStreamState
             : createStreamingUIMessageState({
                 lastMessage:
-                  trigger === 'resume-stream' ||
-                  trigger === 'regenerate-message'
-                    ? undefined
-                    : this.state.snapshot(responseMessage),
+                  trigger === 'resume-stream'
+                    ? resumableResponseMessage
+                    : trigger === 'regenerate-message'
+                      ? undefined
+                      : this.state.snapshot(responseMessage),
                 messageId: this.generateId(),
               }),
         abortController,

package/src/ui/http-chat-transport.ts

@@ -4,10 +4,19 @@ import {
   type FetchFunction,
   type Resolvable,
 } from '@ai-sdk/provider-utils';
+import { InvalidArgumentError } from '../error/invalid-argument-error';
 import type { UIMessageChunk } from '../ui-message-stream/ui-message-chunks';
 import type { ChatTransport } from './chat-transport';
 import type { UIMessage } from './ui-messages';
 
+function appendPathToUrl(url: string, path: string): string {
+  const queryOrFragmentStart = url.search(/[?#]/);
+
+  return queryOrFragmentStart === -1
+    ? `${url}${path}`
+    : `${url.slice(0, queryOrFragmentStart)}${path}${url.slice(queryOrFragmentStart)}`;
+}
+
 export type PrepareSendMessagesRequest<UI_MESSAGE extends UIMessage> = (
   options: {
     id: string;
@@ -233,7 +242,24 @@ export abstract class HttpChatTransport<
       requestMetadata: options.metadata,
     });
 
-    const api = preparedRequest?.api ?? `${this.api}/${options.chatId}/stream`;
+    let api = preparedRequest?.api;
+    if (api == null) {
+      // encodeURIComponent leaves dot segments unchanged, and URL parsers
+      // normalize them even when their dots are percent-encoded.
+      if (options.chatId === '.' || options.chatId === '..') {
+        throw new InvalidArgumentError({
+          parameter: 'chatId',
+          value: options.chatId,
+          message:
+            'Chat IDs must not be "." or ".." when using the default reconnect URL.',
+        });
+      }
+
+      api = appendPathToUrl(
+        this.api,
+        `/${encodeURIComponent(options.chatId)}/stream`,
+      );
+    }
     const headers =
       preparedRequest?.headers !== undefined
         ? normalizeHeaders(preparedRequest.headers)

package/src/ui/last-assistant-message-is-complete-with-tool-calls.ts

@@ -3,7 +3,9 @@ import { isToolUIPart, type UIMessage } from './ui-messages';
 /**
  * Check if the last message is an assistant message with completed tool calls.
  * The last step of the message must have at least one tool invocation and
- * all tool invocations must have a result.
+ * all tool invocations must have a result. Completed model text may follow
+ * the tool invocation because client-side tool results update parts in place,
+ * but trailing text without a completed stream state prevents resubmission.
  */
 export function lastAssistantMessageIsCompleteWithToolCalls({
   messages,
@@ -24,13 +26,23 @@ export function lastAssistantMessageIsCompleteWithToolCalls({
     return part.type === 'step-start' ? index : lastIndex;
   }, -1);
 
-  const lastStepToolInvocations = message.parts
-    .slice(lastStepStartIndex + 1)
+  const lastStepParts = message.parts.slice(lastStepStartIndex + 1);
+
+  const lastStepToolInvocations = lastStepParts
     .filter(isToolUIPart)
     .filter(part => !part.providerExecuted);
 
+  const lastToolInvocationIndex = lastStepParts.reduce(
+    (lastIndex, part, index) =>
+      isToolUIPart(part) && !part.providerExecuted ? index : lastIndex,
+    -1,
+  );
+
   return (
     lastStepToolInvocations.length > 0 &&
+    !lastStepParts
+      .slice(lastToolInvocationIndex + 1)
+      .some(part => part.type === 'text' && part.state !== 'done') &&
     lastStepToolInvocations.every(
       part =>
         (part.state === 'output-available' && part.preliminary !== true) ||

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

@@ -60,22 +60,55 @@ export function createStreamingUIMessageState<UI_MESSAGE extends UIMessage>({
   lastMessage: UI_MESSAGE | undefined;
   messageId: string;
 }): StreamingUIMessageState<UI_MESSAGE> {
+  const message =
+    lastMessage?.role === 'assistant'
+      ? lastMessage
+      : ({
+          id: messageId,
+          metadata: undefined,
+          role: 'assistant',
+          parts: [] as UIMessagePart<
+            InferUIMessageData<UI_MESSAGE>,
+            InferUIMessageTools<UI_MESSAGE>
+          >[],
+        } as UI_MESSAGE);
+  const partialToolCalls: StreamingUIMessageState<UI_MESSAGE>['partialToolCalls'] =
+    createIdMap();
+  const lastStepStartIndex = message.parts.findLastIndex(
+    part => part.type === 'step-start',
+  );
+  let staticToolIndex = 0;
+
+  for (const part of message.parts.slice(lastStepStartIndex + 1)) {
+    if (!isToolUIPart(part)) {
+      continue;
+    }
+
+    const index = staticToolIndex;
+    if (isStaticToolUIPart(part)) {
+      staticToolIndex++;
+    }
+
+    if (part.state !== 'input-streaming') {
+      continue;
+    }
+
+    partialToolCalls[part.toolCallId] = {
+      text: part.rawInput ?? '',
+      index,
+      toolName:
+        part.type === 'dynamic-tool' ? part.toolName : getStaticToolName(part),
+      dynamic: part.type === 'dynamic-tool',
+      title: part.title,
+      toolMetadata: part.toolMetadata,
+    };
+  }
+
   return {
-    message:
-      lastMessage?.role === 'assistant'
-        ? lastMessage
-        : ({
-            id: messageId,
-            metadata: undefined,
-            role: 'assistant',
-            parts: [] as UIMessagePart<
-              InferUIMessageData<UI_MESSAGE>,
-              InferUIMessageTools<UI_MESSAGE>
-            >[],
-          } as UI_MESSAGE),
+    message,
     activeTextParts: createIdMap(),
     activeReasoningParts: createIdMap(),
-    partialToolCalls: createIdMap(),
+    partialToolCalls,
   };
 }
 
@@ -167,6 +200,7 @@ export function processUIMessageStream<UI_MESSAGE extends UIMessage>({
               | {
                   state: 'input-streaming';
                   input: unknown;
+                  rawInput?: string;
                   providerExecuted?: boolean;
                   providerMetadata?: ProviderMetadata;
                 }
@@ -282,6 +316,7 @@ export function processUIMessageStream<UI_MESSAGE extends UIMessage>({
               | {
                   state: 'input-streaming';
                   input: unknown;
+                  rawInput?: string;
                   providerMetadata?: ProviderMetadata;
                 }
               | {
@@ -322,7 +357,7 @@ export function processUIMessageStream<UI_MESSAGE extends UIMessage>({
               anyPart.input = anyOptions.input;
               anyPart.output = anyOptions.output;
               anyPart.errorText = anyOptions.errorText;
-              anyPart.rawInput = anyOptions.rawInput ?? anyPart.rawInput;
+              anyPart.rawInput = anyOptions.rawInput;
               anyPart.preliminary = anyOptions.preliminary;
               if (options.title !== undefined) {
                 anyPart.title = options.title;
@@ -615,6 +650,7 @@ export function processUIMessageStream<UI_MESSAGE extends UIMessage>({
                   toolName: partialToolCall.toolName,
                   state: 'input-streaming',
                   input: partialArgs,
+                  rawInput: partialToolCall.text,
                   title: partialToolCall.title,
                   toolMetadata: partialToolCall.toolMetadata,
                 });
@@ -624,6 +660,7 @@ export function processUIMessageStream<UI_MESSAGE extends UIMessage>({
                   toolName: partialToolCall.toolName,
                   state: 'input-streaming',
                   input: partialArgs,
+                  rawInput: partialToolCall.text,
                   title: partialToolCall.title,
                   toolMetadata: partialToolCall.toolMetadata,
                 });

package/src/ui/ui-messages.ts

@@ -238,6 +238,13 @@ export type UIToolInvocation<TOOL extends UITool | Tool> = {
   | {
       state: 'input-streaming';
       input?: DeepPartial<asUITool<TOOL>['input']> | undefined;
+      /**
+       * The raw tool input text received so far.
+       *
+       * This is used to continue input streaming when a message is persisted
+       * and later resumed.
+       */
+      rawInput?: string;
       output?: never;
       errorText?: never;
       callProviderMetadata?: ProviderMetadata;
@@ -361,6 +368,13 @@ export type DynamicToolUIPart = {
   | {
       state: 'input-streaming';
       input?: unknown;
+      /**
+       * The raw tool input text received so far.
+       *
+       * This is used to continue input streaming when a message is persisted
+       * and later resumed.
+       */
+      rawInput?: string;
       output?: never;
       errorText?: never;
       callProviderMetadata?: ProviderMetadata;

package/src/ui/validate-ui-messages.ts

@@ -139,6 +139,7 @@ const uiMessagesSchema = lazySchema(() => {
                   toolMetadata: toolMetadataSchema.optional(),
                   state: z.literal('input-streaming'),
                   input: z.unknown().optional(),
+                  rawInput: z.string().optional(),
                   providerExecuted: z.boolean().optional(),
                   callProviderMetadata: providerMetadataSchema.optional(),
                   output: z.never().optional(),
@@ -235,6 +236,7 @@ const uiMessagesSchema = lazySchema(() => {
                   providerExecuted: z.boolean().optional(),
                   callProviderMetadata: providerMetadataSchema.optional(),
                   input: z.unknown().optional(),
+                  rawInput: z.string().optional(),
                   output: z.never().optional(),
                   errorText: z.never().optional(),
                   approval: z.never().optional(),