@axonflow/sdk 2.3.0 → 3.3.1

package/dist/esm/interceptors/bedrock.d.ts

@@ -1,142 +0,0 @@
-import { BaseInterceptor } from './base';
-import { AIRequest } from '../types';
-/**
- * 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({...}));
- * ```
- */
-export declare class BedrockInterceptor extends BaseInterceptor {
-    canHandle(aiCall: any): boolean;
-    extractRequest(aiCall: any): AIRequest;
-    executeWithModifications(aiCall: any, _modifications: any): Promise<any>;
-    getProvider(): string;
-}
-/**
- * Bedrock model providers and their model IDs
- */
-export declare const BedrockModels: {
-    readonly CLAUDE_3_OPUS: "anthropic.claude-3-opus-20240229-v1:0";
-    readonly CLAUDE_3_SONNET: "anthropic.claude-3-sonnet-20240229-v1:0";
-    readonly CLAUDE_3_HAIKU: "anthropic.claude-3-haiku-20240307-v1:0";
-    readonly CLAUDE_2_1: "anthropic.claude-v2:1";
-    readonly CLAUDE_2: "anthropic.claude-v2";
-    readonly CLAUDE_INSTANT: "anthropic.claude-instant-v1";
-    readonly TITAN_TEXT_EXPRESS: "amazon.titan-text-express-v1";
-    readonly TITAN_TEXT_LITE: "amazon.titan-text-lite-v1";
-    readonly TITAN_TEXT_PREMIER: "amazon.titan-text-premier-v1:0";
-    readonly LLAMA2_13B: "meta.llama2-13b-chat-v1";
-    readonly LLAMA2_70B: "meta.llama2-70b-chat-v1";
-    readonly LLAMA3_8B: "meta.llama3-8b-instruct-v1:0";
-    readonly LLAMA3_70B: "meta.llama3-70b-instruct-v1:0";
-    readonly COHERE_COMMAND: "cohere.command-text-v14";
-    readonly COHERE_COMMAND_LIGHT: "cohere.command-light-text-v14";
-    readonly AI21_JURASSIC_ULTRA: "ai21.j2-ultra-v1";
-    readonly AI21_JURASSIC_MID: "ai21.j2-mid-v1";
-};
-/**
- * Bedrock invoke model request body for Claude
- */
-export interface BedrockClaudeRequest {
-    anthropic_version: string;
-    max_tokens: number;
-    messages: Array<{
-        role: 'user' | 'assistant';
-        content: string | Array<{
-            type: 'text';
-            text: string;
-        }>;
-    }>;
-    temperature?: number;
-    top_p?: number;
-    top_k?: number;
-    stop_sequences?: string[];
-    system?: string;
-}
-/**
- * Bedrock invoke model response for Claude
- */
-export interface BedrockClaudeResponse {
-    id: string;
-    type: string;
-    role: string;
-    content: Array<{
-        type: 'text';
-        text: string;
-    }>;
-    model: string;
-    stop_reason: string;
-    usage: {
-        input_tokens: number;
-        output_tokens: number;
-    };
-}
-/**
- * Bedrock invoke model request body for Titan
- */
-export interface BedrockTitanRequest {
-    inputText: string;
-    textGenerationConfig?: {
-        maxTokenCount?: number;
-        temperature?: number;
-        topP?: number;
-        stopSequences?: string[];
-    };
-}
-/**
- * Bedrock invoke model response for Titan
- */
-export interface BedrockTitanResponse {
-    inputTextTokenCount: number;
-    results: Array<{
-        tokenCount: number;
-        outputText: string;
-        completionReason: string;
-    }>;
-}
-/**
- * Wraps an AWS Bedrock client with AxonFlow governance.
- *
- * @deprecated This function is deprecated and will be removed in v2.0.0.
- * JavaScript Proxy-based wrapping has compatibility issues with modern SDK versions.
- *
- * Use Gateway Mode or Proxy Mode instead:
- *
- * Gateway Mode (recommended):
- * ```typescript
- * const context = await axonflow.getPolicyApprovedContext({ query, userToken });
- * const response = await bedrockClient.send(new InvokeModelCommand({...}));
- * await axonflow.auditLLMCall({ contextId: context.contextId, ... });
- * ```
- *
- * Proxy Mode:
- * ```typescript
- * const response = await axonflow.executeQuery({
- *   query,
- *   userToken,
- *   context: { provider: 'bedrock', model: 'anthropic.claude-3-sonnet' }
- * });
- * ```
- *
- * See: https://docs.getaxonflow.com/sdk/gateway-mode
- */
-export declare function wrapBedrockClient(bedrockClient: any, axonflow: any): any;
-/**
- * Creates a governed Bedrock invoke function with full governance flow.
- *
- * Use this for fine-grained control with pre-check and audit.
- */
-export declare function createGovernedBedrockInvoke(bedrockClient: any, axonflow: any, userToken?: string): (modelId: string, body: any) => Promise<any>;
-//# sourceMappingURL=bedrock.d.ts.map

