ai 5.0.0-canary.14 → 5.0.0-canary.16

package/dist/index.d.ts

@@ -1,11 +1,11 @@
-import { ToolCall, ToolResult, FetchFunction, Validator, IDGenerator } from '@ai-sdk/provider-utils';
-export { IDGenerator, ToolCall, ToolResult, createIdGenerator, generateId } from '@ai-sdk/provider-utils';
-import { EmbeddingModelV2, EmbeddingModelV2Embedding, ImageModelV2, ImageModelV2CallWarning, LanguageModelV2, LanguageModelV2FinishReason, LanguageModelV2CallWarning, LanguageModelV2Source, SharedV2ProviderMetadata, SharedV2ProviderOptions, TranscriptionModelV1, TranscriptionModelV1CallWarning, SpeechModelV1, SpeechModelV1CallWarning, JSONValue as JSONValue$1, JSONObject, LanguageModelV2CallOptions, AISDKError, LanguageModelV2ToolCall, JSONSchema7 as JSONSchema7$1, JSONParseError, TypeValidationError, LanguageModelV2Middleware, ProviderV2, NoSuchModelError } from '@ai-sdk/provider';
-export { AISDKError, APICallError, EmptyResponseBodyError, InvalidPromptError, InvalidResponseDataError, JSONParseError, LoadAPIKeyError, NoContentGeneratedError, NoSuchModelError, TypeValidationError, UnsupportedFunctionalityError } from '@ai-sdk/provider';
+import { ToolCall, ToolResult, FetchFunction, Schema, IDGenerator } from '@ai-sdk/provider-utils';
+export { IDGenerator, Schema, ToolCall, ToolResult, asSchema, createIdGenerator, generateId, jsonSchema } from '@ai-sdk/provider-utils';
+import * as _ai_sdk_provider from '@ai-sdk/provider';
+import { EmbeddingModelV2, EmbeddingModelV2Embedding, ImageModelV2, ImageModelV2CallWarning, JSONValue as JSONValue$1, LanguageModelV2, LanguageModelV2FinishReason, LanguageModelV2CallWarning, LanguageModelV2Source, SharedV2ProviderMetadata, SharedV2ProviderOptions, SpeechModelV1, SpeechModelV1CallWarning, TranscriptionModelV1, TranscriptionModelV1CallWarning, JSONObject, LanguageModelV2CallOptions, AISDKError, LanguageModelV2ToolCall, JSONSchema7, JSONParseError, TypeValidationError, LanguageModelV2Middleware, ProviderV2, NoSuchModelError } from '@ai-sdk/provider';
+export { AISDKError, APICallError, EmptyResponseBodyError, InvalidPromptError, InvalidResponseDataError, JSONParseError, JSONSchema7, LoadAPIKeyError, NoContentGeneratedError, NoSuchModelError, TypeValidationError, UnsupportedFunctionalityError } from '@ai-sdk/provider';
 import { ServerResponse } from 'node:http';
 import { AttributeValue, Tracer } from '@opentelemetry/api';
 import { z } from 'zod';
-import { JSONSchema7 } from 'json-schema';
 import { ServerResponse as ServerResponse$1 } from 'http';
 
 /**
@@ -42,6 +42,8 @@ type ImageModelResponseMetadata = {
     headers?: Record<string, string>;
 };
 
+type JSONValue = JSONValue$1;
+
 /**
 Language model that is used by the AI SDK Core functions.
 */
