@warlock.js/ai-google 4.1.1

package/esm/model.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"model.mjs","names":[],"sources":["../../../../../@warlock.js/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 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 { mapFinishReason, toGoogleContents, toGoogleTools, wrapGoogleError } from \"./utils\";\n\nconst LOG_MODULE = \"ai.google\";\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    this.capabilities = {\n      structuredOutput: config.structuredOutput ?? true,\n      vision: config.vision ?? inferVisionCapability(config.name),\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          this.applyUsage(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    };\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   * Cache-read tokens are surfaced as `cachedTokens` only when\n   * non-zero. Absent usage 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      this.applyUsage(usage, response.usageMetadata);\n    }\n\n    return usage;\n  }\n\n  /**\n   * Fold a Gemini `usageMetadata` block into the running neutral\n   * `Usage` accumulator. Shared by `complete()` and the streaming\n   * loop (where the final chunk carries cumulative totals).\n   */\n  private applyUsage(\n    usage: Usage,\n    raw: NonNullable<GenerateContentResponse[\"usageMetadata\"]>,\n  ): 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\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":";;;;;;;;;AAsBA,MAAM,aAAa;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAiCnB,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,KAAK,eAAe;GAClB,kBAAkB,OAAO,oBAAoB;GAC7C,QAAQ,OAAO,UAAU,sBAAsB,OAAO,IAAI;EAC5D;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,IAAI,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,IAAI,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,IAAI;IAE/C,IAAI,iBACF,kBAAkB;IAGpB,IAAI,MAAM,eACR,KAAK,WAAW,OAAO,MAAM,aAAa;GAE9C;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;EACvD;CACF;;;;;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,IAAI,SAAS,SAAS,CAAC,GAExD,KAAK,SAAS,KAAK,eAAe,IAAI,CAAC,EACvC,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;;;;;;CAOA,AAAQ,aAAa,UAA0C;EAC7D,MAAM,QAAe;GAAE,OAAO;GAAG,QAAQ;GAAG,OAAO;EAAE;EAErD,IAAI,SAAS,eACX,KAAK,WAAW,OAAO,SAAS,aAAa;EAG/C,OAAO;CACT;;;;;;CAOA,AAAQ,WACN,OACA,KACM;EACN,MAAM,QAAQ,IAAI,oBAAoB,MAAM;EAC5C,MAAM,SAAS,IAAI,wBAAwB,MAAM;EACjD,MAAM,QAAQ,IAAI,mBAAmB,MAAM,QAAQ,MAAM;EAEzD,MAAM,SAAS,IAAI;EAEnB,IAAI,UAAU,SAAS,GACrB,MAAM,eAAe;CAEzB;;;;;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/sdk.d.mts

@@ -0,0 +1,62 @@
+import { GoogleEmbedderConfig, GoogleModelConfig, GoogleSDKConfig } from "./config.type.mjs";
+import { EmbedderContract, ModelContract, SDKAdapterContract } from "@warlock.js/ai";
+
+//#region ../../@warlock.js/ai-google/src/sdk.d.ts
+/**
+ * Google Gemini-backed implementation of `SDKAdapterContract`.
+ *
+ * **Role.** The package entry point for Gemini models via the
+ * `@google/genai` SDK. A single `GoogleSDK` holds one live
+ * `GoogleGenAI` client, shared by every `ModelContract` /
+ * `EmbedderContract` it produces. Construct one SDK per
+ * account/project and reuse it everywhere.
+ *
+ * **Responsibility.**
+ * - Owns: a long-lived `GoogleGenAI` client (auth, Vertex vs Gemini
+ *   API) and its lifetime. Factory for `GoogleModel` /
+ *   `GoogleEmbedder` instances sharing that client.
+ * - Does NOT own: anything per-call — those live in `GoogleModel` /
+ *   `GoogleEmbedder` and the agent runtime.
+ *
+ * Modeled as a class (see §4.2 of code-style.md — "long-lived state
+ * across many calls"), fronted by FP usage like the other adapters.
+ *
+ * @example
+ * const google = new GoogleSDK({ apiKey: process.env.GEMINI_API_KEY! });
+ * const model = google.model({ name: "gemini-2.5-flash", temperature: 0.7 });
+ * const embedder = google.embedder({ name: "gemini-embedding-001" });
+ */
+declare class GoogleSDK implements SDKAdapterContract {
+  private readonly ai;
+  private readonly provider;
+  private readonly pricing?;
+  constructor(config: GoogleSDKConfig);
+  /**
+   * Build a `GoogleModel` bound to this SDK's client. Each call
+   * returns a fresh instance; all instances share the underlying
+   * `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).
+   */
+  model(config: GoogleModelConfig): ModelContract;
+  /**
+   * Rough token-count estimate. Uses the character-heuristic
+   * (`approximateTokenCount`) from the core package — Gemini's
+   * `countTokens` is a network round-trip; `count()` is intentionally
+   * offline. Good for budgeting/quota guards, not billing.
+   */
+  count(text: string, _model?: string): Promise<number>;
+  /**
+   * Build a `GoogleEmbedder` bound to this SDK's client.
+   *
+   * @example
+   * const embedder = google.embedder({ name: "gemini-embedding-001" });
+   * const { vector } = await embedder.embed("Hello world");
+   */
+  embedder(config: GoogleEmbedderConfig): EmbedderContract;
+}
+//#endregion
+export { GoogleSDK };
+//# sourceMappingURL=sdk.d.mts.map

package/esm/sdk.d.mts.map

@@ -0,0 +1 @@
+{"version":3,"file":"sdk.d.mts","names":[],"sources":["../../../../../@warlock.js/ai-google/src/sdk.ts"],"mappings":";;;;;;AAwCA;;;;;;;;;;;;;;;;;;;;;;cAAa,SAAA,YAAqB,kBAAA;EAAA,iBACf,EAAA;EAAA,iBACA,QAAA;EAAA,iBACA,OAAA;cAEE,MAAA,EAAQ,eAAA;EA0CpB;;;;;AAAwD;;;;EAzBxD,KAAA,CAAM,MAAA,EAAQ,iBAAA,GAAoB,aAAA;;;;;;;EAc5B,KAAA,CAAM,IAAA,UAAc,MAAA,YAAkB,OAAA;;;;;;;;EAW5C,QAAA,CAAS,MAAA,EAAQ,oBAAA,GAAuB,gBAAA;AAAA"}

package/esm/sdk.mjs

@@ -0,0 +1,78 @@
+import { GoogleEmbedder } from "./embedder.mjs";
+import { GoogleModel } from "./model.mjs";
+import { GoogleGenAI } from "@google/genai";
+import { approximateTokenCount } from "@warlock.js/ai";
+
+//#region ../../@warlock.js/ai-google/src/sdk.ts
+/**
+* Google Gemini-backed implementation of `SDKAdapterContract`.
+*
+* **Role.** The package entry point for Gemini models via the
+* `@google/genai` SDK. A single `GoogleSDK` holds one live
+* `GoogleGenAI` client, shared by every `ModelContract` /
+* `EmbedderContract` it produces. Construct one SDK per
+* account/project and reuse it everywhere.
+*
+* **Responsibility.**
+* - Owns: a long-lived `GoogleGenAI` client (auth, Vertex vs Gemini
+*   API) and its lifetime. Factory for `GoogleModel` /
+*   `GoogleEmbedder` instances sharing that client.
+* - Does NOT own: anything per-call — those live in `GoogleModel` /
+*   `GoogleEmbedder` and the agent runtime.
+*
+* Modeled as a class (see §4.2 of code-style.md — "long-lived state
+* across many calls"), fronted by FP usage like the other adapters.
+*
+* @example
+* const google = new GoogleSDK({ apiKey: process.env.GEMINI_API_KEY! });
+* const model = google.model({ name: "gemini-2.5-flash", temperature: 0.7 });
+* const embedder = google.embedder({ name: "gemini-embedding-001" });
+*/
+var GoogleSDK = class {
+	constructor(config) {
+		const { provider, pricing, ...clientOptions } = config;
+		this.ai = new GoogleGenAI(clientOptions);
+		this.provider = provider ?? "google";
+		this.pricing = pricing;
+	}
+	/**
+	* Build a `GoogleModel` bound to this SDK's client. Each call
+	* returns a fresh instance; all instances share the underlying
+	* `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).
+	*/
+	model(config) {
+		const resolvedPricing = config.pricing ?? this.pricing?.[config.name];
+		const resolvedConfig = resolvedPricing === config.pricing ? config : {
+			...config,
+			pricing: resolvedPricing
+		};
+		return new GoogleModel(this.ai, resolvedConfig, this.provider);
+	}
+	/**
+	* Rough token-count estimate. Uses the character-heuristic
+	* (`approximateTokenCount`) from the core package — Gemini's
+	* `countTokens` is a network round-trip; `count()` is intentionally
+	* offline. Good for budgeting/quota guards, not billing.
+	*/
+	async count(text, _model) {
+		return approximateTokenCount(text);
+	}
+	/**
+	* Build a `GoogleEmbedder` bound to this SDK's client.
+	*
+	* @example
+	* const embedder = google.embedder({ name: "gemini-embedding-001" });
+	* const { vector } = await embedder.embed("Hello world");
+	*/
+	embedder(config) {
+		return new GoogleEmbedder(this.ai, config, this.provider);
+	}
+};
+
+//#endregion
+export { GoogleSDK };
+//# sourceMappingURL=sdk.mjs.map

package/esm/sdk.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"sdk.mjs","names":[],"sources":["../../../../../@warlock.js/ai-google/src/sdk.ts"],"sourcesContent":["import { GoogleGenAI } from \"@google/genai\";\nimport type {\n  EmbedderContract,\n  ModelContract,\n  ModelPricing,\n  SDKAdapterContract,\n} from \"@warlock.js/ai\";\nimport { approximateTokenCount } from \"@warlock.js/ai\";\nimport type {\n  GoogleEmbedderConfig,\n  GoogleModelConfig,\n  GoogleSDKConfig,\n} from \"./config.type\";\nimport { GoogleEmbedder } from \"./embedder\";\nimport { GoogleModel } from \"./model\";\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"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAwCA,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;AACF"}

package/esm/utils/index.mjs

@@ -0,0 +1,6 @@
+import { mapFinishReason } from "./map-finish-reason.mjs";
+import { toGoogleContents } from "./to-google-contents.mjs";
+import { toGoogleTools } from "./to-google-tools.mjs";
+import { wrapGoogleError } from "./wrap-google-error.mjs";
+
+export {  };

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

@@ -0,0 +1,34 @@
+//#region ../../@warlock.js/ai-google/src/utils/map-finish-reason.ts
+const finishReasonMap = {
+	STOP: "stop",
+	MAX_TOKENS: "length"
+};
+/**
+* Map Gemini's `FinishReason` enum value to the normalized
+* `FinishReason` union.
+*
+* `STOP` is the natural terminal. `MAX_TOKENS` maps to `length`.
+* Everything else — `SAFETY`, `RECITATION`, `BLOCKLIST`,
+* `PROHIBITED_CONTENT`, `SPII`, `MALFORMED_FUNCTION_CALL`,
+* `UNEXPECTED_TOOL_CALL`, `LANGUAGE`, `OTHER`,
+* `FINISH_REASON_UNSPECIFIED`, `null`, or any future value — falls
+* through to `"error"`.
+*
+* Note: Gemini reports `STOP` even when the turn ended in a function
+* call (it has no `tool_use` reason). `GoogleModel` overrides the
+* mapped reason to `"tool_calls"` when the response carries function
+* calls — this map intentionally stays purely about the raw signal.
+*
+* @example
+* mapFinishReason("STOP");        // "stop"
+* mapFinishReason("MAX_TOKENS");  // "length"
+* mapFinishReason("SAFETY");      // "error"
+* mapFinishReason(undefined);     // "error"
+*/
+function mapFinishReason(raw) {
+	return finishReasonMap[raw ?? ""] ?? "error";
+}
+
+//#endregion
+export { mapFinishReason };
+//# sourceMappingURL=map-finish-reason.mjs.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"map-finish-reason.mjs","names":[],"sources":["../../../../../../@warlock.js/ai-google/src/utils/map-finish-reason.ts"],"sourcesContent":["import type { FinishReason } from \"@warlock.js/ai\";\n\nconst finishReasonMap: Record<string, FinishReason> = {\n  STOP: \"stop\",\n  MAX_TOKENS: \"length\",\n};\n\n/**\n * Map Gemini's `FinishReason` enum value to the normalized\n * `FinishReason` union.\n *\n * `STOP` is the natural terminal. `MAX_TOKENS` maps to `length`.\n * Everything else — `SAFETY`, `RECITATION`, `BLOCKLIST`,\n * `PROHIBITED_CONTENT`, `SPII`, `MALFORMED_FUNCTION_CALL`,\n * `UNEXPECTED_TOOL_CALL`, `LANGUAGE`, `OTHER`,\n * `FINISH_REASON_UNSPECIFIED`, `null`, or any future value — falls\n * through to `\"error\"`.\n *\n * Note: Gemini reports `STOP` even when the turn ended in a function\n * call (it has no `tool_use` reason). `GoogleModel` overrides the\n * mapped reason to `\"tool_calls\"` when the response carries function\n * calls — this map intentionally stays purely about the raw signal.\n *\n * @example\n * mapFinishReason(\"STOP\");        // \"stop\"\n * mapFinishReason(\"MAX_TOKENS\");  // \"length\"\n * mapFinishReason(\"SAFETY\");      // \"error\"\n * mapFinishReason(undefined);     // \"error\"\n */\nexport function mapFinishReason(raw: string | null | undefined): FinishReason {\n  return finishReasonMap[raw ?? \"\"] ?? \"error\";\n}\n"],"mappings":";AAEA,MAAM,kBAAgD;CACpD,MAAM;CACN,YAAY;AACd;;;;;;;;;;;;;;;;;;;;;;;AAwBA,SAAgB,gBAAgB,KAA8C;CAC5E,OAAO,gBAAgB,OAAO,OAAO;AACvC"}

package/esm/utils/to-google-contents.mjs

@@ -0,0 +1,120 @@
+import { InvalidRequestError, safeJsonParse } from "@warlock.js/ai";
+
+//#region ../../@warlock.js/ai-google/src/utils/to-google-contents.ts
+/**
+* Convert vendor-neutral `Message[]` into Gemini's request shape.
+*
+* Gemini specifics this function absorbs:
+*
+* 1. **No `system` role.** System messages concatenate into the
+*    separate `systemInstruction` config field.
+* 2. **Role names differ.** Neutral `assistant` → Gemini `"model"`;
+*    `user` stays `"user"`.
+* 3. **Tool results are `user` turns.** A neutral `tool` message
+*    becomes a `"user"` content with a single `functionResponse` part.
+* 4. **Tool calls are `functionCall` parts.** An assistant message
+*    with `toolCalls` becomes a `"model"` content: an optional leading
+*    `text` part followed by one `functionCall` part per call.
+*
+* @example
+* const { systemInstruction, contents } = toGoogleContents([
+*   { role: "system", content: "Be concise." },
+*   { role: "user", content: "Hi" },
+* ]);
+*/
+function toGoogleContents(messages) {
+	const systemParts = [];
+	const contents = [];
+	for (const message of messages) {
+		if (message.role === "system") {
+			systemParts.push(stringifyContent(message.content));
+			continue;
+		}
+		if (message.role === "tool") {
+			contents.push({
+				role: "user",
+				parts: [{ functionResponse: {
+					name: message.toolCallId ?? "",
+					response: toResponseObject(stringifyContent(message.content))
+				} }]
+			});
+			continue;
+		}
+		if (message.role === "assistant" && message.toolCalls && message.toolCalls.length > 0) {
+			const parts = [];
+			const text = stringifyContent(message.content);
+			if (text) parts.push({ text });
+			for (const toolCall of message.toolCalls) {
+				const thoughtSignature = toolCall.providerMetadata?.thoughtSignature;
+				parts.push({
+					...typeof thoughtSignature === "string" ? { thoughtSignature } : {},
+					functionCall: {
+						name: toolCall.name,
+						args: toolCall.input ?? {}
+					}
+				});
+			}
+			contents.push({
+				role: "model",
+				parts
+			});
+			continue;
+		}
+		if (message.role === "user" && Array.isArray(message.content)) {
+			contents.push({
+				role: "user",
+				parts: message.content.map(toGooglePart)
+			});
+			continue;
+		}
+		contents.push({
+			role: message.role === "assistant" ? "model" : "user",
+			parts: [{ text: stringifyContent(message.content) }]
+		});
+	}
+	return {
+		systemInstruction: systemParts.length > 0 ? systemParts.join("\n\n") : void 0,
+		contents
+	};
+}
+/**
+* Multipart content is only meaningful on user messages — for any
+* other role collapse a `ContentPart[]` to concatenated text. Plain
+* strings pass through unchanged.
+*/
+function stringifyContent(content) {
+	if (typeof content === "string") return content;
+	return content.filter((part) => part.type === "text").map((part) => part.text).join("");
+}
+/**
+* Gemini's `functionResponse.response` must be a JSON object. Tool
+* results arrive as a string (usually stringified JSON) — parse it
+* when it is a JSON object, otherwise wrap the raw string under a
+* `result` key so the model always receives a well-formed object.
+*/
+function toResponseObject(raw) {
+	const parsed = safeJsonParse(raw, void 0);
+	if (parsed !== null && typeof parsed === "object" && !Array.isArray(parsed)) return parsed;
+	return { result: raw };
+}
+/**
+* Map a resolved `ContentPart` to a Gemini `Part`. Images are sent as
+* inline base64 (`inlineData`). Gemini's `generateContent` does not
+* fetch arbitrary remote URLs (only Files API / GCS URIs via
+* `fileData`), so a neutral `{ url }` image surfaces a typed
+* `InvalidRequestError` upfront rather than a downstream Gemini fault.
+* The agent resolves attachments before this point, so nothing is
+* read or fetched here.
+*/
+function toGooglePart(part) {
+	if (part.type === "text") return { text: part.text };
+	if ("url" in part.source) throw new InvalidRequestError("Gemini generateContent does not fetch remote-URL images; supply base64 image bytes instead.");
+	return { inlineData: {
+		mimeType: part.source.mediaType,
+		data: part.source.base64
+	} };
+}
+
+//#endregion
+export { toGoogleContents };
+//# sourceMappingURL=to-google-contents.mjs.map

package/esm/utils/to-google-contents.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"to-google-contents.mjs","names":[],"sources":["../../../../../../@warlock.js/ai-google/src/utils/to-google-contents.ts"],"sourcesContent":["import { InvalidRequestError, safeJsonParse, type ContentPart, type Message } from \"@warlock.js/ai\";\nimport type { Content, Part } from \"@google/genai\";\n\n/**\n * Result of splitting a vendor-neutral `Message[]` for Gemini's\n * `generateContent`: the system prompt is hoisted to a separate\n * `systemInstruction` string (Gemini has no `\"system\"` role — content\n * roles must be `\"user\"` or `\"model\"`), and the remaining turns map to\n * `Content[]`.\n */\nexport type GoogleContents = {\n  systemInstruction: string | undefined;\n  contents: Content[];\n};\n\n/**\n * Convert vendor-neutral `Message[]` into Gemini's request shape.\n *\n * Gemini specifics this function absorbs:\n *\n * 1. **No `system` role.** System messages concatenate into the\n *    separate `systemInstruction` config field.\n * 2. **Role names differ.** Neutral `assistant` → Gemini `\"model\"`;\n *    `user` stays `\"user\"`.\n * 3. **Tool results are `user` turns.** A neutral `tool` message\n *    becomes a `\"user\"` content with a single `functionResponse` part.\n * 4. **Tool calls are `functionCall` parts.** An assistant message\n *    with `toolCalls` becomes a `\"model\"` content: an optional leading\n *    `text` part followed by one `functionCall` part per call.\n *\n * @example\n * const { systemInstruction, contents } = toGoogleContents([\n *   { role: \"system\", content: \"Be concise.\" },\n *   { role: \"user\", content: \"Hi\" },\n * ]);\n */\nexport function toGoogleContents(messages: Message[]): GoogleContents {\n  const systemParts: string[] = [];\n  const contents: Content[] = [];\n\n  for (const message of messages) {\n    if (message.role === \"system\") {\n      systemParts.push(stringifyContent(message.content));\n\n      continue;\n    }\n\n    if (message.role === \"tool\") {\n      contents.push({\n        role: \"user\",\n        parts: [\n          {\n            // Gemini matches a `functionResponse` to its `functionCall`\n            // by `name` (the Developer API has no call ids). `name` is\n            // the neutral `toolCallId`, which `GoogleModel` set to the\n            // function name. The wire `id` is intentionally omitted —\n            // an empty/synthetic id is rejected as an invalid argument.\n            functionResponse: {\n              name: message.toolCallId ?? \"\",\n              response: toResponseObject(stringifyContent(message.content)),\n            },\n          },\n        ],\n      });\n\n      continue;\n    }\n\n    if (message.role === \"assistant\" && message.toolCalls && message.toolCalls.length > 0) {\n      const parts: Part[] = [];\n      const text = stringifyContent(message.content);\n\n      if (text) {\n        parts.push({ text });\n      }\n\n      for (const toolCall of message.toolCalls) {\n        // Replay the opaque `thoughtSignature` Gemini attached to this\n        // function call on the original turn. Thinking models reject\n        // the follow-up request with a 400 if the signature is missing\n        // from the echoed `functionCall` part. Captured by\n        // `GoogleModel.partToToolCall` into `providerMetadata`.\n        const thoughtSignature = toolCall.providerMetadata?.thoughtSignature;\n\n        parts.push({\n          ...(typeof thoughtSignature === \"string\" ? { thoughtSignature } : {}),\n          // `id` omitted deliberately — Gemini Developer API function\n          // calls have no ids; echoing an empty/synthetic one is\n          // rejected as an invalid argument. Matched by `name`.\n          functionCall: {\n            name: toolCall.name,\n            args: (toolCall.input ?? {}) as Record<string, unknown>,\n          },\n        });\n      }\n\n      contents.push({ role: \"model\", parts });\n\n      continue;\n    }\n\n    if (message.role === \"user\" && Array.isArray(message.content)) {\n      contents.push({ role: \"user\", parts: message.content.map(toGooglePart) });\n\n      continue;\n    }\n\n    contents.push({\n      role: message.role === \"assistant\" ? \"model\" : \"user\",\n      parts: [{ text: stringifyContent(message.content) }],\n    });\n  }\n\n  return {\n    systemInstruction: systemParts.length > 0 ? systemParts.join(\"\\n\\n\") : undefined,\n    contents,\n  };\n}\n\n/**\n * Multipart content is only meaningful on user messages — for any\n * other role collapse a `ContentPart[]` to concatenated text. Plain\n * strings pass through unchanged.\n */\nfunction stringifyContent(content: string | ContentPart[]): string {\n  if (typeof content === \"string\") {\n    return content;\n  }\n\n  return content\n    .filter((part): part is { type: \"text\"; text: string } => part.type === \"text\")\n    .map((part) => part.text)\n    .join(\"\");\n}\n\n/**\n * Gemini's `functionResponse.response` must be a JSON object. Tool\n * results arrive as a string (usually stringified JSON) — parse it\n * when it is a JSON object, otherwise wrap the raw string under a\n * `result` key so the model always receives a well-formed object.\n */\nfunction toResponseObject(raw: string): Record<string, unknown> {\n  const parsed = safeJsonParse<unknown>(raw, undefined);\n\n  if (parsed !== null && typeof parsed === \"object\" && !Array.isArray(parsed)) {\n    return parsed as Record<string, unknown>;\n  }\n\n  return { result: raw };\n}\n\n/**\n * Map a resolved `ContentPart` to a Gemini `Part`. Images are sent as\n * inline base64 (`inlineData`). Gemini's `generateContent` does not\n * fetch arbitrary remote URLs (only Files API / GCS URIs via\n * `fileData`), so a neutral `{ url }` image surfaces a typed\n * `InvalidRequestError` upfront rather than a downstream Gemini fault.\n * The agent resolves attachments before this point, so nothing is\n * read or fetched here.\n */\nfunction toGooglePart(part: ContentPart): Part {\n  if (part.type === \"text\") {\n    return { text: part.text };\n  }\n\n  if (\"url\" in part.source) {\n    throw new InvalidRequestError(\n      \"Gemini generateContent does not fetch remote-URL images; supply base64 image bytes instead.\",\n    );\n  }\n\n  return {\n    inlineData: { mimeType: part.source.mediaType, data: part.source.base64 },\n  };\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;AAoCA,SAAgB,iBAAiB,UAAqC;CACpE,MAAM,cAAwB,CAAC;CAC/B,MAAM,WAAsB,CAAC;CAE7B,KAAK,MAAM,WAAW,UAAU;EAC9B,IAAI,QAAQ,SAAS,UAAU;GAC7B,YAAY,KAAK,iBAAiB,QAAQ,OAAO,CAAC;GAElD;EACF;EAEA,IAAI,QAAQ,SAAS,QAAQ;GAC3B,SAAS,KAAK;IACZ,MAAM;IACN,OAAO,CACL,EAME,kBAAkB;KAChB,MAAM,QAAQ,cAAc;KAC5B,UAAU,iBAAiB,iBAAiB,QAAQ,OAAO,CAAC;IAC9D,EACF,CACF;GACF,CAAC;GAED;EACF;EAEA,IAAI,QAAQ,SAAS,eAAe,QAAQ,aAAa,QAAQ,UAAU,SAAS,GAAG;GACrF,MAAM,QAAgB,CAAC;GACvB,MAAM,OAAO,iBAAiB,QAAQ,OAAO;GAE7C,IAAI,MACF,MAAM,KAAK,EAAE,KAAK,CAAC;GAGrB,KAAK,MAAM,YAAY,QAAQ,WAAW;IAMxC,MAAM,mBAAmB,SAAS,kBAAkB;IAEpD,MAAM,KAAK;KACT,GAAI,OAAO,qBAAqB,WAAW,EAAE,iBAAiB,IAAI,CAAC;KAInE,cAAc;MACZ,MAAM,SAAS;MACf,MAAO,SAAS,SAAS,CAAC;KAC5B;IACF,CAAC;GACH;GAEA,SAAS,KAAK;IAAE,MAAM;IAAS;GAAM,CAAC;GAEtC;EACF;EAEA,IAAI,QAAQ,SAAS,UAAU,MAAM,QAAQ,QAAQ,OAAO,GAAG;GAC7D,SAAS,KAAK;IAAE,MAAM;IAAQ,OAAO,QAAQ,QAAQ,IAAI,YAAY;GAAE,CAAC;GAExE;EACF;EAEA,SAAS,KAAK;GACZ,MAAM,QAAQ,SAAS,cAAc,UAAU;GAC/C,OAAO,CAAC,EAAE,MAAM,iBAAiB,QAAQ,OAAO,EAAE,CAAC;EACrD,CAAC;CACH;CAEA,OAAO;EACL,mBAAmB,YAAY,SAAS,IAAI,YAAY,KAAK,MAAM,IAAI;EACvE;CACF;AACF;;;;;;AAOA,SAAS,iBAAiB,SAAyC;CACjE,IAAI,OAAO,YAAY,UACrB,OAAO;CAGT,OAAO,QACJ,QAAQ,SAAiD,KAAK,SAAS,MAAM,EAC7E,KAAK,SAAS,KAAK,IAAI,EACvB,KAAK,EAAE;AACZ;;;;;;;AAQA,SAAS,iBAAiB,KAAsC;CAC9D,MAAM,SAAS,cAAuB,KAAK,MAAS;CAEpD,IAAI,WAAW,QAAQ,OAAO,WAAW,YAAY,CAAC,MAAM,QAAQ,MAAM,GACxE,OAAO;CAGT,OAAO,EAAE,QAAQ,IAAI;AACvB;;;;;;;;;;AAWA,SAAS,aAAa,MAAyB;CAC7C,IAAI,KAAK,SAAS,QAChB,OAAO,EAAE,MAAM,KAAK,KAAK;CAG3B,IAAI,SAAS,KAAK,QAChB,MAAM,IAAI,oBACR,6FACF;CAGF,OAAO,EACL,YAAY;EAAE,UAAU,KAAK,OAAO;EAAW,MAAM,KAAK,OAAO;CAAO,EAC1E;AACF"}

package/esm/utils/to-google-tools.mjs

@@ -0,0 +1,41 @@
+import { extractJsonSchema } from "@warlock.js/ai";
+
+//#region ../../@warlock.js/ai-google/src/utils/to-google-tools.ts
+/**
+* Convert vendor-neutral `ToolConfig[]` into Gemini's `tools` array —
+* a single `Tool` carrying one `functionDeclarations` entry per tool.
+*
+* The input schema is forwarded via `parametersJsonSchema` (raw JSON
+* Schema, mutually exclusive with Gemini's typed `parameters`).
+* Non-object extractions degrade to a parameterless object so
+* registration never fails.
+*
+* Returns `undefined` when there are no tools so the caller can omit
+* `config.tools` entirely.
+*
+* @example
+* const tools = toGoogleTools([weatherTool]);
+* await ai.models.generateContent({ model, contents, config: { tools } });
+*/
+function toGoogleTools(tools) {
+	if (!tools || tools.length === 0) return;
+	return [{ functionDeclarations: tools.map((tool) => ({
+		name: tool.name,
+		description: tool.description,
+		parametersJsonSchema: toJsonSchema(tool.input)
+	})) }];
+}
+/**
+* Resolve a tool's input schema to a JSON-Schema object. Gemini wants
+* an object root for function parameters; anything else (or a failed
+* extraction) degrades to a parameterless object.
+*/
+function toJsonSchema(input) {
+	const schema = extractJsonSchema(input);
+	if (schema && schema.type === "object") return schema;
+	return { type: "object" };
+}
+
+//#endregion
+export { toGoogleTools };
+//# sourceMappingURL=to-google-tools.mjs.map

package/esm/utils/to-google-tools.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"to-google-tools.mjs","names":[],"sources":["../../../../../../@warlock.js/ai-google/src/utils/to-google-tools.ts"],"sourcesContent":["import { extractJsonSchema, type ToolConfig } from \"@warlock.js/ai\";\nimport type { Tool } from \"@google/genai\";\n\n/**\n * Convert vendor-neutral `ToolConfig[]` into Gemini's `tools` array —\n * a single `Tool` carrying one `functionDeclarations` entry per tool.\n *\n * The input schema is forwarded via `parametersJsonSchema` (raw JSON\n * Schema, mutually exclusive with Gemini's typed `parameters`).\n * Non-object extractions degrade to a parameterless object so\n * registration never fails.\n *\n * Returns `undefined` when there are no tools so the caller can omit\n * `config.tools` entirely.\n *\n * @example\n * const tools = toGoogleTools([weatherTool]);\n * await ai.models.generateContent({ model, contents, config: { tools } });\n */\nexport function toGoogleTools(\n  tools: ToolConfig<unknown, unknown>[] | undefined,\n): Tool[] | undefined {\n  if (!tools || tools.length === 0) {\n    return undefined;\n  }\n\n  return [\n    {\n      functionDeclarations: tools.map((tool) => ({\n        name: tool.name,\n        description: tool.description,\n        parametersJsonSchema: toJsonSchema(tool.input),\n      })),\n    },\n  ];\n}\n\n/**\n * Resolve a tool's input schema to a JSON-Schema object. Gemini wants\n * an object root for function parameters; anything else (or a failed\n * extraction) degrades to a parameterless object.\n */\nfunction toJsonSchema(input: ToolConfig<unknown, unknown>[\"input\"]): Record<string, unknown> {\n  const schema = extractJsonSchema(input);\n\n  if (schema && schema.type === \"object\") {\n    return schema;\n  }\n\n  return { type: \"object\" };\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;AAmBA,SAAgB,cACd,OACoB;CACpB,IAAI,CAAC,SAAS,MAAM,WAAW,GAC7B;CAGF,OAAO,CACL,EACE,sBAAsB,MAAM,KAAK,UAAU;EACzC,MAAM,KAAK;EACX,aAAa,KAAK;EAClB,sBAAsB,aAAa,KAAK,KAAK;CAC/C,EAAE,EACJ,CACF;AACF;;;;;;AAOA,SAAS,aAAa,OAAuE;CAC3F,MAAM,SAAS,kBAAkB,KAAK;CAEtC,IAAI,UAAU,OAAO,SAAS,UAC5B,OAAO;CAGT,OAAO,EAAE,MAAM,SAAS;AAC1B"}

package/esm/utils/wrap-google-error.mjs

@@ -0,0 +1,108 @@
+import { ApiError } from "@google/genai";
+import { AIError, ContextLengthExceededError, InvalidRequestError, ProviderAuthError, ProviderError, ProviderRateLimitError, ProviderTimeoutError } from "@warlock.js/ai";
+
+//#region ../../@warlock.js/ai-google/src/utils/wrap-google-error.ts
+/**
+* Wrap any thrown value caught inside the Gemini adapter into the
+* appropriate `@warlock.js/ai` `AIError` subclass.
+*
+* **Dispatch strategy.** Gemini has no machine error `code`; the
+* signals are the HTTP `status` and the canonical status phrase Google
+* embeds in `message` (`PERMISSION_DENIED`, `RESOURCE_EXHAUSTED`,
+* `INVALID_ARGUMENT`, …). Dispatch keys on `status`, using the message
+* phrase as the tie-breaker for the two 400 sub-cases
+* (context-length vs generic) and for status-less auth/quota errors.
+*
+* `AIError` instances pass through unchanged so `catch/throw wrap(e)`
+* pipelines never double-wrap.
+*
+* @example
+* try {
+*   return await this.ai.models.generateContent(...);
+* } catch (thrown) {
+*   throw wrapGoogleError(thrown);
+* }
+*/
+function wrapGoogleError(thrown) {
+	if (thrown instanceof AIError) return thrown;
+	const shape = toShape(thrown);
+	const context = buildContext(shape);
+	const message = shape.message ?? (thrown instanceof Error ? thrown.message : String(thrown));
+	if (isTimeout(shape)) return new ProviderTimeoutError(message, {
+		cause: thrown,
+		context
+	});
+	if (shape.status === 401 || shape.status === 403 || /permission_denied|api key not valid|unauthenticated/i.test(message)) return new ProviderAuthError(message, {
+		cause: thrown,
+		context
+	});
+	if (shape.status === 429 || /resource_exhausted|quota/i.test(message)) return new ProviderRateLimitError(message, {
+		cause: thrown,
+		context
+	});
+	if (shape.status === 400) {
+		if (/token count|context length|exceeds the maximum|input is too long/i.test(message)) return new ContextLengthExceededError(message, {
+			cause: thrown,
+			context
+		});
+		return new InvalidRequestError(message, {
+			cause: thrown,
+			context
+		});
+	}
+	if (shape.status === 404 || isClientStatus(shape.status)) return new InvalidRequestError(message, {
+		cause: thrown,
+		context
+	});
+	return new ProviderError(message, {
+		cause: thrown,
+		context
+	});
+}
+/**
+* Read the raw error shape. The Gemini SDK's `ApiError` carries a
+* numeric `status`; flattened/proxied errors may carry it (or `code`)
+* loosely.
+*/
+function toShape(thrown) {
+	if (thrown instanceof ApiError) return {
+		status: thrown.status,
+		message: thrown.message,
+		name: thrown.name
+	};
+	if (typeof thrown === "object" && thrown !== null) {
+		const raw = thrown;
+		return {
+			status: typeof raw.status === "number" ? raw.status : void 0,
+			message: typeof raw.message === "string" ? raw.message : void 0,
+			name: typeof raw.name === "string" ? raw.name : void 0,
+			code: typeof raw.code === "string" ? raw.code : void 0
+		};
+	}
+	return {};
+}
+/**
+* Decide whether the error is a timeout. Gemini maps gateway timeouts
+* to HTTP 504 (`DEADLINE_EXCEEDED`); transport aborts surface as
+* `AbortError` / `ETIMEDOUT` / `ECONNABORTED`.
+*/
+function isTimeout(shape) {
+	if (shape.status === 504) return true;
+	if (shape.name === "AbortError" || /deadline_exceeded/i.test(shape.message ?? "")) return true;
+	return shape.code === "ETIMEDOUT" || shape.code === "ECONNABORTED";
+}
+/** True for HTTP 4xx — a client-side request problem, not a server fault. */
+function isClientStatus(status) {
+	return typeof status === "number" && status >= 400 && status < 500;
+}
+/** Attach the diagnostic fields to `error.context`. */
+function buildContext(shape) {
+	const context = {};
+	if (shape.status !== void 0) context.status = shape.status;
+	if (shape.name) context.code = shape.name;
+	return context;
+}
+
+//#endregion
+export { wrapGoogleError };
+//# sourceMappingURL=wrap-google-error.mjs.map

package/esm/utils/wrap-google-error.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"wrap-google-error.mjs","names":[],"sources":["../../../../../../@warlock.js/ai-google/src/utils/wrap-google-error.ts"],"sourcesContent":["import {\n  AIError,\n  ContextLengthExceededError,\n  InvalidRequestError,\n  ProviderAuthError,\n  ProviderError,\n  ProviderRateLimitError,\n  ProviderTimeoutError,\n} from \"@warlock.js/ai\";\nimport { ApiError } from \"@google/genai\";\n\n/**\n * Raw-error fields the wrapper reads off a Gemini SDK error.\n * `@google/genai`'s `ApiError` exposes `status` (HTTP code) +\n * `message`; transport aborts surface as `AbortError` / `ETIMEDOUT`.\n * We duck-type so proxied / re-thrown errors still classify.\n */\ntype GoogleErrorShape = {\n  status?: number;\n  message?: string;\n  name?: string;\n  code?: string;\n};\n\n/**\n * Wrap any thrown value caught inside the Gemini adapter into the\n * appropriate `@warlock.js/ai` `AIError` subclass.\n *\n * **Dispatch strategy.** Gemini has no machine error `code`; the\n * signals are the HTTP `status` and the canonical status phrase Google\n * embeds in `message` (`PERMISSION_DENIED`, `RESOURCE_EXHAUSTED`,\n * `INVALID_ARGUMENT`, …). Dispatch keys on `status`, using the message\n * phrase as the tie-breaker for the two 400 sub-cases\n * (context-length vs generic) and for status-less auth/quota errors.\n *\n * `AIError` instances pass through unchanged so `catch/throw wrap(e)`\n * pipelines never double-wrap.\n *\n * @example\n * try {\n *   return await this.ai.models.generateContent(...);\n * } catch (thrown) {\n *   throw wrapGoogleError(thrown);\n * }\n */\nexport function wrapGoogleError(thrown: unknown): AIError {\n  if (thrown instanceof AIError) {\n    return thrown;\n  }\n\n  const shape = toShape(thrown);\n  const context = buildContext(shape);\n  const message = shape.message ?? (thrown instanceof Error ? thrown.message : String(thrown));\n\n  if (isTimeout(shape)) {\n    return new ProviderTimeoutError(message, { cause: thrown, context });\n  }\n\n  if (\n    shape.status === 401 ||\n    shape.status === 403 ||\n    /permission_denied|api key not valid|unauthenticated/i.test(message)\n  ) {\n    return new ProviderAuthError(message, { cause: thrown, context });\n  }\n\n  if (shape.status === 429 || /resource_exhausted|quota/i.test(message)) {\n    return new ProviderRateLimitError(message, { cause: thrown, context });\n  }\n\n  if (shape.status === 400) {\n    if (/token count|context length|exceeds the maximum|input is too long/i.test(message)) {\n      return new ContextLengthExceededError(message, { cause: thrown, context });\n    }\n\n    return new InvalidRequestError(message, { cause: thrown, context });\n  }\n\n  if (shape.status === 404 || isClientStatus(shape.status)) {\n    return new InvalidRequestError(message, { cause: thrown, context });\n  }\n\n  return new ProviderError(message, { cause: thrown, context });\n}\n\n/**\n * Read the raw error shape. The Gemini SDK's `ApiError` carries a\n * numeric `status`; flattened/proxied errors may carry it (or `code`)\n * loosely.\n */\nfunction toShape(thrown: unknown): GoogleErrorShape {\n  if (thrown instanceof ApiError) {\n    return { status: thrown.status, message: thrown.message, name: thrown.name };\n  }\n\n  if (typeof thrown === \"object\" && thrown !== null) {\n    const raw = thrown as Record<string, unknown>;\n\n    return {\n      status: typeof raw.status === \"number\" ? raw.status : undefined,\n      message: typeof raw.message === \"string\" ? raw.message : undefined,\n      name: typeof raw.name === \"string\" ? raw.name : undefined,\n      code: typeof raw.code === \"string\" ? raw.code : undefined,\n    };\n  }\n\n  return {};\n}\n\n/**\n * Decide whether the error is a timeout. Gemini maps gateway timeouts\n * to HTTP 504 (`DEADLINE_EXCEEDED`); transport aborts surface as\n * `AbortError` / `ETIMEDOUT` / `ECONNABORTED`.\n */\nfunction isTimeout(shape: GoogleErrorShape): boolean {\n  if (shape.status === 504) {\n    return true;\n  }\n\n  if (shape.name === \"AbortError\" || /deadline_exceeded/i.test(shape.message ?? \"\")) {\n    return true;\n  }\n\n  return shape.code === \"ETIMEDOUT\" || shape.code === \"ECONNABORTED\";\n}\n\n/** True for HTTP 4xx — a client-side request problem, not a server fault. */\nfunction isClientStatus(status: number | undefined): boolean {\n  return typeof status === \"number\" && status >= 400 && status < 500;\n}\n\n/** Attach the diagnostic fields to `error.context`. */\nfunction buildContext(shape: GoogleErrorShape): Record<string, unknown> {\n  const context: Record<string, unknown> = {};\n\n  if (shape.status !== undefined) {\n    context.status = shape.status;\n  }\n\n  if (shape.name) {\n    context.code = shape.name;\n  }\n\n  return context;\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;AA6CA,SAAgB,gBAAgB,QAA0B;CACxD,IAAI,kBAAkB,SACpB,OAAO;CAGT,MAAM,QAAQ,QAAQ,MAAM;CAC5B,MAAM,UAAU,aAAa,KAAK;CAClC,MAAM,UAAU,MAAM,YAAY,kBAAkB,QAAQ,OAAO,UAAU,OAAO,MAAM;CAE1F,IAAI,UAAU,KAAK,GACjB,OAAO,IAAI,qBAAqB,SAAS;EAAE,OAAO;EAAQ;CAAQ,CAAC;CAGrE,IACE,MAAM,WAAW,OACjB,MAAM,WAAW,OACjB,uDAAuD,KAAK,OAAO,GAEnE,OAAO,IAAI,kBAAkB,SAAS;EAAE,OAAO;EAAQ;CAAQ,CAAC;CAGlE,IAAI,MAAM,WAAW,OAAO,4BAA4B,KAAK,OAAO,GAClE,OAAO,IAAI,uBAAuB,SAAS;EAAE,OAAO;EAAQ;CAAQ,CAAC;CAGvE,IAAI,MAAM,WAAW,KAAK;EACxB,IAAI,oEAAoE,KAAK,OAAO,GAClF,OAAO,IAAI,2BAA2B,SAAS;GAAE,OAAO;GAAQ;EAAQ,CAAC;EAG3E,OAAO,IAAI,oBAAoB,SAAS;GAAE,OAAO;GAAQ;EAAQ,CAAC;CACpE;CAEA,IAAI,MAAM,WAAW,OAAO,eAAe,MAAM,MAAM,GACrD,OAAO,IAAI,oBAAoB,SAAS;EAAE,OAAO;EAAQ;CAAQ,CAAC;CAGpE,OAAO,IAAI,cAAc,SAAS;EAAE,OAAO;EAAQ;CAAQ,CAAC;AAC9D;;;;;;AAOA,SAAS,QAAQ,QAAmC;CAClD,IAAI,kBAAkB,UACpB,OAAO;EAAE,QAAQ,OAAO;EAAQ,SAAS,OAAO;EAAS,MAAM,OAAO;CAAK;CAG7E,IAAI,OAAO,WAAW,YAAY,WAAW,MAAM;EACjD,MAAM,MAAM;EAEZ,OAAO;GACL,QAAQ,OAAO,IAAI,WAAW,WAAW,IAAI,SAAS;GACtD,SAAS,OAAO,IAAI,YAAY,WAAW,IAAI,UAAU;GACzD,MAAM,OAAO,IAAI,SAAS,WAAW,IAAI,OAAO;GAChD,MAAM,OAAO,IAAI,SAAS,WAAW,IAAI,OAAO;EAClD;CACF;CAEA,OAAO,CAAC;AACV;;;;;;AAOA,SAAS,UAAU,OAAkC;CACnD,IAAI,MAAM,WAAW,KACnB,OAAO;CAGT,IAAI,MAAM,SAAS,gBAAgB,qBAAqB,KAAK,MAAM,WAAW,EAAE,GAC9E,OAAO;CAGT,OAAO,MAAM,SAAS,eAAe,MAAM,SAAS;AACtD;;AAGA,SAAS,eAAe,QAAqC;CAC3D,OAAO,OAAO,WAAW,YAAY,UAAU,OAAO,SAAS;AACjE;;AAGA,SAAS,aAAa,OAAkD;CACtE,MAAM,UAAmC,CAAC;CAE1C,IAAI,MAAM,WAAW,QACnB,QAAQ,SAAS,MAAM;CAGzB,IAAI,MAAM,MACR,QAAQ,OAAO,MAAM;CAGvB,OAAO;AACT"}