package/dist/esm/interceptors/bedrock.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"bedrock.d.ts","sourceRoot":"","sources":["../../../src/interceptors/bedrock.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,MAAM,QAAQ,CAAC;AACzC,OAAO,EAAE,SAAS,EAAE,MAAM,UAAU,CAAC;AAErC;;;;;;;;;;;;;;;;;GAiBG;AACH,qBAAa,kBAAmB,SAAQ,eAAe;IACrD,SAAS,CAAC,MAAM,EAAE,GAAG,GAAG,OAAO;IAW/B,cAAc,CAAC,MAAM,EAAE,GAAG,GAAG,SAAS;IAsBtC,wBAAwB,CAAC,MAAM,EAAE,GAAG,EAAE,cAAc,EAAE,GAAG,GAAG,OAAO,CAAC,GAAG,CAAC;IAIxE,WAAW,IAAI,MAAM;CAGtB;AAED;;GAEG;AACH,eAAO,MAAM,aAAa;;;;;;;;;;;;;;;;;;CA2BhB,CAAC;AAEX;;GAEG;AACH,MAAM,WAAW,oBAAoB;IACnC,iBAAiB,EAAE,MAAM,CAAC;IAC1B,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,EAAE,KAAK,CAAC;QACd,IAAI,EAAE,MAAM,GAAG,WAAW,CAAC;QAC3B,OAAO,EAAE,MAAM,GAAG,KAAK,CAAC;YAAE,IAAI,EAAE,MAAM,CAAC;YAAC,IAAI,EAAE,MAAM,CAAA;SAAE,CAAC,CAAC;KACzD,CAAC,CAAC;IACH,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,cAAc,CAAC,EAAE,MAAM,EAAE,CAAC;IAC1B,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB;AAED;;GAEG;AACH,MAAM,WAAW,qBAAqB;IACpC,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IAC/C,KAAK,EAAE,MAAM,CAAC;IACd,WAAW,EAAE,MAAM,CAAC;IACpB,KAAK,EAAE;QACL,YAAY,EAAE,MAAM,CAAC;QACrB,aAAa,EAAE,MAAM,CAAC;KACvB,CAAC;CACH;AAED;;GAEG;AACH,MAAM,WAAW,mBAAmB;IAClC,SAAS,EAAE,MAAM,CAAC;IAClB,oBAAoB,CAAC,EAAE;QACrB,aAAa,CAAC,EAAE,MAAM,CAAC;QACvB,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB,IAAI,CAAC,EAAE,MAAM,CAAC;QACd,aAAa,CAAC,EAAE,MAAM,EAAE,CAAC;KAC1B,CAAC;CACH;AAED;;GAEG;AACH,MAAM,WAAW,oBAAoB;IACnC,mBAAmB,EAAE,MAAM,CAAC;IAC5B,OAAO,EAAE,KAAK,CAAC;QACb,UAAU,EAAE,MAAM,CAAC;QACnB,UAAU,EAAE,MAAM,CAAC;QACnB,gBAAgB,EAAE,MAAM,CAAC;KAC1B,CAAC,CAAC;CACJ;AAED;;;;;;;;;;;;;;;;;;;;;;;;;GAyBG;AACH,wBAAgB,iBAAiB,CAAC,aAAa,EAAE,GAAG,EAAE,QAAQ,EAAE,GAAG,GAAG,GAAG,CAsDxE;AAED;;;;GAIG;AACH,wBAAgB,2BAA2B,CACzC,aAAa,EAAE,GAAG,EAClB,QAAQ,EAAE,GAAG,EACb,SAAS,GAAE,MAAW,GACrB,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,KAAK,OAAO,CAAC,GAAG,CAAC,CA8E9C"}

package/dist/esm/interceptors/bedrock.js

