@axonflow/sdk 1.2.0 → 1.3.0

package/dist/cjs/interceptors/bedrock.js

@@ -0,0 +1,259 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.BedrockModels = exports.BedrockInterceptor = void 0;
+exports.wrapBedrockClient = wrapBedrockClient;
+exports.createGovernedBedrockInvoke = createGovernedBedrockInvoke;
+const base_1 = require("./base");
+/**
+ * Interceptor for AWS Bedrock API calls
+ *
+ * Bedrock uses AWS IAM authentication (no API keys required).
+ * Supports multiple model providers: Anthropic Claude, Amazon Titan, Meta Llama, etc.
+ *
+ * @example
+ * ```typescript
+ * import { BedrockRuntimeClient, InvokeModelCommand } from '@aws-sdk/client-bedrock-runtime';
+ * import { AxonFlow, wrapBedrockClient } from '@axonflow/sdk';
+ *
+ * const bedrockClient = new BedrockRuntimeClient({ region: 'us-east-1' });
+ * const axonflow = new AxonFlow({ endpoint: 'http://localhost:8080' });
+ *
+ * const wrapped = wrapBedrockClient(bedrockClient, axonflow);
+ * const response = await wrapped.send(new InvokeModelCommand({...}));
+ * ```
+ */
+class BedrockInterceptor extends base_1.BaseInterceptor {
+    canHandle(aiCall) {
+        const callString = aiCall.toString();
+        return (callString.includes('bedrock') ||
+            callString.includes('BedrockRuntime') ||
+            callString.includes('InvokeModel') ||
+            callString.includes('anthropic.claude') ||
+            callString.includes('amazon.titan'));
+    }
+    extractRequest(aiCall) {
+        const callString = aiCall.toString();
+        let model = 'anthropic.claude-3-sonnet';
+        if (callString.includes('titan')) {
+            model = 'amazon.titan-text-express-v1';
+        }
+        else if (callString.includes('llama')) {
+            model = 'meta.llama2-70b-chat-v1';
+        }
+        else if (callString.includes('claude-3-opus')) {
+            model = 'anthropic.claude-3-opus';
+        }
+        else if (callString.includes('claude-3-haiku')) {
+            model = 'anthropic.claude-3-haiku';
+        }
+        return {
+            provider: 'bedrock',
+            model,
+            prompt: callString,
+            parameters: {},
+        };
+    }
+    executeWithModifications(aiCall, _modifications) {
+        return aiCall();
+    }
+    getProvider() {
+        return 'bedrock';
+    }
+}
+exports.BedrockInterceptor = BedrockInterceptor;
+/**
+ * Bedrock model providers and their model IDs
+ */
+exports.BedrockModels = {
+    // Anthropic Claude models
+    CLAUDE_3_OPUS: 'anthropic.claude-3-opus-20240229-v1:0',
+    CLAUDE_3_SONNET: 'anthropic.claude-3-sonnet-20240229-v1:0',
+    CLAUDE_3_HAIKU: 'anthropic.claude-3-haiku-20240307-v1:0',
+    CLAUDE_2_1: 'anthropic.claude-v2:1',
+    CLAUDE_2: 'anthropic.claude-v2',
+    CLAUDE_INSTANT: 'anthropic.claude-instant-v1',
+    // Amazon Titan models
+    TITAN_TEXT_EXPRESS: 'amazon.titan-text-express-v1',
+    TITAN_TEXT_LITE: 'amazon.titan-text-lite-v1',
+    TITAN_TEXT_PREMIER: 'amazon.titan-text-premier-v1:0',
+    // Meta Llama models
+    LLAMA2_13B: 'meta.llama2-13b-chat-v1',
+    LLAMA2_70B: 'meta.llama2-70b-chat-v1',
+    LLAMA3_8B: 'meta.llama3-8b-instruct-v1:0',
+    LLAMA3_70B: 'meta.llama3-70b-instruct-v1:0',
+    // Cohere models
+    COHERE_COMMAND: 'cohere.command-text-v14',
+    COHERE_COMMAND_LIGHT: 'cohere.command-light-text-v14',
+    // AI21 models
+    AI21_JURASSIC_ULTRA: 'ai21.j2-ultra-v1',
+    AI21_JURASSIC_MID: 'ai21.j2-mid-v1',
+};
+/**
+ * Wraps an AWS Bedrock client with AxonFlow governance.
+ *
+ * @example
+ * ```typescript
+ * import { BedrockRuntimeClient, InvokeModelCommand } from '@aws-sdk/client-bedrock-runtime';
+ * import { AxonFlow, wrapBedrockClient } from '@axonflow/sdk';
+ *
+ * const bedrock = new BedrockRuntimeClient({ region: 'us-east-1' });
+ * const axonflow = new AxonFlow({ endpoint: 'http://localhost:8080' });
+ *
+ * const wrapped = wrapBedrockClient(bedrock, axonflow);
+ *
+ * const command = new InvokeModelCommand({
+ *   modelId: 'anthropic.claude-3-sonnet-20240229-v1:0',
+ *   body: JSON.stringify({
+ *     anthropic_version: 'bedrock-2023-05-31',
+ *     max_tokens: 1024,
+ *     messages: [{ role: 'user', content: 'Hello!' }]
+ *   })
+ * });
+ *
+ * const response = await wrapped.send(command);
+ * ```
+ */
+function wrapBedrockClient(bedrockClient, axonflow) {
+    const originalSend = bedrockClient.send.bind(bedrockClient);
+    bedrockClient.send = async (command) => {
+        // Check if this is an InvokeModelCommand
+        const commandName = command.constructor?.name || '';
+        if (!commandName.includes('InvokeModel')) {
+            return originalSend(command);
+        }
+        // Extract model ID and prompt for governance context
+        const modelId = command.input?.modelId || 'unknown';
+        let prompt = '';
+        try {
+            const body = JSON.parse(typeof command.input?.body === 'string'
+                ? command.input.body
+                : new TextDecoder().decode(command.input?.body));
+            // Extract prompt based on model type
+            if (body.messages) {
+                // Claude format
+                prompt = body.messages
+                    .map((m) => (typeof m.content === 'string' ? m.content : m.content?.[0]?.text || ''))
+                    .join(' ');
+            }
+            else if (body.inputText) {
+                // Titan format
+                prompt = body.inputText;
+            }
+            else if (body.prompt) {
+                // Generic format
+                prompt = body.prompt;
+            }
+        }
+        catch {
+            // If we can't parse the body, continue without prompt extraction
+        }
+        // Protect the call with AxonFlow, passing context
+        return axonflow.protect(async () => {
+            return originalSend(command);
+        }, { provider: 'bedrock', model: modelId, query: prompt });
+    };
+    return bedrockClient;
+}
+/**
+ * Creates a governed Bedrock invoke function with full governance flow.
+ *
+ * Use this for fine-grained control with pre-check and audit.
+ */
+function createGovernedBedrockInvoke(bedrockClient, axonflow, userToken = '') {
+    return async (modelId, body) => {
+        // Extract prompt
+        let prompt = '';
+        if (body.messages) {
+            prompt = body.messages
+                .map((m) => (typeof m.content === 'string' ? m.content : m.content?.[0]?.text || ''))
+                .join(' ');
+        }
+        else if (body.inputText) {
+            prompt = body.inputText;
+        }
+        // Pre-check with AxonFlow
+        const preCheck = await axonflow.getPolicyApprovedContext({
+            userToken,
+            query: prompt,
+            context: {
+                provider: 'bedrock',
+                model: modelId,
+            },
+        });
+        if (!preCheck.approved) {
+            throw new Error(`Request blocked by policy: ${preCheck.blockReason}`);
+        }
+        // Execute the call
+        const startTime = Date.now();
+        // Dynamically import to avoid requiring AWS SDK as a dependency
+        const { InvokeModelCommand } = await Promise.resolve().then(() => __importStar(require('@aws-sdk/client-bedrock-runtime')));
+        const command = new InvokeModelCommand({
+            modelId,
+            body: JSON.stringify(body),
+            contentType: 'application/json',
+            accept: 'application/json',
+        });
+        const response = await bedrockClient.send(command);
+        const latencyMs = Date.now() - startTime;
+        // Parse response
+        const responseBody = JSON.parse(new TextDecoder().decode(response.body));
+        // Audit the call
+        if (preCheck.contextId) {
+            let summary = '';
+            let promptTokens = 0;
+            let completionTokens = 0;
+            if (responseBody.content) {
+                // Claude response
+                summary = responseBody.content[0]?.text?.substring(0, 200) || '';
+                promptTokens = responseBody.usage?.input_tokens || 0;
+                completionTokens = responseBody.usage?.output_tokens || 0;
+            }
+            else if (responseBody.results) {
+                // Titan response
+                summary = responseBody.results[0]?.outputText?.substring(0, 200) || '';
+                promptTokens = responseBody.inputTextTokenCount || 0;
+                completionTokens = responseBody.results[0]?.tokenCount || 0;
+            }
+            await axonflow.auditLLMCall(preCheck.contextId, summary, 'bedrock', modelId, {
+                promptTokens,
+                completionTokens,
+                totalTokens: promptTokens + completionTokens,
+            }, latencyMs);
+        }
+        return responseBody;
+    };
+}
+//# sourceMappingURL=bedrock.js.map

