@trustgateai/sdk 1.0.0 → 1.1.0

package/dist/{chunk-UUGIIO3Q.mjs → chunk-HWMHN2XV.mjs}

@@ -1,7 +1,7 @@
 import {
   buildContextHeaders,
   getWorkflowContextFromEnv
-} from "./chunk-SV5FTXAH.mjs";
+} from "./chunk-Z2D2AATH.mjs";
 
 // src/middleware/vercel-ai.ts
 function getTrustGateHeaders(workflowContext) {

package/dist/chunk-OCDNSTSM.mjs

@@ -0,0 +1,31 @@
+import {
+  buildContextHeaders,
+  getWorkflowContextFromEnv
+} from "./chunk-Z2D2AATH.mjs";
+
+// src/middleware/google.ts
+function getGeminiHeaders(config) {
+  const ctx = config.workflowContext ?? getWorkflowContextFromEnv();
+  const headers = {
+    ...config.headers,
+    ...buildContextHeaders(ctx)
+  };
+  if (config.apiKey) {
+    headers["x-api-key"] = config.apiKey;
+  }
+  return headers;
+}
+function createTrustGateGeminiConfig(config) {
+  const baseUrl = config.baseUrl.replace(/\/$/, "");
+  const headers = getGeminiHeaders(config);
+  return {
+    baseUrl,
+    apiKey: config.apiKey ?? null,
+    headers: Object.keys(headers).length ? headers : void 0
+  };
+}
+
+export {
+  getGeminiHeaders,
+  createTrustGateGeminiConfig
+};

package/dist/chunk-QJUVF52Z.mjs

@@ -0,0 +1,31 @@
+import {
+  buildContextHeaders,
+  getWorkflowContextFromEnv
+} from "./chunk-Z2D2AATH.mjs";
+
+// src/middleware/anthropic.ts
+function getAnthropicHeaders(config) {
+  const ctx = config.workflowContext ?? getWorkflowContextFromEnv();
+  const headers = {
+    ...config.headers,
+    ...buildContextHeaders(ctx)
+  };
+  if (config.apiKey) {
+    headers["x-api-key"] = config.apiKey;
+  }
+  return headers;
+}
+function createTrustGateAnthropicConfig(config) {
+  const baseURL = config.baseUrl.replace(/\/$/, "");
+  const headers = getAnthropicHeaders(config);
+  return {
+    baseURL,
+    apiKey: config.apiKey ?? null,
+    defaultHeaders: Object.keys(headers).length ? headers : void 0
+  };
+}
+
+export {
+  getAnthropicHeaders,
+  createTrustGateAnthropicConfig
+};

package/dist/{chunk-R2YXTU4L.mjs → chunk-STJ53FX3.mjs}

@@ -1,7 +1,7 @@
 import {
   buildContextHeaders,
   getWorkflowContextFromEnv
-} from "./chunk-SV5FTXAH.mjs";
+} from "./chunk-Z2D2AATH.mjs";
 
 // src/middleware/langchain.ts
 function createTrustGateLangChainConfig(config) {

package/dist/{chunk-SV5FTXAH.mjs → chunk-Z2D2AATH.mjs}

@@ -1,5 +1,5 @@
 // src/version.ts
-var SDK_VERSION = "1.0.0";
+var SDK_VERSION = "1.1.0";
 
 // src/context.ts
 var HEADER_PREFIX = "x-trustgate";

package/dist/index.d.mts

@@ -1,7 +1,9 @@
-import { T as TrustGateConfig, R as RequestOptions, W as WorkflowContext, N as N8nHeaderSnippet } from './types-BxyYog9f.mjs';
-export { a as RuntimeMetadata } from './types-BxyYog9f.mjs';
+import { T as TrustGateConfig, R as RequestOptions, W as WorkflowContext, N as N8nHeaderSnippet } from './types-QFksyF5p.mjs';
+export { P as Provider, a as RuntimeMetadata } from './types-QFksyF5p.mjs';
 export { TrustGateOpenAIOptions, createTrustGateOpenAIOptions, getTrustGateHeaders } from './middleware/vercel-ai.mjs';
 export { createTrustGateLangChainConfig } from './middleware/langchain.mjs';
+export { createTrustGateAnthropicConfig, getAnthropicHeaders } from './middleware/anthropic.mjs';
+export { createTrustGateGeminiConfig, getGeminiHeaders } from './middleware/google.mjs';
 
 /**
  * TrustGate Node.js client. Forwards requests to the TrustGate gateway with
@@ -24,6 +26,8 @@ declare class TrustGate {
     getWorkflowContext(options?: RequestOptions): WorkflowContext;
     /**
      * Perform a non-streaming JSON request through TrustGate.
+     * @param options.provider - Optional provider for this request ('openai' | 'anthropic' | 'google');
+     *   controls auth header format (e.g. x-api-key only for anthropic/google).
      */
     fetch(path: string, init?: RequestInit, options?: RequestOptions): Promise<Response>;
     /**

package/dist/index.d.ts

@@ -1,7 +1,9 @@
-import { T as TrustGateConfig, R as RequestOptions, W as WorkflowContext, N as N8nHeaderSnippet } from './types-BxyYog9f.js';
-export { a as RuntimeMetadata } from './types-BxyYog9f.js';
+import { T as TrustGateConfig, R as RequestOptions, W as WorkflowContext, N as N8nHeaderSnippet } from './types-QFksyF5p.js';
+export { P as Provider, a as RuntimeMetadata } from './types-QFksyF5p.js';
 export { TrustGateOpenAIOptions, createTrustGateOpenAIOptions, getTrustGateHeaders } from './middleware/vercel-ai.js';
 export { createTrustGateLangChainConfig } from './middleware/langchain.js';
+export { createTrustGateAnthropicConfig, getAnthropicHeaders } from './middleware/anthropic.js';
+export { createTrustGateGeminiConfig, getGeminiHeaders } from './middleware/google.js';
 
 /**
  * TrustGate Node.js client. Forwards requests to the TrustGate gateway with
@@ -24,6 +26,8 @@ declare class TrustGate {
     getWorkflowContext(options?: RequestOptions): WorkflowContext;
     /**
      * Perform a non-streaming JSON request through TrustGate.
+     * @param options.provider - Optional provider for this request ('openai' | 'anthropic' | 'google');
+     *   controls auth header format (e.g. x-api-key only for anthropic/google).
      */
     fetch(path: string, init?: RequestInit, options?: RequestOptions): Promise<Response>;
     /**

package/dist/index.js

@@ -25,8 +25,12 @@ __export(index_exports, {
   TrustGate: () => TrustGate,
   buildContextHeaders: () => buildContextHeaders,
   buildSourceToolJson: () => buildSourceToolJson,
+  createTrustGateAnthropicConfig: () => createTrustGateAnthropicConfig,
+  createTrustGateGeminiConfig: () => createTrustGateGeminiConfig,
   createTrustGateLangChainConfig: () => createTrustGateLangChainConfig,
   createTrustGateOpenAIOptions: () => createTrustGateOpenAIOptions,
+  getAnthropicHeaders: () => getAnthropicHeaders,
+  getGeminiHeaders: () => getGeminiHeaders,
   getN8nHeaderSnippet: () => getN8nHeaderSnippet,
   getTrustGateHeaders: () => getTrustGateHeaders,
   getWorkflowContextFromEnv: () => getWorkflowContextFromEnv
@@ -34,7 +38,7 @@ __export(index_exports, {
 module.exports = __toCommonJS(index_exports);
 
 // src/version.ts
-var SDK_VERSION = "1.0.0";
+var SDK_VERSION = "1.1.0";
 
 // src/context.ts
 var HEADER_PREFIX = "x-trustgate";
@@ -142,6 +146,19 @@ function buildContextHeaders(ctx) {
 function resolveContext(config, options) {
   return options?.workflowContext ?? config.workflowContext ?? getWorkflowContextFromEnv();
 }
+function authHeadersForProvider(apiKey, provider) {
+  switch (provider) {
+    case "anthropic":
+    case "google":
+      return { "x-api-key": apiKey };
+    case "openai":
+    default:
+      return {
+        authorization: `Bearer ${apiKey}`,
+        "x-api-key": apiKey
+      };
+  }
+}
 function mergeHeaders(config, options, context) {
   const out = {
     "content-type": "application/json",
@@ -149,8 +166,8 @@ function mergeHeaders(config, options, context) {
     ...options?.headers
   };
   if (config.apiKey) {
-    out["authorization"] = `Bearer ${config.apiKey}`;
-    out["x-api-key"] = config.apiKey;
+    const provider = options?.provider ?? config.provider;
+    Object.assign(out, authHeadersForProvider(config.apiKey, provider));
   }
   if (context) {
     Object.assign(out, buildContextHeaders(context));
@@ -175,6 +192,8 @@ var TrustGate = class {
   }
   /**
    * Perform a non-streaming JSON request through TrustGate.
+   * @param options.provider - Optional provider for this request ('openai' | 'anthropic' | 'google');
+   *   controls auth header format (e.g. x-api-key only for anthropic/google).
    */
   async fetch(path, init = {}, options) {
     const context = resolveContext(this.config, options);
@@ -299,6 +318,50 @@ function createTrustGateLangChainConfig(config) {
     }
   };
 }
