@tepa/provider-gemini 0.1.0

package/README.md

@@ -0,0 +1,59 @@
+# @tepa/provider-gemini
+
+Google Gemini LLM provider for the Tepa agent pipeline. Uses the [@google/genai](https://www.npmjs.com/package/@google/genai) SDK.
+
+## Install
+
+```bash
+npm install @tepa/provider-gemini
+```
+
+## Setup
+
+Set your Gemini API key as an environment variable:
+
+```bash
+export GEMINI_API_KEY=...
+```
+
+Or use `GOOGLE_API_KEY`, or pass it directly in the constructor.
+
+## Usage
+
+```typescript
+import { Tepa } from "tepa";
+import { GeminiProvider } from "@tepa/provider-gemini";
+
+const tepa = new Tepa({
+  tools: [/* ... */],
+  provider: new GeminiProvider(),
+});
+```
+
+### Provider Options
+
+```typescript
+const provider = new GeminiProvider({
+  apiKey: "...",            // Defaults to GEMINI_API_KEY or GOOGLE_API_KEY env var
+  maxRetries: 3,            // Default: 3
+  retryBaseDelayMs: 1000,   // Base delay for exponential backoff
+});
+```
+
+## Logging
+
+Every LLM call is automatically logged to a JSONL file in `.tepa/logs/`. You can disable the default file logger, add custom log listeners, or send logs to external services like Prometheus, NewRelic, or Datadog using the `onLog()` method:
+
+```typescript
+const provider = new GeminiProvider({ defaultLog: false });
+
+provider.onLog((entry) => {
+  externalLogger.send(entry);
+});
+```
+
+See [`@tepa/provider-core`](../provider-core) for full logging documentation.
+
+## Native Tool Use
+
+This provider supports native tool calling via Gemini's function calling API. When tool schemas are passed via `options.tools`, they are forwarded as function declarations. The LLM returns structured `functionCall` parts with pre-parsed parameters, eliminating text-based JSON parsing errors.

package/dist/index.cjs

@@ -0,0 +1,173 @@
+"use strict";
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+
+// src/index.ts
+var index_exports = {};
+__export(index_exports, {
+  GeminiProvider: () => GeminiProvider,
+  extractText: () => extractText,
+  extractToolUse: () => extractToolUse,
+  toFinishReason: () => toFinishReason,
+  toGeminiContents: () => toGeminiContents,
+  toGeminiTools: () => toGeminiTools
+});
+module.exports = __toCommonJS(index_exports);
+
+// src/gemini.ts
+var import_genai = require("@google/genai");
+var import_provider_core = require("@tepa/provider-core");
+
+// src/formatting.ts
+function toGeminiContents(messages) {
+  return messages.map((msg) => ({
+    role: msg.role === "assistant" ? "model" : "user",
+    parts: [{ text: msg.content }]
+  }));
+}
+function toFinishReason(reason) {
+  switch (reason) {
+    case "MAX_TOKENS":
+      return "max_tokens";
+    case "STOP":
+    default:
+      return "end_turn";
+  }
+}
+function extractText(response) {
+  return response.text ?? "";
+}
+function extractToolUse(response) {
+  const parts = response.candidates?.[0]?.content?.parts ?? [];
+  const blocks = [];
+  let index = 0;
+  for (const part of parts) {
+    if (part.functionCall?.name) {
+      blocks.push({
+        id: `gemini-call-${index++}`,
+        name: part.functionCall.name,
+        input: part.functionCall.args ?? {}
+      });
+    }
+  }
+  return blocks;
+}
+function toGeminiTools(tools) {
+  const functionDeclarations = tools.map((tool) => {
+    const properties = {};
+    const required = [];
+    for (const [name, param] of Object.entries(tool.parameters)) {
+      properties[name] = {
+        type: param.type.toUpperCase(),
+        description: param.description
+      };
+      if (param.required !== false) {
+        required.push(name);
+      }
+    }
+    return {
+      name: tool.name,
+      description: tool.description,
+      parameters: {
+        type: "OBJECT",
+        properties,
+        required
+      }
+    };
+  });
+  return [{ functionDeclarations }];
+}
+
+// src/gemini.ts
+var DEFAULT_MODEL = "gemini-3-flash-preview";
+var DEFAULT_MAX_TOKENS = 64e3;
+var GeminiProvider = class extends import_provider_core.BaseLLMProvider {
+  providerName = "gemini";
+  client;
+  constructor(options = {}) {
+    super(options);
+    const apiKey = options.apiKey ?? process.env.GEMINI_API_KEY ?? process.env.GOOGLE_API_KEY;
+    this.client = new import_genai.GoogleGenAI({ apiKey });
+  }
+  async doComplete(messages, options) {
+    const contents = toGeminiContents(messages);
+    const config = {
+      maxOutputTokens: options.maxTokens ?? DEFAULT_MAX_TOKENS
+    };
+    if (options.temperature !== void 0) {
+      config.temperature = options.temperature;
+    }
+    if (options.systemPrompt) {
+      config.systemInstruction = options.systemPrompt;
+    }
+    const params = {
+      model: options.model || DEFAULT_MODEL,
+      contents,
+      config
+    };
+    if (options.tools && options.tools.length > 0) {
+      params.tools = toGeminiTools(options.tools);
+    }
+    const response = await this.client.models.generateContent(params);
+    const candidates = response.candidates ?? [];
+    const finishReason = candidates[0]?.finishReason ?? null;
+    const usage = response.usageMetadata ?? {};
+    const toolUse = extractToolUse(response);
+    const hasToolUse = toolUse.length > 0;
+    return {
+      text: extractText(response),
+      tokensUsed: {
+        input: usage.promptTokenCount ?? 0,
+        output: usage.candidatesTokenCount ?? 0
+      },
+      finishReason: hasToolUse ? "tool_use" : toFinishReason(finishReason),
+      ...hasToolUse && { toolUse }
+    };
+  }
+  isRetryable(error) {
+    if (error instanceof TypeError) {
+      return true;
+    }
+    if (error instanceof import_genai.ApiError) {
+      const status = error.status;
+      if (status === 400 || status === 401 || status === 403 || status === 404) {
+        return false;
+      }
+      if (status === 429 || status !== void 0 && status >= 500) {
+        return true;
+      }
+    }
+    return false;
+  }
+  getRetryAfterMs(_error) {
+    return null;
+  }
+  isRateLimitError(error) {
+    return error instanceof import_genai.ApiError && error.status === 429;
+  }
+};
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  GeminiProvider,
+  extractText,
+  extractToolUse,
+  toFinishReason,
+  toGeminiContents,
+  toGeminiTools
+});
+//# sourceMappingURL=index.cjs.map

