@adaline/vertex 0.15.0 → 0.17.0

package/dist/index.d.mts

@@ -1,7 +1,7 @@
 import { z } from 'zod';
 import { BaseChatModel } from '@adaline/google';
 import { ChatModelSchemaType, HeadersType, EmbeddingModelV1, EmbeddingModelSchemaType, UrlType, ParamsType, ProviderV1, ChatModelV1 } from '@adaline/provider';
-import { EmbeddingRequestsType, ConfigType, EmbeddingResponseType } from '@adaline/types';
+import { ChatModelPriceType, EmbeddingRequestsType, ConfigType, EmbeddingResponseType } from '@adaline/types';
 
 declare const BaseChatModelOptions: z.ZodObject<{
     accessToken: z.ZodString;
@@ -36,17 +36,32 @@ declare class BaseChatModelVertex extends BaseChatModel {
     private readonly publisher;
     constructor(modelSchema: ChatModelSchemaType, options: BaseChatModelOptionsType);
     getDefaultHeaders(): HeadersType;
+    getModelPricing(): ChatModelPriceType;
 }
 
 declare const Gemini1_0Pro001Literal: "gemini-1.0-pro-001";
 declare const Gemini1_0Pro001Schema: {
-    description: string;
-    maxOutputTokens: number;
-    name: string;
-    roles: Partial<Record<"system" | "user" | "assistant" | "tool", string | undefined>>;
-    modalities: ["text" | "tool-call" | "tool-response", ...("text" | "tool-call" | "tool-response")[]];
-    maxInputTokens: number;
-    config: {
+    readonly price: {
+        modelName: string;
+        currency: string;
+        tokenRanges: {
+            minTokens: number;
+            maxTokens: null;
+            prices: {
+                base: {
+                    inputPricePerMillion: number;
+                    outputPricePerMillion: number;
+                };
+            };
+        }[];
+    };
+    readonly description: string;
+    readonly maxOutputTokens: number;
+    readonly name: string;
+    readonly roles: Partial<Record<"system" | "user" | "assistant" | "tool", string | undefined>>;
+    readonly modalities: ["text" | "tool-call" | "tool-response", ...("text" | "tool-call" | "tool-response")[]];
+    readonly maxInputTokens: number;
+    readonly config: {
         def: Record<string, {
             type: "multi-string";
             param: string;
@@ -84,6 +99,7 @@ declare const Gemini1_0Pro001Schema: {
         }>;
         schema: z.ZodObject<z.ZodRawShape, z.UnknownKeysParam, z.ZodTypeAny, unknown, unknown>;
     };
+    readonly maxReasoningTokens?: number | undefined;
 };
 declare const Gemini1_0Pro001Options: z.ZodObject<{
     accessToken: z.ZodString;
@@ -114,6 +130,20 @@ declare class Gemini1_0Pro001 extends BaseChatModelVertex {
 
 declare const Gemini1_0ProVisionLiteral: "gemini-1.0-pro-vision";
 declare const Gemini1_0ProVisionSchema: {
+    price: {
+        modelName: string;
+        currency: string;
+        tokenRanges: {
+            minTokens: number;
+            maxTokens: null;
+            prices: {
+                base: {
+                    inputPricePerMillion: number;
+                    outputPricePerMillion: number;
+                };
+            };
+        }[];
+    };
     description: string;
     maxOutputTokens: number;
     name: string;
@@ -158,6 +188,7 @@ declare const Gemini1_0ProVisionSchema: {
         }>;
         schema: z.ZodObject<z.ZodRawShape, z.UnknownKeysParam, z.ZodTypeAny, unknown, unknown>;
     };
+    maxReasoningTokens?: number | undefined;
 };
 declare const Gemini1_0ProVisionOptions: z.ZodObject<{
     accessToken: z.ZodString;
@@ -188,13 +219,27 @@ declare class Gemini1_0ProVision extends BaseChatModelVertex {
 
 declare const Gemini1_0ProLiteral: "gemini-1.0-pro";
 declare const Gemini1_0ProSchema: {
-    description: string;
-    maxOutputTokens: number;
-    name: string;
-    roles: Partial<Record<"system" | "user" | "assistant" | "tool", string | undefined>>;
-    modalities: ["text" | "tool-call" | "tool-response", ...("text" | "tool-call" | "tool-response")[]];
-    maxInputTokens: number;
-    config: {
+    readonly price: {
+        modelName: string;
+        currency: string;
+        tokenRanges: {
+            minTokens: number;
+            maxTokens: null;
+            prices: {
+                base: {
+                    inputPricePerMillion: number;
+                    outputPricePerMillion: number;
+                };
+            };
+        }[];
+    };
+    readonly description: string;
+    readonly maxOutputTokens: number;
+    readonly name: string;
+    readonly roles: Partial<Record<"system" | "user" | "assistant" | "tool", string | undefined>>;
+    readonly modalities: ["text" | "tool-call" | "tool-response", ...("text" | "tool-call" | "tool-response")[]];
+    readonly maxInputTokens: number;
+    readonly config: {
         def: Record<string, {
             type: "multi-string";
             param: string;
@@ -232,6 +277,7 @@ declare const Gemini1_0ProSchema: {
         }>;
         schema: z.ZodObject<z.ZodRawShape, z.UnknownKeysParam, z.ZodTypeAny, unknown, unknown>;
     };
+    readonly maxReasoningTokens?: number | undefined;
 };
 declare const Gemini1_0ProOptions: z.ZodObject<{
     accessToken: z.ZodString;
@@ -262,13 +308,36 @@ declare class Gemini1_0Pro extends BaseChatModelVertex {
 
 declare const Gemini1_5Flash001Literal: "gemini-1.5-flash-001";
 declare const Gemini1_5Flash001Schema: {
-    description: string;
-    maxOutputTokens: number;
-    name: string;
-    roles: Partial<Record<"system" | "user" | "assistant" | "tool", string | undefined>>;
-    modalities: ["text" | "image" | "tool-call" | "tool-response", ...("text" | "image" | "tool-call" | "tool-response")[]];
-    maxInputTokens: number;
-    config: {
+    readonly price: {
+        modelName: string;
+        currency: string;
+        tokenRanges: ({
+            minTokens: number;
+            maxTokens: number;
+            prices: {
+                base: {
+                    inputPricePerMillion: number;
+                    outputPricePerMillion: number;
+                };
+            };
+        } | {
+            minTokens: number;
+            maxTokens: null;
+            prices: {
+                base: {
+                    inputPricePerMillion: number;
+                    outputPricePerMillion: number;
+                };
+            };
+        })[];
+    };
+    readonly description: string;
+    readonly maxOutputTokens: number;
+    readonly name: string;
+    readonly roles: Partial<Record<"system" | "user" | "assistant" | "tool", string | undefined>>;
+    readonly modalities: ["text" | "image" | "tool-call" | "tool-response", ...("text" | "image" | "tool-call" | "tool-response")[]];
+    readonly maxInputTokens: number;
+    readonly config: {
         def: Record<string, {
             type: "multi-string";
             param: string;
@@ -306,6 +375,7 @@ declare const Gemini1_5Flash001Schema: {
         }>;
         schema: z.ZodObject<z.ZodRawShape, z.UnknownKeysParam, z.ZodTypeAny, unknown, unknown>;
     };
+    readonly maxReasoningTokens?: number | undefined;
 };
 declare const Gemini1_5Flash001Options: z.ZodObject<{
     accessToken: z.ZodString;
@@ -336,6 +406,29 @@ declare class Gemini1_5Flash001 extends BaseChatModelVertex {
 
 declare const Gemini1_5Flash002Literal: "gemini-1.5-flash-002";
 declare const Gemini1_5Flash002Schema: {
+    price: {
+        modelName: string;
+        currency: string;
+        tokenRanges: ({
+            minTokens: number;
+            maxTokens: number;
+            prices: {
+                base: {
+                    inputPricePerMillion: number;
+                    outputPricePerMillion: number;
+                };
+            };
+        } | {
+            minTokens: number;
+            maxTokens: null;
+            prices: {
+                base: {
+                    inputPricePerMillion: number;
+                    outputPricePerMillion: number;
+                };
+            };
+        })[];
+    };
     description: string;
     maxOutputTokens: number;
     name: string;
@@ -380,6 +473,7 @@ declare const Gemini1_5Flash002Schema: {
         }>;
         schema: z.ZodObject<z.ZodRawShape, z.UnknownKeysParam, z.ZodTypeAny, unknown, unknown>;
     };
+    maxReasoningTokens?: number | undefined;
 };
 declare const Gemini1_5Flash002Options: z.ZodObject<{
     accessToken: z.ZodString;
@@ -409,52 +503,7 @@ declare class Gemini1_5Flash002 extends BaseChatModelVertex {
 }
 
 declare const Gemini1_5FlashLiteral: "gemini-1.5-flash";
-declare const Gemini1_5FlashSchema: {
-    description: string;
-    maxOutputTokens: number;
-    name: string;
-    roles: Partial<Record<"system" | "user" | "assistant" | "tool", string | undefined>>;
-    modalities: ["text" | "image" | "tool-call" | "tool-response", ...("text" | "image" | "tool-call" | "tool-response")[]];
-    maxInputTokens: number;
-    config: {
-        def: Record<string, {
-            type: "multi-string";
-            param: string;
-            title: string;
-            description: string;
-            max: number;
-        } | {
-            type: "object-schema";
-            param: string;
-            title: string;
-            description: string;
-            objectSchema?: any;
-        } | {
-            type: "range";
-            param: string;
-            title: string;
-            description: string;
-            max: number;
-            default: number;
-            min: number;
-            step: number;
-        } | {
-            type: "select-boolean";
-            param: string;
-            title: string;
-            description: string;
-            default: boolean | null;
-        } | {
-            type: "select-string";
-            param: string;
-            title: string;
-            description: string;
-            default: string;
-            choices: string[];
-        }>;
-        schema: z.ZodObject<z.ZodRawShape, z.UnknownKeysParam, z.ZodTypeAny, unknown, unknown>;
-    };
-};
+declare const Gemini1_5FlashSchema: ChatModelSchemaType;
 declare const Gemini1_5FlashOptions: z.ZodObject<{
     accessToken: z.ZodString;
     modelName: z.ZodString;
@@ -484,13 +533,36 @@ declare class Gemini1_5Flash extends BaseChatModelVertex {
 
 declare const Gemini1_5Pro001Literal: "gemini-1.5-pro-001";
 declare const Gemini1_5Pro001Schema: {
-    description: string;
-    maxOutputTokens: number;
-    name: string;
-    roles: Partial<Record<"system" | "user" | "assistant" | "tool", string | undefined>>;
-    modalities: ["text" | "image" | "tool-call" | "tool-response", ...("text" | "image" | "tool-call" | "tool-response")[]];
-    maxInputTokens: number;
-    config: {
+    readonly price: {
+        modelName: string;
+        currency: string;
+        tokenRanges: ({
+            minTokens: number;
+            maxTokens: number;
+            prices: {
+                base: {
+                    inputPricePerMillion: number;
+                    outputPricePerMillion: number;
+                };
+            };
+        } | {
+            minTokens: number;
+            maxTokens: null;
+            prices: {
+                base: {
+                    inputPricePerMillion: number;
+                    outputPricePerMillion: number;
+                };
+            };
+        })[];
+    };
+    readonly description: string;
+    readonly maxOutputTokens: number;
+    readonly name: string;
+    readonly roles: Partial<Record<"system" | "user" | "assistant" | "tool", string | undefined>>;
+    readonly modalities: ["text" | "image" | "tool-call" | "tool-response", ...("text" | "image" | "tool-call" | "tool-response")[]];
+    readonly maxInputTokens: number;
+    readonly config: {
         def: Record<string, {
             type: "multi-string";
             param: string;
@@ -528,6 +600,7 @@ declare const Gemini1_5Pro001Schema: {
         }>;
         schema: z.ZodObject<z.ZodRawShape, z.UnknownKeysParam, z.ZodTypeAny, unknown, unknown>;
     };
+    readonly maxReasoningTokens?: number | undefined;
 };
 declare const Gemini1_5Pro001Options: z.ZodObject<{
     accessToken: z.ZodString;
@@ -558,13 +631,36 @@ declare class Gemini1_5Pro001 extends BaseChatModelVertex {
 
 declare const Gemini1_5Pro002Literal: "gemini-1.5-pro-002";
 declare const Gemini1_5Pro002Schema: {
-    description: string;
-    maxOutputTokens: number;
-    name: string;
-    roles: Partial<Record<"system" | "user" | "assistant" | "tool", string | undefined>>;
-    modalities: ["text" | "image" | "tool-call" | "tool-response", ...("text" | "image" | "tool-call" | "tool-response")[]];
-    maxInputTokens: number;
-    config: {
+    readonly price: {
+        modelName: string;
+        currency: string;
+        tokenRanges: ({
+            minTokens: number;
+            maxTokens: number;
+            prices: {
+                base: {
+                    inputPricePerMillion: number;
+                    outputPricePerMillion: number;
+                };
+            };
+        } | {
+            minTokens: number;
+            maxTokens: null;
+            prices: {
+                base: {
+                    inputPricePerMillion: number;
+                    outputPricePerMillion: number;
+                };
+            };
+        })[];
+    };
+    readonly description: string;
+    readonly maxOutputTokens: number;
+    readonly name: string;
+    readonly roles: Partial<Record<"system" | "user" | "assistant" | "tool", string | undefined>>;
+    readonly modalities: ["text" | "image" | "tool-call" | "tool-response", ...("text" | "image" | "tool-call" | "tool-response")[]];
+    readonly maxInputTokens: number;
+    readonly config: {
         def: Record<string, {
             type: "multi-string";
             param: string;
@@ -602,6 +698,7 @@ declare const Gemini1_5Pro002Schema: {
         }>;
         schema: z.ZodObject<z.ZodRawShape, z.UnknownKeysParam, z.ZodTypeAny, unknown, unknown>;
     };
+    readonly maxReasoningTokens?: number | undefined;
 };
 declare const Gemini1_5Pro002Options: z.ZodObject<{
     accessToken: z.ZodString;
@@ -632,13 +729,36 @@ declare class Gemini1_5Pro002 extends BaseChatModelVertex {
 
 declare const Gemini1_5ProLiteral: "gemini-1.5-pro";
 declare const Gemini1_5ProSchema: {
-    description: string;
-    maxOutputTokens: number;
-    name: string;
-    roles: Partial<Record<"system" | "user" | "assistant" | "tool", string | undefined>>;
-    modalities: ["text" | "image" | "tool-call" | "tool-response", ...("text" | "image" | "tool-call" | "tool-response")[]];
-    maxInputTokens: number;
-    config: {
+    readonly price: {
+        modelName: string;
+        currency: string;
+        tokenRanges: ({
+            minTokens: number;
+            maxTokens: number;
+            prices: {
+                base: {
+                    inputPricePerMillion: number;
+                    outputPricePerMillion: number;
+                };
+            };
+        } | {
+            minTokens: number;
+            maxTokens: null;
+            prices: {
+                base: {
+                    inputPricePerMillion: number;
+                    outputPricePerMillion: number;
+                };
+            };
+        })[];
+    };
+    readonly description: string;
+    readonly maxOutputTokens: number;
+    readonly name: string;
+    readonly roles: Partial<Record<"system" | "user" | "assistant" | "tool", string | undefined>>;
+    readonly modalities: ["text" | "image" | "tool-call" | "tool-response", ...("text" | "image" | "tool-call" | "tool-response")[]];
+    readonly maxInputTokens: number;
+    readonly config: {
         def: Record<string, {
             type: "multi-string";
             param: string;
@@ -676,6 +796,7 @@ declare const Gemini1_5ProSchema: {
         }>;
         schema: z.ZodObject<z.ZodRawShape, z.UnknownKeysParam, z.ZodTypeAny, unknown, unknown>;
     };
+    readonly maxReasoningTokens?: number | undefined;
 };
 declare const Gemini1_5ProOptions: z.ZodObject<{
     accessToken: z.ZodString;
@@ -706,13 +827,27 @@ declare class Gemini1_5Pro extends BaseChatModelVertex {
 
 declare const Gemini2_0FlashExpLiteral: "gemini-2.0-flash-exp";
 declare const Gemini2_0FlashExpSchema: {
-    description: string;
-    maxOutputTokens: number;
-    name: string;
-    roles: Partial<Record<"system" | "user" | "assistant" | "tool", string | undefined>>;
-    modalities: ["text" | "image" | "tool-call" | "tool-response", ...("text" | "image" | "tool-call" | "tool-response")[]];
-    maxInputTokens: number;
-    config: {
+    readonly price: {
+        modelName: string;
+        currency: string;
+        tokenRanges: {
+            minTokens: number;
+            maxTokens: null;
+            prices: {
+                base: {
+                    inputPricePerMillion: number;
+                    outputPricePerMillion: number;
+                };
+            };
+        }[];
+    };
+    readonly description: string;
+    readonly maxOutputTokens: number;
+    readonly name: string;
+    readonly roles: Partial<Record<"system" | "user" | "assistant" | "tool", string | undefined>>;
+    readonly modalities: ["text" | "image" | "tool-call" | "tool-response", ...("text" | "image" | "tool-call" | "tool-response")[]];
+    readonly maxInputTokens: number;
+    readonly config: {
         def: Record<string, {
             type: "multi-string";
             param: string;
@@ -750,6 +885,7 @@ declare const Gemini2_0FlashExpSchema: {
         }>;
         schema: z.ZodObject<z.ZodRawShape, z.UnknownKeysParam, z.ZodTypeAny, unknown, unknown>;
     };
+    readonly maxReasoningTokens?: number | undefined;
 };
 declare const Gemini2_0FlashExpOptions: z.ZodObject<{
     accessToken: z.ZodString;