ai 3.1.0-canary.2 → 3.1.0-canary.3

package/core/dist/index.d.ts

@@ -51,6 +51,10 @@ type LanguageModelV1CallSettings = {
      * by the model, calls will generate deterministic results.
      */
     seed?: number;
+    /**
+     * Abort signal for cancelling the operation.
+     */
+    abortSignal?: AbortSignal;
 };
 
 /**
@@ -106,9 +110,9 @@ interface LanguageModelV1TextPart {
 interface LanguageModelV1ImagePart {
     type: 'image';
     /**
-     * Image data as a Uint8Array.
+     * Image data as a Uint8Array (e.g. from a Blob or Buffer) or a URL.
      */
-    image: Uint8Array;
+    image: Uint8Array | URL;
     /**
      * Optional mime type of the image.
      */
@@ -234,6 +238,32 @@ type LanguageModelV1 = {
          * model has only generated text.
          */
         toolCalls?: Array<LanguageModelV1FunctionToolCall>;
+        /**
+         * Finish reason.
+         */
+        finishReason: LanguageModelV1FinishReason;
+        /**
+         * Usage information.
+         */
+        usage: {
+            promptTokens: number;
+            completionTokens: number;
+        };
+        /**
+         * Raw prompt and setting information for observability provider integration.
+         */
+        rawCall: {
+            /**
+             * Raw prompt after expansion and conversion to the format that the
+             * provider uses to send the information to their API.
+             */
+            rawPrompt: unknown;
+            /**
+             * Raw settings that are used for the API call. Includes provider-specific
+             * settings.
+             */
+            rawSettings: Record<string, unknown>;
+        };
         warnings?: LanguageModelV1CallWarning[];
     }>;
     /**
@@ -246,6 +276,21 @@ type LanguageModelV1 = {
      */
     doStream(options: LanguageModelV1CallOptions): PromiseLike<{
         stream: ReadableStream<LanguageModelV1StreamPart>;
+        /**
+         * Raw prompt and setting information for observability provider integration.
+         */
+        rawCall: {
+            /**
+             * Raw prompt after expansion and conversion to the format that the
+             * provider uses to send the information to their API.
+             */
+            rawPrompt: unknown;
+            /**
+             * Raw settings that are used for the API call. Includes provider-specific
+             * settings.
+             */
+            rawSettings: Record<string, unknown>;
+        };
         warnings?: LanguageModelV1CallWarning[];
     }>;
 };
@@ -271,6 +316,12 @@ type LanguageModelV1StreamPart = {
     error: unknown;
 };
 
+type TokenUsage = {
+    promptTokens: number;
+    completionTokens: number;
+    totalTokens: number;
+};
+
 type CallSettings = {
     /**
      * Maximum number of tokens to generate.
@@ -313,6 +364,14 @@ type CallSettings = {
      * by the model, calls will generate deterministic results.
      */
     seed?: number;
+    /**
+     * Maximum number of retries. Set to 0 to disable retries. Default is 2.
+     */
+    maxRetries?: number;
+    /**
+     * Abort signal.
+     */
+    abortSignal?: AbortSignal;
 };
 
 /**
@@ -332,9 +391,12 @@ interface TextPart {
 interface ImagePart {
     type: 'image';
     /**
-     * Image data. Can either be a base64-encoded string, a Uint8Array, an ArrayBuffer, or a Buffer.
+     * Image data. Can either be:
+     *
+     * - data: a base64-encoded string, a Uint8Array, an ArrayBuffer, or a Buffer
+     * - URL: a URL that points to the image
      */
-    image: DataContent;
+    image: DataContent | URL;
     /**
      * Optional mime type of the image.
      */