package/dist/cjs/interceptors/bedrock.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"bedrock.js","sourceRoot":"","sources":["../../../src/interceptors/bedrock.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAmLA,8CAgDC;AAOD,kEAkFC;AA5TD,iCAAyC;AAGzC;;;;;;;;;;;;;;;;;GAiBG;AACH,MAAa,kBAAmB,SAAQ,sBAAe;IACrD,SAAS,CAAC,MAAW;QACnB,MAAM,UAAU,GAAG,MAAM,CAAC,QAAQ,EAAE,CAAC;QACrC,OAAO,CACL,UAAU,CAAC,QAAQ,CAAC,SAAS,CAAC;YAC9B,UAAU,CAAC,QAAQ,CAAC,gBAAgB,CAAC;YACrC,UAAU,CAAC,QAAQ,CAAC,aAAa,CAAC;YAClC,UAAU,CAAC,QAAQ,CAAC,kBAAkB,CAAC;YACvC,UAAU,CAAC,QAAQ,CAAC,cAAc,CAAC,CACpC,CAAC;IACJ,CAAC;IAED,cAAc,CAAC,MAAW;QACxB,MAAM,UAAU,GAAG,MAAM,CAAC,QAAQ,EAAE,CAAC;QAErC,IAAI,KAAK,GAAG,2BAA2B,CAAC;QACxC,IAAI,UAAU,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;YACjC,KAAK,GAAG,8BAA8B,CAAC;QACzC,CAAC;aAAM,IAAI,UAAU,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;YACxC,KAAK,GAAG,yBAAyB,CAAC;QACpC,CAAC;aAAM,IAAI,UAAU,CAAC,QAAQ,CAAC,eAAe,CAAC,EAAE,CAAC;YAChD,KAAK,GAAG,yBAAyB,CAAC;QACpC,CAAC;aAAM,IAAI,UAAU,CAAC,QAAQ,CAAC,gBAAgB,CAAC,EAAE,CAAC;YACjD,KAAK,GAAG,0BAA0B,CAAC;QACrC,CAAC;QAED,OAAO;YACL,QAAQ,EAAE,SAAS;YACnB,KAAK;YACL,MAAM,EAAE,UAAU;YAClB,UAAU,EAAE,EAAE;SACf,CAAC;IACJ,CAAC;IAED,wBAAwB,CAAC,MAAW,EAAE,cAAmB;QACvD,OAAO,MAAM,EAAE,CAAC;IAClB,CAAC;IAED,WAAW;QACT,OAAO,SAAS,CAAC;IACnB,CAAC;CACF;AAzCD,gDAyCC;AAED;;GAEG;AACU,QAAA,aAAa,GAAG;IAC3B,0BAA0B;IAC1B,aAAa,EAAE,uCAAuC;IACtD,eAAe,EAAE,yCAAyC;IAC1D,cAAc,EAAE,wCAAwC;IACxD,UAAU,EAAE,uBAAuB;IACnC,QAAQ,EAAE,qBAAqB;IAC/B,cAAc,EAAE,6BAA6B;IAE7C,sBAAsB;IACtB,kBAAkB,EAAE,8BAA8B;IAClD,eAAe,EAAE,2BAA2B;IAC5C,kBAAkB,EAAE,gCAAgC;IAEpD,oBAAoB;IACpB,UAAU,EAAE,yBAAyB;IACrC,UAAU,EAAE,yBAAyB;IACrC,SAAS,EAAE,8BAA8B;IACzC,UAAU,EAAE,+BAA+B;IAE3C,gBAAgB;IAChB,cAAc,EAAE,yBAAyB;IACzC,oBAAoB,EAAE,+BAA+B;IAErD,cAAc;IACd,mBAAmB,EAAE,kBAAkB;IACvC,iBAAiB,EAAE,gBAAgB;CAC3B,CAAC;AA4DX;;;;;;;;;;;;;;;;;;;;;;;;GAwBG;AACH,SAAgB,iBAAiB,CAAC,aAAkB,EAAE,QAAa;IACjE,MAAM,YAAY,GAAG,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;IAE5D,aAAa,CAAC,IAAI,GAAG,KAAK,EAAE,OAAY,EAAE,EAAE;QAC1C,yCAAyC;QACzC,MAAM,WAAW,GAAG,OAAO,CAAC,WAAW,EAAE,IAAI,IAAI,EAAE,CAAC;QACpD,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,aAAa,CAAC,EAAE,CAAC;YACzC,OAAO,YAAY,CAAC,OAAO,CAAC,CAAC;QAC/B,CAAC;QAED,qDAAqD;QACrD,MAAM,OAAO,GAAG,OAAO,CAAC,KAAK,EAAE,OAAO,IAAI,SAAS,CAAC;QACpD,IAAI,MAAM,GAAG,EAAE,CAAC;QAEhB,IAAI,CAAC;YACH,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CACrB,OAAO,OAAO,CAAC,KAAK,EAAE,IAAI,KAAK,QAAQ;gBACrC,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI;gBACpB,CAAC,CAAC,IAAI,WAAW,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC,KAAK,EAAE,IAAI,CAAC,CAClD,CAAC;YAEF,qCAAqC;YACrC,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;gBAClB,gBAAgB;gBAChB,MAAM,GAAG,IAAI,CAAC,QAAQ;qBACnB,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,IAAI,EAAE,CAAC,CAAC;qBACzF,IAAI,CAAC,GAAG,CAAC,CAAC;YACf,CAAC;iBAAM,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;gBAC1B,eAAe;gBACf,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC;YAC1B,CAAC;iBAAM,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;gBACvB,iBAAiB;gBACjB,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC;YACvB,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,iEAAiE;QACnE,CAAC;QAED,kDAAkD;QAClD,OAAO,QAAQ,CAAC,OAAO,CACrB,KAAK,IAAI,EAAE;YACT,OAAO,YAAY,CAAC,OAAO,CAAC,CAAC;QAC/B,CAAC,EACD,EAAE,QAAQ,EAAE,SAAS,EAAE,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,CACvD,CAAC;IACJ,CAAC,CAAC;IAEF,OAAO,aAAa,CAAC;AACvB,CAAC;AAED;;;;GAIG;AACH,SAAgB,2BAA2B,CACzC,aAAkB,EAClB,QAAa,EACb,YAAoB,EAAE;IAEtB,OAAO,KAAK,EAAE,OAAe,EAAE,IAAS,EAAgB,EAAE;QACxD,iBAAiB;QACjB,IAAI,MAAM,GAAG,EAAE,CAAC;QAChB,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAClB,MAAM,GAAG,IAAI,CAAC,QAAQ;iBACnB,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,IAAI,EAAE,CAAC,CAAC;iBACzF,IAAI,CAAC,GAAG,CAAC,CAAC;QACf,CAAC;aAAM,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YAC1B,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC;QAC1B,CAAC;QAED,0BAA0B;QAC1B,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,wBAAwB,CAAC;YACvD,SAAS;YACT,KAAK,EAAE,MAAM;YACb,OAAO,EAAE;gBACP,QAAQ,EAAE,SAAS;gBACnB,KAAK,EAAE,OAAO;aACf;SACF,CAAC,CAAC;QAEH,IAAI,CAAC,QAAQ,CAAC,QAAQ,EAAE,CAAC;YACvB,MAAM,IAAI,KAAK,CAAC,8BAA8B,QAAQ,CAAC,WAAW,EAAE,CAAC,CAAC;QACxE,CAAC;QAED,mBAAmB;QACnB,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAE7B,gEAAgE;QAChE,MAAM,EAAE,kBAAkB,EAAE,GAAG,wDAAa,iCAAiC,GAAC,CAAC;QAC/E,MAAM,OAAO,GAAG,IAAI,kBAAkB,CAAC;YACrC,OAAO;YACP,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;YAC1B,WAAW,EAAE,kBAAkB;YAC/B,MAAM,EAAE,kBAAkB;SAC3B,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG,MAAM,aAAa,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACnD,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;QAEzC,iBAAiB;QACjB,MAAM,YAAY,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,WAAW,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC;QAEzE,iBAAiB;QACjB,IAAI,QAAQ,CAAC,SAAS,EAAE,CAAC;YACvB,IAAI,OAAO,GAAG,EAAE,CAAC;YACjB,IAAI,YAAY,GAAG,CAAC,CAAC;YACrB,IAAI,gBAAgB,GAAG,CAAC,CAAC;YAEzB,IAAI,YAAY,CAAC,OAAO,EAAE,CAAC;gBACzB,kBAAkB;gBAClB,OAAO,GAAG,YAAY,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,SAAS,CAAC,CAAC,EAAE,GAAG,CAAC,IAAI,EAAE,CAAC;gBACjE,YAAY,GAAG,YAAY,CAAC,KAAK,EAAE,YAAY,IAAI,CAAC,CAAC;gBACrD,gBAAgB,GAAG,YAAY,CAAC,KAAK,EAAE,aAAa,IAAI,CAAC,CAAC;YAC5D,CAAC;iBAAM,IAAI,YAAY,CAAC,OAAO,EAAE,CAAC;gBAChC,iBAAiB;gBACjB,OAAO,GAAG,YAAY,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,SAAS,CAAC,CAAC,EAAE,GAAG,CAAC,IAAI,EAAE,CAAC;gBACvE,YAAY,GAAG,YAAY,CAAC,mBAAmB,IAAI,CAAC,CAAC;gBACrD,gBAAgB,GAAG,YAAY,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,UAAU,IAAI,CAAC,CAAC;YAC9D,CAAC;YAED,MAAM,QAAQ,CAAC,YAAY,CACzB,QAAQ,CAAC,SAAS,EAClB,OAAO,EACP,SAAS,EACT,OAAO,EACP;gBACE,YAAY;gBACZ,gBAAgB;gBAChB,WAAW,EAAE,YAAY,GAAG,gBAAgB;aAC7C,EACD,SAAS,CACV,CAAC;QACJ,CAAC;QAED,OAAO,YAAY,CAAC;IACtB,CAAC,CAAC;AACJ,CAAC"}

