@warlock.js/ai-google 5.29.0 → 5.31.0

package/esm/image.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"image.mjs","names":[],"sources":["../../../../../../ai-google/src/image.ts"],"sourcesContent":["import {\n  ContentFilterError,\n  ProviderError,\n  type GeneratedImage,\n  type ImageGenerationOptions,\n  type ImageGenerationResponse,\n  type ImageModelContract,\n  type ImageModelPricing,\n} from \"@warlock.js/ai\";\nimport { log, type Logger } from \"@warlock.js/logger\";\nimport type { GenerateImagesConfig, GoogleGenAI } from \"@google/genai\";\nimport type { GoogleImageConfig } from \"./config.type\";\nimport { wrapGoogleError } from \"./utils\";\n\nconst LOG_MODULE = \"ai.google\";\n\n/** Map a neutral output container hint to an IANA media type. */\nfunction mediaTypeFor(format: string | undefined): string | undefined {\n  switch (format) {\n    case \"png\":\n      return \"image/png\";\n    case \"jpeg\":\n    case \"jpg\":\n      return \"image/jpeg\";\n    case \"webp\":\n      return \"image/webp\";\n    default:\n      return undefined;\n  }\n}\n\n/**\n * Google Imagen-backed implementation of `ImageModelContract`, via\n * `ai.models.generateImages`. Imagen is per-image-metered and returns\n * base64 image bytes (no hosted URL, no token usage).\n *\n * **No model-id validation.** `config.name` is passed through to\n * `ai.models.generateImages` exactly as given — the constructor never\n * inspects it. Google adds and retires image model ids on its own\n * schedule, so an id this adapter does not recognize is not the\n * adapter's call to refuse; an unsupported id surfaces as a provider\n * error from Google (wrapped into the typed `AIError` hierarchy by\n * `generate()`), not as a local one.\n *\n * **Safety filtering.** When Imagen filters every candidate for safety\n * (`raiFilteredReason`), this surfaces a typed `ContentFilterError`\n * carrying the reason, rather than returning an empty success.\n *\n * @example\n * const model = new GoogleImageModel(ai, { name: \"imagen-4.0-generate-001\" }, \"google\");\n * const { images } = await model.generate(\"a watercolor lighthouse at dawn\");\n */\nexport class GoogleImageModel implements ImageModelContract {\n  public readonly name: string;\n  public readonly provider: string;\n  public readonly pricing?: ImageModelPricing;\n\n  private readonly ai: GoogleGenAI;\n  private readonly logger: Logger = log;\n\n  public constructor(ai: GoogleGenAI, config: GoogleImageConfig, provider: string = \"google\") {\n    this.ai = ai;\n    this.name = config.name;\n    this.provider = provider;\n    this.pricing = config.pricing;\n  }\n\n  public async generate(\n    prompt: string,\n    options?: ImageGenerationOptions,\n  ): Promise<ImageGenerationResponse> {\n    const config: GenerateImagesConfig = {};\n\n    if (options?.count !== undefined) config.numberOfImages = options.count;\n    if (options?.aspectRatio !== undefined) config.aspectRatio = options.aspectRatio;\n    if (options?.negativePrompt !== undefined) config.negativePrompt = options.negativePrompt;\n    if (options?.signal !== undefined) config.abortSignal = options.signal;\n\n    const outputMimeType = mediaTypeFor(options?.format);\n    if (outputMimeType !== undefined) config.outputMimeType = outputMimeType;\n\n    // Imagen sizing is `imageSize` (\"1K\"/\"2K\") — a distinct concept from\n    // OpenAI's WxH `size`, so we honor only an explicit passthrough.\n    if (typeof options?.imageSize === \"string\") config.imageSize = options.imageSize;\n    if (typeof options?.personGeneration === \"string\") {\n      config.personGeneration = options.personGeneration as GenerateImagesConfig[\"personGeneration\"];\n    }\n\n    this.logger.debug(LOG_MODULE, \"image.request\", \"models.generateImages\", {\n      model: this.name,\n      count: options?.count ?? 1,\n    });\n\n    let response: Awaited<ReturnType<GoogleGenAI[\"models\"][\"generateImages\"]>>;\n\n    try {\n      response = await this.ai.models.generateImages({ model: this.name, prompt, config });\n    } catch (thrown) {\n      const wrapped = wrapGoogleError(thrown);\n\n      this.logger.error(LOG_MODULE, \"image.error\", wrapped.message, {\n        code: wrapped.code,\n        context: wrapped.context,\n      });\n\n      throw wrapped;\n    }\n\n    const generated = response.generatedImages ?? [];\n    const images: GeneratedImage[] = [];\n\n    for (const candidate of generated) {\n      const bytes = candidate.image?.imageBytes;\n      if (!bytes) continue;\n\n      images.push({\n        type: \"base64\",\n        base64: bytes,\n        mediaType: candidate.image?.mimeType ?? outputMimeType ?? \"image/png\",\n        ...(candidate.enhancedPrompt ? { revisedPrompt: candidate.enhancedPrompt } : {}),\n      });\n    }\n\n    if (images.length === 0) {\n      const filtered = generated.find((candidate) => candidate.raiFilteredReason);\n\n      if (filtered?.raiFilteredReason) {\n        throw new ContentFilterError(\n          `Imagen filtered all candidates: ${filtered.raiFilteredReason}`,\n          { reason: filtered.raiFilteredReason },\n        );\n      }\n\n      throw new ProviderError(\"Imagen returned no images.\");\n    }\n\n    this.logger.debug(LOG_MODULE, \"image.response\", \"models.generateImages succeeded\", {\n      images: images.length,\n    });\n\n    // Imagen returns no token usage — honest zero (priced per image).\n    return { images, usage: { input: 0, output: 0, total: 0 } };\n  }\n}\n"],"mappings":";;;;;;AAcA,MAAM,aAAa;;AAGnB,SAAS,aAAa,QAAgD;CACpE,QAAQ,QAAR;EACE,KAAK,OACH,OAAO;EACT,KAAK;EACL,KAAK,OACH,OAAO;EACT,KAAK,QACH,OAAO;EACT,SACE;CACJ;AACF;;;;;;;;;;;;;;;;;;;;;;AAuBA,IAAa,mBAAb,MAA4D;CAQ1D,AAAO,YAAY,IAAiB,QAA2B,WAAmB,UAAU;gBAF1D;EAGhC,KAAK,KAAK;EACV,KAAK,OAAO,OAAO;EACnB,KAAK,WAAW;EAChB,KAAK,UAAU,OAAO;CACxB;CAEA,MAAa,SACX,QACA,SACkC;EAClC,MAAM,SAA+B,CAAC;EAEtC,IAAI,SAAS,UAAU,QAAW,OAAO,iBAAiB,QAAQ;EAClE,IAAI,SAAS,gBAAgB,QAAW,OAAO,cAAc,QAAQ;EACrE,IAAI,SAAS,mBAAmB,QAAW,OAAO,iBAAiB,QAAQ;EAC3E,IAAI,SAAS,WAAW,QAAW,OAAO,cAAc,QAAQ;EAEhE,MAAM,iBAAiB,aAAa,SAAS,MAAM;EACnD,IAAI,mBAAmB,QAAW,OAAO,iBAAiB;EAI1D,IAAI,OAAO,SAAS,cAAc,UAAU,OAAO,YAAY,QAAQ;EACvE,IAAI,OAAO,SAAS,qBAAqB,UACvC,OAAO,mBAAmB,QAAQ;EAGpC,KAAK,OAAO,MAAM,YAAY,iBAAiB,yBAAyB;GACtE,OAAO,KAAK;GACZ,OAAO,SAAS,SAAS;EAC3B,CAAC;EAED,IAAI;EAEJ,IAAI;GACF,WAAW,MAAM,KAAK,GAAG,OAAO,eAAe;IAAE,OAAO,KAAK;IAAM;IAAQ;GAAO,CAAC;EACrF,SAAS,QAAQ;GACf,MAAM,UAAU,gBAAgB,MAAM;GAEtC,KAAK,OAAO,MAAM,YAAY,eAAe,QAAQ,SAAS;IAC5D,MAAM,QAAQ;IACd,SAAS,QAAQ;GACnB,CAAC;GAED,MAAM;EACR;EAEA,MAAM,YAAY,SAAS,mBAAmB,CAAC;EAC/C,MAAM,SAA2B,CAAC;EAElC,KAAK,MAAM,aAAa,WAAW;GACjC,MAAM,QAAQ,UAAU,OAAO;GAC/B,IAAI,CAAC,OAAO;GAEZ,OAAO,KAAK;IACV,MAAM;IACN,QAAQ;IACR,WAAW,UAAU,OAAO,YAAY,kBAAkB;IAC1D,GAAI,UAAU,iBAAiB,EAAE,eAAe,UAAU,eAAe,IAAI,CAAC;GAChF,CAAC;EACH;EAEA,IAAI,OAAO,WAAW,GAAG;GACvB,MAAM,WAAW,UAAU,MAAM,cAAc,UAAU,iBAAiB;GAE1E,IAAI,UAAU,mBACZ,MAAM,IAAI,mBACR,mCAAmC,SAAS,qBAC5C,EAAE,QAAQ,SAAS,kBAAkB,CACvC;GAGF,MAAM,IAAI,cAAc,4BAA4B;EACtD;EAEA,KAAK,OAAO,MAAM,YAAY,kBAAkB,mCAAmC,EACjF,QAAQ,OAAO,OACjB,CAAC;EAGD,OAAO;GAAE;GAAQ,OAAO;IAAE,OAAO;IAAG,QAAQ;IAAG,OAAO;GAAE;EAAE;CAC5D;AACF"}
+{"version":3,"file":"image.mjs","names":[],"sources":["../../../../../../../packages/ai-google/src/image.ts"],"sourcesContent":["import {\n  ContentFilterError,\n  ProviderError,\n  type GeneratedImage,\n  type ImageGenerationOptions,\n  type ImageGenerationResponse,\n  type ImageModelContract,\n  type ImageModelPricing,\n} from \"@warlock.js/ai\";\nimport { log, type Logger } from \"@warlock.js/logger\";\nimport type { GenerateImagesConfig, GoogleGenAI } from \"@google/genai\";\nimport type { GoogleImageConfig } from \"./config.type\";\nimport { wrapGoogleError } from \"./utils\";\n\nconst LOG_MODULE = \"ai.google\";\n\n/** Map a neutral output container hint to an IANA media type. */\nfunction mediaTypeFor(format: string | undefined): string | undefined {\n  switch (format) {\n    case \"png\":\n      return \"image/png\";\n    case \"jpeg\":\n    case \"jpg\":\n      return \"image/jpeg\";\n    case \"webp\":\n      return \"image/webp\";\n    default:\n      return undefined;\n  }\n}\n\n/**\n * Google Imagen-backed implementation of `ImageModelContract`, via\n * `ai.models.generateImages`. Imagen is per-image-metered and returns\n * base64 image bytes (no hosted URL, no token usage).\n *\n * **No model-id validation.** `config.name` is passed through to\n * `ai.models.generateImages` exactly as given — the constructor never\n * inspects it. Google adds and retires image model ids on its own\n * schedule, so an id this adapter does not recognize is not the\n * adapter's call to refuse; an unsupported id surfaces as a provider\n * error from Google (wrapped into the typed `AIError` hierarchy by\n * `generate()`), not as a local one.\n *\n * **Safety filtering.** When Imagen filters every candidate for safety\n * (`raiFilteredReason`), this surfaces a typed `ContentFilterError`\n * carrying the reason, rather than returning an empty success.\n *\n * @example\n * const model = new GoogleImageModel(ai, { name: \"imagen-4.0-generate-001\" }, \"google\");\n * const { images } = await model.generate(\"a watercolor lighthouse at dawn\");\n */\nexport class GoogleImageModel implements ImageModelContract {\n  public readonly name: string;\n  public readonly provider: string;\n  public readonly pricing?: ImageModelPricing;\n\n  private readonly ai: GoogleGenAI;\n  private readonly logger: Logger = log;\n\n  public constructor(ai: GoogleGenAI, config: GoogleImageConfig, provider: string = \"google\") {\n    this.ai = ai;\n    this.name = config.name;\n    this.provider = provider;\n    this.pricing = config.pricing;\n  }\n\n  public async generate(\n    prompt: string,\n    options?: ImageGenerationOptions,\n  ): Promise<ImageGenerationResponse> {\n    const config: GenerateImagesConfig = {};\n\n    if (options?.count !== undefined) config.numberOfImages = options.count;\n    if (options?.aspectRatio !== undefined) config.aspectRatio = options.aspectRatio;\n    if (options?.negativePrompt !== undefined) config.negativePrompt = options.negativePrompt;\n    if (options?.signal !== undefined) config.abortSignal = options.signal;\n\n    const outputMimeType = mediaTypeFor(options?.format);\n    if (outputMimeType !== undefined) config.outputMimeType = outputMimeType;\n\n    // Imagen sizing is `imageSize` (\"1K\"/\"2K\") — a distinct concept from\n    // OpenAI's WxH `size`, so we honor only an explicit passthrough.\n    if (typeof options?.imageSize === \"string\") config.imageSize = options.imageSize;\n    if (typeof options?.personGeneration === \"string\") {\n      config.personGeneration =\n        options.personGeneration as GenerateImagesConfig[\"personGeneration\"];\n    }\n\n    this.logger.debug(LOG_MODULE, \"image.request\", \"models.generateImages\", {\n      model: this.name,\n      count: options?.count ?? 1,\n    });\n\n    let response: Awaited<ReturnType<GoogleGenAI[\"models\"][\"generateImages\"]>>;\n\n    try {\n      response = await this.ai.models.generateImages({ model: this.name, prompt, config });\n    } catch (thrown) {\n      const wrapped = wrapGoogleError(thrown);\n\n      this.logger.error(LOG_MODULE, \"image.error\", wrapped.message, {\n        code: wrapped.code,\n        context: wrapped.context,\n      });\n\n      throw wrapped;\n    }\n\n    const generated = response.generatedImages ?? [];\n    const images: GeneratedImage[] = [];\n\n    for (const candidate of generated) {\n      const bytes = candidate.image?.imageBytes;\n      if (!bytes) continue;\n\n      images.push({\n        type: \"base64\",\n        base64: bytes,\n        mediaType: candidate.image?.mimeType ?? outputMimeType ?? \"image/png\",\n        ...(candidate.enhancedPrompt ? { revisedPrompt: candidate.enhancedPrompt } : {}),\n      });\n    }\n\n    if (images.length === 0) {\n      const filtered = generated.find((candidate) => candidate.raiFilteredReason);\n\n      if (filtered?.raiFilteredReason) {\n        throw new ContentFilterError(\n          `Imagen filtered all candidates: ${filtered.raiFilteredReason}`,\n          { reason: filtered.raiFilteredReason },\n        );\n      }\n\n      throw new ProviderError(\"Imagen returned no images.\");\n    }\n\n    this.logger.debug(LOG_MODULE, \"image.response\", \"models.generateImages succeeded\", {\n      images: images.length,\n    });\n\n    // Imagen returns no token usage — honest zero (priced per image).\n    return { images, usage: { input: 0, output: 0, total: 0 } };\n  }\n}\n"],"mappings":";;;;;;AAcA,MAAM,aAAa;;AAGnB,SAAS,aAAa,QAAgD;CACpE,QAAQ,QAAR;EACE,KAAK,OACH,OAAO;EACT,KAAK;EACL,KAAK,OACH,OAAO;EACT,KAAK,QACH,OAAO;EACT,SACE;CACJ;AACF;;;;;;;;;;;;;;;;;;;;;;AAuBA,IAAa,mBAAb,MAA4D;CAQ1D,AAAO,YAAY,IAAiB,QAA2B,WAAmB,UAAU;gBAF1D;EAGhC,KAAK,KAAK;EACV,KAAK,OAAO,OAAO;EACnB,KAAK,WAAW;EAChB,KAAK,UAAU,OAAO;CACxB;CAEA,MAAa,SACX,QACA,SACkC;EAClC,MAAM,SAA+B,CAAC;EAEtC,IAAI,SAAS,UAAU,QAAW,OAAO,iBAAiB,QAAQ;EAClE,IAAI,SAAS,gBAAgB,QAAW,OAAO,cAAc,QAAQ;EACrE,IAAI,SAAS,mBAAmB,QAAW,OAAO,iBAAiB,QAAQ;EAC3E,IAAI,SAAS,WAAW,QAAW,OAAO,cAAc,QAAQ;EAEhE,MAAM,iBAAiB,aAAa,SAAS,MAAM;EACnD,IAAI,mBAAmB,QAAW,OAAO,iBAAiB;EAI1D,IAAI,OAAO,SAAS,cAAc,UAAU,OAAO,YAAY,QAAQ;EACvE,IAAI,OAAO,SAAS,qBAAqB,UACvC,OAAO,mBACL,QAAQ;EAGZ,KAAK,OAAO,MAAM,YAAY,iBAAiB,yBAAyB;GACtE,OAAO,KAAK;GACZ,OAAO,SAAS,SAAS;EAC3B,CAAC;EAED,IAAI;EAEJ,IAAI;GACF,WAAW,MAAM,KAAK,GAAG,OAAO,eAAe;IAAE,OAAO,KAAK;IAAM;IAAQ;GAAO,CAAC;EACrF,SAAS,QAAQ;GACf,MAAM,UAAU,gBAAgB,MAAM;GAEtC,KAAK,OAAO,MAAM,YAAY,eAAe,QAAQ,SAAS;IAC5D,MAAM,QAAQ;IACd,SAAS,QAAQ;GACnB,CAAC;GAED,MAAM;EACR;EAEA,MAAM,YAAY,SAAS,mBAAmB,CAAC;EAC/C,MAAM,SAA2B,CAAC;EAElC,KAAK,MAAM,aAAa,WAAW;GACjC,MAAM,QAAQ,UAAU,OAAO;GAC/B,IAAI,CAAC,OAAO;GAEZ,OAAO,KAAK;IACV,MAAM;IACN,QAAQ;IACR,WAAW,UAAU,OAAO,YAAY,kBAAkB;IAC1D,GAAI,UAAU,iBAAiB,EAAE,eAAe,UAAU,eAAe,IAAI,CAAC;GAChF,CAAC;EACH;EAEA,IAAI,OAAO,WAAW,GAAG;GACvB,MAAM,WAAW,UAAU,MAAM,cAAc,UAAU,iBAAiB;GAE1E,IAAI,UAAU,mBACZ,MAAM,IAAI,mBACR,mCAAmC,SAAS,qBAC5C,EAAE,QAAQ,SAAS,kBAAkB,CACvC;GAGF,MAAM,IAAI,cAAc,4BAA4B;EACtD;EAEA,KAAK,OAAO,MAAM,YAAY,kBAAkB,mCAAmC,EACjF,QAAQ,OAAO,OACjB,CAAC;EAGD,OAAO;GAAE;GAAQ,OAAO;IAAE,OAAO;IAAG,QAAQ;IAAG,OAAO;GAAE;EAAE;CAC5D;AACF"}

