ai 3.2.34 → 3.2.36

package/dist/index.d.mts

@@ -1,14 +1,34 @@
 import { DeepPartial, Attachment, JSONValue, CreateMessage, FunctionCall as FunctionCall$1, AssistantMessage, DataMessage } from '@ai-sdk/ui-utils';
 export { AssistantMessage, AssistantStatus, ChatRequest, ChatRequestOptions, CreateMessage, DataMessage, DeepPartial, Function, FunctionCall, FunctionCallHandler, IdGenerator, JSONValue, Message, RequestOptions, StreamPart, Tool, ToolCall, ToolCallHandler, ToolChoice, ToolInvocation, UseAssistantOptions, formatStreamPart, parseComplexResponse, parseStreamPart, readDataStream } from '@ai-sdk/ui-utils';
+import { AttributeValue, Span } from '@opentelemetry/api';
 import { EmbeddingModelV1, EmbeddingModelV1Embedding, LanguageModelV1, LanguageModelV1FinishReason, LanguageModelV1LogProbs, LanguageModelV1CallWarning, LanguageModelV1StreamPart } from '@ai-sdk/provider';
 export { APICallError, EmptyResponseBodyError, InvalidArgumentError, InvalidDataContentError, InvalidPromptError, InvalidResponseDataError, InvalidToolArgumentsError, JSONParseError, LoadAPIKeyError, NoObjectGeneratedError, NoSuchToolError, RetryError, ToolCallParseError, TypeValidationError, UnsupportedFunctionalityError, UnsupportedJSONSchemaError } from '@ai-sdk/provider';
 import { z } from 'zod';
-import { AttributeValue, Span } from '@opentelemetry/api';
+import { Validator } from '@ai-sdk/provider-utils';
+import { JSONSchema7 } from 'json-schema';
 import { ServerResponse } from 'http';
 import { ServerResponse as ServerResponse$1 } from 'node:http';
 import { AssistantStream } from 'openai/lib/AssistantStream';
 import { Run } from 'openai/resources/beta/threads/runs/runs';
 
+/**
+ * Telemetry configuration.
+ */
+type TelemetrySettings = {
+    /**
+     * Enable or disable telemetry. Disabled by default while experimental.
+     */
+    isEnabled?: boolean;
+    /**
+     * Identifier for this function. Used to group telemetry data by function.
+     */
+    functionId?: string;
+    /**
+     * Additional information to include in the telemetry data.
+     */
+    metadata?: Record<string, AttributeValue>;
+};
+
 /**
 Represents the number of tokens used in a prompt and completion.
  */
@@ -129,7 +149,7 @@ Embed a value using an embedding model. The type of the value is defined by the
 
 @returns A result object that contains the embedding, the value, and additional information.
  */