package/dist/cjs/interceptors/gemini.d.ts

@@ -0,0 +1,81 @@
+import { BaseInterceptor } from './base';
+import { AIRequest } from '../types';
+/**
+ * Interceptor for Google Gemini API calls
+ *
+ * Supports both @google/generative-ai SDK and direct API calls.
+ */
+export declare class GeminiInterceptor extends BaseInterceptor {
+    canHandle(aiCall: any): boolean;
+    extractRequest(aiCall: any): AIRequest;
+    executeWithModifications(aiCall: any, _modifications: any): Promise<any>;
+    getProvider(): string;
+}
+/**
+ * Gemini content part types
+ */
+export interface GeminiTextPart {
+    text: string;
+}
+export interface GeminiInlineDataPart {
+    inlineData: {
+        mimeType: string;
+        data: string;
+    };
+}
+export type GeminiPart = GeminiTextPart | GeminiInlineDataPart;
+export interface GeminiContent {
+    role?: 'user' | 'model';
+    parts: GeminiPart[];
+}
+export interface GeminiGenerateContentRequest {
+    contents: GeminiContent[];
+    generationConfig?: {
+        temperature?: number;
+        topP?: number;
+        topK?: number;
+        maxOutputTokens?: number;
+        stopSequences?: string[];
+    };
+    safetySettings?: Array<{
+        category: string;
+        threshold: string;
+    }>;
+}
+export interface GeminiCandidate {
+    content: GeminiContent;
+    finishReason: string;
+    safetyRatings?: Array<{
+        category: string;
+        probability: string;
+    }>;
+}
+export interface GeminiUsageMetadata {
+    promptTokenCount: number;
+    candidatesTokenCount: number;
+    totalTokenCount: number;
+}
+export interface GeminiGenerateContentResponse {
+    candidates: GeminiCandidate[];
+    usageMetadata?: GeminiUsageMetadata;
+}
+/**
+ * Helper to wrap Gemini GenerativeModel for easier interception
+ *
+ * @example
+ * ```typescript
+ * import { GoogleGenerativeAI } from '@google/generative-ai';
+ * import { AxonFlow, wrapGeminiModel } from '@axonflow/sdk';
+ *
+ * const genAI = new GoogleGenerativeAI(process.env.GEMINI_API_KEY);
+ * const model = genAI.getGenerativeModel({ model: 'gemini-pro' });
+ * const axonflow = new AxonFlow({ apiKey: process.env.AXONFLOW_API_KEY });
+ *
+ * const wrappedModel = wrapGeminiModel(model, axonflow);
+ *
+ * // Use as normal - governance is automatic
+ * const result = await wrappedModel.generateContent('What is AI governance?');
+ * ```
+ */
+export declare function wrapGeminiModel(geminiModel: any, axonflow: any): any;
+//# sourceMappingURL=gemini.d.ts.map