@@ -159,36 +161,63 @@ provider-specific functionality that can be fully encapsulated in the provider.
 type ProviderOptions = SharedV2ProviderOptions;
 
 /**
-Represents the number of tokens used in a prompt and completion.
- */
-type LanguageModelUsage$1 = {
+Speech model that is used by the AI SDK Core functions.
+  */
+type SpeechModel = SpeechModelV1;
+/**
+Warning from the model provider for this call. The call will proceed, but e.g.
+some settings might not be supported, which can lead to suboptimal results.
+  */
+type SpeechWarning = SpeechModelV1CallWarning;
+
+type SpeechModelResponseMetadata = {
     /**
-  The number of tokens used in the prompt.
+  Timestamp for the start of the generated response.
      */
-    promptTokens: number;
+    timestamp: Date;
     /**
-  The number of tokens used in the completion.
-   */
-    completionTokens: number;
+  The ID of the response model that was used to generate the response.
+     */
+    modelId: string;
     /**
-  The total number of tokens used (promptTokens + completionTokens).
+  Response headers.
      */
-    totalTokens: number;
+    headers?: Record<string, string>;
+    /**
+  Response body.
+     */
+    body?: unknown;
 };
+
 /**
-Represents the number of tokens used in an embedding.
- */
-type EmbeddingModelUsage = {
+Transcription model that is used by the AI SDK Core functions.
+  */
+type TranscriptionModel = TranscriptionModelV1;
+/**
+Warning from the model provider for this call. The call will proceed, but e.g.
+some settings might not be supported, which can lead to suboptimal results.
+  */
+type TranscriptionWarning = TranscriptionModelV1CallWarning;
+
+type TranscriptionModelResponseMetadata = {
     /**
-  The number of tokens used in the embedding.
+  Timestamp for the start of the generated response.
      */
-    tokens: number;
+    timestamp: Date;
+    /**
+  The ID of the response model that was used to generate the response.
+     */
+    modelId: string;
+    /**
+  Response headers.
+     */
+    headers?: Record<string, string>;
 };
 
 /**
 Represents the number of tokens used in a prompt and completion.
  */
-type LanguageModelUsage = {
+type LanguageModelUsage$1 = {
     /**
   The number of tokens used in the prompt.
      */
@@ -244,7 +273,7 @@ interface Attachment {
 /**
  * AI SDK UI Messages. They are used in the client and to communicate between the frontend and the API routes.
  */
-interface Message {
+interface UIMessage {
     /**
   A unique identifier for the message.
      */
@@ -258,53 +287,26 @@ interface Message {
      */
     content: string;
     /**
-  Reasoning for the message.
-  
-  @deprecated Use `parts` instead.
-     */
-    reasoning?: string;
-    /**
-     * Additional attachments to be sent along with the message.
+  Additional attachments to be sent along with the message.
      */
     experimental_attachments?: Attachment[];
     /**
-  The 'data' role is deprecated.
+  The role of the message.
      */
-    role: 'system' | 'user' | 'assistant' | 'data';
+    role: 'system' | 'user' | 'assistant';
     /**
-  For data messages.
-  
-  @deprecated Data messages will be removed.
+  Additional message-specific information added on the server via StreamData
      */
-    data?: JSONValue;
+    annotations?: JSONValue$1[] | undefined;
     /**
-     * Additional message-specific information added on the server via StreamData
-     */
-    annotations?: JSONValue[] | undefined;
-    /**
-  Tool invocations (that can be tool calls or tool results, depending on whether or not the invocation has finished)
-  that the assistant made as part of this message.
+  The parts of the message. Use this for rendering the message in the UI.
   
-  @deprecated Use `parts` instead.
-     */
-    toolInvocations?: Array<ToolInvocation>;
-    /**
-     * The parts of the message. Use this for rendering the message in the UI.
-     *
-     * Assistant messages can have text, reasoning and tool invocation parts.
-     * User messages can have text parts.
+  Assistant messages can have text, reasoning and tool invocation parts.
+  User messages can have text parts.
      */
-    parts?: Array<TextUIPart | ReasoningUIPart | ToolInvocationUIPart | SourceUIPart | FileUIPart | StepStartUIPart>;
+    parts: Array<UIMessagePart>;
 }
-type UIMessage = Message & {
-    /**
-     * The parts of the message. Use this for rendering the message in the UI.
-     *
-     * Assistant messages can have text, reasoning and tool invocation parts.
-     * User messages can have text parts.
-     */
-    parts: Array<TextUIPart | ReasoningUIPart | ToolInvocationUIPart | SourceUIPart | FileUIPart | StepStartUIPart>;
-};
+type UIMessagePart = TextUIPart | ReasoningUIPart | ToolInvocationUIPart | SourceUIPart | FileUIPart | StepStartUIPart;
 /**
  * A text part of a message.
  */
@@ -323,7 +325,7 @@ type ReasoningUIPart = {
     /**
      * The reasoning text.
      */
-    reasoning: string;
+    text: string;
     /**
      * The provider metadata.
      */
@@ -371,8 +373,8 @@ type FileUIPart = {
 type StepStartUIPart = {
     type: 'step-start';
 };
-type CreateMessage = Omit<Message, 'id'> & {
-    id?: Message['id'];
+type CreateUIMessage = Omit<UIMessage, 'id'> & {
+    id?: UIMessage['id'];
 };
 type ChatRequest = {
     /**
@@ -386,11 +388,11 @@ type ChatRequest = {
     /**
   The messages of the chat.
      */
-    messages: Message[];
+    messages: UIMessage[];
     /**
   Additional data to be sent to the server.
      */
-    data?: JSONValue;
+    data?: JSONValue$1;
 };
 type RequestOptions = {
     /**
@@ -414,7 +416,7 @@ type ChatRequestOptions = {
     /**
   Additional data to be sent to the API endpoint.
      */
-    data?: JSONValue;
+    data?: JSONValue$1;
     /**
      * Additional files to be sent to the server.
      */
@@ -445,7 +447,7 @@ type UseChatOptions = {
     /**
      * Initial messages of the chat. Useful to load an existing chat history.
      */
-    initialMessages?: Message[];
+    initialMessages?: UIMessage[];
     /**
      * Initial input of the chat.
      */
@@ -471,8 +473,8 @@ type UseChatOptions = {
      * @param options.usage The token usage of the message.
      * @param options.finishReason The finish reason of the message.
      */
-    onFinish?: (message: Message, options: {
-        usage: LanguageModelUsage;
+    onFinish?: (message: UIMessage, options: {
+        usage: LanguageModelUsage$1;
         finishReason: LanguageModelV2FinishReason;
     }) => void;
     /**
@@ -588,66 +590,32 @@ type UseCompletionOptions = {
       */
     fetch?: FetchFunction;
 };
-/**
-A JSON value can be a string, number, boolean, object, array, or null.
-JSON values can be serialized and deserialized by the JSON.stringify and JSON.parse methods.
- */
-type JSONValue = null | string | number | boolean | {
-    [value: string]: JSONValue;
-} | Array<JSONValue>;
 
 /**
-Transcription model that is used by the AI SDK Core functions.
-  */
-type TranscriptionModel = TranscriptionModelV1;
-/**
-Warning from the model provider for this call. The call will proceed, but e.g.
-some settings might not be supported, which can lead to suboptimal results.
-  */
-type TranscriptionWarning = TranscriptionModelV1CallWarning;
-
-type TranscriptionModelResponseMetadata = {
+Represents the number of tokens used in a prompt and completion.
+ */
+type LanguageModelUsage = {
     /**
-  Timestamp for the start of the generated response.
+  The number of tokens used in the prompt.
      */
-    timestamp: Date;
+    promptTokens: number;
     /**
-  The ID of the response model that was used to generate the response.
-     */
-    modelId: string;
+  The number of tokens used in the completion.
+   */
+    completionTokens: number;
     /**
-  Response headers.
+  The total number of tokens used (promptTokens + completionTokens).
      */
-    headers?: Record<string, string>;
+    totalTokens: number;
 };
-
 /**
-Speech model that is used by the AI SDK Core functions.
-  */
-type SpeechModel = SpeechModelV1;
-/**
-Warning from the model provider for this call. The call will proceed, but e.g.
-some settings might not be supported, which can lead to suboptimal results.
-  */
-type SpeechWarning = SpeechModelV1CallWarning;
-
-type SpeechModelResponseMetadata = {
-    /**
-  Timestamp for the start of the generated response.
-     */
-    timestamp: Date;
-    /**
-  The ID of the response model that was used to generate the response.
-     */
-    modelId: string;
-    /**
-  Response headers.
-     */
-    headers?: Record<string, string>;
+Represents the number of tokens used in an embedding.
+ */
+type EmbeddingModelUsage = {
     /**
-  Response body.
+  The number of tokens used in the embedding.
      */
-    body?: unknown;
+    tokens: number;
 };
 
 declare const getOriginalFetch$1: () => typeof fetch;
@@ -662,7 +630,7 @@ declare function callChatApi({ api, body, streamProtocol, credentials, headers,
     onResponse: ((response: Response) => void | Promise<void>) | undefined;
     onUpdate: (options: {
         message: UIMessage;
-        data: JSONValue[] | undefined;
+        data: JSONValue$1[] | undefined;
         replaceLastMessage: boolean;
     }) => void;
     onFinish: UseChatOptions['onFinish'];
@@ -700,7 +668,7 @@ interface DataStreamPart<CODE extends string, NAME extends string, TYPE> {
         value: TYPE;
     };
 }
-declare const dataStreamParts: readonly [DataStreamPart<"0", "text", string>, DataStreamPart<"2", "data", JSONValue[]>, DataStreamPart<"3", "error", string>, DataStreamPart<"8", "message_annotations", JSONValue[]>, DataStreamPart<"9", "tool_call", ToolCall<string, any>>, DataStreamPart<"a", "tool_result", Omit<ToolResult<string, any, any>, "args" | "toolName">>, DataStreamPart<"b", "tool_call_streaming_start", {
+declare const dataStreamParts: readonly [DataStreamPart<"0", "text", string>, DataStreamPart<"2", "data", _ai_sdk_provider.JSONValue[]>, DataStreamPart<"3", "error", string>, DataStreamPart<"8", "message_annotations", _ai_sdk_provider.JSONValue[]>, DataStreamPart<"9", "tool_call", ToolCall<string, any>>, DataStreamPart<"a", "tool_result", Omit<ToolResult<string, any, any>, "args" | "toolName">>, DataStreamPart<"b", "tool_call_streaming_start", {
     toolCallId: string;
     toolName: string;
 }>, DataStreamPart<"c", "tool_call_delta", {
@@ -799,10 +767,6 @@ type PartialObject<ObjectType extends object> = {
 
 declare function extractMaxToolInvocationStep(toolInvocations: ToolInvocation[] | undefined): number | undefined;
 
-declare function fillMessageParts(messages: Message[]): UIMessage[];
-
-declare function getMessageParts(message: Message | CreateMessage | UIMessage): (TextUIPart | ReasoningUIPart | ToolInvocationUIPart | SourceUIPart | FileUIPart | StepStartUIPart)[];
-
 /**
  * Performs a deep-equal comparison of two parsed JSON objects.
  *
@@ -874,39 +838,19 @@ declare function processTextStream({ stream, onTextPart, }: {
 }): Promise<void>;
 
 /**
- * Used to mark schemas so we can support both Zod and custom schemas.
- */
-declare const schemaSymbol: unique symbol;
-type Schema<OBJECT = unknown> = Validator<OBJECT> & {
-    /**
-     * Used to mark schemas so we can support both Zod and custom schemas.
-     */
-    [schemaSymbol]: true;
-    /**
-     * Schema type for inference.
-     */
-    _type: OBJECT;
-    /**
-     * The JSON Schema for the schema. It is passed to the providers.
-     */
-    readonly jsonSchema: JSONSchema7;
-};
-/**
- * Create a schema using a JSON Schema.
+ * Updates the result of a specific tool invocation in the last message of the given messages array.
  *
- * @param jsonSchema The JSON Schema for the schema.
- * @param options.validate Optional. A validation function for the schema.
+ * @param {object} params - The parameters object.
+ * @param {UIMessage[]} params.messages - An array of messages, from which the last one is updated.
+ * @param {string} params.toolCallId - The unique identifier for the tool invocation to update.
+ * @param {unknown} params.toolResult - The result object to attach to the tool invocation.
+ * @returns {void} This function does not return anything.
  */
-declare function jsonSchema<OBJECT = unknown>(jsonSchema: JSONSchema7, { validate, }?: {
-    validate?: (value: unknown) => {
-        success: true;
-        value: OBJECT;
-    } | {
-        success: false;
-        error: Error;
-    };
-}): Schema<OBJECT>;
-declare function asSchema<OBJECT>(schema: z.Schema<OBJECT, z.ZodTypeDef, any> | Schema<OBJECT> | undefined): Schema<OBJECT>;
+declare function updateToolCallResult({ messages, toolCallId, toolResult: result, }: {
+    messages: UIMessage[];
+    toolCallId: string;
+    toolResult: unknown;
+}): void;
 
 declare function shouldResubmitMessages({ originalMaxToolInvocationStep, originalMessageCount, maxSteps, messages, }: {
     originalMaxToolInvocationStep: number | undefined;
@@ -923,31 +867,6 @@ declare function isAssistantMessageWithCompletedToolCalls(message: UIMessage): m
     role: 'assistant';
 };
 
-/**
- * Updates the result of a specific tool invocation in the last message of the given messages array.
- *
- * @param {object} params - The parameters object.
- * @param {UIMessage[]} params.messages - An array of messages, from which the last one is updated.
- * @param {string} params.toolCallId - The unique identifier for the tool invocation to update.
- * @param {unknown} params.toolResult - The result object to attach to the tool invocation.
- * @returns {void} This function does not return anything.
- */
-declare function updateToolCallResult({ messages, toolCallId, toolResult: result, }: {
-    messages: UIMessage[];
-    toolCallId: string;
-    toolResult: unknown;
-}): void;
-
-declare function zodSchema<OBJECT>(zodSchema: z.Schema<OBJECT, z.ZodTypeDef, any>, options?: {
-    /**
-     * Enables support for references in the schema.
-     * This is required for recursive schemas, e.g. with `z.lazy`.
-     * However, not all language models and providers support such references.
-     * Defaults to `false`.
-     */
-    useReferences?: boolean;
-}): Schema<OBJECT>;
-
 interface DataStreamWriter {
     /**
      * Appends a data part to the stream.
@@ -1532,7 +1451,7 @@ type Prompt = {
     /**
   A list of messages. You can either use `prompt` or `messages` but not both.
      */
-    messages?: Array<CoreMessage> | Array<Omit<Message, 'id'>>;
+    messages?: Array<CoreMessage> | Array<Omit<UIMessage, 'id'>>;
 };
 
 /**
@@ -1628,20 +1547,20 @@ declare const JSONRPCErrorSchema: z.ZodObject<{
         data?: unknown;
     }>;
 }, "strict", z.ZodTypeAny, {
+    id: string | number;
     error: {
         code: number;
         message: string;
         data?: unknown;
     };
-    id: string | number;
     jsonrpc: "2.0";
 }, {
+    id: string | number;
     error: {
         code: number;
         message: string;
         data?: unknown;
     };
-    id: string | number;
     jsonrpc: "2.0";
 }>;
 type JSONRPCError = z.infer<typeof JSONRPCErrorSchema>;
@@ -1762,20 +1681,20 @@ declare const JSONRPCMessageSchema: z.ZodUnion<[z.ZodObject<z.objectUtil.extendS
         data?: unknown;
     }>;
 }, "strict", z.ZodTypeAny, {
+    id: string | number;
     error: {
         code: number;
         message: string;
         data?: unknown;
     };
-    id: string | number;
     jsonrpc: "2.0";
 }, {
+    id: string | number;
     error: {
         code: number;
         message: string;
         data?: unknown;
     };
-    id: string | number;
     jsonrpc: "2.0";
 }>]>;
 type JSONRPCMessage = z.infer<typeof JSONRPCMessageSchema>;
@@ -2643,9 +2562,13 @@ declare function createMCPClient(config: MCPClientConfig): Promise<MCPClient>;
  * Tool parameters are automatically inferred from the server's JSON schema
  * if not explicitly provided in the tools configuration
  *
+ * This client is meant to be used to communicate with a single server. To communicate and fetch tools across multiple servers, it's recommended to create a new client instance per server.
+ *
  * Not supported:
  * - Client options (e.g. sampling, roots) as they are not needed for tool conversion
  * - Accepting notifications
+ * - Session management (when passing a sessionId to an instance of the Streamable HTTP transport)
+ * - Resumable SSE streams
  */
 declare class MCPClient {
     private transport;
@@ -2658,6 +2581,7 @@ declare class MCPClient {
     constructor({ transport: transportConfig, name, onUncaughtError, }: MCPClientConfig);
     init(): Promise<this>;
     close(): Promise<void>;
+    private assertCapability;
     private request;
     private listTools;
     private callTool;
@@ -2792,7 +2716,7 @@ type StepResult<TOOLS extends ToolSet> = {
     /**
   The token usage of the generated text.
   */
-    readonly usage: LanguageModelUsage$1;
+    readonly usage: LanguageModelUsage;
     /**
   Warnings from the model provider (e.g. unsupported settings).
   */
@@ -2880,7 +2804,7 @@ interface GenerateTextResult<TOOLS extends ToolSet, OUTPUT> {
     /**
     The token usage of the generated text.
      */
-    readonly usage: LanguageModelUsage$1;
+    readonly usage: LanguageModelUsage;
     /**
     Warnings from the model provider (e.g. unsupported settings)
      */
@@ -2933,7 +2857,7 @@ interface Output$1<OUTPUT, PARTIAL> {
         text: string;
     }, context: {
         response: LanguageModelResponseMetadata;
-        usage: LanguageModelUsage$1;
+        usage: LanguageModelUsage;
         finishReason: FinishReason;
     }): Promise<OUTPUT>;
 }
@@ -2985,9 +2909,9 @@ declare class NoSuchToolError extends AISDKError {
  * Otherwise, the new message will be appended.
  */
 declare function appendClientMessage({ messages, message, }: {
-    messages: Message[];
-    message: Message;
-}): Message[];
+    messages: UIMessage[];
+    message: UIMessage;
+}): UIMessage[];
 
 /**
  * Appends the ResponseMessage[] from the response to a Message[] (for useChat).
@@ -2997,7 +2921,7 @@ declare function appendClientMessage({ messages, message, }: {
  * @returns A new Message[] with the response messages appended.
  */
 declare function appendResponseMessages({ messages, responseMessages, _internal: { currentDate }, }: {
-    messages: Message[];
+    messages: UIMessage[];
     responseMessages: ResponseMessage[];
     /**
   Internal. For test use only. May change without notice.
@@ -3005,13 +2929,13 @@ declare function appendResponseMessages({ messages, responseMessages, _internal:
     _internal?: {
         currentDate?: () => Date;
     };
-}): Message[];
+}): UIMessage[];
 
 /**
 Converts an array of messages from useChat into an array of CoreMessages that can be used
 with the AI core functions (e.g. `streamText`).
  */
-declare function convertToCoreMessages<TOOLS extends ToolSet = never>(messages: Array<Omit<Message, 'id'>>, options?: {
+declare function convertToCoreMessages<TOOLS extends ToolSet = never>(messages: Array<Omit<UIMessage, 'id'>>, options?: {
     tools?: TOOLS;
 }): CoreMessage[];
 
@@ -3035,7 +2959,7 @@ type ToolCallRepairFunction<TOOLS extends ToolSet> = (options: {
     tools: TOOLS;
     parameterSchema: (options: {
         toolName: string;
-    }) => JSONSchema7$1;
+    }) => JSONSchema7;
     error: NoSuchToolError | InvalidToolArgumentsError;
 }) => Promise<LanguageModelV2ToolCall | null>;
 
@@ -3253,7 +3177,7 @@ interface StreamTextResult<TOOLS extends ToolSet, PARTIAL_OUTPUT> {
   
   Resolved when the response is finished.
        */
-    readonly usage: Promise<LanguageModelUsage$1>;
+    readonly usage: Promise<LanguageModelUsage>;
     /**
   Sources that have been used as input to generate the response.
   For multi-step generation, the sources are accumulated from all steps.
@@ -3470,14 +3394,14 @@ type TextStreamPart<TOOLS extends ToolSet> = {
     request: LanguageModelRequestMetadata;
     warnings: CallWarning[] | undefined;
     response: LanguageModelResponseMetadata;
-    usage: LanguageModelUsage$1;
+    usage: LanguageModelUsage;
     finishReason: FinishReason;
     providerMetadata: ProviderMetadata | undefined;
     isContinued: boolean;
 } | {
     type: 'finish';
     finishReason: FinishReason;
-    usage: LanguageModelUsage$1;
+    usage: LanguageModelUsage;
     providerMetadata: ProviderMetadata | undefined;
     /**
      * @deprecated use response on step-finish instead
@@ -3790,7 +3714,7 @@ declare function generateImage({ model, prompt, n, size, aspectRatio, seed, prov
   }
   ```
        */
-    providerOptions?: Record<string, Record<string, JSONValue$1>>;
+    providerOptions?: ProviderOptions;
     /**
   Maximum number of retries per embedding model call. Set to 0 to disable retries.
   
@@ -3823,7 +3747,7 @@ interface GenerateObjectResult<OBJECT> {
     /**
     The token usage of the generated text.
        */
-    readonly usage: LanguageModelUsage$1;
+    readonly usage: LanguageModelUsage;
     /**
     Warnings from the model provider (e.g. unsupported settings).
        */
@@ -3951,7 +3875,7 @@ interface StreamObjectResult<PARTIAL, RESULT, ELEMENT_STREAM> {
     /**
     The token usage of the generated response. Resolved when the response is finished.
        */
-    readonly usage: Promise<LanguageModelUsage$1>;
+    readonly usage: Promise<LanguageModelUsage>;
     /**
   Additional provider-specific metadata. They are passed through
   from the provider to the AI SDK and enable provider-specific
@@ -4022,7 +3946,7 @@ type ObjectStreamPart<PARTIAL> = {
 } | {
     type: 'finish';
     finishReason: FinishReason;
-    usage: LanguageModelUsage$1;
+    usage: LanguageModelUsage;
     response: LanguageModelResponseMetadata;
     providerMetadata?: ProviderMetadata;
 };
@@ -4044,7 +3968,7 @@ type StreamObjectOnFinishCallback<RESULT> = (event: {
     /**
   The token usage of the generated response.
   */
-    usage: LanguageModelUsage$1;
+    usage: LanguageModelUsage;
     /**
   The generated object. Can be undefined if the final object does not match the schema.
   */
@@ -4074,7 +3998,14 @@ This function streams the output. If you do not want to stream the output, use `
 @return
 A result object for accessing the partial object stream and additional information.
  */
-declare function streamObject<RESULT extends SCHEMA extends z.Schema ? Output extends 'array' ? Array<z.infer<SCHEMA>> : z.infer<SCHEMA> : SCHEMA extends Schema<infer T> ? Output extends 'array' ? Array<T> : T : never, SCHEMA extends z.Schema | Schema = z.Schema<JSONValue$1>, Output extends 'object' | 'array' | 'no-schema' = 'object'>(options: Omit<CallSettings, 'stopSequences'> & Prompt & (Output extends 'no-schema' ? {} : {
+declare function streamObject<RESULT extends SCHEMA extends z.Schema ? Output extends 'array' ? Array<z.infer<SCHEMA>> : z.infer<SCHEMA> : SCHEMA extends Schema<infer T> ? Output extends 'array' ? Array<T> : T : never, SCHEMA extends z.Schema | Schema = z.Schema<JSONValue$1>, Output extends 'object' | 'array' | 'enum' | 'no-schema' = RESULT extends string ? 'enum' : 'object'>(options: Omit<CallSettings, 'stopSequences'> & Prompt & (Output extends 'enum' ? {
+    /**
+The enum values that the model should use.
+  */
+    enum: Array<RESULT>;
+    mode?: 'json';
+    output: 'enum';
+} : Output extends 'no-schema' ? {} : {
     /**
 The schema of the object that the model should generate.
 */
@@ -4139,7 +4070,7 @@ Callback that is called when the LLM response and the final object validation ar
         currentDate?: () => Date;
         now?: () => number;
     };
-}): StreamObjectResult<Output extends 'array' ? RESULT : DeepPartial<RESULT>, Output extends 'array' ? RESULT : RESULT, Output extends 'array' ? RESULT extends Array<infer U> ? AsyncIterableStream<U> : never : never>;
+}): StreamObjectResult<Output extends 'enum' ? string : Output extends 'array' ? RESULT : DeepPartial<RESULT>, Output extends 'array' ? RESULT : RESULT, Output extends 'array' ? RESULT extends Array<infer U> ? AsyncIterableStream<U> : never : never>;
 
 /**
  * A generated audio file.
@@ -4473,6 +4404,8 @@ declare function createProviderRegistry<PROVIDERS extends Record<string, Provide
  */
 declare const experimental_createProviderRegistry: typeof createProviderRegistry;
 
+declare function getToolInvocations(message: UIMessage): ToolInvocation[];
+
 /**
  * Calculates the cosine similarity between two vectors. This is a useful metric for
  * comparing the similarity of two vectors such as embeddings.
@@ -4556,7 +4489,7 @@ type SingleRequestTextStreamPart<TOOLS extends ToolSet> = {
 } | {
     type: 'finish';
     finishReason: FinishReason;
-    usage: LanguageModelUsage$1;
+    usage: LanguageModelUsage;
     providerMetadata?: ProviderMetadata;
 } | {
     type: 'error';
@@ -4620,7 +4553,7 @@ declare class NoObjectGeneratedError extends AISDKError {
     /**
     The usage of the model.
      */
-    readonly usage: LanguageModelUsage$1 | undefined;
+    readonly usage: LanguageModelUsage | undefined;
     /**
     Reason why the model finished generating a response.
      */
@@ -4630,7 +4563,7 @@ declare class NoObjectGeneratedError extends AISDKError {
         cause?: Error;
         text?: string;
         response: LanguageModelResponseMetadata;
-        usage: LanguageModelUsage$1;
+        usage: LanguageModelUsage;
         finishReason: FinishReason;
     });
     static isInstance(error: unknown): error is NoObjectGeneratedError;
@@ -4716,9 +4649,9 @@ declare class InvalidMessageRoleError extends AISDKError {
 declare const symbol$2: unique symbol;
 declare class MessageConversionError extends AISDKError {
     private readonly [symbol$2];
-    readonly originalMessage: Omit<Message, 'id'>;
+    readonly originalMessage: Omit<UIMessage, 'id'>;
     constructor({ originalMessage, message, }: {
-        originalMessage: Omit<Message, 'id'>;
+        originalMessage: Omit<UIMessage, 'id'>;
         message: string;
     });
     static isInstance(error: unknown): error is MessageConversionError;
@@ -4755,4 +4688,4 @@ declare class RetryError extends AISDKError {
     static isInstance(error: unknown): error is RetryError;
 }
 
-export { AssistantContent, Attachment, CallSettings, CallWarning, ChatRequest, ChatRequestOptions, ChunkDetector, CoreAssistantMessage, CoreMessage, CoreSystemMessage, CoreToolMessage, CoreUserMessage, CreateMessage, DataContent, DataStreamOptions, DataStreamPart, DataStreamString, DataStreamWriter, DeepPartial, DownloadError, EmbedManyResult, EmbedResult, Embedding, EmbeddingModel, EmbeddingModelUsage, GenerateImageResult as Experimental_GenerateImageResult, GeneratedFile as Experimental_GeneratedImage, SpeechResult as Experimental_SpeechResult, TranscriptionResult as Experimental_TranscriptionResult, FilePart, FileUIPart, FinishReason, GenerateObjectResult, GenerateTextOnStepFinishCallback, GenerateTextResult, GeneratedAudioFile, GeneratedFile, IdGenerator, ImageModel, ImageGenerationWarning as ImageModelCallWarning, ImageModelResponseMetadata, ImagePart, InvalidArgumentError, InvalidDataContentError, InvalidMessageRoleError, InvalidStreamPartError, InvalidToolArgumentsError, JSONRPCError, JSONRPCMessage, JSONRPCNotification, JSONRPCRequest, JSONRPCResponse, JSONValue, LanguageModel, LanguageModelRequestMetadata, LanguageModelResponseMetadata, LanguageModelUsage$1 as LanguageModelUsage, MCPClientError, MCPTransport, Message, MessageConversionError, NoImageGeneratedError, NoObjectGeneratedError, NoOutputSpecifiedError, NoSuchProviderError, NoSuchToolError, ObjectStreamPart, output as Output, Prompt, Provider, ProviderMetadata, ProviderOptions, ProviderRegistryProvider, ReasoningUIPart, RepairTextFunction, RequestOptions, RetryError, Schema, SourceUIPart, SpeechModel, SpeechModelResponseMetadata, SpeechWarning, StepResult, StepStartUIPart, StreamData, StreamObjectOnFinishCallback, StreamObjectResult, StreamTextOnChunkCallback, StreamTextOnErrorCallback, StreamTextOnFinishCallback, StreamTextOnStepFinishCallback, StreamTextResult, StreamTextTransform, TelemetrySettings, TextPart, TextStreamPart, TextUIPart, Tool, ToolCallPart, ToolCallRepairError, ToolCallRepairFunction, ToolCallUnion, ToolChoice, ToolContent, ToolExecutionError, ToolExecutionOptions, ToolInvocation, ToolInvocationUIPart, ToolResultPart, ToolResultUnion, ToolSet, TranscriptionModel, TranscriptionModelResponseMetadata, TranscriptionWarning, UIMessage, UseChatOptions, UseCompletionOptions, UserContent, appendClientMessage, appendResponseMessages, asSchema, callChatApi, callCompletionApi, convertToCoreMessages, coreAssistantMessageSchema, coreMessageSchema, coreSystemMessageSchema, coreToolMessageSchema, coreUserMessageSchema, cosineSimilarity, createDataStream, createDataStreamResponse, createProviderRegistry, customProvider, defaultSettingsMiddleware, embed, embedMany, createMCPClient as experimental_createMCPClient, experimental_createProviderRegistry, experimental_customProvider, generateImage as experimental_generateImage, generateSpeech as experimental_generateSpeech, transcribe as experimental_transcribe, extractMaxToolInvocationStep, extractReasoningMiddleware, fillMessageParts, formatDataStreamPart, generateObject, generateText, getMessageParts, getTextFromDataUrl, isAssistantMessageWithCompletedToolCalls, isDeepEqualData, jsonSchema, parseDataStreamPart, parsePartialJson, pipeDataStreamToResponse, prepareAttachmentsForRequest, processDataStream, processTextStream, shouldResubmitMessages, simulateReadableStream, simulateStreamingMiddleware, smoothStream, streamObject, streamText, tool, updateToolCallResult, wrapLanguageModel, zodSchema };
+export { AssistantContent, Attachment, CallSettings, CallWarning, ChatRequest, ChatRequestOptions, ChunkDetector, CoreAssistantMessage, CoreMessage, CoreSystemMessage, CoreToolMessage, CoreUserMessage, CreateUIMessage, DataContent, DataStreamOptions, DataStreamPart, DataStreamString, DataStreamWriter, DeepPartial, DownloadError, EmbedManyResult, EmbedResult, Embedding, EmbeddingModel, EmbeddingModelUsage, GenerateImageResult as Experimental_GenerateImageResult, GeneratedFile as Experimental_GeneratedImage, SpeechResult as Experimental_SpeechResult, TranscriptionResult as Experimental_TranscriptionResult, FilePart, FileUIPart, FinishReason, GenerateObjectResult, GenerateTextOnStepFinishCallback, GenerateTextResult, GeneratedAudioFile, GeneratedFile, IdGenerator, ImageModel, ImageGenerationWarning as ImageModelCallWarning, ImageModelResponseMetadata, ImagePart, InvalidArgumentError, InvalidDataContentError, InvalidMessageRoleError, InvalidStreamPartError, InvalidToolArgumentsError, JSONRPCError, JSONRPCMessage, JSONRPCNotification, JSONRPCRequest, JSONRPCResponse, JSONValue, LanguageModel, LanguageModelRequestMetadata, LanguageModelResponseMetadata, LanguageModelUsage, MCPClientError, MCPTransport, MessageConversionError, NoImageGeneratedError, NoObjectGeneratedError, NoOutputSpecifiedError, NoSuchProviderError, NoSuchToolError, ObjectStreamPart, output as Output, Prompt, Provider, ProviderMetadata, ProviderOptions, ProviderRegistryProvider, ReasoningUIPart, RepairTextFunction, RequestOptions, RetryError, SourceUIPart, SpeechModel, SpeechModelResponseMetadata, SpeechWarning, StepResult, StepStartUIPart, StreamData, StreamObjectOnFinishCallback, StreamObjectResult, StreamTextOnChunkCallback, StreamTextOnErrorCallback, StreamTextOnFinishCallback, StreamTextOnStepFinishCallback, StreamTextResult, StreamTextTransform, TelemetrySettings, TextPart, TextStreamPart, TextUIPart, Tool, ToolCallPart, ToolCallRepairError, ToolCallRepairFunction, ToolCallUnion, ToolChoice, ToolContent, ToolExecutionError, ToolExecutionOptions, ToolInvocation, ToolInvocationUIPart, ToolResultPart, ToolResultUnion, ToolSet, TranscriptionModel, TranscriptionModelResponseMetadata, TranscriptionWarning, UIMessage, UIMessagePart, UseChatOptions, UseCompletionOptions, UserContent, appendClientMessage, appendResponseMessages, callChatApi, callCompletionApi, convertToCoreMessages, coreAssistantMessageSchema, coreMessageSchema, coreSystemMessageSchema, coreToolMessageSchema, coreUserMessageSchema, cosineSimilarity, createDataStream, createDataStreamResponse, createProviderRegistry, customProvider, defaultSettingsMiddleware, embed, embedMany, createMCPClient as experimental_createMCPClient, experimental_createProviderRegistry, experimental_customProvider, generateImage as experimental_generateImage, generateSpeech as experimental_generateSpeech, transcribe as experimental_transcribe, extractMaxToolInvocationStep, extractReasoningMiddleware, formatDataStreamPart, generateObject, generateText, getTextFromDataUrl, getToolInvocations, isAssistantMessageWithCompletedToolCalls, isDeepEqualData, parseDataStreamPart, parsePartialJson, pipeDataStreamToResponse, prepareAttachmentsForRequest, processDataStream, processTextStream, shouldResubmitMessages, simulateReadableStream, simulateStreamingMiddleware, smoothStream, streamObject, streamText, tool, updateToolCallResult, wrapLanguageModel };