@tanstack/ai-gemini 0.20.1 → 0.22.0

package/README.md

@@ -1,5 +1,19 @@
 <div align="center">
-  <img src="https://raw.githubusercontent.com/TanStack/ai/main/media/header_ai.png" alt="TanStack AI" />
+  <picture>
+    <source
+      media="(prefers-color-scheme: dark)"
+      srcset="https://tanstack.com/api/readme/ai.png?theme=dark"
+    />
+    <source
+      media="(prefers-color-scheme: light)"
+      srcset="https://tanstack.com/api/readme/ai.png"
+    />
+    <img
+      src="https://tanstack.com/api/readme/ai.png"
+      alt="TanStack AI"
+      width="900"
+    />
+  </picture>
 </div>
 
 <br />

package/dist/esm/adapters/audio.js

@@ -1,66 +1,118 @@
-import { BaseAudioAdapter } from "@tanstack/ai/adapters";
-import { createGeminiClient, generateId, getGeminiApiKeyFromEnv } from "../utils/client.js";
 import { buildGeminiUsage } from "../usage.js";
-class GeminiAudioAdapter extends BaseAudioAdapter {
-  name = "gemini";
-  client;
-  constructor(config, model) {
-    super(model, config);
-    this.client = createGeminiClient(config);
-  }
-  async generateAudio(options) {
-    const { model, prompt, modelOptions, logger } = options;
-    logger.request(`activity=generateAudio provider=gemini model=${model}`, {
-      provider: "gemini",
-      model
-    });
-    try {
-      const response = await this.client.models.generateContent({
-        model,
-        contents: [{ role: "user", parts: [{ text: prompt }] }],
-        config: {
-          responseModalities: ["AUDIO", "TEXT"],
-          ...modelOptions?.seed != null ? { seed: modelOptions.seed } : {}
-        }
-      });
-      const parts = response.candidates?.[0]?.content?.parts ?? [];
-      const audioPart = parts.find(
-        (part) => part.inlineData?.mimeType?.startsWith("audio/")
-      );
-      if (!audioPart?.inlineData?.data) {
-        throw new Error("No audio data in Gemini Lyria response");
-      }
-      const contentType = audioPart.inlineData.mimeType;
-      return {
-        id: generateId(this.name),
-        model,
-        audio: {
-          b64Json: audioPart.inlineData.data,
-          ...contentType !== void 0 && { contentType }
-        },
-        // Surface token usage (with per-modality breakdown) when Gemini reports
-        // it. Spread conditionally for exactOptionalPropertyTypes.
-        ...response.usageMetadata ? { usage: buildGeminiUsage(response.usageMetadata) } : {}
-      };
-    } catch (error) {
-      logger.errors("gemini.generateAudio fatal", {
-        error,
-        source: "gemini.generateAudio"
-      });
-      throw error;
-    }
-  }
-}
+import { createGeminiClient, generateId, getGeminiApiKeyFromEnv } from "../utils/client.js";
+import "../utils/index.js";
+import { BaseAudioAdapter } from "@tanstack/ai/adapters";
+//#region src/adapters/audio.ts
+/**
+* Gemini Lyria Music Generation Adapter.
+*
+* Tree-shakeable adapter for Google Lyria music generation via the Gemini API.
+*
+* Models:
+* - `lyria-3-pro-preview` — flagship model, full-length songs with verses,
+*   choruses, and bridges. Outputs MP3 or WAV at 48 kHz stereo.
+* - `lyria-3-clip-preview` — 30-second clips in MP3.
+*
+* @see https://ai.google.dev/gemini-api/docs/music-generation
+*
+* @example
+* ```typescript
+* const adapter = geminiAudio('lyria-3-pro-preview')
+* const result = await generateAudio({
+*   adapter,
+*   prompt: 'An upbeat jazz track with saxophone and drums',
+* })
+* ```
+*/
+var GeminiAudioAdapter = class extends BaseAudioAdapter {
+	name = "gemini";
+	client;
+	constructor(config, model) {
+		super(model, config);
+		this.client = createGeminiClient(config);
+	}
+	async generateAudio(options) {
+		const { model, prompt, modelOptions, logger } = options;
+		logger.request(`activity=generateAudio provider=gemini model=${model}`, {
+			provider: "gemini",
+			model
+		});
+		try {
+			const response = await this.client.models.generateContent({
+				model,
+				contents: [{
+					role: "user",
+					parts: [{ text: prompt }]
+				}],
+				config: {
+					responseModalities: ["AUDIO", "TEXT"],
+					...modelOptions?.seed != null ? { seed: modelOptions.seed } : {}
+				}
+			});
+			const audioPart = (response.candidates?.[0]?.content?.parts ?? []).find((part) => part.inlineData?.mimeType?.startsWith("audio/"));
+			if (!audioPart?.inlineData?.data) throw new Error("No audio data in Gemini Lyria response");
+			const contentType = audioPart.inlineData.mimeType;
+			return {
+				id: generateId(this.name),
+				model,
+				audio: {
+					b64Json: audioPart.inlineData.data,
+					...contentType !== void 0 && { contentType }
+				},
+				...response.usageMetadata ? { usage: buildGeminiUsage(response.usageMetadata) } : {}
+			};
+		} catch (error) {
+			logger.errors("gemini.generateAudio fatal", {
+				error,
+				source: "gemini.generateAudio"
+			});
+			throw error;
+		}
+	}
+};
+/**
+* Creates a Gemini Lyria audio adapter with an explicit API key.
+*
+* @param model - The Lyria model name (e.g., 'lyria-3-pro-preview')
+* @param apiKey - Your Google API key
+* @param config - Optional additional configuration
+*
+* @example
+* ```typescript
+* const adapter = createGeminiAudio('lyria-3-pro-preview', 'your-api-key')
+* const result = await generateAudio({
+*   adapter,
+*   prompt: 'Ambient electronic music with soft pads',
+* })
+* ```
+*/
 function createGeminiAudio(model, apiKey, config) {
-  return new GeminiAudioAdapter({ ...config, apiKey }, model);
+	return new GeminiAudioAdapter({
+		...config,
+		apiKey
+	}, model);
 }