package/dist/cjs/interceptors/gemini.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"gemini.d.ts","sourceRoot":"","sources":["../../../src/interceptors/gemini.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,MAAM,QAAQ,CAAC;AACzC,OAAO,EAAE,SAAS,EAAE,MAAM,UAAU,CAAC;AAErC;;;;GAIG;AACH,qBAAa,iBAAkB,SAAQ,eAAe;IACpD,SAAS,CAAC,MAAM,EAAE,GAAG,GAAG,OAAO;IAY/B,cAAc,CAAC,MAAM,EAAE,GAAG,GAAG,SAAS;IAwBtC,wBAAwB,CAAC,MAAM,EAAE,GAAG,EAAE,cAAc,EAAE,GAAG,GAAG,OAAO,CAAC,GAAG,CAAC;IAKxE,WAAW,IAAI,MAAM;CAGtB;AAED;;GAEG;AACH,MAAM,WAAW,cAAc;IAC7B,IAAI,EAAE,MAAM,CAAC;CACd;AAED,MAAM,WAAW,oBAAoB;IACnC,UAAU,EAAE;QACV,QAAQ,EAAE,MAAM,CAAC;QACjB,IAAI,EAAE,MAAM,CAAC;KACd,CAAC;CACH;AAED,MAAM,MAAM,UAAU,GAAG,cAAc,GAAG,oBAAoB,CAAC;AAE/D,MAAM,WAAW,aAAa;IAC5B,IAAI,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC;IACxB,KAAK,EAAE,UAAU,EAAE,CAAC;CACrB;AAED,MAAM,WAAW,4BAA4B;IAC3C,QAAQ,EAAE,aAAa,EAAE,CAAC;IAC1B,gBAAgB,CAAC,EAAE;QACjB,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB,IAAI,CAAC,EAAE,MAAM,CAAC;QACd,IAAI,CAAC,EAAE,MAAM,CAAC;QACd,eAAe,CAAC,EAAE,MAAM,CAAC;QACzB,aAAa,CAAC,EAAE,MAAM,EAAE,CAAC;KAC1B,CAAC;IACF,cAAc,CAAC,EAAE,KAAK,CAAC;QACrB,QAAQ,EAAE,MAAM,CAAC;QACjB,SAAS,EAAE,MAAM,CAAC;KACnB,CAAC,CAAC;CACJ;AAED,MAAM,WAAW,eAAe;IAC9B,OAAO,EAAE,aAAa,CAAC;IACvB,YAAY,EAAE,MAAM,CAAC;IACrB,aAAa,CAAC,EAAE,KAAK,CAAC;QACpB,QAAQ,EAAE,MAAM,CAAC;QACjB,WAAW,EAAE,MAAM,CAAC;KACrB,CAAC,CAAC;CACJ;AAED,MAAM,WAAW,mBAAmB;IAClC,gBAAgB,EAAE,MAAM,CAAC;IACzB,oBAAoB,EAAE,MAAM,CAAC;IAC7B,eAAe,EAAE,MAAM,CAAC;CACzB;AAED,MAAM,WAAW,6BAA6B;IAC5C,UAAU,EAAE,eAAe,EAAE,CAAC;IAC9B,aAAa,CAAC,EAAE,mBAAmB,CAAC;CACrC;AAED;;;;;;;;;;;;;;;;;GAiBG;AACH,wBAAgB,eAAe,CAAC,WAAW,EAAE,GAAG,EAAE,QAAQ,EAAE,GAAG,GAAG,GAAG,CA2BpE"}

