tolvyn 1.0.2 → 1.0.3

package/README.md

@@ -1,16 +1,17 @@
-# tolvyn (Node.js / TypeScript SDK)
+# tolvyn · npm
 
-Drop-in replacement for the `openai` and `@anthropic-ai/sdk` packages. Add cost metering, team attribution, and budget enforcement to your AI calls in one line.
+[![npm version](https://img.shields.io/npm/v/tolvyn.svg)](https://www.npmjs.com/package/tolvyn)
+
+Drop-in replacement for `openai` and `@anthropic-ai/sdk`.
+One line change. Every AI call metered, attributed, and governed.
+
+## Install
 
 ```bash
 npm install tolvyn
 ```
 
----
-
-## Quick Start
-
-### OpenAI (ESM / TypeScript)
+## Quick start
 
 ```typescript
 // Before
@@ -20,7 +21,7 @@ const client = new OpenAI();
 // After — one line change
 import { OpenAI } from "tolvyn";
 const client = new OpenAI({
-  tolvynApiKey: "tlv_live_...",    // or set TOLVYN_API_KEY env var
+  tolvynApiKey: "tlv_live_...",
   team: "backend",
   service: "summariser",
 });
@@ -32,152 +33,31 @@ const response = await client.chat.completions.create({
 });
 ```
 
-### OpenAI (CommonJS)
-
-```javascript
-const { OpenAI } = require("tolvyn");
-
-const client = new OpenAI({
-  tolvynApiKey: process.env.TOLVYN_API_KEY,
-  team: "backend",
-  service: "summariser",
-});
-```
-
-### Anthropic (ESM / TypeScript)
+Works the same way for Anthropic:
 
 ```typescript
-// Before
-import Anthropic from "@anthropic-ai/sdk";
-const client = new Anthropic();
-
-// After
 import { Anthropic } from "tolvyn";
-const client = new Anthropic({
-  tolvynApiKey: "tlv_live_...",
-  team: "ml-team",
-  service: "classifier",
-});
-
-const message = await client.messages.create({
-  model: "claude-sonnet-4-6",
-  max_tokens: 1024,
-  messages: [{ role: "user", content: "Hello" }],
-});
+const client = new Anthropic({ tolvynApiKey: "tlv_live_...", team: "ml", service: "classifier" });
 ```
 
----
-
-## Classes
-
-| Class              | Extends                       | Provider  |
-|--------------------|-------------------------------|-----------|
-| `tolvyn.OpenAI`    | `openai` OpenAI               | OpenAI    |
-| `tolvyn.Anthropic` | `@anthropic-ai/sdk` Anthropic | Anthropic |
-
-Both classes are strict drop-ins. Every method, event, and behaviour of the underlying SDK is preserved.
-
----
-
-## Parameters
-
-### `OpenAI` (`TolvynOpenAIOptions`)
-
-Extends `openai.ClientOptions` (omitting `apiKey` and `baseURL`).
-
-| Parameter      | Type                  | Default     | Description                                                                                               |
-|----------------|-----------------------|-------------|-----------------------------------------------------------------------------------------------------------|
-| `tolvynApiKey` | `string or undefined` | `undefined` | Your TOLVYN API key. Falls back to `TOLVYN_API_KEY` env var. Required.                                    |
-| `proxyUrl`     | `string or undefined` | `undefined` | TOLVYN proxy URL. Falls back to `TOLVYN_PROXY_URL` env var, then `http://localhost:8081/v1/proxy/openai`. |
-| `team`         | `string or undefined` | `undefined` | Team name for cost attribution. Sent as `X-Tolvyn-Team` header.                                           |
-| `service`      | `string or undefined` | `undefined` | Service name. Sent as `X-Tolvyn-Service` header.                                                          |
-| `feature`      | `string or undefined` | `undefined` | Feature name. Sent as `X-Tolvyn-Feature` header.                                                          |
-| `agent`        | `string or undefined` | `undefined` | Agent name. Sent as `X-Tolvyn-Agent` header.                                                              |
-| `failOpen`     | `boolean`             | `true`      | If `true` and the proxy is unreachable, retry directly against OpenAI using `openAIApiKey`.               |
-| `openAIApiKey` | `string or undefined` | `undefined` | OpenAI key used only for fail-open fallback. Falls back to `OPENAI_API_KEY` env var.                      |
-| `...rest`      |  any                  |     —       | All other `openai.ClientOptions` fields are forwarded to the underlying client.                           |
-
-### `Anthropic` (`TolvynAnthropicOptions`)
-
-Extends `@anthropic-ai/sdk` ClientOptions (omitting `apiKey` and `baseURL`).
-
-| Parameter        | Type                  | Default     | Description                                                                                                  |
-|------------------|-----------------------|-------------|--------------------------------------------------------------------------------------------------------------|
-| `tolvynApiKey`   | `string or undefined` | `undefined` | Your TOLVYN API key. Falls back to `TOLVYN_API_KEY` env var. Required.                                       |
-| `proxyUrl`       | `string or undefined` | `undefined` | TOLVYN proxy URL. Falls back to `TOLVYN_PROXY_URL` env var, then `http://localhost:8081/v1/proxy/anthropic`. |
-| `team`           | `string or undefined` | `undefined` | Team name for cost attribution.                                                                              |
-| `service`        | `string or undefined` | `undefined` | Service name.                                                                                                |
-| `feature`        | `string or undefined` | `undefined` | Feature name.                                                                                                |
-| `agent`          | `string or undefined` | `undefined` | Agent name.                                                                                                  |
-| `failOpen`       | `boolean`             | `true`      | If `true` and the proxy is unreachable, retry directly against Anthropic.                                    |
-| `anthropicApiKey`| `string or undefined` | `undefined` | Anthropic key used only for fail-open fallback. Falls back to `ANTHROPIC_API_KEY` env var.                   |
-| `...rest`        |  any                  |      —      | All other Anthropic ClientOptions fields are forwarded.                                                      |
-
----
-
-## Tagging
-
-```typescript
-const client = new OpenAI({
-  tolvynApiKey: "tlv_live_...",
-  team: "search-team",         // Maps to a team in TOLVYN → budget applies
-  service: "semantic-search",  // Sub-component (e.g. microservice name)
-  feature: "query-expansion",  // Feature within the service
-  agent: "reranker-v2",        // Agent name for multi-agent pipelines
-});
-```
-
-All four tags are optional and independent. They appear in `tolvyn tail` output, the dashboard, and usage breakdown endpoints.
-
----
-
-## Fail-open behaviour
-
-By default, `failOpen: true`. When the TOLVYN proxy is unreachable (connection refused, HTTP 503), the SDK retries the request directly against the AI provider using the fallback API key.
-
-A proxy outage **never breaks your application**. Requests that bypass the proxy are not metered; they appear in the provider's billing but not in TOLVYN.
-
-To hard-fail on proxy errors:
-
-```typescript
-const client = new OpenAI({ tolvynApiKey: "tlv_live_...", failOpen: false });
-```
-
----
-
-## Imports
-
-### ESM
-
-```typescript
-import { OpenAI, Anthropic } from "tolvyn";
-import type { TolvynOpenAIOptions, TolvynAnthropicOptions } from "tolvyn";
-```
-
-### CommonJS
+CommonJS:
 
 ```javascript
-const { OpenAI, Anthropic } = require("tolvyn");
+const { OpenAI } = require("tolvyn");
+const client = new OpenAI({ tolvynApiKey: process.env.TOLVYN_API_KEY, team: "backend" });
 ```
 
-Both ESM (`dist/esm/`) and CJS (`dist/cjs/`) builds are included.
-
----
+## What you get
 
-## Environment variables
+- **Cost metering** — every request logged with exact token counts and cost in microdollars
+- **Team attribution** — see spend by team and service, not just a total invoice number
+- **Budget enforcement** — set hard limits that block requests before they hit your provider
+- **Immutable ledger** — hash-chained audit trail, verifiable at any time
+- **Drop-in** — no changes to your existing API calls, models, or response handling
 
-| Variable            | Used by           | Description                          |
-|---------------------|-------------------|--------------------------------------|
-| `TOLVYN_API_KEY`    | All classes       | TOLVYN API key                       |
-| `TOLVYN_PROXY_URL`  | All classes       | Proxy URL override                   |
-| `OPENAI_API_KEY`    | OpenAI classes    | OpenAI key for fail-open fallback    |
-| `ANTHROPIC_API_KEY` | Anthropic classes | Anthropic key for fail-open fallback |
+Full docs: [docs.tolvyn.io/nodejs-sdk](https://docs.tolvyn.io/nodejs-sdk)
+Free trial: [tolvyn.io](https://tolvyn.io)
 
 ---
 
-## Requirements
-
-- Node.js 18+
-- `openai >= 4.0.0` (peer dependency)
-- `@anthropic-ai/sdk >= 0.20.0` (peer dependency)
-- TypeScript 5.0+ (if using TypeScript)
+© 2026 TOLVYN. All rights reserved.

package/dist/cjs/google.d.ts

@@ -0,0 +1,23 @@
+/**
+ * TOLVYN Google wrapper — thin drop-in over @google/generative-ai.
+ */
+import { GoogleGenerativeAI, ModelParams, RequestOptions } from '@google/generative-ai';
+export interface TolvynGoogleOptions {
+    tolvynApiKey?: string;
+    proxyUrl?: string;
+    team?: string;
+    service?: string;
+    feature?: string;
+    agent?: string;
+    failOpen?: boolean;
+    googleApiKey?: string;
+}
+export declare class TolvynGoogle extends GoogleGenerativeAI {
+    readonly _tolvynFailOpen: boolean;
+    readonly _tolvynFallbackKey: string | undefined;
+    private readonly _tolvynProxyUrl;
+    private readonly _tolvynHeaders;
+    constructor(options?: TolvynGoogleOptions);
+    getGenerativeModel(modelParams: ModelParams, requestOptions?: RequestOptions): import("@google/generative-ai").GenerativeModel;
+}
+//# sourceMappingURL=google.d.ts.map

package/dist/cjs/google.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"google.d.ts","sourceRoot":"","sources":["../../src/google.ts"],"names":[],"mappings":"AAAA;;GAEG;AACH,OAAO,EACL,kBAAkB,EAClB,WAAW,EACX,cAAc,EACf,MAAM,uBAAuB,CAAC;AAM/B,MAAM,WAAW,mBAAmB;IAClC,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,QAAQ,CAAC,EAAE,OAAO,CAAC;IACnB,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB;AAED,qBAAa,YAAa,SAAQ,kBAAkB;IAClD,SAAgB,eAAe,EAAE,OAAO,CAAC;IACzC,SAAgB,kBAAkB,EAAE,MAAM,GAAG,SAAS,CAAC;IACvD,OAAO,CAAC,QAAQ,CAAC,eAAe,CAAS;IACzC,OAAO,CAAC,QAAQ,CAAC,cAAc,CAAyB;gBAE5C,OAAO,GAAE,mBAAwB;IA4BpC,kBAAkB,CACzB,WAAW,EAAE,WAAW,EACxB,cAAc,CAAC,EAAE,cAAc;CASlC"}

package/dist/cjs/google.js

@@ -0,0 +1,47 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.TolvynGoogle = void 0;
+/**
+ * TOLVYN Google wrapper — thin drop-in over @google/generative-ai.
+ */
+const generative_ai_1 = require("@google/generative-ai");
+// The proxy base URL is prepended to Google API paths (/v1beta/models/...).
+// The TOLVYN proxy strips /v1/proxy/google and forwards the remainder to Google.
+const GOOGLE_DEFAULT_PROXY_URL = 'https://proxy.tolvyn.io/v1/proxy/google';
+class TolvynGoogle extends generative_ai_1.GoogleGenerativeAI {
+    constructor(options = {}) {
+        const tolvynApiKey = options.tolvynApiKey ?? process.env['TOLVYN_API_KEY'];
+        if (!tolvynApiKey) {
+            throw new Error('tolvynApiKey required. Set TOLVYN_API_KEY env var or pass tolvynApiKey.');
+        }
+        // GoogleGenerativeAI sends this value as x-goog-api-key.
+        // The TOLVYN proxy's extractBearer reads x-goog-api-key as a fallback.
+        super(tolvynApiKey);
+        this._tolvynProxyUrl =
+            options.proxyUrl ??
+                process.env['TOLVYN_PROXY_URL'] ??
+                GOOGLE_DEFAULT_PROXY_URL;
+        this._tolvynHeaders = {};
+        if (options.team)
+            this._tolvynHeaders['X-Tolvyn-Team'] = options.team;
+        if (options.service)
+            this._tolvynHeaders['X-Tolvyn-Service'] = options.service;
+        if (options.feature)
+            this._tolvynHeaders['X-Tolvyn-Feature'] = options.feature;
+        if (options.agent)
+            this._tolvynHeaders['X-Tolvyn-Agent'] = options.agent;
+        this._tolvynFailOpen = options.failOpen ?? true;
+        this._tolvynFallbackKey =
+            options.googleApiKey ?? process.env['GOOGLE_API_KEY'];
+    }
+    getGenerativeModel(modelParams, requestOptions) {
+        const mergedOptions = {
+            baseUrl: this._tolvynProxyUrl,
+            customHeaders: this._tolvynHeaders,
+            ...requestOptions,
+        };
+        return super.getGenerativeModel(modelParams, mergedOptions);
+    }
+}
+exports.TolvynGoogle = TolvynGoogle;
+//# sourceMappingURL=google.js.map

package/dist/cjs/google.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"google.js","sourceRoot":"","sources":["../../src/google.ts"],"names":[],"mappings":";;;AAAA;;GAEG;AACH,yDAI+B;AAE/B,4EAA4E;AAC5E,iFAAiF;AACjF,MAAM,wBAAwB,GAAG,yCAAyC,CAAC;AAa3E,MAAa,YAAa,SAAQ,kCAAkB;IAMlD,YAAY,UAA+B,EAAE;QAC3C,MAAM,YAAY,GAAG,OAAO,CAAC,YAAY,IAAI,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;QAC3E,IAAI,CAAC,YAAY,EAAE,CAAC;YAClB,MAAM,IAAI,KAAK,CACb,yEAAyE,CAC1E,CAAC;QACJ,CAAC;QAED,yDAAyD;QACzD,uEAAuE;QACvE,KAAK,CAAC,YAAY,CAAC,CAAC;QAEpB,IAAI,CAAC,eAAe;YAClB,OAAO,CAAC,QAAQ;gBAChB,OAAO,CAAC,GAAG,CAAC,kBAAkB,CAAC;gBAC/B,wBAAwB,CAAC;QAE3B,IAAI,CAAC,cAAc,GAAG,EAAE,CAAC;QACzB,IAAI,OAAO,CAAC,IAAI;YAAK,IAAI,CAAC,cAAc,CAAC,eAAe,CAAC,GAAM,OAAO,CAAC,IAAI,CAAC;QAC5E,IAAI,OAAO,CAAC,OAAO;YAAE,IAAI,CAAC,cAAc,CAAC,kBAAkB,CAAC,GAAG,OAAO,CAAC,OAAO,CAAC;QAC/E,IAAI,OAAO,CAAC,OAAO;YAAE,IAAI,CAAC,cAAc,CAAC,kBAAkB,CAAC,GAAG,OAAO,CAAC,OAAO,CAAC;QAC/E,IAAI,OAAO,CAAC,KAAK;YAAI,IAAI,CAAC,cAAc,CAAC,gBAAgB,CAAC,GAAK,OAAO,CAAC,KAAK,CAAC;QAE7E,IAAI,CAAC,eAAe,GAAG,OAAO,CAAC,QAAQ,IAAI,IAAI,CAAC;QAChD,IAAI,CAAC,kBAAkB;YACrB,OAAO,CAAC,YAAY,IAAI,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;IAC1D,CAAC;IAEQ,kBAAkB,CACzB,WAAwB,EACxB,cAA+B;QAE/B,MAAM,aAAa,GAAmB;YACpC,OAAO,EAAE,IAAI,CAAC,eAAe;YAC7B,aAAa,EAAE,IAAI,CAAC,cAAc;YAClC,GAAG,cAAc;SAClB,CAAC;QACF,OAAO,KAAK,CAAC,kBAAkB,CAAC,WAAW,EAAE,aAAa,CAAC,CAAC;IAC9D,CAAC;CACF;AA7CD,oCA6CC"}

package/dist/cjs/index.d.ts

@@ -2,4 +2,6 @@ export { OpenAI } from './client';
 export type { TolvynOpenAIOptions } from './client';
 export { Anthropic } from './anthropic';
 export type { TolvynAnthropicOptions } from './anthropic';
+export { TolvynGoogle as Google } from './google';
+export type { TolvynGoogleOptions } from './google';
 //# sourceMappingURL=index.d.ts.map

package/dist/cjs/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,UAAU,CAAC;AAClC,YAAY,EAAE,mBAAmB,EAAE,MAAM,UAAU,CAAC;AACpD,OAAO,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AACxC,YAAY,EAAE,sBAAsB,EAAE,MAAM,aAAa,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,UAAU,CAAC;AAClC,YAAY,EAAE,mBAAmB,EAAE,MAAM,UAAU,CAAC;AACpD,OAAO,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AACxC,YAAY,EAAE,sBAAsB,EAAE,MAAM,aAAa,CAAC;AAC1D,OAAO,EAAE,YAAY,IAAI,MAAM,EAAE,MAAM,UAAU,CAAC;AAClD,YAAY,EAAE,mBAAmB,EAAE,MAAM,UAAU,CAAC"}

package/dist/cjs/index.js

@@ -1,8 +1,10 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.Anthropic = exports.OpenAI = void 0;
+exports.Google = exports.Anthropic = exports.OpenAI = void 0;
 var client_1 = require("./client");
 Object.defineProperty(exports, "OpenAI", { enumerable: true, get: function () { return client_1.OpenAI; } });
 var anthropic_1 = require("./anthropic");
 Object.defineProperty(exports, "Anthropic", { enumerable: true, get: function () { return anthropic_1.Anthropic; } });
+var google_1 = require("./google");
+Object.defineProperty(exports, "Google", { enumerable: true, get: function () { return google_1.TolvynGoogle; } });
 //# sourceMappingURL=index.js.map

package/dist/cjs/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":";;;AAAA,mCAAkC;AAAzB,gGAAA,MAAM,OAAA;AAEf,yCAAwC;AAA/B,sGAAA,SAAS,OAAA"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":";;;AAAA,mCAAkC;AAAzB,gGAAA,MAAM,OAAA;AAEf,yCAAwC;AAA/B,sGAAA,SAAS,OAAA;AAElB,mCAAkD;AAAzC,gGAAA,YAAY,OAAU"}

package/dist/esm/google.d.ts

@@ -0,0 +1,23 @@
+/**
+ * TOLVYN Google wrapper — thin drop-in over @google/generative-ai.
+ */
+import { GoogleGenerativeAI, ModelParams, RequestOptions } from '@google/generative-ai';
+export interface TolvynGoogleOptions {
+    tolvynApiKey?: string;
+    proxyUrl?: string;
+    team?: string;
+    service?: string;
+    feature?: string;
+    agent?: string;
+    failOpen?: boolean;
+    googleApiKey?: string;
+}
+export declare class TolvynGoogle extends GoogleGenerativeAI {
+    readonly _tolvynFailOpen: boolean;
+    readonly _tolvynFallbackKey: string | undefined;
+    private readonly _tolvynProxyUrl;
+    private readonly _tolvynHeaders;
+    constructor(options?: TolvynGoogleOptions);
+    getGenerativeModel(modelParams: ModelParams, requestOptions?: RequestOptions): import("@google/generative-ai").GenerativeModel;
+}
+//# sourceMappingURL=google.d.ts.map

package/dist/esm/google.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"google.d.ts","sourceRoot":"","sources":["../../src/google.ts"],"names":[],"mappings":"AAAA;;GAEG;AACH,OAAO,EACL,kBAAkB,EAClB,WAAW,EACX,cAAc,EACf,MAAM,uBAAuB,CAAC;AAM/B,MAAM,WAAW,mBAAmB;IAClC,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,QAAQ,CAAC,EAAE,OAAO,CAAC;IACnB,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB;AAED,qBAAa,YAAa,SAAQ,kBAAkB;IAClD,SAAgB,eAAe,EAAE,OAAO,CAAC;IACzC,SAAgB,kBAAkB,EAAE,MAAM,GAAG,SAAS,CAAC;IACvD,OAAO,CAAC,QAAQ,CAAC,eAAe,CAAS;IACzC,OAAO,CAAC,QAAQ,CAAC,cAAc,CAAyB;gBAE5C,OAAO,GAAE,mBAAwB;IA4BpC,kBAAkB,CACzB,WAAW,EAAE,WAAW,EACxB,cAAc,CAAC,EAAE,cAAc;CASlC"}