+
+// src/middleware/anthropic.ts
+function getAnthropicHeaders(config) {
+  const ctx = config.workflowContext ?? getWorkflowContextFromEnv();
+  const headers = {
+    ...config.headers,
+    ...buildContextHeaders(ctx)
+  };
+  if (config.apiKey) {
+    headers["x-api-key"] = config.apiKey;
+  }
+  return headers;
+}
+function createTrustGateAnthropicConfig(config) {
+  const baseURL = config.baseUrl.replace(/\/$/, "");
+  const headers = getAnthropicHeaders(config);
+  return {
+    baseURL,
+    apiKey: config.apiKey ?? null,
+    defaultHeaders: Object.keys(headers).length ? headers : void 0
+  };
+}
+
+// src/middleware/google.ts
+function getGeminiHeaders(config) {
+  const ctx = config.workflowContext ?? getWorkflowContextFromEnv();
+  const headers = {
+    ...config.headers,
+    ...buildContextHeaders(ctx)
+  };
+  if (config.apiKey) {
+    headers["x-api-key"] = config.apiKey;
+  }
+  return headers;
+}
+function createTrustGateGeminiConfig(config) {
+  const baseUrl = config.baseUrl.replace(/\/$/, "");
+  const headers = getGeminiHeaders(config);
+  return {
+    baseUrl,
+    apiKey: config.apiKey ?? null,
+    headers: Object.keys(headers).length ? headers : void 0
+  };
+}
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   HEADER_PREFIX,
@@ -306,8 +369,12 @@ function createTrustGateLangChainConfig(config) {
   TrustGate,
   buildContextHeaders,
   buildSourceToolJson,
+  createTrustGateAnthropicConfig,
+  createTrustGateGeminiConfig,
   createTrustGateLangChainConfig,
   createTrustGateOpenAIOptions,
+  getAnthropicHeaders,
+  getGeminiHeaders,
   getN8nHeaderSnippet,
   getTrustGateHeaders,
   getWorkflowContextFromEnv

package/dist/index.mjs

@@ -1,22 +1,43 @@
+import {
+  createTrustGateAnthropicConfig,
+  getAnthropicHeaders
+} from "./chunk-QJUVF52Z.mjs";
+import {
+  createTrustGateGeminiConfig,
+  getGeminiHeaders
+} from "./chunk-OCDNSTSM.mjs";
 import {
   createTrustGateLangChainConfig
-} from "./chunk-R2YXTU4L.mjs";
+} from "./chunk-STJ53FX3.mjs";
 import {
   createTrustGateOpenAIOptions,
   getTrustGateHeaders
-} from "./chunk-UUGIIO3Q.mjs";
+} from "./chunk-HWMHN2XV.mjs";
 import {
   HEADER_PREFIX,
   SOURCE_TOOL_HEADER,
   buildContextHeaders,
   buildSourceToolJson,
   getWorkflowContextFromEnv
-} from "./chunk-SV5FTXAH.mjs";
+} from "./chunk-Z2D2AATH.mjs";
 
 // src/client.ts
 function resolveContext(config, options) {
   return options?.workflowContext ?? config.workflowContext ?? getWorkflowContextFromEnv();
 }