package/dist/cjs/interceptors/gemini.js

@@ -0,0 +1,110 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.GeminiInterceptor = void 0;
+exports.wrapGeminiModel = wrapGeminiModel;
+const base_1 = require("./base");
+/**
+ * Interceptor for Google Gemini API calls
+ *
+ * Supports both @google/generative-ai SDK and direct API calls.
+ */
+class GeminiInterceptor extends base_1.BaseInterceptor {
+    canHandle(aiCall) {
+        // Check if this looks like a Gemini call
+        const callString = aiCall.toString();
+        return (callString.includes('gemini') ||
+            callString.includes('generateContent') ||
+            callString.includes('GenerativeModel') ||
+            callString.includes('google') ||
+            callString.includes('palm'));
+    }
+    extractRequest(aiCall) {
+        // Try to extract Gemini-specific details
+        const callString = aiCall.toString();
+        // Try to detect model
+        let model = 'gemini-pro';
+        if (callString.includes('gemini-1.5-pro')) {
+            model = 'gemini-1.5-pro';
+        }
+        else if (callString.includes('gemini-1.5-flash')) {
+            model = 'gemini-1.5-flash';
+        }
+        else if (callString.includes('gemini-pro-vision')) {
+            model = 'gemini-pro-vision';
+        }
+        return {
+            provider: 'gemini',
+            model,
+            prompt: callString,
+            parameters: {
+            // Would extract temperature, topP, etc. in production
+            },
+        };
+    }
+    executeWithModifications(aiCall, _modifications) {
+        // Execute the call with any modifications from governance
+        return aiCall();
+    }
+    getProvider() {
+        return 'gemini';
+    }
+}
+exports.GeminiInterceptor = GeminiInterceptor;
+/**
+ * Helper to wrap Gemini GenerativeModel for easier interception
+ *
+ * @example
+ * ```typescript
+ * import { GoogleGenerativeAI } from '@google/generative-ai';
+ * import { AxonFlow, wrapGeminiModel } from '@axonflow/sdk';
+ *
+ * const genAI = new GoogleGenerativeAI(process.env.GEMINI_API_KEY);
+ * const model = genAI.getGenerativeModel({ model: 'gemini-pro' });
+ * const axonflow = new AxonFlow({ apiKey: process.env.AXONFLOW_API_KEY });
+ *
+ * const wrappedModel = wrapGeminiModel(model, axonflow);
+ *
+ * // Use as normal - governance is automatic
+ * const result = await wrappedModel.generateContent('What is AI governance?');
+ * ```
+ */
+function wrapGeminiModel(geminiModel, axonflow) {
+    return new Proxy(geminiModel, {
+        get(target, prop, receiver) {
+            const original = Reflect.get(target, prop, receiver);
+            // Intercept generateContent and generateContentStream
+            if (typeof original === 'function' &&
+                ['generateContent', 'generateContentStream', 'sendMessage'].includes(prop.toString())) {
+                return async (...args) => {
+                    // Protect the call with AxonFlow
+                    return axonflow.protect(() => original.apply(target, args));
+                };
+            }
+            // For chat sessions, wrap the returned chat object
+            if (prop === 'startChat' && typeof original === 'function') {
+                return (...args) => {
+                    const chat = original.apply(target, args);
+                    return wrapGeminiChat(chat, axonflow);
+                };
+            }
+            return original;
+        },
+    });
+}
+/**
+ * Helper to wrap Gemini ChatSession for multi-turn conversations
+ */
+function wrapGeminiChat(chatSession, axonflow) {
+    return new Proxy(chatSession, {
+        get(target, prop, receiver) {
+            const original = Reflect.get(target, prop, receiver);
+            if (typeof original === 'function' && prop === 'sendMessage') {
+                return async (...args) => {
+                    return axonflow.protect(() => original.apply(target, args));
+                };
+            }
+            return original;
+        },
+    });
+}
+//# sourceMappingURL=gemini.js.map