package/dist/esm/google.js

@@ -0,0 +1,43 @@
+/**
+ * TOLVYN Google wrapper — thin drop-in over @google/generative-ai.
+ */
+import { GoogleGenerativeAI, } from '@google/generative-ai';
+// The proxy base URL is prepended to Google API paths (/v1beta/models/...).
+// The TOLVYN proxy strips /v1/proxy/google and forwards the remainder to Google.
+const GOOGLE_DEFAULT_PROXY_URL = 'https://proxy.tolvyn.io/v1/proxy/google';
+export class TolvynGoogle extends GoogleGenerativeAI {
+    constructor(options = {}) {
+        const tolvynApiKey = options.tolvynApiKey ?? process.env['TOLVYN_API_KEY'];
+        if (!tolvynApiKey) {
+            throw new Error('tolvynApiKey required. Set TOLVYN_API_KEY env var or pass tolvynApiKey.');
+        }
+        // GoogleGenerativeAI sends this value as x-goog-api-key.
+        // The TOLVYN proxy's extractBearer reads x-goog-api-key as a fallback.
+        super(tolvynApiKey);
+        this._tolvynProxyUrl =
+            options.proxyUrl ??
+                process.env['TOLVYN_PROXY_URL'] ??
+                GOOGLE_DEFAULT_PROXY_URL;
+        this._tolvynHeaders = {};
+        if (options.team)
+            this._tolvynHeaders['X-Tolvyn-Team'] = options.team;
+        if (options.service)
+            this._tolvynHeaders['X-Tolvyn-Service'] = options.service;
+        if (options.feature)
+            this._tolvynHeaders['X-Tolvyn-Feature'] = options.feature;
+        if (options.agent)
+            this._tolvynHeaders['X-Tolvyn-Agent'] = options.agent;
+        this._tolvynFailOpen = options.failOpen ?? true;
+        this._tolvynFallbackKey =
+            options.googleApiKey ?? process.env['GOOGLE_API_KEY'];
+    }
+    getGenerativeModel(modelParams, requestOptions) {
+        const mergedOptions = {
+            baseUrl: this._tolvynProxyUrl,
+            customHeaders: this._tolvynHeaders,
+            ...requestOptions,
+        };
+        return super.getGenerativeModel(modelParams, mergedOptions);
+    }
+}
+//# sourceMappingURL=google.js.map