package/dist/index.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/index.ts","../src/gemini.ts","../src/formatting.ts"],"sourcesContent":["export { GeminiProvider, type GeminiProviderOptions } from \"./gemini.js\";\nexport { toGeminiContents, toGeminiTools, toFinishReason, extractText, extractToolUse } from \"./formatting.js\";\n","import { GoogleGenAI, ApiError } from \"@google/genai\";\nimport type { LLMMessage, LLMRequestOptions, LLMResponse } from \"@tepa/types\";\nimport { BaseLLMProvider, type BaseLLMProviderOptions } from \"@tepa/provider-core\";\nimport { toGeminiContents, toGeminiTools, toFinishReason, extractText, extractToolUse } from \"./formatting.js\";\n\nconst DEFAULT_MODEL = \"gemini-3-flash-preview\";\nconst DEFAULT_MAX_TOKENS = 64_000;\n\nexport interface GeminiProviderOptions extends BaseLLMProviderOptions {\n  /** Gemini API key. Falls back to GEMINI_API_KEY or GOOGLE_API_KEY env variables. */\n  apiKey?: string;\n}\n\n/** LLM provider implementation for Google Gemini models. */\nexport class GeminiProvider extends BaseLLMProvider {\n  protected readonly providerName = \"gemini\";\n  private readonly client: GoogleGenAI;\n\n  constructor(options: GeminiProviderOptions = {}) {\n    super(options);\n    const apiKey = options.apiKey ?? process.env.GEMINI_API_KEY ?? process.env.GOOGLE_API_KEY;\n    this.client = new GoogleGenAI({ apiKey });\n  }\n\n  protected async doComplete(\n    messages: LLMMessage[],\n    options: LLMRequestOptions,\n  ): Promise<LLMResponse> {\n    const contents = toGeminiContents(messages);\n\n    const config: Record<string, unknown> = {\n      maxOutputTokens: options.maxTokens ?? DEFAULT_MAX_TOKENS,\n    };\n\n    if (options.temperature !== undefined) {\n      config.temperature = options.temperature;\n    }\n\n    if (options.systemPrompt) {\n      config.systemInstruction = options.systemPrompt;\n    }\n\n    const params: Record<string, unknown> = {\n      model: options.model || DEFAULT_MODEL,\n      contents,\n      config,\n    };\n\n    if (options.tools && options.tools.length > 0) {\n      params.tools = toGeminiTools(options.tools);\n    }\n\n    const response = await this.client.models.generateContent(params as any);\n\n    const candidates = response.candidates ?? [];\n    const finishReason = candidates[0]?.finishReason ?? null;\n    const usage = response.usageMetadata ?? {};\n\n    const toolUse = extractToolUse(response);\n    const hasToolUse = toolUse.length > 0;\n\n    return {\n      text: extractText(response),\n      tokensUsed: {\n        input: usage.promptTokenCount ?? 0,\n        output: usage.candidatesTokenCount ?? 0,\n      },\n      finishReason: hasToolUse ? \"tool_use\" : toFinishReason(finishReason),\n      ...(hasToolUse && { toolUse }),\n    };\n  }\n\n  protected isRetryable(error: unknown): boolean {\n    if (error instanceof TypeError) {\n      return true;\n    }\n    if (error instanceof ApiError) {\n      const status = error.status;\n      if (status === 400 || status === 401 || status === 403 || status === 404) {\n        return false;\n      }\n      if (status === 429 || (status !== undefined && status >= 500)) {\n        return true;\n      }\n    }\n    return false;\n  }\n\n  protected getRetryAfterMs(_error: unknown): number | null {\n    return null;\n  }\n\n  protected isRateLimitError(error: unknown): boolean {\n    return error instanceof ApiError && error.status === 429;\n  }\n}\n","import type { LLMMessage, LLMToolUseBlock, ToolSchema } from \"@tepa/types\";\n\nexport interface GeminiContent {\n  role: \"user\" | \"model\";\n  parts: { text: string }[];\n}\n\n/**\n * Convert Tepa LLMMessage array to Gemini contents format.\n * Maps \"assistant\" role to \"model\".\n */\nexport function toGeminiContents(messages: LLMMessage[]): GeminiContent[] {\n  return messages.map((msg) => ({\n    role: msg.role === \"assistant\" ? \"model\" : \"user\",\n    parts: [{ text: msg.content }],\n  }));\n}\n\n/**\n * Map Gemini finish reason to Tepa finishReason.\n */\nexport function toFinishReason(\n  reason: string | null | undefined,\n): \"end_turn\" | \"max_tokens\" | \"stop_sequence\" | \"tool_use\" {\n  switch (reason) {\n    case \"MAX_TOKENS\":\n      return \"max_tokens\";\n    case \"STOP\":\n    default:\n      return \"end_turn\";\n  }\n}\n\n/**\n * Extract text from a Gemini response.\n */\nexport function extractText(response: { text?: string }): string {\n  return response.text ?? \"\";\n}\n\n/**\n * Extract tool use blocks from a Gemini response.\n */\nexport function extractToolUse(response: any): LLMToolUseBlock[] {\n  const parts = response.candidates?.[0]?.content?.parts ?? [];\n  const blocks: LLMToolUseBlock[] = [];\n  let index = 0;\n\n  for (const part of parts) {\n    if (part.functionCall?.name) {\n      blocks.push({\n        id: `gemini-call-${index++}`,\n        name: part.functionCall.name,\n        input: part.functionCall.args ?? {},\n      });\n    }\n  }\n\n  return blocks;\n}\n\n/**\n * Convert Tepa ToolSchema to Gemini function declarations.\n */\nexport function toGeminiTools(tools: ToolSchema[]): Record<string, unknown>[] {\n  const functionDeclarations = tools.map((tool) => {\n    const properties: Record<string, unknown> = {};\n    const required: string[] = [];\n\n    for (const [name, param] of Object.entries(tool.parameters)) {\n      properties[name] = {\n        type: param.type.toUpperCase(),\n        description: param.description,\n      };\n      if (param.required !== false) {\n        required.push(name);\n      }\n    }\n\n    return {\n      name: tool.name,\n      description: tool.description,\n      parameters: {\n        type: \"OBJECT\",\n        properties,\n        required,\n      },\n    };\n  });\n\n  return [{ functionDeclarations }];\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACAA,mBAAsC;AAEtC,2BAA6D;;;ACStD,SAAS,iBAAiB,UAAyC;AACxE,SAAO,SAAS,IAAI,CAAC,SAAS;AAAA,IAC5B,MAAM,IAAI,SAAS,cAAc,UAAU;AAAA,IAC3C,OAAO,CAAC,EAAE,MAAM,IAAI,QAAQ,CAAC;AAAA,EAC/B,EAAE;AACJ;AAKO,SAAS,eACd,QAC0D;AAC1D,UAAQ,QAAQ;AAAA,IACd,KAAK;AACH,aAAO;AAAA,IACT,KAAK;AAAA,IACL;AACE,aAAO;AAAA,EACX;AACF;AAKO,SAAS,YAAY,UAAqC;AAC/D,SAAO,SAAS,QAAQ;AAC1B;AAKO,SAAS,eAAe,UAAkC;AAC/D,QAAM,QAAQ,SAAS,aAAa,CAAC,GAAG,SAAS,SAAS,CAAC;AAC3D,QAAM,SAA4B,CAAC;AACnC,MAAI,QAAQ;AAEZ,aAAW,QAAQ,OAAO;AACxB,QAAI,KAAK,cAAc,MAAM;AAC3B,aAAO,KAAK;AAAA,QACV,IAAI,eAAe,OAAO;AAAA,QAC1B,MAAM,KAAK,aAAa;AAAA,QACxB,OAAO,KAAK,aAAa,QAAQ,CAAC;AAAA,MACpC,CAAC;AAAA,IACH;AAAA,EACF;AAEA,SAAO;AACT;AAKO,SAAS,cAAc,OAAgD;AAC5E,QAAM,uBAAuB,MAAM,IAAI,CAAC,SAAS;AAC/C,UAAM,aAAsC,CAAC;AAC7C,UAAM,WAAqB,CAAC;AAE5B,eAAW,CAAC,MAAM,KAAK,KAAK,OAAO,QAAQ,KAAK,UAAU,GAAG;AAC3D,iBAAW,IAAI,IAAI;AAAA,QACjB,MAAM,MAAM,KAAK,YAAY;AAAA,QAC7B,aAAa,MAAM;AAAA,MACrB;AACA,UAAI,MAAM,aAAa,OAAO;AAC5B,iBAAS,KAAK,IAAI;AAAA,MACpB;AAAA,IACF;AAEA,WAAO;AAAA,MACL,MAAM,KAAK;AAAA,MACX,aAAa,KAAK;AAAA,MAClB,YAAY;AAAA,QACV,MAAM;AAAA,QACN;AAAA,QACA;AAAA,MACF;AAAA,IACF;AAAA,EACF,CAAC;AAED,SAAO,CAAC,EAAE,qBAAqB,CAAC;AAClC;;;ADtFA,IAAM,gBAAgB;AACtB,IAAM,qBAAqB;AAQpB,IAAM,iBAAN,cAA6B,qCAAgB;AAAA,EAC/B,eAAe;AAAA,EACjB;AAAA,EAEjB,YAAY,UAAiC,CAAC,GAAG;AAC/C,UAAM,OAAO;AACb,UAAM,SAAS,QAAQ,UAAU,QAAQ,IAAI,kBAAkB,QAAQ,IAAI;AAC3E,SAAK,SAAS,IAAI,yBAAY,EAAE,OAAO,CAAC;AAAA,EAC1C;AAAA,EAEA,MAAgB,WACd,UACA,SACsB;AACtB,UAAM,WAAW,iBAAiB,QAAQ;AAE1C,UAAM,SAAkC;AAAA,MACtC,iBAAiB,QAAQ,aAAa;AAAA,IACxC;AAEA,QAAI,QAAQ,gBAAgB,QAAW;AACrC,aAAO,cAAc,QAAQ;AAAA,IAC/B;AAEA,QAAI,QAAQ,cAAc;AACxB,aAAO,oBAAoB,QAAQ;AAAA,IACrC;AAEA,UAAM,SAAkC;AAAA,MACtC,OAAO,QAAQ,SAAS;AAAA,MACxB;AAAA,MACA;AAAA,IACF;AAEA,QAAI,QAAQ,SAAS,QAAQ,MAAM,SAAS,GAAG;AAC7C,aAAO,QAAQ,cAAc,QAAQ,KAAK;AAAA,IAC5C;AAEA,UAAM,WAAW,MAAM,KAAK,OAAO,OAAO,gBAAgB,MAAa;AAEvE,UAAM,aAAa,SAAS,cAAc,CAAC;AAC3C,UAAM,eAAe,WAAW,CAAC,GAAG,gBAAgB;AACpD,UAAM,QAAQ,SAAS,iBAAiB,CAAC;AAEzC,UAAM,UAAU,eAAe,QAAQ;AACvC,UAAM,aAAa,QAAQ,SAAS;AAEpC,WAAO;AAAA,MACL,MAAM,YAAY,QAAQ;AAAA,MAC1B,YAAY;AAAA,QACV,OAAO,MAAM,oBAAoB;AAAA,QACjC,QAAQ,MAAM,wBAAwB;AAAA,MACxC;AAAA,MACA,cAAc,aAAa,aAAa,eAAe,YAAY;AAAA,MACnE,GAAI,cAAc,EAAE,QAAQ;AAAA,IAC9B;AAAA,EACF;AAAA,EAEU,YAAY,OAAyB;AAC7C,QAAI,iBAAiB,WAAW;AAC9B,aAAO;AAAA,IACT;AACA,QAAI,iBAAiB,uBAAU;AAC7B,YAAM,SAAS,MAAM;AACrB,UAAI,WAAW,OAAO,WAAW,OAAO,WAAW,OAAO,WAAW,KAAK;AACxE,eAAO;AAAA,MACT;AACA,UAAI,WAAW,OAAQ,WAAW,UAAa,UAAU,KAAM;AAC7D,eAAO;AAAA,MACT;AAAA,IACF;AACA,WAAO;AAAA,EACT;AAAA,EAEU,gBAAgB,QAAgC;AACxD,WAAO;AAAA,EACT;AAAA,EAEU,iBAAiB,OAAyB;AAClD,WAAO,iBAAiB,yBAAY,MAAM,WAAW;AAAA,EACvD;AACF;","names":[]}

