@tanstack/ai-gemini 0.8.9 → 0.9.1

package/dist/esm/adapters/image.js

@@ -9,20 +9,35 @@ class GeminiImageAdapter extends BaseImageAdapter {
     this.client = createGeminiClient(config);
   }
   async generateImages(options) {
-    const { model, prompt } = options;
-    validatePrompt({ prompt, model });
-    if (this.isGeminiImageModel(model)) {
-      return this.generateWithGeminiApi(options);
+    const { model, prompt, logger } = options;
+    logger.request(
+      `activity=generateImage provider=gemini model=${this.model}`,
+      {
+        provider: "gemini",
+        model: this.model
+      }
+    );
+    try {
+      validatePrompt({ prompt, model });
+      if (this.isGeminiImageModel(model)) {
+        return await 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.transformImagenResponse(model, response);
+    } catch (error) {
+      logger.errors("gemini.generateImage fatal", {
+        error,
+        source: "gemini.generateImage"
+      });
+      throw error;
     }
-    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.transformImagenResponse(model, response);
   }
   isGeminiImageModel(model) {
     return model.startsWith("gemini-");

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  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;"}
+{"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, logger } = options\n\n    logger.request(\n      `activity=generateImage provider=gemini model=${this.model}`,\n      {\n        provider: 'gemini',\n        model: this.model,\n      },\n    )\n\n    try {\n      validatePrompt({ prompt, model })\n\n      if (this.isGeminiImageModel(model)) {\n        return await 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    } catch (error) {\n      logger.errors('gemini.generateImage fatal', {\n        error,\n        source: 'gemini.generateImage',\n      })\n      throw error\n    }\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,QAAQ,OAAA,IAAW;AAElC,WAAO;AAAA,MACL,gDAAgD,KAAK,KAAK;AAAA,MAC1D;AAAA,QACE,UAAU;AAAA,QACV,OAAO,KAAK;AAAA,MAAA;AAAA,IACd;AAGF,QAAI;AACF,qBAAe,EAAE,QAAQ,OAAO;AAEhC,UAAI,KAAK,mBAAmB,KAAK,GAAG;AAClC,eAAO,MAAM,KAAK,sBAAsB,OAAO;AAAA,MACjD;AAGA,wBAAkB,OAAO,QAAQ,IAAI;AACrC,6BAAuB,OAAO,QAAQ,cAAc;AAEpD,YAAM,SAAS,KAAK,kBAAkB,OAAO;AAE7C,YAAM,WAAW,MAAM,KAAK,OAAO,OAAO,eAAe;AAAA,QACvD;AAAA,QACA;AAAA,QACA;AAAA,MAAA,CACD;AAED,aAAO,KAAK,wBAAwB,OAAO,QAAQ;AAAA,IACrD,SAAS,OAAO;AACd,aAAO,OAAO,8BAA8B;AAAA,QAC1C;AAAA,QACA,QAAQ;AAAA,MAAA,CACT;AACD,YAAM;AAAA,IACR;AAAA,EACF;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.js

@@ -16,41 +16,55 @@ class GeminiSummarizeAdapter {
     this.model = model;
   }
   async summarize(options) {
+    const { logger } = options;
     const model = options.model;
+    logger.request(`activity=summarize provider=gemini`, {
+      provider: "gemini",
+      model
+    });
     const formatInstructions = this.getFormatInstructions(options.style);
     const lengthInstructions = options.maxLength ? ` Keep the summary under ${options.maxLength} tokens.` : "";
     const systemPrompt = `You are a helpful assistant that summarizes text. ${formatInstructions}${lengthInstructions}`;
-    const response = await this.client.models.generateContent({
-      model,
-      contents: [
-        {
-          role: "user",
-          parts: [
-            { text: `Please summarize the following:
+    try {
+      const response = await this.client.models.generateContent({
+        model,
+        contents: [
+          {
+            role: "user",
+            parts: [
+              { text: `Please summarize the following:
 
 ${options.text}` }
-          ]
+            ]
+          }
+        ],
+        config: {
+          systemInstruction: systemPrompt
         }
-      ],
-      config: {
-        systemInstruction: systemPrompt
-      }
-    });
-    const summary = response.text ?? "";
-    const inputTokens = response.usageMetadata?.promptTokenCount ?? 0;
-    const outputTokens = response.usageMetadata?.candidatesTokenCount ?? 0;
-    return {
-      id: generateId("sum"),
-      model,
-      summary,
-      usage: {
-        promptTokens: inputTokens,
-        completionTokens: outputTokens,
-        totalTokens: inputTokens + outputTokens
-      }
-    };
+      });
+      const summary = response.text ?? "";
+      const inputTokens = response.usageMetadata?.promptTokenCount ?? 0;
+      const outputTokens = response.usageMetadata?.candidatesTokenCount ?? 0;
+      return {
+        id: generateId("sum"),
+        model,
+        summary,
+        usage: {
+          promptTokens: inputTokens,
+          completionTokens: outputTokens,
+          totalTokens: inputTokens + outputTokens
+        }
+      };
+    } catch (error) {
+      logger.errors("gemini.summarize fatal", {
+        error,
+        source: "gemini.summarize"
+      });
+      throw error;
+    }
   }
   async *summarizeStream(options) {
+    const { logger } = options;
     const model = options.model;
     const id = generateId("sum");
     let accumulatedContent = "";
@@ -59,57 +73,71 @@ ${options.text}` }
     const formatInstructions = this.getFormatInstructions(options.style);
     const lengthInstructions = options.maxLength ? ` Keep the summary under ${options.maxLength} words.` : "";
     const systemPrompt = `You are a helpful assistant that summarizes text. ${formatInstructions}${lengthInstructions}`;
-    const result = await this.client.models.generateContentStream({
+    logger.request(`activity=summarize provider=gemini`, {
+      provider: "gemini",
       model,
-      contents: [
-        {
-          role: "user",
-          parts: [
-            { text: `Please summarize the following:
+      stream: true
+    });
+    try {
+      const result = await this.client.models.generateContentStream({
+        model,
+        contents: [
+          {
+            role: "user",
+            parts: [
+              { text: `Please summarize the following:
 
 ${options.text}` }
-          ]
-        }
-      ],
-      config: {
-        systemInstruction: systemPrompt
-      }
-    });
-    for await (const chunk of result) {
-      if (chunk.usageMetadata) {
-        inputTokens = chunk.usageMetadata.promptTokenCount ?? inputTokens;
-        outputTokens = chunk.usageMetadata.candidatesTokenCount ?? outputTokens;
-      }
-      if (chunk.candidates?.[0]?.content?.parts) {
-        for (const part of chunk.candidates[0].content.parts) {
-          if (part.text) {
-            accumulatedContent += part.text;
-            yield asChunk({
-              type: "TEXT_MESSAGE_CONTENT",
-              messageId: id,
-              model,
-              timestamp: Date.now(),
-              delta: part.text,
-              content: accumulatedContent
-            });
+            ]
           }
+        ],
+        config: {
+          systemInstruction: systemPrompt
         }
-      }
-      const finishReason = chunk.candidates?.[0]?.finishReason;
-      if (finishReason === FinishReason.STOP || finishReason === FinishReason.MAX_TOKENS || finishReason === FinishReason.SAFETY) {
-        yield asChunk({
-          type: "RUN_FINISHED",
-          runId: id,
-          model,
-          timestamp: Date.now(),
-          finishReason: finishReason === FinishReason.STOP ? "stop" : finishReason === FinishReason.MAX_TOKENS ? "length" : "content_filter",
-          usage: {
-            promptTokens: inputTokens,
-            completionTokens: outputTokens,
-            totalTokens: inputTokens + outputTokens
+      });
+      for await (const chunk of result) {
+        logger.provider(`provider=gemini`, { chunk });
+        if (chunk.usageMetadata) {
+          inputTokens = chunk.usageMetadata.promptTokenCount ?? inputTokens;
+          outputTokens = chunk.usageMetadata.candidatesTokenCount ?? outputTokens;
+        }
+        if (chunk.candidates?.[0]?.content?.parts) {
+          for (const part of chunk.candidates[0].content.parts) {
+            if (part.text) {
+              accumulatedContent += part.text;
+              yield asChunk({
+                type: "TEXT_MESSAGE_CONTENT",
+                messageId: id,
+                model,
+                timestamp: Date.now(),
+                delta: part.text,
+                content: accumulatedContent
+              });
+            }
           }
-        });
+        }
+        const finishReason = chunk.candidates?.[0]?.finishReason;
+        if (finishReason === FinishReason.STOP || finishReason === FinishReason.MAX_TOKENS || finishReason === FinishReason.SAFETY) {
+          yield asChunk({
+            type: "RUN_FINISHED",
+            runId: id,
+            model,
+            timestamp: Date.now(),
+            finishReason: finishReason === FinishReason.STOP ? "stop" : finishReason === FinishReason.MAX_TOKENS ? "length" : "content_filter",
+            usage: {
+              promptTokens: inputTokens,
+              completionTokens: outputTokens,
+              totalTokens: inputTokens + outputTokens
+            }
+          });
+        }
       }
+    } catch (error) {
+      logger.errors("gemini.summarize fatal", {
+        error,
+        source: "gemini.summarize"
+      });
+      throw error;
     }
   }
   getFormatInstructions(style) {

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/** Cast an event object to StreamChunk. */\nconst asChunk = (chunk: Record<string, unknown>) =>\n  chunk as unknown as StreamChunk\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 asChunk({\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 asChunk({\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":";;AAgBA,MAAM,UAAU,CAAC,UACf;AASK,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,QAAQ;AAAA,cACZ,MAAM;AAAA,cACN,WAAW;AAAA,cACX;AAAA,cACA,WAAW,KAAK,IAAA;AAAA,cAChB,OAAO,KAAK;AAAA,cACZ,SAAS;AAAA,YAAA,CACV;AAAA,UACH;AAAA,QACF;AAAA,MACF;AAGA,YAAM,eAAe,MAAM,aAAa,CAAC,GAAG;AAC5C,UACE,iBAAiB,aAAa,QAC9B,iBAAiB,aAAa,cAC9B,iBAAiB,aAAa,QAC9B;AACA,cAAM,QAAQ;AAAA,UACZ,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,CACD;AAAA,MACH;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/** Cast an event object to StreamChunk. */\nconst asChunk = (chunk: Record<string, unknown>) =>\n  chunk as unknown as StreamChunk\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 { logger } = options\n    const model = options.model\n\n    logger.request(`activity=summarize provider=gemini`, {\n      provider: 'gemini',\n      model,\n    })\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    try {\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    } catch (error) {\n      logger.errors('gemini.summarize fatal', {\n        error,\n        source: 'gemini.summarize',\n      })\n      throw error\n    }\n  }\n\n  async *summarizeStream(\n    options: SummarizationOptions,\n  ): AsyncIterable<StreamChunk> {\n    const { logger } = options\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    logger.request(`activity=summarize provider=gemini`, {\n      provider: 'gemini',\n      model,\n      stream: true,\n    })\n\n    try {\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        logger.provider(`provider=gemini`, { chunk })\n        // Track usage metadata\n        if (chunk.usageMetadata) {\n          inputTokens = chunk.usageMetadata.promptTokenCount ?? inputTokens\n          outputTokens =\n            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 asChunk({\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 asChunk({\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    } catch (error) {\n      logger.errors('gemini.summarize fatal', {\n        error,\n        source: 'gemini.summarize',\n      })\n      throw error\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":";;AAgBA,MAAM,UAAU,CAAC,UACf;AASK,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,EAAE,WAAW;AACnB,UAAM,QAAQ,QAAQ;AAEtB,WAAO,QAAQ,sCAAsC;AAAA,MACnD,UAAU;AAAA,MACV;AAAA,IAAA,CACD;AAGD,UAAM,qBAAqB,KAAK,sBAAsB,QAAQ,KAAK;AACnE,UAAM,qBAAqB,QAAQ,YAC/B,2BAA2B,QAAQ,SAAS,aAC5C;AAEJ,UAAM,eAAe,qDAAqD,kBAAkB,GAAG,kBAAkB;AAEjH,QAAI;AACF,YAAM,WAAW,MAAM,KAAK,OAAO,OAAO,gBAAgB;AAAA,QACxD;AAAA,QACA,UAAU;AAAA,UACR;AAAA,YACE,MAAM;AAAA,YACN,OAAO;AAAA,cACL,EAAE,MAAM;AAAA;AAAA,EAAsC,QAAQ,IAAI,GAAA;AAAA,YAAG;AAAA,UAC/D;AAAA,QACF;AAAA,QAEF,QAAQ;AAAA,UACN,mBAAmB;AAAA,QAAA;AAAA,MACrB,CACD;AAED,YAAM,UAAU,SAAS,QAAQ;AACjC,YAAM,cAAc,SAAS,eAAe,oBAAoB;AAChE,YAAM,eAAe,SAAS,eAAe,wBAAwB;AAErE,aAAO;AAAA,QACL,IAAI,WAAW,KAAK;AAAA,QACpB;AAAA,QACA;AAAA,QACA,OAAO;AAAA,UACL,cAAc;AAAA,UACd,kBAAkB;AAAA,UAClB,aAAa,cAAc;AAAA,QAAA;AAAA,MAC7B;AAAA,IAEJ,SAAS,OAAO;AACd,aAAO,OAAO,0BAA0B;AAAA,QACtC;AAAA,QACA,QAAQ;AAAA,MAAA,CACT;AACD,YAAM;AAAA,IACR;AAAA,EACF;AAAA,EAEA,OAAO,gBACL,SAC4B;AAC5B,UAAM,EAAE,WAAW;AACnB,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,WAAO,QAAQ,sCAAsC;AAAA,MACnD,UAAU;AAAA,MACV;AAAA,MACA,QAAQ;AAAA,IAAA,CACT;AAED,QAAI;AACF,YAAM,SAAS,MAAM,KAAK,OAAO,OAAO,sBAAsB;AAAA,QAC5D;AAAA,QACA,UAAU;AAAA,UACR;AAAA,YACE,MAAM;AAAA,YACN,OAAO;AAAA,cACL,EAAE,MAAM;AAAA;AAAA,EAAsC,QAAQ,IAAI,GAAA;AAAA,YAAG;AAAA,UAC/D;AAAA,QACF;AAAA,QAEF,QAAQ;AAAA,UACN,mBAAmB;AAAA,QAAA;AAAA,MACrB,CACD;AAED,uBAAiB,SAAS,QAAQ;AAChC,eAAO,SAAS,mBAAmB,EAAE,MAAA,CAAO;AAE5C,YAAI,MAAM,eAAe;AACvB,wBAAc,MAAM,cAAc,oBAAoB;AACtD,yBACE,MAAM,cAAc,wBAAwB;AAAA,QAChD;AAEA,YAAI,MAAM,aAAa,CAAC,GAAG,SAAS,OAAO;AACzC,qBAAW,QAAQ,MAAM,WAAW,CAAC,EAAE,QAAQ,OAAO;AACpD,gBAAI,KAAK,MAAM;AACb,oCAAsB,KAAK;AAC3B,oBAAM,QAAQ;AAAA,gBACZ,MAAM;AAAA,gBACN,WAAW;AAAA,gBACX;AAAA,gBACA,WAAW,KAAK,IAAA;AAAA,gBAChB,OAAO,KAAK;AAAA,gBACZ,SAAS;AAAA,cAAA,CACV;AAAA,YACH;AAAA,UACF;AAAA,QACF;AAGA,cAAM,eAAe,MAAM,aAAa,CAAC,GAAG;AAC5C,YACE,iBAAiB,aAAa,QAC9B,iBAAiB,aAAa,cAC9B,iBAAiB,aAAa,QAC9B;AACA,gBAAM,QAAQ;AAAA,YACZ,MAAM;AAAA,YACN,OAAO;AAAA,YACP;AAAA,YACA,WAAW,KAAK,IAAA;AAAA,YAChB,cACE,iBAAiB,aAAa,OAC1B,SACA,iBAAiB,aAAa,aAC5B,WACA;AAAA,YACR,OAAO;AAAA,cACL,cAAc;AAAA,cACd,kBAAkB;AAAA,cAClB,aAAa,cAAc;AAAA,YAAA;AAAA,UAC7B,CACD;AAAA,QACH;AAAA,MACF;AAAA,IACF,SAAS,OAAO;AACd,aAAO,OAAO,0BAA0B;AAAA,QACtC;AAAA,QACA,QAAQ;AAAA,MAAA,CACT;AACD,YAAM;AAAA,IACR;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/adapters/text.d.ts

@@ -1,5 +1,5 @@
 import { BaseTextAdapter, StructuredOutputOptions, StructuredOutputResult } from '@tanstack/ai/adapters';
-import { GEMINI_MODELS, GeminiChatModelProviderOptionsByName, GeminiModelInputModalitiesByName } from '../model-meta.js';
+import { GEMINI_MODELS, GeminiChatModelProviderOptionsByName, GeminiChatModelToolCapabilitiesByName, GeminiModelInputModalitiesByName } from '../model-meta.js';
 import { Modality, StreamChunk, TextOptions } from '@tanstack/ai';
 import { ExternalTextProviderOptions } from '../text/text-provider-options.js';
 import { GeminiMessageMetadataByModality } from '../message-types.js';
@@ -23,13 +23,18 @@ type ResolveProviderOptions<TModel extends string> = TModel extends keyof Gemini
  * If the model has explicit modalities in the map, use those; otherwise use all modalities.
  */
 type ResolveInputModalities<TModel extends string> = TModel extends keyof GeminiModelInputModalitiesByName ? GeminiModelInputModalitiesByName[TModel] : readonly ['text', 'image', 'audio', 'video', 'document'];
+/**
+ * Resolve tool capabilities for a specific model.
+ * If the model has explicit tools in the map, use those; otherwise use empty tuple.
+ */
+type ResolveToolCapabilities<TModel extends string> = TModel extends keyof GeminiChatModelToolCapabilitiesByName ? NonNullable<GeminiChatModelToolCapabilitiesByName[TModel]> : readonly [];
 /**
  * Gemini Text (Chat) Adapter
  *
  * Tree-shakeable adapter for Gemini chat/text completion functionality.
  * Import only what you need for smaller bundle sizes.
  */
-export declare class GeminiTextAdapter<TModel extends (typeof GEMINI_MODELS)[number], TProviderOptions extends object = ResolveProviderOptions<TModel>, TInputModalities extends ReadonlyArray<Modality> = ResolveInputModalities<TModel>> extends BaseTextAdapter<TModel, TProviderOptions, TInputModalities, GeminiMessageMetadataByModality> {
+export declare class GeminiTextAdapter<TModel extends (typeof GEMINI_MODELS)[number], TProviderOptions extends Record<string, any> = ResolveProviderOptions<TModel>, TInputModalities extends ReadonlyArray<Modality> = ResolveInputModalities<TModel>, TToolCapabilities extends ReadonlyArray<string> = ResolveToolCapabilities<TModel>> extends BaseTextAdapter<TModel, TProviderOptions, TInputModalities, GeminiMessageMetadataByModality, TToolCapabilities> {
     readonly kind: "text";
     readonly name: "gemini";
     private client;
@@ -64,10 +69,10 @@ export declare class GeminiTextAdapter<TModel extends (typeof GEMINI_MODELS)[num
  * Creates a Gemini text adapter with explicit API key.
  * Type resolution happens here at the call site.
  */
-export declare function createGeminiChat<TModel extends (typeof GEMINI_MODELS)[number]>(model: TModel, apiKey: string, config?: Omit<GeminiTextConfig, 'apiKey'>): GeminiTextAdapter<TModel, ResolveProviderOptions<TModel>, ResolveInputModalities<TModel>>;
+export declare function createGeminiChat<TModel extends (typeof GEMINI_MODELS)[number]>(model: TModel, apiKey: string, config?: Omit<GeminiTextConfig, 'apiKey'>): GeminiTextAdapter<TModel, ResolveProviderOptions<TModel>, ResolveInputModalities<TModel>, ResolveToolCapabilities<TModel>>;
 /**
  * Creates a Gemini text adapter with automatic API key detection.
  * Type resolution happens here at the call site.
  */
-export declare function geminiText<TModel extends (typeof GEMINI_MODELS)[number]>(model: TModel, config?: Omit<GeminiTextConfig, 'apiKey'>): GeminiTextAdapter<TModel, ResolveProviderOptions<TModel>, ResolveInputModalities<TModel>>;
+export declare function geminiText<TModel extends (typeof GEMINI_MODELS)[number]>(model: TModel, config?: Omit<GeminiTextConfig, 'apiKey'>): GeminiTextAdapter<TModel, ResolveProviderOptions<TModel>, ResolveInputModalities<TModel>, ResolveToolCapabilities<TModel>>;
 export {};

package/dist/esm/adapters/text.js

@@ -12,11 +12,20 @@ class GeminiTextAdapter extends BaseTextAdapter {
   }
   async *chatStream(options) {
     const mappedOptions = this.mapCommonOptionsToGemini(options);
+    const { logger } = options;
     try {
+      logger.request(
+        `activity=chat provider=gemini model=${this.model} messages=${options.messages.length} tools=${options.tools?.length ?? 0} stream=true`,
+        { provider: "gemini", model: this.model }
+      );
       const result = await this.client.models.generateContentStream(mappedOptions);
-      yield* this.processStreamChunks(result, options);
+      yield* this.processStreamChunks(result, options, logger);
     } catch (error) {
       const timestamp = Date.now();
+      logger.errors("gemini.chatStream fatal", {
+        error,
+        source: "gemini.chatStream"
+      });
       yield asChunk({
         type: "RUN_ERROR",
         model: options.model,
@@ -35,8 +44,13 @@ class GeminiTextAdapter extends BaseTextAdapter {
    */
   async structuredOutput(options) {
     const { chatOptions, outputSchema } = options;
+    const { logger } = chatOptions;
     const mappedOptions = this.mapCommonOptionsToGemini(chatOptions);
     try {
+      logger.request(
+        `activity=chat provider=gemini model=${this.model} messages=${chatOptions.messages.length} tools=${chatOptions.tools?.length ?? 0} stream=false`,
+        { provider: "gemini", model: this.model }
+      );
       const result = await this.client.models.generateContent({
         ...mappedOptions,
         config: {
@@ -59,6 +73,10 @@ class GeminiTextAdapter extends BaseTextAdapter {
         rawText
       };
     } catch (error) {
+      logger.errors("gemini.structuredOutput fatal", {
+        error,
+        source: "gemini.structuredOutput"
+      });
       throw new Error(
         error instanceof Error ? error.message : "An unknown error occurred during structured output generation."
       );
@@ -78,7 +96,7 @@ class GeminiTextAdapter extends BaseTextAdapter {
     }
     return textContent;
   }
-  async *processStreamChunks(result, options) {
+  async *processStreamChunks(result, options, logger) {
     const model = options.model;
     const timestamp = Date.now();
     let accumulatedContent = "";
@@ -95,6 +113,7 @@ class GeminiTextAdapter extends BaseTextAdapter {
     let hasEmittedTextMessageStart = false;
     let hasEmittedStepStarted = false;
     for await (const chunk of result) {
+      logger.provider(`provider=gemini`, { chunk });
       if (!hasEmittedRunStarted) {
         hasEmittedRunStarted = true;
         yield asChunk({