@@ -1,224 +0,0 @@
-import { BaseInterceptor } from './base.js';
-/**
- * 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({...}));
- * ```
- */
-export class BedrockInterceptor extends 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';
-    }
-}
-/**
- * Bedrock model providers and their model IDs
- */
-export const 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.
- *
- * @deprecated This function is deprecated and will be removed in v2.0.0.
- * JavaScript Proxy-based wrapping has compatibility issues with modern SDK versions.
- *
- * Use Gateway Mode or Proxy Mode instead:
- *
- * Gateway Mode (recommended):
- * ```typescript
- * const context = await axonflow.getPolicyApprovedContext({ query, userToken });
- * const response = await bedrockClient.send(new InvokeModelCommand({...}));
- * await axonflow.auditLLMCall({ contextId: context.contextId, ... });
- * ```
- *
- * Proxy Mode:
- * ```typescript
- * const response = await axonflow.executeQuery({
- *   query,
- *   userToken,
- *   context: { provider: 'bedrock', model: 'anthropic.claude-3-sonnet' }
- * });
- * ```
- *
- * See: https://docs.getaxonflow.com/sdk/gateway-mode
- */
-export function wrapBedrockClient(bedrockClient, axonflow) {
-    console.warn('[AxonFlow] wrapBedrockClient is deprecated and will be removed in v2.0.0. ' +
-        'Use Gateway Mode (getPolicyApprovedContext + auditLLMCall) or Proxy Mode (executeQuery) instead. ' +
-        'See: https://docs.getaxonflow.com/sdk/gateway-mode');
-    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.
- */
-export 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 import('@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/esm/interceptors/bedrock.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"bedrock.js","sourceRoot":"","sources":["../../../src/interceptors/bedrock.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,MAAM,QAAQ,CAAC;AAGzC;;;;;;;;;;;;;;;;;GAiBG;AACH,MAAM,OAAO,kBAAmB,SAAQ,eAAe;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;AAED;;GAEG;AACH,MAAM,CAAC,MAAM,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;;;;;;;;;;;;;;;;;;;;;;;;;GAyBG;AACH,MAAM,UAAU,iBAAiB,CAAC,aAAkB,EAAE,QAAa;IACjE,OAAO,CAAC,IAAI,CACV,4EAA4E;QAC1E,mGAAmG;QACnG,oDAAoD,CACvD,CAAC;IAEF,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,MAAM,UAAU,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,MAAM,MAAM,CAAC,iCAAiC,CAAC,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/esm/interceptors/gemini.d.ts

@@ -1,89 +0,0 @@
-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
- *
- * @deprecated This function is deprecated and will be removed in v2.0.0.
- * JavaScript Proxy-based wrapping has compatibility issues with modern SDK versions.
- *
- * Use Gateway Mode or Proxy Mode instead:
- *
- * Gateway Mode (recommended):
- * ```typescript
- * const context = await axonflow.getPolicyApprovedContext({ query, userToken });
- * const response = await model.generateContent(query);
- * await axonflow.auditLLMCall({ contextId: context.contextId, ... });
- * ```
- *
- * Proxy Mode:
- * ```typescript
- * const response = await axonflow.executeQuery({
- *   query,
- *   userToken,
- *   context: { provider: 'gemini', model: 'gemini-2.0-flash' }
- * });
- * ```
- *
- * See: https://docs.getaxonflow.com/sdk/gateway-mode
- */
-export declare function wrapGeminiModel(geminiModel: any, axonflow: any): any;
-//# sourceMappingURL=gemini.d.ts.map

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

@@ -1 +0,0 @@
-{"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;;;;;;;;;;;;;;;;;;;;;;;;;GAyBG;AACH,wBAAgB,eAAe,CAAC,WAAW,EAAE,GAAG,EAAE,QAAQ,EAAE,GAAG,GAAG,GAAG,CAgCpE"}

package/dist/esm/interceptors/gemini.js

@@ -1,116 +0,0 @@
-import { BaseInterceptor } from './base.js';
-/**
- * Interceptor for Google Gemini API calls
- *
- * Supports both @google/generative-ai SDK and direct API calls.
- */
-export class GeminiInterceptor extends 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';
-    }
-}
-/**
- * Helper to wrap Gemini GenerativeModel for easier interception
- *
- * @deprecated This function is deprecated and will be removed in v2.0.0.
- * JavaScript Proxy-based wrapping has compatibility issues with modern SDK versions.
- *
- * Use Gateway Mode or Proxy Mode instead:
- *
- * Gateway Mode (recommended):
- * ```typescript
- * const context = await axonflow.getPolicyApprovedContext({ query, userToken });
- * const response = await model.generateContent(query);
- * await axonflow.auditLLMCall({ contextId: context.contextId, ... });
- * ```
- *
- * Proxy Mode:
- * ```typescript
- * const response = await axonflow.executeQuery({
- *   query,
- *   userToken,
- *   context: { provider: 'gemini', model: 'gemini-2.0-flash' }
- * });
- * ```
- *
- * See: https://docs.getaxonflow.com/sdk/gateway-mode
- */
-export function wrapGeminiModel(geminiModel, axonflow) {
-    console.warn('[AxonFlow] wrapGeminiModel is deprecated and will be removed in v2.0.0. ' +
-        'Use Gateway Mode (getPolicyApprovedContext + auditLLMCall) or Proxy Mode (executeQuery) instead. ' +
-        'See: https://docs.getaxonflow.com/sdk/gateway-mode');
-    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/esm/interceptors/gemini.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"gemini.js","sourceRoot":"","sources":["../../../src/interceptors/gemini.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,MAAM,QAAQ,CAAC;AAGzC;;;;GAIG;AACH,MAAM,OAAO,iBAAkB,SAAQ,eAAe;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;AA0DD;;;;;;;;;;;;;;;;;;;;;;;;;GAyBG;AACH,MAAM,UAAU,eAAe,CAAC,WAAgB,EAAE,QAAa;IAC7D,OAAO,CAAC,IAAI,CACV,0EAA0E;QACxE,mGAAmG;QACnG,oDAAoD,CACvD,CAAC;IACF,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"}