package/dist/cjs/interceptors/gemini.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"gemini.js","sourceRoot":"","sources":["../../../src/interceptors/gemini.ts"],"names":[],"mappings":";;;AAiIA,0CA2BC;AA5JD,iCAAyC;AAGzC;;;;GAIG;AACH,MAAa,iBAAkB,SAAQ,sBAAe;IACpD,SAAS,CAAC,MAAW;QACnB,yCAAyC;QACzC,MAAM,UAAU,GAAG,MAAM,CAAC,QAAQ,EAAE,CAAC;QACrC,OAAO,CACL,UAAU,CAAC,QAAQ,CAAC,QAAQ,CAAC;YAC7B,UAAU,CAAC,QAAQ,CAAC,iBAAiB,CAAC;YACtC,UAAU,CAAC,QAAQ,CAAC,iBAAiB,CAAC;YACtC,UAAU,CAAC,QAAQ,CAAC,QAAQ,CAAC;YAC7B,UAAU,CAAC,QAAQ,CAAC,MAAM,CAAC,CAC5B,CAAC;IACJ,CAAC;IAED,cAAc,CAAC,MAAW;QACxB,yCAAyC;QACzC,MAAM,UAAU,GAAG,MAAM,CAAC,QAAQ,EAAE,CAAC;QAErC,sBAAsB;QACtB,IAAI,KAAK,GAAG,YAAY,CAAC;QACzB,IAAI,UAAU,CAAC,QAAQ,CAAC,gBAAgB,CAAC,EAAE,CAAC;YAC1C,KAAK,GAAG,gBAAgB,CAAC;QAC3B,CAAC;aAAM,IAAI,UAAU,CAAC,QAAQ,CAAC,kBAAkB,CAAC,EAAE,CAAC;YACnD,KAAK,GAAG,kBAAkB,CAAC;QAC7B,CAAC;aAAM,IAAI,UAAU,CAAC,QAAQ,CAAC,mBAAmB,CAAC,EAAE,CAAC;YACpD,KAAK,GAAG,mBAAmB,CAAC;QAC9B,CAAC;QAED,OAAO;YACL,QAAQ,EAAE,QAAQ;YAClB,KAAK;YACL,MAAM,EAAE,UAAU;YAClB,UAAU,EAAE;YACV,sDAAsD;aACvD;SACF,CAAC;IACJ,CAAC;IAED,wBAAwB,CAAC,MAAW,EAAE,cAAmB;QACvD,0DAA0D;QAC1D,OAAO,MAAM,EAAE,CAAC;IAClB,CAAC;IAED,WAAW;QACT,OAAO,QAAQ,CAAC;IAClB,CAAC;CACF;AA7CD,8CA6CC;AA0DD;;;;;;;;;;;;;;;;;GAiBG;AACH,SAAgB,eAAe,CAAC,WAAgB,EAAE,QAAa;IAC7D,OAAO,IAAI,KAAK,CAAC,WAAW,EAAE;QAC5B,GAAG,CAAC,MAAM,EAAE,IAAI,EAAE,QAAQ;YACxB,MAAM,QAAQ,GAAG,OAAO,CAAC,GAAG,CAAC,MAAM,EAAE,IAAI,EAAE,QAAQ,CAAC,CAAC;YAErD,sDAAsD;YACtD,IACE,OAAO,QAAQ,KAAK,UAAU;gBAC9B,CAAC,iBAAiB,EAAE,uBAAuB,EAAE,aAAa,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC,EACrF,CAAC;gBACD,OAAO,KAAK,EAAE,GAAG,IAAW,EAAE,EAAE;oBAC9B,iCAAiC;oBACjC,OAAO,QAAQ,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC,KAAK,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC,CAAC;gBAC9D,CAAC,CAAC;YACJ,CAAC;YAED,mDAAmD;YACnD,IAAI,IAAI,KAAK,WAAW,IAAI,OAAO,QAAQ,KAAK,UAAU,EAAE,CAAC;gBAC3D,OAAO,CAAC,GAAG,IAAW,EAAE,EAAE;oBACxB,MAAM,IAAI,GAAG,QAAQ,CAAC,KAAK,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;oBAC1C,OAAO,cAAc,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;gBACxC,CAAC,CAAC;YACJ,CAAC;YAED,OAAO,QAAQ,CAAC;QAClB,CAAC;KACF,CAAC,CAAC;AACL,CAAC;AAED;;GAEG;AACH,SAAS,cAAc,CAAC,WAAgB,EAAE,QAAa;IACrD,OAAO,IAAI,KAAK,CAAC,WAAW,EAAE;QAC5B,GAAG,CAAC,MAAM,EAAE,IAAI,EAAE,QAAQ;YACxB,MAAM,QAAQ,GAAG,OAAO,CAAC,GAAG,CAAC,MAAM,EAAE,IAAI,EAAE,QAAQ,CAAC,CAAC;YAErD,IAAI,OAAO,QAAQ,KAAK,UAAU,IAAI,IAAI,KAAK,aAAa,EAAE,CAAC;gBAC7D,OAAO,KAAK,EAAE,GAAG,IAAW,EAAE,EAAE;oBAC9B,OAAO,QAAQ,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC,KAAK,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC,CAAC;gBAC9D,CAAC,CAAC;YACJ,CAAC;YAED,OAAO,QAAQ,CAAC;QAClB,CAAC;KACF,CAAC,CAAC;AACL,CAAC"}