package/dist/index.d.cts

@@ -0,0 +1,49 @@
+import { LLMMessage, LLMRequestOptions, LLMResponse, LLMToolUseBlock, ToolSchema } from '@tepa/types';
+import { BaseLLMProvider, BaseLLMProviderOptions } from '@tepa/provider-core';
+
+interface GeminiProviderOptions extends BaseLLMProviderOptions {
+    /** Gemini API key. Falls back to GEMINI_API_KEY or GOOGLE_API_KEY env variables. */
+    apiKey?: string;
+}
+/** LLM provider implementation for Google Gemini models. */
+declare class GeminiProvider extends BaseLLMProvider {
+    protected readonly providerName = "gemini";
+    private readonly client;
+    constructor(options?: GeminiProviderOptions);
+    protected doComplete(messages: LLMMessage[], options: LLMRequestOptions): Promise<LLMResponse>;
+    protected isRetryable(error: unknown): boolean;
+    protected getRetryAfterMs(_error: unknown): number | null;
+    protected isRateLimitError(error: unknown): boolean;
+}
+
+interface GeminiContent {
+    role: "user" | "model";
+    parts: {
+        text: string;
+    }[];
+}
+/**
+ * Convert Tepa LLMMessage array to Gemini contents format.
+ * Maps "assistant" role to "model".
+ */
+declare function toGeminiContents(messages: LLMMessage[]): GeminiContent[];
+/**
+ * Map Gemini finish reason to Tepa finishReason.
+ */
+declare function toFinishReason(reason: string | null | undefined): "end_turn" | "max_tokens" | "stop_sequence" | "tool_use";
+/**
+ * Extract text from a Gemini response.
+ */
+declare function extractText(response: {
+    text?: string;
+}): string;
+/**
+ * Extract tool use blocks from a Gemini response.
+ */
+declare function extractToolUse(response: any): LLMToolUseBlock[];
+/**
+ * Convert Tepa ToolSchema to Gemini function declarations.
+ */
+declare function toGeminiTools(tools: ToolSchema[]): Record<string, unknown>[];
+
+export { GeminiProvider, type GeminiProviderOptions, extractText, extractToolUse, toFinishReason, toGeminiContents, toGeminiTools };