@@ -379,22 +441,26 @@ type Prompt = {
 /**
  * Generate a structured, typed object using a language model.
  */
-declare function generateObject<T>({ model, schema, mode, system, prompt, messages, ...settings }: CallSettings & Prompt & {
+declare function generateObject<T>({ model, schema, mode, system, prompt, messages, maxRetries, abortSignal, ...settings }: CallSettings & Prompt & {
     model: LanguageModelV1;
     schema: z.Schema<T>;
     mode?: 'auto' | 'json' | 'tool' | 'grammar';
 }): Promise<GenerateObjectResult<T>>;
 declare class GenerateObjectResult<T> {
     readonly object: T;
+    readonly finishReason: LanguageModelV1FinishReason;
+    readonly usage: TokenUsage;
     constructor(options: {
         object: T;
+        finishReason: LanguageModelV1FinishReason;
+        usage: TokenUsage;
     });
 }
 
 /**
  * Stream an object as a partial object stream.
  */
-declare function streamObject<T>({ model, schema, mode, system, prompt, messages, ...settings }: CallSettings & Prompt & {
+declare function streamObject<T>({ model, schema, mode, system, prompt, messages, maxRetries, abortSignal, ...settings }: CallSettings & Prompt & {
     model: LanguageModelV1;
     schema: z.Schema<T>;
     mode?: 'auto' | 'json' | 'tool' | 'grammar';
@@ -477,7 +543,7 @@ type ToToolResultArray<TOOLS extends Record<string, Tool>> = Array<ToToolResult<
 /**
  * Generate a text and call tools using a language model.
  */
-declare function generateText<TOOLS extends Record<string, Tool>>({ model, tools, system, prompt, messages, ...settings }: CallSettings & Prompt & {
+declare function generateText<TOOLS extends Record<string, Tool>>({ model, tools, system, prompt, messages, maxRetries, abortSignal, ...settings }: CallSettings & Prompt & {
     model: LanguageModelV1;
     tools?: TOOLS;
 }): Promise<GenerateTextResult<TOOLS>>;
@@ -485,21 +551,21 @@ declare class GenerateTextResult<TOOLS extends Record<string, Tool>> {
     readonly text: string;
     readonly toolCalls: ToToolCallArray<TOOLS>;
     readonly toolResults: ToToolResultArray<TOOLS>;
+    readonly finishReason: LanguageModelV1FinishReason;
+    readonly usage: TokenUsage;
     constructor(options: {
         text: string;
         toolCalls: ToToolCallArray<TOOLS>;
         toolResults: ToToolResultArray<TOOLS>;
+        finishReason: LanguageModelV1FinishReason;
+        usage: TokenUsage;
     });
 }
 
-declare class StreamTextHttpResponse extends Response {
-    constructor(messageStream: ReadableStream<TextStreamPart<any>>);
-}
-
 /**
  * Stream text generated by a language model.
  */
-declare function streamText<TOOLS extends Record<string, Tool>>({ model, tools, system, prompt, messages, ...settings }: CallSettings & Prompt & {
+declare function streamText<TOOLS extends Record<string, Tool>>({ model, tools, system, prompt, messages, maxRetries, abortSignal, ...settings }: CallSettings & Prompt & {
     model: LanguageModelV1;
     tools?: TOOLS;
 }): Promise<StreamTextResult<TOOLS>>;
@@ -519,7 +585,6 @@ declare class StreamTextResult<TOOLS extends Record<string, Tool>> {
     readonly textStream: AsyncIterable<string>;
     readonly fullStream: AsyncIterable<TextStreamPart<TOOLS>>;
     constructor(stream: ReadableStream<TextStreamPart<TOOLS>>);
-    toResponse(): StreamTextHttpResponse;
 }
 
-export { AssistantContent, AssistantMessage, DataContent, ErrorStreamPart, GenerateObjectResult, GenerateTextResult, ImagePart, Message, StreamObjectResult, StreamTextHttpResponse, StreamTextResult, TextPart, TextStreamPart, Tool, ToolCallPart, ToolContent, ToolMessage, ToolResultPart, UserContent, UserMessage, convertDataContentToBase64String, convertDataContentToUint8Array, generateObject, generateText, streamObject, streamText, tool };
+export { AssistantContent, AssistantMessage, DataContent, ErrorStreamPart, GenerateObjectResult, GenerateTextResult, ImagePart, Message, StreamObjectResult, StreamTextResult, TextPart, TextStreamPart, Tool, ToolCallPart, ToolContent, ToolMessage, ToolResultPart, UserContent, UserMessage, convertDataContentToBase64String, convertDataContentToUint8Array, generateObject, generateText, streamObject, streamText, tool };