package/dist/cjs/interceptors/ollama.d.ts

@@ -0,0 +1,143 @@
+import { BaseInterceptor } from './base';
+import { AIRequest } from '../types';
+/**
+ * Interceptor for Ollama API calls
+ *
+ * Ollama is a local LLM server that runs on localhost:11434 by default.
+ * No authentication is required.
+ *
+ * @example
+ * ```typescript
+ * import { AxonFlow, wrapOllamaClient } from '@axonflow/sdk';
+ * import Ollama from 'ollama';
+ *
+ * const ollama = new Ollama({ host: 'http://localhost:11434' });
+ * const axonflow = new AxonFlow({ endpoint: 'http://localhost:8080' });
+ *
+ * const wrapped = wrapOllamaClient(ollama, axonflow);
+ * const response = await wrapped.chat({
+ *   model: 'llama2',
+ *   messages: [{ role: 'user', content: 'Hello!' }]
+ * });
+ * ```
+ */
+export declare class OllamaInterceptor extends BaseInterceptor {
+    canHandle(aiCall: any): boolean;
+    extractRequest(aiCall: any): AIRequest;
+    executeWithModifications(aiCall: any, _modifications: any): Promise<any>;
+    getProvider(): string;
+}
+/**
+ * Ollama chat message
+ */
+export interface OllamaMessage {
+    role: 'system' | 'user' | 'assistant';
+    content: string;
+    images?: string[];
+}
+/**
+ * Ollama chat request
+ */
+export interface OllamaChatRequest {
+    model: string;
+    messages: OllamaMessage[];
+    stream?: boolean;
+    format?: 'json';
+    options?: {
+        temperature?: number;
+        top_p?: number;
+        top_k?: number;
+        num_predict?: number;
+        stop?: string[];
+    };
+}
+/**
+ * Ollama chat response
+ */
+export interface OllamaChatResponse {
+    model: string;
+    created_at: string;
+    message: OllamaMessage;
+    done: boolean;
+    total_duration?: number;
+    load_duration?: number;
+    prompt_eval_count?: number;
+    prompt_eval_duration?: number;
+    eval_count?: number;
+    eval_duration?: number;
+}
+/**
+ * Ollama generate request (for completions)
+ */
+export interface OllamaGenerateRequest {
+    model: string;
+    prompt: string;
+    stream?: boolean;
+    format?: 'json';
+    options?: {
+        temperature?: number;
+        top_p?: number;
+        top_k?: number;
+        num_predict?: number;
+        stop?: string[];
+    };
+}
+/**
+ * Ollama generate response
+ */
+export interface OllamaGenerateResponse {
+    model: string;
+    created_at: string;
+    response: string;
+    done: boolean;
+    total_duration?: number;
+    load_duration?: number;
+    prompt_eval_count?: number;
+    prompt_eval_duration?: number;
+    eval_count?: number;
+    eval_duration?: number;
+}
+/**
+ * Wraps an Ollama client with AxonFlow governance.
+ *
+ * @example
+ * ```typescript
+ * import Ollama from 'ollama';
+ * import { AxonFlow, wrapOllamaClient } from '@axonflow/sdk';
+ *
+ * const ollama = new Ollama({ host: 'http://localhost:11434' });
+ * const axonflow = new AxonFlow({ endpoint: 'http://localhost:8080' });
+ *
+ * const wrapped = wrapOllamaClient(ollama, axonflow);
+ *
+ * // Chat API
+ * const chatResponse = await wrapped.chat({
+ *   model: 'llama2',
+ *   messages: [{ role: 'user', content: 'What is AI governance?' }]
+ * });
+ *
+ * // Generate API
+ * const genResponse = await wrapped.generate({
+ *   model: 'llama2',
+ *   prompt: 'What is AI governance?'
+ * });
+ * ```
+ */
+export declare function wrapOllamaClient(ollamaClient: any, axonflow: any): any;
+/**
+ * Creates a governed Ollama chat function.
+ *
+ * Use this when you want fine-grained control over individual calls
+ * rather than wrapping the entire client.
+ *
+ * @example
+ * ```typescript
+ * const governedChat = createGovernedOllamaChat(ollamaClient, axonflow, 'user-123');
+ * const response = await governedChat({
+ *   model: 'llama2',
+ *   messages: [{ role: 'user', content: 'Hello!' }]
+ * });
+ * ```
+ */
+export declare function createGovernedOllamaChat(ollamaClient: any, axonflow: any, userToken?: string): (request: OllamaChatRequest) => Promise<OllamaChatResponse>;
+//# sourceMappingURL=ollama.d.ts.map