package/dist/index.d.ts

@@ -0,0 +1,49 @@
+import { LLMMessage, LLMRequestOptions, LLMResponse, LLMToolUseBlock, ToolSchema } from '@tepa/types';
+import { BaseLLMProvider, BaseLLMProviderOptions } from '@tepa/provider-core';
+
+interface GeminiProviderOptions extends BaseLLMProviderOptions {
+    /** Gemini API key. Falls back to GEMINI_API_KEY or GOOGLE_API_KEY env variables. */
+    apiKey?: string;
+}
+/** LLM provider implementation for Google Gemini models. */
+declare class GeminiProvider extends BaseLLMProvider {
+    protected readonly providerName = "gemini";
+    private readonly client;
+    constructor(options?: GeminiProviderOptions);
+    protected doComplete(messages: LLMMessage[], options: LLMRequestOptions): Promise<LLMResponse>;
+    protected isRetryable(error: unknown): boolean;
+    protected getRetryAfterMs(_error: unknown): number | null;
+    protected isRateLimitError(error: unknown): boolean;
+}
+
+interface GeminiContent {
+    role: "user" | "model";
+    parts: {
+        text: string;
+    }[];
+}
+/**
+ * Convert Tepa LLMMessage array to Gemini contents format.
+ * Maps "assistant" role to "model".
+ */
+declare function toGeminiContents(messages: LLMMessage[]): GeminiContent[];
+/**
+ * Map Gemini finish reason to Tepa finishReason.
+ */
+declare function toFinishReason(reason: string | null | undefined): "end_turn" | "max_tokens" | "stop_sequence" | "tool_use";
+/**
+ * Extract text from a Gemini response.
+ */
+declare function extractText(response: {
+    text?: string;
+}): string;
+/**
+ * Extract tool use blocks from a Gemini response.
+ */
+declare function extractToolUse(response: any): LLMToolUseBlock[];
+/**
+ * Convert Tepa ToolSchema to Gemini function declarations.
+ */
+declare function toGeminiTools(tools: ToolSchema[]): Record<string, unknown>[];
+
+export { GeminiProvider, type GeminiProviderOptions, extractText, extractToolUse, toFinishReason, toGeminiContents, toGeminiTools };

