@tanstack/ai-gemini 0.6.0 → 0.8.0

package/dist/esm/adapters/image.d.ts

@@ -13,14 +13,16 @@ export type GeminiImageModel = (typeof GEMINI_IMAGE_MODELS)[number];
 /**
  * Gemini Image Generation Adapter
  *
- * Tree-shakeable adapter for Gemini Imagen image generation functionality.
- * Supports Imagen 3 and Imagen 4 models.
+ * Tree-shakeable adapter for Gemini image generation functionality.
+ * Supports Imagen 3/4 models (via generateImages API) and Gemini native
+ * image models like Nano Banana 2 (via generateContent API).
  *
  * Features:
  * - Aspect ratio-based image sizing
  * - Person generation controls
  * - Safety filtering
  * - Watermark options
+ * - Extended resolution tiers (Nano Banana 2)
  */
 export declare class GeminiImageAdapter<TModel extends GeminiImageModel> extends BaseImageAdapter<TModel, GeminiImageProviderOptions, GeminiImageModelProviderOptionsByName, GeminiImageModelSizeByName> {
     readonly kind: "image";
@@ -33,8 +35,11 @@ export declare class GeminiImageAdapter<TModel extends GeminiImageModel> extends
     private client;
     constructor(config: GeminiImageConfig, model: TModel);
     generateImages(options: ImageGenerationOptions<GeminiImageProviderOptions>): Promise<ImageGenerationResult>;
-    private buildConfig;
-    private transformResponse;
+    private isGeminiImageModel;
+    private generateWithGeminiApi;
+    private transformGeminiResponse;
+    private buildImagenConfig;
+    private transformImagenResponse;
 }
 /**
  * Creates a Gemini image adapter with explicit API key.

package/dist/esm/adapters/image.js

@@ -1,6 +1,6 @@
 import { BaseImageAdapter } from "@tanstack/ai/adapters";
 import { createGeminiClient, generateId, getGeminiApiKeyFromEnv } from "../utils/client.js";
-import { validatePrompt, validateImageSize, validateNumberOfImages, sizeToAspectRatio } from "../image/image-provider-options.js";
+import { validatePrompt, validateImageSize, validateNumberOfImages, parseNativeImageSize, sizeToAspectRatio } from "../image/image-provider-options.js";
 class GeminiImageAdapter extends BaseImageAdapter {
   constructor(config, model) {
     super({}, model);
@@ -9,19 +9,66 @@ class GeminiImageAdapter extends BaseImageAdapter {
     this.client = createGeminiClient(config);
   }
   async generateImages(options) {
-    const { model, prompt, numberOfImages, size } = options;
+    const { model, prompt } = options;
     validatePrompt({ prompt, model });
-    validateImageSize(model, size);
-    validateNumberOfImages(model, numberOfImages);
-    const config = this.buildConfig(options);
+    if (this.isGeminiImageModel(model)) {
+      return this.generateWithGeminiApi(options);
+    }
+    validateImageSize(model, options.size);
+    validateNumberOfImages(model, options.numberOfImages);
+    const config = this.buildImagenConfig(options);
     const response = await this.client.models.generateImages({
       model,
       prompt,
       config
     });
-    return this.transformResponse(model, response);
+    return this.transformImagenResponse(model, response);
   }
-  buildConfig(options) {
+  isGeminiImageModel(model) {
+    return model.startsWith("gemini-");
+  }
+  async generateWithGeminiApi(options) {
+    const { model, prompt, size, numberOfImages, modelOptions } = options;
+    const parsedSize = size ? parseNativeImageSize(size) : void 0;
+    const augmentedPrompt = numberOfImages && numberOfImages > 1 ? `${prompt} Generate ${numberOfImages} distinct images.` : prompt;
+    const config = {
+      // Include TEXT so the model can interleave descriptions between images
+      responseModalities: ["TEXT", "IMAGE"],
+      ...parsedSize && {
+        imageConfig: {
+          ...parsedSize.aspectRatio && {
+            aspectRatio: parsedSize.aspectRatio
+          },
+          ...parsedSize.resolution && {
+            imageSize: parsedSize.resolution
+          }
+        }
+      },
+      ...modelOptions
+    };
+    const response = await this.client.models.generateContent({
+      model,
+      contents: augmentedPrompt,
+      config
+    });
+    return this.transformGeminiResponse(model, response);
+  }
+  transformGeminiResponse(model, response) {
+    const images = [];
+    const parts = response.candidates?.[0]?.content?.parts ?? [];
+    for (const part of parts) {
+      if (part.inlineData?.data && typeof part.inlineData.data === "string" && part.inlineData.data.length > 0) {
+        images.push({ b64Json: part.inlineData.data });
+      }
+    }
+    return {
+      id: generateId(this.name),
+      model,
+      images,
+      usage: void 0
+    };
+  }
+  buildImagenConfig(options) {
     const { size, numberOfImages, modelOptions } = options;
     return {
       numberOfImages: numberOfImages ?? 1,
@@ -30,7 +77,7 @@ class GeminiImageAdapter extends BaseImageAdapter {
       ...modelOptions
     };
   }
-  transformResponse(model, response) {
+  transformImagenResponse(model, response) {
     const images = (response.generatedImages ?? []).map(
       (item) => ({
         b64Json: item.image?.imageBytes,

package/dist/esm/adapters/image.js.map

@@ -1 +1 @@
-{"version":3,"file":"image.js","sources":["../../../src/adapters/image.ts"],"sourcesContent":["import { BaseImageAdapter } from '@tanstack/ai/adapters'\nimport {\n  createGeminiClient,\n  generateId,\n  getGeminiApiKeyFromEnv,\n} from '../utils'\nimport {\n  sizeToAspectRatio,\n  validateImageSize,\n  validateNumberOfImages,\n  validatePrompt,\n} from '../image/image-provider-options'\nimport type { GEMINI_IMAGE_MODELS } from '../model-meta'\nimport type {\n  GeminiImageModelProviderOptionsByName,\n  GeminiImageModelSizeByName,\n  GeminiImageProviderOptions,\n} from '../image/image-provider-options'\nimport type {\n  GeneratedImage,\n  ImageGenerationOptions,\n  ImageGenerationResult,\n} from '@tanstack/ai'\nimport type {\n  GenerateImagesConfig,\n  GenerateImagesResponse,\n  GoogleGenAI,\n} from '@google/genai'\nimport type { GeminiClientConfig } from '../utils'\n\n/**\n * Configuration for Gemini image adapter\n */\nexport interface GeminiImageConfig extends GeminiClientConfig {}\n\n/** Model type for Gemini Image */\nexport type GeminiImageModel = (typeof GEMINI_IMAGE_MODELS)[number]\n\n/**\n * Gemini Image Generation Adapter\n *\n * Tree-shakeable adapter for Gemini Imagen image generation functionality.\n * Supports Imagen 3 and Imagen 4 models.\n *\n * Features:\n * - Aspect ratio-based image sizing\n * - Person generation controls\n * - Safety filtering\n * - Watermark options\n */\nexport class GeminiImageAdapter<\n  TModel extends GeminiImageModel,\n> extends BaseImageAdapter<\n  TModel,\n  GeminiImageProviderOptions,\n  GeminiImageModelProviderOptionsByName,\n  GeminiImageModelSizeByName\n> {\n  readonly kind = 'image' as const\n  readonly name = 'gemini' as const\n\n  // Type-only property - never assigned at runtime\n  declare '~types': {\n    providerOptions: GeminiImageProviderOptions\n    modelProviderOptionsByName: GeminiImageModelProviderOptionsByName\n    modelSizeByName: GeminiImageModelSizeByName\n  }\n\n  private client: GoogleGenAI\n\n  constructor(config: GeminiImageConfig, model: TModel) {\n    super({}, model)\n    this.client = createGeminiClient(config)\n  }\n\n  async generateImages(\n    options: ImageGenerationOptions<GeminiImageProviderOptions>,\n  ): Promise<ImageGenerationResult> {\n    const { model, prompt, numberOfImages, size } = options\n\n    // Validate inputs\n    validatePrompt({ prompt, model })\n    validateImageSize(model, size)\n    validateNumberOfImages(model, numberOfImages)\n\n    // Build request config\n    const config = this.buildConfig(options)\n\n    const response = await this.client.models.generateImages({\n      model,\n      prompt,\n      config,\n    })\n\n    return this.transformResponse(model, response)\n  }\n\n  private buildConfig(\n    options: ImageGenerationOptions<GeminiImageProviderOptions>,\n  ): GenerateImagesConfig {\n    const { size, numberOfImages, modelOptions } = options\n\n    return {\n      numberOfImages: numberOfImages ?? 1,\n      // Map size to aspect ratio if provided (modelOptions.aspectRatio will override)\n      aspectRatio: size ? sizeToAspectRatio(size) : undefined,\n      ...modelOptions,\n    }\n  }\n\n  private transformResponse(\n    model: string,\n    response: GenerateImagesResponse,\n  ): ImageGenerationResult {\n    const images: Array<GeneratedImage> = (response.generatedImages ?? []).map(\n      (item) => ({\n        b64Json: item.image?.imageBytes,\n        revisedPrompt: item.enhancedPrompt,\n      }),\n    )\n\n    return {\n      id: generateId(this.name),\n      model,\n      images,\n      usage: undefined,\n    }\n  }\n}\n\n/**\n * Creates a Gemini image adapter with explicit API key.\n * Type resolution happens here at the call site.\n *\n * @param model - The model name (e.g., 'imagen-3.0-generate-002')\n * @param apiKey - Your Google API key\n * @param config - Optional additional configuration\n * @returns Configured Gemini image adapter instance with resolved types\n *\n * @example\n * ```typescript\n * const adapter = createGeminiImage('imagen-3.0-generate-002', \"your-api-key\");\n *\n * const result = await generateImage({\n *   adapter,\n *   prompt: 'A cute baby sea otter'\n * });\n * ```\n */\nexport function createGeminiImage<TModel extends GeminiImageModel>(\n  model: TModel,\n  apiKey: string,\n  config?: Omit<GeminiImageConfig, 'apiKey'>,\n): GeminiImageAdapter<TModel> {\n  return new GeminiImageAdapter({ apiKey, ...config }, model)\n}\n\n/**\n * Creates a Gemini image adapter with automatic API key detection from environment variables.\n * Type resolution happens here at the call site.\n *\n * Looks for `GOOGLE_API_KEY` or `GEMINI_API_KEY` in:\n * - `process.env` (Node.js)\n * - `window.env` (Browser with injected env)\n *\n * @param model - The model name (e.g., 'imagen-4.0-generate-001')\n * @param config - Optional configuration (excluding apiKey which is auto-detected)\n * @returns Configured Gemini image adapter instance with resolved types\n * @throws Error if GOOGLE_API_KEY or GEMINI_API_KEY is not found in environment\n *\n * @example\n * ```typescript\n * // Automatically uses GOOGLE_API_KEY from environment\n * const adapter = geminiImage('imagen-4.0-generate-001');\n *\n * const result = await generateImage({\n *   adapter,\n *   prompt: 'A beautiful sunset over mountains'\n * });\n * ```\n */\nexport function geminiImage<TModel extends GeminiImageModel>(\n  model: TModel,\n  config?: Omit<GeminiImageConfig, 'apiKey'>,\n): GeminiImageAdapter<TModel> {\n  const apiKey = getGeminiApiKeyFromEnv()\n  return createGeminiImage(model, apiKey, config)\n}\n"],"names":[],"mappings":";;;AAkDO,MAAM,2BAEH,iBAKR;AAAA,EAaA,YAAY,QAA2B,OAAe;AACpD,UAAM,CAAA,GAAI,KAAK;AAbjB,SAAS,OAAO;AAChB,SAAS,OAAO;AAad,SAAK,SAAS,mBAAmB,MAAM;AAAA,EACzC;AAAA,EAEA,MAAM,eACJ,SACgC;AAChC,UAAM,EAAE,OAAO,QAAQ,gBAAgB,SAAS;AAGhD,mBAAe,EAAE,QAAQ,OAAO;AAChC,sBAAkB,OAAO,IAAI;AAC7B,2BAAuB,OAAO,cAAc;AAG5C,UAAM,SAAS,KAAK,YAAY,OAAO;AAEvC,UAAM,WAAW,MAAM,KAAK,OAAO,OAAO,eAAe;AAAA,MACvD;AAAA,MACA;AAAA,MACA;AAAA,IAAA,CACD;AAED,WAAO,KAAK,kBAAkB,OAAO,QAAQ;AAAA,EAC/C;AAAA,EAEQ,YACN,SACsB;AACtB,UAAM,EAAE,MAAM,gBAAgB,aAAA,IAAiB;AAE/C,WAAO;AAAA,MACL,gBAAgB,kBAAkB;AAAA;AAAA,MAElC,aAAa,OAAO,kBAAkB,IAAI,IAAI;AAAA,MAC9C,GAAG;AAAA,IAAA;AAAA,EAEP;AAAA,EAEQ,kBACN,OACA,UACuB;AACvB,UAAM,UAAiC,SAAS,mBAAmB,CAAA,GAAI;AAAA,MACrE,CAAC,UAAU;AAAA,QACT,SAAS,KAAK,OAAO;AAAA,QACrB,eAAe,KAAK;AAAA,MAAA;AAAA,IACtB;AAGF,WAAO;AAAA,MACL,IAAI,WAAW,KAAK,IAAI;AAAA,MACxB;AAAA,MACA;AAAA,MACA,OAAO;AAAA,IAAA;AAAA,EAEX;AACF;AAqBO,SAAS,kBACd,OACA,QACA,QAC4B;AAC5B,SAAO,IAAI,mBAAmB,EAAE,QAAQ,GAAG,OAAA,GAAU,KAAK;AAC5D;AA0BO,SAAS,YACd,OACA,QAC4B;AAC5B,QAAM,SAAS,uBAAA;AACf,SAAO,kBAAkB,OAAO,QAAQ,MAAM;AAChD;"}
+{"version":3,"file":"image.js","sources":["../../../src/adapters/image.ts"],"sourcesContent":["import { BaseImageAdapter } from '@tanstack/ai/adapters'\nimport {\n  createGeminiClient,\n  generateId,\n  getGeminiApiKeyFromEnv,\n} from '../utils'\nimport {\n  parseNativeImageSize,\n  sizeToAspectRatio,\n  validateImageSize,\n  validateNumberOfImages,\n  validatePrompt,\n} from '../image/image-provider-options'\nimport type { GEMINI_IMAGE_MODELS } from '../model-meta'\nimport type {\n  GeminiImageModelProviderOptionsByName,\n  GeminiImageModelSizeByName,\n  GeminiImageProviderOptions,\n} from '../image/image-provider-options'\nimport type {\n  GeneratedImage,\n  ImageGenerationOptions,\n  ImageGenerationResult,\n} from '@tanstack/ai'\nimport type {\n  GenerateContentConfig,\n  GenerateContentResponse,\n  GenerateImagesConfig,\n  GenerateImagesResponse,\n  GoogleGenAI,\n} from '@google/genai'\nimport type { GeminiClientConfig } from '../utils'\n\n/**\n * Configuration for Gemini image adapter\n */\nexport interface GeminiImageConfig extends GeminiClientConfig {}\n\n/** Model type for Gemini Image */\nexport type GeminiImageModel = (typeof GEMINI_IMAGE_MODELS)[number]\n\n/**\n * Gemini Image Generation Adapter\n *\n * Tree-shakeable adapter for Gemini image generation functionality.\n * Supports Imagen 3/4 models (via generateImages API) and Gemini native\n * image models like Nano Banana 2 (via generateContent API).\n *\n * Features:\n * - Aspect ratio-based image sizing\n * - Person generation controls\n * - Safety filtering\n * - Watermark options\n * - Extended resolution tiers (Nano Banana 2)\n */\nexport class GeminiImageAdapter<\n  TModel extends GeminiImageModel,\n> extends BaseImageAdapter<\n  TModel,\n  GeminiImageProviderOptions,\n  GeminiImageModelProviderOptionsByName,\n  GeminiImageModelSizeByName\n> {\n  readonly kind = 'image' as const\n  readonly name = 'gemini' as const\n\n  // Type-only property - never assigned at runtime\n  declare '~types': {\n    providerOptions: GeminiImageProviderOptions\n    modelProviderOptionsByName: GeminiImageModelProviderOptionsByName\n    modelSizeByName: GeminiImageModelSizeByName\n  }\n\n  private client: GoogleGenAI\n\n  constructor(config: GeminiImageConfig, model: TModel) {\n    super({}, model)\n    this.client = createGeminiClient(config)\n  }\n\n  async generateImages(\n    options: ImageGenerationOptions<GeminiImageProviderOptions>,\n  ): Promise<ImageGenerationResult> {\n    const { model, prompt } = options\n\n    validatePrompt({ prompt, model })\n\n    if (this.isGeminiImageModel(model)) {\n      return this.generateWithGeminiApi(options)\n    }\n\n    // Imagen models path (generateImages API)\n    validateImageSize(model, options.size)\n    validateNumberOfImages(model, options.numberOfImages)\n\n    const config = this.buildImagenConfig(options)\n\n    const response = await this.client.models.generateImages({\n      model,\n      prompt,\n      config,\n    })\n\n    return this.transformImagenResponse(model, response)\n  }\n\n  private isGeminiImageModel(model: string): boolean {\n    return model.startsWith('gemini-')\n  }\n\n  private async generateWithGeminiApi(\n    options: ImageGenerationOptions<GeminiImageProviderOptions>,\n  ): Promise<ImageGenerationResult> {\n    const { model, prompt, size, numberOfImages, modelOptions } = options\n\n    const parsedSize = size ? parseNativeImageSize(size) : undefined\n\n    // The generateContent API has no numberOfImages parameter.\n    // Instead, augment the prompt to request multiple images when needed.\n    const augmentedPrompt =\n      numberOfImages && numberOfImages > 1\n        ? `${prompt} Generate ${numberOfImages} distinct images.`\n        : prompt\n\n    const config: GenerateContentConfig = {\n      // Include TEXT so the model can interleave descriptions between images\n      responseModalities: ['TEXT', 'IMAGE'],\n      ...(parsedSize && {\n        imageConfig: {\n          ...(parsedSize.aspectRatio && {\n            aspectRatio: parsedSize.aspectRatio,\n          }),\n          ...(parsedSize.resolution && {\n            imageSize: parsedSize.resolution,\n          }),\n        },\n      }),\n      ...modelOptions,\n    }\n\n    const response = await this.client.models.generateContent({\n      model,\n      contents: augmentedPrompt,\n      config,\n    })\n\n    return this.transformGeminiResponse(model, response)\n  }\n\n  private transformGeminiResponse(\n    model: string,\n    response: GenerateContentResponse,\n  ): ImageGenerationResult {\n    const images: Array<GeneratedImage> = []\n    const parts = response.candidates?.[0]?.content?.parts ?? []\n\n    for (const part of parts) {\n      if (\n        part.inlineData?.data &&\n        typeof part.inlineData.data === 'string' &&\n        part.inlineData.data.length > 0\n      ) {\n        images.push({ b64Json: part.inlineData.data })\n      }\n    }\n\n    return {\n      id: generateId(this.name),\n      model,\n      images,\n      usage: undefined,\n    }\n  }\n\n  private buildImagenConfig(\n    options: ImageGenerationOptions<GeminiImageProviderOptions>,\n  ): GenerateImagesConfig {\n    const { size, numberOfImages, modelOptions } = options\n\n    return {\n      numberOfImages: numberOfImages ?? 1,\n      // Map size to aspect ratio if provided (modelOptions.aspectRatio will override)\n      aspectRatio: size ? sizeToAspectRatio(size) : undefined,\n      ...modelOptions,\n    }\n  }\n\n  private transformImagenResponse(\n    model: string,\n    response: GenerateImagesResponse,\n  ): ImageGenerationResult {\n    const images: Array<GeneratedImage> = (response.generatedImages ?? []).map(\n      (item) => ({\n        b64Json: item.image?.imageBytes,\n        revisedPrompt: item.enhancedPrompt,\n      }),\n    )\n\n    return {\n      id: generateId(this.name),\n      model,\n      images,\n      usage: undefined,\n    }\n  }\n}\n\n/**\n * Creates a Gemini image adapter with explicit API key.\n * Type resolution happens here at the call site.\n *\n * @param model - The model name (e.g., 'imagen-3.0-generate-002')\n * @param apiKey - Your Google API key\n * @param config - Optional additional configuration\n * @returns Configured Gemini image adapter instance with resolved types\n *\n * @example\n * ```typescript\n * const adapter = createGeminiImage('imagen-3.0-generate-002', \"your-api-key\");\n *\n * const result = await generateImage({\n *   adapter,\n *   prompt: 'A cute baby sea otter'\n * });\n * ```\n */\nexport function createGeminiImage<TModel extends GeminiImageModel>(\n  model: TModel,\n  apiKey: string,\n  config?: Omit<GeminiImageConfig, 'apiKey'>,\n): GeminiImageAdapter<TModel> {\n  return new GeminiImageAdapter({ apiKey, ...config }, model)\n}\n\n/**\n * Creates a Gemini image adapter with automatic API key detection from environment variables.\n * Type resolution happens here at the call site.\n *\n * Looks for `GOOGLE_API_KEY` or `GEMINI_API_KEY` in:\n * - `process.env` (Node.js)\n * - `window.env` (Browser with injected env)\n *\n * @param model - The model name (e.g., 'imagen-4.0-generate-001')\n * @param config - Optional configuration (excluding apiKey which is auto-detected)\n * @returns Configured Gemini image adapter instance with resolved types\n * @throws Error if GOOGLE_API_KEY or GEMINI_API_KEY is not found in environment\n *\n * @example\n * ```typescript\n * // Automatically uses GOOGLE_API_KEY from environment\n * const adapter = geminiImage('imagen-4.0-generate-001');\n *\n * const result = await generateImage({\n *   adapter,\n *   prompt: 'A beautiful sunset over mountains'\n * });\n * ```\n */\nexport function geminiImage<TModel extends GeminiImageModel>(\n  model: TModel,\n  config?: Omit<GeminiImageConfig, 'apiKey'>,\n): GeminiImageAdapter<TModel> {\n  const apiKey = getGeminiApiKeyFromEnv()\n  return createGeminiImage(model, apiKey, config)\n}\n"],"names":[],"mappings":";;;AAuDO,MAAM,2BAEH,iBAKR;AAAA,EAaA,YAAY,QAA2B,OAAe;AACpD,UAAM,CAAA,GAAI,KAAK;AAbjB,SAAS,OAAO;AAChB,SAAS,OAAO;AAad,SAAK,SAAS,mBAAmB,MAAM;AAAA,EACzC;AAAA,EAEA,MAAM,eACJ,SACgC;AAChC,UAAM,EAAE,OAAO,OAAA,IAAW;AAE1B,mBAAe,EAAE,QAAQ,OAAO;AAEhC,QAAI,KAAK,mBAAmB,KAAK,GAAG;AAClC,aAAO,KAAK,sBAAsB,OAAO;AAAA,IAC3C;AAGA,sBAAkB,OAAO,QAAQ,IAAI;AACrC,2BAAuB,OAAO,QAAQ,cAAc;AAEpD,UAAM,SAAS,KAAK,kBAAkB,OAAO;AAE7C,UAAM,WAAW,MAAM,KAAK,OAAO,OAAO,eAAe;AAAA,MACvD;AAAA,MACA;AAAA,MACA;AAAA,IAAA,CACD;AAED,WAAO,KAAK,wBAAwB,OAAO,QAAQ;AAAA,EACrD;AAAA,EAEQ,mBAAmB,OAAwB;AACjD,WAAO,MAAM,WAAW,SAAS;AAAA,EACnC;AAAA,EAEA,MAAc,sBACZ,SACgC;AAChC,UAAM,EAAE,OAAO,QAAQ,MAAM,gBAAgB,iBAAiB;AAE9D,UAAM,aAAa,OAAO,qBAAqB,IAAI,IAAI;AAIvD,UAAM,kBACJ,kBAAkB,iBAAiB,IAC/B,GAAG,MAAM,aAAa,cAAc,sBACpC;AAEN,UAAM,SAAgC;AAAA;AAAA,MAEpC,oBAAoB,CAAC,QAAQ,OAAO;AAAA,MACpC,GAAI,cAAc;AAAA,QAChB,aAAa;AAAA,UACX,GAAI,WAAW,eAAe;AAAA,YAC5B,aAAa,WAAW;AAAA,UAAA;AAAA,UAE1B,GAAI,WAAW,cAAc;AAAA,YAC3B,WAAW,WAAW;AAAA,UAAA;AAAA,QACxB;AAAA,MACF;AAAA,MAEF,GAAG;AAAA,IAAA;AAGL,UAAM,WAAW,MAAM,KAAK,OAAO,OAAO,gBAAgB;AAAA,MACxD;AAAA,MACA,UAAU;AAAA,MACV;AAAA,IAAA,CACD;AAED,WAAO,KAAK,wBAAwB,OAAO,QAAQ;AAAA,EACrD;AAAA,EAEQ,wBACN,OACA,UACuB;AACvB,UAAM,SAAgC,CAAA;AACtC,UAAM,QAAQ,SAAS,aAAa,CAAC,GAAG,SAAS,SAAS,CAAA;AAE1D,eAAW,QAAQ,OAAO;AACxB,UACE,KAAK,YAAY,QACjB,OAAO,KAAK,WAAW,SAAS,YAChC,KAAK,WAAW,KAAK,SAAS,GAC9B;AACA,eAAO,KAAK,EAAE,SAAS,KAAK,WAAW,MAAM;AAAA,MAC/C;AAAA,IACF;AAEA,WAAO;AAAA,MACL,IAAI,WAAW,KAAK,IAAI;AAAA,MACxB;AAAA,MACA;AAAA,MACA,OAAO;AAAA,IAAA;AAAA,EAEX;AAAA,EAEQ,kBACN,SACsB;AACtB,UAAM,EAAE,MAAM,gBAAgB,aAAA,IAAiB;AAE/C,WAAO;AAAA,MACL,gBAAgB,kBAAkB;AAAA;AAAA,MAElC,aAAa,OAAO,kBAAkB,IAAI,IAAI;AAAA,MAC9C,GAAG;AAAA,IAAA;AAAA,EAEP;AAAA,EAEQ,wBACN,OACA,UACuB;AACvB,UAAM,UAAiC,SAAS,mBAAmB,CAAA,GAAI;AAAA,MACrE,CAAC,UAAU;AAAA,QACT,SAAS,KAAK,OAAO;AAAA,QACrB,eAAe,KAAK;AAAA,MAAA;AAAA,IACtB;AAGF,WAAO;AAAA,MACL,IAAI,WAAW,KAAK,IAAI;AAAA,MACxB;AAAA,MACA;AAAA,MACA,OAAO;AAAA,IAAA;AAAA,EAEX;AACF;AAqBO,SAAS,kBACd,OACA,QACA,QAC4B;AAC5B,SAAO,IAAI,mBAAmB,EAAE,QAAQ,GAAG,OAAA,GAAU,KAAK;AAC5D;AA0BO,SAAS,YACd,OACA,QAC4B;AAC5B,QAAM,SAAS,uBAAA;AACf,SAAO,kBAAkB,OAAO,QAAQ,MAAM;AAChD;"}

package/dist/esm/adapters/summarize.d.ts

@@ -9,7 +9,7 @@ export interface GeminiSummarizeConfig extends GeminiClientConfig {
 /**
  * Available Gemini models for summarization
  */
-export declare const GeminiSummarizeModels: readonly ["gemini-2.0-flash", "gemini-1.5-flash", "gemini-1.5-pro", "gemini-2.0-flash-lite"];
+export declare const GeminiSummarizeModels: readonly ["gemini-3.1-flash-lite-preview", "gemini-2.0-flash", "gemini-1.5-flash", "gemini-1.5-pro", "gemini-2.0-flash-lite"];
 export type GeminiSummarizeModel = (typeof GeminiSummarizeModels)[number];
 /**
  * Provider-specific options for Gemini summarization

package/dist/esm/adapters/summarize.js

@@ -1,6 +1,7 @@
 import { FinishReason } from "@google/genai";
 import { createGeminiClient, generateId, getGeminiApiKeyFromEnv } from "../utils/client.js";
 const GeminiSummarizeModels = [
+  "gemini-3.1-flash-lite-preview",
   "gemini-2.0-flash",
   "gemini-1.5-flash",
   "gemini-1.5-pro",

package/dist/esm/adapters/summarize.js.map

@@ -1 +1 @@
-{"version":3,"file":"summarize.js","sources":["../../../src/adapters/summarize.ts"],"sourcesContent":["import { FinishReason } from '@google/genai'\nimport {\n  createGeminiClient,\n  generateId,\n  getGeminiApiKeyFromEnv,\n} from '../utils'\nimport type { GoogleGenAI } from '@google/genai'\nimport type { GeminiClientConfig } from '../utils'\nimport type { SummarizeAdapter } from '@tanstack/ai/adapters'\nimport type {\n  StreamChunk,\n  SummarizationOptions,\n  SummarizationResult,\n} from '@tanstack/ai'\n\n/**\n * Configuration for Gemini summarize adapter\n */\nexport interface GeminiSummarizeConfig extends GeminiClientConfig {}\n/**\n * Available Gemini models for summarization\n */\nexport const GeminiSummarizeModels = [\n  'gemini-2.0-flash',\n  'gemini-1.5-flash',\n  'gemini-1.5-pro',\n  'gemini-2.0-flash-lite',\n] as const\n\nexport type GeminiSummarizeModel = (typeof GeminiSummarizeModels)[number]\n\n/**\n * Provider-specific options for Gemini summarization\n */\nexport interface GeminiSummarizeProviderOptions {\n  /** Generation configuration */\n  generationConfig?: {\n    temperature?: number\n    topP?: number\n    topK?: number\n    maxOutputTokens?: number\n    stopSequences?: Array<string>\n  }\n  /** Safety settings */\n  safetySettings?: Array<{\n    category: string\n    threshold: string\n  }>\n}\n\nexport interface GeminiSummarizeAdapterOptions {\n  // Additional adapter options can be added here\n}\n\n/**\n * Gemini Summarize Adapter\n * A tree-shakeable summarization adapter for Google Gemini\n */\nexport class GeminiSummarizeAdapter<\n  TModel extends GeminiSummarizeModel,\n> implements SummarizeAdapter<TModel, GeminiSummarizeProviderOptions> {\n  readonly kind = 'summarize' as const\n  readonly name = 'gemini' as const\n  readonly model: TModel\n\n  // Type-only property - never assigned at runtime\n  declare '~types': {\n    providerOptions: GeminiSummarizeProviderOptions\n  }\n\n  private client: GoogleGenAI\n\n  constructor(config: GeminiSummarizeConfig, model: TModel) {\n    this.client = createGeminiClient(config)\n    this.model = model\n  }\n\n  async summarize(options: SummarizationOptions): Promise<SummarizationResult> {\n    const model = options.model\n\n    // Build the system prompt based on format\n    const formatInstructions = this.getFormatInstructions(options.style)\n    const lengthInstructions = options.maxLength\n      ? ` Keep the summary under ${options.maxLength} tokens.`\n      : ''\n\n    const systemPrompt = `You are a helpful assistant that summarizes text. ${formatInstructions}${lengthInstructions}`\n\n    const response = await this.client.models.generateContent({\n      model,\n      contents: [\n        {\n          role: 'user',\n          parts: [\n            { text: `Please summarize the following:\\n\\n${options.text}` },\n          ],\n        },\n      ],\n      config: {\n        systemInstruction: systemPrompt,\n      },\n    })\n\n    const summary = response.text ?? ''\n    const inputTokens = response.usageMetadata?.promptTokenCount ?? 0\n    const outputTokens = response.usageMetadata?.candidatesTokenCount ?? 0\n\n    return {\n      id: generateId('sum'),\n      model,\n      summary,\n      usage: {\n        promptTokens: inputTokens,\n        completionTokens: outputTokens,\n        totalTokens: inputTokens + outputTokens,\n      },\n    }\n  }\n\n  async *summarizeStream(\n    options: SummarizationOptions,\n  ): AsyncIterable<StreamChunk> {\n    const model = options.model\n    const id = generateId('sum')\n    let accumulatedContent = ''\n    let inputTokens = 0\n    let outputTokens = 0\n\n    // Build the system prompt based on format\n    const formatInstructions = this.getFormatInstructions(options.style)\n    const lengthInstructions = options.maxLength\n      ? ` Keep the summary under ${options.maxLength} words.`\n      : ''\n\n    const systemPrompt = `You are a helpful assistant that summarizes text. ${formatInstructions}${lengthInstructions}`\n\n    const result = await this.client.models.generateContentStream({\n      model,\n      contents: [\n        {\n          role: 'user',\n          parts: [\n            { text: `Please summarize the following:\\n\\n${options.text}` },\n          ],\n        },\n      ],\n      config: {\n        systemInstruction: systemPrompt,\n      },\n    })\n\n    for await (const chunk of result) {\n      // Track usage metadata\n      if (chunk.usageMetadata) {\n        inputTokens = chunk.usageMetadata.promptTokenCount ?? inputTokens\n        outputTokens = chunk.usageMetadata.candidatesTokenCount ?? outputTokens\n      }\n\n      if (chunk.candidates?.[0]?.content?.parts) {\n        for (const part of chunk.candidates[0].content.parts) {\n          if (part.text) {\n            accumulatedContent += part.text\n            yield {\n              type: 'TEXT_MESSAGE_CONTENT',\n              messageId: id,\n              model,\n              timestamp: Date.now(),\n              delta: part.text,\n              content: accumulatedContent,\n            }\n          }\n        }\n      }\n\n      // Check for finish reason\n      const finishReason = chunk.candidates?.[0]?.finishReason\n      if (\n        finishReason === FinishReason.STOP ||\n        finishReason === FinishReason.MAX_TOKENS ||\n        finishReason === FinishReason.SAFETY\n      ) {\n        yield {\n          type: 'RUN_FINISHED',\n          runId: id,\n          model,\n          timestamp: Date.now(),\n          finishReason:\n            finishReason === FinishReason.STOP\n              ? 'stop'\n              : finishReason === FinishReason.MAX_TOKENS\n                ? 'length'\n                : 'content_filter',\n          usage: {\n            promptTokens: inputTokens,\n            completionTokens: outputTokens,\n            totalTokens: inputTokens + outputTokens,\n          },\n        }\n      }\n    }\n  }\n\n  private getFormatInstructions(\n    style?: 'paragraph' | 'bullet-points' | 'concise',\n  ): string {\n    switch (style) {\n      case 'bullet-points':\n        return 'Provide the summary as bullet points.'\n      case 'concise':\n        return 'Provide a very brief one or two sentence summary.'\n      case 'paragraph':\n      default:\n        return 'Provide the summary in paragraph form.'\n    }\n  }\n}\n\n/**\n * Creates a Gemini summarize adapter with explicit API key and model\n */\nexport function createGeminiSummarize<TModel extends GeminiSummarizeModel>(\n  apiKey: string,\n  model: TModel,\n  config?: Omit<GeminiSummarizeConfig, 'apiKey'>,\n): GeminiSummarizeAdapter<TModel> {\n  return new GeminiSummarizeAdapter({ ...config, apiKey }, model)\n}\n\n/**\n * Creates a Gemini summarize adapter with API key from environment and required model\n */\nexport function geminiSummarize<TModel extends GeminiSummarizeModel>(\n  model: TModel,\n  config?: Omit<GeminiSummarizeConfig, 'apiKey'>,\n): GeminiSummarizeAdapter<TModel> {\n  const apiKey = getGeminiApiKeyFromEnv()\n  return new GeminiSummarizeAdapter({ ...config, apiKey }, model)\n}\n"],"names":[],"mappings":";;AAsBO,MAAM,wBAAwB;AAAA,EACnC;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACF;AA+BO,MAAM,uBAEyD;AAAA,EAYpE,YAAY,QAA+B,OAAe;AAX1D,SAAS,OAAO;AAChB,SAAS,OAAO;AAWd,SAAK,SAAS,mBAAmB,MAAM;AACvC,SAAK,QAAQ;AAAA,EACf;AAAA,EAEA,MAAM,UAAU,SAA6D;AAC3E,UAAM,QAAQ,QAAQ;AAGtB,UAAM,qBAAqB,KAAK,sBAAsB,QAAQ,KAAK;AACnE,UAAM,qBAAqB,QAAQ,YAC/B,2BAA2B,QAAQ,SAAS,aAC5C;AAEJ,UAAM,eAAe,qDAAqD,kBAAkB,GAAG,kBAAkB;AAEjH,UAAM,WAAW,MAAM,KAAK,OAAO,OAAO,gBAAgB;AAAA,MACxD;AAAA,MACA,UAAU;AAAA,QACR;AAAA,UACE,MAAM;AAAA,UACN,OAAO;AAAA,YACL,EAAE,MAAM;AAAA;AAAA,EAAsC,QAAQ,IAAI,GAAA;AAAA,UAAG;AAAA,QAC/D;AAAA,MACF;AAAA,MAEF,QAAQ;AAAA,QACN,mBAAmB;AAAA,MAAA;AAAA,IACrB,CACD;AAED,UAAM,UAAU,SAAS,QAAQ;AACjC,UAAM,cAAc,SAAS,eAAe,oBAAoB;AAChE,UAAM,eAAe,SAAS,eAAe,wBAAwB;AAErE,WAAO;AAAA,MACL,IAAI,WAAW,KAAK;AAAA,MACpB;AAAA,MACA;AAAA,MACA,OAAO;AAAA,QACL,cAAc;AAAA,QACd,kBAAkB;AAAA,QAClB,aAAa,cAAc;AAAA,MAAA;AAAA,IAC7B;AAAA,EAEJ;AAAA,EAEA,OAAO,gBACL,SAC4B;AAC5B,UAAM,QAAQ,QAAQ;AACtB,UAAM,KAAK,WAAW,KAAK;AAC3B,QAAI,qBAAqB;AACzB,QAAI,cAAc;AAClB,QAAI,eAAe;AAGnB,UAAM,qBAAqB,KAAK,sBAAsB,QAAQ,KAAK;AACnE,UAAM,qBAAqB,QAAQ,YAC/B,2BAA2B,QAAQ,SAAS,YAC5C;AAEJ,UAAM,eAAe,qDAAqD,kBAAkB,GAAG,kBAAkB;AAEjH,UAAM,SAAS,MAAM,KAAK,OAAO,OAAO,sBAAsB;AAAA,MAC5D;AAAA,MACA,UAAU;AAAA,QACR;AAAA,UACE,MAAM;AAAA,UACN,OAAO;AAAA,YACL,EAAE,MAAM;AAAA;AAAA,EAAsC,QAAQ,IAAI,GAAA;AAAA,UAAG;AAAA,QAC/D;AAAA,MACF;AAAA,MAEF,QAAQ;AAAA,QACN,mBAAmB;AAAA,MAAA;AAAA,IACrB,CACD;AAED,qBAAiB,SAAS,QAAQ;AAEhC,UAAI,MAAM,eAAe;AACvB,sBAAc,MAAM,cAAc,oBAAoB;AACtD,uBAAe,MAAM,cAAc,wBAAwB;AAAA,MAC7D;AAEA,UAAI,MAAM,aAAa,CAAC,GAAG,SAAS,OAAO;AACzC,mBAAW,QAAQ,MAAM,WAAW,CAAC,EAAE,QAAQ,OAAO;AACpD,cAAI,KAAK,MAAM;AACb,kCAAsB,KAAK;AAC3B,kBAAM;AAAA,cACJ,MAAM;AAAA,cACN,WAAW;AAAA,cACX;AAAA,cACA,WAAW,KAAK,IAAA;AAAA,cAChB,OAAO,KAAK;AAAA,cACZ,SAAS;AAAA,YAAA;AAAA,UAEb;AAAA,QACF;AAAA,MACF;AAGA,YAAM,eAAe,MAAM,aAAa,CAAC,GAAG;AAC5C,UACE,iBAAiB,aAAa,QAC9B,iBAAiB,aAAa,cAC9B,iBAAiB,aAAa,QAC9B;AACA,cAAM;AAAA,UACJ,MAAM;AAAA,UACN,OAAO;AAAA,UACP;AAAA,UACA,WAAW,KAAK,IAAA;AAAA,UAChB,cACE,iBAAiB,aAAa,OAC1B,SACA,iBAAiB,aAAa,aAC5B,WACA;AAAA,UACR,OAAO;AAAA,YACL,cAAc;AAAA,YACd,kBAAkB;AAAA,YAClB,aAAa,cAAc;AAAA,UAAA;AAAA,QAC7B;AAAA,MAEJ;AAAA,IACF;AAAA,EACF;AAAA,EAEQ,sBACN,OACQ;AACR,YAAQ,OAAA;AAAA,MACN,KAAK;AACH,eAAO;AAAA,MACT,KAAK;AACH,eAAO;AAAA,MACT,KAAK;AAAA,MACL;AACE,eAAO;AAAA,IAAA;AAAA,EAEb;AACF;AAKO,SAAS,sBACd,QACA,OACA,QACgC;AAChC,SAAO,IAAI,uBAAuB,EAAE,GAAG,QAAQ,OAAA,GAAU,KAAK;AAChE;AAKO,SAAS,gBACd,OACA,QACgC;AAChC,QAAM,SAAS,uBAAA;AACf,SAAO,IAAI,uBAAuB,EAAE,GAAG,QAAQ,OAAA,GAAU,KAAK;AAChE;"}
+{"version":3,"file":"summarize.js","sources":["../../../src/adapters/summarize.ts"],"sourcesContent":["import { FinishReason } from '@google/genai'\nimport {\n  createGeminiClient,\n  generateId,\n  getGeminiApiKeyFromEnv,\n} from '../utils'\nimport type { GoogleGenAI } from '@google/genai'\nimport type { GeminiClientConfig } from '../utils'\nimport type { SummarizeAdapter } from '@tanstack/ai/adapters'\nimport type {\n  StreamChunk,\n  SummarizationOptions,\n  SummarizationResult,\n} from '@tanstack/ai'\n\n/**\n * Configuration for Gemini summarize adapter\n */\nexport interface GeminiSummarizeConfig extends GeminiClientConfig {}\n/**\n * Available Gemini models for summarization\n */\nexport const GeminiSummarizeModels = [\n  'gemini-3.1-flash-lite-preview',\n  'gemini-2.0-flash',\n  'gemini-1.5-flash',\n  'gemini-1.5-pro',\n  'gemini-2.0-flash-lite',\n] as const\n\nexport type GeminiSummarizeModel = (typeof GeminiSummarizeModels)[number]\n\n/**\n * Provider-specific options for Gemini summarization\n */\nexport interface GeminiSummarizeProviderOptions {\n  /** Generation configuration */\n  generationConfig?: {\n    temperature?: number\n    topP?: number\n    topK?: number\n    maxOutputTokens?: number\n    stopSequences?: Array<string>\n  }\n  /** Safety settings */\n  safetySettings?: Array<{\n    category: string\n    threshold: string\n  }>\n}\n\nexport interface GeminiSummarizeAdapterOptions {\n  // Additional adapter options can be added here\n}\n\n/**\n * Gemini Summarize Adapter\n * A tree-shakeable summarization adapter for Google Gemini\n */\nexport class GeminiSummarizeAdapter<\n  TModel extends GeminiSummarizeModel,\n> implements SummarizeAdapter<TModel, GeminiSummarizeProviderOptions> {\n  readonly kind = 'summarize' as const\n  readonly name = 'gemini' as const\n  readonly model: TModel\n\n  // Type-only property - never assigned at runtime\n  declare '~types': {\n    providerOptions: GeminiSummarizeProviderOptions\n  }\n\n  private client: GoogleGenAI\n\n  constructor(config: GeminiSummarizeConfig, model: TModel) {\n    this.client = createGeminiClient(config)\n    this.model = model\n  }\n\n  async summarize(options: SummarizationOptions): Promise<SummarizationResult> {\n    const model = options.model\n\n    // Build the system prompt based on format\n    const formatInstructions = this.getFormatInstructions(options.style)\n    const lengthInstructions = options.maxLength\n      ? ` Keep the summary under ${options.maxLength} tokens.`\n      : ''\n\n    const systemPrompt = `You are a helpful assistant that summarizes text. ${formatInstructions}${lengthInstructions}`\n\n    const response = await this.client.models.generateContent({\n      model,\n      contents: [\n        {\n          role: 'user',\n          parts: [\n            { text: `Please summarize the following:\\n\\n${options.text}` },\n          ],\n        },\n      ],\n      config: {\n        systemInstruction: systemPrompt,\n      },\n    })\n\n    const summary = response.text ?? ''\n    const inputTokens = response.usageMetadata?.promptTokenCount ?? 0\n    const outputTokens = response.usageMetadata?.candidatesTokenCount ?? 0\n\n    return {\n      id: generateId('sum'),\n      model,\n      summary,\n      usage: {\n        promptTokens: inputTokens,\n        completionTokens: outputTokens,\n        totalTokens: inputTokens + outputTokens,\n      },\n    }\n  }\n\n  async *summarizeStream(\n    options: SummarizationOptions,\n  ): AsyncIterable<StreamChunk> {\n    const model = options.model\n    const id = generateId('sum')\n    let accumulatedContent = ''\n    let inputTokens = 0\n    let outputTokens = 0\n\n    // Build the system prompt based on format\n    const formatInstructions = this.getFormatInstructions(options.style)\n    const lengthInstructions = options.maxLength\n      ? ` Keep the summary under ${options.maxLength} words.`\n      : ''\n\n    const systemPrompt = `You are a helpful assistant that summarizes text. ${formatInstructions}${lengthInstructions}`\n\n    const result = await this.client.models.generateContentStream({\n      model,\n      contents: [\n        {\n          role: 'user',\n          parts: [\n            { text: `Please summarize the following:\\n\\n${options.text}` },\n          ],\n        },\n      ],\n      config: {\n        systemInstruction: systemPrompt,\n      },\n    })\n\n    for await (const chunk of result) {\n      // Track usage metadata\n      if (chunk.usageMetadata) {\n        inputTokens = chunk.usageMetadata.promptTokenCount ?? inputTokens\n        outputTokens = chunk.usageMetadata.candidatesTokenCount ?? outputTokens\n      }\n\n      if (chunk.candidates?.[0]?.content?.parts) {\n        for (const part of chunk.candidates[0].content.parts) {\n          if (part.text) {\n            accumulatedContent += part.text\n            yield {\n              type: 'TEXT_MESSAGE_CONTENT',\n              messageId: id,\n              model,\n              timestamp: Date.now(),\n              delta: part.text,\n              content: accumulatedContent,\n            }\n          }\n        }\n      }\n\n      // Check for finish reason\n      const finishReason = chunk.candidates?.[0]?.finishReason\n      if (\n        finishReason === FinishReason.STOP ||\n        finishReason === FinishReason.MAX_TOKENS ||\n        finishReason === FinishReason.SAFETY\n      ) {\n        yield {\n          type: 'RUN_FINISHED',\n          runId: id,\n          model,\n          timestamp: Date.now(),\n          finishReason:\n            finishReason === FinishReason.STOP\n              ? 'stop'\n              : finishReason === FinishReason.MAX_TOKENS\n                ? 'length'\n                : 'content_filter',\n          usage: {\n            promptTokens: inputTokens,\n            completionTokens: outputTokens,\n            totalTokens: inputTokens + outputTokens,\n          },\n        }\n      }\n    }\n  }\n\n  private getFormatInstructions(\n    style?: 'paragraph' | 'bullet-points' | 'concise',\n  ): string {\n    switch (style) {\n      case 'bullet-points':\n        return 'Provide the summary as bullet points.'\n      case 'concise':\n        return 'Provide a very brief one or two sentence summary.'\n      case 'paragraph':\n      default:\n        return 'Provide the summary in paragraph form.'\n    }\n  }\n}\n\n/**\n * Creates a Gemini summarize adapter with explicit API key and model\n */\nexport function createGeminiSummarize<TModel extends GeminiSummarizeModel>(\n  apiKey: string,\n  model: TModel,\n  config?: Omit<GeminiSummarizeConfig, 'apiKey'>,\n): GeminiSummarizeAdapter<TModel> {\n  return new GeminiSummarizeAdapter({ ...config, apiKey }, model)\n}\n\n/**\n * Creates a Gemini summarize adapter with API key from environment and required model\n */\nexport function geminiSummarize<TModel extends GeminiSummarizeModel>(\n  model: TModel,\n  config?: Omit<GeminiSummarizeConfig, 'apiKey'>,\n): GeminiSummarizeAdapter<TModel> {\n  const apiKey = getGeminiApiKeyFromEnv()\n  return new GeminiSummarizeAdapter({ ...config, apiKey }, model)\n}\n"],"names":[],"mappings":";;AAsBO,MAAM,wBAAwB;AAAA,EACnC;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACF;AA+BO,MAAM,uBAEyD;AAAA,EAYpE,YAAY,QAA+B,OAAe;AAX1D,SAAS,OAAO;AAChB,SAAS,OAAO;AAWd,SAAK,SAAS,mBAAmB,MAAM;AACvC,SAAK,QAAQ;AAAA,EACf;AAAA,EAEA,MAAM,UAAU,SAA6D;AAC3E,UAAM,QAAQ,QAAQ;AAGtB,UAAM,qBAAqB,KAAK,sBAAsB,QAAQ,KAAK;AACnE,UAAM,qBAAqB,QAAQ,YAC/B,2BAA2B,QAAQ,SAAS,aAC5C;AAEJ,UAAM,eAAe,qDAAqD,kBAAkB,GAAG,kBAAkB;AAEjH,UAAM,WAAW,MAAM,KAAK,OAAO,OAAO,gBAAgB;AAAA,MACxD;AAAA,MACA,UAAU;AAAA,QACR;AAAA,UACE,MAAM;AAAA,UACN,OAAO;AAAA,YACL,EAAE,MAAM;AAAA;AAAA,EAAsC,QAAQ,IAAI,GAAA;AAAA,UAAG;AAAA,QAC/D;AAAA,MACF;AAAA,MAEF,QAAQ;AAAA,QACN,mBAAmB;AAAA,MAAA;AAAA,IACrB,CACD;AAED,UAAM,UAAU,SAAS,QAAQ;AACjC,UAAM,cAAc,SAAS,eAAe,oBAAoB;AAChE,UAAM,eAAe,SAAS,eAAe,wBAAwB;AAErE,WAAO;AAAA,MACL,IAAI,WAAW,KAAK;AAAA,MACpB;AAAA,MACA;AAAA,MACA,OAAO;AAAA,QACL,cAAc;AAAA,QACd,kBAAkB;AAAA,QAClB,aAAa,cAAc;AAAA,MAAA;AAAA,IAC7B;AAAA,EAEJ;AAAA,EAEA,OAAO,gBACL,SAC4B;AAC5B,UAAM,QAAQ,QAAQ;AACtB,UAAM,KAAK,WAAW,KAAK;AAC3B,QAAI,qBAAqB;AACzB,QAAI,cAAc;AAClB,QAAI,eAAe;AAGnB,UAAM,qBAAqB,KAAK,sBAAsB,QAAQ,KAAK;AACnE,UAAM,qBAAqB,QAAQ,YAC/B,2BAA2B,QAAQ,SAAS,YAC5C;AAEJ,UAAM,eAAe,qDAAqD,kBAAkB,GAAG,kBAAkB;AAEjH,UAAM,SAAS,MAAM,KAAK,OAAO,OAAO,sBAAsB;AAAA,MAC5D;AAAA,MACA,UAAU;AAAA,QACR;AAAA,UACE,MAAM;AAAA,UACN,OAAO;AAAA,YACL,EAAE,MAAM;AAAA;AAAA,EAAsC,QAAQ,IAAI,GAAA;AAAA,UAAG;AAAA,QAC/D;AAAA,MACF;AAAA,MAEF,QAAQ;AAAA,QACN,mBAAmB;AAAA,MAAA;AAAA,IACrB,CACD;AAED,qBAAiB,SAAS,QAAQ;AAEhC,UAAI,MAAM,eAAe;AACvB,sBAAc,MAAM,cAAc,oBAAoB;AACtD,uBAAe,MAAM,cAAc,wBAAwB;AAAA,MAC7D;AAEA,UAAI,MAAM,aAAa,CAAC,GAAG,SAAS,OAAO;AACzC,mBAAW,QAAQ,MAAM,WAAW,CAAC,EAAE,QAAQ,OAAO;AACpD,cAAI,KAAK,MAAM;AACb,kCAAsB,KAAK;AAC3B,kBAAM;AAAA,cACJ,MAAM;AAAA,cACN,WAAW;AAAA,cACX;AAAA,cACA,WAAW,KAAK,IAAA;AAAA,cAChB,OAAO,KAAK;AAAA,cACZ,SAAS;AAAA,YAAA;AAAA,UAEb;AAAA,QACF;AAAA,MACF;AAGA,YAAM,eAAe,MAAM,aAAa,CAAC,GAAG;AAC5C,UACE,iBAAiB,aAAa,QAC9B,iBAAiB,aAAa,cAC9B,iBAAiB,aAAa,QAC9B;AACA,cAAM;AAAA,UACJ,MAAM;AAAA,UACN,OAAO;AAAA,UACP;AAAA,UACA,WAAW,KAAK,IAAA;AAAA,UAChB,cACE,iBAAiB,aAAa,OAC1B,SACA,iBAAiB,aAAa,aAC5B,WACA;AAAA,UACR,OAAO;AAAA,YACL,cAAc;AAAA,YACd,kBAAkB;AAAA,YAClB,aAAa,cAAc;AAAA,UAAA;AAAA,QAC7B;AAAA,MAEJ;AAAA,IACF;AAAA,EACF;AAAA,EAEQ,sBACN,OACQ;AACR,YAAQ,OAAA;AAAA,MACN,KAAK;AACH,eAAO;AAAA,MACT,KAAK;AACH,eAAO;AAAA,MACT,KAAK;AAAA,MACL;AACE,eAAO;AAAA,IAAA;AAAA,EAEb;AACF;AAKO,SAAS,sBACd,QACA,OACA,QACgC;AAChC,SAAO,IAAI,uBAAuB,EAAE,GAAG,QAAQ,OAAA,GAAU,KAAK;AAChE;AAKO,SAAS,gBACd,OACA,QACgC;AAChC,QAAM,SAAS,uBAAA;AACf,SAAO,IAAI,uBAAuB,EAAE,GAAG,QAAQ,OAAA,GAAU,KAAK;AAChE;"}

package/dist/esm/image/image-provider-options.d.ts

@@ -103,11 +103,32 @@ export type GeminiImageModelProviderOptionsByName = {
  */
 export type GeminiImageSize = '1024x1024' | '512x512' | '1024x768' | '1536x1024' | '1792x1024' | '1920x1080' | '768x1024' | '1024x1536' | '1024x1792' | '1080x1920';
 /**
- * Model-specific size options mapping
- * All Imagen models use the same size options
+ * Aspect ratios supported by Gemini native image models (via generateContent API).
+ * Matches the SDK's ImageConfig.aspectRatio values.
+ */
+export type GeminiNativeImageAspectRatio = '1:1' | '2:3' | '3:2' | '3:4' | '4:3' | '9:16' | '16:9' | '21:9';
+/**
+ * Resolution tiers for Gemini native image models.
+ * Matches the SDK's ImageConfig.imageSize values.
+ */
+export type GeminiNativeImageResolution = '1K' | '2K' | '4K';
+/**
+ * Template literal size type for Gemini native image models: "16:9_4K", "1:1_2K", etc.
+ */
+export type GeminiNativeImageSize = `${GeminiNativeImageAspectRatio}_${GeminiNativeImageResolution}`;
+/**
+ * Gemini native image models that use the generateContent API path.
+ * These models support template literal sizes (aspectRatio_resolution).
+ */
+export type GeminiNativeImageModels = 'gemini-3.1-flash-image-preview' | 'gemini-3-pro-image-preview' | 'gemini-2.5-flash-image' | 'gemini-2.0-flash-preview-image-generation';
+/**
+ * Model-specific size options mapping.
+ * Gemini native image models use template literal sizes, Imagen models use pixel sizes.
  */
 export type GeminiImageModelSizeByName = {
-    [K in GeminiImageModels]: GeminiImageSize;
+    [K in GeminiNativeImageModels]: GeminiNativeImageSize;
+} & {
+    [K in Exclude<GeminiImageModels, GeminiNativeImageModels>]: GeminiImageSize;
 };
 /**
  * Valid sizes for Gemini Imagen models
@@ -137,3 +158,11 @@ export declare function validatePrompt(options: {
     prompt: string;
     model: string;
 }): void;
+/**
+ * Parses a Gemini native image size string into its components.
+ * Format: "aspectRatio_resolution" e.g. "16:9_4K" → { aspectRatio: "16:9", resolution: "4K" }
+ */
+export declare function parseNativeImageSize(size: string): {
+    aspectRatio: string;
+    resolution: string;
+} | undefined;

package/dist/esm/image/image-provider-options.js

@@ -43,8 +43,14 @@ function validatePrompt(options) {
     throw new Error(`Prompt cannot be empty for model "${model}".`);
   }
 }
+function parseNativeImageSize(size) {
+  const match = size.match(/^(\d+:\d+)_(.+)$/);
+  if (!match) return void 0;
+  return { aspectRatio: match[1], resolution: match[2] };
+}
 export {
   GEMINI_SIZE_TO_ASPECT_RATIO,
+  parseNativeImageSize,
   sizeToAspectRatio,
   validateImageSize,
   validateNumberOfImages,

package/dist/esm/image/image-provider-options.js.map

@@ -1 +1 @@
-{"version":3,"file":"image-provider-options.js","sources":["../../../src/image/image-provider-options.ts"],"sourcesContent":["import type { GeminiImageModels } from '../model-meta'\nimport type {\n  ImagePromptLanguage,\n  PersonGeneration,\n  SafetyFilterLevel,\n} from '@google/genai'\n\n// Re-export SDK types so users can use them directly\nexport type { ImagePromptLanguage, PersonGeneration, SafetyFilterLevel }\n\n/**\n * Gemini Imagen aspect ratio options\n * Controls the aspect ratio of generated images\n */\nexport type GeminiAspectRatio =\n  | '1:1'\n  | '3:4'\n  | '4:3'\n  | '9:16'\n  | '16:9'\n  | '9:21'\n  | '21:9'\n\n/**\n * Provider options for Gemini image generation\n * These options match the @google/genai GenerateImagesConfig interface\n * and can be spread directly into the API request.\n */\nexport interface GeminiImageProviderOptions {\n  /**\n   * The aspect ratio of generated images\n   * @default '1:1'\n   */\n  aspectRatio?: GeminiAspectRatio\n\n  /**\n   * Controls whether people can appear in generated images\n   * Use PersonGeneration enum values: DONT_ALLOW, ALLOW_ADULT, ALLOW_ALL\n   * @default 'ALLOW_ADULT'\n   */\n  personGeneration?: PersonGeneration\n\n  /**\n   * Safety filter level for content filtering\n   * Use SafetyFilterLevel enum values\n   */\n  safetyFilterLevel?: SafetyFilterLevel\n\n  /**\n   * Optional seed for reproducible image generation\n   * When the same seed is used with the same prompt and settings,\n   * you should get similar (though not identical) results\n   */\n  seed?: number\n\n  /**\n   * Whether to add a SynthID watermark to generated images\n   * SynthID helps identify AI-generated content\n   * @default true\n   */\n  addWatermark?: boolean\n\n  /**\n   * Language of the prompt\n   * Use ImagePromptLanguage enum values\n   */\n  language?: ImagePromptLanguage\n\n  /**\n   * Negative prompt - what to avoid in the generated image\n   * Not all models support negative prompts\n   */\n  negativePrompt?: string\n\n  /**\n   * Output MIME type for the generated image\n   * @default 'image/png'\n   */\n  outputMimeType?: 'image/png' | 'image/jpeg' | 'image/webp'\n\n  /**\n   * Compression quality for JPEG outputs (0-100)\n   * Higher values mean better quality but larger file sizes\n   * @default 75\n   */\n  outputCompressionQuality?: number\n\n  /**\n   * Controls how much the model adheres to the text prompt\n   * Large values increase output and prompt alignment,\n   * but may compromise image quality\n   */\n  guidanceScale?: number\n\n  /**\n   * Whether to use the prompt rewriting logic\n   */\n  enhancePrompt?: boolean\n\n  /**\n   * Whether to report the safety scores of each generated image\n   * and the positive prompt in the response\n   */\n  includeSafetyAttributes?: boolean\n\n  /**\n   * Whether to include the Responsible AI filter reason\n   * if the image is filtered out of the response\n   */\n  includeRaiReason?: boolean\n\n  /**\n   * Cloud Storage URI used to store the generated images\n   */\n  outputGcsUri?: string\n\n  /**\n   * User specified labels to track billing usage\n   */\n  labels?: Record<string, string>\n}\n\n/**\n * Model-specific provider options mapping\n * Currently all Imagen models use the same options structure\n */\nexport type GeminiImageModelProviderOptionsByName = {\n  [K in GeminiImageModels]: GeminiImageProviderOptions\n}\n\n/**\n * Supported size strings for Gemini Imagen models\n * These map to aspect ratios internally\n */\nexport type GeminiImageSize =\n  | '1024x1024'\n  | '512x512'\n  | '1024x768'\n  | '1536x1024'\n  | '1792x1024'\n  | '1920x1080'\n  | '768x1024'\n  | '1024x1536'\n  | '1024x1792'\n  | '1080x1920'\n\n/**\n * Model-specific size options mapping\n * All Imagen models use the same size options\n */\nexport type GeminiImageModelSizeByName = {\n  [K in GeminiImageModels]: GeminiImageSize\n}\n\n/**\n * Valid sizes for Gemini Imagen models\n * Gemini uses aspect ratios, but we map common WIDTHxHEIGHT formats to aspect ratios\n * These are approximate mappings based on common image dimensions\n */\nexport const GEMINI_SIZE_TO_ASPECT_RATIO: Record<string, GeminiAspectRatio> = {\n  // Square\n  '1024x1024': '1:1',\n  '512x512': '1:1',\n  // Landscape\n  '1024x768': '4:3',\n  '1536x1024': '4:3',\n  '1792x1024': '16:9',\n  '1920x1080': '16:9',\n  // Portrait\n  '768x1024': '3:4',\n  '1024x1536': '3:4', // Inverted\n  '1024x1792': '9:16',\n  '1080x1920': '9:16',\n}\n\n/**\n * Maps a WIDTHxHEIGHT size string to a Gemini aspect ratio\n * Returns undefined if the size cannot be mapped\n */\nexport function sizeToAspectRatio(\n  size: string | undefined,\n): GeminiAspectRatio | undefined {\n  if (!size) return undefined\n  return GEMINI_SIZE_TO_ASPECT_RATIO[size]\n}\n\n/**\n * Validates that the provided size can be mapped to an aspect ratio\n * Throws an error if the size is invalid\n */\nexport function validateImageSize(\n  model: string,\n  size: string | undefined,\n): void {\n  if (!size) return\n\n  const aspectRatio = sizeToAspectRatio(size)\n  if (!aspectRatio) {\n    const validSizes = Object.keys(GEMINI_SIZE_TO_ASPECT_RATIO)\n    throw new Error(\n      `Invalid size \"${size}\" for model \"${model}\". ` +\n        `Gemini Imagen uses aspect ratios. Valid sizes that map to aspect ratios: ${validSizes.join(', ')}. ` +\n        `Alternatively, use providerOptions.aspectRatio directly with values: 1:1, 3:4, 4:3, 9:16, 16:9, 9:21, 21:9`,\n    )\n  }\n}\n\n/**\n * Validates the number of images requested\n * Imagen models support 1-8 images per request (varies by model)\n */\nexport function validateNumberOfImages(\n  model: string,\n  numberOfImages: number | undefined,\n): void {\n  if (numberOfImages === undefined) return\n\n  // Most Imagen models support 1-4 images, some support up to 8\n  const maxImages = 4\n  if (numberOfImages < 1 || numberOfImages > maxImages) {\n    throw new Error(\n      `Invalid numberOfImages \"${numberOfImages}\" for model \"${model}\". ` +\n        `Must be between 1 and ${maxImages}.`,\n    )\n  }\n}\n\n/**\n * Validates the prompt is not empty\n */\nexport function validatePrompt(options: {\n  prompt: string\n  model: string\n}): void {\n  const { prompt, model } = options\n  if (!prompt || prompt.trim().length === 0) {\n    throw new Error(`Prompt cannot be empty for model \"${model}\".`)\n  }\n}\n"],"names":[],"mappings":"AA+JO,MAAM,8BAAiE;AAAA;AAAA,EAE5E,aAAa;AAAA,EACb,WAAW;AAAA;AAAA,EAEX,YAAY;AAAA,EACZ,aAAa;AAAA,EACb,aAAa;AAAA,EACb,aAAa;AAAA;AAAA,EAEb,YAAY;AAAA,EACZ,aAAa;AAAA;AAAA,EACb,aAAa;AAAA,EACb,aAAa;AACf;AAMO,SAAS,kBACd,MAC+B;AAC/B,MAAI,CAAC,KAAM,QAAO;AAClB,SAAO,4BAA4B,IAAI;AACzC;AAMO,SAAS,kBACd,OACA,MACM;AACN,MAAI,CAAC,KAAM;AAEX,QAAM,cAAc,kBAAkB,IAAI;AAC1C,MAAI,CAAC,aAAa;AAChB,UAAM,aAAa,OAAO,KAAK,2BAA2B;AAC1D,UAAM,IAAI;AAAA,MACR,iBAAiB,IAAI,gBAAgB,KAAK,+EACoC,WAAW,KAAK,IAAI,CAAC;AAAA,IAAA;AAAA,EAGvG;AACF;AAMO,SAAS,uBACd,OACA,gBACM;AACN,MAAI,mBAAmB,OAAW;AAGlC,QAAM,YAAY;AAClB,MAAI,iBAAiB,KAAK,iBAAiB,WAAW;AACpD,UAAM,IAAI;AAAA,MACR,2BAA2B,cAAc,gBAAgB,KAAK,4BACnC,SAAS;AAAA,IAAA;AAAA,EAExC;AACF;AAKO,SAAS,eAAe,SAGtB;AACP,QAAM,EAAE,QAAQ,MAAA,IAAU;AAC1B,MAAI,CAAC,UAAU,OAAO,KAAA,EAAO,WAAW,GAAG;AACzC,UAAM,IAAI,MAAM,qCAAqC,KAAK,IAAI;AAAA,EAChE;AACF;"}
+{"version":3,"file":"image-provider-options.js","sources":["../../../src/image/image-provider-options.ts"],"sourcesContent":["import type { GeminiImageModels } from '../model-meta'\nimport type {\n  ImagePromptLanguage,\n  PersonGeneration,\n  SafetyFilterLevel,\n} from '@google/genai'\n\n// Re-export SDK types so users can use them directly\nexport type { ImagePromptLanguage, PersonGeneration, SafetyFilterLevel }\n\n/**\n * Gemini Imagen aspect ratio options\n * Controls the aspect ratio of generated images\n */\nexport type GeminiAspectRatio =\n  | '1:1'\n  | '3:4'\n  | '4:3'\n  | '9:16'\n  | '16:9'\n  | '9:21'\n  | '21:9'\n\n/**\n * Provider options for Gemini image generation\n * These options match the @google/genai GenerateImagesConfig interface\n * and can be spread directly into the API request.\n */\nexport interface GeminiImageProviderOptions {\n  /**\n   * The aspect ratio of generated images\n   * @default '1:1'\n   */\n  aspectRatio?: GeminiAspectRatio\n\n  /**\n   * Controls whether people can appear in generated images\n   * Use PersonGeneration enum values: DONT_ALLOW, ALLOW_ADULT, ALLOW_ALL\n   * @default 'ALLOW_ADULT'\n   */\n  personGeneration?: PersonGeneration\n\n  /**\n   * Safety filter level for content filtering\n   * Use SafetyFilterLevel enum values\n   */\n  safetyFilterLevel?: SafetyFilterLevel\n\n  /**\n   * Optional seed for reproducible image generation\n   * When the same seed is used with the same prompt and settings,\n   * you should get similar (though not identical) results\n   */\n  seed?: number\n\n  /**\n   * Whether to add a SynthID watermark to generated images\n   * SynthID helps identify AI-generated content\n   * @default true\n   */\n  addWatermark?: boolean\n\n  /**\n   * Language of the prompt\n   * Use ImagePromptLanguage enum values\n   */\n  language?: ImagePromptLanguage\n\n  /**\n   * Negative prompt - what to avoid in the generated image\n   * Not all models support negative prompts\n   */\n  negativePrompt?: string\n\n  /**\n   * Output MIME type for the generated image\n   * @default 'image/png'\n   */\n  outputMimeType?: 'image/png' | 'image/jpeg' | 'image/webp'\n\n  /**\n   * Compression quality for JPEG outputs (0-100)\n   * Higher values mean better quality but larger file sizes\n   * @default 75\n   */\n  outputCompressionQuality?: number\n\n  /**\n   * Controls how much the model adheres to the text prompt\n   * Large values increase output and prompt alignment,\n   * but may compromise image quality\n   */\n  guidanceScale?: number\n\n  /**\n   * Whether to use the prompt rewriting logic\n   */\n  enhancePrompt?: boolean\n\n  /**\n   * Whether to report the safety scores of each generated image\n   * and the positive prompt in the response\n   */\n  includeSafetyAttributes?: boolean\n\n  /**\n   * Whether to include the Responsible AI filter reason\n   * if the image is filtered out of the response\n   */\n  includeRaiReason?: boolean\n\n  /**\n   * Cloud Storage URI used to store the generated images\n   */\n  outputGcsUri?: string\n\n  /**\n   * User specified labels to track billing usage\n   */\n  labels?: Record<string, string>\n}\n\n/**\n * Model-specific provider options mapping\n * Currently all Imagen models use the same options structure\n */\nexport type GeminiImageModelProviderOptionsByName = {\n  [K in GeminiImageModels]: GeminiImageProviderOptions\n}\n\n/**\n * Supported size strings for Gemini Imagen models\n * These map to aspect ratios internally\n */\nexport type GeminiImageSize =\n  | '1024x1024'\n  | '512x512'\n  | '1024x768'\n  | '1536x1024'\n  | '1792x1024'\n  | '1920x1080'\n  | '768x1024'\n  | '1024x1536'\n  | '1024x1792'\n  | '1080x1920'\n\n/**\n * Aspect ratios supported by Gemini native image models (via generateContent API).\n * Matches the SDK's ImageConfig.aspectRatio values.\n */\nexport type GeminiNativeImageAspectRatio =\n  | '1:1'\n  | '2:3'\n  | '3:2'\n  | '3:4'\n  | '4:3'\n  | '9:16'\n  | '16:9'\n  | '21:9'\n\n/**\n * Resolution tiers for Gemini native image models.\n * Matches the SDK's ImageConfig.imageSize values.\n */\nexport type GeminiNativeImageResolution = '1K' | '2K' | '4K'\n\n/**\n * Template literal size type for Gemini native image models: \"16:9_4K\", \"1:1_2K\", etc.\n */\nexport type GeminiNativeImageSize =\n  `${GeminiNativeImageAspectRatio}_${GeminiNativeImageResolution}`\n\n/**\n * Gemini native image models that use the generateContent API path.\n * These models support template literal sizes (aspectRatio_resolution).\n */\nexport type GeminiNativeImageModels =\n  | 'gemini-3.1-flash-image-preview'\n  | 'gemini-3-pro-image-preview'\n  | 'gemini-2.5-flash-image'\n  | 'gemini-2.0-flash-preview-image-generation'\n\n/**\n * Model-specific size options mapping.\n * Gemini native image models use template literal sizes, Imagen models use pixel sizes.\n */\nexport type GeminiImageModelSizeByName = {\n  [K in GeminiNativeImageModels]: GeminiNativeImageSize\n} & {\n  [K in Exclude<GeminiImageModels, GeminiNativeImageModels>]: GeminiImageSize\n}\n\n/**\n * Valid sizes for Gemini Imagen models\n * Gemini uses aspect ratios, but we map common WIDTHxHEIGHT formats to aspect ratios\n * These are approximate mappings based on common image dimensions\n */\nexport const GEMINI_SIZE_TO_ASPECT_RATIO: Record<string, GeminiAspectRatio> = {\n  // Square\n  '1024x1024': '1:1',\n  '512x512': '1:1',\n  // Landscape\n  '1024x768': '4:3',\n  '1536x1024': '4:3',\n  '1792x1024': '16:9',\n  '1920x1080': '16:9',\n  // Portrait\n  '768x1024': '3:4',\n  '1024x1536': '3:4', // Inverted\n  '1024x1792': '9:16',\n  '1080x1920': '9:16',\n}\n\n/**\n * Maps a WIDTHxHEIGHT size string to a Gemini aspect ratio\n * Returns undefined if the size cannot be mapped\n */\nexport function sizeToAspectRatio(\n  size: string | undefined,\n): GeminiAspectRatio | undefined {\n  if (!size) return undefined\n  return GEMINI_SIZE_TO_ASPECT_RATIO[size]\n}\n\n/**\n * Validates that the provided size can be mapped to an aspect ratio\n * Throws an error if the size is invalid\n */\nexport function validateImageSize(\n  model: string,\n  size: string | undefined,\n): void {\n  if (!size) return\n\n  const aspectRatio = sizeToAspectRatio(size)\n  if (!aspectRatio) {\n    const validSizes = Object.keys(GEMINI_SIZE_TO_ASPECT_RATIO)\n    throw new Error(\n      `Invalid size \"${size}\" for model \"${model}\". ` +\n        `Gemini Imagen uses aspect ratios. Valid sizes that map to aspect ratios: ${validSizes.join(', ')}. ` +\n        `Alternatively, use providerOptions.aspectRatio directly with values: 1:1, 3:4, 4:3, 9:16, 16:9, 9:21, 21:9`,\n    )\n  }\n}\n\n/**\n * Validates the number of images requested\n * Imagen models support 1-8 images per request (varies by model)\n */\nexport function validateNumberOfImages(\n  model: string,\n  numberOfImages: number | undefined,\n): void {\n  if (numberOfImages === undefined) return\n\n  // Most Imagen models support 1-4 images, some support up to 8\n  const maxImages = 4\n  if (numberOfImages < 1 || numberOfImages > maxImages) {\n    throw new Error(\n      `Invalid numberOfImages \"${numberOfImages}\" for model \"${model}\". ` +\n        `Must be between 1 and ${maxImages}.`,\n    )\n  }\n}\n\n/**\n * Validates the prompt is not empty\n */\nexport function validatePrompt(options: {\n  prompt: string\n  model: string\n}): void {\n  const { prompt, model } = options\n  if (!prompt || prompt.trim().length === 0) {\n    throw new Error(`Prompt cannot be empty for model \"${model}\".`)\n  }\n}\n\n/**\n * Parses a Gemini native image size string into its components.\n * Format: \"aspectRatio_resolution\" e.g. \"16:9_4K\" → { aspectRatio: \"16:9\", resolution: \"4K\" }\n */\nexport function parseNativeImageSize(\n  size: string,\n): { aspectRatio: string; resolution: string } | undefined {\n  const match = size.match(/^(\\d+:\\d+)_(.+)$/)\n  if (!match) return undefined\n  return { aspectRatio: match[1]!, resolution: match[2]! }\n}\n"],"names":[],"mappings":"AAqMO,MAAM,8BAAiE;AAAA;AAAA,EAE5E,aAAa;AAAA,EACb,WAAW;AAAA;AAAA,EAEX,YAAY;AAAA,EACZ,aAAa;AAAA,EACb,aAAa;AAAA,EACb,aAAa;AAAA;AAAA,EAEb,YAAY;AAAA,EACZ,aAAa;AAAA;AAAA,EACb,aAAa;AAAA,EACb,aAAa;AACf;AAMO,SAAS,kBACd,MAC+B;AAC/B,MAAI,CAAC,KAAM,QAAO;AAClB,SAAO,4BAA4B,IAAI;AACzC;AAMO,SAAS,kBACd,OACA,MACM;AACN,MAAI,CAAC,KAAM;AAEX,QAAM,cAAc,kBAAkB,IAAI;AAC1C,MAAI,CAAC,aAAa;AAChB,UAAM,aAAa,OAAO,KAAK,2BAA2B;AAC1D,UAAM,IAAI;AAAA,MACR,iBAAiB,IAAI,gBAAgB,KAAK,+EACoC,WAAW,KAAK,IAAI,CAAC;AAAA,IAAA;AAAA,EAGvG;AACF;AAMO,SAAS,uBACd,OACA,gBACM;AACN,MAAI,mBAAmB,OAAW;AAGlC,QAAM,YAAY;AAClB,MAAI,iBAAiB,KAAK,iBAAiB,WAAW;AACpD,UAAM,IAAI;AAAA,MACR,2BAA2B,cAAc,gBAAgB,KAAK,4BACnC,SAAS;AAAA,IAAA;AAAA,EAExC;AACF;AAKO,SAAS,eAAe,SAGtB;AACP,QAAM,EAAE,QAAQ,MAAA,IAAU;AAC1B,MAAI,CAAC,UAAU,OAAO,KAAA,EAAO,WAAW,GAAG;AACzC,UAAM,IAAI,MAAM,qCAAqC,KAAK,IAAI;AAAA,EAChE;AACF;AAMO,SAAS,qBACd,MACyD;AACzD,QAAM,QAAQ,KAAK,MAAM,kBAAkB;AAC3C,MAAI,CAAC,MAAO,QAAO;AACnB,SAAO,EAAE,aAAa,MAAM,CAAC,GAAI,YAAY,MAAM,CAAC,EAAA;AACtD;"}

package/dist/esm/model-meta.d.ts

@@ -1,4 +1,23 @@
 import { GeminiCachedContentOptions, GeminiCommonConfigOptions, GeminiSafetyOptions, GeminiStructuredOutputOptions, GeminiThinkingAdvancedOptions, GeminiThinkingOptions, GeminiToolConfigOptions } from './text/text-provider-options.js';
+declare const GEMINI_3_1_PRO: {
+    readonly name: "gemini-3.1-pro-preview";
+    readonly max_input_tokens: 1048576;
+    readonly max_output_tokens: 65536;
+    readonly knowledge_cutoff: "2025-01-01";
+    readonly supports: {
+        readonly input: ["text", "image", "audio", "video", "document"];
+        readonly output: ["text"];
+        readonly capabilities: ["batch_api", "caching", "code_execution", "file_search", "function_calling", "search_grounding", "structured_output", "thinking", "url_context"];
+    };
+    readonly pricing: {
+        readonly input: {
+            readonly normal: 2.5;
+        };
+        readonly output: {
+            readonly normal: 15;
+        };
+    };
+};
 declare const GEMINI_3_PRO: {
     readonly name: "gemini-3-pro-preview";
     readonly max_input_tokens: 1048576;
@@ -37,6 +56,25 @@ declare const GEMINI_3_FLASH: {
         };
     };
 };
+declare const GEMINI_3_1_FLASH_LITE: {
+    readonly name: "gemini-3.1-flash-lite-preview";
+    readonly max_input_tokens: 1048576;
+    readonly max_output_tokens: 65536;
+    readonly knowledge_cutoff: "2025-01-01";
+    readonly supports: {
+        readonly input: ["text", "image", "audio", "video", "document"];
+        readonly output: ["text"];
+        readonly capabilities: ["batch_api", "caching", "code_execution", "file_search", "function_calling", "search_grounding", "structured_output", "thinking", "url_context"];
+    };
+    readonly pricing: {
+        readonly input: {
+            readonly normal: 0.25;
+        };
+        readonly output: {
+            readonly normal: 1.5;
+        };
+    };
+};
 declare const GEMINI_2_5_PRO: {
     readonly name: "gemini-2.5-pro";
     readonly max_input_tokens: 1048576;
@@ -284,10 +322,10 @@ const VEO_2 = {
   GeminiGenerationConfigOptions &
   GeminiCachedContentOptions
 > */
-export declare const GEMINI_MODELS: readonly ["gemini-3-pro-preview", "gemini-3-flash-preview", "gemini-2.5-pro", "gemini-2.5-flash", "gemini-2.5-flash-preview-09-2025", "gemini-2.5-flash-lite", "gemini-2.5-flash-lite-preview-09-2025", "gemini-2.0-flash", "gemini-2.0-flash-lite"];
+export declare const GEMINI_MODELS: readonly ["gemini-3.1-pro-preview", "gemini-3-pro-preview", "gemini-3-flash-preview", "gemini-3.1-flash-lite-preview", "gemini-2.5-pro", "gemini-2.5-flash", "gemini-2.5-flash-preview-09-2025", "gemini-2.5-flash-lite", "gemini-2.5-flash-lite-preview-09-2025", "gemini-2.0-flash", "gemini-2.0-flash-lite"];
 export type GeminiModels = (typeof GEMINI_MODELS)[number];
 export type GeminiImageModels = (typeof GEMINI_IMAGE_MODELS)[number];
-export declare const GEMINI_IMAGE_MODELS: readonly ["gemini-3-pro-image-preview", "gemini-2.5-flash-image", "gemini-2.0-flash-preview-image-generation", "imagen-3.0-generate-002", "imagen-4.0-generate-001", "imagen-4.0-fast-generate-001", "imagen-4.0-ultra-generate-001"];
+export declare const GEMINI_IMAGE_MODELS: readonly ["gemini-3.1-flash-image-preview", "gemini-3-pro-image-preview", "gemini-2.5-flash-image", "gemini-2.0-flash-preview-image-generation", "imagen-3.0-generate-002", "imagen-4.0-generate-001", "imagen-4.0-fast-generate-001", "imagen-4.0-ultra-generate-001"];
 /**
  * Text-to-speech models
  * @experimental Gemini TTS is an experimental feature and may change.
@@ -300,8 +338,10 @@ export declare const GEMINI_TTS_MODELS: readonly ["gemini-2.5-flash-preview-tts"
 export declare const GEMINI_TTS_VOICES: readonly ["Zephyr", "Puck", "Charon", "Kore", "Fenrir", "Leda", "Orus", "Aoede", "Callirrhoe", "Autonoe", "Enceladus", "Iapetus", "Umbriel", "Algieba", "Despina", "Erinome", "Algenib", "Rasalgethi", "Laomedeia", "Achernar", "Alnilam", "Schedar", "Gacrux", "Pulcherrima", "Achird", "Zubenelgenubi", "Vindemiatrix", "Sadachbia", "Sadaltager", "Sulafat"];
 export type GeminiTTSVoice = (typeof GEMINI_TTS_VOICES)[number];
 export type GeminiChatModelProviderOptionsByName = {
+    [GEMINI_3_1_PRO.name]: GeminiToolConfigOptions & GeminiSafetyOptions & GeminiCommonConfigOptions & GeminiCachedContentOptions & GeminiStructuredOutputOptions & GeminiThinkingOptions & GeminiThinkingAdvancedOptions;
     [GEMINI_3_PRO.name]: GeminiToolConfigOptions & GeminiSafetyOptions & GeminiCommonConfigOptions & GeminiCachedContentOptions & GeminiStructuredOutputOptions & GeminiThinkingOptions & GeminiThinkingAdvancedOptions;
     [GEMINI_3_FLASH.name]: GeminiToolConfigOptions & GeminiSafetyOptions & GeminiCommonConfigOptions & GeminiCachedContentOptions & GeminiStructuredOutputOptions & GeminiThinkingOptions & GeminiThinkingAdvancedOptions;
+    [GEMINI_3_1_FLASH_LITE.name]: GeminiToolConfigOptions & GeminiSafetyOptions & GeminiCommonConfigOptions & GeminiCachedContentOptions & GeminiStructuredOutputOptions & GeminiThinkingOptions;
     [GEMINI_2_5_PRO.name]: GeminiToolConfigOptions & GeminiSafetyOptions & GeminiCommonConfigOptions & GeminiCachedContentOptions & GeminiStructuredOutputOptions & GeminiThinkingOptions;
     [GEMINI_2_5_FLASH.name]: GeminiToolConfigOptions & GeminiSafetyOptions & GeminiCommonConfigOptions & GeminiCachedContentOptions & GeminiStructuredOutputOptions & GeminiThinkingOptions;
     [GEMINI_2_5_FLASH_PREVIEW.name]: GeminiToolConfigOptions & GeminiSafetyOptions & GeminiCommonConfigOptions & GeminiCachedContentOptions & GeminiStructuredOutputOptions & GeminiThinkingOptions;
@@ -324,8 +364,10 @@ export type GeminiChatModelProviderOptionsByName = {
  * @see https://ai.google.dev/gemini-api/docs/document-processing
  */
 export type GeminiModelInputModalitiesByName = {
+    [GEMINI_3_1_PRO.name]: typeof GEMINI_3_1_PRO.supports.input;
     [GEMINI_3_PRO.name]: typeof GEMINI_3_PRO.supports.input;
     [GEMINI_3_FLASH.name]: typeof GEMINI_3_FLASH.supports.input;
+    [GEMINI_3_1_FLASH_LITE.name]: typeof GEMINI_3_1_FLASH_LITE.supports.input;
     [GEMINI_2_5_PRO.name]: typeof GEMINI_2_5_PRO.supports.input;
     [GEMINI_2_5_FLASH_LITE.name]: typeof GEMINI_2_5_FLASH_LITE.supports.input;
     [GEMINI_2_5_FLASH_LITE_PREVIEW.name]: typeof GEMINI_2_5_FLASH_LITE_PREVIEW.supports.input;

package/dist/esm/model-meta.js

@@ -1,3 +1,6 @@
+const GEMINI_3_1_PRO = {
+  name: "gemini-3.1-pro-preview"
+};
 const GEMINI_3_PRO = {
   name: "gemini-3-pro-preview"
 };
@@ -7,6 +10,12 @@ const GEMINI_3_FLASH = {
 const GEMINI_3_PRO_IMAGE = {
   name: "gemini-3-pro-image-preview"
 };
+const GEMINI_3_1_FLASH_IMAGE = {
+  name: "gemini-3.1-flash-image-preview"
+};
+const GEMINI_3_1_FLASH_LITE = {
+  name: "gemini-3.1-flash-lite-preview"
+};
 const GEMINI_2_5_PRO = {
   name: "gemini-2.5-pro"
 };
@@ -53,8 +62,10 @@ const IMAGEN_3 = {
   name: "imagen-3.0-generate-002"
 };
 const GEMINI_MODELS = [
+  GEMINI_3_1_PRO.name,
   GEMINI_3_PRO.name,
   GEMINI_3_FLASH.name,
+  GEMINI_3_1_FLASH_LITE.name,
   GEMINI_2_5_PRO.name,
   GEMINI_2_5_FLASH.name,
   GEMINI_2_5_FLASH_PREVIEW.name,
@@ -64,6 +75,7 @@ const GEMINI_MODELS = [
   GEMINI_2_FLASH_LITE.name
 ];
 const GEMINI_IMAGE_MODELS = [
+  GEMINI_3_1_FLASH_IMAGE.name,
   GEMINI_3_PRO_IMAGE.name,
   GEMINI_2_5_FLASH_IMAGE.name,
   GEMINI_2_FLASH_IMAGE.name,

package/dist/esm/model-meta.js.map

@@ -1 +1 @@
-{"version":3,"file":"model-meta.js","sources":["../../src/model-meta.ts"],"sourcesContent":["import type {\n  GeminiCachedContentOptions,\n  GeminiCommonConfigOptions,\n  GeminiSafetyOptions,\n  GeminiStructuredOutputOptions,\n  GeminiThinkingAdvancedOptions,\n  GeminiThinkingOptions,\n  GeminiToolConfigOptions,\n} from './text/text-provider-options'\n\ninterface ModelMeta<TProviderOptions = unknown> {\n  name: string\n  supports: {\n    input: Array<'text' | 'image' | 'audio' | 'video' | 'document'>\n    output: Array<'text' | 'image' | 'audio' | 'video'>\n    capabilities?: Array<\n      | 'audio_generation'\n      | 'batch_api'\n      | 'caching'\n      | 'code_execution'\n      | 'file_search'\n      | 'function_calling'\n      | 'grounding_with_gmaps'\n      | 'image_generation'\n      | 'live_api'\n      | 'search_grounding'\n      | 'structured_output'\n      | 'thinking'\n      | 'url_context'\n    >\n  }\n  max_input_tokens?: number\n  max_output_tokens?: number\n  knowledge_cutoff?: string\n  pricing?: {\n    input: {\n      normal: number\n      cached?: number\n    }\n    output: {\n      normal: number\n    }\n  }\n  /**\n   * Type-level description of which provider options this model supports.\n   */\n  providerOptions?: TProviderOptions\n}\n\nconst GEMINI_3_PRO = {\n  name: 'gemini-3-pro-preview',\n  max_input_tokens: 1_048_576,\n  max_output_tokens: 65_536,\n  knowledge_cutoff: '2025-01-01',\n  supports: {\n    input: ['text', 'image', 'audio', 'video', 'document'],\n    output: ['text'],\n    capabilities: [\n      'batch_api',\n      'caching',\n      'code_execution',\n      'file_search',\n      'function_calling',\n      'search_grounding',\n      'structured_output',\n      'thinking',\n      'url_context',\n    ],\n  },\n  pricing: {\n    input: {\n      normal: 2.5,\n    },\n    output: {\n      normal: 15,\n    },\n  },\n} as const satisfies ModelMeta<\n  GeminiToolConfigOptions &\n    GeminiSafetyOptions &\n    GeminiCommonConfigOptions &\n    GeminiCachedContentOptions &\n    GeminiStructuredOutputOptions &\n    GeminiThinkingOptions &\n    GeminiThinkingAdvancedOptions\n>\n\nconst GEMINI_3_FLASH = {\n  name: 'gemini-3-flash-preview',\n  max_input_tokens: 1_048_576,\n  max_output_tokens: 65_536,\n  knowledge_cutoff: '2025-01-01',\n  supports: {\n    input: ['text', 'image', 'audio', 'video', 'document'],\n    output: ['text'],\n    capabilities: [\n      'batch_api',\n      'caching',\n      'code_execution',\n      'file_search',\n      'function_calling',\n      'search_grounding',\n      'structured_output',\n      'thinking',\n      'url_context',\n    ],\n  },\n  pricing: {\n    input: {\n      normal: 0.5,\n    },\n    output: {\n      normal: 3,\n    },\n  },\n} as const satisfies ModelMeta<\n  GeminiToolConfigOptions &\n    GeminiSafetyOptions &\n    GeminiCommonConfigOptions &\n    GeminiCachedContentOptions &\n    GeminiStructuredOutputOptions &\n    GeminiThinkingOptions &\n    GeminiThinkingAdvancedOptions\n>\n\nconst GEMINI_3_PRO_IMAGE = {\n  name: 'gemini-3-pro-image-preview',\n  max_input_tokens: 65_536,\n  max_output_tokens: 32_768,\n  knowledge_cutoff: '2025-01-01',\n  supports: {\n    input: ['text', 'image'],\n    output: ['text', 'image'],\n    capabilities: [\n      'batch_api',\n      'image_generation',\n      'search_grounding',\n      'structured_output',\n      'thinking',\n    ],\n  },\n  pricing: {\n    input: {\n      normal: 2,\n    },\n    output: {\n      normal: 0.134,\n    },\n  },\n} as const satisfies ModelMeta<\n  GeminiToolConfigOptions &\n    GeminiSafetyOptions &\n    GeminiCommonConfigOptions &\n    GeminiCachedContentOptions &\n    GeminiStructuredOutputOptions &\n    GeminiThinkingOptions &\n    GeminiThinkingAdvancedOptions\n>\n\nconst GEMINI_2_5_PRO = {\n  name: 'gemini-2.5-pro',\n  max_input_tokens: 1_048_576,\n  max_output_tokens: 65_536,\n  knowledge_cutoff: '2025-01-01',\n  supports: {\n    input: ['text', 'image', 'audio', 'video', 'document'],\n    output: ['text'],\n    capabilities: [\n      'batch_api',\n      'caching',\n      'code_execution',\n      'file_search',\n      'function_calling',\n      'grounding_with_gmaps',\n      'search_grounding',\n      'structured_output',\n      'thinking',\n      'url_context',\n    ],\n  },\n  pricing: {\n    input: {\n      normal: 2.5,\n    },\n    output: {\n      normal: 15,\n    },\n  },\n} as const satisfies ModelMeta<\n  GeminiToolConfigOptions &\n    GeminiSafetyOptions &\n    GeminiCommonConfigOptions &\n    GeminiCachedContentOptions &\n    GeminiStructuredOutputOptions &\n    GeminiThinkingOptions\n>\n\nconst GEMINI_2_5_PRO_TTS = {\n  name: 'gemini-2.5-pro-preview-tts',\n  max_input_tokens: 8_192,\n  max_output_tokens: 16_384,\n  knowledge_cutoff: '2025-05-01',\n  supports: {\n    input: ['text'],\n    output: ['audio'],\n    capabilities: ['audio_generation', 'file_search'],\n  },\n  pricing: {\n    input: {\n      normal: 2.5,\n    },\n    output: {\n      normal: 15,\n    },\n  },\n} as const satisfies ModelMeta<\n  GeminiToolConfigOptions &\n    GeminiSafetyOptions &\n    GeminiCommonConfigOptions &\n    GeminiCachedContentOptions\n>\n\nconst GEMINI_2_5_FLASH = {\n  name: 'gemini-2.5-flash',\n  max_input_tokens: 1_048_576,\n  max_output_tokens: 65_536,\n  knowledge_cutoff: '2025-01-01',\n  supports: {\n    input: ['text', 'image', 'audio', 'video'],\n    output: ['text'],\n    capabilities: [\n      'batch_api',\n      'caching',\n      'code_execution',\n      'file_search',\n      'function_calling',\n      'grounding_with_gmaps',\n      'search_grounding',\n      'structured_output',\n      'thinking',\n      'url_context',\n    ],\n  },\n  pricing: {\n    input: {\n      normal: 1,\n    },\n    output: {\n      normal: 2.5,\n    },\n  },\n} as const satisfies ModelMeta<\n  GeminiToolConfigOptions &\n    GeminiSafetyOptions &\n    GeminiCommonConfigOptions &\n    GeminiCachedContentOptions &\n    GeminiStructuredOutputOptions &\n    GeminiThinkingOptions\n>\n\nconst GEMINI_2_5_FLASH_PREVIEW = {\n  name: 'gemini-2.5-flash-preview-09-2025',\n  max_input_tokens: 1_048_576,\n  max_output_tokens: 65_536,\n  knowledge_cutoff: '2025-01-01',\n  supports: {\n    input: ['text', 'image', 'audio', 'video'],\n    output: ['text'],\n    capabilities: [\n      'batch_api',\n      'caching',\n      'code_execution',\n      'file_search',\n      'function_calling',\n      'search_grounding',\n      'structured_output',\n      'thinking',\n      'url_context',\n    ],\n  },\n  pricing: {\n    input: {\n      normal: 1,\n    },\n    output: {\n      normal: 2.5,\n    },\n  },\n} as const satisfies ModelMeta<\n  GeminiToolConfigOptions &\n    GeminiSafetyOptions &\n    GeminiCommonConfigOptions &\n    GeminiCachedContentOptions &\n    GeminiStructuredOutputOptions &\n    GeminiThinkingOptions\n>\n\nconst GEMINI_2_5_FLASH_IMAGE = {\n  name: 'gemini-2.5-flash-image',\n  max_input_tokens: 1_048_576,\n  max_output_tokens: 65_536,\n  knowledge_cutoff: '2025-06-01',\n  supports: {\n    input: ['text', 'image'],\n    output: ['text', 'image'],\n    capabilities: [\n      'batch_api',\n      'caching',\n      'file_search',\n      'image_generation',\n      'structured_output',\n    ],\n  },\n  pricing: {\n    input: {\n      normal: 0.3,\n    },\n    output: {\n      normal: 0.4,\n    },\n  },\n} as const satisfies ModelMeta<\n  GeminiToolConfigOptions &\n    GeminiSafetyOptions &\n    GeminiCommonConfigOptions &\n    GeminiCachedContentOptions\n>\n/**\nconst GEMINI_2_5_FLASH_LIVE = {\n  name: 'gemini-2.5-flash-native-audio-preview-09-2025',\n  max_input_tokens: 141_072,\n  max_output_tokens: 8_192,\n  knowledge_cutoff: '2025-01-01',\n  supports: {\n    input: ['text', 'audio', 'video'],\n    output: ['text', 'audio'],\n    capabilities: [\n      'audio_generation',\n      'file_search',\n      'function_calling',\n      'live_api',\n      'search_grounding',\n      'thinking',\n    ],\n  },\n  pricing: {\n    // todo find this info\n    input: {\n      normal: 0,\n    },\n    output: {\n      normal: 0,\n    },\n  },\n} as const satisfies ModelMeta<\n  GeminiToolConfigOptions &\n  GeminiSafetyOptions &\n  GeminiGenerationConfigOptions &\n  GeminiCachedContentOptions &\n  GeminiThinkingOptions\n>\n*/\nconst GEMINI_2_5_FLASH_TTS = {\n  name: 'gemini-2.5-flash-preview-tts',\n  max_input_tokens: 8_192,\n  max_output_tokens: 16_384,\n  knowledge_cutoff: '2025-05-01',\n  supports: {\n    input: ['text'],\n    output: ['audio'],\n    capabilities: ['audio_generation', 'batch_api', 'file_search'],\n  },\n  pricing: {\n    input: {\n      normal: 1,\n    },\n    output: {\n      normal: 2.5,\n    },\n  },\n} as const satisfies ModelMeta<\n  GeminiToolConfigOptions &\n    GeminiSafetyOptions &\n    GeminiCommonConfigOptions &\n    GeminiCachedContentOptions\n>\n\nconst GEMINI_2_5_FLASH_LITE = {\n  name: 'gemini-2.5-flash-lite',\n  max_input_tokens: 1_048_576,\n  max_output_tokens: 65_536,\n  knowledge_cutoff: '2025-01-01',\n  supports: {\n    input: ['text', 'image', 'audio', 'video', 'document'],\n    output: ['text'],\n    capabilities: [\n      'batch_api',\n      'caching',\n      'code_execution',\n      'function_calling',\n      'grounding_with_gmaps',\n      'search_grounding',\n      'structured_output',\n      'thinking',\n      'url_context',\n    ],\n  },\n  pricing: {\n    input: {\n      normal: 0.1,\n    },\n    output: {\n      normal: 0.4,\n    },\n  },\n} as const satisfies ModelMeta<\n  GeminiToolConfigOptions &\n    GeminiSafetyOptions &\n    GeminiCommonConfigOptions &\n    GeminiCachedContentOptions &\n    GeminiStructuredOutputOptions &\n    GeminiThinkingOptions\n>\n\nconst GEMINI_2_5_FLASH_LITE_PREVIEW = {\n  name: 'gemini-2.5-flash-lite-preview-09-2025',\n  max_input_tokens: 1_048_576,\n  max_output_tokens: 65_536,\n  knowledge_cutoff: '2025-01-01',\n  supports: {\n    input: ['text', 'image', 'audio', 'video', 'document'],\n    output: ['text'],\n    capabilities: [\n      'batch_api',\n      'caching',\n      'code_execution',\n      'function_calling',\n      'search_grounding',\n      'structured_output',\n      'thinking',\n      'url_context',\n    ],\n  },\n  pricing: {\n    input: {\n      normal: 0.1,\n    },\n    output: {\n      normal: 0.4,\n    },\n  },\n} as const satisfies ModelMeta<\n  GeminiToolConfigOptions &\n    GeminiSafetyOptions &\n    GeminiCommonConfigOptions &\n    GeminiCachedContentOptions &\n    GeminiStructuredOutputOptions &\n    GeminiThinkingOptions\n>\n\nconst GEMINI_2_FLASH = {\n  name: 'gemini-2.0-flash',\n  max_input_tokens: 1_048_576,\n  max_output_tokens: 8_192,\n  knowledge_cutoff: '2024-08-01',\n  supports: {\n    input: ['text', 'image', 'audio', 'video'],\n    output: ['text'],\n    capabilities: [\n      'batch_api',\n      'caching',\n      'code_execution',\n      'function_calling',\n      'grounding_with_gmaps',\n      'live_api',\n      'search_grounding',\n      'structured_output',\n    ],\n  },\n  pricing: {\n    input: {\n      normal: 0.1,\n    },\n    output: {\n      normal: 0.4,\n    },\n  },\n} as const satisfies ModelMeta<\n  GeminiToolConfigOptions &\n    GeminiSafetyOptions &\n    GeminiCommonConfigOptions &\n    GeminiCachedContentOptions &\n    GeminiStructuredOutputOptions\n>\n\nconst GEMINI_2_FLASH_IMAGE = {\n  name: 'gemini-2.0-flash-preview-image-generation',\n  max_input_tokens: 32_768,\n  max_output_tokens: 8_192,\n  knowledge_cutoff: '2024-08-01',\n  supports: {\n    input: ['text', 'image', 'audio', 'video'],\n    output: ['text'],\n    capabilities: [\n      'batch_api',\n      'caching',\n      'image_generation',\n      'structured_output',\n    ],\n  },\n  pricing: {\n    input: {\n      normal: 0.1,\n    },\n    output: {\n      normal: 0.039,\n    },\n  },\n} as const satisfies ModelMeta<\n  GeminiToolConfigOptions &\n    GeminiSafetyOptions &\n    GeminiCommonConfigOptions &\n    GeminiCachedContentOptions\n>\n/* \nconst GEMINI_2_FLASH_LIVE = {\n  name: 'gemini-2.0-flash-live-001',\n  max_input_tokens: 1_048_576,\n  max_output_tokens: 8_192,\n  knowledge_cutoff: '2024-08-01',\n  supports: {\n    input: ['text', 'audio', 'video'],\n    output: ['text', 'audio'],\n    capabilities: [\n      'audio_generation',\n      'code_execution',\n      'function_calling',\n      'live_api',\n      'search_grounding',\n      'structured_output',\n      'url_context',\n    ],\n  },\n  pricing: {\n    // todo find this info\n    input: {\n      normal: 0,\n    },\n    output: {\n      normal: 0,\n    },\n  },\n} as const satisfies ModelMeta<\n  GeminiToolConfigOptions &\n  GeminiSafetyOptions &\n  GeminiGenerationConfigOptions &\n  GeminiCachedContentOptions\n> */\n\nconst GEMINI_2_FLASH_LITE = {\n  name: 'gemini-2.0-flash-lite',\n  max_input_tokens: 1_048_576,\n  max_output_tokens: 8_192,\n  knowledge_cutoff: '2024-08-01',\n  supports: {\n    input: ['text', 'audio', 'video', 'image'],\n    output: ['text'],\n    capabilities: [\n      'batch_api',\n      'caching',\n      'function_calling',\n      'structured_output',\n    ],\n  },\n  pricing: {\n    input: {\n      normal: 0.075,\n    },\n    output: {\n      normal: 0.3,\n    },\n  },\n} as const satisfies ModelMeta<\n  GeminiToolConfigOptions &\n    GeminiSafetyOptions &\n    GeminiCommonConfigOptions &\n    GeminiCachedContentOptions &\n    GeminiStructuredOutputOptions\n>\n\nconst IMAGEN_4_GENERATE = {\n  name: 'imagen-4.0-generate-001',\n  max_input_tokens: 480,\n  max_output_tokens: 4,\n  supports: {\n    input: ['text'],\n    output: ['image'],\n  },\n  pricing: {\n    input: {\n      normal: 0,\n    },\n    output: {\n      normal: 0.4,\n    },\n  },\n} as const satisfies ModelMeta<\n  GeminiToolConfigOptions &\n    GeminiSafetyOptions &\n    GeminiCommonConfigOptions &\n    GeminiCachedContentOptions\n>\n\nconst IMAGEN_4_GENERATE_ULTRA = {\n  name: 'imagen-4.0-ultra-generate-001',\n  max_input_tokens: 480,\n  max_output_tokens: 4,\n  supports: {\n    input: ['text'],\n    output: ['image'],\n  },\n  pricing: {\n    input: {\n      normal: 0,\n    },\n    output: {\n      normal: 0.6,\n    },\n  },\n} as const satisfies ModelMeta<\n  GeminiToolConfigOptions &\n    GeminiSafetyOptions &\n    GeminiCommonConfigOptions &\n    GeminiCachedContentOptions\n>\n\nconst IMAGEN_4_GENERATE_FAST = {\n  name: 'imagen-4.0-fast-generate-001',\n  max_input_tokens: 480,\n  max_output_tokens: 4,\n  supports: {\n    input: ['text'],\n    output: ['image'],\n  },\n  pricing: {\n    input: {\n      normal: 0,\n    },\n    output: {\n      normal: 0.2,\n    },\n  },\n} as const satisfies ModelMeta<\n  GeminiToolConfigOptions &\n    GeminiSafetyOptions &\n    GeminiCommonConfigOptions &\n    GeminiCachedContentOptions\n>\n\nconst IMAGEN_3 = {\n  name: 'imagen-3.0-generate-002',\n  max_output_tokens: 4,\n  supports: {\n    input: ['text'],\n    output: ['image'],\n  },\n  pricing: {\n    input: {\n      normal: 0,\n    },\n    output: {\n      normal: 0.03,\n    },\n  },\n} as const satisfies ModelMeta<\n  GeminiToolConfigOptions &\n    GeminiSafetyOptions &\n    GeminiCommonConfigOptions &\n    GeminiCachedContentOptions\n>\n/** \nconst VEO_3_1_PREVIEW = {\n  name: 'veo-3.1-generate-preview',\n  max_input_tokens: 1024,\n  max_output_tokens: 1,\n  supports: {\n    input: ['text', 'image'],\n    output: ['video', 'audio'],\n  },\n  pricing: {\n    input: {\n      normal: 0,\n    },\n    output: {\n      normal: 0.4,\n    },\n  },\n} as const satisfies ModelMeta<\n  GeminiToolConfigOptions &\n  GeminiSafetyOptions &\n  GeminiGenerationConfigOptions &\n  GeminiCachedContentOptions\n>\n\nconst VEO_3_1_FAST_PREVIEW = {\n  name: 'veo-3.1-fast-generate-preview',\n  max_input_tokens: 1024,\n  max_output_tokens: 1,\n  supports: {\n    input: ['text', 'image'],\n    output: ['video', 'audio'],\n  },\n  pricing: {\n    input: {\n      normal: 0,\n    },\n    output: {\n      normal: 0.15,\n    },\n  },\n} as const satisfies ModelMeta<\n  GeminiToolConfigOptions &\n  GeminiSafetyOptions &\n  GeminiGenerationConfigOptions &\n  GeminiCachedContentOptions\n>\n\nconst VEO_3 = {\n  name: 'veo-3.0-generate-001',\n  max_input_tokens: 1024,\n  max_output_tokens: 1,\n  supports: {\n    input: ['text', 'image'],\n    output: ['video', 'audio'],\n  },\n  pricing: {\n    input: {\n      normal: 0,\n    },\n    output: {\n      normal: 0.4,\n    },\n  },\n} as const satisfies ModelMeta<\n  GeminiToolConfigOptions &\n  GeminiSafetyOptions &\n  GeminiGenerationConfigOptions &\n  GeminiCachedContentOptions\n>\n\nconst VEO_3_FAST = {\n  name: 'veo-3.0-fast-generate-001',\n  max_input_tokens: 1024,\n  max_output_tokens: 1,\n  supports: {\n    input: ['text', 'image'],\n    output: ['video', 'audio'],\n  },\n  pricing: {\n    input: {\n      normal: 0,\n    },\n    output: {\n      normal: 0.15,\n    },\n  },\n} as const satisfies ModelMeta<\n  GeminiToolConfigOptions &\n  GeminiSafetyOptions &\n  GeminiGenerationConfigOptions &\n  GeminiCachedContentOptions\n>\n\nconst VEO_2 = {\n  name: 'veo-2.0-generate-001',\n  max_output_tokens: 2,\n  supports: {\n    input: ['text', 'image'],\n    output: ['video'],\n  },\n  pricing: {\n    input: {\n      normal: 0,\n    },\n    output: {\n      normal: 0.35,\n    },\n  },\n} as const satisfies ModelMeta<\n  GeminiToolConfigOptions &\n  GeminiSafetyOptions &\n  GeminiGenerationConfigOptions &\n  GeminiCachedContentOptions\n> */\n\n/* const GEMINI_MODEL_META = {\n  [GEMINI_3_PRO.name]: GEMINI_3_PRO,\n  [GEMINI_2_5_PRO.name]: GEMINI_2_5_PRO,\n  [GEMINI_2_5_PRO_TTS.name]: GEMINI_2_5_PRO_TTS,\n  [GEMINI_2_5_FLASH.name]: GEMINI_2_5_FLASH,\n  [GEMINI_2_5_FLASH_PREVIEW.name]: GEMINI_2_5_FLASH_PREVIEW,\n  [GEMINI_2_5_FLASH_IMAGE.name]: GEMINI_2_5_FLASH_IMAGE,\n  [GEMINI_2_5_FLASH_LIVE.name]: GEMINI_2_5_FLASH_LIVE,\n  [GEMINI_2_5_FLASH_TTS.name]: GEMINI_2_5_FLASH_TTS,\n  [GEMINI_2_5_FLASH_LITE.name]: GEMINI_2_5_FLASH_LITE,\n  [GEMINI_2_5_FLASH_LITE_PREVIEW.name]: GEMINI_2_5_FLASH_LITE_PREVIEW,\n  [GEMINI_2_FLASH.name]: GEMINI_2_FLASH,\n  [GEMINI_2_FLASH_IMAGE.name]: GEMINI_2_FLASH_IMAGE,\n  [GEMINI_2_FLASH_LIVE.name]: GEMINI_2_FLASH_LIVE,\n  [GEMINI_2_FLASH_LITE.name]: GEMINI_2_FLASH_LITE,\n  [IMAGEN_4_GENERATE.name]: IMAGEN_4_GENERATE,\n  [IMAGEN_4_GENERATE_ULTRA.name]: IMAGEN_4_GENERATE_ULTRA,\n  [IMAGEN_4_GENERATE_FAST.name]: IMAGEN_4_GENERATE_FAST,\n  [IMAGEN_3.name]: IMAGEN_3,\n  [VEO_3_1_PREVIEW.name]: VEO_3_1_PREVIEW,\n  [VEO_3_1_FAST_PREVIEW.name]: VEO_3_1_FAST_PREVIEW,\n  [VEO_3.name]: VEO_3,\n  [VEO_3_FAST.name]: VEO_3_FAST,\n  [VEO_2.name]: VEO_2,\n} as const */\n\nexport const GEMINI_MODELS = [\n  GEMINI_3_PRO.name,\n  GEMINI_3_FLASH.name,\n  GEMINI_2_5_PRO.name,\n  GEMINI_2_5_FLASH.name,\n  GEMINI_2_5_FLASH_PREVIEW.name,\n  GEMINI_2_5_FLASH_LITE.name,\n  GEMINI_2_5_FLASH_LITE_PREVIEW.name,\n  GEMINI_2_FLASH.name,\n  GEMINI_2_FLASH_LITE.name,\n] as const\n\nexport type GeminiModels = (typeof GEMINI_MODELS)[number]\n\nexport type GeminiImageModels = (typeof GEMINI_IMAGE_MODELS)[number]\n\nexport const GEMINI_IMAGE_MODELS = [\n  GEMINI_3_PRO_IMAGE.name,\n  GEMINI_2_5_FLASH_IMAGE.name,\n  GEMINI_2_FLASH_IMAGE.name,\n  IMAGEN_3.name,\n  IMAGEN_4_GENERATE.name,\n  IMAGEN_4_GENERATE_FAST.name,\n  IMAGEN_4_GENERATE_ULTRA.name,\n] as const\n\n/**\n * Text-to-speech models\n * @experimental Gemini TTS is an experimental feature and may change.\n */\nexport const GEMINI_TTS_MODELS = [\n  GEMINI_2_5_FLASH_TTS.name,\n  GEMINI_2_5_PRO_TTS.name,\n] as const\n\n/**\n * Available voice names for Gemini TTS\n * @see https://ai.google.dev/gemini-api/docs/speech-generation\n */\nexport const GEMINI_TTS_VOICES = [\n  'Zephyr',\n  'Puck',\n  'Charon',\n  'Kore',\n  'Fenrir',\n  'Leda',\n  'Orus',\n  'Aoede',\n  'Callirrhoe',\n  'Autonoe',\n  'Enceladus',\n  'Iapetus',\n  'Umbriel',\n  'Algieba',\n  'Despina',\n  'Erinome',\n  'Algenib',\n  'Rasalgethi',\n  'Laomedeia',\n  'Achernar',\n  'Alnilam',\n  'Schedar',\n  'Gacrux',\n  'Pulcherrima',\n  'Achird',\n  'Zubenelgenubi',\n  'Vindemiatrix',\n  'Sadachbia',\n  'Sadaltager',\n  'Sulafat',\n] as const\n\nexport type GeminiTTSVoice = (typeof GEMINI_TTS_VOICES)[number]\n\n/*   const GEMINI_AUDIO_MODELS = [\n  GEMINI_2_5_PRO_TTS.name,\n  GEMINI_2_5_FLASH_TTS.name,\n  GEMINI_2_5_FLASH_LIVE.name,\n  GEMINI_2_FLASH_LIVE.name,\n] as const\n\n  const GEMINI_VIDEO_MODELS = [\n  VEO_3_1_PREVIEW.name,\n  VEO_3_1_FAST_PREVIEW.name,\n  VEO_3.name,\n  VEO_3_FAST.name,\n  VEO_2.name,\n] as const */\n\n// Manual type map for per-model provider options\nexport type GeminiChatModelProviderOptionsByName = {\n  // Models with thinking and structured output support\n  [GEMINI_3_PRO.name]: GeminiToolConfigOptions &\n    GeminiSafetyOptions &\n    GeminiCommonConfigOptions &\n    GeminiCachedContentOptions &\n    GeminiStructuredOutputOptions &\n    GeminiThinkingOptions &\n    GeminiThinkingAdvancedOptions\n  [GEMINI_3_FLASH.name]: GeminiToolConfigOptions &\n    GeminiSafetyOptions &\n    GeminiCommonConfigOptions &\n    GeminiCachedContentOptions &\n    GeminiStructuredOutputOptions &\n    GeminiThinkingOptions &\n    GeminiThinkingAdvancedOptions\n  [GEMINI_2_5_PRO.name]: GeminiToolConfigOptions &\n    GeminiSafetyOptions &\n    GeminiCommonConfigOptions &\n    GeminiCachedContentOptions &\n    GeminiStructuredOutputOptions &\n    GeminiThinkingOptions\n  [GEMINI_2_5_FLASH.name]: GeminiToolConfigOptions &\n    GeminiSafetyOptions &\n    GeminiCommonConfigOptions &\n    GeminiCachedContentOptions &\n    GeminiStructuredOutputOptions &\n    GeminiThinkingOptions\n  [GEMINI_2_5_FLASH_PREVIEW.name]: GeminiToolConfigOptions &\n    GeminiSafetyOptions &\n    GeminiCommonConfigOptions &\n    GeminiCachedContentOptions &\n    GeminiStructuredOutputOptions &\n    GeminiThinkingOptions\n  [GEMINI_2_5_FLASH_LITE.name]: GeminiToolConfigOptions &\n    GeminiSafetyOptions &\n    GeminiCommonConfigOptions &\n    GeminiCachedContentOptions &\n    GeminiStructuredOutputOptions &\n    GeminiThinkingOptions\n  [GEMINI_2_5_FLASH_LITE_PREVIEW.name]: GeminiToolConfigOptions &\n    GeminiSafetyOptions &\n    GeminiCommonConfigOptions &\n    GeminiCachedContentOptions &\n    GeminiStructuredOutputOptions &\n    GeminiThinkingOptions\n  // Models with structured output but no thinking support\n  [GEMINI_2_FLASH.name]: GeminiToolConfigOptions &\n    GeminiSafetyOptions &\n    GeminiCommonConfigOptions &\n    GeminiCachedContentOptions &\n    GeminiStructuredOutputOptions\n  [GEMINI_2_FLASH_LITE.name]: GeminiToolConfigOptions &\n    GeminiSafetyOptions &\n    GeminiCommonConfigOptions &\n    GeminiCachedContentOptions &\n    GeminiStructuredOutputOptions\n}\n\n/**\n * Type-only map from chat model name to its supported input modalities.\n * Based on the 'supports.input' arrays defined for each model.\n * Note: 'document' in the model meta is mapped to 'document' modality.\n * Used by the core AI types to constrain ContentPart types based on the selected model.\n * Note: These must be inlined as readonly arrays (not typeof) because the model\n * constants are not exported and typeof references don't work in .d.ts files\n * when consumed by external packages.\n *\n * @see https://ai.google.dev/gemini-api/docs/vision\n * @see https://ai.google.dev/gemini-api/docs/audio\n * @see https://ai.google.dev/gemini-api/docs/document-processing\n */\nexport type GeminiModelInputModalitiesByName = {\n  // Models with full multimodal support (text, image, audio, video, document)\n  [GEMINI_3_PRO.name]: typeof GEMINI_3_PRO.supports.input\n  [GEMINI_3_FLASH.name]: typeof GEMINI_3_FLASH.supports.input\n  [GEMINI_2_5_PRO.name]: typeof GEMINI_2_5_PRO.supports.input\n  [GEMINI_2_5_FLASH_LITE.name]: typeof GEMINI_2_5_FLASH_LITE.supports.input\n  [GEMINI_2_5_FLASH_LITE_PREVIEW.name]: typeof GEMINI_2_5_FLASH_LITE_PREVIEW.supports.input\n\n  // Models with text, image, audio, video (no document)\n  [GEMINI_2_5_FLASH.name]: typeof GEMINI_2_5_FLASH.supports.input\n  [GEMINI_2_5_FLASH_PREVIEW.name]: typeof GEMINI_2_5_FLASH_PREVIEW.supports.input\n  [GEMINI_2_FLASH.name]: typeof GEMINI_2_FLASH.supports.input\n  [GEMINI_2_FLASH_LITE.name]: typeof GEMINI_2_FLASH_LITE.supports.input\n}\n"],"names":[],"mappings":"AAiDA,MAAM,eAAe;AAAA,EACnB,MAAM;AA2BR;AAUA,MAAM,iBAAiB;AAAA,EACrB,MAAM;AA2BR;AAUA,MAAM,qBAAqB;AAAA,EACzB,MAAM;AAuBR;AAUA,MAAM,iBAAiB;AAAA,EACrB,MAAM;AA4BR;AASA,MAAM,qBAAqB;AAAA,EACzB,MAAM;AAiBR;AAOA,MAAM,mBAAmB;AAAA,EACvB,MAAM;AA4BR;AASA,MAAM,2BAA2B;AAAA,EAC/B,MAAM;AA2BR;AASA,MAAM,yBAAyB;AAAA,EAC7B,MAAM;AAuBR;AAyCA,MAAM,uBAAuB;AAAA,EAC3B,MAAM;AAiBR;AAOA,MAAM,wBAAwB;AAAA,EAC5B,MAAM;AA2BR;AASA,MAAM,gCAAgC;AAAA,EACpC,MAAM;AA0BR;AASA,MAAM,iBAAiB;AAAA,EACrB,MAAM;AA0BR;AAQA,MAAM,uBAAuB;AAAA,EAC3B,MAAM;AAsBR;AAyCA,MAAM,sBAAsB;AAAA,EAC1B,MAAM;AAsBR;AAQA,MAAM,oBAAoB;AAAA,EACxB,MAAM;AAeR;AAOA,MAAM,0BAA0B;AAAA,EAC9B,MAAM;AAeR;AAOA,MAAM,yBAAyB;AAAA,EAC7B,MAAM;AAeR;AAOA,MAAM,WAAW;AAAA,EACf,MAAM;AAcR;AAmJO,MAAM,gBAAgB;AAAA,EAC3B,aAAa;AAAA,EACb,eAAe;AAAA,EACf,eAAe;AAAA,EACf,iBAAiB;AAAA,EACjB,yBAAyB;AAAA,EACzB,sBAAsB;AAAA,EACtB,8BAA8B;AAAA,EAC9B,eAAe;AAAA,EACf,oBAAoB;AACtB;AAMO,MAAM,sBAAsB;AAAA,EACjC,mBAAmB;AAAA,EACnB,uBAAuB;AAAA,EACvB,qBAAqB;AAAA,EACrB,SAAS;AAAA,EACT,kBAAkB;AAAA,EAClB,uBAAuB;AAAA,EACvB,wBAAwB;AAC1B;AAMO,MAAM,oBAAoB;AAAA,EAC/B,qBAAqB;AAAA,EACrB,mBAAmB;AACrB;AAMO,MAAM,oBAAoB;AAAA,EAC/B;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACF;"}
+{"version":3,"file":"model-meta.js","sources":["../../src/model-meta.ts"],"sourcesContent":["import type {\n  GeminiCachedContentOptions,\n  GeminiCommonConfigOptions,\n  GeminiSafetyOptions,\n  GeminiStructuredOutputOptions,\n  GeminiThinkingAdvancedOptions,\n  GeminiThinkingOptions,\n  GeminiToolConfigOptions,\n} from './text/text-provider-options'\n\ninterface ModelMeta<TProviderOptions = unknown> {\n  name: string\n  supports: {\n    input: Array<'text' | 'image' | 'audio' | 'video' | 'document'>\n    output: Array<'text' | 'image' | 'audio' | 'video'>\n    capabilities?: Array<\n      | 'audio_generation'\n      | 'batch_api'\n      | 'caching'\n      | 'code_execution'\n      | 'file_search'\n      | 'function_calling'\n      | 'grounding_with_gmaps'\n      | 'image_generation'\n      | 'live_api'\n      | 'search_grounding'\n      | 'structured_output'\n      | 'thinking'\n      | 'url_context'\n    >\n  }\n  max_input_tokens?: number\n  max_output_tokens?: number\n  knowledge_cutoff?: string\n  pricing?: {\n    input: {\n      normal: number\n      cached?: number\n    }\n    output: {\n      normal: number\n    }\n  }\n  /**\n   * Type-level description of which provider options this model supports.\n   */\n  providerOptions?: TProviderOptions\n}\n\nconst GEMINI_3_1_PRO = {\n  name: 'gemini-3.1-pro-preview',\n  max_input_tokens: 1_048_576,\n  max_output_tokens: 65_536,\n  knowledge_cutoff: '2025-01-01',\n  supports: {\n    input: ['text', 'image', 'audio', 'video', 'document'],\n    output: ['text'],\n    capabilities: [\n      'batch_api',\n      'caching',\n      'code_execution',\n      'file_search',\n      'function_calling',\n      'search_grounding',\n      'structured_output',\n      'thinking',\n      'url_context',\n    ],\n  },\n  pricing: {\n    input: {\n      normal: 2.5,\n    },\n    output: {\n      normal: 15,\n    },\n  },\n} as const satisfies ModelMeta<\n  GeminiToolConfigOptions &\n    GeminiSafetyOptions &\n    GeminiCommonConfigOptions &\n    GeminiCachedContentOptions &\n    GeminiStructuredOutputOptions &\n    GeminiThinkingOptions &\n    GeminiThinkingAdvancedOptions\n>\n\nconst GEMINI_3_PRO = {\n  name: 'gemini-3-pro-preview',\n  max_input_tokens: 1_048_576,\n  max_output_tokens: 65_536,\n  knowledge_cutoff: '2025-01-01',\n  supports: {\n    input: ['text', 'image', 'audio', 'video', 'document'],\n    output: ['text'],\n    capabilities: [\n      'batch_api',\n      'caching',\n      'code_execution',\n      'file_search',\n      'function_calling',\n      'search_grounding',\n      'structured_output',\n      'thinking',\n      'url_context',\n    ],\n  },\n  pricing: {\n    input: {\n      normal: 2.5,\n    },\n    output: {\n      normal: 15,\n    },\n  },\n} as const satisfies ModelMeta<\n  GeminiToolConfigOptions &\n    GeminiSafetyOptions &\n    GeminiCommonConfigOptions &\n    GeminiCachedContentOptions &\n    GeminiStructuredOutputOptions &\n    GeminiThinkingOptions &\n    GeminiThinkingAdvancedOptions\n>\n\nconst GEMINI_3_FLASH = {\n  name: 'gemini-3-flash-preview',\n  max_input_tokens: 1_048_576,\n  max_output_tokens: 65_536,\n  knowledge_cutoff: '2025-01-01',\n  supports: {\n    input: ['text', 'image', 'audio', 'video', 'document'],\n    output: ['text'],\n    capabilities: [\n      'batch_api',\n      'caching',\n      'code_execution',\n      'file_search',\n      'function_calling',\n      'search_grounding',\n      'structured_output',\n      'thinking',\n      'url_context',\n    ],\n  },\n  pricing: {\n    input: {\n      normal: 0.5,\n    },\n    output: {\n      normal: 3,\n    },\n  },\n} as const satisfies ModelMeta<\n  GeminiToolConfigOptions &\n    GeminiSafetyOptions &\n    GeminiCommonConfigOptions &\n    GeminiCachedContentOptions &\n    GeminiStructuredOutputOptions &\n    GeminiThinkingOptions &\n    GeminiThinkingAdvancedOptions\n>\n\nconst GEMINI_3_PRO_IMAGE = {\n  name: 'gemini-3-pro-image-preview',\n  max_input_tokens: 65_536,\n  max_output_tokens: 32_768,\n  knowledge_cutoff: '2025-01-01',\n  supports: {\n    input: ['text', 'image'],\n    output: ['text', 'image'],\n    capabilities: [\n      'batch_api',\n      'image_generation',\n      'search_grounding',\n      'structured_output',\n      'thinking',\n    ],\n  },\n  pricing: {\n    input: {\n      normal: 2,\n    },\n    output: {\n      normal: 0.134,\n    },\n  },\n} as const satisfies ModelMeta<\n  GeminiToolConfigOptions &\n    GeminiSafetyOptions &\n    GeminiCommonConfigOptions &\n    GeminiCachedContentOptions &\n    GeminiStructuredOutputOptions &\n    GeminiThinkingOptions &\n    GeminiThinkingAdvancedOptions\n>\n\nconst GEMINI_3_1_FLASH_IMAGE = {\n  name: 'gemini-3.1-flash-image-preview',\n  max_input_tokens: 65_536,\n  max_output_tokens: 65_536,\n  knowledge_cutoff: '2025-01-01',\n  supports: {\n    input: ['text', 'image'],\n    output: ['text', 'image'],\n    capabilities: [\n      'batch_api',\n      'image_generation',\n      'search_grounding',\n      'structured_output',\n      'thinking',\n    ],\n  },\n  pricing: {\n    input: {\n      normal: 0.25,\n    },\n    output: {\n      normal: 1.5,\n    },\n  },\n} as const satisfies ModelMeta<\n  GeminiToolConfigOptions &\n    GeminiSafetyOptions &\n    GeminiCommonConfigOptions &\n    GeminiCachedContentOptions &\n    GeminiStructuredOutputOptions &\n    GeminiThinkingOptions\n>\n\nconst GEMINI_3_1_FLASH_LITE = {\n  name: 'gemini-3.1-flash-lite-preview',\n  max_input_tokens: 1_048_576,\n  max_output_tokens: 65_536,\n  knowledge_cutoff: '2025-01-01',\n  supports: {\n    input: ['text', 'image', 'audio', 'video', 'document'],\n    output: ['text'],\n    capabilities: [\n      'batch_api',\n      'caching',\n      'code_execution',\n      'file_search',\n      'function_calling',\n      'search_grounding',\n      'structured_output',\n      'thinking',\n      'url_context',\n    ],\n  },\n  pricing: {\n    input: {\n      normal: 0.25,\n    },\n    output: {\n      normal: 1.5,\n    },\n  },\n} as const satisfies ModelMeta<\n  GeminiToolConfigOptions &\n    GeminiSafetyOptions &\n    GeminiCommonConfigOptions &\n    GeminiCachedContentOptions &\n    GeminiStructuredOutputOptions &\n    GeminiThinkingOptions\n>\n\nconst GEMINI_2_5_PRO = {\n  name: 'gemini-2.5-pro',\n  max_input_tokens: 1_048_576,\n  max_output_tokens: 65_536,\n  knowledge_cutoff: '2025-01-01',\n  supports: {\n    input: ['text', 'image', 'audio', 'video', 'document'],\n    output: ['text'],\n    capabilities: [\n      'batch_api',\n      'caching',\n      'code_execution',\n      'file_search',\n      'function_calling',\n      'grounding_with_gmaps',\n      'search_grounding',\n      'structured_output',\n      'thinking',\n      'url_context',\n    ],\n  },\n  pricing: {\n    input: {\n      normal: 2.5,\n    },\n    output: {\n      normal: 15,\n    },\n  },\n} as const satisfies ModelMeta<\n  GeminiToolConfigOptions &\n    GeminiSafetyOptions &\n    GeminiCommonConfigOptions &\n    GeminiCachedContentOptions &\n    GeminiStructuredOutputOptions &\n    GeminiThinkingOptions\n>\n\nconst GEMINI_2_5_PRO_TTS = {\n  name: 'gemini-2.5-pro-preview-tts',\n  max_input_tokens: 8_192,\n  max_output_tokens: 16_384,\n  knowledge_cutoff: '2025-05-01',\n  supports: {\n    input: ['text'],\n    output: ['audio'],\n    capabilities: ['audio_generation', 'file_search'],\n  },\n  pricing: {\n    input: {\n      normal: 2.5,\n    },\n    output: {\n      normal: 15,\n    },\n  },\n} as const satisfies ModelMeta<\n  GeminiToolConfigOptions &\n    GeminiSafetyOptions &\n    GeminiCommonConfigOptions &\n    GeminiCachedContentOptions\n>\n\nconst GEMINI_2_5_FLASH = {\n  name: 'gemini-2.5-flash',\n  max_input_tokens: 1_048_576,\n  max_output_tokens: 65_536,\n  knowledge_cutoff: '2025-01-01',\n  supports: {\n    input: ['text', 'image', 'audio', 'video'],\n    output: ['text'],\n    capabilities: [\n      'batch_api',\n      'caching',\n      'code_execution',\n      'file_search',\n      'function_calling',\n      'grounding_with_gmaps',\n      'search_grounding',\n      'structured_output',\n      'thinking',\n      'url_context',\n    ],\n  },\n  pricing: {\n    input: {\n      normal: 1,\n    },\n    output: {\n      normal: 2.5,\n    },\n  },\n} as const satisfies ModelMeta<\n  GeminiToolConfigOptions &\n    GeminiSafetyOptions &\n    GeminiCommonConfigOptions &\n    GeminiCachedContentOptions &\n    GeminiStructuredOutputOptions &\n    GeminiThinkingOptions\n>\n\nconst GEMINI_2_5_FLASH_PREVIEW = {\n  name: 'gemini-2.5-flash-preview-09-2025',\n  max_input_tokens: 1_048_576,\n  max_output_tokens: 65_536,\n  knowledge_cutoff: '2025-01-01',\n  supports: {\n    input: ['text', 'image', 'audio', 'video'],\n    output: ['text'],\n    capabilities: [\n      'batch_api',\n      'caching',\n      'code_execution',\n      'file_search',\n      'function_calling',\n      'search_grounding',\n      'structured_output',\n      'thinking',\n      'url_context',\n    ],\n  },\n  pricing: {\n    input: {\n      normal: 1,\n    },\n    output: {\n      normal: 2.5,\n    },\n  },\n} as const satisfies ModelMeta<\n  GeminiToolConfigOptions &\n    GeminiSafetyOptions &\n    GeminiCommonConfigOptions &\n    GeminiCachedContentOptions &\n    GeminiStructuredOutputOptions &\n    GeminiThinkingOptions\n>\n\nconst GEMINI_2_5_FLASH_IMAGE = {\n  name: 'gemini-2.5-flash-image',\n  max_input_tokens: 1_048_576,\n  max_output_tokens: 65_536,\n  knowledge_cutoff: '2025-06-01',\n  supports: {\n    input: ['text', 'image'],\n    output: ['text', 'image'],\n    capabilities: [\n      'batch_api',\n      'caching',\n      'file_search',\n      'image_generation',\n      'structured_output',\n    ],\n  },\n  pricing: {\n    input: {\n      normal: 0.3,\n    },\n    output: {\n      normal: 0.4,\n    },\n  },\n} as const satisfies ModelMeta<\n  GeminiToolConfigOptions &\n    GeminiSafetyOptions &\n    GeminiCommonConfigOptions &\n    GeminiCachedContentOptions\n>\n/**\nconst GEMINI_2_5_FLASH_LIVE = {\n  name: 'gemini-2.5-flash-native-audio-preview-09-2025',\n  max_input_tokens: 141_072,\n  max_output_tokens: 8_192,\n  knowledge_cutoff: '2025-01-01',\n  supports: {\n    input: ['text', 'audio', 'video'],\n    output: ['text', 'audio'],\n    capabilities: [\n      'audio_generation',\n      'file_search',\n      'function_calling',\n      'live_api',\n      'search_grounding',\n      'thinking',\n    ],\n  },\n  pricing: {\n    // todo find this info\n    input: {\n      normal: 0,\n    },\n    output: {\n      normal: 0,\n    },\n  },\n} as const satisfies ModelMeta<\n  GeminiToolConfigOptions &\n  GeminiSafetyOptions &\n  GeminiGenerationConfigOptions &\n  GeminiCachedContentOptions &\n  GeminiThinkingOptions\n>\n*/\nconst GEMINI_2_5_FLASH_TTS = {\n  name: 'gemini-2.5-flash-preview-tts',\n  max_input_tokens: 8_192,\n  max_output_tokens: 16_384,\n  knowledge_cutoff: '2025-05-01',\n  supports: {\n    input: ['text'],\n    output: ['audio'],\n    capabilities: ['audio_generation', 'batch_api', 'file_search'],\n  },\n  pricing: {\n    input: {\n      normal: 1,\n    },\n    output: {\n      normal: 2.5,\n    },\n  },\n} as const satisfies ModelMeta<\n  GeminiToolConfigOptions &\n    GeminiSafetyOptions &\n    GeminiCommonConfigOptions &\n    GeminiCachedContentOptions\n>\n\nconst GEMINI_2_5_FLASH_LITE = {\n  name: 'gemini-2.5-flash-lite',\n  max_input_tokens: 1_048_576,\n  max_output_tokens: 65_536,\n  knowledge_cutoff: '2025-01-01',\n  supports: {\n    input: ['text', 'image', 'audio', 'video', 'document'],\n    output: ['text'],\n    capabilities: [\n      'batch_api',\n      'caching',\n      'code_execution',\n      'function_calling',\n      'grounding_with_gmaps',\n      'search_grounding',\n      'structured_output',\n      'thinking',\n      'url_context',\n    ],\n  },\n  pricing: {\n    input: {\n      normal: 0.1,\n    },\n    output: {\n      normal: 0.4,\n    },\n  },\n} as const satisfies ModelMeta<\n  GeminiToolConfigOptions &\n    GeminiSafetyOptions &\n    GeminiCommonConfigOptions &\n    GeminiCachedContentOptions &\n    GeminiStructuredOutputOptions &\n    GeminiThinkingOptions\n>\n\nconst GEMINI_2_5_FLASH_LITE_PREVIEW = {\n  name: 'gemini-2.5-flash-lite-preview-09-2025',\n  max_input_tokens: 1_048_576,\n  max_output_tokens: 65_536,\n  knowledge_cutoff: '2025-01-01',\n  supports: {\n    input: ['text', 'image', 'audio', 'video', 'document'],\n    output: ['text'],\n    capabilities: [\n      'batch_api',\n      'caching',\n      'code_execution',\n      'function_calling',\n      'search_grounding',\n      'structured_output',\n      'thinking',\n      'url_context',\n    ],\n  },\n  pricing: {\n    input: {\n      normal: 0.1,\n    },\n    output: {\n      normal: 0.4,\n    },\n  },\n} as const satisfies ModelMeta<\n  GeminiToolConfigOptions &\n    GeminiSafetyOptions &\n    GeminiCommonConfigOptions &\n    GeminiCachedContentOptions &\n    GeminiStructuredOutputOptions &\n    GeminiThinkingOptions\n>\n\nconst GEMINI_2_FLASH = {\n  name: 'gemini-2.0-flash',\n  max_input_tokens: 1_048_576,\n  max_output_tokens: 8_192,\n  knowledge_cutoff: '2024-08-01',\n  supports: {\n    input: ['text', 'image', 'audio', 'video'],\n    output: ['text'],\n    capabilities: [\n      'batch_api',\n      'caching',\n      'code_execution',\n      'function_calling',\n      'grounding_with_gmaps',\n      'live_api',\n      'search_grounding',\n      'structured_output',\n    ],\n  },\n  pricing: {\n    input: {\n      normal: 0.1,\n    },\n    output: {\n      normal: 0.4,\n    },\n  },\n} as const satisfies ModelMeta<\n  GeminiToolConfigOptions &\n    GeminiSafetyOptions &\n    GeminiCommonConfigOptions &\n    GeminiCachedContentOptions &\n    GeminiStructuredOutputOptions\n>\n\nconst GEMINI_2_FLASH_IMAGE = {\n  name: 'gemini-2.0-flash-preview-image-generation',\n  max_input_tokens: 32_768,\n  max_output_tokens: 8_192,\n  knowledge_cutoff: '2024-08-01',\n  supports: {\n    input: ['text', 'image', 'audio', 'video'],\n    output: ['text'],\n    capabilities: [\n      'batch_api',\n      'caching',\n      'image_generation',\n      'structured_output',\n    ],\n  },\n  pricing: {\n    input: {\n      normal: 0.1,\n    },\n    output: {\n      normal: 0.039,\n    },\n  },\n} as const satisfies ModelMeta<\n  GeminiToolConfigOptions &\n    GeminiSafetyOptions &\n    GeminiCommonConfigOptions &\n    GeminiCachedContentOptions\n>\n/* \nconst GEMINI_2_FLASH_LIVE = {\n  name: 'gemini-2.0-flash-live-001',\n  max_input_tokens: 1_048_576,\n  max_output_tokens: 8_192,\n  knowledge_cutoff: '2024-08-01',\n  supports: {\n    input: ['text', 'audio', 'video'],\n    output: ['text', 'audio'],\n    capabilities: [\n      'audio_generation',\n      'code_execution',\n      'function_calling',\n      'live_api',\n      'search_grounding',\n      'structured_output',\n      'url_context',\n    ],\n  },\n  pricing: {\n    // todo find this info\n    input: {\n      normal: 0,\n    },\n    output: {\n      normal: 0,\n    },\n  },\n} as const satisfies ModelMeta<\n  GeminiToolConfigOptions &\n  GeminiSafetyOptions &\n  GeminiGenerationConfigOptions &\n  GeminiCachedContentOptions\n> */\n\nconst GEMINI_2_FLASH_LITE = {\n  name: 'gemini-2.0-flash-lite',\n  max_input_tokens: 1_048_576,\n  max_output_tokens: 8_192,\n  knowledge_cutoff: '2024-08-01',\n  supports: {\n    input: ['text', 'audio', 'video', 'image'],\n    output: ['text'],\n    capabilities: [\n      'batch_api',\n      'caching',\n      'function_calling',\n      'structured_output',\n    ],\n  },\n  pricing: {\n    input: {\n      normal: 0.075,\n    },\n    output: {\n      normal: 0.3,\n    },\n  },\n} as const satisfies ModelMeta<\n  GeminiToolConfigOptions &\n    GeminiSafetyOptions &\n    GeminiCommonConfigOptions &\n    GeminiCachedContentOptions &\n    GeminiStructuredOutputOptions\n>\n\nconst IMAGEN_4_GENERATE = {\n  name: 'imagen-4.0-generate-001',\n  max_input_tokens: 480,\n  max_output_tokens: 4,\n  supports: {\n    input: ['text'],\n    output: ['image'],\n  },\n  pricing: {\n    input: {\n      normal: 0,\n    },\n    output: {\n      normal: 0.4,\n    },\n  },\n} as const satisfies ModelMeta<\n  GeminiToolConfigOptions &\n    GeminiSafetyOptions &\n    GeminiCommonConfigOptions &\n    GeminiCachedContentOptions\n>\n\nconst IMAGEN_4_GENERATE_ULTRA = {\n  name: 'imagen-4.0-ultra-generate-001',\n  max_input_tokens: 480,\n  max_output_tokens: 4,\n  supports: {\n    input: ['text'],\n    output: ['image'],\n  },\n  pricing: {\n    input: {\n      normal: 0,\n    },\n    output: {\n      normal: 0.6,\n    },\n  },\n} as const satisfies ModelMeta<\n  GeminiToolConfigOptions &\n    GeminiSafetyOptions &\n    GeminiCommonConfigOptions &\n    GeminiCachedContentOptions\n>\n\nconst IMAGEN_4_GENERATE_FAST = {\n  name: 'imagen-4.0-fast-generate-001',\n  max_input_tokens: 480,\n  max_output_tokens: 4,\n  supports: {\n    input: ['text'],\n    output: ['image'],\n  },\n  pricing: {\n    input: {\n      normal: 0,\n    },\n    output: {\n      normal: 0.2,\n    },\n  },\n} as const satisfies ModelMeta<\n  GeminiToolConfigOptions &\n    GeminiSafetyOptions &\n    GeminiCommonConfigOptions &\n    GeminiCachedContentOptions\n>\n\nconst IMAGEN_3 = {\n  name: 'imagen-3.0-generate-002',\n  max_output_tokens: 4,\n  supports: {\n    input: ['text'],\n    output: ['image'],\n  },\n  pricing: {\n    input: {\n      normal: 0,\n    },\n    output: {\n      normal: 0.03,\n    },\n  },\n} as const satisfies ModelMeta<\n  GeminiToolConfigOptions &\n    GeminiSafetyOptions &\n    GeminiCommonConfigOptions &\n    GeminiCachedContentOptions\n>\n/** \nconst VEO_3_1_PREVIEW = {\n  name: 'veo-3.1-generate-preview',\n  max_input_tokens: 1024,\n  max_output_tokens: 1,\n  supports: {\n    input: ['text', 'image'],\n    output: ['video', 'audio'],\n  },\n  pricing: {\n    input: {\n      normal: 0,\n    },\n    output: {\n      normal: 0.4,\n    },\n  },\n} as const satisfies ModelMeta<\n  GeminiToolConfigOptions &\n  GeminiSafetyOptions &\n  GeminiGenerationConfigOptions &\n  GeminiCachedContentOptions\n>\n\nconst VEO_3_1_FAST_PREVIEW = {\n  name: 'veo-3.1-fast-generate-preview',\n  max_input_tokens: 1024,\n  max_output_tokens: 1,\n  supports: {\n    input: ['text', 'image'],\n    output: ['video', 'audio'],\n  },\n  pricing: {\n    input: {\n      normal: 0,\n    },\n    output: {\n      normal: 0.15,\n    },\n  },\n} as const satisfies ModelMeta<\n  GeminiToolConfigOptions &\n  GeminiSafetyOptions &\n  GeminiGenerationConfigOptions &\n  GeminiCachedContentOptions\n>\n\nconst VEO_3 = {\n  name: 'veo-3.0-generate-001',\n  max_input_tokens: 1024,\n  max_output_tokens: 1,\n  supports: {\n    input: ['text', 'image'],\n    output: ['video', 'audio'],\n  },\n  pricing: {\n    input: {\n      normal: 0,\n    },\n    output: {\n      normal: 0.4,\n    },\n  },\n} as const satisfies ModelMeta<\n  GeminiToolConfigOptions &\n  GeminiSafetyOptions &\n  GeminiGenerationConfigOptions &\n  GeminiCachedContentOptions\n>\n\nconst VEO_3_FAST = {\n  name: 'veo-3.0-fast-generate-001',\n  max_input_tokens: 1024,\n  max_output_tokens: 1,\n  supports: {\n    input: ['text', 'image'],\n    output: ['video', 'audio'],\n  },\n  pricing: {\n    input: {\n      normal: 0,\n    },\n    output: {\n      normal: 0.15,\n    },\n  },\n} as const satisfies ModelMeta<\n  GeminiToolConfigOptions &\n  GeminiSafetyOptions &\n  GeminiGenerationConfigOptions &\n  GeminiCachedContentOptions\n>\n\nconst VEO_2 = {\n  name: 'veo-2.0-generate-001',\n  max_output_tokens: 2,\n  supports: {\n    input: ['text', 'image'],\n    output: ['video'],\n  },\n  pricing: {\n    input: {\n      normal: 0,\n    },\n    output: {\n      normal: 0.35,\n    },\n  },\n} as const satisfies ModelMeta<\n  GeminiToolConfigOptions &\n  GeminiSafetyOptions &\n  GeminiGenerationConfigOptions &\n  GeminiCachedContentOptions\n> */\n\n/* const GEMINI_MODEL_META = {\n  [GEMINI_3_PRO.name]: GEMINI_3_PRO,\n  [GEMINI_2_5_PRO.name]: GEMINI_2_5_PRO,\n  [GEMINI_2_5_PRO_TTS.name]: GEMINI_2_5_PRO_TTS,\n  [GEMINI_2_5_FLASH.name]: GEMINI_2_5_FLASH,\n  [GEMINI_2_5_FLASH_PREVIEW.name]: GEMINI_2_5_FLASH_PREVIEW,\n  [GEMINI_2_5_FLASH_IMAGE.name]: GEMINI_2_5_FLASH_IMAGE,\n  [GEMINI_2_5_FLASH_LIVE.name]: GEMINI_2_5_FLASH_LIVE,\n  [GEMINI_2_5_FLASH_TTS.name]: GEMINI_2_5_FLASH_TTS,\n  [GEMINI_2_5_FLASH_LITE.name]: GEMINI_2_5_FLASH_LITE,\n  [GEMINI_2_5_FLASH_LITE_PREVIEW.name]: GEMINI_2_5_FLASH_LITE_PREVIEW,\n  [GEMINI_2_FLASH.name]: GEMINI_2_FLASH,\n  [GEMINI_2_FLASH_IMAGE.name]: GEMINI_2_FLASH_IMAGE,\n  [GEMINI_2_FLASH_LIVE.name]: GEMINI_2_FLASH_LIVE,\n  [GEMINI_2_FLASH_LITE.name]: GEMINI_2_FLASH_LITE,\n  [IMAGEN_4_GENERATE.name]: IMAGEN_4_GENERATE,\n  [IMAGEN_4_GENERATE_ULTRA.name]: IMAGEN_4_GENERATE_ULTRA,\n  [IMAGEN_4_GENERATE_FAST.name]: IMAGEN_4_GENERATE_FAST,\n  [IMAGEN_3.name]: IMAGEN_3,\n  [VEO_3_1_PREVIEW.name]: VEO_3_1_PREVIEW,\n  [VEO_3_1_FAST_PREVIEW.name]: VEO_3_1_FAST_PREVIEW,\n  [VEO_3.name]: VEO_3,\n  [VEO_3_FAST.name]: VEO_3_FAST,\n  [VEO_2.name]: VEO_2,\n} as const */\n\nexport const GEMINI_MODELS = [\n  GEMINI_3_1_PRO.name,\n  GEMINI_3_PRO.name,\n  GEMINI_3_FLASH.name,\n  GEMINI_3_1_FLASH_LITE.name,\n  GEMINI_2_5_PRO.name,\n  GEMINI_2_5_FLASH.name,\n  GEMINI_2_5_FLASH_PREVIEW.name,\n  GEMINI_2_5_FLASH_LITE.name,\n  GEMINI_2_5_FLASH_LITE_PREVIEW.name,\n  GEMINI_2_FLASH.name,\n  GEMINI_2_FLASH_LITE.name,\n] as const\n\nexport type GeminiModels = (typeof GEMINI_MODELS)[number]\n\nexport type GeminiImageModels = (typeof GEMINI_IMAGE_MODELS)[number]\n\nexport const GEMINI_IMAGE_MODELS = [\n  GEMINI_3_1_FLASH_IMAGE.name,\n  GEMINI_3_PRO_IMAGE.name,\n  GEMINI_2_5_FLASH_IMAGE.name,\n  GEMINI_2_FLASH_IMAGE.name,\n  IMAGEN_3.name,\n  IMAGEN_4_GENERATE.name,\n  IMAGEN_4_GENERATE_FAST.name,\n  IMAGEN_4_GENERATE_ULTRA.name,\n] as const\n\n/**\n * Text-to-speech models\n * @experimental Gemini TTS is an experimental feature and may change.\n */\nexport const GEMINI_TTS_MODELS = [\n  GEMINI_2_5_FLASH_TTS.name,\n  GEMINI_2_5_PRO_TTS.name,\n] as const\n\n/**\n * Available voice names for Gemini TTS\n * @see https://ai.google.dev/gemini-api/docs/speech-generation\n */\nexport const GEMINI_TTS_VOICES = [\n  'Zephyr',\n  'Puck',\n  'Charon',\n  'Kore',\n  'Fenrir',\n  'Leda',\n  'Orus',\n  'Aoede',\n  'Callirrhoe',\n  'Autonoe',\n  'Enceladus',\n  'Iapetus',\n  'Umbriel',\n  'Algieba',\n  'Despina',\n  'Erinome',\n  'Algenib',\n  'Rasalgethi',\n  'Laomedeia',\n  'Achernar',\n  'Alnilam',\n  'Schedar',\n  'Gacrux',\n  'Pulcherrima',\n  'Achird',\n  'Zubenelgenubi',\n  'Vindemiatrix',\n  'Sadachbia',\n  'Sadaltager',\n  'Sulafat',\n] as const\n\nexport type GeminiTTSVoice = (typeof GEMINI_TTS_VOICES)[number]\n\n/*   const GEMINI_AUDIO_MODELS = [\n  GEMINI_2_5_PRO_TTS.name,\n  GEMINI_2_5_FLASH_TTS.name,\n  GEMINI_2_5_FLASH_LIVE.name,\n  GEMINI_2_FLASH_LIVE.name,\n] as const\n\n  const GEMINI_VIDEO_MODELS = [\n  VEO_3_1_PREVIEW.name,\n  VEO_3_1_FAST_PREVIEW.name,\n  VEO_3.name,\n  VEO_3_FAST.name,\n  VEO_2.name,\n] as const */\n\n// Manual type map for per-model provider options\nexport type GeminiChatModelProviderOptionsByName = {\n  // Models with thinking and structured output support\n  [GEMINI_3_1_PRO.name]: GeminiToolConfigOptions &\n    GeminiSafetyOptions &\n    GeminiCommonConfigOptions &\n    GeminiCachedContentOptions &\n    GeminiStructuredOutputOptions &\n    GeminiThinkingOptions &\n    GeminiThinkingAdvancedOptions\n  [GEMINI_3_PRO.name]: GeminiToolConfigOptions &\n    GeminiSafetyOptions &\n    GeminiCommonConfigOptions &\n    GeminiCachedContentOptions &\n    GeminiStructuredOutputOptions &\n    GeminiThinkingOptions &\n    GeminiThinkingAdvancedOptions\n  [GEMINI_3_FLASH.name]: GeminiToolConfigOptions &\n    GeminiSafetyOptions &\n    GeminiCommonConfigOptions &\n    GeminiCachedContentOptions &\n    GeminiStructuredOutputOptions &\n    GeminiThinkingOptions &\n    GeminiThinkingAdvancedOptions\n  [GEMINI_3_1_FLASH_LITE.name]: GeminiToolConfigOptions &\n    GeminiSafetyOptions &\n    GeminiCommonConfigOptions &\n    GeminiCachedContentOptions &\n    GeminiStructuredOutputOptions &\n    GeminiThinkingOptions\n  [GEMINI_2_5_PRO.name]: GeminiToolConfigOptions &\n    GeminiSafetyOptions &\n    GeminiCommonConfigOptions &\n    GeminiCachedContentOptions &\n    GeminiStructuredOutputOptions &\n    GeminiThinkingOptions\n  [GEMINI_2_5_FLASH.name]: GeminiToolConfigOptions &\n    GeminiSafetyOptions &\n    GeminiCommonConfigOptions &\n    GeminiCachedContentOptions &\n    GeminiStructuredOutputOptions &\n    GeminiThinkingOptions\n  [GEMINI_2_5_FLASH_PREVIEW.name]: GeminiToolConfigOptions &\n    GeminiSafetyOptions &\n    GeminiCommonConfigOptions &\n    GeminiCachedContentOptions &\n    GeminiStructuredOutputOptions &\n    GeminiThinkingOptions\n  [GEMINI_2_5_FLASH_LITE.name]: GeminiToolConfigOptions &\n    GeminiSafetyOptions &\n    GeminiCommonConfigOptions &\n    GeminiCachedContentOptions &\n    GeminiStructuredOutputOptions &\n    GeminiThinkingOptions\n  [GEMINI_2_5_FLASH_LITE_PREVIEW.name]: GeminiToolConfigOptions &\n    GeminiSafetyOptions &\n    GeminiCommonConfigOptions &\n    GeminiCachedContentOptions &\n    GeminiStructuredOutputOptions &\n    GeminiThinkingOptions\n  // Models with structured output but no thinking support\n  [GEMINI_2_FLASH.name]: GeminiToolConfigOptions &\n    GeminiSafetyOptions &\n    GeminiCommonConfigOptions &\n    GeminiCachedContentOptions &\n    GeminiStructuredOutputOptions\n  [GEMINI_2_FLASH_LITE.name]: GeminiToolConfigOptions &\n    GeminiSafetyOptions &\n    GeminiCommonConfigOptions &\n    GeminiCachedContentOptions &\n    GeminiStructuredOutputOptions\n}\n\n/**\n * Type-only map from chat model name to its supported input modalities.\n * Based on the 'supports.input' arrays defined for each model.\n * Note: 'document' in the model meta is mapped to 'document' modality.\n * Used by the core AI types to constrain ContentPart types based on the selected model.\n * Note: These must be inlined as readonly arrays (not typeof) because the model\n * constants are not exported and typeof references don't work in .d.ts files\n * when consumed by external packages.\n *\n * @see https://ai.google.dev/gemini-api/docs/vision\n * @see https://ai.google.dev/gemini-api/docs/audio\n * @see https://ai.google.dev/gemini-api/docs/document-processing\n */\nexport type GeminiModelInputModalitiesByName = {\n  // Models with full multimodal support (text, image, audio, video, document)\n  [GEMINI_3_1_PRO.name]: typeof GEMINI_3_1_PRO.supports.input\n  [GEMINI_3_PRO.name]: typeof GEMINI_3_PRO.supports.input\n  [GEMINI_3_FLASH.name]: typeof GEMINI_3_FLASH.supports.input\n  [GEMINI_3_1_FLASH_LITE.name]: typeof GEMINI_3_1_FLASH_LITE.supports.input\n  [GEMINI_2_5_PRO.name]: typeof GEMINI_2_5_PRO.supports.input\n  [GEMINI_2_5_FLASH_LITE.name]: typeof GEMINI_2_5_FLASH_LITE.supports.input\n  [GEMINI_2_5_FLASH_LITE_PREVIEW.name]: typeof GEMINI_2_5_FLASH_LITE_PREVIEW.supports.input\n\n  // Models with text, image, audio, video (no document)\n  [GEMINI_2_5_FLASH.name]: typeof GEMINI_2_5_FLASH.supports.input\n  [GEMINI_2_5_FLASH_PREVIEW.name]: typeof GEMINI_2_5_FLASH_PREVIEW.supports.input\n  [GEMINI_2_FLASH.name]: typeof GEMINI_2_FLASH.supports.input\n  [GEMINI_2_FLASH_LITE.name]: typeof GEMINI_2_FLASH_LITE.supports.input\n}\n"],"names":[],"mappings":"AAiDA,MAAM,iBAAiB;AAAA,EACrB,MAAM;AA2BR;AAUA,MAAM,eAAe;AAAA,EACnB,MAAM;AA2BR;AAUA,MAAM,iBAAiB;AAAA,EACrB,MAAM;AA2BR;AAUA,MAAM,qBAAqB;AAAA,EACzB,MAAM;AAuBR;AAUA,MAAM,yBAAyB;AAAA,EAC7B,MAAM;AAuBR;AASA,MAAM,wBAAwB;AAAA,EAC5B,MAAM;AA2BR;AASA,MAAM,iBAAiB;AAAA,EACrB,MAAM;AA4BR;AASA,MAAM,qBAAqB;AAAA,EACzB,MAAM;AAiBR;AAOA,MAAM,mBAAmB;AAAA,EACvB,MAAM;AA4BR;AASA,MAAM,2BAA2B;AAAA,EAC/B,MAAM;AA2BR;AASA,MAAM,yBAAyB;AAAA,EAC7B,MAAM;AAuBR;AAyCA,MAAM,uBAAuB;AAAA,EAC3B,MAAM;AAiBR;AAOA,MAAM,wBAAwB;AAAA,EAC5B,MAAM;AA2BR;AASA,MAAM,gCAAgC;AAAA,EACpC,MAAM;AA0BR;AASA,MAAM,iBAAiB;AAAA,EACrB,MAAM;AA0BR;AAQA,MAAM,uBAAuB;AAAA,EAC3B,MAAM;AAsBR;AAyCA,MAAM,sBAAsB;AAAA,EAC1B,MAAM;AAsBR;AAQA,MAAM,oBAAoB;AAAA,EACxB,MAAM;AAeR;AAOA,MAAM,0BAA0B;AAAA,EAC9B,MAAM;AAeR;AAOA,MAAM,yBAAyB;AAAA,EAC7B,MAAM;AAeR;AAOA,MAAM,WAAW;AAAA,EACf,MAAM;AAcR;AAmJO,MAAM,gBAAgB;AAAA,EAC3B,eAAe;AAAA,EACf,aAAa;AAAA,EACb,eAAe;AAAA,EACf,sBAAsB;AAAA,EACtB,eAAe;AAAA,EACf,iBAAiB;AAAA,EACjB,yBAAyB;AAAA,EACzB,sBAAsB;AAAA,EACtB,8BAA8B;AAAA,EAC9B,eAAe;AAAA,EACf,oBAAoB;AACtB;AAMO,MAAM,sBAAsB;AAAA,EACjC,uBAAuB;AAAA,EACvB,mBAAmB;AAAA,EACnB,uBAAuB;AAAA,EACvB,qBAAqB;AAAA,EACrB,SAAS;AAAA,EACT,kBAAkB;AAAA,EAClB,uBAAuB;AAAA,EACvB,wBAAwB;AAC1B;AAMO,MAAM,oBAAoB;AAAA,EAC/B,qBAAqB;AAAA,EACrB,mBAAmB;AACrB;AAMO,MAAM,oBAAoB;AAAA,EAC/B;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACF;"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@tanstack/ai-gemini",
-  "version": "0.6.0",
+  "version": "0.8.0",
   "description": "Google Gemini adapter for TanStack AI",
   "author": "",
   "license": "MIT",
@@ -30,15 +30,15 @@
     "adapter"
   ],
   "dependencies": {
-    "@google/genai": "^1.30.0"
+    "@google/genai": "^1.43.0"
   },
   "peerDependencies": {
-    "@tanstack/ai": "^0.6.1"
+    "@tanstack/ai": "^0.6.2"
   },
   "devDependencies": {
     "@vitest/coverage-v8": "4.0.14",
     "vite": "^7.2.7",
-    "@tanstack/ai": "0.6.1"
+    "@tanstack/ai": "0.6.2"
   },
   "scripts": {
     "build": "vite build",

package/src/adapters/image.ts

@@ -5,6 +5,7 @@ import {
   getGeminiApiKeyFromEnv,
 } from '../utils'
 import {
+  parseNativeImageSize,
   sizeToAspectRatio,
   validateImageSize,
   validateNumberOfImages,
@@ -22,6 +23,8 @@ import type {
   ImageGenerationResult,
 } from '@tanstack/ai'
 import type {
+  GenerateContentConfig,
+  GenerateContentResponse,
   GenerateImagesConfig,
   GenerateImagesResponse,
   GoogleGenAI,
@@ -39,14 +42,16 @@ export type GeminiImageModel = (typeof GEMINI_IMAGE_MODELS)[number]
 /**
  * Gemini Image Generation Adapter
  *
- * Tree-shakeable adapter for Gemini Imagen image generation functionality.
- * Supports Imagen 3 and Imagen 4 models.
+ * Tree-shakeable adapter for Gemini image generation functionality.
+ * Supports Imagen 3/4 models (via generateImages API) and Gemini native
+ * image models like Nano Banana 2 (via generateContent API).
  *
  * Features:
  * - Aspect ratio-based image sizing
  * - Person generation controls
  * - Safety filtering
  * - Watermark options
+ * - Extended resolution tiers (Nano Banana 2)
  */
 export class GeminiImageAdapter<
   TModel extends GeminiImageModel,
@@ -76,15 +81,19 @@ export class GeminiImageAdapter<
   async generateImages(
     options: ImageGenerationOptions<GeminiImageProviderOptions>,
   ): Promise<ImageGenerationResult> {
-    const { model, prompt, numberOfImages, size } = options
+    const { model, prompt } = options
 
-    // Validate inputs
     validatePrompt({ prompt, model })
-    validateImageSize(model, size)
-    validateNumberOfImages(model, numberOfImages)
 
-    // Build request config
-    const config = this.buildConfig(options)
+    if (this.isGeminiImageModel(model)) {
+      return this.generateWithGeminiApi(options)
+    }
+
+    // Imagen models path (generateImages API)
+    validateImageSize(model, options.size)
+    validateNumberOfImages(model, options.numberOfImages)
+
+    const config = this.buildImagenConfig(options)
 
     const response = await this.client.models.generateImages({
       model,
@@ -92,10 +101,78 @@ export class GeminiImageAdapter<
       config,
     })
 
-    return this.transformResponse(model, response)
+    return this.transformImagenResponse(model, response)
+  }
+
+  private isGeminiImageModel(model: string): boolean {
+    return model.startsWith('gemini-')
+  }
+
+  private async generateWithGeminiApi(
+    options: ImageGenerationOptions<GeminiImageProviderOptions>,
+  ): Promise<ImageGenerationResult> {
+    const { model, prompt, size, numberOfImages, modelOptions } = options
+
+    const parsedSize = size ? parseNativeImageSize(size) : undefined
+
+    // The generateContent API has no numberOfImages parameter.
+    // Instead, augment the prompt to request multiple images when needed.
+    const augmentedPrompt =
+      numberOfImages && numberOfImages > 1
+        ? `${prompt} Generate ${numberOfImages} distinct images.`
+        : prompt
+
+    const config: GenerateContentConfig = {
+      // Include TEXT so the model can interleave descriptions between images
+      responseModalities: ['TEXT', 'IMAGE'],
+      ...(parsedSize && {
+        imageConfig: {
+          ...(parsedSize.aspectRatio && {
+            aspectRatio: parsedSize.aspectRatio,
+          }),
+          ...(parsedSize.resolution && {
+            imageSize: parsedSize.resolution,
+          }),
+        },
+      }),
+      ...modelOptions,
+    }
+
+    const response = await this.client.models.generateContent({
+      model,
+      contents: augmentedPrompt,
+      config,
+    })
+
+    return this.transformGeminiResponse(model, response)
+  }
+
+  private transformGeminiResponse(
+    model: string,
+    response: GenerateContentResponse,
+  ): ImageGenerationResult {
+    const images: Array<GeneratedImage> = []
+    const parts = response.candidates?.[0]?.content?.parts ?? []
+
+    for (const part of parts) {
+      if (
+        part.inlineData?.data &&
+        typeof part.inlineData.data === 'string' &&
+        part.inlineData.data.length > 0
+      ) {
+        images.push({ b64Json: part.inlineData.data })
+      }
+    }
+
+    return {
+      id: generateId(this.name),
+      model,
+      images,
+      usage: undefined,
+    }
   }
 
-  private buildConfig(
+  private buildImagenConfig(
     options: ImageGenerationOptions<GeminiImageProviderOptions>,
   ): GenerateImagesConfig {
     const { size, numberOfImages, modelOptions } = options
@@ -108,7 +185,7 @@ export class GeminiImageAdapter<
     }
   }
 
-  private transformResponse(
+  private transformImagenResponse(
     model: string,
     response: GenerateImagesResponse,
   ): ImageGenerationResult {

package/src/adapters/summarize.ts

@@ -21,6 +21,7 @@ export interface GeminiSummarizeConfig extends GeminiClientConfig {}
  * Available Gemini models for summarization
  */
 export const GeminiSummarizeModels = [
+  'gemini-3.1-flash-lite-preview',
   'gemini-2.0-flash',
   'gemini-1.5-flash',
   'gemini-1.5-pro',

package/src/image/image-provider-options.ts

@@ -145,11 +145,49 @@ export type GeminiImageSize =
   | '1080x1920'
 
 /**
- * Model-specific size options mapping
- * All Imagen models use the same size options
+ * Aspect ratios supported by Gemini native image models (via generateContent API).
+ * Matches the SDK's ImageConfig.aspectRatio values.
+ */
+export type GeminiNativeImageAspectRatio =
+  | '1:1'
+  | '2:3'
+  | '3:2'
+  | '3:4'
+  | '4:3'
+  | '9:16'
+  | '16:9'
+  | '21:9'
+
+/**
+ * Resolution tiers for Gemini native image models.
+ * Matches the SDK's ImageConfig.imageSize values.
+ */
+export type GeminiNativeImageResolution = '1K' | '2K' | '4K'
+
+/**
+ * Template literal size type for Gemini native image models: "16:9_4K", "1:1_2K", etc.
+ */
+export type GeminiNativeImageSize =
+  `${GeminiNativeImageAspectRatio}_${GeminiNativeImageResolution}`
+
+/**
+ * Gemini native image models that use the generateContent API path.
+ * These models support template literal sizes (aspectRatio_resolution).
+ */
+export type GeminiNativeImageModels =
+  | 'gemini-3.1-flash-image-preview'
+  | 'gemini-3-pro-image-preview'
+  | 'gemini-2.5-flash-image'
+  | 'gemini-2.0-flash-preview-image-generation'
+
+/**
+ * Model-specific size options mapping.
+ * Gemini native image models use template literal sizes, Imagen models use pixel sizes.
  */
 export type GeminiImageModelSizeByName = {
-  [K in GeminiImageModels]: GeminiImageSize
+  [K in GeminiNativeImageModels]: GeminiNativeImageSize
+} & {
+  [K in Exclude<GeminiImageModels, GeminiNativeImageModels>]: GeminiImageSize
 }
 
 /**
@@ -237,3 +275,15 @@ export function validatePrompt(options: {
     throw new Error(`Prompt cannot be empty for model "${model}".`)
   }
 }
+
+/**
+ * Parses a Gemini native image size string into its components.
+ * Format: "aspectRatio_resolution" e.g. "16:9_4K" → { aspectRatio: "16:9", resolution: "4K" }
+ */
+export function parseNativeImageSize(
+  size: string,
+): { aspectRatio: string; resolution: string } | undefined {
+  const match = size.match(/^(\d+:\d+)_(.+)$/)
+  if (!match) return undefined
+  return { aspectRatio: match[1]!, resolution: match[2]! }
+}

package/src/model-meta.ts

@@ -47,6 +47,44 @@ interface ModelMeta<TProviderOptions = unknown> {
   providerOptions?: TProviderOptions
 }
 
+const GEMINI_3_1_PRO = {
+  name: 'gemini-3.1-pro-preview',
+  max_input_tokens: 1_048_576,
+  max_output_tokens: 65_536,
+  knowledge_cutoff: '2025-01-01',
+  supports: {
+    input: ['text', 'image', 'audio', 'video', 'document'],
+    output: ['text'],
+    capabilities: [
+      'batch_api',
+      'caching',
+      'code_execution',
+      'file_search',
+      'function_calling',
+      'search_grounding',
+      'structured_output',
+      'thinking',
+      'url_context',
+    ],
+  },
+  pricing: {
+    input: {
+      normal: 2.5,
+    },
+    output: {
+      normal: 15,
+    },
+  },
+} as const satisfies ModelMeta<
+  GeminiToolConfigOptions &
+    GeminiSafetyOptions &
+    GeminiCommonConfigOptions &
+    GeminiCachedContentOptions &
+    GeminiStructuredOutputOptions &
+    GeminiThinkingOptions &
+    GeminiThinkingAdvancedOptions
+>
+
 const GEMINI_3_PRO = {
   name: 'gemini-3-pro-preview',
   max_input_tokens: 1_048_576,
@@ -157,6 +195,76 @@ const GEMINI_3_PRO_IMAGE = {
     GeminiThinkingAdvancedOptions
 >
 
+const GEMINI_3_1_FLASH_IMAGE = {
+  name: 'gemini-3.1-flash-image-preview',
+  max_input_tokens: 65_536,
+  max_output_tokens: 65_536,
+  knowledge_cutoff: '2025-01-01',
+  supports: {
+    input: ['text', 'image'],
+    output: ['text', 'image'],
+    capabilities: [
+      'batch_api',
+      'image_generation',
+      'search_grounding',
+      'structured_output',
+      'thinking',
+    ],
+  },
+  pricing: {
+    input: {
+      normal: 0.25,
+    },
+    output: {
+      normal: 1.5,
+    },
+  },
+} as const satisfies ModelMeta<
+  GeminiToolConfigOptions &
+    GeminiSafetyOptions &
+    GeminiCommonConfigOptions &
+    GeminiCachedContentOptions &
+    GeminiStructuredOutputOptions &
+    GeminiThinkingOptions
+>
+
+const GEMINI_3_1_FLASH_LITE = {
+  name: 'gemini-3.1-flash-lite-preview',
+  max_input_tokens: 1_048_576,
+  max_output_tokens: 65_536,
+  knowledge_cutoff: '2025-01-01',
+  supports: {
+    input: ['text', 'image', 'audio', 'video', 'document'],
+    output: ['text'],
+    capabilities: [
+      'batch_api',
+      'caching',
+      'code_execution',
+      'file_search',
+      'function_calling',
+      'search_grounding',
+      'structured_output',
+      'thinking',
+      'url_context',
+    ],
+  },
+  pricing: {
+    input: {
+      normal: 0.25,
+    },
+    output: {
+      normal: 1.5,
+    },
+  },
+} as const satisfies ModelMeta<
+  GeminiToolConfigOptions &
+    GeminiSafetyOptions &
+    GeminiCommonConfigOptions &
+    GeminiCachedContentOptions &
+    GeminiStructuredOutputOptions &
+    GeminiThinkingOptions
+>
+
 const GEMINI_2_5_PRO = {
   name: 'gemini-2.5-pro',
   max_input_tokens: 1_048_576,
@@ -820,8 +928,10 @@ const VEO_2 = {
 } as const */
 
 export const GEMINI_MODELS = [
+  GEMINI_3_1_PRO.name,
   GEMINI_3_PRO.name,
   GEMINI_3_FLASH.name,
+  GEMINI_3_1_FLASH_LITE.name,
   GEMINI_2_5_PRO.name,
   GEMINI_2_5_FLASH.name,
   GEMINI_2_5_FLASH_PREVIEW.name,
@@ -836,6 +946,7 @@ export type GeminiModels = (typeof GEMINI_MODELS)[number]
 export type GeminiImageModels = (typeof GEMINI_IMAGE_MODELS)[number]
 
 export const GEMINI_IMAGE_MODELS = [
+  GEMINI_3_1_FLASH_IMAGE.name,
   GEMINI_3_PRO_IMAGE.name,
   GEMINI_2_5_FLASH_IMAGE.name,
   GEMINI_2_FLASH_IMAGE.name,
@@ -911,6 +1022,13 @@ export type GeminiTTSVoice = (typeof GEMINI_TTS_VOICES)[number]
 // Manual type map for per-model provider options
 export type GeminiChatModelProviderOptionsByName = {
   // Models with thinking and structured output support
+  [GEMINI_3_1_PRO.name]: GeminiToolConfigOptions &
+    GeminiSafetyOptions &
+    GeminiCommonConfigOptions &
+    GeminiCachedContentOptions &
+    GeminiStructuredOutputOptions &
+    GeminiThinkingOptions &
+    GeminiThinkingAdvancedOptions
   [GEMINI_3_PRO.name]: GeminiToolConfigOptions &
     GeminiSafetyOptions &
     GeminiCommonConfigOptions &
@@ -925,6 +1043,12 @@ export type GeminiChatModelProviderOptionsByName = {
     GeminiStructuredOutputOptions &
     GeminiThinkingOptions &
     GeminiThinkingAdvancedOptions
+  [GEMINI_3_1_FLASH_LITE.name]: GeminiToolConfigOptions &
+    GeminiSafetyOptions &
+    GeminiCommonConfigOptions &
+    GeminiCachedContentOptions &
+    GeminiStructuredOutputOptions &
+    GeminiThinkingOptions
   [GEMINI_2_5_PRO.name]: GeminiToolConfigOptions &
     GeminiSafetyOptions &
     GeminiCommonConfigOptions &
@@ -983,8 +1107,10 @@ export type GeminiChatModelProviderOptionsByName = {
  */
 export type GeminiModelInputModalitiesByName = {
   // Models with full multimodal support (text, image, audio, video, document)
+  [GEMINI_3_1_PRO.name]: typeof GEMINI_3_1_PRO.supports.input
   [GEMINI_3_PRO.name]: typeof GEMINI_3_PRO.supports.input
   [GEMINI_3_FLASH.name]: typeof GEMINI_3_FLASH.supports.input
+  [GEMINI_3_1_FLASH_LITE.name]: typeof GEMINI_3_1_FLASH_LITE.supports.input
   [GEMINI_2_5_PRO.name]: typeof GEMINI_2_5_PRO.supports.input
   [GEMINI_2_5_FLASH_LITE.name]: typeof GEMINI_2_5_FLASH_LITE.supports.input
   [GEMINI_2_5_FLASH_LITE_PREVIEW.name]: typeof GEMINI_2_5_FLASH_LITE_PREVIEW.supports.input