@axemere/gateway-google 0.1.9

package/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 Axemere LLC
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

package/README.md

@@ -0,0 +1,30 @@
+# @axemere/gateway-google
+
+Google Generative AI drop-in wrapper for the [Axemere AI Gateway](https://axemere.ai).
+
+## Installation
+
+```bash
+npm install @axemere/gateway-google @google/generative-ai
+```
+
+## Usage
+
+```ts
+import { genaiClient } from "@axemere/gateway-google";
+import { AiGatewayConfig } from "@axemere/gateway";
+
+const ai = genaiClient(new AiGatewayConfig());
+const model = ai.getGenerativeModel({ model: "gemini-1.5-flash" });
+const result = await model.generateContent("Explain quantum entanglement in one sentence.");
+console.log(result.response.text());
+```
+
+The returned client is a standard `GoogleGenerativeAI` instance — no other code changes needed.
+
+## Configuration
+
+Reads `AXEMERE_GATEWAY_URL` and `AXEMERE_GATEWAY_TOKEN` from the environment, or pass an
+`AiGatewayConfig` instance explicitly.
+
+See [@axemere/gateway](https://www.npmjs.com/package/@axemere/gateway) for all configuration options.

package/dist/index.d.ts

@@ -0,0 +1,33 @@
+import { GoogleGenerativeAI } from "@google/generative-ai";
+import type { ModelParams, RequestOptions } from "@google/generative-ai";
+import { AiGatewayConfig } from "@axemere/gateway";
+export { AiGatewayConfig };
+/**
+ * GoogleGenerativeAI subclass that automatically routes all model requests
+ * through the Axemere AI Gateway. Gateway auth headers and base URL are injected
+ * into every getGenerativeModel() call without requiring the caller to pass
+ * requestOptions manually.
+ */
+declare class GatewayGenerativeAI extends GoogleGenerativeAI {
+    private readonly _gatewayRequestOptions;
+    constructor(apiKey: string, gatewayRequestOptions: RequestOptions);
+    getGenerativeModel(modelParams: ModelParams, requestOptions?: RequestOptions): import("@google/generative-ai").GenerativeModel;
+}
+/**
+ * Creates a GoogleGenerativeAI client pre-configured to route through the Axemere AI Gateway.
+ *
+ * The returned client is a standard GoogleGenerativeAI instance — use it exactly as you would
+ * the official @google/generative-ai package. All requests are proxied through the gateway,
+ * which applies policy, budgets, and records usage.
+ *
+ * @param config - AiGatewayConfig instance (reads env vars if omitted).
+ * @returns A GoogleGenerativeAI instance with gateway routing applied to every model request.
+ *
+ * @example
+ * ```ts
+ * const ai = genaiClient(new AiGatewayConfig());
+ * const model = ai.getGenerativeModel({ model: "gemini-1.5-flash" });
+ * const result = await model.generateContent("Hello");
+ * ```
+ */
+export declare function genaiClient(config?: AiGatewayConfig): GatewayGenerativeAI;

package/dist/index.js

@@ -0,0 +1,64 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AiGatewayConfig = void 0;
+exports.genaiClient = genaiClient;
+const generative_ai_1 = require("@google/generative-ai");
+const gateway_1 = require("@axemere/gateway");
+Object.defineProperty(exports, "AiGatewayConfig", { enumerable: true, get: function () { return gateway_1.AiGatewayConfig; } });
+/**
+ * GoogleGenerativeAI subclass that automatically routes all model requests
+ * through the Axemere AI Gateway. Gateway auth headers and base URL are injected
+ * into every getGenerativeModel() call without requiring the caller to pass
+ * requestOptions manually.
+ */
+class GatewayGenerativeAI extends generative_ai_1.GoogleGenerativeAI {
+    constructor(apiKey, gatewayRequestOptions) {
+        super(apiKey);
+        this._gatewayRequestOptions = gatewayRequestOptions;
+    }
+    getGenerativeModel(modelParams, requestOptions) {
+        // Merge gateway defaults under caller-supplied options so callers can
+        // still override baseUrl or add their own custom headers.
+        const merged = {
+            ...this._gatewayRequestOptions,
+            ...requestOptions,
+        };
+        // Merge customHeaders: gateway headers first, caller headers win on conflict.
+        // Note: x-goog-api-key cannot be set via customHeaders (SDK enforces this),
+        // so the placeholder key is forwarded as-is. The gateway recognises
+        // PLACEHOLDER_API_KEY ("axemere-proxy-managed") and does not treat it as BYOK.
+        const gwHeaders = new Headers(this._gatewayRequestOptions.customHeaders ?? {});
+        if (requestOptions?.customHeaders) {
+            const callerHeaders = new Headers(requestOptions.customHeaders);
+            callerHeaders.forEach((val, key) => gwHeaders.set(key, val));
+        }
+        merged.customHeaders = gwHeaders;
+        return super.getGenerativeModel(modelParams, merged);
+    }
+}
+/**
+ * Creates a GoogleGenerativeAI client pre-configured to route through the Axemere AI Gateway.
+ *
+ * The returned client is a standard GoogleGenerativeAI instance — use it exactly as you would
+ * the official @google/generative-ai package. All requests are proxied through the gateway,
+ * which applies policy, budgets, and records usage.
+ *
+ * @param config - AiGatewayConfig instance (reads env vars if omitted).
+ * @returns A GoogleGenerativeAI instance with gateway routing applied to every model request.
+ *
+ * @example
+ * ```ts
+ * const ai = genaiClient(new AiGatewayConfig());
+ * const model = ai.getGenerativeModel({ model: "gemini-1.5-flash" });
+ * const result = await model.generateContent("Hello");
+ * ```
+ */
+function genaiClient(config) {
+    const cfg = config ?? new gateway_1.AiGatewayConfig();
+    const headers = (0, gateway_1.aiGatewayHeaders)(cfg, { targetHost: "generativelanguage.googleapis.com" });
+    const gatewayRequestOptions = {
+        baseUrl: cfg.gateway_url,
+        customHeaders: new Headers(headers),
+    };
+    return new GatewayGenerativeAI(gateway_1.PLACEHOLDER_API_KEY, gatewayRequestOptions);
+}

package/dist/index.test.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/index.test.js

@@ -0,0 +1,54 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const generative_ai_1 = require("@google/generative-ai");
+const gateway_1 = require("@axemere/gateway");
+const index_1 = require("./index");
+describe("genaiClient()", () => {
+    it("returns a GoogleGenerativeAI instance", () => {
+        const cfg = new gateway_1.AiGatewayConfig({ gateway_url: "http://localhost:7080" });
+        const client = (0, index_1.genaiClient)(cfg);
+        expect(client).toBeInstanceOf(generative_ai_1.GoogleGenerativeAI);
+    });
+    it("constructs with default config when no config passed", () => {
+        delete process.env["AXEMERE_GATEWAY_URL"];
+        delete process.env["AXEMERE_GATEWAY_TOKEN"];
+        expect(() => (0, index_1.genaiClient)()).not.toThrow();
+    });
+    it("injects Authorization header from gateway_token", () => {
+        const cfg = new gateway_1.AiGatewayConfig({
+            gateway_url: "http://localhost:7080",
+            gateway_token: "tok_test",
+        });
+        const client = (0, index_1.genaiClient)(cfg);
+        // Retrieve the options injected into getGenerativeModel by calling it and
+        // inspecting the GenerativeModel's private _requestOptions.
+        const model = client.getGenerativeModel({ model: "gemini-1.5-flash" });
+        const opts = model
+            ._requestOptions;
+        expect(opts.customHeaders.get("Authorization")).toBe("Bearer tok_test");
+    });
+    it("sets baseUrl to the gateway URL", () => {
+        const cfg = new gateway_1.AiGatewayConfig({ gateway_url: "http://localhost:7080" });
+        const client = (0, index_1.genaiClient)(cfg);
+        const model = client.getGenerativeModel({ model: "gemini-1.5-flash" });
+        const opts = model
+            ._requestOptions;
+        expect(opts.baseUrl).toBe("http://localhost:7080");
+    });
+    it("sets X-MVGC-Target-Host to generativelanguage.googleapis.com", () => {
+        const cfg = new gateway_1.AiGatewayConfig({ gateway_url: "http://localhost:7080" });
+        const client = (0, index_1.genaiClient)(cfg);
+        const model = client.getGenerativeModel({ model: "gemini-1.5-flash" });
+        const opts = model
+            ._requestOptions;
+        expect(opts.customHeaders.get("X-MVGC-Target-Host")).toBe("generativelanguage.googleapis.com");
+    });
+    it("caller requestOptions override gateway defaults", () => {
+        const cfg = new gateway_1.AiGatewayConfig({ gateway_url: "http://localhost:7080" });
+        const client = (0, index_1.genaiClient)(cfg);
+        const model = client.getGenerativeModel({ model: "gemini-1.5-flash" }, { baseUrl: "http://other-gateway:9090" });
+        const opts = model
+            ._requestOptions;
+        expect(opts.baseUrl).toBe("http://other-gateway:9090");
+    });
+});

package/package.json

@@ -0,0 +1,53 @@
+{
+  "name": "@axemere/gateway-google",
+  "version": "0.1.9",
+  "description": "Axemere AI Gateway SDK — Google Generative AI wrapper",
+  "main": "dist/index.js",
+  "types": "dist/index.d.ts",
+  "files": [
+    "dist",
+    "README.md"
+  ],
+  "license": "MIT",
+  "engines": {
+    "node": ">=18"
+  },
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/Axemere-LLC/axemere-node.git",
+    "directory": "packages/gateway-google"
+  },
+  "publishConfig": {
+    "access": "public"
+  },
+  "peerDependencies": {
+    "@google/generative-ai": ">=0.21.0"
+  },
+  "dependencies": {
+    "@axemere/gateway": "0.1.9"
+  },
+  "devDependencies": {
+    "@google/generative-ai": "^0.24.0",
+    "typescript": "^5.4.0",
+    "@types/node": "^20.0.0",
+    "@types/jest": "^29.0.0",
+    "jest": "^29.0.0",
+    "ts-jest": "^29.0.0"
+  },
+  "jest": {
+    "preset": "ts-jest",
+    "testEnvironment": "node",
+    "transform": {
+      "^.+\\.tsx?$": [
+        "ts-jest",
+        {
+          "useESM": false
+        }
+      ]
+    }
+  },
+  "scripts": {
+    "build": "tsc",
+    "test": "jest"
+  }
+}