package/dist/index.js

@@ -0,0 +1,141 @@
+// src/gemini.ts
+import { GoogleGenAI, ApiError } from "@google/genai";
+import { BaseLLMProvider } from "@tepa/provider-core";
+
+// src/formatting.ts
+function toGeminiContents(messages) {
+  return messages.map((msg) => ({
+    role: msg.role === "assistant" ? "model" : "user",
+    parts: [{ text: msg.content }]
+  }));
+}
+function toFinishReason(reason) {
+  switch (reason) {
+    case "MAX_TOKENS":
+      return "max_tokens";
+    case "STOP":
+    default:
+      return "end_turn";
+  }
+}
+function extractText(response) {
+  return response.text ?? "";
+}
+function extractToolUse(response) {
+  const parts = response.candidates?.[0]?.content?.parts ?? [];
+  const blocks = [];
+  let index = 0;
+  for (const part of parts) {
+    if (part.functionCall?.name) {
+      blocks.push({
+        id: `gemini-call-${index++}`,
+        name: part.functionCall.name,
+        input: part.functionCall.args ?? {}
+      });
+    }
+  }
+  return blocks;
+}
+function toGeminiTools(tools) {
+  const functionDeclarations = tools.map((tool) => {
+    const properties = {};
+    const required = [];
+    for (const [name, param] of Object.entries(tool.parameters)) {
+      properties[name] = {
+        type: param.type.toUpperCase(),
+        description: param.description
+      };
+      if (param.required !== false) {
+        required.push(name);
+      }
+    }
+    return {
+      name: tool.name,
+      description: tool.description,
+      parameters: {
+        type: "OBJECT",
+        properties,
+        required
+      }
+    };
+  });
+  return [{ functionDeclarations }];
+}
+
+// src/gemini.ts
+var DEFAULT_MODEL = "gemini-3-flash-preview";
+var DEFAULT_MAX_TOKENS = 64e3;
+var GeminiProvider = class extends BaseLLMProvider {
+  providerName = "gemini";
+  client;
+  constructor(options = {}) {
+    super(options);
+    const apiKey = options.apiKey ?? process.env.GEMINI_API_KEY ?? process.env.GOOGLE_API_KEY;
+    this.client = new GoogleGenAI({ apiKey });
+  }
+  async doComplete(messages, options) {
+    const contents = toGeminiContents(messages);
+    const config = {
+      maxOutputTokens: options.maxTokens ?? DEFAULT_MAX_TOKENS
+    };
+    if (options.temperature !== void 0) {
+      config.temperature = options.temperature;
+    }
+    if (options.systemPrompt) {
+      config.systemInstruction = options.systemPrompt;
+    }
+    const params = {
+      model: options.model || DEFAULT_MODEL,
+      contents,
+      config
+    };
+    if (options.tools && options.tools.length > 0) {
+      params.tools = toGeminiTools(options.tools);
+    }
+    const response = await this.client.models.generateContent(params);
+    const candidates = response.candidates ?? [];
+    const finishReason = candidates[0]?.finishReason ?? null;
+    const usage = response.usageMetadata ?? {};
+    const toolUse = extractToolUse(response);
+    const hasToolUse = toolUse.length > 0;
+    return {
+      text: extractText(response),
+      tokensUsed: {
+        input: usage.promptTokenCount ?? 0,
+        output: usage.candidatesTokenCount ?? 0
+      },
+      finishReason: hasToolUse ? "tool_use" : toFinishReason(finishReason),
+      ...hasToolUse && { toolUse }
+    };
+  }
+  isRetryable(error) {
+    if (error instanceof TypeError) {
+      return true;
+    }
+    if (error instanceof ApiError) {
+      const status = error.status;
+      if (status === 400 || status === 401 || status === 403 || status === 404) {
+        return false;
+      }
+      if (status === 429 || status !== void 0 && status >= 500) {
+        return true;
+      }
+    }
+    return false;
+  }
+  getRetryAfterMs(_error) {
+    return null;
+  }
+  isRateLimitError(error) {
+    return error instanceof ApiError && error.status === 429;
+  }
+};
+export {
+  GeminiProvider,
+  extractText,
+  extractToolUse,
+  toFinishReason,
+  toGeminiContents,
+  toGeminiTools
+};
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/gemini.ts","../src/formatting.ts"],"sourcesContent":["import { GoogleGenAI, ApiError } from \"@google/genai\";\nimport type { LLMMessage, LLMRequestOptions, LLMResponse } from \"@tepa/types\";\nimport { BaseLLMProvider, type BaseLLMProviderOptions } from \"@tepa/provider-core\";\nimport { toGeminiContents, toGeminiTools, toFinishReason, extractText, extractToolUse } from \"./formatting.js\";\n\nconst DEFAULT_MODEL = \"gemini-3-flash-preview\";\nconst DEFAULT_MAX_TOKENS = 64_000;\n\nexport interface GeminiProviderOptions extends BaseLLMProviderOptions {\n  /** Gemini API key. Falls back to GEMINI_API_KEY or GOOGLE_API_KEY env variables. */\n  apiKey?: string;\n}\n\n/** LLM provider implementation for Google Gemini models. */\nexport class GeminiProvider extends BaseLLMProvider {\n  protected readonly providerName = \"gemini\";\n  private readonly client: GoogleGenAI;\n\n  constructor(options: GeminiProviderOptions = {}) {\n    super(options);\n    const apiKey = options.apiKey ?? process.env.GEMINI_API_KEY ?? process.env.GOOGLE_API_KEY;\n    this.client = new GoogleGenAI({ apiKey });\n  }\n\n  protected async doComplete(\n    messages: LLMMessage[],\n    options: LLMRequestOptions,\n  ): Promise<LLMResponse> {\n    const contents = toGeminiContents(messages);\n\n    const config: Record<string, unknown> = {\n      maxOutputTokens: options.maxTokens ?? DEFAULT_MAX_TOKENS,\n    };\n\n    if (options.temperature !== undefined) {\n      config.temperature = options.temperature;\n    }\n\n    if (options.systemPrompt) {\n      config.systemInstruction = options.systemPrompt;\n    }\n\n    const params: Record<string, unknown> = {\n      model: options.model || DEFAULT_MODEL,\n      contents,\n      config,\n    };\n\n    if (options.tools && options.tools.length > 0) {\n      params.tools = toGeminiTools(options.tools);\n    }\n\n    const response = await this.client.models.generateContent(params as any);\n\n    const candidates = response.candidates ?? [];\n    const finishReason = candidates[0]?.finishReason ?? null;\n    const usage = response.usageMetadata ?? {};\n\n    const toolUse = extractToolUse(response);\n    const hasToolUse = toolUse.length > 0;\n\n    return {\n      text: extractText(response),\n      tokensUsed: {\n        input: usage.promptTokenCount ?? 0,\n        output: usage.candidatesTokenCount ?? 0,\n      },\n      finishReason: hasToolUse ? \"tool_use\" : toFinishReason(finishReason),\n      ...(hasToolUse && { toolUse }),\n    };\n  }\n\n  protected isRetryable(error: unknown): boolean {\n    if (error instanceof TypeError) {\n      return true;\n    }\n    if (error instanceof ApiError) {\n      const status = error.status;\n      if (status === 400 || status === 401 || status === 403 || status === 404) {\n        return false;\n      }\n      if (status === 429 || (status !== undefined && status >= 500)) {\n        return true;\n      }\n    }\n    return false;\n  }\n\n  protected getRetryAfterMs(_error: unknown): number | null {\n    return null;\n  }\n\n  protected isRateLimitError(error: unknown): boolean {\n    return error instanceof ApiError && error.status === 429;\n  }\n}\n","import type { LLMMessage, LLMToolUseBlock, ToolSchema } from \"@tepa/types\";\n\nexport interface GeminiContent {\n  role: \"user\" | \"model\";\n  parts: { text: string }[];\n}\n\n/**\n * Convert Tepa LLMMessage array to Gemini contents format.\n * Maps \"assistant\" role to \"model\".\n */\nexport function toGeminiContents(messages: LLMMessage[]): GeminiContent[] {\n  return messages.map((msg) => ({\n    role: msg.role === \"assistant\" ? \"model\" : \"user\",\n    parts: [{ text: msg.content }],\n  }));\n}\n\n/**\n * Map Gemini finish reason to Tepa finishReason.\n */\nexport function toFinishReason(\n  reason: string | null | undefined,\n): \"end_turn\" | \"max_tokens\" | \"stop_sequence\" | \"tool_use\" {\n  switch (reason) {\n    case \"MAX_TOKENS\":\n      return \"max_tokens\";\n    case \"STOP\":\n    default:\n      return \"end_turn\";\n  }\n}\n\n/**\n * Extract text from a Gemini response.\n */\nexport function extractText(response: { text?: string }): string {\n  return response.text ?? \"\";\n}\n\n/**\n * Extract tool use blocks from a Gemini response.\n */\nexport function extractToolUse(response: any): LLMToolUseBlock[] {\n  const parts = response.candidates?.[0]?.content?.parts ?? [];\n  const blocks: LLMToolUseBlock[] = [];\n  let index = 0;\n\n  for (const part of parts) {\n    if (part.functionCall?.name) {\n      blocks.push({\n        id: `gemini-call-${index++}`,\n        name: part.functionCall.name,\n        input: part.functionCall.args ?? {},\n      });\n    }\n  }\n\n  return blocks;\n}\n\n/**\n * Convert Tepa ToolSchema to Gemini function declarations.\n */\nexport function toGeminiTools(tools: ToolSchema[]): Record<string, unknown>[] {\n  const functionDeclarations = tools.map((tool) => {\n    const properties: Record<string, unknown> = {};\n    const required: string[] = [];\n\n    for (const [name, param] of Object.entries(tool.parameters)) {\n      properties[name] = {\n        type: param.type.toUpperCase(),\n        description: param.description,\n      };\n      if (param.required !== false) {\n        required.push(name);\n      }\n    }\n\n    return {\n      name: tool.name,\n      description: tool.description,\n      parameters: {\n        type: \"OBJECT\",\n        properties,\n        required,\n      },\n    };\n  });\n\n  return [{ functionDeclarations }];\n}\n"],"mappings":";AAAA,SAAS,aAAa,gBAAgB;AAEtC,SAAS,uBAAoD;;;ACStD,SAAS,iBAAiB,UAAyC;AACxE,SAAO,SAAS,IAAI,CAAC,SAAS;AAAA,IAC5B,MAAM,IAAI,SAAS,cAAc,UAAU;AAAA,IAC3C,OAAO,CAAC,EAAE,MAAM,IAAI,QAAQ,CAAC;AAAA,EAC/B,EAAE;AACJ;AAKO,SAAS,eACd,QAC0D;AAC1D,UAAQ,QAAQ;AAAA,IACd,KAAK;AACH,aAAO;AAAA,IACT,KAAK;AAAA,IACL;AACE,aAAO;AAAA,EACX;AACF;AAKO,SAAS,YAAY,UAAqC;AAC/D,SAAO,SAAS,QAAQ;AAC1B;AAKO,SAAS,eAAe,UAAkC;AAC/D,QAAM,QAAQ,SAAS,aAAa,CAAC,GAAG,SAAS,SAAS,CAAC;AAC3D,QAAM,SAA4B,CAAC;AACnC,MAAI,QAAQ;AAEZ,aAAW,QAAQ,OAAO;AACxB,QAAI,KAAK,cAAc,MAAM;AAC3B,aAAO,KAAK;AAAA,QACV,IAAI,eAAe,OAAO;AAAA,QAC1B,MAAM,KAAK,aAAa;AAAA,QACxB,OAAO,KAAK,aAAa,QAAQ,CAAC;AAAA,MACpC,CAAC;AAAA,IACH;AAAA,EACF;AAEA,SAAO;AACT;AAKO,SAAS,cAAc,OAAgD;AAC5E,QAAM,uBAAuB,MAAM,IAAI,CAAC,SAAS;AAC/C,UAAM,aAAsC,CAAC;AAC7C,UAAM,WAAqB,CAAC;AAE5B,eAAW,CAAC,MAAM,KAAK,KAAK,OAAO,QAAQ,KAAK,UAAU,GAAG;AAC3D,iBAAW,IAAI,IAAI;AAAA,QACjB,MAAM,MAAM,KAAK,YAAY;AAAA,QAC7B,aAAa,MAAM;AAAA,MACrB;AACA,UAAI,MAAM,aAAa,OAAO;AAC5B,iBAAS,KAAK,IAAI;AAAA,MACpB;AAAA,IACF;AAEA,WAAO;AAAA,MACL,MAAM,KAAK;AAAA,MACX,aAAa,KAAK;AAAA,MAClB,YAAY;AAAA,QACV,MAAM;AAAA,QACN;AAAA,QACA;AAAA,MACF;AAAA,IACF;AAAA,EACF,CAAC;AAED,SAAO,CAAC,EAAE,qBAAqB,CAAC;AAClC;;;ADtFA,IAAM,gBAAgB;AACtB,IAAM,qBAAqB;AAQpB,IAAM,iBAAN,cAA6B,gBAAgB;AAAA,EAC/B,eAAe;AAAA,EACjB;AAAA,EAEjB,YAAY,UAAiC,CAAC,GAAG;AAC/C,UAAM,OAAO;AACb,UAAM,SAAS,QAAQ,UAAU,QAAQ,IAAI,kBAAkB,QAAQ,IAAI;AAC3E,SAAK,SAAS,IAAI,YAAY,EAAE,OAAO,CAAC;AAAA,EAC1C;AAAA,EAEA,MAAgB,WACd,UACA,SACsB;AACtB,UAAM,WAAW,iBAAiB,QAAQ;AAE1C,UAAM,SAAkC;AAAA,MACtC,iBAAiB,QAAQ,aAAa;AAAA,IACxC;AAEA,QAAI,QAAQ,gBAAgB,QAAW;AACrC,aAAO,cAAc,QAAQ;AAAA,IAC/B;AAEA,QAAI,QAAQ,cAAc;AACxB,aAAO,oBAAoB,QAAQ;AAAA,IACrC;AAEA,UAAM,SAAkC;AAAA,MACtC,OAAO,QAAQ,SAAS;AAAA,MACxB;AAAA,MACA;AAAA,IACF;AAEA,QAAI,QAAQ,SAAS,QAAQ,MAAM,SAAS,GAAG;AAC7C,aAAO,QAAQ,cAAc,QAAQ,KAAK;AAAA,IAC5C;AAEA,UAAM,WAAW,MAAM,KAAK,OAAO,OAAO,gBAAgB,MAAa;AAEvE,UAAM,aAAa,SAAS,cAAc,CAAC;AAC3C,UAAM,eAAe,WAAW,CAAC,GAAG,gBAAgB;AACpD,UAAM,QAAQ,SAAS,iBAAiB,CAAC;AAEzC,UAAM,UAAU,eAAe,QAAQ;AACvC,UAAM,aAAa,QAAQ,SAAS;AAEpC,WAAO;AAAA,MACL,MAAM,YAAY,QAAQ;AAAA,MAC1B,YAAY;AAAA,QACV,OAAO,MAAM,oBAAoB;AAAA,QACjC,QAAQ,MAAM,wBAAwB;AAAA,MACxC;AAAA,MACA,cAAc,aAAa,aAAa,eAAe,YAAY;AAAA,MACnE,GAAI,cAAc,EAAE,QAAQ;AAAA,IAC9B;AAAA,EACF;AAAA,EAEU,YAAY,OAAyB;AAC7C,QAAI,iBAAiB,WAAW;AAC9B,aAAO;AAAA,IACT;AACA,QAAI,iBAAiB,UAAU;AAC7B,YAAM,SAAS,MAAM;AACrB,UAAI,WAAW,OAAO,WAAW,OAAO,WAAW,OAAO,WAAW,KAAK;AACxE,eAAO;AAAA,MACT;AACA,UAAI,WAAW,OAAQ,WAAW,UAAa,UAAU,KAAM;AAC7D,eAAO;AAAA,MACT;AAAA,IACF;AACA,WAAO;AAAA,EACT;AAAA,EAEU,gBAAgB,QAAgC;AACxD,WAAO;AAAA,EACT;AAAA,EAEU,iBAAiB,OAAyB;AAClD,WAAO,iBAAiB,YAAY,MAAM,WAAW;AAAA,EACvD;AACF;","names":[]}