-declare function embed<VALUE>({ model, value, maxRetries, abortSignal, headers, }: {
+declare function embed<VALUE>({ model, value, maxRetries, abortSignal, headers, experimental_telemetry: telemetry, }: {
     /**
   The embedding model to use.
        */
@@ -153,6 +173,10 @@ declare function embed<VALUE>({ model, value, maxRetries, abortSignal, headers,
   Only applicable for HTTP-based providers.
    */
     headers?: Record<string, string>;
+    /**
+     * Optional telemetry configuration (experimental).
+     */
+    experimental_telemetry?: TelemetrySettings;
 }): Promise<EmbedResult<VALUE>>;
 
 /**
@@ -190,7 +214,7 @@ has a limit on how many embeddings can be generated in a single call.
 
 @returns A result object that contains the embeddings, the value, and additional information.
  */
-declare function embedMany<VALUE>({ model, values, maxRetries, abortSignal, headers, }: {
+declare function embedMany<VALUE>({ model, values, maxRetries, abortSignal, headers, experimental_telemetry: telemetry, }: {
     /**
   The embedding model to use.
        */
@@ -214,6 +238,10 @@ declare function embedMany<VALUE>({ model, values, maxRetries, abortSignal, head
   Only applicable for HTTP-based providers.
    */
     headers?: Record<string, string>;
+    /**
+     * Optional telemetry configuration (experimental).
+     */
+    experimental_telemetry?: TelemetrySettings;
 }): Promise<EmbedManyResult<VALUE>>;
 
 type CallSettings = {
@@ -473,22 +501,38 @@ type Prompt = {
 };
 
 /**
- * Telemetry configuration.
+ * Used to mark schemas so we can support both Zod and custom schemas.
  */
-type TelemetrySettings = {
+declare const schemaSymbol: unique symbol;
+type Schema<OBJECT = unknown> = Validator<OBJECT> & {
     /**
-     * Enable or disable telemetry. Disabled by default while experimental.
+     * Used to mark schemas so we can support both Zod and custom schemas.
      */
-    isEnabled?: boolean;
+    [schemaSymbol]: true;
     /**
-     * Identifier for this function. Used to group telemetry data by function.
+     * Schema type for inference.
      */
-    functionId?: string;
+    _type: OBJECT;
     /**
-     * Additional information to include in the telemetry data.
+     * The JSON Schema for the schema. It is passed to the providers.
      */
-    metadata?: Record<string, AttributeValue>;
+    readonly jsonSchema: JSONSchema7;
 };
+/**
+ * Create a schema using a JSON Schema.
+ *
+ * @param jsonSchema The JSON Schema for the schema.
+ * @param options.validate Optional. A validation function for the schema.
+ */
+declare function jsonSchema<OBJECT = unknown>(jsonSchema: JSONSchema7, { validate, }?: {
+    validate?: (value: unknown) => {
+        success: true;
+        value: OBJECT;
+    } | {
+        success: false;
+        error: Error;
+    };
+}): Schema<OBJECT>;
 
 /**
 The result of a `generateObject` call.
@@ -571,7 +615,7 @@ If set and supported by the model, calls will generate deterministic results.
 @returns
 A result object that contains the generated object, the finish reason, the token usage, and additional information.
  */
-declare function generateObject<T>({ model, schema, mode, system, prompt, messages, maxRetries, abortSignal, headers, experimental_telemetry: telemetry, ...settings }: Omit<CallSettings, 'stopSequences'> & Prompt & {
+declare function generateObject<T>({ model, schema: inputSchema, mode, system, prompt, messages, maxRetries, abortSignal, headers, experimental_telemetry: telemetry, ...settings }: Omit<CallSettings, 'stopSequences'> & Prompt & {
     /**
 The language model to use.
      */
@@ -579,11 +623,11 @@ The language model to use.
     /**
 The schema of the object that the model should generate.
      */
-    schema: z.Schema<T>;
+    schema: z.Schema<T> | Schema<T>;
     /**
 The mode to use for object generation.
 
-The Zod schema is converted in a JSON schema and used in one of the following ways
+The schema is converted in a JSON schema and used in one of the following ways
 
 - 'auto': The provider will choose the best mode for the model.
 - 'tool': A tool with the JSON schema as parameters is is provided and the provider is instructed to use it.
@@ -748,7 +792,7 @@ If set and supported by the model, calls will generate deterministic results.
 @return
 A result object for accessing the partial object stream and additional information.
  */
-declare function streamObject<T>({ model, schema, mode, system, prompt, messages, maxRetries, abortSignal, headers, onFinish, ...settings }: Omit<CallSettings, 'stopSequences'> & Prompt & {
+declare function streamObject<T>({ model, schema: inputSchema, mode, system, prompt, messages, maxRetries, abortSignal, headers, experimental_telemetry: telemetry, onFinish, ...settings }: Omit<CallSettings, 'stopSequences'> & Prompt & {
     /**
 The language model to use.
      */
@@ -756,11 +800,11 @@ The language model to use.
     /**
 The schema of the object that the model should generate.
  */
-    schema: z.Schema<T>;
+    schema: z.Schema<T> | Schema<T>;
     /**
 The mode to use for object generation.
 
-The Zod schema is converted in a JSON schema and used in one of the following ways
+The schema is converted in a JSON schema and used in one of the following ways
 
 - 'auto': The provider will choose the best mode for the model.
 - 'tool': A tool with the JSON schema as parameters is is provided and the provider is instructed to use it.
@@ -772,6 +816,10 @@ Default and recommended: 'auto' (best mode for the model).
      */
     mode?: 'auto' | 'json' | 'tool';
     /**
+Optional telemetry configuration (experimental).
+     */
+    experimental_telemetry?: TelemetrySettings;
+    /**
 Callback that is called when the LLM response and the final object validation are finished.
      */
     onFinish?: (event: {
@@ -808,12 +856,14 @@ declare class DefaultStreamObjectResult<T> implements StreamObjectResult<T> {
     readonly warnings: StreamObjectResult<T>['warnings'];
     readonly usage: StreamObjectResult<T>['usage'];
     readonly rawResponse: StreamObjectResult<T>['rawResponse'];
-    constructor({ stream, warnings, rawResponse, schema, onFinish, }: {
+    constructor({ stream, warnings, rawResponse, schema, onFinish, rootSpan, doStreamSpan, }: {
         stream: ReadableStream<string | Omit<LanguageModelV1StreamPart, 'text-delta'>>;
         warnings: StreamObjectResult<T>['warnings'];
         rawResponse?: StreamObjectResult<T>['rawResponse'];
-        schema: z.Schema<T>;
+        schema: z.Schema<T> | Schema<T>;
         onFinish: Parameters<typeof streamObject<T>>[0]['onFinish'];
+        rootSpan: Span;
+        doStreamSpan: Span;
     });
     get object(): Promise<T>;
     get partialObjectStream(): AsyncIterableStream<DeepPartial<T>>;
@@ -830,13 +880,15 @@ declare class DefaultStreamObjectResult<T> implements StreamObjectResult<T> {
  */
 declare const experimental_streamObject: typeof streamObject;
 
+type Parameters$1 = z.ZodTypeAny | Schema<any>;
+type inferParameters<PARAMETERS extends Parameters$1> = PARAMETERS extends Schema<any> ? PARAMETERS['_type'] : PARAMETERS extends z.ZodTypeAny ? z.infer<PARAMETERS> : never;
 /**
 A tool contains the description and the schema of the input that the tool expects.
 This enables the language model to generate the input.
 
 The tool can also contain an optional execute function for the actual execution function of the tool.
  */
-interface CoreTool<PARAMETERS extends z.ZodTypeAny = any, RESULT = any> {
+interface CoreTool<PARAMETERS extends Parameters$1 = any, RESULT = any> {
     /**
   An optional description of what the tool does. Will be used by the language model to decide whether to use the tool.
      */
@@ -851,17 +903,17 @@ interface CoreTool<PARAMETERS extends z.ZodTypeAny = any, RESULT = any> {
   An async function that is called with the arguments from the tool call and produces a result.
   If not provided, the tool will not be executed automatically.
      */
-    execute?: (args: z.infer<PARAMETERS>) => PromiseLike<RESULT>;
+    execute?: (args: inferParameters<PARAMETERS>) => PromiseLike<RESULT>;
 }
 /**
 Helper function for inferring the execute args of a tool.
  */
-declare function tool<PARAMETERS extends z.ZodTypeAny, RESULT>(tool: CoreTool<PARAMETERS, RESULT> & {
-    execute: (args: z.infer<PARAMETERS>) => PromiseLike<RESULT>;
+declare function tool<PARAMETERS extends Parameters$1, RESULT>(tool: CoreTool<PARAMETERS, RESULT> & {
+    execute: (args: inferParameters<PARAMETERS>) => PromiseLike<RESULT>;
 }): CoreTool<PARAMETERS, RESULT> & {
-    execute: (args: z.infer<PARAMETERS>) => PromiseLike<RESULT>;
+    execute: (args: inferParameters<PARAMETERS>) => PromiseLike<RESULT>;
 };
-declare function tool<PARAMETERS extends z.ZodTypeAny, RESULT>(tool: CoreTool<PARAMETERS, RESULT> & {
+declare function tool<PARAMETERS extends Parameters$1, RESULT>(tool: CoreTool<PARAMETERS, RESULT> & {
     execute?: undefined;
 }): CoreTool<PARAMETERS, RESULT> & {
     execute: undefined;
@@ -948,7 +1000,7 @@ type ToToolResultObject<TOOLS extends Record<string, CoreTool>> = ValueOf<{
         type: 'tool-result';
         toolCallId: string;
         toolName: NAME & string;
-        args: z.infer<TOOLS[NAME]['parameters']>;
+        args: inferParameters<TOOLS[NAME]['parameters']>;
         result: Awaited<ReturnType<Exclude<TOOLS[NAME]['execute'], undefined>>>;
     };
 }>;
@@ -986,7 +1038,7 @@ type ToToolCall<TOOLS extends Record<string, CoreTool>> = ValueOf<{
         type: 'tool-call';
         toolCallId: string;
         toolName: NAME & string;
-        args: z.infer<TOOLS[NAME]['parameters']>;
+        args: inferParameters<TOOLS[NAME]['parameters']>;
     };
 }>;
 type ToToolCallArray<TOOLS extends Record<string, CoreTool>> = Array<ToToolCall<TOOLS>>;
@@ -2312,4 +2364,4 @@ declare const generateId: (size?: number | undefined) => string;
  */
 declare const nanoid: (size?: number | undefined) => string;
 
-export { AIStream, AIStreamCallbacksAndOptions, AIStreamParser, AIStreamParserOptions, AWSBedrockAnthropicMessagesStream, AWSBedrockAnthropicStream, AWSBedrockCohereStream, AWSBedrockLlama2Stream, AWSBedrockStream, AnthropicStream, AssistantContent, AssistantResponse, CallWarning, CohereStream, CompletionTokenUsage, CompletionUsage, CoreAssistantMessage, CoreMessage, CoreSystemMessage, CoreTool, CoreToolChoice, CoreToolMessage, CoreUserMessage, DataContent, EmbedManyResult, EmbedResult, Embedding, EmbeddingModel, EmbeddingTokenUsage, ExperimentalAssistantMessage, ExperimentalMessage, ExperimentalTool, ExperimentalToolMessage, ExperimentalUserMessage, FinishReason, FunctionCallPayload, GenerateObjectResult, GenerateTextResult, GoogleGenerativeAIStream, HuggingFaceStream, ImagePart, InkeepAIStreamCallbacksAndOptions, InkeepChatResultCallbacks, InkeepOnFinalMetadata, InkeepStream, InvalidMessageRoleError, InvalidModelIdError, langchainAdapter as LangChainAdapter, LangChainStream, LanguageModel, LogProbs, MistralStream, NoSuchModelError, NoSuchProviderError, ObjectStreamInputPart, ObjectStreamPart, OpenAIStream, OpenAIStreamCallbacks, ReplicateStream, StreamData, StreamObjectResult, StreamTextResult, StreamingTextResponse, TextPart$1 as TextPart, TextStreamPart, TokenUsage, ToolCallPart, ToolCallPayload, ToolContent, ToolResultPart, UserContent, convertDataContentToBase64String, convertDataContentToUint8Array, convertToCoreMessages, convertUint8ArrayToText, cosineSimilarity, createCallbacksTransformer, createEventStreamTransformer, createStreamDataTransformer, embed, embedMany, experimental_AssistantResponse, experimental_ModelRegistry, experimental_ProviderRegistry, experimental_StreamData, experimental_createModelRegistry, experimental_createProviderRegistry, experimental_generateObject, experimental_generateText, experimental_streamObject, experimental_streamText, generateId, generateObject, generateText, nanoid, readableFromAsyncIterable, streamObject, streamText, streamToResponse, tool, trimStartOfStreamHelper };
+export { AIStream, AIStreamCallbacksAndOptions, AIStreamParser, AIStreamParserOptions, AWSBedrockAnthropicMessagesStream, AWSBedrockAnthropicStream, AWSBedrockCohereStream, AWSBedrockLlama2Stream, AWSBedrockStream, AnthropicStream, AssistantContent, AssistantResponse, CallWarning, CohereStream, CompletionTokenUsage, CompletionUsage, CoreAssistantMessage, CoreMessage, CoreSystemMessage, CoreTool, CoreToolChoice, CoreToolMessage, CoreUserMessage, DataContent, EmbedManyResult, EmbedResult, Embedding, EmbeddingModel, EmbeddingTokenUsage, ExperimentalAssistantMessage, ExperimentalMessage, ExperimentalTool, ExperimentalToolMessage, ExperimentalUserMessage, FinishReason, FunctionCallPayload, GenerateObjectResult, GenerateTextResult, GoogleGenerativeAIStream, HuggingFaceStream, ImagePart, InkeepAIStreamCallbacksAndOptions, InkeepChatResultCallbacks, InkeepOnFinalMetadata, InkeepStream, InvalidMessageRoleError, InvalidModelIdError, langchainAdapter as LangChainAdapter, LangChainStream, LanguageModel, LogProbs, MistralStream, NoSuchModelError, NoSuchProviderError, ObjectStreamInputPart, ObjectStreamPart, OpenAIStream, OpenAIStreamCallbacks, ReplicateStream, StreamData, StreamObjectResult, StreamTextResult, StreamingTextResponse, TextPart$1 as TextPart, TextStreamPart, TokenUsage, ToolCallPart, ToolCallPayload, ToolContent, ToolResultPart, UserContent, convertDataContentToBase64String, convertDataContentToUint8Array, convertToCoreMessages, convertUint8ArrayToText, cosineSimilarity, createCallbacksTransformer, createEventStreamTransformer, createStreamDataTransformer, embed, embedMany, experimental_AssistantResponse, experimental_ModelRegistry, experimental_ProviderRegistry, experimental_StreamData, experimental_createModelRegistry, experimental_createProviderRegistry, experimental_generateObject, experimental_generateText, experimental_streamObject, experimental_streamText, generateId, generateObject, generateText, jsonSchema, nanoid, readableFromAsyncIterable, streamObject, streamText, streamToResponse, tool, trimStartOfStreamHelper };

package/dist/index.d.ts

@@ -1,14 +1,34 @@
 import { DeepPartial, Attachment, JSONValue, CreateMessage, FunctionCall as FunctionCall$1, AssistantMessage, DataMessage } from '@ai-sdk/ui-utils';
 export { AssistantMessage, AssistantStatus, ChatRequest, ChatRequestOptions, CreateMessage, DataMessage, DeepPartial, Function, FunctionCall, FunctionCallHandler, IdGenerator, JSONValue, Message, RequestOptions, StreamPart, Tool, ToolCall, ToolCallHandler, ToolChoice, ToolInvocation, UseAssistantOptions, formatStreamPart, parseComplexResponse, parseStreamPart, readDataStream } from '@ai-sdk/ui-utils';
+import { AttributeValue, Span } from '@opentelemetry/api';
 import { EmbeddingModelV1, EmbeddingModelV1Embedding, LanguageModelV1, LanguageModelV1FinishReason, LanguageModelV1LogProbs, LanguageModelV1CallWarning, LanguageModelV1StreamPart } from '@ai-sdk/provider';
 export { APICallError, EmptyResponseBodyError, InvalidArgumentError, InvalidDataContentError, InvalidPromptError, InvalidResponseDataError, InvalidToolArgumentsError, JSONParseError, LoadAPIKeyError, NoObjectGeneratedError, NoSuchToolError, RetryError, ToolCallParseError, TypeValidationError, UnsupportedFunctionalityError, UnsupportedJSONSchemaError } from '@ai-sdk/provider';
 import { z } from 'zod';
-import { AttributeValue, Span } from '@opentelemetry/api';
+import { Validator } from '@ai-sdk/provider-utils';
+import { JSONSchema7 } from 'json-schema';
 import { ServerResponse } from 'http';
 import { ServerResponse as ServerResponse$1 } from 'node:http';
 import { AssistantStream } from 'openai/lib/AssistantStream';
 import { Run } from 'openai/resources/beta/threads/runs/runs';
 
+/**
+ * Telemetry configuration.
+ */
+type TelemetrySettings = {
+    /**
+     * Enable or disable telemetry. Disabled by default while experimental.
+     */
+    isEnabled?: boolean;
+    /**
+     * Identifier for this function. Used to group telemetry data by function.
+     */
+    functionId?: string;
+    /**
+     * Additional information to include in the telemetry data.
+     */
+    metadata?: Record<string, AttributeValue>;
+};
+
 /**
 Represents the number of tokens used in a prompt and completion.
  */
@@ -129,7 +149,7 @@ Embed a value using an embedding model. The type of the value is defined by the
 
 @returns A result object that contains the embedding, the value, and additional information.
  */
-declare function embed<VALUE>({ model, value, maxRetries, abortSignal, headers, }: {
+declare function embed<VALUE>({ model, value, maxRetries, abortSignal, headers, experimental_telemetry: telemetry, }: {
     /**
   The embedding model to use.
        */
@@ -153,6 +173,10 @@ declare function embed<VALUE>({ model, value, maxRetries, abortSignal, headers,
   Only applicable for HTTP-based providers.
    */
     headers?: Record<string, string>;
+    /**
+     * Optional telemetry configuration (experimental).
+     */
+    experimental_telemetry?: TelemetrySettings;
 }): Promise<EmbedResult<VALUE>>;
 
 /**
@@ -190,7 +214,7 @@ has a limit on how many embeddings can be generated in a single call.
 
 @returns A result object that contains the embeddings, the value, and additional information.
  */
-declare function embedMany<VALUE>({ model, values, maxRetries, abortSignal, headers, }: {
+declare function embedMany<VALUE>({ model, values, maxRetries, abortSignal, headers, experimental_telemetry: telemetry, }: {
     /**
   The embedding model to use.
        */
@@ -214,6 +238,10 @@ declare function embedMany<VALUE>({ model, values, maxRetries, abortSignal, head
   Only applicable for HTTP-based providers.
    */
     headers?: Record<string, string>;
+    /**
+     * Optional telemetry configuration (experimental).
+     */
+    experimental_telemetry?: TelemetrySettings;
 }): Promise<EmbedManyResult<VALUE>>;
 
 type CallSettings = {
@@ -473,22 +501,38 @@ type Prompt = {
 };
 
 /**
- * Telemetry configuration.
+ * Used to mark schemas so we can support both Zod and custom schemas.
  */
-type TelemetrySettings = {
+declare const schemaSymbol: unique symbol;
+type Schema<OBJECT = unknown> = Validator<OBJECT> & {
     /**
-     * Enable or disable telemetry. Disabled by default while experimental.
+     * Used to mark schemas so we can support both Zod and custom schemas.
      */
-    isEnabled?: boolean;
+    [schemaSymbol]: true;
     /**
-     * Identifier for this function. Used to group telemetry data by function.
+     * Schema type for inference.
      */
-    functionId?: string;
+    _type: OBJECT;
     /**
-     * Additional information to include in the telemetry data.
+     * The JSON Schema for the schema. It is passed to the providers.
      */
-    metadata?: Record<string, AttributeValue>;
+    readonly jsonSchema: JSONSchema7;
 };
+/**
+ * Create a schema using a JSON Schema.
+ *
+ * @param jsonSchema The JSON Schema for the schema.
+ * @param options.validate Optional. A validation function for the schema.
+ */
+declare function jsonSchema<OBJECT = unknown>(jsonSchema: JSONSchema7, { validate, }?: {
+    validate?: (value: unknown) => {
+        success: true;
+        value: OBJECT;
+    } | {
+        success: false;
+        error: Error;
+    };
+}): Schema<OBJECT>;
 
 /**
 The result of a `generateObject` call.
@@ -571,7 +615,7 @@ If set and supported by the model, calls will generate deterministic results.
 @returns
 A result object that contains the generated object, the finish reason, the token usage, and additional information.
  */
-declare function generateObject<T>({ model, schema, mode, system, prompt, messages, maxRetries, abortSignal, headers, experimental_telemetry: telemetry, ...settings }: Omit<CallSettings, 'stopSequences'> & Prompt & {
+declare function generateObject<T>({ model, schema: inputSchema, mode, system, prompt, messages, maxRetries, abortSignal, headers, experimental_telemetry: telemetry, ...settings }: Omit<CallSettings, 'stopSequences'> & Prompt & {
     /**
 The language model to use.
      */
@@ -579,11 +623,11 @@ The language model to use.
     /**
 The schema of the object that the model should generate.
      */
-    schema: z.Schema<T>;
+    schema: z.Schema<T> | Schema<T>;
     /**
 The mode to use for object generation.
 
-The Zod schema is converted in a JSON schema and used in one of the following ways
+The schema is converted in a JSON schema and used in one of the following ways
 
 - 'auto': The provider will choose the best mode for the model.
 - 'tool': A tool with the JSON schema as parameters is is provided and the provider is instructed to use it.
@@ -748,7 +792,7 @@ If set and supported by the model, calls will generate deterministic results.
 @return
 A result object for accessing the partial object stream and additional information.
  */
-declare function streamObject<T>({ model, schema, mode, system, prompt, messages, maxRetries, abortSignal, headers, onFinish, ...settings }: Omit<CallSettings, 'stopSequences'> & Prompt & {
+declare function streamObject<T>({ model, schema: inputSchema, mode, system, prompt, messages, maxRetries, abortSignal, headers, experimental_telemetry: telemetry, onFinish, ...settings }: Omit<CallSettings, 'stopSequences'> & Prompt & {
     /**
 The language model to use.
      */
@@ -756,11 +800,11 @@ The language model to use.
     /**
 The schema of the object that the model should generate.
  */
-    schema: z.Schema<T>;
+    schema: z.Schema<T> | Schema<T>;
     /**
 The mode to use for object generation.
 
-The Zod schema is converted in a JSON schema and used in one of the following ways
+The schema is converted in a JSON schema and used in one of the following ways
 
 - 'auto': The provider will choose the best mode for the model.
 - 'tool': A tool with the JSON schema as parameters is is provided and the provider is instructed to use it.
@@ -772,6 +816,10 @@ Default and recommended: 'auto' (best mode for the model).
      */
     mode?: 'auto' | 'json' | 'tool';
     /**
+Optional telemetry configuration (experimental).
+     */
+    experimental_telemetry?: TelemetrySettings;
+    /**
 Callback that is called when the LLM response and the final object validation are finished.
      */
     onFinish?: (event: {
@@ -808,12 +856,14 @@ declare class DefaultStreamObjectResult<T> implements StreamObjectResult<T> {
     readonly warnings: StreamObjectResult<T>['warnings'];
     readonly usage: StreamObjectResult<T>['usage'];
     readonly rawResponse: StreamObjectResult<T>['rawResponse'];
-    constructor({ stream, warnings, rawResponse, schema, onFinish, }: {
+    constructor({ stream, warnings, rawResponse, schema, onFinish, rootSpan, doStreamSpan, }: {
         stream: ReadableStream<string | Omit<LanguageModelV1StreamPart, 'text-delta'>>;
         warnings: StreamObjectResult<T>['warnings'];
         rawResponse?: StreamObjectResult<T>['rawResponse'];
-        schema: z.Schema<T>;
+        schema: z.Schema<T> | Schema<T>;
         onFinish: Parameters<typeof streamObject<T>>[0]['onFinish'];
+        rootSpan: Span;
+        doStreamSpan: Span;
     });
     get object(): Promise<T>;
     get partialObjectStream(): AsyncIterableStream<DeepPartial<T>>;
@@ -830,13 +880,15 @@ declare class DefaultStreamObjectResult<T> implements StreamObjectResult<T> {
  */
 declare const experimental_streamObject: typeof streamObject;
 
+type Parameters$1 = z.ZodTypeAny | Schema<any>;
+type inferParameters<PARAMETERS extends Parameters$1> = PARAMETERS extends Schema<any> ? PARAMETERS['_type'] : PARAMETERS extends z.ZodTypeAny ? z.infer<PARAMETERS> : never;
 /**
 A tool contains the description and the schema of the input that the tool expects.
 This enables the language model to generate the input.
 
 The tool can also contain an optional execute function for the actual execution function of the tool.
  */
-interface CoreTool<PARAMETERS extends z.ZodTypeAny = any, RESULT = any> {
+interface CoreTool<PARAMETERS extends Parameters$1 = any, RESULT = any> {
     /**
   An optional description of what the tool does. Will be used by the language model to decide whether to use the tool.
      */
@@ -851,17 +903,17 @@ interface CoreTool<PARAMETERS extends z.ZodTypeAny = any, RESULT = any> {
   An async function that is called with the arguments from the tool call and produces a result.
   If not provided, the tool will not be executed automatically.
      */
-    execute?: (args: z.infer<PARAMETERS>) => PromiseLike<RESULT>;
+    execute?: (args: inferParameters<PARAMETERS>) => PromiseLike<RESULT>;
 }
 /**
 Helper function for inferring the execute args of a tool.
  */
-declare function tool<PARAMETERS extends z.ZodTypeAny, RESULT>(tool: CoreTool<PARAMETERS, RESULT> & {
-    execute: (args: z.infer<PARAMETERS>) => PromiseLike<RESULT>;
+declare function tool<PARAMETERS extends Parameters$1, RESULT>(tool: CoreTool<PARAMETERS, RESULT> & {
+    execute: (args: inferParameters<PARAMETERS>) => PromiseLike<RESULT>;
 }): CoreTool<PARAMETERS, RESULT> & {
-    execute: (args: z.infer<PARAMETERS>) => PromiseLike<RESULT>;
+    execute: (args: inferParameters<PARAMETERS>) => PromiseLike<RESULT>;
 };
-declare function tool<PARAMETERS extends z.ZodTypeAny, RESULT>(tool: CoreTool<PARAMETERS, RESULT> & {
+declare function tool<PARAMETERS extends Parameters$1, RESULT>(tool: CoreTool<PARAMETERS, RESULT> & {
     execute?: undefined;
 }): CoreTool<PARAMETERS, RESULT> & {
     execute: undefined;
@@ -948,7 +1000,7 @@ type ToToolResultObject<TOOLS extends Record<string, CoreTool>> = ValueOf<{
         type: 'tool-result';
         toolCallId: string;
         toolName: NAME & string;
-        args: z.infer<TOOLS[NAME]['parameters']>;
+        args: inferParameters<TOOLS[NAME]['parameters']>;
         result: Awaited<ReturnType<Exclude<TOOLS[NAME]['execute'], undefined>>>;
     };
 }>;
@@ -986,7 +1038,7 @@ type ToToolCall<TOOLS extends Record<string, CoreTool>> = ValueOf<{
         type: 'tool-call';
         toolCallId: string;
         toolName: NAME & string;
-        args: z.infer<TOOLS[NAME]['parameters']>;
+        args: inferParameters<TOOLS[NAME]['parameters']>;
     };
 }>;
 type ToToolCallArray<TOOLS extends Record<string, CoreTool>> = Array<ToToolCall<TOOLS>>;
@@ -2312,4 +2364,4 @@ declare const generateId: (size?: number | undefined) => string;
  */
 declare const nanoid: (size?: number | undefined) => string;
 
-export { AIStream, AIStreamCallbacksAndOptions, AIStreamParser, AIStreamParserOptions, AWSBedrockAnthropicMessagesStream, AWSBedrockAnthropicStream, AWSBedrockCohereStream, AWSBedrockLlama2Stream, AWSBedrockStream, AnthropicStream, AssistantContent, AssistantResponse, CallWarning, CohereStream, CompletionTokenUsage, CompletionUsage, CoreAssistantMessage, CoreMessage, CoreSystemMessage, CoreTool, CoreToolChoice, CoreToolMessage, CoreUserMessage, DataContent, EmbedManyResult, EmbedResult, Embedding, EmbeddingModel, EmbeddingTokenUsage, ExperimentalAssistantMessage, ExperimentalMessage, ExperimentalTool, ExperimentalToolMessage, ExperimentalUserMessage, FinishReason, FunctionCallPayload, GenerateObjectResult, GenerateTextResult, GoogleGenerativeAIStream, HuggingFaceStream, ImagePart, InkeepAIStreamCallbacksAndOptions, InkeepChatResultCallbacks, InkeepOnFinalMetadata, InkeepStream, InvalidMessageRoleError, InvalidModelIdError, langchainAdapter as LangChainAdapter, LangChainStream, LanguageModel, LogProbs, MistralStream, NoSuchModelError, NoSuchProviderError, ObjectStreamInputPart, ObjectStreamPart, OpenAIStream, OpenAIStreamCallbacks, ReplicateStream, StreamData, StreamObjectResult, StreamTextResult, StreamingTextResponse, TextPart$1 as TextPart, TextStreamPart, TokenUsage, ToolCallPart, ToolCallPayload, ToolContent, ToolResultPart, UserContent, convertDataContentToBase64String, convertDataContentToUint8Array, convertToCoreMessages, convertUint8ArrayToText, cosineSimilarity, createCallbacksTransformer, createEventStreamTransformer, createStreamDataTransformer, embed, embedMany, experimental_AssistantResponse, experimental_ModelRegistry, experimental_ProviderRegistry, experimental_StreamData, experimental_createModelRegistry, experimental_createProviderRegistry, experimental_generateObject, experimental_generateText, experimental_streamObject, experimental_streamText, generateId, generateObject, generateText, nanoid, readableFromAsyncIterable, streamObject, streamText, streamToResponse, tool, trimStartOfStreamHelper };
+export { AIStream, AIStreamCallbacksAndOptions, AIStreamParser, AIStreamParserOptions, AWSBedrockAnthropicMessagesStream, AWSBedrockAnthropicStream, AWSBedrockCohereStream, AWSBedrockLlama2Stream, AWSBedrockStream, AnthropicStream, AssistantContent, AssistantResponse, CallWarning, CohereStream, CompletionTokenUsage, CompletionUsage, CoreAssistantMessage, CoreMessage, CoreSystemMessage, CoreTool, CoreToolChoice, CoreToolMessage, CoreUserMessage, DataContent, EmbedManyResult, EmbedResult, Embedding, EmbeddingModel, EmbeddingTokenUsage, ExperimentalAssistantMessage, ExperimentalMessage, ExperimentalTool, ExperimentalToolMessage, ExperimentalUserMessage, FinishReason, FunctionCallPayload, GenerateObjectResult, GenerateTextResult, GoogleGenerativeAIStream, HuggingFaceStream, ImagePart, InkeepAIStreamCallbacksAndOptions, InkeepChatResultCallbacks, InkeepOnFinalMetadata, InkeepStream, InvalidMessageRoleError, InvalidModelIdError, langchainAdapter as LangChainAdapter, LangChainStream, LanguageModel, LogProbs, MistralStream, NoSuchModelError, NoSuchProviderError, ObjectStreamInputPart, ObjectStreamPart, OpenAIStream, OpenAIStreamCallbacks, ReplicateStream, StreamData, StreamObjectResult, StreamTextResult, StreamingTextResponse, TextPart$1 as TextPart, TextStreamPart, TokenUsage, ToolCallPart, ToolCallPayload, ToolContent, ToolResultPart, UserContent, convertDataContentToBase64String, convertDataContentToUint8Array, convertToCoreMessages, convertUint8ArrayToText, cosineSimilarity, createCallbacksTransformer, createEventStreamTransformer, createStreamDataTransformer, embed, embedMany, experimental_AssistantResponse, experimental_ModelRegistry, experimental_ProviderRegistry, experimental_StreamData, experimental_createModelRegistry, experimental_createProviderRegistry, experimental_generateObject, experimental_generateText, experimental_streamObject, experimental_streamText, generateId, generateObject, generateText, jsonSchema, nanoid, readableFromAsyncIterable, streamObject, streamText, streamToResponse, tool, trimStartOfStreamHelper };