package/dist/cjs/interceptors/ollama.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ollama.d.ts","sourceRoot":"","sources":["../../../src/interceptors/ollama.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,MAAM,QAAQ,CAAC;AACzC,OAAO,EAAE,SAAS,EAAE,MAAM,UAAU,CAAC;AAErC;;;;;;;;;;;;;;;;;;;;GAoBG;AACH,qBAAa,iBAAkB,SAAQ,eAAe;IACpD,SAAS,CAAC,MAAM,EAAE,GAAG,GAAG,OAAO;IAW/B,cAAc,CAAC,MAAM,EAAE,GAAG,GAAG,SAAS;IAoBtC,wBAAwB,CAAC,MAAM,EAAE,GAAG,EAAE,cAAc,EAAE,GAAG,GAAG,OAAO,CAAC,GAAG,CAAC;IAIxE,WAAW,IAAI,MAAM;CAGtB;AAED;;GAEG;AACH,MAAM,WAAW,aAAa;IAC5B,IAAI,EAAE,QAAQ,GAAG,MAAM,GAAG,WAAW,CAAC;IACtC,OAAO,EAAE,MAAM,CAAC;IAChB,MAAM,CAAC,EAAE,MAAM,EAAE,CAAC;CACnB;AAED;;GAEG;AACH,MAAM,WAAW,iBAAiB;IAChC,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,EAAE,aAAa,EAAE,CAAC;IAC1B,MAAM,CAAC,EAAE,OAAO,CAAC;IACjB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,OAAO,CAAC,EAAE;QACR,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB,KAAK,CAAC,EAAE,MAAM,CAAC;QACf,KAAK,CAAC,EAAE,MAAM,CAAC;QACf,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC;KACjB,CAAC;CACH;AAED;;GAEG;AACH,MAAM,WAAW,kBAAkB;IACjC,KAAK,EAAE,MAAM,CAAC;IACd,UAAU,EAAE,MAAM,CAAC;IACnB,OAAO,EAAE,aAAa,CAAC;IACvB,IAAI,EAAE,OAAO,CAAC;IACd,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,oBAAoB,CAAC,EAAE,MAAM,CAAC;IAC9B,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,aAAa,CAAC,EAAE,MAAM,CAAC;CACxB;AAED;;GAEG;AACH,MAAM,WAAW,qBAAqB;IACpC,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,EAAE,MAAM,CAAC;IACf,MAAM,CAAC,EAAE,OAAO,CAAC;IACjB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,OAAO,CAAC,EAAE;QACR,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB,KAAK,CAAC,EAAE,MAAM,CAAC;QACf,KAAK,CAAC,EAAE,MAAM,CAAC;QACf,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC;KACjB,CAAC;CACH;AAED;;GAEG;AACH,MAAM,WAAW,sBAAsB;IACrC,KAAK,EAAE,MAAM,CAAC;IACd,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,EAAE,MAAM,CAAC;IACjB,IAAI,EAAE,OAAO,CAAC;IACd,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,oBAAoB,CAAC,EAAE,MAAM,CAAC;IAC9B,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,aAAa,CAAC,EAAE,MAAM,CAAC;CACxB;AAED;;;;;;;;;;;;;;;;;;;;;;;;;GAyBG;AACH,wBAAgB,gBAAgB,CAAC,YAAY,EAAE,GAAG,EAAE,QAAQ,EAAE,GAAG,GAAG,GAAG,CAetE;AAED;;;;;;;;;;;;;;GAcG;AACH,wBAAgB,wBAAwB,CACtC,YAAY,EAAE,GAAG,EACjB,QAAQ,EAAE,GAAG,EACb,SAAS,GAAE,MAAW,GACrB,CAAC,OAAO,EAAE,iBAAiB,KAAK,OAAO,CAAC,kBAAkB,CAAC,CA0C7D"}