package/package.json

@@ -0,0 +1,55 @@
+{
+  "name": "@tepa/provider-gemini",
+  "version": "0.1.0",
+  "description": "Google Gemini LLM provider for the Tepa autonomous agent",
+  "license": "MIT",
+  "author": "Frandi <frandi.tech@gmail.com>",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/frandi/tepa-ai.git",
+    "directory": "packages/provider-gemini"
+  },
+  "homepage": "https://github.com/frandi/tepa-ai/tree/main/packages/provider-gemini#readme",
+  "bugs": {
+    "url": "https://github.com/frandi/tepa-ai/issues"
+  },
+  "keywords": ["tepa", "ai", "llm", "agent", "gemini", "google", "provider"],
+  "type": "module",
+  "main": "./dist/index.cjs",
+  "module": "./dist/index.js",
+  "types": "./dist/index.d.ts",
+  "exports": {
+    ".": {
+      "import": {
+        "types": "./dist/index.d.ts",
+        "default": "./dist/index.js"
+      },
+      "require": {
+        "types": "./dist/index.d.cts",
+        "default": "./dist/index.cjs"
+      }
+    }
+  },
+  "files": ["dist"],
+  "sideEffects": false,
+  "engines": {
+    "node": ">=18"
+  },
+  "publishConfig": {
+    "access": "public"
+  },
+  "scripts": {
+    "build": "tsup",
+    "dev": "tsup --watch",
+    "prepublishOnly": "npm run build"
+  },
+  "dependencies": {
+    "@tepa/types": "^0.1.0",
+    "@tepa/provider-core": "^0.1.0",
+    "@google/genai": "^1.x"
+  },
+  "devDependencies": {
+    "tsup": "^8.0.0",
+    "typescript": "^5.5.0"
+  }
+}