package/dist/esm/google.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"google.js","sourceRoot":"","sources":["../../src/google.ts"],"names":[],"mappings":"AAAA;;GAEG;AACH,OAAO,EACL,kBAAkB,GAGnB,MAAM,uBAAuB,CAAC;AAE/B,4EAA4E;AAC5E,iFAAiF;AACjF,MAAM,wBAAwB,GAAG,yCAAyC,CAAC;AAa3E,MAAM,OAAO,YAAa,SAAQ,kBAAkB;IAMlD,YAAY,UAA+B,EAAE;QAC3C,MAAM,YAAY,GAAG,OAAO,CAAC,YAAY,IAAI,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;QAC3E,IAAI,CAAC,YAAY,EAAE,CAAC;YAClB,MAAM,IAAI,KAAK,CACb,yEAAyE,CAC1E,CAAC;QACJ,CAAC;QAED,yDAAyD;QACzD,uEAAuE;QACvE,KAAK,CAAC,YAAY,CAAC,CAAC;QAEpB,IAAI,CAAC,eAAe;YAClB,OAAO,CAAC,QAAQ;gBAChB,OAAO,CAAC,GAAG,CAAC,kBAAkB,CAAC;gBAC/B,wBAAwB,CAAC;QAE3B,IAAI,CAAC,cAAc,GAAG,EAAE,CAAC;QACzB,IAAI,OAAO,CAAC,IAAI;YAAK,IAAI,CAAC,cAAc,CAAC,eAAe,CAAC,GAAM,OAAO,CAAC,IAAI,CAAC;QAC5E,IAAI,OAAO,CAAC,OAAO;YAAE,IAAI,CAAC,cAAc,CAAC,kBAAkB,CAAC,GAAG,OAAO,CAAC,OAAO,CAAC;QAC/E,IAAI,OAAO,CAAC,OAAO;YAAE,IAAI,CAAC,cAAc,CAAC,kBAAkB,CAAC,GAAG,OAAO,CAAC,OAAO,CAAC;QAC/E,IAAI,OAAO,CAAC,KAAK;YAAI,IAAI,CAAC,cAAc,CAAC,gBAAgB,CAAC,GAAK,OAAO,CAAC,KAAK,CAAC;QAE7E,IAAI,CAAC,eAAe,GAAG,OAAO,CAAC,QAAQ,IAAI,IAAI,CAAC;QAChD,IAAI,CAAC,kBAAkB;YACrB,OAAO,CAAC,YAAY,IAAI,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;IAC1D,CAAC;IAEQ,kBAAkB,CACzB,WAAwB,EACxB,cAA+B;QAE/B,MAAM,aAAa,GAAmB;YACpC,OAAO,EAAE,IAAI,CAAC,eAAe;YAC7B,aAAa,EAAE,IAAI,CAAC,cAAc;YAClC,GAAG,cAAc;SAClB,CAAC;QACF,OAAO,KAAK,CAAC,kBAAkB,CAAC,WAAW,EAAE,aAAa,CAAC,CAAC;IAC9D,CAAC;CACF"}