package/esm/index.d.mts

@@ -1,5 +1,7 @@
 import { GoogleEmbedderConfig, GoogleImageConfig, GoogleModelConfig, GoogleSDKConfig } from "./config.type.mjs";
 import { GoogleSDK } from "./sdk.mjs";
+import { GOOGLE_MODEL_PRICING, googlePricingFor } from "./known-pricing.mjs";
 import { GeminiImageModel } from "./gemini-image.mjs";
 import { GoogleImageModel } from "./image.mjs";
-export { GeminiImageModel, type GoogleEmbedderConfig, type GoogleImageConfig, GoogleImageModel, type GoogleModelConfig, GoogleSDK, type GoogleSDKConfig };
+import { GoogleCallOptions } from "./provider-options.mjs";
+export { GOOGLE_MODEL_PRICING, GeminiImageModel, type GoogleCallOptions, type GoogleEmbedderConfig, type GoogleImageConfig, GoogleImageModel, type GoogleModelConfig, GoogleSDK, type GoogleSDKConfig, googlePricingFor };

package/esm/index.mjs

@@ -1,5 +1,6 @@
 import { GeminiImageModel } from "./gemini-image.mjs";
 import { GoogleImageModel } from "./image.mjs";
+import { GOOGLE_MODEL_PRICING, googlePricingFor } from "./known-pricing.mjs";
 import { GoogleSDK } from "./sdk.mjs";
 
-export { GeminiImageModel, GoogleImageModel, GoogleSDK };
+export { GOOGLE_MODEL_PRICING, GeminiImageModel, GoogleImageModel, GoogleSDK, googlePricingFor };

package/esm/known-pricing.d.mts

@@ -0,0 +1,33 @@
+import { ModelPricing } from "@warlock.js/ai";
+
+//#region ../../packages/ai-google/src/known-pricing.d.ts
+/**
+ * Built-in Gemini API list prices (paid tier, Standard), USD per
+ * 1,000,000 tokens, keyed by exact model id. `cachedInput` is the
+ * context-cache read rate; the hourly cache-storage charge is not a token
+ * rate and is never included.
+ *
+ * Source: https://ai.google.dev/gemini-api/docs/pricing, fetched
+ * 2026-10-17. Only models with one flat token price are listed. Left out
+ * on purpose — pass `pricing` yourself for them:
+ *
+ * - Models priced by prompt length (the Pro models, split at 200k
+ *   prompt tokens).
+ * - Models with a higher audio-input rate (most Flash / Flash-Lite
+ *   models): `Usage` does not split input by modality.
+ * - Models with a scheduled price change (gemini-4-argon, the 3.8 / 3.6
+ *   Flash models).
+ */
+declare const GOOGLE_MODEL_PRICING: Readonly<Record<string, ModelPricing>>;
+/**
+ * Look up the built-in list price for a Gemini model id (exact match),
+ * or `undefined` so cost stays unknown rather than wrong.
+ *
+ * @example
+ * googlePricingFor("gemini-3.5-flash-lite"); // { input: 0.3, output: 2.5, cachedInput: 0.03 }
+ * googlePricingFor("gemini-2.5-pro"); // undefined — priced by prompt length
+ */
+declare function googlePricingFor(modelName: string): ModelPricing | undefined;
+//#endregion
+export { GOOGLE_MODEL_PRICING, googlePricingFor };
+//# sourceMappingURL=known-pricing.d.mts.map

package/esm/known-pricing.d.mts.map

@@ -0,0 +1 @@
+{"version":3,"file":"known-pricing.d.mts","names":[],"sources":["../../../../../../../packages/ai-google/src/known-pricing.ts"],"mappings":";;;;;AAmBA;;;;;;;;;;;;AAAuE;AAYvE;;cAZa,oBAAA,EAAsB,QAAA,CAAS,MAAA,SAAe,YAAA;;AAYM;;;;;;;iBAAjD,gBAAA,CAAiB,SAAA,WAAoB,YAAY"}

package/esm/known-pricing.mjs

@@ -0,0 +1,38 @@
+//#region ../../packages/ai-google/src/known-pricing.ts
+/**
+* Built-in Gemini API list prices (paid tier, Standard), USD per
+* 1,000,000 tokens, keyed by exact model id. `cachedInput` is the
+* context-cache read rate; the hourly cache-storage charge is not a token
+* rate and is never included.
+*
+* Source: https://ai.google.dev/gemini-api/docs/pricing, fetched
+* 2026-10-17. Only models with one flat token price are listed. Left out
+* on purpose — pass `pricing` yourself for them:
+*
+* - Models priced by prompt length (the Pro models, split at 200k
+*   prompt tokens).
+* - Models with a higher audio-input rate (most Flash / Flash-Lite
+*   models): `Usage` does not split input by modality.
+* - Models with a scheduled price change (gemini-4-argon, the 3.8 / 3.6
+*   Flash models).
+*/
+const GOOGLE_MODEL_PRICING = { "gemini-3.5-flash-lite": {
+	input: .3,
+	output: 2.5,
+	cachedInput: .03
+} };
+/**
+* Look up the built-in list price for a Gemini model id (exact match),
+* or `undefined` so cost stays unknown rather than wrong.
+*
+* @example
+* googlePricingFor("gemini-3.5-flash-lite"); // { input: 0.3, output: 2.5, cachedInput: 0.03 }
+* googlePricingFor("gemini-2.5-pro"); // undefined — priced by prompt length
+*/
+function googlePricingFor(modelName) {
+	return GOOGLE_MODEL_PRICING[modelName];
+}
+
+//#endregion
+export { GOOGLE_MODEL_PRICING, googlePricingFor };
+//# sourceMappingURL=known-pricing.mjs.map

package/esm/known-pricing.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"known-pricing.mjs","names":[],"sources":["../../../../../../../packages/ai-google/src/known-pricing.ts"],"sourcesContent":["import type { ModelPricing } from \"@warlock.js/ai\";\n\n/**\n * Built-in Gemini API list prices (paid tier, Standard), USD per\n * 1,000,000 tokens, keyed by exact model id. `cachedInput` is the\n * context-cache read rate; the hourly cache-storage charge is not a token\n * rate and is never included.\n *\n * Source: https://ai.google.dev/gemini-api/docs/pricing, fetched\n * 2026-10-17. Only models with one flat token price are listed. Left out\n * on purpose — pass `pricing` yourself for them:\n *\n * - Models priced by prompt length (the Pro models, split at 200k\n *   prompt tokens).\n * - Models with a higher audio-input rate (most Flash / Flash-Lite\n *   models): `Usage` does not split input by modality.\n * - Models with a scheduled price change (gemini-4-argon, the 3.8 / 3.6\n *   Flash models).\n */\nexport const GOOGLE_MODEL_PRICING: Readonly<Record<string, ModelPricing>> = {\n  \"gemini-3.5-flash-lite\": { input: 0.3, output: 2.5, cachedInput: 0.03 },\n};\n\n/**\n * Look up the built-in list price for a Gemini model id (exact match),\n * or `undefined` so cost stays unknown rather than wrong.\n *\n * @example\n * googlePricingFor(\"gemini-3.5-flash-lite\"); // { input: 0.3, output: 2.5, cachedInput: 0.03 }\n * googlePricingFor(\"gemini-2.5-pro\"); // undefined — priced by prompt length\n */\nexport function googlePricingFor(modelName: string): ModelPricing | undefined {\n  return GOOGLE_MODEL_PRICING[modelName];\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;AAmBA,MAAa,uBAA+D,EAC1E,yBAAyB;CAAE,OAAO;CAAK,QAAQ;CAAK,aAAa;AAAK,EACxE;;;;;;;;;AAUA,SAAgB,iBAAiB,WAA6C;CAC5E,OAAO,qBAAqB;AAC9B"}

package/esm/known-vision-models.mjs

@@ -1,4 +1,4 @@
-//#region ../ai-google/src/known-vision-models.ts
+//#region ../../packages/ai-google/src/known-vision-models.ts
 /**
 * Substrings identifying Gemini model ids whose family accepts image
 * input (vision).

package/esm/known-vision-models.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"known-vision-models.mjs","names":[],"sources":["../../../../../../ai-google/src/known-vision-models.ts"],"sourcesContent":["/**\n * Substrings identifying Gemini model ids whose family accepts image\n * input (vision).\n *\n * Every Gemini 1.5, 2.x, and 2.5 model is natively multimodal, as is\n * the legacy `gemini-pro-vision`. Only the original text-only\n * `gemini-pro` / `gemini-1.0-pro` is excluded. A substring match\n * tolerates the date/preview suffixes Google appends\n * (`gemini-2.5-flash-preview-05-20`). Override per-model via\n * `google.model({ name, vision: true | false })`.\n */\nconst VISION_CAPABLE_SUBSTRINGS = [\n  \"gemini-1.5\",\n  \"gemini-2\",\n  \"gemini-exp\",\n  \"gemini-pro-vision\",\n  \"gemini-flash\",\n];\n\n/**\n * Infer whether a Gemini model id supports vision based on the known\n * multimodal-family substrings. Unknown ids default to `false` so\n * passing an image attachment to an unsupported model surfaces a\n * clear, agent-side capability error instead of an opaque Gemini 400.\n *\n * @example\n * inferVisionCapability(\"gemini-2.5-flash\");          // → true\n * inferVisionCapability(\"gemini-1.5-pro-002\");        // → true\n * inferVisionCapability(\"gemini-1.0-pro\");            // → false\n * inferVisionCapability(\"text-embedding-004\");        // → false\n */\nexport function inferVisionCapability(modelId: string): boolean {\n  const normalized = modelId.toLowerCase();\n\n  return VISION_CAPABLE_SUBSTRINGS.some((fragment) => normalized.includes(fragment));\n}\n"],"mappings":";;;;;;;;;;;;AAWA,MAAM,4BAA4B;CAChC;CACA;CACA;CACA;CACA;AACF;;;;;;;;;;;;;AAcA,SAAgB,sBAAsB,SAA0B;CAC9D,MAAM,aAAa,QAAQ,YAAY;CAEvC,OAAO,0BAA0B,MAAM,aAAa,WAAW,SAAS,QAAQ,CAAC;AACnF"}
+{"version":3,"file":"known-vision-models.mjs","names":[],"sources":["../../../../../../../packages/ai-google/src/known-vision-models.ts"],"sourcesContent":["/**\n * Substrings identifying Gemini model ids whose family accepts image\n * input (vision).\n *\n * Every Gemini 1.5, 2.x, and 2.5 model is natively multimodal, as is\n * the legacy `gemini-pro-vision`. Only the original text-only\n * `gemini-pro` / `gemini-1.0-pro` is excluded. A substring match\n * tolerates the date/preview suffixes Google appends\n * (`gemini-2.5-flash-preview-05-20`). Override per-model via\n * `google.model({ name, vision: true | false })`.\n */\nconst VISION_CAPABLE_SUBSTRINGS = [\n  \"gemini-1.5\",\n  \"gemini-2\",\n  \"gemini-exp\",\n  \"gemini-pro-vision\",\n  \"gemini-flash\",\n];\n\n/**\n * Infer whether a Gemini model id supports vision based on the known\n * multimodal-family substrings. Unknown ids default to `false` so\n * passing an image attachment to an unsupported model surfaces a\n * clear, agent-side capability error instead of an opaque Gemini 400.\n *\n * @example\n * inferVisionCapability(\"gemini-2.5-flash\");          // → true\n * inferVisionCapability(\"gemini-1.5-pro-002\");        // → true\n * inferVisionCapability(\"gemini-1.0-pro\");            // → false\n * inferVisionCapability(\"text-embedding-004\");        // → false\n */\nexport function inferVisionCapability(modelId: string): boolean {\n  const normalized = modelId.toLowerCase();\n\n  return VISION_CAPABLE_SUBSTRINGS.some((fragment) => normalized.includes(fragment));\n}\n"],"mappings":";;;;;;;;;;;;AAWA,MAAM,4BAA4B;CAChC;CACA;CACA;CACA;CACA;AACF;;;;;;;;;;;;;AAcA,SAAgB,sBAAsB,SAA0B;CAC9D,MAAM,aAAa,QAAQ,YAAY;CAEvC,OAAO,0BAA0B,MAAM,aAAa,WAAW,SAAS,QAAQ,CAAC;AACnF"}

package/esm/model.mjs

@@ -4,11 +4,15 @@ import { toGoogleContents } from "./utils/to-google-contents.mjs";
 import { toGoogleTools } from "./utils/to-google-tools.mjs";
 import { wrapGoogleError } from "./utils/wrap-google-error.mjs";
 import "./utils/index.mjs";
+import { mergeProviderBody } from "./provider-options.mjs";
 import { inferVisionCapability } from "./known-vision-models.mjs";
+import { InvalidRequestError, attachCost, resolveStreamChunks, selectProviderTools } from "@warlock.js/ai";
 import { log } from "@warlock.js/logger";
 
-//#region ../ai-google/src/model.ts
+//#region ../../packages/ai-google/src/model.ts
 const LOG_MODULE = "ai.google";