+function authHeadersForProvider(apiKey, provider) {
+  switch (provider) {
+    case "anthropic":
+    case "google":
+      return { "x-api-key": apiKey };
+    case "openai":
+    default:
+      return {
+        authorization: `Bearer ${apiKey}`,
+        "x-api-key": apiKey
+      };
+  }
+}
 function mergeHeaders(config, options, context) {
   const out = {
     "content-type": "application/json",
@@ -24,8 +45,8 @@ function mergeHeaders(config, options, context) {
     ...options?.headers
   };
   if (config.apiKey) {
-    out["authorization"] = `Bearer ${config.apiKey}`;
-    out["x-api-key"] = config.apiKey;
+    const provider = options?.provider ?? config.provider;
+    Object.assign(out, authHeadersForProvider(config.apiKey, provider));
   }
   if (context) {
     Object.assign(out, buildContextHeaders(context));
@@ -50,6 +71,8 @@ var TrustGate = class {
   }
   /**
    * Perform a non-streaming JSON request through TrustGate.
+   * @param options.provider - Optional provider for this request ('openai' | 'anthropic' | 'google');
+   *   controls auth header format (e.g. x-api-key only for anthropic/google).
    */
   async fetch(path, init = {}, options) {
     const context = resolveContext(this.config, options);
@@ -136,8 +159,12 @@ export {
   TrustGate,
   buildContextHeaders,
   buildSourceToolJson,
+  createTrustGateAnthropicConfig,
+  createTrustGateGeminiConfig,
   createTrustGateLangChainConfig,
   createTrustGateOpenAIOptions,
+  getAnthropicHeaders,
+  getGeminiHeaders,
   getN8nHeaderSnippet,
   getTrustGateHeaders,
   getWorkflowContextFromEnv

package/dist/middleware/anthropic.d.mts

@@ -0,0 +1,38 @@
+import { T as TrustGateConfig } from '../types-QFksyF5p.mjs';
+
+/**
+ * Anthropic middleware: use TrustGate as the base URL for @anthropic-ai/sdk.
+ *
+ * Pass the returned config to the Anthropic client so all requests go through
+ * TrustGate with workflow context and x-api-key for Shadow AI detection.
+ *
+ * Example:
+ *
+ *   import Anthropic from '@anthropic-ai/sdk';
+ *   import { createTrustGateAnthropicConfig } from '@trustgateai/sdk/middleware/anthropic';
+ *
+ *   const config = createTrustGateAnthropicConfig({
+ *     baseUrl: process.env.TRUSTGATE_BASE_URL!,
+ *     apiKey: process.env.TRUSTGATE_API_KEY,
+ *   });
+ *
+ *   const anthropic = new Anthropic(config);
+ *   const message = await anthropic.messages.create({ model: 'claude-3-5-sonnet-20241022', ... });
+ */
+
+/**
+ * Returns headers compatible with @anthropic-ai/sdk. Uses x-api-key only (no
+ * Authorization header). Use with defaultHeaders or when building requests manually.
+ */
+declare function getAnthropicHeaders(config: TrustGateConfig): Record<string, string>;
+/**
+ * Returns a config object for the Anthropic SDK (baseURL, apiKey, defaultHeaders).
+ * Injects TrustGate context headers and x-api-key so requests go through the gateway.
+ */
+declare function createTrustGateAnthropicConfig(config: TrustGateConfig): {
+    baseURL: string;
+    apiKey: string | null;
+    defaultHeaders?: Record<string, string>;
+};
+
+export { createTrustGateAnthropicConfig, getAnthropicHeaders };

package/dist/middleware/anthropic.d.ts

@@ -0,0 +1,38 @@
+import { T as TrustGateConfig } from '../types-QFksyF5p.js';
+
+/**
+ * Anthropic middleware: use TrustGate as the base URL for @anthropic-ai/sdk.
+ *
+ * Pass the returned config to the Anthropic client so all requests go through
+ * TrustGate with workflow context and x-api-key for Shadow AI detection.
+ *
+ * Example:
+ *
+ *   import Anthropic from '@anthropic-ai/sdk';
+ *   import { createTrustGateAnthropicConfig } from '@trustgateai/sdk/middleware/anthropic';
+ *
+ *   const config = createTrustGateAnthropicConfig({
+ *     baseUrl: process.env.TRUSTGATE_BASE_URL!,
+ *     apiKey: process.env.TRUSTGATE_API_KEY,
+ *   });
+ *
+ *   const anthropic = new Anthropic(config);
+ *   const message = await anthropic.messages.create({ model: 'claude-3-5-sonnet-20241022', ... });
+ */
+
+/**
+ * Returns headers compatible with @anthropic-ai/sdk. Uses x-api-key only (no
+ * Authorization header). Use with defaultHeaders or when building requests manually.
+ */
+declare function getAnthropicHeaders(config: TrustGateConfig): Record<string, string>;
+/**
+ * Returns a config object for the Anthropic SDK (baseURL, apiKey, defaultHeaders).
+ * Injects TrustGate context headers and x-api-key so requests go through the gateway.
+ */
+declare function createTrustGateAnthropicConfig(config: TrustGateConfig): {
+    baseURL: string;
+    apiKey: string | null;
+    defaultHeaders?: Record<string, string>;
+};
+
+export { createTrustGateAnthropicConfig, getAnthropicHeaders };

package/dist/middleware/anthropic.js

@@ -0,0 +1,158 @@
+"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/middleware/anthropic.ts
+var anthropic_exports = {};
+__export(anthropic_exports, {
+  createTrustGateAnthropicConfig: () => createTrustGateAnthropicConfig,
+  getAnthropicHeaders: () => getAnthropicHeaders
+});
+module.exports = __toCommonJS(anthropic_exports);
+
+// src/version.ts
+var SDK_VERSION = "1.1.0";
+
+// src/context.ts
+var HEADER_PREFIX = "x-trustgate";
+var SOURCE_TOOL_HEADER = "x-trustgate-source-tool";
+var N8N_KEYS = [
+  "N8N_WORKFLOW_ID",
+  "N8N_EXECUTION_ID",
+  "N8N_WORKFLOW_NAME",
+  "N8N_EXECUTION_MODE"
+];
+var VERCEL_KEYS = [
+  "VERCEL",
+  "VERCEL_ENV",
+  "VERCEL_URL",
+  "VERCEL_GIT_COMMIT_REF",
+  "VERCEL_GIT_REPO_ID",
+  "VERCEL_GIT_COMMIT_SHA"
+];
+var GITHUB_KEYS = [
+  "GITHUB_ACTION",
+  "GITHUB_ACTIONS",
+  "GITHUB_WORKFLOW",
+  "GITHUB_RUN_ID",
+  "GITHUB_RUN_NUMBER",
+  "GITHUB_REPOSITORY",
+  "GITHUB_SHA"
+];
+function pickEnv(keys) {
+  const out = {};
+  for (const key of keys) {
+    const v = process.env[key];
+    if (v !== void 0 && v !== "") out[key] = v;
+  }
+  return out;
+}
+function getBaseMetadata() {
+  return {
+    sdk_version: SDK_VERSION,
+    node_version: process.version,
+    os: process.platform,
+    arch: process.arch
+  };
+}
+function getWorkflowContextFromEnv() {
+  const baseMeta = getBaseMetadata();
+  const n8n = pickEnv(N8N_KEYS);
+  if (Object.keys(n8n).length > 0) {
+    return {
+      source: "n8n",
+      workflow_id: n8n.N8N_WORKFLOW_ID,
+      execution_id: n8n.N8N_EXECUTION_ID,
+      metadata: { ...baseMeta, ...n8n }
+    };
+  }
+  const vercel = pickEnv(VERCEL_KEYS);
+  if (Object.keys(vercel).length > 0) {
+    return {
+      source: "vercel",
+      workflow_id: vercel.VERCEL_GIT_REPO_ID || vercel.VERCEL_URL,
+      execution_id: vercel.VERCEL_GIT_COMMIT_SHA,
+      metadata: { ...baseMeta, ...vercel }
+    };
+  }
+  const github = pickEnv(GITHUB_KEYS);
+  if (Object.keys(github).length > 0) {
+    return {
+      source: "github",
+      workflow_id: github.GITHUB_WORKFLOW || github.GITHUB_REPOSITORY,
+      execution_id: github.GITHUB_RUN_ID || github.GITHUB_RUN_NUMBER,
+      metadata: { ...baseMeta, ...github }
+    };
+  }
+  return {
+    source: "local_script",
+    metadata: baseMeta
+  };
+}
+function buildSourceToolJson(ctx) {
+  const meta = ctx.metadata ?? getBaseMetadata();
+  const payload = {
+    source: ctx.source,
+    metadata: meta
+  };
+  if (ctx.workflow_id !== void 0) payload.workflow_id = ctx.workflow_id;
+  if (ctx.execution_id !== void 0) payload.execution_id = ctx.execution_id;
+  if (ctx.workflow_step !== void 0) payload.workflow_step = ctx.workflow_step;
+  return JSON.stringify(payload);
+}
+function buildContextHeaders(ctx) {
+  const meta = ctx.metadata ?? getBaseMetadata();
+  const h = {
+    [SOURCE_TOOL_HEADER]: buildSourceToolJson({ ...ctx, metadata: meta }),
+    [`${HEADER_PREFIX}-source`]: ctx.source
+  };
+  if (ctx.workflow_id) h[`${HEADER_PREFIX}-workflow-id`] = ctx.workflow_id;
+  if (ctx.execution_id) h[`${HEADER_PREFIX}-execution-id`] = ctx.execution_id;
+  if (ctx.workflow_step) h[`${HEADER_PREFIX}-workflow-step`] = ctx.workflow_step;
+  for (const [k, v] of Object.entries(meta)) {
+    if (v) h[`${HEADER_PREFIX}-meta-${k.toLowerCase()}`] = String(v);
+  }
+  return h;
+}
+
+// src/middleware/anthropic.ts
+function getAnthropicHeaders(config) {
+  const ctx = config.workflowContext ?? getWorkflowContextFromEnv();
+  const headers = {
+    ...config.headers,
+    ...buildContextHeaders(ctx)
+  };
+  if (config.apiKey) {
+    headers["x-api-key"] = config.apiKey;
+  }
+  return headers;
+}
+function createTrustGateAnthropicConfig(config) {
+  const baseURL = config.baseUrl.replace(/\/$/, "");
+  const headers = getAnthropicHeaders(config);
+  return {
+    baseURL,
+    apiKey: config.apiKey ?? null,
+    defaultHeaders: Object.keys(headers).length ? headers : void 0
+  };
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  createTrustGateAnthropicConfig,
+  getAnthropicHeaders
+});

package/dist/middleware/anthropic.mjs

@@ -0,0 +1,9 @@
+import {
+  createTrustGateAnthropicConfig,
+  getAnthropicHeaders
+} from "../chunk-QJUVF52Z.mjs";
+import "../chunk-Z2D2AATH.mjs";
+export {
+  createTrustGateAnthropicConfig,
+  getAnthropicHeaders
+};

package/dist/middleware/google.d.mts

@@ -0,0 +1,45 @@
+import { T as TrustGateConfig } from '../types-QFksyF5p.mjs';
+
+/**
+ * Google Gemini middleware: use TrustGate with @google/generative-ai.
+ *
+ * Pass the returned config when creating the client or when making requests.
+ * The legacy @google/generative-ai SDK constructor accepts (apiKey); use the
+ * returned apiKey to point at TrustGate (gateway key), and pass headers via
+ * a custom fetch if your setup supports it.
+ *
+ * Example (gateway as proxy; SDK may require custom fetch for full header control):
+ *
+ *   import { GoogleGenerativeAI } from '@google/generative-ai';
+ *   import { createTrustGateGeminiConfig } from '@trustgateai/sdk/middleware/google';
+ *
+ *   const config = createTrustGateGeminiConfig({
+ *     baseUrl: process.env.TRUSTGATE_BASE_URL!,
+ *     apiKey: process.env.TRUSTGATE_API_KEY,
+ *   });
+ *
+ *   const genAI = new GoogleGenerativeAI(config.apiKey ?? '');
+ *   const model = genAI.getGenerativeModel({ model: 'gemini-1.5-flash' });
+ *
+ * For full TrustGate header injection (x-trustgate-source-tool, etc.), use
+ * the TrustGate client's fetch with provider: 'google' or pass config.headers
+ * via a custom fetch implementation if the SDK supports it.
+ */
+
+/**
+ * Returns headers compatible with @google/generative-ai. Uses x-api-key and
+ * TrustGate context. Use when building requests or with a custom fetch.
+ */
+declare function getGeminiHeaders(config: TrustGateConfig): Record<string, string>;
+/**
+ * Returns a config object for use with @google/generative-ai (or compatible clients).
+ * Injects TrustGate context headers and x-api-key. Use apiKey for the gateway and
+ * baseUrl/headers when using a custom fetch or wrapper.
+ */
+declare function createTrustGateGeminiConfig(config: TrustGateConfig): {
+    baseUrl: string;
+    apiKey: string | null;
+    headers?: Record<string, string>;
+};
+
+export { createTrustGateGeminiConfig, getGeminiHeaders };

package/dist/middleware/google.d.ts

@@ -0,0 +1,45 @@
+import { T as TrustGateConfig } from '../types-QFksyF5p.js';
+
+/**
+ * Google Gemini middleware: use TrustGate with @google/generative-ai.
+ *
+ * Pass the returned config when creating the client or when making requests.
+ * The legacy @google/generative-ai SDK constructor accepts (apiKey); use the
+ * returned apiKey to point at TrustGate (gateway key), and pass headers via
+ * a custom fetch if your setup supports it.
+ *
+ * Example (gateway as proxy; SDK may require custom fetch for full header control):
+ *
+ *   import { GoogleGenerativeAI } from '@google/generative-ai';
+ *   import { createTrustGateGeminiConfig } from '@trustgateai/sdk/middleware/google';
+ *
+ *   const config = createTrustGateGeminiConfig({
+ *     baseUrl: process.env.TRUSTGATE_BASE_URL!,
+ *     apiKey: process.env.TRUSTGATE_API_KEY,
+ *   });
+ *
+ *   const genAI = new GoogleGenerativeAI(config.apiKey ?? '');
+ *   const model = genAI.getGenerativeModel({ model: 'gemini-1.5-flash' });
+ *
+ * For full TrustGate header injection (x-trustgate-source-tool, etc.), use
+ * the TrustGate client's fetch with provider: 'google' or pass config.headers
+ * via a custom fetch implementation if the SDK supports it.
+ */
+
+/**
+ * Returns headers compatible with @google/generative-ai. Uses x-api-key and
+ * TrustGate context. Use when building requests or with a custom fetch.
+ */
+declare function getGeminiHeaders(config: TrustGateConfig): Record<string, string>;
+/**
+ * Returns a config object for use with @google/generative-ai (or compatible clients).
+ * Injects TrustGate context headers and x-api-key. Use apiKey for the gateway and
+ * baseUrl/headers when using a custom fetch or wrapper.
+ */
+declare function createTrustGateGeminiConfig(config: TrustGateConfig): {
+    baseUrl: string;
+    apiKey: string | null;
+    headers?: Record<string, string>;
+};
+
+export { createTrustGateGeminiConfig, getGeminiHeaders };