package/dist/esm/index.d.ts

@@ -2,4 +2,6 @@ export { OpenAI } from './client';
 export type { TolvynOpenAIOptions } from './client';
 export { Anthropic } from './anthropic';
 export type { TolvynAnthropicOptions } from './anthropic';
+export { TolvynGoogle as Google } from './google';
+export type { TolvynGoogleOptions } from './google';
 //# sourceMappingURL=index.d.ts.map

package/dist/esm/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,UAAU,CAAC;AAClC,YAAY,EAAE,mBAAmB,EAAE,MAAM,UAAU,CAAC;AACpD,OAAO,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AACxC,YAAY,EAAE,sBAAsB,EAAE,MAAM,aAAa,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,UAAU,CAAC;AAClC,YAAY,EAAE,mBAAmB,EAAE,MAAM,UAAU,CAAC;AACpD,OAAO,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AACxC,YAAY,EAAE,sBAAsB,EAAE,MAAM,aAAa,CAAC;AAC1D,OAAO,EAAE,YAAY,IAAI,MAAM,EAAE,MAAM,UAAU,CAAC;AAClD,YAAY,EAAE,mBAAmB,EAAE,MAAM,UAAU,CAAC"}

package/dist/esm/index.js

@@ -1,3 +1,4 @@
 export { OpenAI } from './client';
 export { Anthropic } from './anthropic';