+/** `provider` value of `ai.providerTool()` entries this adapter sends verbatim. */
+const GOOGLE_WIRE_FORMAT = "google";
 /**
 * Bucketed `thinkingBudget` (token caps) for the neutral
 * `reasoning.effort` levels when the caller gives no explicit
@@ -102,7 +106,7 @@ var GoogleModel = class {
 		}
 		const toolCalls = this.extractToolCalls(response);
 		const finishReason = toolCalls ? "tool_calls" : mapFinishReason(response.candidates?.[0]?.finishReason);
-		const usage = this.extractUsage(response);
+		const usage = attachCost(this.extractUsage(response), this.pricing);
 		this.logger.debug(LOG_MODULE, "response", "generateContent call succeeded", {
 			finishReason,
 			usage
@@ -114,11 +118,39 @@ var GoogleModel = class {
 			toolCalls
 		};
 	}
+	/** Count native input tokens with the same converted content, system and tools. */
+	async countTokens(messages, options) {
+		const { systemInstruction, contents } = toGoogleContents(messages);
+		const config = this.buildConfig(systemInstruction, options);
+		if (!this.ai.vertexai && (config.systemInstruction !== void 0 || config.tools !== void 0)) throw new InvalidRequestError("The Google SDK countTokens endpoint in Developer API mode cannot count system instructions or tools. Use Vertex mode for this preflight, or SDK.count() for an offline estimate.");
+		if (config.cachedContent !== void 0) throw new InvalidRequestError("The Google SDK countTokens endpoint cannot count a cachedContent reference.");
+		try {
+			const response = await this.ai.models.countTokens({
+				model: this.name,
+				contents,
+				config: {
+					...config.systemInstruction !== void 0 ? { systemInstruction: config.systemInstruction } : {},
+					...config.tools !== void 0 ? { tools: config.tools } : {},
+					...config.abortSignal ? { abortSignal: config.abortSignal } : {},
+					...config.httpOptions ? { httpOptions: config.httpOptions } : {}
+				}
+			});
+			if (response.totalTokens === void 0) throw new InvalidRequestError("Google countTokens returned no totalTokens.");
+			return response.totalTokens;
+		} catch (thrown) {
+			throw this.logAndWrap(thrown);
+		}
+	}
 	/**
 	* Incremental streaming completion via `generateContentStream`.
 	* Yields neutral `ModelStreamChunk`s — `delta` for text, `tool-call`
 	* per function call (Gemini emits a fully-formed call, not partial
 	* JSON), and a terminal `done` with the final finish reason + usage.
+	*
+	* With `options.streamChunks` it also yields `reasoning` for thought
+	* parts (only sent when `thinkingConfig.includeThoughts` is on — set it
+	* through `providerOptions.google.body`) and `usage` whenever a chunk
+	* carries `usageMetadata`. No `tool-args`: calls arrive whole.
 	*/
 	async *stream(messages, options) {
 		this.logger.debug(LOG_MODULE, "request", "Starting generateContentStream call", {
@@ -145,8 +177,15 @@ var GoogleModel = class {
 			output: 0,
 			total: 0
 		};
+		const want = resolveStreamChunks(options);
 		try {
 			for await (const chunk of iterable) {
+				if (want.reasoning) {
+					for (const part of chunk.candidates?.[0]?.content?.parts ?? []) if (part.thought === true && typeof part.text === "string" && part.text !== "") yield {
+						type: "reasoning",
+						delta: part.text
+					};
+				}
 				const text = chunk.text;
 				if (text) yield {
 					type: "delta",
@@ -166,12 +205,19 @@ var GoogleModel = class {
 				}
 				const candidateFinish = chunk.candidates?.[0]?.finishReason;
 				if (candidateFinish) rawFinishReason = candidateFinish;
-				if (chunk.usageMetadata) applyGoogleUsage(usage, chunk.usageMetadata);
+				if (chunk.usageMetadata) {
+					applyGoogleUsage(usage, chunk.usageMetadata);
+					if (want.usage) yield {
+						type: "usage",
+						usage: { ...usage }
+					};
+				}
 			}
 		} catch (thrown) {
 			throw this.logAndWrap(thrown);
 		}
 		const finishReason = sawToolCall ? "tool_calls" : mapFinishReason(rawFinishReason);
+		attachCost(usage, this.pricing);
 		this.logger.debug(LOG_MODULE, "response", "generateContentStream call succeeded", {
 			finishReason,
 			usage
@@ -191,15 +237,15 @@ var GoogleModel = class {
 	buildConfig(systemInstruction, options) {
 		const temperature = options?.temperature ?? this.config.temperature;
 		const maxOutputTokens = options?.maxTokens ?? this.config.maxTokens;
-		return {
+		return mergeProviderBody({
 			...systemInstruction ? { systemInstruction } : {},
 			...temperature !== void 0 ? { temperature } : {},
 			...maxOutputTokens !== void 0 ? { maxOutputTokens } : {},
 			...options?.signal ? { abortSignal: options.signal } : {},
-			...this.buildTools(options?.tools),
+			...this.buildTools(options),
 			...this.buildStructuredOutput(options?.responseSchema),
 			...this.buildThinking(options?.reasoning)
-		};
+		}, options);
 	}
 	/**
 	* Translate the neutral `reasoning` option into Gemini's
@@ -224,11 +270,21 @@ var GoogleModel = class {
 	}
 	/**
 	* Spread-friendly tools fragment. Empty object when no tools were
-	* supplied so the caller can unconditionally spread it.
+	* supplied so the caller can unconditionally spread it. Function tools
+	* come first; `ai.providerTool({ provider: "google" })` definitions
+	* (`{ googleSearch: {} }`, `{ codeExecution: {} }`, `{ urlContext: {} }`,
+	* …) follow verbatim as their own `Tool` entries. Other providers'
+	* definitions are skipped with a warning.
 	*/
-	buildTools(tools) {
-		const mapped = toGoogleTools(tools);
-		return mapped ? { tools: mapped } : {};
+	buildTools(options) {
+		const { tools, definitions } = selectProviderTools(options, {
+			accepts: [GOOGLE_WIRE_FORMAT, this.provider],
+			module: LOG_MODULE,
+			model: this.name,
+			logger: this.logger
+		});
+		const mapped = [...toGoogleTools(tools) ?? [], ...definitions];
+		return mapped.length > 0 ? { tools: mapped } : {};
 	}
 	/**
 	* Translate the neutral `responseSchema` into Gemini's native JSON

package/esm/model.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"model.mjs","names":[],"sources":["../../../../../../ai-google/src/model.ts"],"sourcesContent":["import {\n  type Message,\n  type ModelCallOptions,\n  type ModelCapabilities,\n  type ModelContract,\n  type ModelPricing,\n  type ModelResponse,\n  type ModelStreamChunk,\n  type ModelToolCallRequest,\n  type ReasoningEffort,\n  type Usage,\n} from \"@warlock.js/ai\";\nimport { log, type Logger } from \"@warlock.js/logger\";\nimport type {\n  GenerateContentConfig,\n  GenerateContentResponse,\n  GoogleGenAI,\n  Part,\n} from \"@google/genai\";\nimport type { GoogleModelConfig } from \"./config.type\";\nimport { inferVisionCapability } from \"./known-vision-models\";\nimport {\n  applyGoogleUsage,\n  mapFinishReason,\n  toGoogleContents,\n  toGoogleTools,\n  wrapGoogleError,\n} from \"./utils\";\n\nconst LOG_MODULE = \"ai.google\";\n\n/**\n * Bucketed `thinkingBudget` (token caps) for the neutral\n * `reasoning.effort` levels when the caller gives no explicit\n * `reasoning.maxTokens`. Gemini 2.5 accepts a positive budget as a cap\n * on the thinking phase; these mirror the spread the OpenAI\n * `reasoning_effort` low/medium/high tiers imply. The installed\n * `@google/genai` `ThinkingConfig` declaration documents that allowed\n * ranges are model-dependent, so retain the existing conservative 24,576\n * cap for the new tiers rather than sending a budget unsupported by Flash.\n */\nconst EFFORT_THINKING_BUDGET: Record<Exclude<ReasoningEffort, \"none\">, number> = {\n  minimal: 1024,\n  low: 1024,\n  medium: 8192,\n  high: 24576,\n  xhigh: 24576,\n  max: 24576,\n};\n\n/**\n * Google Gemini-backed implementation of `ModelContract`.\n *\n * **Role.** The provider-facing bridge between the vendor-neutral\n * `@warlock.js/ai` agent runtime and the `@google/genai` SDK\n * (`models.generateContent` / `generateContentStream`).\n *\n * **Responsibility.**\n * - Owns: a long-lived `GoogleGenAI` client + frozen `ModelConfig`\n *   (name, temperature, maxTokens) used as per-call defaults.\n * - Owns: translating vendor-neutral `Message[]` / `ToolConfig[]` into\n *   Gemini shapes (systemInstruction hoisting, `model` role,\n *   `functionCall` / `functionResponse` parts, inline image bytes) on\n *   the way out, and Gemini's candidate/parts response (text, function\n *   calls, finish reason, token usage) back into neutral shapes on the\n *   way in.\n * - Does NOT own: dispatching tools, looping, history, retries — those\n *   are agent concerns. The model is a per-call protocol adapter.\n *\n * Modeled as a class (see §4.2 of code-style.md — \"long-lived state\n * across calls\"): the `GoogleGenAI` client is reused for the SDK's\n * lifetime.\n *\n * @example\n * import { GoogleGenAI } from \"@google/genai\";\n * const ai = new GoogleGenAI({ apiKey: process.env.GEMINI_API_KEY });\n * const model = new GoogleModel(ai, { name: \"gemini-2.5-flash\" });\n *\n * const myAgent = agent({ model, tools: [searchTool] });\n * const result = await myAgent.execute(\"Summarize today's news.\");\n */\nexport class GoogleModel implements ModelContract {\n  public readonly name: string;\n  public readonly provider: string;\n  public readonly capabilities: ModelCapabilities;\n  public readonly pricing?: ModelPricing;\n\n  private readonly ai: GoogleGenAI;\n  private readonly config: GoogleModelConfig;\n  private readonly logger: Logger = log;\n\n  public constructor(ai: GoogleGenAI, config: GoogleModelConfig, provider: string = \"google\") {\n    this.ai = ai;\n    this.config = config;\n    this.name = config.name;\n    this.provider = provider;\n    this.pricing = config.pricing;\n    const multimodal = config.vision ?? inferVisionCapability(config.name);\n\n    this.capabilities = {\n      structuredOutput: config.structuredOutput ?? true,\n      vision: multimodal,\n      // Every Gemini 2.5 model thinks; older families harmlessly ignore\n      // an empty thinking budget. Defaulting `true` lets the agent\n      // forward reasoning options; an explicit `false` opts a model out.\n      reasoning: config.reasoning ?? true,\n      // Gemini reports cache-read hits (`cachedContentTokenCount`) on\n      // every call via implicit caching, and accepts explicit context\n      // caching. Read-side accounting is always honored.\n      promptCaching: true,\n      // The multimodal Gemini families that accept images also accept\n      // audio and PDF/document parts. Mirror the vision inference unless\n      // explicitly overridden.\n      audio: config.audio ?? multimodal,\n      pdf: config.pdf ?? multimodal,\n    };\n  }\n\n  /**\n   * Single-shot completion. Sends the full message list to\n   * `generateContent`, waits for the terminal response, and reshapes\n   * it into a vendor-neutral `ModelResponse`. Per-call `options`\n   * override the instance defaults for this call only.\n   */\n  public async complete(messages: Message[], options?: ModelCallOptions): Promise<ModelResponse> {\n    this.logger.debug(LOG_MODULE, \"request\", \"Starting generateContent call\", {\n      model: this.name,\n      messageCount: messages.length,\n      streaming: false,\n      toolCount: options?.tools?.length ?? 0,\n    });\n\n    const { systemInstruction, contents } = toGoogleContents(messages);\n\n    let response: GenerateContentResponse;\n\n    try {\n      response = await this.ai.models.generateContent({\n        model: this.name,\n        contents,\n        config: this.buildConfig(systemInstruction, options),\n      });\n    } catch (thrown) {\n      throw this.logAndWrap(thrown);\n    }\n\n    const toolCalls = this.extractToolCalls(response);\n    const finishReason = toolCalls\n      ? \"tool_calls\"\n      : mapFinishReason(response.candidates?.[0]?.finishReason);\n    const usage = this.extractUsage(response);\n\n    this.logger.debug(LOG_MODULE, \"response\", \"generateContent call succeeded\", {\n      finishReason,\n      usage,\n    });\n\n    return {\n      content: response.text ?? \"\",\n      finishReason,\n      usage,\n      toolCalls,\n    };\n  }\n\n  /**\n   * Incremental streaming completion via `generateContentStream`.\n   * Yields neutral `ModelStreamChunk`s — `delta` for text, `tool-call`\n   * per function call (Gemini emits a fully-formed call, not partial\n   * JSON), and a terminal `done` with the final finish reason + usage.\n   */\n  public async *stream(\n    messages: Message[],\n    options?: ModelCallOptions,\n  ): AsyncIterable<ModelStreamChunk> {\n    this.logger.debug(LOG_MODULE, \"request\", \"Starting generateContentStream call\", {\n      model: this.name,\n      messageCount: messages.length,\n      streaming: true,\n      toolCount: options?.tools?.length ?? 0,\n    });\n\n    const { systemInstruction, contents } = toGoogleContents(messages);\n\n    let iterable: AsyncGenerator<GenerateContentResponse>;\n\n    try {\n      iterable = await this.ai.models.generateContentStream({\n        model: this.name,\n        contents,\n        config: this.buildConfig(systemInstruction, options),\n      });\n    } catch (thrown) {\n      throw this.logAndWrap(thrown);\n    }\n\n    let rawFinishReason: string | undefined;\n    let sawToolCall = false;\n    const usage: Usage = { input: 0, output: 0, total: 0 };\n\n    try {\n      for await (const chunk of iterable) {\n        const text = chunk.text;\n\n        if (text) {\n          yield { type: \"delta\", content: text };\n        }\n\n        for (const part of chunk.candidates?.[0]?.content?.parts ?? []) {\n          const toolCall = this.partToToolCall(part);\n\n          if (!toolCall) {\n            continue;\n          }\n\n          sawToolCall = true;\n\n          yield {\n            type: \"tool-call\",\n            id: toolCall.id,\n            name: toolCall.name,\n            input: toolCall.input,\n            ...(toolCall.providerMetadata\n              ? { providerMetadata: toolCall.providerMetadata }\n              : {}),\n          };\n        }\n\n        const candidateFinish = chunk.candidates?.[0]?.finishReason;\n\n        if (candidateFinish) {\n          rawFinishReason = candidateFinish;\n        }\n\n        if (chunk.usageMetadata) {\n          applyGoogleUsage(usage, chunk.usageMetadata);\n        }\n      }\n    } catch (thrown) {\n      throw this.logAndWrap(thrown);\n    }\n\n    const finishReason = sawToolCall ? \"tool_calls\" : mapFinishReason(rawFinishReason);\n\n    this.logger.debug(LOG_MODULE, \"response\", \"generateContentStream call succeeded\", {\n      finishReason,\n      usage,\n    });\n\n    yield { type: \"done\", finishReason, usage };\n  }\n\n  /**\n   * Assemble the `GenerateContentConfig` shared by `complete()` and\n   * `stream()`: inference params, hoisted system instruction,\n   * cancellation signal, and conditional tools + native structured\n   * output.\n   */\n  private buildConfig(\n    systemInstruction: string | undefined,\n    options: ModelCallOptions | undefined,\n  ): GenerateContentConfig {\n    const temperature = options?.temperature ?? this.config.temperature;\n    const maxOutputTokens = options?.maxTokens ?? this.config.maxTokens;\n\n    return {\n      ...(systemInstruction ? { systemInstruction } : {}),\n      ...(temperature !== undefined ? { temperature } : {}),\n      ...(maxOutputTokens !== undefined ? { maxOutputTokens } : {}),\n      ...(options?.signal ? { abortSignal: options.signal } : {}),\n      ...this.buildTools(options?.tools),\n      ...this.buildStructuredOutput(options?.responseSchema),\n      ...this.buildThinking(options?.reasoning),\n    };\n  }\n\n  /**\n   * Translate the neutral `reasoning` option into Gemini's\n   * `thinkingConfig`. `reasoning.maxTokens` maps directly to\n   * `thinkingBudget` (token cap on the thinking phase); when only\n   * `reasoning.effort` is given it is bucketed into a budget. Emitted\n   * only when the model is `reasoning`-capable — a `false` capability\n   * (config override) drops it so a non-thinking model never receives\n   * an unsupported `thinkingConfig`.\n   *\n   * Gemini's `thinkingBudget` semantics: `0` disables thinking, `-1`\n   * lets the model decide automatically. A positive value caps the\n   * thinking tokens. The neutral `effort: \"none\"` (\"run without\n   * reasoning\") maps to `thinkingBudget: 0`.\n   */\n  private buildThinking(\n    reasoning: ModelCallOptions[\"reasoning\"],\n  ): Pick<GenerateContentConfig, \"thinkingConfig\"> {\n    if (!reasoning || !this.capabilities.reasoning) {\n      return {};\n    }\n\n    // `effort: \"none\"` = explicit \"run without reasoning\". Gemini disables\n    // thinking with `thinkingBudget: 0` (its native off switch), so emit\n    // that rather than omitting the config — an omitted config lets a\n    // thinking model reason at its default budget.\n    if (reasoning.effort === \"none\") {\n      return { thinkingConfig: { thinkingBudget: 0 } };\n    }\n\n    const thinkingBudget =\n      reasoning.maxTokens ?? (reasoning.effort ? EFFORT_THINKING_BUDGET[reasoning.effort] : undefined);\n\n    if (thinkingBudget === undefined) {\n      return {};\n    }\n\n    return { thinkingConfig: { thinkingBudget } };\n  }\n\n  /**\n   * Spread-friendly tools fragment. Empty object when no tools were\n   * supplied so the caller can unconditionally spread it.\n   */\n  private buildTools(tools: ModelCallOptions[\"tools\"]): Pick<GenerateContentConfig, \"tools\"> {\n    const mapped = toGoogleTools(tools);\n\n    return mapped ? { tools: mapped } : {};\n  }\n\n  /**\n   * Translate the neutral `responseSchema` into Gemini's native JSON\n   * structured output (`responseMimeType: \"application/json\"` +\n   * `responseJsonSchema`, which takes a raw JSON Schema directly).\n   * Emitted only when the model is `structuredOutput`-capable and the\n   * schema is an object root — otherwise the agent's soft prompt hint\n   * + client-side `validate()` carry shape.\n   */\n  private buildStructuredOutput(\n    responseSchema: Record<string, unknown> | undefined,\n  ): Pick<GenerateContentConfig, \"responseMimeType\" | \"responseJsonSchema\"> {\n    if (!responseSchema || !this.capabilities.structuredOutput) {\n      return {};\n    }\n\n    if (responseSchema.type !== \"object\" || typeof responseSchema.properties !== \"object\") {\n      return {};\n    }\n\n    return {\n      responseMimeType: \"application/json\",\n      responseJsonSchema: responseSchema,\n    };\n  }\n\n  /**\n   * Reshape Gemini's function-call content parts into the neutral\n   * `ModelToolCallRequest[]`. Returns `undefined` when the model\n   * requested no functions so callers can branch on presence.\n   *\n   * Reads `candidates[0].content.parts` directly rather than the\n   * `response.functionCalls` getter: the getter discards the\n   * part-level `thoughtSignature`, and Gemini \"thinking\" models 400\n   * the follow-up turn if that signature is not echoed back. See\n   * `partToToolCall`.\n   */\n  private extractToolCalls(\n    response: GenerateContentResponse,\n  ): ModelToolCallRequest[] | undefined {\n    const parts = response.candidates?.[0]?.content?.parts ?? [];\n    const toolCalls = parts\n      .map((part) => this.partToToolCall(part))\n      .filter((call): call is ModelToolCallRequest => call !== undefined);\n\n    return toolCalls.length > 0 ? toolCalls : undefined;\n  }\n\n  /**\n   * Map a single Gemini `Part` to a neutral `ModelToolCallRequest`,\n   * or `undefined` when the part is not a function call. The part's\n   * `thoughtSignature` (opaque, set by thinking models) is carried on\n   * `providerMetadata` so `toGoogleContents` can replay it on the\n   * assistant turn — Gemini rejects the next request without it.\n   */\n  private partToToolCall(part: Part): ModelToolCallRequest | undefined {\n    if (!part.functionCall) {\n      return undefined;\n    }\n\n    const call = part.functionCall;\n\n    return {\n      // The Gemini Developer API does not assign function-call ids\n      // (only Vertex parallel-calling does). Fall back to the function\n      // name so the neutral `toolCallId` is non-empty and the echoed\n      // `functionResponse.name` resolves — Gemini matches a result to\n      // its call by name. See decisions §49.\n      id: call.id ?? call.name ?? \"\",\n      name: call.name ?? \"\",\n      input: (call.args ?? {}) as Record<string, unknown>,\n      ...(part.thoughtSignature\n        ? { providerMetadata: { thoughtSignature: part.thoughtSignature } }\n        : {}),\n    };\n  }\n\n  /**\n   * Normalize Gemini's `usageMetadata` into the neutral `Usage` shape\n   * via the shared {@link applyGoogleUsage} mapper (the same one the\n   * streaming loop and the Gemini image model use). Absent usage\n   * collapses to zeros.\n   */\n  private extractUsage(response: GenerateContentResponse): Usage {\n    const usage: Usage = { input: 0, output: 0, total: 0 };\n\n    if (response.usageMetadata) {\n      applyGoogleUsage(usage, response.usageMetadata);\n    }\n\n    return usage;\n  }\n\n  /**\n   * Wrap a thrown provider error into the typed `AIError` hierarchy\n   * and emit the standard error log line before it propagates.\n   */\n  private logAndWrap(thrown: unknown) {\n    const wrapped = wrapGoogleError(thrown);\n\n    this.logger.error(LOG_MODULE, \"error\", wrapped.message, {\n      code: wrapped.code,\n      context: wrapped.context,\n    });\n\n    return wrapped;\n  }\n}\n"],"mappings":";;;;;;;;;;AA6BA,MAAM,aAAa;;;;;;;;;;;AAYnB,MAAM,yBAA2E;CAC/E,SAAS;CACT,KAAK;CACL,QAAQ;CACR,MAAM;CACN,OAAO;CACP,KAAK;AACP;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAiCA,IAAa,cAAb,MAAkD;CAUhD,AAAO,YAAY,IAAiB,QAA2B,WAAmB,UAAU;gBAF1D;EAGhC,KAAK,KAAK;EACV,KAAK,SAAS;EACd,KAAK,OAAO,OAAO;EACnB,KAAK,WAAW;EAChB,KAAK,UAAU,OAAO;EACtB,MAAM,aAAa,OAAO,UAAU,sBAAsB,OAAO,IAAI;EAErE,KAAK,eAAe;GAClB,kBAAkB,OAAO,oBAAoB;GAC7C,QAAQ;GAIR,WAAW,OAAO,aAAa;GAI/B,eAAe;GAIf,OAAO,OAAO,SAAS;GACvB,KAAK,OAAO,OAAO;EACrB;CACF;;;;;;;CAQA,MAAa,SAAS,UAAqB,SAAoD;EAC7F,KAAK,OAAO,MAAM,YAAY,WAAW,iCAAiC;GACxE,OAAO,KAAK;GACZ,cAAc,SAAS;GACvB,WAAW;GACX,WAAW,SAAS,OAAO,UAAU;EACvC,CAAC;EAED,MAAM,EAAE,mBAAmB,aAAa,iBAAiB,QAAQ;EAEjE,IAAI;EAEJ,IAAI;GACF,WAAW,MAAM,KAAK,GAAG,OAAO,gBAAgB;IAC9C,OAAO,KAAK;IACZ;IACA,QAAQ,KAAK,YAAY,mBAAmB,OAAO;GACrD,CAAC;EACH,SAAS,QAAQ;GACf,MAAM,KAAK,WAAW,MAAM;EAC9B;EAEA,MAAM,YAAY,KAAK,iBAAiB,QAAQ;EAChD,MAAM,eAAe,YACjB,eACA,gBAAgB,SAAS,aAAa,EAAE,EAAE,YAAY;EAC1D,MAAM,QAAQ,KAAK,aAAa,QAAQ;EAExC,KAAK,OAAO,MAAM,YAAY,YAAY,kCAAkC;GAC1E;GACA;EACF,CAAC;EAED,OAAO;GACL,SAAS,SAAS,QAAQ;GAC1B;GACA;GACA;EACF;CACF;;;;;;;CAQA,OAAc,OACZ,UACA,SACiC;EACjC,KAAK,OAAO,MAAM,YAAY,WAAW,uCAAuC;GAC9E,OAAO,KAAK;GACZ,cAAc,SAAS;GACvB,WAAW;GACX,WAAW,SAAS,OAAO,UAAU;EACvC,CAAC;EAED,MAAM,EAAE,mBAAmB,aAAa,iBAAiB,QAAQ;EAEjE,IAAI;EAEJ,IAAI;GACF,WAAW,MAAM,KAAK,GAAG,OAAO,sBAAsB;IACpD,OAAO,KAAK;IACZ;IACA,QAAQ,KAAK,YAAY,mBAAmB,OAAO;GACrD,CAAC;EACH,SAAS,QAAQ;GACf,MAAM,KAAK,WAAW,MAAM;EAC9B;EAEA,IAAI;EACJ,IAAI,cAAc;EAClB,MAAM,QAAe;GAAE,OAAO;GAAG,QAAQ;GAAG,OAAO;EAAE;EAErD,IAAI;GACF,WAAW,MAAM,SAAS,UAAU;IAClC,MAAM,OAAO,MAAM;IAEnB,IAAI,MACF,MAAM;KAAE,MAAM;KAAS,SAAS;IAAK;IAGvC,KAAK,MAAM,QAAQ,MAAM,aAAa,EAAE,EAAE,SAAS,SAAS,CAAC,GAAG;KAC9D,MAAM,WAAW,KAAK,eAAe,IAAI;KAEzC,IAAI,CAAC,UACH;KAGF,cAAc;KAEd,MAAM;MACJ,MAAM;MACN,IAAI,SAAS;MACb,MAAM,SAAS;MACf,OAAO,SAAS;MAChB,GAAI,SAAS,mBACT,EAAE,kBAAkB,SAAS,iBAAiB,IAC9C,CAAC;KACP;IACF;IAEA,MAAM,kBAAkB,MAAM,aAAa,EAAE,EAAE;IAE/C,IAAI,iBACF,kBAAkB;IAGpB,IAAI,MAAM,eACR,iBAAiB,OAAO,MAAM,aAAa;GAE/C;EACF,SAAS,QAAQ;GACf,MAAM,KAAK,WAAW,MAAM;EAC9B;EAEA,MAAM,eAAe,cAAc,eAAe,gBAAgB,eAAe;EAEjF,KAAK,OAAO,MAAM,YAAY,YAAY,wCAAwC;GAChF;GACA;EACF,CAAC;EAED,MAAM;GAAE,MAAM;GAAQ;GAAc;EAAM;CAC5C;;;;;;;CAQA,AAAQ,YACN,mBACA,SACuB;EACvB,MAAM,cAAc,SAAS,eAAe,KAAK,OAAO;EACxD,MAAM,kBAAkB,SAAS,aAAa,KAAK,OAAO;EAE1D,OAAO;GACL,GAAI,oBAAoB,EAAE,kBAAkB,IAAI,CAAC;GACjD,GAAI,gBAAgB,SAAY,EAAE,YAAY,IAAI,CAAC;GACnD,GAAI,oBAAoB,SAAY,EAAE,gBAAgB,IAAI,CAAC;GAC3D,GAAI,SAAS,SAAS,EAAE,aAAa,QAAQ,OAAO,IAAI,CAAC;GACzD,GAAG,KAAK,WAAW,SAAS,KAAK;GACjC,GAAG,KAAK,sBAAsB,SAAS,cAAc;GACrD,GAAG,KAAK,cAAc,SAAS,SAAS;EAC1C;CACF;;;;;;;;;;;;;;;CAgBA,AAAQ,cACN,WAC+C;EAC/C,IAAI,CAAC,aAAa,CAAC,KAAK,aAAa,WACnC,OAAO,CAAC;EAOV,IAAI,UAAU,WAAW,QACvB,OAAO,EAAE,gBAAgB,EAAE,gBAAgB,EAAE,EAAE;EAGjD,MAAM,iBACJ,UAAU,cAAc,UAAU,SAAS,uBAAuB,UAAU,UAAU;EAExF,IAAI,mBAAmB,QACrB,OAAO,CAAC;EAGV,OAAO,EAAE,gBAAgB,EAAE,eAAe,EAAE;CAC9C;;;;;CAMA,AAAQ,WAAW,OAAwE;EACzF,MAAM,SAAS,cAAc,KAAK;EAElC,OAAO,SAAS,EAAE,OAAO,OAAO,IAAI,CAAC;CACvC;;;;;;;;;CAUA,AAAQ,sBACN,gBACwE;EACxE,IAAI,CAAC,kBAAkB,CAAC,KAAK,aAAa,kBACxC,OAAO,CAAC;EAGV,IAAI,eAAe,SAAS,YAAY,OAAO,eAAe,eAAe,UAC3E,OAAO,CAAC;EAGV,OAAO;GACL,kBAAkB;GAClB,oBAAoB;EACtB;CACF;;;;;;;;;;;;CAaA,AAAQ,iBACN,UACoC;EAEpC,MAAM,aADQ,SAAS,aAAa,EAAE,EAAE,SAAS,SAAS,CAAC,EACpC,CACpB,KAAK,SAAS,KAAK,eAAe,IAAI,CAAC,CAAC,CACxC,QAAQ,SAAuC,SAAS,MAAS;EAEpE,OAAO,UAAU,SAAS,IAAI,YAAY;CAC5C;;;;;;;;CASA,AAAQ,eAAe,MAA8C;EACnE,IAAI,CAAC,KAAK,cACR;EAGF,MAAM,OAAO,KAAK;EAElB,OAAO;GAML,IAAI,KAAK,MAAM,KAAK,QAAQ;GAC5B,MAAM,KAAK,QAAQ;GACnB,OAAQ,KAAK,QAAQ,CAAC;GACtB,GAAI,KAAK,mBACL,EAAE,kBAAkB,EAAE,kBAAkB,KAAK,iBAAiB,EAAE,IAChE,CAAC;EACP;CACF;;;;;;;CAQA,AAAQ,aAAa,UAA0C;EAC7D,MAAM,QAAe;GAAE,OAAO;GAAG,QAAQ;GAAG,OAAO;EAAE;EAErD,IAAI,SAAS,eACX,iBAAiB,OAAO,SAAS,aAAa;EAGhD,OAAO;CACT;;;;;CAMA,AAAQ,WAAW,QAAiB;EAClC,MAAM,UAAU,gBAAgB,MAAM;EAEtC,KAAK,OAAO,MAAM,YAAY,SAAS,QAAQ,SAAS;GACtD,MAAM,QAAQ;GACd,SAAS,QAAQ;EACnB,CAAC;EAED,OAAO;CACT;AACF"}
+{"version":3,"file":"model.mjs","names":[],"sources":["../../../../../../../packages/ai-google/src/model.ts"],"sourcesContent":["import {\n  attachCost,\n  InvalidRequestError,\n  resolveStreamChunks,\n  selectProviderTools,\n  type Message,\n  type ModelCallOptions,\n  type ModelCapabilities,\n  type ModelContract,\n  type ModelPricing,\n  type ModelResponse,\n  type ModelStreamChunk,\n  type ModelToolCallRequest,\n  type ReasoningEffort,\n  type Usage,\n} from \"@warlock.js/ai\";\nimport { log, type Logger } from \"@warlock.js/logger\";\nimport type {\n  GenerateContentConfig,\n  GenerateContentResponse,\n  GoogleGenAI,\n  Part,\n  Tool,\n} from \"@google/genai\";\nimport { mergeProviderBody } from \"./provider-options\";\nimport type { GoogleModelConfig } from \"./config.type\";\nimport { inferVisionCapability } from \"./known-vision-models\";\nimport {\n  applyGoogleUsage,\n  mapFinishReason,\n  toGoogleContents,\n  toGoogleTools,\n  wrapGoogleError,\n} from \"./utils\";\n\nconst LOG_MODULE = \"ai.google\";\n\n/** `provider` value of `ai.providerTool()` entries this adapter sends verbatim. */\nconst GOOGLE_WIRE_FORMAT = \"google\";\n\n/**\n * Bucketed `thinkingBudget` (token caps) for the neutral\n * `reasoning.effort` levels when the caller gives no explicit\n * `reasoning.maxTokens`. Gemini 2.5 accepts a positive budget as a cap\n * on the thinking phase; these mirror the spread the OpenAI\n * `reasoning_effort` low/medium/high tiers imply. The installed\n * `@google/genai` `ThinkingConfig` declaration documents that allowed\n * ranges are model-dependent, so retain the existing conservative 24,576\n * cap for the new tiers rather than sending a budget unsupported by Flash.\n */\nconst EFFORT_THINKING_BUDGET: Record<Exclude<ReasoningEffort, \"none\">, number> = {\n  minimal: 1024,\n  low: 1024,\n  medium: 8192,\n  high: 24576,\n  xhigh: 24576,\n  max: 24576,\n};\n\n/**\n * Google Gemini-backed implementation of `ModelContract`.\n *\n * **Role.** The provider-facing bridge between the vendor-neutral\n * `@warlock.js/ai` agent runtime and the `@google/genai` SDK\n * (`models.generateContent` / `generateContentStream`).\n *\n * **Responsibility.**\n * - Owns: a long-lived `GoogleGenAI` client + frozen `ModelConfig`\n *   (name, temperature, maxTokens) used as per-call defaults.\n * - Owns: translating vendor-neutral `Message[]` / `ToolConfig[]` into\n *   Gemini shapes (systemInstruction hoisting, `model` role,\n *   `functionCall` / `functionResponse` parts, inline image bytes) on\n *   the way out, and Gemini's candidate/parts response (text, function\n *   calls, finish reason, token usage) back into neutral shapes on the\n *   way in.\n * - Does NOT own: dispatching tools, looping, history, retries — those\n *   are agent concerns. The model is a per-call protocol adapter.\n *\n * Modeled as a class (see §4.2 of code-style.md — \"long-lived state\n * across calls\"): the `GoogleGenAI` client is reused for the SDK's\n * lifetime.\n *\n * @example\n * import { GoogleGenAI } from \"@google/genai\";\n * const ai = new GoogleGenAI({ apiKey: process.env.GEMINI_API_KEY });\n * const model = new GoogleModel(ai, { name: \"gemini-2.5-flash\" });\n *\n * const myAgent = agent({ model, tools: [searchTool] });\n * const result = await myAgent.execute(\"Summarize today's news.\");\n */\nexport class GoogleModel implements ModelContract {\n  public readonly name: string;\n  public readonly provider: string;\n  public readonly capabilities: ModelCapabilities;\n  public readonly pricing?: ModelPricing;\n\n  private readonly ai: GoogleGenAI;\n  private readonly config: GoogleModelConfig;\n  private readonly logger: Logger = log;\n\n  public constructor(ai: GoogleGenAI, config: GoogleModelConfig, provider: string = \"google\") {\n    this.ai = ai;\n    this.config = config;\n    this.name = config.name;\n    this.provider = provider;\n    this.pricing = config.pricing;\n    const multimodal = config.vision ?? inferVisionCapability(config.name);\n\n    this.capabilities = {\n      structuredOutput: config.structuredOutput ?? true,\n      vision: multimodal,\n      // Every Gemini 2.5 model thinks; older families harmlessly ignore\n      // an empty thinking budget. Defaulting `true` lets the agent\n      // forward reasoning options; an explicit `false` opts a model out.\n      reasoning: config.reasoning ?? true,\n      // Gemini reports cache-read hits (`cachedContentTokenCount`) on\n      // every call via implicit caching, and accepts explicit context\n      // caching. Read-side accounting is always honored.\n      promptCaching: true,\n      // The multimodal Gemini families that accept images also accept\n      // audio and PDF/document parts. Mirror the vision inference unless\n      // explicitly overridden.\n      audio: config.audio ?? multimodal,\n      pdf: config.pdf ?? multimodal,\n    };\n  }\n\n  /**\n   * Single-shot completion. Sends the full message list to\n   * `generateContent`, waits for the terminal response, and reshapes\n   * it into a vendor-neutral `ModelResponse`. Per-call `options`\n   * override the instance defaults for this call only.\n   */\n  public async complete(messages: Message[], options?: ModelCallOptions): Promise<ModelResponse> {\n    this.logger.debug(LOG_MODULE, \"request\", \"Starting generateContent call\", {\n      model: this.name,\n      messageCount: messages.length,\n      streaming: false,\n      toolCount: options?.tools?.length ?? 0,\n    });\n\n    const { systemInstruction, contents } = toGoogleContents(messages);\n\n    let response: GenerateContentResponse;\n\n    try {\n      response = await this.ai.models.generateContent({\n        model: this.name,\n        contents,\n        config: this.buildConfig(systemInstruction, options),\n      });\n    } catch (thrown) {\n      throw this.logAndWrap(thrown);\n    }\n\n    const toolCalls = this.extractToolCalls(response);\n    const finishReason = toolCalls\n      ? \"tool_calls\"\n      : mapFinishReason(response.candidates?.[0]?.finishReason);\n    const usage = attachCost(this.extractUsage(response), this.pricing);\n\n    this.logger.debug(LOG_MODULE, \"response\", \"generateContent call succeeded\", {\n      finishReason,\n      usage,\n    });\n\n    return {\n      content: response.text ?? \"\",\n      finishReason,\n      usage,\n      toolCalls,\n    };\n  }\n\n  /** Count native input tokens with the same converted content, system and tools. */\n  public async countTokens(messages: Message[], options?: ModelCallOptions): Promise<number> {\n    const { systemInstruction, contents } = toGoogleContents(messages);\n    const config = this.buildConfig(systemInstruction, options);\n    if (\n      !this.ai.vertexai &&\n      (config.systemInstruction !== undefined || config.tools !== undefined)\n    ) {\n      throw new InvalidRequestError(\n        \"The Google SDK countTokens endpoint in Developer API mode cannot count system instructions or tools. Use Vertex mode for this preflight, or SDK.count() for an offline estimate.\",\n      );\n    }\n    if (config.cachedContent !== undefined) {\n      throw new InvalidRequestError(\n        \"The Google SDK countTokens endpoint cannot count a cachedContent reference.\",\n      );\n    }\n    try {\n      const response = await this.ai.models.countTokens({\n        model: this.name,\n        contents,\n        config: {\n          ...(config.systemInstruction !== undefined\n            ? { systemInstruction: config.systemInstruction }\n            : {}),\n          ...(config.tools !== undefined ? { tools: config.tools as Tool[] } : {}),\n          ...(config.abortSignal ? { abortSignal: config.abortSignal } : {}),\n          ...(config.httpOptions ? { httpOptions: config.httpOptions } : {}),\n        },\n      });\n      if (response.totalTokens === undefined)\n        throw new InvalidRequestError(\"Google countTokens returned no totalTokens.\");\n      return response.totalTokens;\n    } catch (thrown) {\n      throw this.logAndWrap(thrown);\n    }\n  }\n\n  /**\n   * Incremental streaming completion via `generateContentStream`.\n   * Yields neutral `ModelStreamChunk`s — `delta` for text, `tool-call`\n   * per function call (Gemini emits a fully-formed call, not partial\n   * JSON), and a terminal `done` with the final finish reason + usage.\n   *\n   * With `options.streamChunks` it also yields `reasoning` for thought\n   * parts (only sent when `thinkingConfig.includeThoughts` is on — set it\n   * through `providerOptions.google.body`) and `usage` whenever a chunk\n   * carries `usageMetadata`. No `tool-args`: calls arrive whole.\n   */\n  public async *stream(\n    messages: Message[],\n    options?: ModelCallOptions,\n  ): AsyncIterable<ModelStreamChunk> {\n    this.logger.debug(LOG_MODULE, \"request\", \"Starting generateContentStream call\", {\n      model: this.name,\n      messageCount: messages.length,\n      streaming: true,\n      toolCount: options?.tools?.length ?? 0,\n    });\n\n    const { systemInstruction, contents } = toGoogleContents(messages);\n\n    let iterable: AsyncGenerator<GenerateContentResponse>;\n\n    try {\n      iterable = await this.ai.models.generateContentStream({\n        model: this.name,\n        contents,\n        config: this.buildConfig(systemInstruction, options),\n      });\n    } catch (thrown) {\n      throw this.logAndWrap(thrown);\n    }\n\n    let rawFinishReason: string | undefined;\n    let sawToolCall = false;\n    const usage: Usage = { input: 0, output: 0, total: 0 };\n    const want = resolveStreamChunks(options);\n\n    try {\n      for await (const chunk of iterable) {\n        // Thought parts (`thought: true`, sent when `thinkingConfig.includeThoughts`\n        // is on) carry the thought summary in `text`; the SDK's `chunk.text`\n        // getter already leaves them out of the answer.\n        if (want.reasoning) {\n          for (const part of chunk.candidates?.[0]?.content?.parts ?? []) {\n            if (part.thought === true && typeof part.text === \"string\" && part.text !== \"\") {\n              yield { type: \"reasoning\", delta: part.text };\n            }\n          }\n        }\n\n        const text = chunk.text;\n\n        if (text) {\n          yield { type: \"delta\", content: text };\n        }\n\n        for (const part of chunk.candidates?.[0]?.content?.parts ?? []) {\n          const toolCall = this.partToToolCall(part);\n\n          if (!toolCall) {\n            continue;\n          }\n\n          sawToolCall = true;\n\n          yield {\n            type: \"tool-call\",\n            id: toolCall.id,\n            name: toolCall.name,\n            input: toolCall.input,\n            ...(toolCall.providerMetadata ? { providerMetadata: toolCall.providerMetadata } : {}),\n          };\n        }\n\n        const candidateFinish = chunk.candidates?.[0]?.finishReason;\n\n        if (candidateFinish) {\n          rawFinishReason = candidateFinish;\n        }\n\n        if (chunk.usageMetadata) {\n          applyGoogleUsage(usage, chunk.usageMetadata);\n\n          if (want.usage) {\n            yield { type: \"usage\", usage: { ...usage } };\n          }\n        }\n      }\n    } catch (thrown) {\n      throw this.logAndWrap(thrown);\n    }\n\n    const finishReason = sawToolCall ? \"tool_calls\" : mapFinishReason(rawFinishReason);\n\n    attachCost(usage, this.pricing);\n\n    this.logger.debug(LOG_MODULE, \"response\", \"generateContentStream call succeeded\", {\n      finishReason,\n      usage,\n    });\n\n    yield { type: \"done\", finishReason, usage };\n  }\n\n  /**\n   * Assemble the `GenerateContentConfig` shared by `complete()` and\n   * `stream()`: inference params, hoisted system instruction,\n   * cancellation signal, and conditional tools + native structured\n   * output.\n   */\n  private buildConfig(\n    systemInstruction: string | undefined,\n    options: ModelCallOptions | undefined,\n  ): GenerateContentConfig {\n    const temperature = options?.temperature ?? this.config.temperature;\n    const maxOutputTokens = options?.maxTokens ?? this.config.maxTokens;\n\n    return mergeProviderBody<GenerateContentConfig>(\n      {\n        ...(systemInstruction ? { systemInstruction } : {}),\n        ...(temperature !== undefined ? { temperature } : {}),\n        ...(maxOutputTokens !== undefined ? { maxOutputTokens } : {}),\n        ...(options?.signal ? { abortSignal: options.signal } : {}),\n        ...this.buildTools(options),\n        ...this.buildStructuredOutput(options?.responseSchema),\n        ...this.buildThinking(options?.reasoning),\n      },\n      options,\n    );\n  }\n\n  /**\n   * Translate the neutral `reasoning` option into Gemini's\n   * `thinkingConfig`. `reasoning.maxTokens` maps directly to\n   * `thinkingBudget` (token cap on the thinking phase); when only\n   * `reasoning.effort` is given it is bucketed into a budget. Emitted\n   * only when the model is `reasoning`-capable — a `false` capability\n   * (config override) drops it so a non-thinking model never receives\n   * an unsupported `thinkingConfig`.\n   *\n   * Gemini's `thinkingBudget` semantics: `0` disables thinking, `-1`\n   * lets the model decide automatically. A positive value caps the\n   * thinking tokens. The neutral `effort: \"none\"` (\"run without\n   * reasoning\") maps to `thinkingBudget: 0`.\n   */\n  private buildThinking(\n    reasoning: ModelCallOptions[\"reasoning\"],\n  ): Pick<GenerateContentConfig, \"thinkingConfig\"> {\n    if (!reasoning || !this.capabilities.reasoning) {\n      return {};\n    }\n\n    // `effort: \"none\"` = explicit \"run without reasoning\". Gemini disables\n    // thinking with `thinkingBudget: 0` (its native off switch), so emit\n    // that rather than omitting the config — an omitted config lets a\n    // thinking model reason at its default budget.\n    if (reasoning.effort === \"none\") {\n      return { thinkingConfig: { thinkingBudget: 0 } };\n    }\n\n    const thinkingBudget =\n      reasoning.maxTokens ??\n      (reasoning.effort ? EFFORT_THINKING_BUDGET[reasoning.effort] : undefined);\n\n    if (thinkingBudget === undefined) {\n      return {};\n    }\n\n    return { thinkingConfig: { thinkingBudget } };\n  }\n\n  /**\n   * Spread-friendly tools fragment. Empty object when no tools were\n   * supplied so the caller can unconditionally spread it. Function tools\n   * come first; `ai.providerTool({ provider: \"google\" })` definitions\n   * (`{ googleSearch: {} }`, `{ codeExecution: {} }`, `{ urlContext: {} }`,\n   * …) follow verbatim as their own `Tool` entries. Other providers'\n   * definitions are skipped with a warning.\n   */\n  private buildTools(options: ModelCallOptions | undefined): Pick<GenerateContentConfig, \"tools\"> {\n    const { tools, definitions } = selectProviderTools(options, {\n      accepts: [GOOGLE_WIRE_FORMAT, this.provider],\n      module: LOG_MODULE,\n      model: this.name,\n      logger: this.logger,\n    });\n    const mapped: Tool[] = [...(toGoogleTools(tools) ?? []), ...(definitions as unknown as Tool[])];\n\n    return mapped.length > 0 ? { tools: mapped } : {};\n  }\n\n  /**\n   * Translate the neutral `responseSchema` into Gemini's native JSON\n   * structured output (`responseMimeType: \"application/json\"` +\n   * `responseJsonSchema`, which takes a raw JSON Schema directly).\n   * Emitted only when the model is `structuredOutput`-capable and the\n   * schema is an object root — otherwise the agent's soft prompt hint\n   * + client-side `validate()` carry shape.\n   */\n  private buildStructuredOutput(\n    responseSchema: Record<string, unknown> | undefined,\n  ): Pick<GenerateContentConfig, \"responseMimeType\" | \"responseJsonSchema\"> {\n    if (!responseSchema || !this.capabilities.structuredOutput) {\n      return {};\n    }\n\n    if (responseSchema.type !== \"object\" || typeof responseSchema.properties !== \"object\") {\n      return {};\n    }\n\n    return {\n      responseMimeType: \"application/json\",\n      responseJsonSchema: responseSchema,\n    };\n  }\n\n  /**\n   * Reshape Gemini's function-call content parts into the neutral\n   * `ModelToolCallRequest[]`. Returns `undefined` when the model\n   * requested no functions so callers can branch on presence.\n   *\n   * Reads `candidates[0].content.parts` directly rather than the\n   * `response.functionCalls` getter: the getter discards the\n   * part-level `thoughtSignature`, and Gemini \"thinking\" models 400\n   * the follow-up turn if that signature is not echoed back. See\n   * `partToToolCall`.\n   */\n  private extractToolCalls(response: GenerateContentResponse): ModelToolCallRequest[] | undefined {\n    const parts = response.candidates?.[0]?.content?.parts ?? [];\n    const toolCalls = parts\n      .map((part) => this.partToToolCall(part))\n      .filter((call): call is ModelToolCallRequest => call !== undefined);\n\n    return toolCalls.length > 0 ? toolCalls : undefined;\n  }\n\n  /**\n   * Map a single Gemini `Part` to a neutral `ModelToolCallRequest`,\n   * or `undefined` when the part is not a function call. The part's\n   * `thoughtSignature` (opaque, set by thinking models) is carried on\n   * `providerMetadata` so `toGoogleContents` can replay it on the\n   * assistant turn — Gemini rejects the next request without it.\n   */\n  private partToToolCall(part: Part): ModelToolCallRequest | undefined {\n    if (!part.functionCall) {\n      return undefined;\n    }\n\n    const call = part.functionCall;\n\n    return {\n      // The Gemini Developer API does not assign function-call ids\n      // (only Vertex parallel-calling does). Fall back to the function\n      // name so the neutral `toolCallId` is non-empty and the echoed\n      // `functionResponse.name` resolves — Gemini matches a result to\n      // its call by name. See decisions §49.\n      id: call.id ?? call.name ?? \"\",\n      name: call.name ?? \"\",\n      input: (call.args ?? {}) as Record<string, unknown>,\n      ...(part.thoughtSignature\n        ? { providerMetadata: { thoughtSignature: part.thoughtSignature } }\n        : {}),\n    };\n  }\n\n  /**\n   * Normalize Gemini's `usageMetadata` into the neutral `Usage` shape\n   * via the shared {@link applyGoogleUsage} mapper (the same one the\n   * streaming loop and the Gemini image model use). Absent usage\n   * collapses to zeros.\n   */\n  private extractUsage(response: GenerateContentResponse): Usage {\n    const usage: Usage = { input: 0, output: 0, total: 0 };\n\n    if (response.usageMetadata) {\n      applyGoogleUsage(usage, response.usageMetadata);\n    }\n\n    return usage;\n  }\n\n  /**\n   * Wrap a thrown provider error into the typed `AIError` hierarchy\n   * and emit the standard error log line before it propagates.\n   */\n  private logAndWrap(thrown: unknown) {\n    const wrapped = wrapGoogleError(thrown);\n\n    this.logger.error(LOG_MODULE, \"error\", wrapped.message, {\n      code: wrapped.code,\n      context: wrapped.context,\n    });\n\n    return wrapped;\n  }\n}\n"],"mappings":";;;;;;;;;;;;AAmCA,MAAM,aAAa;;AAGnB,MAAM,qBAAqB;;;;;;;;;;;AAY3B,MAAM,yBAA2E;CAC/E,SAAS;CACT,KAAK;CACL,QAAQ;CACR,MAAM;CACN,OAAO;CACP,KAAK;AACP;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAiCA,IAAa,cAAb,MAAkD;CAUhD,AAAO,YAAY,IAAiB,QAA2B,WAAmB,UAAU;gBAF1D;EAGhC,KAAK,KAAK;EACV,KAAK,SAAS;EACd,KAAK,OAAO,OAAO;EACnB,KAAK,WAAW;EAChB,KAAK,UAAU,OAAO;EACtB,MAAM,aAAa,OAAO,UAAU,sBAAsB,OAAO,IAAI;EAErE,KAAK,eAAe;GAClB,kBAAkB,OAAO,oBAAoB;GAC7C,QAAQ;GAIR,WAAW,OAAO,aAAa;GAI/B,eAAe;GAIf,OAAO,OAAO,SAAS;GACvB,KAAK,OAAO,OAAO;EACrB;CACF;;;;;;;CAQA,MAAa,SAAS,UAAqB,SAAoD;EAC7F,KAAK,OAAO,MAAM,YAAY,WAAW,iCAAiC;GACxE,OAAO,KAAK;GACZ,cAAc,SAAS;GACvB,WAAW;GACX,WAAW,SAAS,OAAO,UAAU;EACvC,CAAC;EAED,MAAM,EAAE,mBAAmB,aAAa,iBAAiB,QAAQ;EAEjE,IAAI;EAEJ,IAAI;GACF,WAAW,MAAM,KAAK,GAAG,OAAO,gBAAgB;IAC9C,OAAO,KAAK;IACZ;IACA,QAAQ,KAAK,YAAY,mBAAmB,OAAO;GACrD,CAAC;EACH,SAAS,QAAQ;GACf,MAAM,KAAK,WAAW,MAAM;EAC9B;EAEA,MAAM,YAAY,KAAK,iBAAiB,QAAQ;EAChD,MAAM,eAAe,YACjB,eACA,gBAAgB,SAAS,aAAa,EAAE,EAAE,YAAY;EAC1D,MAAM,QAAQ,WAAW,KAAK,aAAa,QAAQ,GAAG,KAAK,OAAO;EAElE,KAAK,OAAO,MAAM,YAAY,YAAY,kCAAkC;GAC1E;GACA;EACF,CAAC;EAED,OAAO;GACL,SAAS,SAAS,QAAQ;GAC1B;GACA;GACA;EACF;CACF;;CAGA,MAAa,YAAY,UAAqB,SAA6C;EACzF,MAAM,EAAE,mBAAmB,aAAa,iBAAiB,QAAQ;EACjE,MAAM,SAAS,KAAK,YAAY,mBAAmB,OAAO;EAC1D,IACE,CAAC,KAAK,GAAG,aACR,OAAO,sBAAsB,UAAa,OAAO,UAAU,SAE5D,MAAM,IAAI,oBACR,kLACF;EAEF,IAAI,OAAO,kBAAkB,QAC3B,MAAM,IAAI,oBACR,6EACF;EAEF,IAAI;GACF,MAAM,WAAW,MAAM,KAAK,GAAG,OAAO,YAAY;IAChD,OAAO,KAAK;IACZ;IACA,QAAQ;KACN,GAAI,OAAO,sBAAsB,SAC7B,EAAE,mBAAmB,OAAO,kBAAkB,IAC9C,CAAC;KACL,GAAI,OAAO,UAAU,SAAY,EAAE,OAAO,OAAO,MAAgB,IAAI,CAAC;KACtE,GAAI,OAAO,cAAc,EAAE,aAAa,OAAO,YAAY,IAAI,CAAC;KAChE,GAAI,OAAO,cAAc,EAAE,aAAa,OAAO,YAAY,IAAI,CAAC;IAClE;GACF,CAAC;GACD,IAAI,SAAS,gBAAgB,QAC3B,MAAM,IAAI,oBAAoB,6CAA6C;GAC7E,OAAO,SAAS;EAClB,SAAS,QAAQ;GACf,MAAM,KAAK,WAAW,MAAM;EAC9B;CACF;;;;;;;;;;;;CAaA,OAAc,OACZ,UACA,SACiC;EACjC,KAAK,OAAO,MAAM,YAAY,WAAW,uCAAuC;GAC9E,OAAO,KAAK;GACZ,cAAc,SAAS;GACvB,WAAW;GACX,WAAW,SAAS,OAAO,UAAU;EACvC,CAAC;EAED,MAAM,EAAE,mBAAmB,aAAa,iBAAiB,QAAQ;EAEjE,IAAI;EAEJ,IAAI;GACF,WAAW,MAAM,KAAK,GAAG,OAAO,sBAAsB;IACpD,OAAO,KAAK;IACZ;IACA,QAAQ,KAAK,YAAY,mBAAmB,OAAO;GACrD,CAAC;EACH,SAAS,QAAQ;GACf,MAAM,KAAK,WAAW,MAAM;EAC9B;EAEA,IAAI;EACJ,IAAI,cAAc;EAClB,MAAM,QAAe;GAAE,OAAO;GAAG,QAAQ;GAAG,OAAO;EAAE;EACrD,MAAM,OAAO,oBAAoB,OAAO;EAExC,IAAI;GACF,WAAW,MAAM,SAAS,UAAU;IAIlC,IAAI,KAAK,WACP;UAAK,MAAM,QAAQ,MAAM,aAAa,EAAE,EAAE,SAAS,SAAS,CAAC,GAC3D,IAAI,KAAK,YAAY,QAAQ,OAAO,KAAK,SAAS,YAAY,KAAK,SAAS,IAC1E,MAAM;MAAE,MAAM;MAAa,OAAO,KAAK;KAAK;IAEhD;IAGF,MAAM,OAAO,MAAM;IAEnB,IAAI,MACF,MAAM;KAAE,MAAM;KAAS,SAAS;IAAK;IAGvC,KAAK,MAAM,QAAQ,MAAM,aAAa,EAAE,EAAE,SAAS,SAAS,CAAC,GAAG;KAC9D,MAAM,WAAW,KAAK,eAAe,IAAI;KAEzC,IAAI,CAAC,UACH;KAGF,cAAc;KAEd,MAAM;MACJ,MAAM;MACN,IAAI,SAAS;MACb,MAAM,SAAS;MACf,OAAO,SAAS;MAChB,GAAI,SAAS,mBAAmB,EAAE,kBAAkB,SAAS,iBAAiB,IAAI,CAAC;KACrF;IACF;IAEA,MAAM,kBAAkB,MAAM,aAAa,EAAE,EAAE;IAE/C,IAAI,iBACF,kBAAkB;IAGpB,IAAI,MAAM,eAAe;KACvB,iBAAiB,OAAO,MAAM,aAAa;KAE3C,IAAI,KAAK,OACP,MAAM;MAAE,MAAM;MAAS,OAAO,EAAE,GAAG,MAAM;KAAE;IAE/C;GACF;EACF,SAAS,QAAQ;GACf,MAAM,KAAK,WAAW,MAAM;EAC9B;EAEA,MAAM,eAAe,cAAc,eAAe,gBAAgB,eAAe;EAEjF,WAAW,OAAO,KAAK,OAAO;EAE9B,KAAK,OAAO,MAAM,YAAY,YAAY,wCAAwC;GAChF;GACA;EACF,CAAC;EAED,MAAM;GAAE,MAAM;GAAQ;GAAc;EAAM;CAC5C;;;;;;;CAQA,AAAQ,YACN,mBACA,SACuB;EACvB,MAAM,cAAc,SAAS,eAAe,KAAK,OAAO;EACxD,MAAM,kBAAkB,SAAS,aAAa,KAAK,OAAO;EAE1D,OAAO,kBACL;GACE,GAAI,oBAAoB,EAAE,kBAAkB,IAAI,CAAC;GACjD,GAAI,gBAAgB,SAAY,EAAE,YAAY,IAAI,CAAC;GACnD,GAAI,oBAAoB,SAAY,EAAE,gBAAgB,IAAI,CAAC;GAC3D,GAAI,SAAS,SAAS,EAAE,aAAa,QAAQ,OAAO,IAAI,CAAC;GACzD,GAAG,KAAK,WAAW,OAAO;GAC1B,GAAG,KAAK,sBAAsB,SAAS,cAAc;GACrD,GAAG,KAAK,cAAc,SAAS,SAAS;EAC1C,GACA,OACF;CACF;;;;;;;;;;;;;;;CAgBA,AAAQ,cACN,WAC+C;EAC/C,IAAI,CAAC,aAAa,CAAC,KAAK,aAAa,WACnC,OAAO,CAAC;EAOV,IAAI,UAAU,WAAW,QACvB,OAAO,EAAE,gBAAgB,EAAE,gBAAgB,EAAE,EAAE;EAGjD,MAAM,iBACJ,UAAU,cACT,UAAU,SAAS,uBAAuB,UAAU,UAAU;EAEjE,IAAI,mBAAmB,QACrB,OAAO,CAAC;EAGV,OAAO,EAAE,gBAAgB,EAAE,eAAe,EAAE;CAC9C;;;;;;;;;CAUA,AAAQ,WAAW,SAA6E;EAC9F,MAAM,EAAE,OAAO,gBAAgB,oBAAoB,SAAS;GAC1D,SAAS,CAAC,oBAAoB,KAAK,QAAQ;GAC3C,QAAQ;GACR,OAAO,KAAK;GACZ,QAAQ,KAAK;EACf,CAAC;EACD,MAAM,SAAiB,CAAC,GAAI,cAAc,KAAK,KAAK,CAAC,GAAI,GAAI,WAAiC;EAE9F,OAAO,OAAO,SAAS,IAAI,EAAE,OAAO,OAAO,IAAI,CAAC;CAClD;;;;;;;;;CAUA,AAAQ,sBACN,gBACwE;EACxE,IAAI,CAAC,kBAAkB,CAAC,KAAK,aAAa,kBACxC,OAAO,CAAC;EAGV,IAAI,eAAe,SAAS,YAAY,OAAO,eAAe,eAAe,UAC3E,OAAO,CAAC;EAGV,OAAO;GACL,kBAAkB;GAClB,oBAAoB;EACtB;CACF;;;;;;;;;;;;CAaA,AAAQ,iBAAiB,UAAuE;EAE9F,MAAM,aADQ,SAAS,aAAa,EAAE,EAAE,SAAS,SAAS,CAAC,EACpC,CACpB,KAAK,SAAS,KAAK,eAAe,IAAI,CAAC,CAAC,CACxC,QAAQ,SAAuC,SAAS,MAAS;EAEpE,OAAO,UAAU,SAAS,IAAI,YAAY;CAC5C;;;;;;;;CASA,AAAQ,eAAe,MAA8C;EACnE,IAAI,CAAC,KAAK,cACR;EAGF,MAAM,OAAO,KAAK;EAElB,OAAO;GAML,IAAI,KAAK,MAAM,KAAK,QAAQ;GAC5B,MAAM,KAAK,QAAQ;GACnB,OAAQ,KAAK,QAAQ,CAAC;GACtB,GAAI,KAAK,mBACL,EAAE,kBAAkB,EAAE,kBAAkB,KAAK,iBAAiB,EAAE,IAChE,CAAC;EACP;CACF;;;;;;;CAQA,AAAQ,aAAa,UAA0C;EAC7D,MAAM,QAAe;GAAE,OAAO;GAAG,QAAQ;GAAG,OAAO;EAAE;EAErD,IAAI,SAAS,eACX,iBAAiB,OAAO,SAAS,aAAa;EAGhD,OAAO;CACT;;;;;CAMA,AAAQ,WAAW,QAAiB;EAClC,MAAM,UAAU,gBAAgB,MAAM;EAEtC,KAAK,OAAO,MAAM,YAAY,SAAS,QAAQ,SAAS;GACtD,MAAM,QAAQ;GACd,SAAS,QAAQ;EACnB,CAAC;EAED,OAAO;CACT;AACF"}

package/esm/provider-options.d.mts

@@ -0,0 +1,8 @@
+//#region ../../packages/ai-google/src/provider-options.d.ts
+/** Adapter-only fields merged after the neutral request parameters. */
+type GoogleCallOptions = {
+  body?: Record<string, unknown>;
+};
+//#endregion
+export { GoogleCallOptions };
+//# sourceMappingURL=provider-options.d.mts.map

package/esm/provider-options.d.mts.map

@@ -0,0 +1 @@
+{"version":3,"file":"provider-options.d.mts","names":[],"sources":["../../../../../../../packages/ai-google/src/provider-options.ts"],"mappings":";;KAGY,iBAAA;EAAsB,IAAA,GAAO,MAAM;AAAA"}

package/esm/provider-options.mjs

@@ -0,0 +1,36 @@
+import { InvalidRequestError } from "@warlock.js/ai";
+
+//#region ../../packages/ai-google/src/provider-options.ts
+const RESERVED_FIELDS = new Set([
+	"systemInstruction",
+	"tools",
+	"abortSignal",
+	"httpOptions"
+]);
+function mergeProviderBody(built, options) {
+	const body = (options?.providerOptions?.google)?.body ?? {};
+	const reserved = Object.keys(body).filter((key) => RESERVED_FIELDS.has(key));
+	if (reserved.length > 0) throw new InvalidRequestError(`providerOptions.google.body cannot set ${reserved.join(", ")}: use the call's messages, tools and model options.`, { context: { fields: reserved } });
+	const defaults = built;
+	const overrides = Object.fromEntries(Object.entries(body).map(([key, value]) => {
+		const previous = defaults[key];
+		return [key, isPlainObject(previous) && isPlainObject(value) ? {
+			...previous,
+			...value
+		} : value];
+	}));
+	return {
+		...built,
+		...overrides
+	};
+}
+/** Merge only plain objects; arrays, null and provider scalar values replace the field. */
+function isPlainObject(value) {
+	if (value === null || typeof value !== "object") return false;
+	const prototype = Object.getPrototypeOf(value);
+	return prototype === Object.prototype || prototype === null;
+}
+
+//#endregion
+export { mergeProviderBody };
+//# sourceMappingURL=provider-options.mjs.map

package/esm/provider-options.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"provider-options.mjs","names":[],"sources":["../../../../../../../packages/ai-google/src/provider-options.ts"],"sourcesContent":["import { InvalidRequestError, type ModelCallOptions } from \"@warlock.js/ai\";\n\n/** Adapter-only fields merged after the neutral request parameters. */\nexport type GoogleCallOptions = { body?: Record<string, unknown> };\n\nconst RESERVED_FIELDS = new Set([\"systemInstruction\", \"tools\", \"abortSignal\", \"httpOptions\"]);\n\nexport function mergeProviderBody<T extends object>(\n  built: T,\n  options: ModelCallOptions | undefined,\n): T {\n  const provider = options?.providerOptions?.google as GoogleCallOptions | undefined;\n  const body = provider?.body ?? {};\n  const reserved = Object.keys(body).filter((key) => RESERVED_FIELDS.has(key));\n  if (reserved.length > 0) {\n    throw new InvalidRequestError(\n      `providerOptions.google.body cannot set ${reserved.join(\", \")}: use the call's messages, tools and model options.`,\n      { context: { fields: reserved } },\n    );\n  }\n  const defaults = built as Record<string, unknown>;\n  const overrides = Object.fromEntries(\n    Object.entries(body).map(([key, value]) => {\n      const previous = defaults[key];\n      return [\n        key,\n        isPlainObject(previous) && isPlainObject(value) ? { ...previous, ...value } : value,\n      ];\n    }),\n  );\n  return { ...built, ...overrides };\n}\n\n/** Merge only plain objects; arrays, null and provider scalar values replace the field. */\nfunction isPlainObject(value: unknown): value is Record<string, unknown> {\n  if (value === null || typeof value !== \"object\") return false;\n  const prototype = Object.getPrototypeOf(value);\n  return prototype === Object.prototype || prototype === null;\n}\n"],"mappings":";;;AAKA,MAAM,kBAAkB,IAAI,IAAI;CAAC;CAAqB;CAAS;CAAe;AAAa,CAAC;AAE5F,SAAgB,kBACd,OACA,SACG;CAEH,MAAM,QADW,SAAS,iBAAiB,OACtB,EAAE,QAAQ,CAAC;CAChC,MAAM,WAAW,OAAO,KAAK,IAAI,CAAC,CAAC,QAAQ,QAAQ,gBAAgB,IAAI,GAAG,CAAC;CAC3E,IAAI,SAAS,SAAS,GACpB,MAAM,IAAI,oBACR,0CAA0C,SAAS,KAAK,IAAI,EAAE,sDAC9D,EAAE,SAAS,EAAE,QAAQ,SAAS,EAAE,CAClC;CAEF,MAAM,WAAW;CACjB,MAAM,YAAY,OAAO,YACvB,OAAO,QAAQ,IAAI,CAAC,CAAC,KAAK,CAAC,KAAK,WAAW;EACzC,MAAM,WAAW,SAAS;EAC1B,OAAO,CACL,KACA,cAAc,QAAQ,KAAK,cAAc,KAAK,IAAI;GAAE,GAAG;GAAU,GAAG;EAAM,IAAI,KAChF;CACF,CAAC,CACH;CACA,OAAO;EAAE,GAAG;EAAO,GAAG;CAAU;AAClC;;AAGA,SAAS,cAAc,OAAkD;CACvE,IAAI,UAAU,QAAQ,OAAO,UAAU,UAAU,OAAO;CACxD,MAAM,YAAY,OAAO,eAAe,KAAK;CAC7C,OAAO,cAAc,OAAO,aAAa,cAAc;AACzD"}

package/esm/sdk.d.mts

@@ -1,7 +1,7 @@
 import { GoogleEmbedderConfig, GoogleImageConfig, GoogleModelConfig, GoogleSDKConfig } from "./config.type.mjs";
 import { EmbedderContract, ImageModelContract, ModelContract, SDKAdapterContract } from "@warlock.js/ai";
 
-//#region ../ai-google/src/sdk.d.ts
+//#region ../../packages/ai-google/src/sdk.d.ts
 /**
  * Google Gemini-backed implementation of `SDKAdapterContract`.
  *
@@ -37,8 +37,9 @@ declare class GoogleSDK implements SDKAdapterContract {
    * `GoogleGenAI` client. The SDK's `provider` label is forwarded.
    *
    * Pricing resolution: per-model `config.pricing` wins; otherwise the
-   * SDK-level registry entry keyed by `config.name`; otherwise
-   * `undefined` (no cost computed).
+   * SDK-level registry entry keyed by `config.name`; otherwise the
+   * built-in list price (`GOOGLE_MODEL_PRICING`); otherwise `undefined`
+   * (no cost computed).
    */
   model(config: GoogleModelConfig): ModelContract;
   /**

package/esm/sdk.d.mts.map

@@ -1 +1 @@
-{"version":3,"file":"sdk.d.mts","names":[],"sources":["../../../../../../ai-google/src/sdk.ts"],"mappings":";;;;;;AA2EA;;;;;;;;;;;;;;;;;;;;;;cAAa,SAAA,YAAqB,kBAAA;EAAA,iBACf,EAAA;EAAA,iBACA,QAAA;EAAA,iBACA,OAAA;cAEE,MAAA,EAAQ,eAAA;EA+BM;;;;;;;;;EAd1B,KAAA,CAAM,MAAA,EAAQ,iBAAA,GAAoB,aAAA;EAoDkB;AAAA;;;;;EAtC9C,KAAA,CAAM,IAAA,UAAc,MAAA,YAAkB,OAAA;;;;;;;;EAW5C,QAAA,CAAS,MAAA,EAAQ,oBAAA,GAAuB,gBAAA;;;;;;;;;;;;;;;;;;;;;;;;EA2BxC,KAAA,CAAM,MAAA,EAAQ,iBAAA,GAAoB,kBAAA;AAAA"}
+{"version":3,"file":"sdk.d.mts","names":[],"sources":["../../../../../../../packages/ai-google/src/sdk.ts"],"mappings":";;;;;;AA+EA;;;;;;;;;;;;;;;;;;;;;;cAAa,SAAA,YAAqB,kBAAA;EAAA,iBACf,EAAA;EAAA,iBACA,QAAA;EAAA,iBACA,OAAA;cAEE,MAAA,EAAQ,eAAA;EAiCM;;;;;;;;;;EAf1B,KAAA,CAAM,MAAA,EAAQ,iBAAA,GAAoB,aAAA;EAqDkB;;;;;;EAtC9C,KAAA,CAAM,IAAA,UAAc,MAAA,YAAkB,OAAA;;;;;;;;EAW5C,QAAA,CAAS,MAAA,EAAQ,oBAAA,GAAuB,gBAAA;;;;;;;;;;;;;;;;;;;;;;;;EA2BxC,KAAA,CAAM,MAAA,EAAQ,iBAAA,GAAoB,kBAAA;AAAA"}

package/esm/sdk.mjs

@@ -1,11 +1,12 @@
 import { GoogleEmbedder } from "./embedder.mjs";
 import { GeminiImageModel } from "./gemini-image.mjs";
 import { GoogleImageModel } from "./image.mjs";
+import { googlePricingFor } from "./known-pricing.mjs";
 import { GoogleModel } from "./model.mjs";
 import { GoogleGenAI } from "@google/genai";
 import { approximateTokenCount } from "@warlock.js/ai";
 
-//#region ../ai-google/src/sdk.ts
+//#region ../../packages/ai-google/src/sdk.ts
 /**
 * Pick the transport for an image model id.
 *
@@ -73,11 +74,12 @@ var GoogleSDK = class {
 	* `GoogleGenAI` client. The SDK's `provider` label is forwarded.
 	*
 	* Pricing resolution: per-model `config.pricing` wins; otherwise the
-	* SDK-level registry entry keyed by `config.name`; otherwise
-	* `undefined` (no cost computed).
+	* SDK-level registry entry keyed by `config.name`; otherwise the
+	* built-in list price (`GOOGLE_MODEL_PRICING`); otherwise `undefined`
+	* (no cost computed).
 	*/
 	model(config) {
-		const resolvedPricing = config.pricing ?? this.pricing?.[config.name];
+		const resolvedPricing = config.pricing ?? this.pricing?.[config.name] ?? googlePricingFor(config.name);
 		const resolvedConfig = resolvedPricing === config.pricing ? config : {
 			...config,
 			pricing: resolvedPricing

package/esm/sdk.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"sdk.mjs","names":[],"sources":["../../../../../../ai-google/src/sdk.ts"],"sourcesContent":["import { GoogleGenAI } from \"@google/genai\";\nimport type {\n  EmbedderContract,\n  ImageModelContract,\n  ModelContract,\n  ModelPricing,\n  SDKAdapterContract,\n} from \"@warlock.js/ai\";\nimport { approximateTokenCount } from \"@warlock.js/ai\";\nimport type {\n  GoogleEmbedderConfig,\n  GoogleImageConfig,\n  GoogleModelConfig,\n  GoogleSDKConfig,\n} from \"./config.type\";\nimport { GoogleEmbedder } from \"./embedder\";\nimport { GeminiImageModel } from \"./gemini-image\";\nimport { GoogleImageModel } from \"./image\";\nimport { GoogleModel } from \"./model\";\n\n/**\n * Pick the transport for an image model id.\n *\n * `ai.models.generateImages` calls `{model}:predict`, and a `gemini-`\n * id sent there comes back `404 … is not supported for predict`\n * (observed verbatim from Google). `generateContent` is what the SDK\n * itself points `generateImages` users at — its deprecation notice\n * reads \"Please use the generateContent method with image models\n * instead\" — so the id has to choose the transport.\n *\n * Runs in this package establish where a `gemini-` id is ACCEPTED, not\n * what it returns: on this transport such an id got as far as a quota\n * error (HTTP 429) instead of the 404. That an image\n * comes back end-to-end once billing is enabled is reported by the\n * maintainer from a locally linked build, not measured here. Whether\n * these models report token usage is still unknown.\n *\n * This is ROUTING, not validation — no id is refused here. An id this\n * function does not recognize takes the `generateImages` route, the\n * only route that existed before Gemini image support landed, so every\n * id that reached Google before still reaches Google the same way and\n * still fails (or succeeds) at the provider.\n *\n * A leading `models/` resource prefix is tolerated, matching the id\n * shapes `inferVisionCapability` already accepts\n * (`models/gemini-1.5-flash-001`).\n */\nfunction usesGeminiImageTransport(name: string): boolean {\n  return name.toLowerCase().replace(/^models\\//, \"\").startsWith(\"gemini-\");\n}\n\n/**\n * Google Gemini-backed implementation of `SDKAdapterContract`.\n *\n * **Role.** The package entry point for Gemini models via the\n * `@google/genai` SDK. A single `GoogleSDK` holds one live\n * `GoogleGenAI` client, shared by every `ModelContract` /\n * `EmbedderContract` it produces. Construct one SDK per\n * account/project and reuse it everywhere.\n *\n * **Responsibility.**\n * - Owns: a long-lived `GoogleGenAI` client (auth, Vertex vs Gemini\n *   API) and its lifetime. Factory for `GoogleModel` /\n *   `GoogleEmbedder` instances sharing that client.\n * - Does NOT own: anything per-call — those live in `GoogleModel` /\n *   `GoogleEmbedder` and the agent runtime.\n *\n * Modeled as a class (see §4.2 of code-style.md — \"long-lived state\n * across many calls\"), fronted by FP usage like the other adapters.\n *\n * @example\n * const google = new GoogleSDK({ apiKey: process.env.GEMINI_API_KEY! });\n * const model = google.model({ name: \"gemini-2.5-flash\", temperature: 0.7 });\n * const embedder = google.embedder({ name: \"gemini-embedding-001\" });\n */\nexport class GoogleSDK implements SDKAdapterContract {\n  private readonly ai: GoogleGenAI;\n  private readonly provider: string;\n  private readonly pricing?: Record<string, ModelPricing>;\n\n  public constructor(config: GoogleSDKConfig) {\n    const { provider, pricing, ...clientOptions } = config;\n\n    this.ai = new GoogleGenAI(clientOptions);\n    this.provider = provider ?? \"google\";\n    this.pricing = pricing;\n  }\n\n  /**\n   * Build a `GoogleModel` bound to this SDK's client. Each call\n   * returns a fresh instance; all instances share the underlying\n   * `GoogleGenAI` client. The SDK's `provider` label is forwarded.\n   *\n   * Pricing resolution: per-model `config.pricing` wins; otherwise the\n   * SDK-level registry entry keyed by `config.name`; otherwise\n   * `undefined` (no cost computed).\n   */\n  public model(config: GoogleModelConfig): ModelContract {\n    const resolvedPricing = config.pricing ?? this.pricing?.[config.name];\n    const resolvedConfig: GoogleModelConfig =\n      resolvedPricing === config.pricing ? config : { ...config, pricing: resolvedPricing };\n\n    return new GoogleModel(this.ai, resolvedConfig, this.provider);\n  }\n\n  /**\n   * Rough token-count estimate. Uses the character-heuristic\n   * (`approximateTokenCount`) from the core package — Gemini's\n   * `countTokens` is a network round-trip; `count()` is intentionally\n   * offline. Good for budgeting/quota guards, not billing.\n   */\n  public async count(text: string, _model?: string): Promise<number> {\n    return approximateTokenCount(text);\n  }\n\n  /**\n   * Build a `GoogleEmbedder` bound to this SDK's client.\n   *\n   * @example\n   * const embedder = google.embedder({ name: \"gemini-embedding-001\" });\n   * const { vector } = await embedder.embed(\"Hello world\");\n   */\n  public embedder(config: GoogleEmbedderConfig): EmbedderContract {\n    return new GoogleEmbedder(this.ai, config, this.provider);\n  }\n\n  /**\n   * Build an image model bound to this SDK's client for use with\n   * `ai.image({ model, prompt })`. `config.name` decides the transport\n   * (see {@link usesGeminiImageTransport}) — a `gemini-` id gets the\n   * `generateContent` implementation, everything else the Imagen\n   * `generateImages` one. No id is rejected locally either way, so an\n   * unsupported model fails at Google, not here.\n   *\n   * The two differ in what usage they can report, which is what the\n   * caller must price for: the Imagen path always returns a zero token\n   * `Usage` (Imagen reports none — price with `{ perImage }`), while the\n   * Gemini path passes through whatever `usageMetadata` Google attaches\n   * (price with `{ input, output }` when tokens come back).\n   *\n   * Pricing resolution mirrors `model()`: per-model `config.pricing`\n   * wins, otherwise the SDK-level registry entry keyed by `config.name`,\n   * otherwise `undefined`.\n   *\n   * @example\n   * const imagen = google.image({ name: \"imagen-4.0-generate-001\" });\n   * const gemini = google.image({ name: \"gemini-3.1-flash-lite-image\" });\n   * const { data } = await ai.image({ model: gemini, prompt: \"a red bicycle\" });\n   */\n  public image(config: GoogleImageConfig): ImageModelContract {\n    const resolvedPricing = config.pricing ?? this.pricing?.[config.name];\n    const resolvedConfig: GoogleImageConfig =\n      resolvedPricing === config.pricing ? config : { ...config, pricing: resolvedPricing };\n\n    if (usesGeminiImageTransport(config.name)) {\n      return new GeminiImageModel(this.ai, resolvedConfig, this.provider);\n    }\n\n    return new GoogleImageModel(this.ai, resolvedConfig, this.provider);\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AA+CA,SAAS,yBAAyB,MAAuB;CACvD,OAAO,KAAK,YAAY,CAAC,CAAC,QAAQ,aAAa,EAAE,CAAC,CAAC,WAAW,SAAS;AACzE;;;;;;;;;;;;;;;;;;;;;;;;;AA0BA,IAAa,YAAb,MAAqD;CAKnD,AAAO,YAAY,QAAyB;EAC1C,MAAM,EAAE,UAAU,SAAS,GAAG,kBAAkB;EAEhD,KAAK,KAAK,IAAI,YAAY,aAAa;EACvC,KAAK,WAAW,YAAY;EAC5B,KAAK,UAAU;CACjB;;;;;;;;;;CAWA,AAAO,MAAM,QAA0C;EACrD,MAAM,kBAAkB,OAAO,WAAW,KAAK,UAAU,OAAO;EAChE,MAAM,iBACJ,oBAAoB,OAAO,UAAU,SAAS;GAAE,GAAG;GAAQ,SAAS;EAAgB;EAEtF,OAAO,IAAI,YAAY,KAAK,IAAI,gBAAgB,KAAK,QAAQ;CAC/D;;;;;;;CAQA,MAAa,MAAM,MAAc,QAAkC;EACjE,OAAO,sBAAsB,IAAI;CACnC;;;;;;;;CASA,AAAO,SAAS,QAAgD;EAC9D,OAAO,IAAI,eAAe,KAAK,IAAI,QAAQ,KAAK,QAAQ;CAC1D;;;;;;;;;;;;;;;;;;;;;;;;CAyBA,AAAO,MAAM,QAA+C;EAC1D,MAAM,kBAAkB,OAAO,WAAW,KAAK,UAAU,OAAO;EAChE,MAAM,iBACJ,oBAAoB,OAAO,UAAU,SAAS;GAAE,GAAG;GAAQ,SAAS;EAAgB;EAEtF,IAAI,yBAAyB,OAAO,IAAI,GACtC,OAAO,IAAI,iBAAiB,KAAK,IAAI,gBAAgB,KAAK,QAAQ;EAGpE,OAAO,IAAI,iBAAiB,KAAK,IAAI,gBAAgB,KAAK,QAAQ;CACpE;AACF"}
+{"version":3,"file":"sdk.mjs","names":[],"sources":["../../../../../../../packages/ai-google/src/sdk.ts"],"sourcesContent":["import { GoogleGenAI } from \"@google/genai\";\nimport type {\n  EmbedderContract,\n  ImageModelContract,\n  ModelContract,\n  ModelPricing,\n  SDKAdapterContract,\n} from \"@warlock.js/ai\";\nimport { approximateTokenCount } from \"@warlock.js/ai\";\nimport type {\n  GoogleEmbedderConfig,\n  GoogleImageConfig,\n  GoogleModelConfig,\n  GoogleSDKConfig,\n} from \"./config.type\";\nimport { GoogleEmbedder } from \"./embedder\";\nimport { GeminiImageModel } from \"./gemini-image\";\nimport { GoogleImageModel } from \"./image\";\nimport { googlePricingFor } from \"./known-pricing\";\nimport { GoogleModel } from \"./model\";\n\n/**\n * Pick the transport for an image model id.\n *\n * `ai.models.generateImages` calls `{model}:predict`, and a `gemini-`\n * id sent there comes back `404 … is not supported for predict`\n * (observed verbatim from Google). `generateContent` is what the SDK\n * itself points `generateImages` users at — its deprecation notice\n * reads \"Please use the generateContent method with image models\n * instead\" — so the id has to choose the transport.\n *\n * Runs in this package establish where a `gemini-` id is ACCEPTED, not\n * what it returns: on this transport such an id got as far as a quota\n * error (HTTP 429) instead of the 404. That an image\n * comes back end-to-end once billing is enabled is reported by the\n * maintainer from a locally linked build, not measured here. Whether\n * these models report token usage is still unknown.\n *\n * This is ROUTING, not validation — no id is refused here. An id this\n * function does not recognize takes the `generateImages` route, the\n * only route that existed before Gemini image support landed, so every\n * id that reached Google before still reaches Google the same way and\n * still fails (or succeeds) at the provider.\n *\n * A leading `models/` resource prefix is tolerated, matching the id\n * shapes `inferVisionCapability` already accepts\n * (`models/gemini-1.5-flash-001`).\n */\nfunction usesGeminiImageTransport(name: string): boolean {\n  return name\n    .toLowerCase()\n    .replace(/^models\\//, \"\")\n    .startsWith(\"gemini-\");\n}\n\n/**\n * Google Gemini-backed implementation of `SDKAdapterContract`.\n *\n * **Role.** The package entry point for Gemini models via the\n * `@google/genai` SDK. A single `GoogleSDK` holds one live\n * `GoogleGenAI` client, shared by every `ModelContract` /\n * `EmbedderContract` it produces. Construct one SDK per\n * account/project and reuse it everywhere.\n *\n * **Responsibility.**\n * - Owns: a long-lived `GoogleGenAI` client (auth, Vertex vs Gemini\n *   API) and its lifetime. Factory for `GoogleModel` /\n *   `GoogleEmbedder` instances sharing that client.\n * - Does NOT own: anything per-call — those live in `GoogleModel` /\n *   `GoogleEmbedder` and the agent runtime.\n *\n * Modeled as a class (see §4.2 of code-style.md — \"long-lived state\n * across many calls\"), fronted by FP usage like the other adapters.\n *\n * @example\n * const google = new GoogleSDK({ apiKey: process.env.GEMINI_API_KEY! });\n * const model = google.model({ name: \"gemini-2.5-flash\", temperature: 0.7 });\n * const embedder = google.embedder({ name: \"gemini-embedding-001\" });\n */\nexport class GoogleSDK implements SDKAdapterContract {\n  private readonly ai: GoogleGenAI;\n  private readonly provider: string;\n  private readonly pricing?: Record<string, ModelPricing>;\n\n  public constructor(config: GoogleSDKConfig) {\n    const { provider, pricing, ...clientOptions } = config;\n\n    this.ai = new GoogleGenAI(clientOptions);\n    this.provider = provider ?? \"google\";\n    this.pricing = pricing;\n  }\n\n  /**\n   * Build a `GoogleModel` bound to this SDK's client. Each call\n   * returns a fresh instance; all instances share the underlying\n   * `GoogleGenAI` client. The SDK's `provider` label is forwarded.\n   *\n   * Pricing resolution: per-model `config.pricing` wins; otherwise the\n   * SDK-level registry entry keyed by `config.name`; otherwise the\n   * built-in list price (`GOOGLE_MODEL_PRICING`); otherwise `undefined`\n   * (no cost computed).\n   */\n  public model(config: GoogleModelConfig): ModelContract {\n    const resolvedPricing =\n      config.pricing ?? this.pricing?.[config.name] ?? googlePricingFor(config.name);\n    const resolvedConfig: GoogleModelConfig =\n      resolvedPricing === config.pricing ? config : { ...config, pricing: resolvedPricing };\n\n    return new GoogleModel(this.ai, resolvedConfig, this.provider);\n  }\n\n  /**\n   * Rough token-count estimate. Uses the character-heuristic\n   * (`approximateTokenCount`) from the core package — Gemini's\n   * `countTokens` is a network round-trip; `count()` is intentionally\n   * offline. Good for budgeting/quota guards, not billing.\n   */\n  public async count(text: string, _model?: string): Promise<number> {\n    return approximateTokenCount(text);\n  }\n\n  /**\n   * Build a `GoogleEmbedder` bound to this SDK's client.\n   *\n   * @example\n   * const embedder = google.embedder({ name: \"gemini-embedding-001\" });\n   * const { vector } = await embedder.embed(\"Hello world\");\n   */\n  public embedder(config: GoogleEmbedderConfig): EmbedderContract {\n    return new GoogleEmbedder(this.ai, config, this.provider);\n  }\n\n  /**\n   * Build an image model bound to this SDK's client for use with\n   * `ai.image({ model, prompt })`. `config.name` decides the transport\n   * (see {@link usesGeminiImageTransport}) — a `gemini-` id gets the\n   * `generateContent` implementation, everything else the Imagen\n   * `generateImages` one. No id is rejected locally either way, so an\n   * unsupported model fails at Google, not here.\n   *\n   * The two differ in what usage they can report, which is what the\n   * caller must price for: the Imagen path always returns a zero token\n   * `Usage` (Imagen reports none — price with `{ perImage }`), while the\n   * Gemini path passes through whatever `usageMetadata` Google attaches\n   * (price with `{ input, output }` when tokens come back).\n   *\n   * Pricing resolution mirrors `model()`: per-model `config.pricing`\n   * wins, otherwise the SDK-level registry entry keyed by `config.name`,\n   * otherwise `undefined`.\n   *\n   * @example\n   * const imagen = google.image({ name: \"imagen-4.0-generate-001\" });\n   * const gemini = google.image({ name: \"gemini-3.1-flash-lite-image\" });\n   * const { data } = await ai.image({ model: gemini, prompt: \"a red bicycle\" });\n   */\n  public image(config: GoogleImageConfig): ImageModelContract {\n    const resolvedPricing = config.pricing ?? this.pricing?.[config.name];\n    const resolvedConfig: GoogleImageConfig =\n      resolvedPricing === config.pricing ? config : { ...config, pricing: resolvedPricing };\n\n    if (usesGeminiImageTransport(config.name)) {\n      return new GeminiImageModel(this.ai, resolvedConfig, this.provider);\n    }\n\n    return new GoogleImageModel(this.ai, resolvedConfig, this.provider);\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAgDA,SAAS,yBAAyB,MAAuB;CACvD,OAAO,KACJ,YAAY,CAAC,CACb,QAAQ,aAAa,EAAE,CAAC,CACxB,WAAW,SAAS;AACzB;;;;;;;;;;;;;;;;;;;;;;;;;AA0BA,IAAa,YAAb,MAAqD;CAKnD,AAAO,YAAY,QAAyB;EAC1C,MAAM,EAAE,UAAU,SAAS,GAAG,kBAAkB;EAEhD,KAAK,KAAK,IAAI,YAAY,aAAa;EACvC,KAAK,WAAW,YAAY;EAC5B,KAAK,UAAU;CACjB;;;;;;;;;;;CAYA,AAAO,MAAM,QAA0C;EACrD,MAAM,kBACJ,OAAO,WAAW,KAAK,UAAU,OAAO,SAAS,iBAAiB,OAAO,IAAI;EAC/E,MAAM,iBACJ,oBAAoB,OAAO,UAAU,SAAS;GAAE,GAAG;GAAQ,SAAS;EAAgB;EAEtF,OAAO,IAAI,YAAY,KAAK,IAAI,gBAAgB,KAAK,QAAQ;CAC/D;;;;;;;CAQA,MAAa,MAAM,MAAc,QAAkC;EACjE,OAAO,sBAAsB,IAAI;CACnC;;;;;;;;CASA,AAAO,SAAS,QAAgD;EAC9D,OAAO,IAAI,eAAe,KAAK,IAAI,QAAQ,KAAK,QAAQ;CAC1D;;;;;;;;;;;;;;;;;;;;;;;;CAyBA,AAAO,MAAM,QAA+C;EAC1D,MAAM,kBAAkB,OAAO,WAAW,KAAK,UAAU,OAAO;EAChE,MAAM,iBACJ,oBAAoB,OAAO,UAAU,SAAS;GAAE,GAAG;GAAQ,SAAS;EAAgB;EAEtF,IAAI,yBAAyB,OAAO,IAAI,GACtC,OAAO,IAAI,iBAAiB,KAAK,IAAI,gBAAgB,KAAK,QAAQ;EAGpE,OAAO,IAAI,iBAAiB,KAAK,IAAI,gBAAgB,KAAK,QAAQ;CACpE;AACF"}

package/esm/utils/apply-google-usage.mjs

@@ -1,4 +1,4 @@
-//#region ../ai-google/src/utils/apply-google-usage.ts
+//#region ../../packages/ai-google/src/utils/apply-google-usage.ts
 /**
 * Fold a Gemini `usageMetadata` block into a running neutral `Usage`
 * accumulator. Shared by every `generateContent`-backed surface — the
@@ -14,8 +14,13 @@
 *
 * `total` falls back to `input + output` when Google omits
 * `totalTokenCount`.
+*
+* `promptTokenCount` "includes the number of tokens in the cached
+* content" (https://ai.google.dev/api/generate-content#UsageMetadata), so
+* the usage is marked `inputIncludesCache: true`.
 */
 function applyGoogleUsage(usage, raw) {
+	usage.inputIncludesCache = true;
 	usage.input = raw.promptTokenCount ?? usage.input;
 	usage.output = raw.candidatesTokenCount ?? usage.output;
 	usage.total = raw.totalTokenCount ?? usage.input + usage.output;

package/esm/utils/apply-google-usage.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"apply-google-usage.mjs","names":[],"sources":["../../../../../../../ai-google/src/utils/apply-google-usage.ts"],"sourcesContent":["import type { Usage } from \"@warlock.js/ai\";\nimport type { GenerateContentResponse } from \"@google/genai\";\n\n/** Gemini's per-response token accounting block, as the SDK types it. */\nexport type GoogleUsageMetadata = NonNullable<GenerateContentResponse[\"usageMetadata\"]>;\n\n/**\n * Fold a Gemini `usageMetadata` block into a running neutral `Usage`\n * accumulator. Shared by every `generateContent`-backed surface — the\n * chat model's `complete()`, its streaming loop (where the final chunk\n * carries cumulative totals), and the Gemini image model — so one\n * mapping decides what a Gemini token report means package-wide.\n *\n * Cache-read hits (`cachedContentTokenCount`, implicit or explicit\n * context caching) surface as `cachedTokens`; the thinking-phase tokens\n * of a reasoning model (`thoughtsTokenCount`) surface as\n * `reasoningTokens`. Both are emitted only when reported `> 0` so an\n * absent channel leaves the field undefined rather than a false zero.\n *\n * `total` falls back to `input + output` when Google omits\n * `totalTokenCount`.\n */\nexport function applyGoogleUsage(usage: Usage, raw: GoogleUsageMetadata): void {\n  usage.input = raw.promptTokenCount ?? usage.input;\n  usage.output = raw.candidatesTokenCount ?? usage.output;\n  usage.total = raw.totalTokenCount ?? usage.input + usage.output;\n\n  const cached = raw.cachedContentTokenCount;\n\n  if (cached && cached > 0) {\n    usage.cachedTokens = cached;\n  }\n\n  const reasoning = raw.thoughtsTokenCount;\n\n  if (reasoning && reasoning > 0) {\n    usage.reasoningTokens = reasoning;\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;AAsBA,SAAgB,iBAAiB,OAAc,KAAgC;CAC7E,MAAM,QAAQ,IAAI,oBAAoB,MAAM;CAC5C,MAAM,SAAS,IAAI,wBAAwB,MAAM;CACjD,MAAM,QAAQ,IAAI,mBAAmB,MAAM,QAAQ,MAAM;CAEzD,MAAM,SAAS,IAAI;CAEnB,IAAI,UAAU,SAAS,GACrB,MAAM,eAAe;CAGvB,MAAM,YAAY,IAAI;CAEtB,IAAI,aAAa,YAAY,GAC3B,MAAM,kBAAkB;AAE5B"}
+{"version":3,"file":"apply-google-usage.mjs","names":[],"sources":["../../../../../../../../packages/ai-google/src/utils/apply-google-usage.ts"],"sourcesContent":["import type { Usage } from \"@warlock.js/ai\";\nimport type { GenerateContentResponse } from \"@google/genai\";\n\n/** Gemini's per-response token accounting block, as the SDK types it. */\nexport type GoogleUsageMetadata = NonNullable<GenerateContentResponse[\"usageMetadata\"]>;\n\n/**\n * Fold a Gemini `usageMetadata` block into a running neutral `Usage`\n * accumulator. Shared by every `generateContent`-backed surface — the\n * chat model's `complete()`, its streaming loop (where the final chunk\n * carries cumulative totals), and the Gemini image model — so one\n * mapping decides what a Gemini token report means package-wide.\n *\n * Cache-read hits (`cachedContentTokenCount`, implicit or explicit\n * context caching) surface as `cachedTokens`; the thinking-phase tokens\n * of a reasoning model (`thoughtsTokenCount`) surface as\n * `reasoningTokens`. Both are emitted only when reported `> 0` so an\n * absent channel leaves the field undefined rather than a false zero.\n *\n * `total` falls back to `input + output` when Google omits\n * `totalTokenCount`.\n *\n * `promptTokenCount` \"includes the number of tokens in the cached\n * content\" (https://ai.google.dev/api/generate-content#UsageMetadata), so\n * the usage is marked `inputIncludesCache: true`.\n */\nexport function applyGoogleUsage(usage: Usage, raw: GoogleUsageMetadata): void {\n  usage.inputIncludesCache = true;\n  usage.input = raw.promptTokenCount ?? usage.input;\n  usage.output = raw.candidatesTokenCount ?? usage.output;\n  usage.total = raw.totalTokenCount ?? usage.input + usage.output;\n\n  const cached = raw.cachedContentTokenCount;\n\n  if (cached && cached > 0) {\n    usage.cachedTokens = cached;\n  }\n\n  const reasoning = raw.thoughtsTokenCount;\n\n  if (reasoning && reasoning > 0) {\n    usage.reasoningTokens = reasoning;\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;AA0BA,SAAgB,iBAAiB,OAAc,KAAgC;CAC7E,MAAM,qBAAqB;CAC3B,MAAM,QAAQ,IAAI,oBAAoB,MAAM;CAC5C,MAAM,SAAS,IAAI,wBAAwB,MAAM;CACjD,MAAM,QAAQ,IAAI,mBAAmB,MAAM,QAAQ,MAAM;CAEzD,MAAM,SAAS,IAAI;CAEnB,IAAI,UAAU,SAAS,GACrB,MAAM,eAAe;CAGvB,MAAM,YAAY,IAAI;CAEtB,IAAI,aAAa,YAAY,GAC3B,MAAM,kBAAkB;AAE5B"}

package/esm/utils/map-finish-reason.mjs

@@ -1,4 +1,4 @@
-//#region ../ai-google/src/utils/map-finish-reason.ts
+//#region ../../packages/ai-google/src/utils/map-finish-reason.ts
 const finishReasonMap = {
 	STOP: "stop",
 	MAX_TOKENS: "length"