+/**
+* Creates a Gemini Lyria audio adapter with automatic API key detection.
+*
+* Looks for `GOOGLE_API_KEY` or `GEMINI_API_KEY` in the environment.
+*
+* @param model - The Lyria model name (e.g., 'lyria-3-pro-preview')
+* @param config - Optional configuration (excluding apiKey)
+*
+* @example
+* ```typescript
+* const adapter = geminiAudio('lyria-3-pro-preview')
+* const result = await generateAudio({
+*   adapter,
+*   prompt: 'An orchestral piece with strings and brass',
+* })
+* ```
+*/
 function geminiAudio(model, config) {
-  const apiKey = getGeminiApiKeyFromEnv();
-  return createGeminiAudio(model, apiKey, config);
+	return createGeminiAudio(model, getGeminiApiKeyFromEnv(), config);
 }
-export {
-  GeminiAudioAdapter,
-  createGeminiAudio,
-  geminiAudio
-};
-//# sourceMappingURL=audio.js.map
+//#endregion
+export { GeminiAudioAdapter, createGeminiAudio, geminiAudio };
+
+//# sourceMappingURL=audio.js.map

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

@@ -1 +1 @@
-{"version":3,"file":"audio.js","sources":["../../../src/adapters/audio.ts"],"sourcesContent":["import { BaseAudioAdapter } from '@tanstack/ai/adapters'\nimport {\n  createGeminiClient,\n  generateId,\n  getGeminiApiKeyFromEnv,\n} from '../utils'\nimport { buildGeminiUsage } from '../usage'\nimport type { GEMINI_AUDIO_MODELS } from '../model-meta'\nimport type {\n  AudioGenerationOptions,\n  AudioGenerationResult,\n} from '@tanstack/ai'\nimport type { GoogleGenAI } from '@google/genai'\nimport type { GeminiClientConfig } from '../utils/client'\n\n/**\n * Provider options for Gemini Lyria music generation.\n *\n * Notes on the Lyria 3 surface area:\n * - `lyria-3-clip-preview` always returns MP3 (30-second clips). It does\n *   not accept `responseMimeType`, and duration is fixed at 30 seconds —\n *   the generic `duration` option on `AudioActivityOptions` is ignored.\n * - `lyria-3-pro-preview` returns MP3 by default. Duration is controlled\n *   via the natural-language prompt, not a separate SDK field, so the\n *   generic `duration` option is similarly ignored.\n * - `negativePrompt` is NOT accepted by `GenerateContentConfig` and has\n *   therefore been removed from this surface to avoid giving callers a\n *   silently-dropped knob.\n *\n * @see https://ai.google.dev/gemini-api/docs/music-generation\n */\nexport interface GeminiAudioProviderOptions {\n  /**\n   * Seed for deterministic generation.\n   */\n  seed?: number\n}\n\nexport interface GeminiAudioConfig extends GeminiClientConfig {}\n\n/** Model type for Gemini Lyria audio generation */\nexport type GeminiAudioModel = (typeof GEMINI_AUDIO_MODELS)[number]\n\n/**\n * Gemini Lyria Music Generation Adapter.\n *\n * Tree-shakeable adapter for Google Lyria music generation via the Gemini API.\n *\n * Models:\n * - `lyria-3-pro-preview` — flagship model, full-length songs with verses,\n *   choruses, and bridges. Outputs MP3 or WAV at 48 kHz stereo.\n * - `lyria-3-clip-preview` — 30-second clips in MP3.\n *\n * @see https://ai.google.dev/gemini-api/docs/music-generation\n *\n * @example\n * ```typescript\n * const adapter = geminiAudio('lyria-3-pro-preview')\n * const result = await generateAudio({\n *   adapter,\n *   prompt: 'An upbeat jazz track with saxophone and drums',\n * })\n * ```\n */\nexport class GeminiAudioAdapter<\n  TModel extends GeminiAudioModel,\n> extends BaseAudioAdapter<TModel, GeminiAudioProviderOptions> {\n  readonly name = 'gemini' as const\n\n  private readonly client: GoogleGenAI\n\n  constructor(config: GeminiAudioConfig, model: TModel) {\n    super(model, config)\n    this.client = createGeminiClient(config)\n  }\n\n  async generateAudio(\n    options: AudioGenerationOptions<GeminiAudioProviderOptions>,\n  ): Promise<AudioGenerationResult> {\n    const { model, prompt, modelOptions, logger } = options\n\n    logger.request(`activity=generateAudio provider=gemini model=${model}`, {\n      provider: 'gemini',\n      model,\n    })\n\n    try {\n      // FIXME (SDK audit): Lyria 3 music generation may not belong on\n      // generateContent at all — @google/genai exposes a `LiveMusicSession`\n      // (`ai.live.music.connect`) with a `musicGenerationConfig` object.\n      // `seed` is valid on GenerateContentConfig, and Lyria always returns\n      // MP3 today, so we don't forward `responseMimeType` either.\n      // The runtime test `emits only GenerateContentConfig-valid fields`\n      // asserts the config shape so a later SDK audit can catch regressions.\n      const response = await this.client.models.generateContent({\n        model,\n        contents: [{ role: 'user', parts: [{ text: prompt }] }],\n        config: {\n          responseModalities: ['AUDIO', 'TEXT'],\n          ...(modelOptions?.seed != null ? { seed: modelOptions.seed } : {}),\n        },\n      })\n\n      const parts = response.candidates?.[0]?.content?.parts ?? []\n      const audioPart = parts.find((part: any) =>\n        part.inlineData?.mimeType?.startsWith('audio/'),\n      )\n\n      if (!audioPart?.inlineData?.data) {\n        throw new Error('No audio data in Gemini Lyria response')\n      }\n\n      // audioPart was selected because mimeType.startsWith('audio/') was\n      // truthy, so the mime type is guaranteed to be a string here. Trust the\n      // value Gemini returned rather than inventing a non-standard\n      // `audio/mp3` fallback (IANA is `audio/mpeg`).\n      const contentType = audioPart.inlineData.mimeType\n\n      return {\n        id: generateId(this.name),\n        model,\n        audio: {\n          b64Json: audioPart.inlineData.data,\n          ...(contentType !== undefined && { contentType }),\n        },\n        // Surface token usage (with per-modality breakdown) when Gemini reports\n        // it. Spread conditionally for exactOptionalPropertyTypes.\n        ...(response.usageMetadata\n          ? { usage: buildGeminiUsage(response.usageMetadata) }\n          : {}),\n      }\n    } catch (error) {\n      logger.errors('gemini.generateAudio fatal', {\n        error,\n        source: 'gemini.generateAudio',\n      })\n      throw error\n    }\n  }\n}\n\n/**\n * Creates a Gemini Lyria audio adapter with an explicit API key.\n *\n * @param model - The Lyria model name (e.g., 'lyria-3-pro-preview')\n * @param apiKey - Your Google API key\n * @param config - Optional additional configuration\n *\n * @example\n * ```typescript\n * const adapter = createGeminiAudio('lyria-3-pro-preview', 'your-api-key')\n * const result = await generateAudio({\n *   adapter,\n *   prompt: 'Ambient electronic music with soft pads',\n * })\n * ```\n */\nexport function createGeminiAudio<TModel extends GeminiAudioModel>(\n  model: TModel,\n  apiKey: string,\n  config?: Omit<GeminiAudioConfig, 'apiKey'>,\n): GeminiAudioAdapter<TModel> {\n  // Put apiKey LAST so caller-supplied config can't silently override the\n  // explicit argument.\n  return new GeminiAudioAdapter({ ...config, apiKey }, model)\n}\n\n/**\n * Creates a Gemini Lyria audio adapter with automatic API key detection.\n *\n * Looks for `GOOGLE_API_KEY` or `GEMINI_API_KEY` in the environment.\n *\n * @param model - The Lyria model name (e.g., 'lyria-3-pro-preview')\n * @param config - Optional configuration (excluding apiKey)\n *\n * @example\n * ```typescript\n * const adapter = geminiAudio('lyria-3-pro-preview')\n * const result = await generateAudio({\n *   adapter,\n *   prompt: 'An orchestral piece with strings and brass',\n * })\n * ```\n */\nexport function geminiAudio<TModel extends GeminiAudioModel>(\n  model: TModel,\n  config?: Omit<GeminiAudioConfig, 'apiKey'>,\n): GeminiAudioAdapter<TModel> {\n  const apiKey = getGeminiApiKeyFromEnv()\n  return createGeminiAudio(model, apiKey, config)\n}\n"],"names":[],"mappings":";;;AAgEO,MAAM,2BAEH,iBAAqD;AAAA,EACpD,OAAO;AAAA,EAEC;AAAA,EAEjB,YAAY,QAA2B,OAAe;AACpD,UAAM,OAAO,MAAM;AACnB,SAAK,SAAS,mBAAmB,MAAM;AAAA,EACzC;AAAA,EAEA,MAAM,cACJ,SACgC;AAChC,UAAM,EAAE,OAAO,QAAQ,cAAc,WAAW;AAEhD,WAAO,QAAQ,gDAAgD,KAAK,IAAI;AAAA,MACtE,UAAU;AAAA,MACV;AAAA,IAAA,CACD;AAED,QAAI;AAQF,YAAM,WAAW,MAAM,KAAK,OAAO,OAAO,gBAAgB;AAAA,QACxD;AAAA,QACA,UAAU,CAAC,EAAE,MAAM,QAAQ,OAAO,CAAC,EAAE,MAAM,OAAA,CAAQ,GAAG;AAAA,QACtD,QAAQ;AAAA,UACN,oBAAoB,CAAC,SAAS,MAAM;AAAA,UACpC,GAAI,cAAc,QAAQ,OAAO,EAAE,MAAM,aAAa,SAAS,CAAA;AAAA,QAAC;AAAA,MAClE,CACD;AAED,YAAM,QAAQ,SAAS,aAAa,CAAC,GAAG,SAAS,SAAS,CAAA;AAC1D,YAAM,YAAY,MAAM;AAAA,QAAK,CAAC,SAC5B,KAAK,YAAY,UAAU,WAAW,QAAQ;AAAA,MAAA;AAGhD,UAAI,CAAC,WAAW,YAAY,MAAM;AAChC,cAAM,IAAI,MAAM,wCAAwC;AAAA,MAC1D;AAMA,YAAM,cAAc,UAAU,WAAW;AAEzC,aAAO;AAAA,QACL,IAAI,WAAW,KAAK,IAAI;AAAA,QACxB;AAAA,QACA,OAAO;AAAA,UACL,SAAS,UAAU,WAAW;AAAA,UAC9B,GAAI,gBAAgB,UAAa,EAAE,YAAA;AAAA,QAAY;AAAA;AAAA;AAAA,QAIjD,GAAI,SAAS,gBACT,EAAE,OAAO,iBAAiB,SAAS,aAAa,MAChD,CAAA;AAAA,MAAC;AAAA,IAET,SAAS,OAAO;AACd,aAAO,OAAO,8BAA8B;AAAA,QAC1C;AAAA,QACA,QAAQ;AAAA,MAAA,CACT;AACD,YAAM;AAAA,IACR;AAAA,EACF;AACF;AAkBO,SAAS,kBACd,OACA,QACA,QAC4B;AAG5B,SAAO,IAAI,mBAAmB,EAAE,GAAG,QAAQ,OAAA,GAAU,KAAK;AAC5D;AAmBO,SAAS,YACd,OACA,QAC4B;AAC5B,QAAM,SAAS,uBAAA;AACf,SAAO,kBAAkB,OAAO,QAAQ,MAAM;AAChD;"}
+{"version":3,"file":"audio.js","names":[],"sources":["../../../src/adapters/audio.ts"],"sourcesContent":["import { BaseAudioAdapter } from '@tanstack/ai/adapters'\nimport {\n  createGeminiClient,\n  generateId,\n  getGeminiApiKeyFromEnv,\n} from '../utils'\nimport { buildGeminiUsage } from '../usage'\nimport type { GEMINI_AUDIO_MODELS } from '../model-meta'\nimport type {\n  AudioGenerationOptions,\n  AudioGenerationResult,\n} from '@tanstack/ai'\nimport type { GoogleGenAI } from '@google/genai'\nimport type { GeminiClientConfig } from '../utils/client'\n\n/**\n * Provider options for Gemini Lyria music generation.\n *\n * Notes on the Lyria 3 surface area:\n * - `lyria-3-clip-preview` always returns MP3 (30-second clips). It does\n *   not accept `responseMimeType`, and duration is fixed at 30 seconds —\n *   the generic `duration` option on `AudioActivityOptions` is ignored.\n * - `lyria-3-pro-preview` returns MP3 by default. Duration is controlled\n *   via the natural-language prompt, not a separate SDK field, so the\n *   generic `duration` option is similarly ignored.\n * - `negativePrompt` is NOT accepted by `GenerateContentConfig` and has\n *   therefore been removed from this surface to avoid giving callers a\n *   silently-dropped knob.\n *\n * @see https://ai.google.dev/gemini-api/docs/music-generation\n */\nexport interface GeminiAudioProviderOptions {\n  /**\n   * Seed for deterministic generation.\n   */\n  seed?: number\n}\n\nexport interface GeminiAudioConfig extends GeminiClientConfig {}\n\n/** Model type for Gemini Lyria audio generation */\nexport type GeminiAudioModel = (typeof GEMINI_AUDIO_MODELS)[number]\n\n/**\n * Gemini Lyria Music Generation Adapter.\n *\n * Tree-shakeable adapter for Google Lyria music generation via the Gemini API.\n *\n * Models:\n * - `lyria-3-pro-preview` — flagship model, full-length songs with verses,\n *   choruses, and bridges. Outputs MP3 or WAV at 48 kHz stereo.\n * - `lyria-3-clip-preview` — 30-second clips in MP3.\n *\n * @see https://ai.google.dev/gemini-api/docs/music-generation\n *\n * @example\n * ```typescript\n * const adapter = geminiAudio('lyria-3-pro-preview')\n * const result = await generateAudio({\n *   adapter,\n *   prompt: 'An upbeat jazz track with saxophone and drums',\n * })\n * ```\n */\nexport class GeminiAudioAdapter<\n  TModel extends GeminiAudioModel,\n> extends BaseAudioAdapter<TModel, GeminiAudioProviderOptions> {\n  readonly name = 'gemini' as const\n\n  private readonly client: GoogleGenAI\n\n  constructor(config: GeminiAudioConfig, model: TModel) {\n    super(model, config)\n    this.client = createGeminiClient(config)\n  }\n\n  async generateAudio(\n    options: AudioGenerationOptions<GeminiAudioProviderOptions>,\n  ): Promise<AudioGenerationResult> {\n    const { model, prompt, modelOptions, logger } = options\n\n    logger.request(`activity=generateAudio provider=gemini model=${model}`, {\n      provider: 'gemini',\n      model,\n    })\n\n    try {\n      // FIXME (SDK audit): Lyria 3 music generation may not belong on\n      // generateContent at all — @google/genai exposes a `LiveMusicSession`\n      // (`ai.live.music.connect`) with a `musicGenerationConfig` object.\n      // `seed` is valid on GenerateContentConfig, and Lyria always returns\n      // MP3 today, so we don't forward `responseMimeType` either.\n      // The runtime test `emits only GenerateContentConfig-valid fields`\n      // asserts the config shape so a later SDK audit can catch regressions.\n      const response = await this.client.models.generateContent({\n        model,\n        contents: [{ role: 'user', parts: [{ text: prompt }] }],\n        config: {\n          responseModalities: ['AUDIO', 'TEXT'],\n          ...(modelOptions?.seed != null ? { seed: modelOptions.seed } : {}),\n        },\n      })\n\n      const parts = response.candidates?.[0]?.content?.parts ?? []\n      const audioPart = parts.find((part: any) =>\n        part.inlineData?.mimeType?.startsWith('audio/'),\n      )\n\n      if (!audioPart?.inlineData?.data) {\n        throw new Error('No audio data in Gemini Lyria response')\n      }\n\n      // audioPart was selected because mimeType.startsWith('audio/') was\n      // truthy, so the mime type is guaranteed to be a string here. Trust the\n      // value Gemini returned rather than inventing a non-standard\n      // `audio/mp3` fallback (IANA is `audio/mpeg`).\n      const contentType = audioPart.inlineData.mimeType\n\n      return {\n        id: generateId(this.name),\n        model,\n        audio: {\n          b64Json: audioPart.inlineData.data,\n          ...(contentType !== undefined && { contentType }),\n        },\n        // Surface token usage (with per-modality breakdown) when Gemini reports\n        // it. Spread conditionally for exactOptionalPropertyTypes.\n        ...(response.usageMetadata\n          ? { usage: buildGeminiUsage(response.usageMetadata) }\n          : {}),\n      }\n    } catch (error) {\n      logger.errors('gemini.generateAudio fatal', {\n        error,\n        source: 'gemini.generateAudio',\n      })\n      throw error\n    }\n  }\n}\n\n/**\n * Creates a Gemini Lyria audio adapter with an explicit API key.\n *\n * @param model - The Lyria model name (e.g., 'lyria-3-pro-preview')\n * @param apiKey - Your Google API key\n * @param config - Optional additional configuration\n *\n * @example\n * ```typescript\n * const adapter = createGeminiAudio('lyria-3-pro-preview', 'your-api-key')\n * const result = await generateAudio({\n *   adapter,\n *   prompt: 'Ambient electronic music with soft pads',\n * })\n * ```\n */\nexport function createGeminiAudio<TModel extends GeminiAudioModel>(\n  model: TModel,\n  apiKey: string,\n  config?: Omit<GeminiAudioConfig, 'apiKey'>,\n): GeminiAudioAdapter<TModel> {\n  // Put apiKey LAST so caller-supplied config can't silently override the\n  // explicit argument.\n  return new GeminiAudioAdapter({ ...config, apiKey }, model)\n}\n\n/**\n * Creates a Gemini Lyria audio adapter with automatic API key detection.\n *\n * Looks for `GOOGLE_API_KEY` or `GEMINI_API_KEY` in the environment.\n *\n * @param model - The Lyria model name (e.g., 'lyria-3-pro-preview')\n * @param config - Optional configuration (excluding apiKey)\n *\n * @example\n * ```typescript\n * const adapter = geminiAudio('lyria-3-pro-preview')\n * const result = await generateAudio({\n *   adapter,\n *   prompt: 'An orchestral piece with strings and brass',\n * })\n * ```\n */\nexport function geminiAudio<TModel extends GeminiAudioModel>(\n  model: TModel,\n  config?: Omit<GeminiAudioConfig, 'apiKey'>,\n): GeminiAudioAdapter<TModel> {\n  const apiKey = getGeminiApiKeyFromEnv()\n  return createGeminiAudio(model, apiKey, config)\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;AAgEA,IAAa,qBAAb,cAEU,iBAAqD;CAC7D,OAAgB;CAEhB;CAEA,YAAY,QAA2B,OAAe;EACpD,MAAM,OAAO,MAAM;EACnB,KAAK,SAAS,mBAAmB,MAAM;CACzC;CAEA,MAAM,cACJ,SACgC;EAChC,MAAM,EAAE,OAAO,QAAQ,cAAc,WAAW;EAEhD,OAAO,QAAQ,gDAAgD,SAAS;GACtE,UAAU;GACV;EACF,CAAC;EAED,IAAI;GAQF,MAAM,WAAW,MAAM,KAAK,OAAO,OAAO,gBAAgB;IACxD;IACA,UAAU,CAAC;KAAE,MAAM;KAAQ,OAAO,CAAC,EAAE,MAAM,OAAO,CAAC;IAAE,CAAC;IACtD,QAAQ;KACN,oBAAoB,CAAC,SAAS,MAAM;KACpC,GAAI,cAAc,QAAQ,OAAO,EAAE,MAAM,aAAa,KAAK,IAAI,CAAC;IAClE;GACF,CAAC;GAGD,MAAM,aADQ,SAAS,aAAa,EAAE,EAAE,SAAS,SAAS,CAAC,EAAA,CACnC,MAAM,SAC5B,KAAK,YAAY,UAAU,WAAW,QAAQ,CAChD;GAEA,IAAI,CAAC,WAAW,YAAY,MAC1B,MAAM,IAAI,MAAM,wCAAwC;GAO1D,MAAM,cAAc,UAAU,WAAW;GAEzC,OAAO;IACL,IAAI,WAAW,KAAK,IAAI;IACxB;IACA,OAAO;KACL,SAAS,UAAU,WAAW;KAC9B,GAAI,gBAAgB,KAAA,KAAa,EAAE,YAAY;IACjD;IAGA,GAAI,SAAS,gBACT,EAAE,OAAO,iBAAiB,SAAS,aAAa,EAAE,IAClD,CAAC;GACP;EACF,SAAS,OAAO;GACd,OAAO,OAAO,8BAA8B;IAC1C;IACA,QAAQ;GACV,CAAC;GACD,MAAM;EACR;CACF;AACF;;;;;;;;;;;;;;;;;AAkBA,SAAgB,kBACd,OACA,QACA,QAC4B;CAG5B,OAAO,IAAI,mBAAmB;EAAE,GAAG;EAAQ;CAAO,GAAG,KAAK;AAC5D;;;;;;;;;;;;;;;;;;AAmBA,SAAgB,YACd,OACA,QAC4B;CAE5B,OAAO,kBAAkB,OADV,uBACiB,GAAQ,MAAM;AAChD"}

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

@@ -0,0 +1,77 @@
+import { BaseEmbeddingAdapter } from '@tanstack/ai/adapters';
+import { EmbeddingOptions, EmbeddingResult } from '@tanstack/ai';
+import { GoogleGenAI } from '@google/genai';
+import { GeminiEmbeddingModel, GeminiEmbeddingModelInputModalitiesByName, GeminiEmbeddingModelProviderOptionsByName } from '../model-meta.js';
+import { GeminiEmbeddingProviderOptions } from '../embedding/embedding-provider-options.js';
+import { GeminiClientConfig } from '../utils/client.js';
+/**
+ * Configuration for Gemini Embedding adapter
+ */
+export interface GeminiEmbeddingConfig extends GeminiClientConfig {
+}
+/**
+ * Gemini Embedding Adapter
+ *
+ * Tree-shakeable adapter for Gemini text embeddings.
+ * Supports gemini-embedding-001.
+ *
+ * Features:
+ * - Batch embedding (one request for the whole input array)
+ * - Matryoshka dimension reduction via the top-level `dimensions` option
+ *   (mapped to the SDK's `outputDimensionality`)
+ * - Task-type hints (`taskType`, `title`) via provider options
+ */
+export declare class GeminiEmbeddingAdapter<TModel extends GeminiEmbeddingModel> extends BaseEmbeddingAdapter<TModel, GeminiEmbeddingProviderOptions, GeminiEmbeddingModelProviderOptionsByName, GeminiEmbeddingModelInputModalitiesByName> {
+    readonly name: "gemini";
+    protected client: GoogleGenAI;
+    constructor(config: GeminiEmbeddingConfig, model: TModel);
+    createEmbeddings(options: EmbeddingOptions<GeminiEmbeddingProviderOptions>): Promise<EmbeddingResult>;
+}
+/**
+ * Creates a Gemini embedding adapter with explicit API key.
+ * Type resolution happens here at the call site.
+ *
+ * @param model - The model name (e.g., 'gemini-embedding-001')
+ * @param apiKey - Your Google API key
+ * @param config - Optional additional configuration
+ * @returns Configured Gemini embedding adapter instance with resolved types
+ *
+ * @example
+ * ```typescript
+ * const adapter = createGeminiEmbedding('gemini-embedding-001', "your-api-key");
+ *
+ * const result = await embed({
+ *   adapter,
+ *   input: 'a red guitar'
+ * });
+ * ```
+ */
+export declare function createGeminiEmbedding<TModel extends GeminiEmbeddingModel>(model: TModel, apiKey: string, config?: Omit<GeminiEmbeddingConfig, 'apiKey'>): GeminiEmbeddingAdapter<TModel>;
+/**
+ * Creates a Gemini embedding adapter with automatic API key detection from environment variables.
+ * Type resolution happens here at the call site.
+ *
+ * Looks for `GOOGLE_API_KEY` or `GEMINI_API_KEY` in:
+ * - `process.env` (Node.js)
+ * - `window.env` (Browser with injected env)
+ *
+ * @param model - The model name (e.g., 'gemini-embedding-001')
+ * @param config - Optional configuration (excluding apiKey which is auto-detected)
+ * @returns Configured Gemini embedding adapter instance with resolved types
+ * @throws Error if GOOGLE_API_KEY or GEMINI_API_KEY is not found in environment
+ *
+ * @example
+ * ```typescript
+ * // Automatically uses GOOGLE_API_KEY from environment
+ * const adapter = geminiEmbedding('gemini-embedding-001');
+ *
+ * const result = await embed({
+ *   adapter,
+ *   input: ['a red guitar', 'a blue drum kit'],
+ *   dimensions: 1536
+ * });
+ *
+ * console.log(result.embeddings[0].vector)
+ * ```
+ */
+export declare function geminiEmbedding<TModel extends GeminiEmbeddingModel>(model: TModel, config?: Omit<GeminiEmbeddingConfig, 'apiKey'>): GeminiEmbeddingAdapter<TModel>;

package/dist/esm/adapters/embedding.js

@@ -0,0 +1,119 @@
+import { createGeminiClient, generateId, getGeminiApiKeyFromEnv } from "../utils/client.js";
+import "../utils/index.js";
+import { requireTextOnlyEmbeddingInput } from "@tanstack/ai";
+import { toRunErrorPayload } from "@tanstack/ai/adapter-internals";
+import { BaseEmbeddingAdapter } from "@tanstack/ai/adapters";
+//#region src/adapters/embedding.ts
+/**
+* Gemini Embedding Adapter
+*
+* Tree-shakeable adapter for Gemini text embeddings.
+* Supports gemini-embedding-001.
+*
+* Features:
+* - Batch embedding (one request for the whole input array)
+* - Matryoshka dimension reduction via the top-level `dimensions` option
+*   (mapped to the SDK's `outputDimensionality`)
+* - Task-type hints (`taskType`, `title`) via provider options
+*/
+var GeminiEmbeddingAdapter = class extends BaseEmbeddingAdapter {
+	name = "gemini";
+	client;
+	constructor(config, model) {
+		super(model, {});
+		this.client = createGeminiClient(config);
+	}
+	async createEmbeddings(options) {
+		const { model, logger, modelOptions } = options;
+		const texts = requireTextOnlyEmbeddingInput(options.input, this.name, model);
+		try {
+			const config = {};
+			if (options.dimensions !== void 0) config.outputDimensionality = options.dimensions;
+			if (modelOptions?.taskType !== void 0) config.taskType = modelOptions.taskType;
+			if (modelOptions?.title !== void 0) config.title = modelOptions.title;
+			logger.request(`activity=embed provider=${this.name} model=${model} inputs=${texts.length}`, {
+				provider: this.name,
+				model
+			});
+			const embeddings = (await this.client.models.embedContent({
+				model,
+				contents: texts,
+				config
+			})).embeddings;
+			if (!embeddings || embeddings.length !== texts.length) throw new Error(`Gemini embedContent returned ${embeddings ? embeddings.length : "no"} embeddings for ${texts.length} inputs (model: ${model}).`);
+			return {
+				id: generateId(this.name),
+				model,
+				embeddings: embeddings.map((embedding, index) => ({
+					vector: embedding.values ?? [],
+					index
+				}))
+			};
+		} catch (error) {
+			logger.errors(`${this.name}.createEmbeddings fatal`, {
+				error: toRunErrorPayload(error, `${this.name}.createEmbeddings failed`),
+				source: `${this.name}.createEmbeddings`
+			});
+			throw error;
+		}
+	}
+};
+/**
+* Creates a Gemini embedding adapter with explicit API key.
+* Type resolution happens here at the call site.
+*
+* @param model - The model name (e.g., 'gemini-embedding-001')
+* @param apiKey - Your Google API key
+* @param config - Optional additional configuration
+* @returns Configured Gemini embedding adapter instance with resolved types
+*
+* @example
+* ```typescript
+* const adapter = createGeminiEmbedding('gemini-embedding-001', "your-api-key");
+*
+* const result = await embed({
+*   adapter,
+*   input: 'a red guitar'
+* });
+* ```
+*/
+function createGeminiEmbedding(model, apiKey, config) {
+	return new GeminiEmbeddingAdapter({
+		...config,
+		apiKey
+	}, model);
+}
+/**
+* Creates a Gemini embedding adapter with automatic API key detection from environment variables.
+* Type resolution happens here at the call site.
+*
+* Looks for `GOOGLE_API_KEY` or `GEMINI_API_KEY` in:
+* - `process.env` (Node.js)
+* - `window.env` (Browser with injected env)
+*
+* @param model - The model name (e.g., 'gemini-embedding-001')
+* @param config - Optional configuration (excluding apiKey which is auto-detected)
+* @returns Configured Gemini embedding adapter instance with resolved types
+* @throws Error if GOOGLE_API_KEY or GEMINI_API_KEY is not found in environment
+*
+* @example
+* ```typescript
+* // Automatically uses GOOGLE_API_KEY from environment
+* const adapter = geminiEmbedding('gemini-embedding-001');
+*
+* const result = await embed({
+*   adapter,
+*   input: ['a red guitar', 'a blue drum kit'],
+*   dimensions: 1536
+* });
+*
+* console.log(result.embeddings[0].vector)
+* ```
+*/
+function geminiEmbedding(model, config) {
+	return createGeminiEmbedding(model, getGeminiApiKeyFromEnv(), config);
+}
+//#endregion
+export { GeminiEmbeddingAdapter, createGeminiEmbedding, geminiEmbedding };
+
+//# sourceMappingURL=embedding.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"embedding.js","names":[],"sources":["../../../src/adapters/embedding.ts"],"sourcesContent":["import { BaseEmbeddingAdapter } from '@tanstack/ai/adapters'\nimport { toRunErrorPayload } from '@tanstack/ai/adapter-internals'\nimport { requireTextOnlyEmbeddingInput } from '@tanstack/ai'\nimport {\n  createGeminiClient,\n  generateId,\n  getGeminiApiKeyFromEnv,\n} from '../utils'\nimport type { EmbeddingOptions, EmbeddingResult } from '@tanstack/ai'\nimport type { EmbedContentConfig, GoogleGenAI } from '@google/genai'\nimport type {\n  GeminiEmbeddingModel,\n  GeminiEmbeddingModelInputModalitiesByName,\n  GeminiEmbeddingModelProviderOptionsByName,\n} from '../model-meta'\nimport type { GeminiEmbeddingProviderOptions } from '../embedding/embedding-provider-options'\nimport type { GeminiClientConfig } from '../utils/client'\n\n/**\n * Configuration for Gemini Embedding adapter\n */\nexport interface GeminiEmbeddingConfig extends GeminiClientConfig {}\n\n/**\n * Gemini Embedding Adapter\n *\n * Tree-shakeable adapter for Gemini text embeddings.\n * Supports gemini-embedding-001.\n *\n * Features:\n * - Batch embedding (one request for the whole input array)\n * - Matryoshka dimension reduction via the top-level `dimensions` option\n *   (mapped to the SDK's `outputDimensionality`)\n * - Task-type hints (`taskType`, `title`) via provider options\n */\nexport class GeminiEmbeddingAdapter<\n  TModel extends GeminiEmbeddingModel,\n> extends BaseEmbeddingAdapter<\n  TModel,\n  GeminiEmbeddingProviderOptions,\n  GeminiEmbeddingModelProviderOptionsByName,\n  GeminiEmbeddingModelInputModalitiesByName\n> {\n  readonly name = 'gemini' as const\n\n  protected client: GoogleGenAI\n\n  constructor(config: GeminiEmbeddingConfig, model: TModel) {\n    super(model, {})\n    this.client = createGeminiClient(config)\n  }\n\n  async createEmbeddings(\n    options: EmbeddingOptions<GeminiEmbeddingProviderOptions>,\n  ): Promise<EmbeddingResult> {\n    const { model, logger, modelOptions } = options\n    const texts = requireTextOnlyEmbeddingInput(options.input, this.name, model)\n\n    try {\n      // Build the config incrementally so possibly-undefined values are never\n      // assigned to optional fields (exactOptionalPropertyTypes).\n      const config: EmbedContentConfig = {}\n      if (options.dimensions !== undefined) {\n        config.outputDimensionality = options.dimensions\n      }\n      if (modelOptions?.taskType !== undefined) {\n        config.taskType = modelOptions.taskType\n      }\n      if (modelOptions?.title !== undefined) {\n        config.title = modelOptions.title\n      }\n\n      logger.request(\n        `activity=embed provider=${this.name} model=${model} inputs=${texts.length}`,\n        { provider: this.name, model },\n      )\n\n      const response = await this.client.models.embedContent({\n        model,\n        contents: texts,\n        config,\n      })\n\n      const embeddings = response.embeddings\n      if (!embeddings || embeddings.length !== texts.length) {\n        throw new Error(\n          `Gemini embedContent returned ${embeddings ? embeddings.length : 'no'} embeddings for ${texts.length} inputs (model: ${model}).`,\n        )\n      }\n\n      // The Gemini embedding API does not report token usage, so `usage` is\n      // omitted from the result.\n      return {\n        id: generateId(this.name),\n        model,\n        embeddings: embeddings.map((embedding, index) => ({\n          vector: embedding.values ?? [],\n          index,\n        })),\n      }\n    } catch (error: unknown) {\n      logger.errors(`${this.name}.createEmbeddings fatal`, {\n        error: toRunErrorPayload(error, `${this.name}.createEmbeddings failed`),\n        source: `${this.name}.createEmbeddings`,\n      })\n      throw error\n    }\n  }\n}\n\n/**\n * Creates a Gemini embedding adapter with explicit API key.\n * Type resolution happens here at the call site.\n *\n * @param model - The model name (e.g., 'gemini-embedding-001')\n * @param apiKey - Your Google API key\n * @param config - Optional additional configuration\n * @returns Configured Gemini embedding adapter instance with resolved types\n *\n * @example\n * ```typescript\n * const adapter = createGeminiEmbedding('gemini-embedding-001', \"your-api-key\");\n *\n * const result = await embed({\n *   adapter,\n *   input: 'a red guitar'\n * });\n * ```\n */\nexport function createGeminiEmbedding<TModel extends GeminiEmbeddingModel>(\n  model: TModel,\n  apiKey: string,\n  config?: Omit<GeminiEmbeddingConfig, 'apiKey'>,\n): GeminiEmbeddingAdapter<TModel> {\n  // Put apiKey LAST so caller-supplied config can't silently override the\n  // explicit argument.\n  return new GeminiEmbeddingAdapter({ ...config, apiKey }, model)\n}\n\n/**\n * Creates a Gemini embedding 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., 'gemini-embedding-001')\n * @param config - Optional configuration (excluding apiKey which is auto-detected)\n * @returns Configured Gemini embedding 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 = geminiEmbedding('gemini-embedding-001');\n *\n * const result = await embed({\n *   adapter,\n *   input: ['a red guitar', 'a blue drum kit'],\n *   dimensions: 1536\n * });\n *\n * console.log(result.embeddings[0].vector)\n * ```\n */\nexport function geminiEmbedding<TModel extends GeminiEmbeddingModel>(\n  model: TModel,\n  config?: Omit<GeminiEmbeddingConfig, 'apiKey'>,\n): GeminiEmbeddingAdapter<TModel> {\n  const apiKey = getGeminiApiKeyFromEnv()\n  return createGeminiEmbedding(model, apiKey, config)\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;AAmCA,IAAa,yBAAb,cAEU,qBAKR;CACA,OAAgB;CAEhB;CAEA,YAAY,QAA+B,OAAe;EACxD,MAAM,OAAO,CAAC,CAAC;EACf,KAAK,SAAS,mBAAmB,MAAM;CACzC;CAEA,MAAM,iBACJ,SAC0B;EAC1B,MAAM,EAAE,OAAO,QAAQ,iBAAiB;EACxC,MAAM,QAAQ,8BAA8B,QAAQ,OAAO,KAAK,MAAM,KAAK;EAE3E,IAAI;GAGF,MAAM,SAA6B,CAAC;GACpC,IAAI,QAAQ,eAAe,KAAA,GACzB,OAAO,uBAAuB,QAAQ;GAExC,IAAI,cAAc,aAAa,KAAA,GAC7B,OAAO,WAAW,aAAa;GAEjC,IAAI,cAAc,UAAU,KAAA,GAC1B,OAAO,QAAQ,aAAa;GAG9B,OAAO,QACL,2BAA2B,KAAK,KAAK,SAAS,MAAM,UAAU,MAAM,UACpE;IAAE,UAAU,KAAK;IAAM;GAAM,CAC/B;GAQA,MAAM,cAAa,MANI,KAAK,OAAO,OAAO,aAAa;IACrD;IACA,UAAU;IACV;GACF,CAAC,EAAA,CAE2B;GAC5B,IAAI,CAAC,cAAc,WAAW,WAAW,MAAM,QAC7C,MAAM,IAAI,MACR,gCAAgC,aAAa,WAAW,SAAS,KAAK,kBAAkB,MAAM,OAAO,kBAAkB,MAAM,GAC/H;GAKF,OAAO;IACL,IAAI,WAAW,KAAK,IAAI;IACxB;IACA,YAAY,WAAW,KAAK,WAAW,WAAW;KAChD,QAAQ,UAAU,UAAU,CAAC;KAC7B;IACF,EAAE;GACJ;EACF,SAAS,OAAgB;GACvB,OAAO,OAAO,GAAG,KAAK,KAAK,0BAA0B;IACnD,OAAO,kBAAkB,OAAO,GAAG,KAAK,KAAK,yBAAyB;IACtE,QAAQ,GAAG,KAAK,KAAK;GACvB,CAAC;GACD,MAAM;EACR;CACF;AACF;;;;;;;;;;;;;;;;;;;;AAqBA,SAAgB,sBACd,OACA,QACA,QACgC;CAGhC,OAAO,IAAI,uBAAuB;EAAE,GAAG;EAAQ;CAAO,GAAG,KAAK;AAChE;;;;;;;;;;;;;;;;;;;;;;;;;;;;AA6BA,SAAgB,gBACd,OACA,QACgC;CAEhC,OAAO,sBAAsB,OADd,uBACqB,GAAQ,MAAM;AACpD"}