+export { TolvynGoogle as Google } from './google';
 //# sourceMappingURL=index.js.map

package/dist/esm/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,UAAU,CAAC;AAElC,OAAO,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,UAAU,CAAC;AAElC,OAAO,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AAExC,OAAO,EAAE,YAAY,IAAI,MAAM,EAAE,MAAM,UAAU,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "tolvyn",
-  "version": "1.0.2",
+  "version": "1.0.3",
   "description": "Drop-in replacement for the OpenAI/Anthropic SDK — routes through TOLVYN for cost attribution, budget enforcement, and audit logging.",
   "main": "./dist/cjs/index.js",
   "module": "./dist/esm/index.js",
@@ -23,6 +23,7 @@
   },
   "dependencies": {
     "@anthropic-ai/sdk": "^0.20.0",
+    "@google/generative-ai": "^0.21.0",
     "openai": "^4.0.0"
   },
   "devDependencies": {
@@ -35,7 +36,9 @@
   "jest": {
     "preset": "ts-jest",
     "testEnvironment": "node",
-    "testMatch": ["**/tests/**/*.test.ts"],
+    "testMatch": [
+      "**/tests/**/*.test.ts"
+    ],
     "globals": {
       "ts-jest": {
         "tsconfig": "tsconfig.cjs.json"

package/src/google.ts

@@ -0,0 +1,70 @@
+/**
+ * TOLVYN Google wrapper — thin drop-in over @google/generative-ai.
+ */
+import {
+  GoogleGenerativeAI,
+  ModelParams,
+  RequestOptions,
+} from '@google/generative-ai';
+
+// The proxy base URL is prepended to Google API paths (/v1beta/models/...).
+// The TOLVYN proxy strips /v1/proxy/google and forwards the remainder to Google.
+const GOOGLE_DEFAULT_PROXY_URL = 'https://proxy.tolvyn.io/v1/proxy/google';
+
+export interface TolvynGoogleOptions {
+  tolvynApiKey?: string;
+  proxyUrl?: string;
+  team?: string;
+  service?: string;
+  feature?: string;
+  agent?: string;
+  failOpen?: boolean;
+  googleApiKey?: string;
+}
+
+export class TolvynGoogle extends GoogleGenerativeAI {
+  public readonly _tolvynFailOpen: boolean;
+  public readonly _tolvynFallbackKey: string | undefined;
+  private readonly _tolvynProxyUrl: string;
+  private readonly _tolvynHeaders: Record<string, string>;
+
+  constructor(options: TolvynGoogleOptions = {}) {
+    const tolvynApiKey = options.tolvynApiKey ?? process.env['TOLVYN_API_KEY'];
+    if (!tolvynApiKey) {
+      throw new Error(
+        'tolvynApiKey required. Set TOLVYN_API_KEY env var or pass tolvynApiKey.'
+      );
+    }
+
+    // GoogleGenerativeAI sends this value as x-goog-api-key.
+    // The TOLVYN proxy's extractBearer reads x-goog-api-key as a fallback.
+    super(tolvynApiKey);
+
+    this._tolvynProxyUrl =
+      options.proxyUrl ??
+      process.env['TOLVYN_PROXY_URL'] ??
+      GOOGLE_DEFAULT_PROXY_URL;
+
+    this._tolvynHeaders = {};
+    if (options.team)    this._tolvynHeaders['X-Tolvyn-Team']    = options.team;
+    if (options.service) this._tolvynHeaders['X-Tolvyn-Service'] = options.service;
+    if (options.feature) this._tolvynHeaders['X-Tolvyn-Feature'] = options.feature;
+    if (options.agent)   this._tolvynHeaders['X-Tolvyn-Agent']   = options.agent;
+
+    this._tolvynFailOpen = options.failOpen ?? true;
+    this._tolvynFallbackKey =
+      options.googleApiKey ?? process.env['GOOGLE_API_KEY'];
+  }
+
+  override getGenerativeModel(
+    modelParams: ModelParams,
+    requestOptions?: RequestOptions
+  ) {
+    const mergedOptions: RequestOptions = {
+      baseUrl: this._tolvynProxyUrl,
+      customHeaders: this._tolvynHeaders,
+      ...requestOptions,
+    };
+    return super.getGenerativeModel(modelParams, mergedOptions);
+  }
+}

package/src/index.ts

@@ -2,3 +2,5 @@ export { OpenAI } from './client';
 export type { TolvynOpenAIOptions } from './client';
 export { Anthropic } from './anthropic';
 export type { TolvynAnthropicOptions } from './anthropic';
+export { TolvynGoogle as Google } from './google';
+export type { TolvynGoogleOptions } from './google';