ai.matey.core 0.2.0

package/dist/cjs/middleware-stack.js

@@ -0,0 +1,256 @@
+"use strict";
+/**
+ * Middleware Stack Implementation
+ *
+ * Composable middleware system for request/response transformation.
+ * Middleware executes in order with "onion" pattern - each middleware
+ * can execute code before and after calling next().
+ *
+ * @module
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.MiddlewareStack = void 0;
+exports.createMiddlewareContext = createMiddlewareContext;
+exports.createStreamingMiddlewareContext = createStreamingMiddlewareContext;
+const ai_matey_errors_1 = require("ai.matey.errors");
+// ============================================================================
+// Middleware Stack
+// ============================================================================
+/**
+ * Middleware stack for composing middleware functions.
+ *
+ * Executes middleware in order with proper error handling and context management.
+ */
+class MiddlewareStack {
+    middleware = [];
+    streamingMiddleware = [];
+    locked = false;
+    /**
+     * Add middleware to the stack.
+     *
+     * @param middleware Middleware function to add
+     * @throws {MiddlewareError} If stack is locked
+     */
+    use(middleware) {
+        if (this.locked) {
+            throw new ai_matey_errors_1.MiddlewareError({
+                message: 'Cannot add middleware after stack is locked',
+                middlewareName: 'unknown',
+            });
+        }
+        this.middleware.push(middleware);
+    }
+    /**
+     * Remove middleware from the stack.
+     *
+     * @param middleware Middleware function to remove
+     * @returns true if middleware was found and removed, false otherwise
+     * @throws {MiddlewareError} If stack is locked
+     */
+    remove(middleware) {
+        if (this.locked) {
+            throw new ai_matey_errors_1.MiddlewareError({
+                message: 'Cannot remove middleware after stack is locked',
+                middlewareName: 'unknown',
+            });
+        }
+        const index = this.middleware.indexOf(middleware);
+        if (index !== -1) {
+            this.middleware.splice(index, 1);
+            return true;
+        }
+        return false;
+    }
+    /**
+     * Add streaming middleware to the stack.
+     *
+     * @param middleware Streaming middleware function to add
+     * @throws {MiddlewareError} If stack is locked
+     */
+    useStreaming(middleware) {
+        if (this.locked) {
+            throw new ai_matey_errors_1.MiddlewareError({
+                message: 'Cannot add streaming middleware after stack is locked',
+                middlewareName: 'unknown',
+            });
+        }
+        this.streamingMiddleware.push(middleware);
+    }
+    /**
+     * Lock the stack to prevent further modifications.
+     *
+     * Called automatically on first request execution.
+     */
+    lock() {
+        this.locked = true;
+    }
+    /**
+     * Get all middleware in the stack.
+     */
+    getMiddleware() {
+        return [...this.middleware];
+    }
+    /**
+     * Get all streaming middleware in the stack.
+     */
+    getStreamingMiddleware() {
+        return [...this.streamingMiddleware];
+    }
+    /**
+     * Check if stack is locked.
+     */
+    isLocked() {
+        return this.locked;
+    }
+    /**
+     * Execute middleware stack for non-streaming requests.
+     *
+     * @param context Middleware context
+     * @param finalHandler Final handler function (backend.execute)
+     * @returns Response after middleware chain
+     */
+    async execute(context, finalHandler) {
+        // Lock stack on first execution
+        if (!this.locked) {
+            this.lock();
+        }
+        // If no middleware, call handler directly
+        if (this.middleware.length === 0) {
+            return finalHandler();
+        }
+        // Compose middleware chain
+        let index = 0;
+        const next = async () => {
+            if (index >= this.middleware.length) {
+                // End of middleware chain, call final handler
+                return finalHandler();
+            }
+            const middlewareFn = this.middleware[index];
+            if (!middlewareFn) {
+                // End of middleware chain, call final handler
+                return finalHandler();
+            }
+            index++;
+            try {
+                return await middlewareFn(context, next);
+            }
+            catch (error) {
+                // Re-throw as MiddlewareError if not already
+                if (error instanceof ai_matey_errors_1.MiddlewareError) {
+                    throw error;
+                }
+                throw new ai_matey_errors_1.MiddlewareError({
+                    message: `Middleware execution failed: ${error instanceof Error ? error.message : String(error)}`,
+                    cause: error instanceof Error ? error : undefined,
+                    irState: {
+                        request: context.request,
+                    },
+                });
+            }
+        };
+        return next();
+    }
+    /**
+     * Execute middleware stack for streaming requests.
+     *
+     * @param context Streaming middleware context
+     * @param finalHandler Final handler function (backend.executeStream)
+     * @returns Stream after middleware chain
+     */
+    async executeStream(context, finalHandler) {
+        // Lock stack on first execution
+        if (!this.locked) {
+            this.lock();
+        }
+        // If no streaming middleware, call handler directly
+        if (this.streamingMiddleware.length === 0) {
+            return finalHandler();
+        }
+        // Compose streaming middleware chain
+        let index = 0;
+        const next = async () => {
+            if (index >= this.streamingMiddleware.length) {
+                // End of middleware chain, call final handler
+                return finalHandler();
+            }
+            const middlewareFn = this.streamingMiddleware[index];
+            if (!middlewareFn) {
+                // End of middleware chain, call final handler
+                return finalHandler();
+            }
+            index++;
+            try {
+                return await middlewareFn(context, next);
+            }
+            catch (error) {
+                // Re-throw as MiddlewareError if not already
+                if (error instanceof ai_matey_errors_1.MiddlewareError) {
+                    throw error;
+                }
+                throw new ai_matey_errors_1.MiddlewareError({
+                    message: `Streaming middleware execution failed: ${error instanceof Error ? error.message : String(error)}`,
+                    cause: error instanceof Error ? error : undefined,
+                    irState: {
+                        request: context.request,
+                    },
+                });
+            }
+        };
+        return next();
+    }
+    /**
+     * Clear all middleware from the stack.
+     *
+     * @throws {MiddlewareError} If stack is locked
+     */
+    clear() {
+        if (this.locked) {
+            throw new ai_matey_errors_1.MiddlewareError({
+                message: 'Cannot clear middleware after stack is locked',
+            });
+        }
+        this.middleware = [];
+        this.streamingMiddleware = [];
+    }
+}
+exports.MiddlewareStack = MiddlewareStack;
+// ============================================================================
+// Helper Functions
+// ============================================================================
+/**
+ * Create middleware context from request and config.
+ *
+ * @param request IR request
+ * @param config Bridge configuration
+ * @param signal Optional abort signal
+ * @returns Middleware context
+ */
+function createMiddlewareContext(request, config, signal) {
+    return {
+        request,
+        isStreaming: request.stream ?? false,
+        state: {},
+        config,
+        signal,
+    };
+}
+/**
+ * Create streaming middleware context from request and config.
+ *
+ * @param request IR request
+ * @param config Bridge configuration
+ * @param signal Optional abort signal
+ * @returns Streaming middleware context
+ */
+function createStreamingMiddlewareContext(request, config, signal) {
+    return {
+        request,
+        isStreaming: true,
+        state: {},
+        config,
+        signal,
+        chunksProcessed: 0,
+        streamComplete: false,
+    };
+}
+//# sourceMappingURL=middleware-stack.js.map

package/dist/cjs/middleware-stack.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"middleware-stack.js","sourceRoot":"","sources":["../../src/middleware-stack.ts"],"names":[],"mappings":";AAAA;;;;;;;;GAQG;;;AA6PH,0DAYC;AAUD,4EAcC;AAtRD,qDAAkD;AAElD,+EAA+E;AAC/E,mBAAmB;AACnB,+EAA+E;AAE/E;;;;GAIG;AACH,MAAa,eAAe;IAClB,UAAU,GAAiB,EAAE,CAAC;IAC9B,mBAAmB,GAA0B,EAAE,CAAC;IAChD,MAAM,GAAY,KAAK,CAAC;IAEhC;;;;;OAKG;IACH,GAAG,CAAC,UAAsB;QACxB,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChB,MAAM,IAAI,iCAAe,CAAC;gBACxB,OAAO,EAAE,6CAA6C;gBACtD,cAAc,EAAE,SAAS;aAC1B,CAAC,CAAC;QACL,CAAC;QACD,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IACnC,CAAC;IAED;;;;;;OAMG;IACH,MAAM,CAAC,UAAsB;QAC3B,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChB,MAAM,IAAI,iCAAe,CAAC;gBACxB,OAAO,EAAE,gDAAgD;gBACzD,cAAc,EAAE,SAAS;aAC1B,CAAC,CAAC;QACL,CAAC;QACD,MAAM,KAAK,GAAG,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;QAClD,IAAI,KAAK,KAAK,CAAC,CAAC,EAAE,CAAC;YACjB,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;YACjC,OAAO,IAAI,CAAC;QACd,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED;;;;;OAKG;IACH,YAAY,CAAC,UAA+B;QAC1C,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChB,MAAM,IAAI,iCAAe,CAAC;gBACxB,OAAO,EAAE,uDAAuD;gBAChE,cAAc,EAAE,SAAS;aAC1B,CAAC,CAAC;QACL,CAAC;QACD,IAAI,CAAC,mBAAmB,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IAC5C,CAAC;IAED;;;;OAIG;IACH,IAAI;QACF,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;IACrB,CAAC;IAED;;OAEG;IACH,aAAa;QACX,OAAO,CAAC,GAAG,IAAI,CAAC,UAAU,CAAC,CAAC;IAC9B,CAAC;IAED;;OAEG;IACH,sBAAsB;QACpB,OAAO,CAAC,GAAG,IAAI,CAAC,mBAAmB,CAAC,CAAC;IACvC,CAAC;IAED;;OAEG;IACH,QAAQ;QACN,OAAO,IAAI,CAAC,MAAM,CAAC;IACrB,CAAC;IAED;;;;;;OAMG;IACH,KAAK,CAAC,OAAO,CACX,OAA0B,EAC1B,YAA2C;QAE3C,gCAAgC;QAChC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,IAAI,CAAC,IAAI,EAAE,CAAC;QACd,CAAC;QAED,0CAA0C;QAC1C,IAAI,IAAI,CAAC,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACjC,OAAO,YAAY,EAAE,CAAC;QACxB,CAAC;QAED,2BAA2B;QAC3B,IAAI,KAAK,GAAG,CAAC,CAAC;QAEd,MAAM,IAAI,GAAmB,KAAK,IAA6B,EAAE;YAC/D,IAAI,KAAK,IAAI,IAAI,CAAC,UAAU,CAAC,MAAM,EAAE,CAAC;gBACpC,8CAA8C;gBAC9C,OAAO,YAAY,EAAE,CAAC;YACxB,CAAC;YAED,MAAM,YAAY,GAAG,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;YAC5C,IAAI,CAAC,YAAY,EAAE,CAAC;gBAClB,8CAA8C;gBAC9C,OAAO,YAAY,EAAE,CAAC;YACxB,CAAC;YACD,KAAK,EAAE,CAAC;YAER,IAAI,CAAC;gBACH,OAAO,MAAM,YAAY,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;YAC3C,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,6CAA6C;gBAC7C,IAAI,KAAK,YAAY,iCAAe,EAAE,CAAC;oBACrC,MAAM,KAAK,CAAC;gBACd,CAAC;gBACD,MAAM,IAAI,iCAAe,CAAC;oBACxB,OAAO,EAAE,gCAAgC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE;oBACjG,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS;oBACjD,OAAO,EAAE;wBACP,OAAO,EAAE,OAAO,CAAC,OAAO;qBACzB;iBACF,CAAC,CAAC;YACL,CAAC;QACH,CAAC,CAAC;QAEF,OAAO,IAAI,EAAE,CAAC;IAChB,CAAC;IAED;;;;;;OAMG;IACH,KAAK,CAAC,aAAa,CACjB,OAAmC,EACnC,YAAyC;QAEzC,gCAAgC;QAChC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,IAAI,CAAC,IAAI,EAAE,CAAC;QACd,CAAC;QAED,oDAAoD;QACpD,IAAI,IAAI,CAAC,mBAAmB,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC1C,OAAO,YAAY,EAAE,CAAC;QACxB,CAAC;QAED,qCAAqC;QACrC,IAAI,KAAK,GAAG,CAAC,CAAC;QAEd,MAAM,IAAI,GAA4B,KAAK,IAA2B,EAAE;YACtE,IAAI,KAAK,IAAI,IAAI,CAAC,mBAAmB,CAAC,MAAM,EAAE,CAAC;gBAC7C,8CAA8C;gBAC9C,OAAO,YAAY,EAAE,CAAC;YACxB,CAAC;YAED,MAAM,YAAY,GAAG,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC,CAAC;YACrD,IAAI,CAAC,YAAY,EAAE,CAAC;gBAClB,8CAA8C;gBAC9C,OAAO,YAAY,EAAE,CAAC;YACxB,CAAC;YACD,KAAK,EAAE,CAAC;YAER,IAAI,CAAC;gBACH,OAAO,MAAM,YAAY,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;YAC3C,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,6CAA6C;gBAC7C,IAAI,KAAK,YAAY,iCAAe,EAAE,CAAC;oBACrC,MAAM,KAAK,CAAC;gBACd,CAAC;gBACD,MAAM,IAAI,iCAAe,CAAC;oBACxB,OAAO,EAAE,0CAA0C,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE;oBAC3G,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS;oBACjD,OAAO,EAAE;wBACP,OAAO,EAAE,OAAO,CAAC,OAAO;qBACzB;iBACF,CAAC,CAAC;YACL,CAAC;QACH,CAAC,CAAC;QAEF,OAAO,IAAI,EAAE,CAAC;IAChB,CAAC;IAED;;;;OAIG;IACH,KAAK;QACH,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChB,MAAM,IAAI,iCAAe,CAAC;gBACxB,OAAO,EAAE,+CAA+C;aACzD,CAAC,CAAC;QACL,CAAC;QACD,IAAI,CAAC,UAAU,GAAG,EAAE,CAAC;QACrB,IAAI,CAAC,mBAAmB,GAAG,EAAE,CAAC;IAChC,CAAC;CACF;AAzND,0CAyNC;AAED,+EAA+E;AAC/E,mBAAmB;AACnB,+EAA+E;AAE/E;;;;;;;GAOG;AACH,SAAgB,uBAAuB,CACrC,OAAsB,EACtB,MAA+B,EAC/B,MAAoB;IAEpB,OAAO;QACL,OAAO;QACP,WAAW,EAAE,OAAO,CAAC,MAAM,IAAI,KAAK;QACpC,KAAK,EAAE,EAAE;QACT,MAAM;QACN,MAAM;KACP,CAAC;AACJ,CAAC;AAED;;;;;;;GAOG;AACH,SAAgB,gCAAgC,CAC9C,OAAsB,EACtB,MAA+B,EAC/B,MAAoB;IAEpB,OAAO;QACL,OAAO;QACP,WAAW,EAAE,IAAI;QACjB,KAAK,EAAE,EAAE;QACT,MAAM;QACN,MAAM;QACN,eAAe,EAAE,CAAC;QAClB,cAAc,EAAE,KAAK;KACtB,CAAC;AACJ,CAAC"}

package/dist/cjs/model-pricing.js

@@ -0,0 +1,350 @@
+"use strict";
+/**
+ * Model Pricing Registry
+ *
+ * Centralized pricing database for AI models across providers.
+ * Prices are in USD per 1,000 tokens.
+ *
+ * @module
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getModelPricing = getModelPricing;
+exports.getModelCapabilities = getModelCapabilities;
+exports.getAllPricedModels = getAllPricedModels;
+exports.getModelsByFamily = getModelsByFamily;
+exports.setPricingOverride = setPricingOverride;
+exports.clearPricingOverride = clearPricingOverride;
+exports.clearAllPricingOverrides = clearAllPricingOverrides;
+exports.getPricingWithOverrides = getPricingWithOverrides;
+/**
+ * Static pricing database for major AI models.
+ * Data sourced from provider APIs and community benchmarks (as of October 2024).
+ */
+const MODEL_PRICING_DATABASE = {
+    // OpenAI Models
+    'gpt-4': {
+        pricing: { input: 0.03, output: 0.06 },
+        latency: { p50: 2000, p95: 4000 },
+        qualityScore: 95,
+        modelFamily: 'gpt-4',
+        releaseDate: '2023-03-14',
+        contextWindow: 8192,
+        maxTokens: 4096,
+        supportsStreaming: true,
+        supportsVision: false,
+        supportsTools: true,
+        supportsJSON: true,
+    },
+    'gpt-4-turbo': {
+        pricing: { input: 0.01, output: 0.03 },
+        latency: { p50: 1800, p95: 3500 },
+        qualityScore: 95,
+        modelFamily: 'gpt-4',
+        releaseDate: '2024-04-09',
+        contextWindow: 128000,
+        maxTokens: 4096,
+        supportsStreaming: true,
+        supportsVision: true,
+        supportsTools: true,
+        supportsJSON: true,
+    },
+    'gpt-4-turbo-preview': {
+        pricing: { input: 0.01, output: 0.03 },
+        latency: { p50: 1800, p95: 3500 },
+        qualityScore: 94,
+        modelFamily: 'gpt-4',
+        releaseDate: '2024-01-25',
+        contextWindow: 128000,
+        maxTokens: 4096,
+        supportsStreaming: true,
+        supportsVision: true,
+        supportsTools: true,
+        supportsJSON: true,
+    },
+    'gpt-3.5-turbo': {
+        pricing: { input: 0.0005, output: 0.0015 },
+        latency: { p50: 800, p95: 1500 },
+        qualityScore: 75,
+        modelFamily: 'gpt-3.5',
+        releaseDate: '2023-03-01',
+        contextWindow: 16385,
+        maxTokens: 4096,
+        supportsStreaming: true,
+        supportsVision: false,
+        supportsTools: true,
+        supportsJSON: true,
+    },
+    'gpt-4o': {
+        pricing: { input: 0.005, output: 0.015 },
+        latency: { p50: 1200, p95: 2500 },
+        qualityScore: 96,
+        modelFamily: 'gpt-4',
+        releaseDate: '2024-05-13',
+        contextWindow: 128000,
+        maxTokens: 4096,
+        supportsStreaming: true,
+        supportsVision: true,
+        supportsTools: true,
+        supportsJSON: true,
+    },
+    'gpt-4o-mini': {
+        pricing: { input: 0.00015, output: 0.0006 },
+        latency: { p50: 600, p95: 1200 },
+        qualityScore: 80,
+        modelFamily: 'gpt-4',
+        releaseDate: '2024-07-18',
+        contextWindow: 128000,
+        maxTokens: 16384,
+        supportsStreaming: true,
+        supportsVision: true,
+        supportsTools: true,
+        supportsJSON: true,
+    },
+    // Anthropic Models
+    'claude-3-5-sonnet-20241022': {
+        pricing: { input: 0.003, output: 0.015 },
+        latency: { p50: 1500, p95: 3000 },
+        qualityScore: 97,
+        modelFamily: 'claude-3',
+        releaseDate: '2024-10-22',
+        contextWindow: 200000,
+        maxTokens: 8192,
+        supportsStreaming: true,
+        supportsVision: true,
+        supportsTools: true,
+        supportsJSON: false,
+    },
+    'claude-3-opus-20240229': {
+        pricing: { input: 0.015, output: 0.075 },
+        latency: { p50: 2200, p95: 4500 },
+        qualityScore: 96,
+        modelFamily: 'claude-3',
+        releaseDate: '2024-02-29',
+        contextWindow: 200000,
+        maxTokens: 4096,
+        supportsStreaming: true,
+        supportsVision: true,
+        supportsTools: true,
+        supportsJSON: false,
+    },
+    'claude-3-sonnet-20240229': {
+        pricing: { input: 0.003, output: 0.015 },
+        latency: { p50: 1600, p95: 3200 },
+        qualityScore: 92,
+        modelFamily: 'claude-3',
+        releaseDate: '2024-02-29',
+        contextWindow: 200000,
+        maxTokens: 4096,
+        supportsStreaming: true,
+        supportsVision: true,
+        supportsTools: true,
+        supportsJSON: false,
+    },
+    'claude-3-haiku-20240307': {
+        pricing: { input: 0.00025, output: 0.00125 },
+        latency: { p50: 400, p95: 800 },
+        qualityScore: 78,
+        modelFamily: 'claude-3',
+        releaseDate: '2024-03-07',
+        contextWindow: 200000,
+        maxTokens: 4096,
+        supportsStreaming: true,
+        supportsVision: true,
+        supportsTools: true,
+        supportsJSON: false,
+    },
+    'claude-3-5-haiku-20241022': {
+        pricing: { input: 0.001, output: 0.005 },
+        latency: { p50: 500, p95: 1000 },
+        qualityScore: 82,
+        modelFamily: 'claude-3',
+        releaseDate: '2024-10-22',
+        contextWindow: 200000,
+        maxTokens: 8192,
+        supportsStreaming: true,
+        supportsVision: false,
+        supportsTools: true,
+        supportsJSON: false,
+    },
+    // Google Gemini Models
+    'gemini-1.5-pro': {
+        pricing: { input: 0.00125, output: 0.005 },
+        latency: { p50: 1400, p95: 2800 },
+        qualityScore: 93,
+        modelFamily: 'gemini-1.5',
+        releaseDate: '2024-02-15',
+        contextWindow: 1000000,
+        maxTokens: 8192,
+        supportsStreaming: true,
+        supportsVision: true,
+        supportsTools: true,
+        supportsJSON: true,
+    },
+    'gemini-1.5-flash': {
+        pricing: { input: 0.000075, output: 0.0003 },
+        latency: { p50: 600, p95: 1200 },
+        qualityScore: 80,
+        modelFamily: 'gemini-1.5',
+        releaseDate: '2024-05-14',
+        contextWindow: 1000000,
+        maxTokens: 8192,
+        supportsStreaming: true,
+        supportsVision: true,
+        supportsTools: true,
+        supportsJSON: true,
+    },
+    'gemini-1.5-flash-8b': {
+        pricing: { input: 0.0000375, output: 0.00015 },
+        latency: { p50: 400, p95: 800 },
+        qualityScore: 70,
+        modelFamily: 'gemini-1.5',
+        releaseDate: '2024-10-03',
+        contextWindow: 1000000,
+        maxTokens: 8192,
+        supportsStreaming: true,
+        supportsVision: true,
+        supportsTools: true,
+        supportsJSON: true,
+    },
+    // Mistral AI Models
+    'mistral-large-latest': {
+        pricing: { input: 0.002, output: 0.006 },
+        latency: { p50: 1300, p95: 2600 },
+        qualityScore: 90,
+        modelFamily: 'mistral',
+        releaseDate: '2024-07-24',
+        contextWindow: 128000,
+        maxTokens: 4096,
+        supportsStreaming: true,
+        supportsVision: false,
+        supportsTools: true,
+        supportsJSON: true,
+    },
+    'mistral-medium-latest': {
+        pricing: { input: 0.0027, output: 0.0081 },
+        latency: { p50: 1100, p95: 2200 },
+        qualityScore: 85,
+        modelFamily: 'mistral',
+        releaseDate: '2023-12-11',
+        contextWindow: 32000,
+        maxTokens: 4096,
+        supportsStreaming: true,
+        supportsVision: false,
+        supportsTools: true,
+        supportsJSON: true,
+    },
+    'mistral-small-latest': {
+        pricing: { input: 0.001, output: 0.003 },
+        latency: { p50: 700, p95: 1400 },
+        qualityScore: 78,
+        modelFamily: 'mistral',
+        releaseDate: '2024-02-26',
+        contextWindow: 32000,
+        maxTokens: 4096,
+        supportsStreaming: true,
+        supportsVision: false,
+        supportsTools: true,
+        supportsJSON: true,
+    },
+    // DeepSeek Models
+    'deepseek-chat': {
+        pricing: { input: 0.00014, output: 0.00028 },
+        latency: { p50: 900, p95: 1800 },
+        qualityScore: 82,
+        modelFamily: 'deepseek',
+        releaseDate: '2024-01-01',
+        contextWindow: 64000,
+        maxTokens: 4096,
+        supportsStreaming: true,
+        supportsVision: false,
+        supportsTools: true,
+        supportsJSON: true,
+    },
+    'deepseek-coder': {
+        pricing: { input: 0.00014, output: 0.00028 },
+        latency: { p50: 900, p95: 1800 },
+        qualityScore: 88,
+        modelFamily: 'deepseek',
+        releaseDate: '2024-01-01',
+        contextWindow: 64000,
+        maxTokens: 4096,
+        supportsStreaming: true,
+        supportsVision: false,
+        supportsTools: true,
+        supportsJSON: true,
+    },
+};
+/**
+ * Get pricing information for a specific model.
+ *
+ * @param modelId - The model identifier (e.g., "gpt-4", "claude-3-opus-20240229")
+ * @returns Pricing data or null if not found
+ */
+function getModelPricing(modelId) {
+    const data = MODEL_PRICING_DATABASE[modelId];
+    return data?.pricing ?? null;
+}
+/**
+ * Get extended capabilities for a specific model.
+ *
+ * @param modelId - The model identifier
+ * @returns Extended capabilities or null if not found
+ */
+function getModelCapabilities(modelId) {
+    return MODEL_PRICING_DATABASE[modelId] ?? null;
+}
+/**
+ * Get all models with pricing data.
+ *
+ * @returns Array of model IDs with pricing information
+ */
+function getAllPricedModels() {
+    return Object.keys(MODEL_PRICING_DATABASE);
+}
+/**
+ * Find models by model family.
+ *
+ * @param family - The model family (e.g., "gpt-4", "claude-3")
+ * @returns Array of model IDs in the family
+ */
+function getModelsByFamily(family) {
+    return Object.keys(MODEL_PRICING_DATABASE).filter((modelId) => MODEL_PRICING_DATABASE[modelId]?.modelFamily === family);
+}
+/**
+ * User-defined pricing overrides.
+ * Allows users to specify custom pricing for models.
+ */
+const pricingOverrides = new Map();
+/**
+ * Set custom pricing for a model (overrides static database).
+ *
+ * @param modelId - The model identifier
+ * @param pricing - Custom pricing data
+ */
+function setPricingOverride(modelId, pricing) {
+    pricingOverrides.set(modelId, pricing);
+}
+/**
+ * Clear pricing override for a model.
+ *
+ * @param modelId - The model identifier
+ */
+function clearPricingOverride(modelId) {
+    pricingOverrides.delete(modelId);
+}
+/**
+ * Clear all pricing overrides.
+ */
+function clearAllPricingOverrides() {
+    pricingOverrides.clear();
+}
+/**
+ * Get pricing with user overrides applied.
+ *
+ * @param modelId - The model identifier
+ * @returns Pricing data (override if set, else from database)
+ */
+function getPricingWithOverrides(modelId) {
+    return pricingOverrides.get(modelId) ?? getModelPricing(modelId);
+}
+//# sourceMappingURL=model-pricing.js.map

package/dist/cjs/model-pricing.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"model-pricing.js","sourceRoot":"","sources":["../../src/model-pricing.ts"],"names":[],"mappings":";AAAA;;;;;;;GAOG;;AAuSH,0CAGC;AAQD,oDAEC;AAOD,gDAEC;AAQD,8CAIC;AAcD,gDAEC;AAOD,oDAEC;AAKD,4DAEC;AAQD,0DAEC;AAzVD;;;GAGG;AACH,MAAM,sBAAsB,GAAuD;IACjF,gBAAgB;IAChB,OAAO,EAAE;QACP,OAAO,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE;QACtC,OAAO,EAAE,EAAE,GAAG,EAAE,IAAI,EAAE,GAAG,EAAE,IAAI,EAAE;QACjC,YAAY,EAAE,EAAE;QAChB,WAAW,EAAE,OAAO;QACpB,WAAW,EAAE,YAAY;QACzB,aAAa,EAAE,IAAI;QACnB,SAAS,EAAE,IAAI;QACf,iBAAiB,EAAE,IAAI;QACvB,cAAc,EAAE,KAAK;QACrB,aAAa,EAAE,IAAI;QACnB,YAAY,EAAE,IAAI;KACnB;IACD,aAAa,EAAE;QACb,OAAO,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE;QACtC,OAAO,EAAE,EAAE,GAAG,EAAE,IAAI,EAAE,GAAG,EAAE,IAAI,EAAE;QACjC,YAAY,EAAE,EAAE;QAChB,WAAW,EAAE,OAAO;QACpB,WAAW,EAAE,YAAY;QACzB,aAAa,EAAE,MAAM;QACrB,SAAS,EAAE,IAAI;QACf,iBAAiB,EAAE,IAAI;QACvB,cAAc,EAAE,IAAI;QACpB,aAAa,EAAE,IAAI;QACnB,YAAY,EAAE,IAAI;KACnB;IACD,qBAAqB,EAAE;QACrB,OAAO,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE;QACtC,OAAO,EAAE,EAAE,GAAG,EAAE,IAAI,EAAE,GAAG,EAAE,IAAI,EAAE;QACjC,YAAY,EAAE,EAAE;QAChB,WAAW,EAAE,OAAO;QACpB,WAAW,EAAE,YAAY;QACzB,aAAa,EAAE,MAAM;QACrB,SAAS,EAAE,IAAI;QACf,iBAAiB,EAAE,IAAI;QACvB,cAAc,EAAE,IAAI;QACpB,aAAa,EAAE,IAAI;QACnB,YAAY,EAAE,IAAI;KACnB;IACD,eAAe,EAAE;QACf,OAAO,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE;QAC1C,OAAO,EAAE,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,IAAI,EAAE;QAChC,YAAY,EAAE,EAAE;QAChB,WAAW,EAAE,SAAS;QACtB,WAAW,EAAE,YAAY;QACzB,aAAa,EAAE,KAAK;QACpB,SAAS,EAAE,IAAI;QACf,iBAAiB,EAAE,IAAI;QACvB,cAAc,EAAE,KAAK;QACrB,aAAa,EAAE,IAAI;QACnB,YAAY,EAAE,IAAI;KACnB;IACD,QAAQ,EAAE;QACR,OAAO,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE;QACxC,OAAO,EAAE,EAAE,GAAG,EAAE,IAAI,EAAE,GAAG,EAAE,IAAI,EAAE;QACjC,YAAY,EAAE,EAAE;QAChB,WAAW,EAAE,OAAO;QACpB,WAAW,EAAE,YAAY;QACzB,aAAa,EAAE,MAAM;QACrB,SAAS,EAAE,IAAI;QACf,iBAAiB,EAAE,IAAI;QACvB,cAAc,EAAE,IAAI;QACpB,aAAa,EAAE,IAAI;QACnB,YAAY,EAAE,IAAI;KACnB;IACD,aAAa,EAAE;QACb,OAAO,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE;QAC3C,OAAO,EAAE,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,IAAI,EAAE;QAChC,YAAY,EAAE,EAAE;QAChB,WAAW,EAAE,OAAO;QACpB,WAAW,EAAE,YAAY;QACzB,aAAa,EAAE,MAAM;QACrB,SAAS,EAAE,KAAK;QAChB,iBAAiB,EAAE,IAAI;QACvB,cAAc,EAAE,IAAI;QACpB,aAAa,EAAE,IAAI;QACnB,YAAY,EAAE,IAAI;KACnB;IAED,mBAAmB;IACnB,4BAA4B,EAAE;QAC5B,OAAO,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE;QACxC,OAAO,EAAE,EAAE,GAAG,EAAE,IAAI,EAAE,GAAG,EAAE,IAAI,EAAE;QACjC,YAAY,EAAE,EAAE;QAChB,WAAW,EAAE,UAAU;QACvB,WAAW,EAAE,YAAY;QACzB,aAAa,EAAE,MAAM;QACrB,SAAS,EAAE,IAAI;QACf,iBAAiB,EAAE,IAAI;QACvB,cAAc,EAAE,IAAI;QACpB,aAAa,EAAE,IAAI;QACnB,YAAY,EAAE,KAAK;KACpB;IACD,wBAAwB,EAAE;QACxB,OAAO,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE;QACxC,OAAO,EAAE,EAAE,GAAG,EAAE,IAAI,EAAE,GAAG,EAAE,IAAI,EAAE;QACjC,YAAY,EAAE,EAAE;QAChB,WAAW,EAAE,UAAU;QACvB,WAAW,EAAE,YAAY;QACzB,aAAa,EAAE,MAAM;QACrB,SAAS,EAAE,IAAI;QACf,iBAAiB,EAAE,IAAI;QACvB,cAAc,EAAE,IAAI;QACpB,aAAa,EAAE,IAAI;QACnB,YAAY,EAAE,KAAK;KACpB;IACD,0BAA0B,EAAE;QAC1B,OAAO,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE;QACxC,OAAO,EAAE,EAAE,GAAG,EAAE,IAAI,EAAE,GAAG,EAAE,IAAI,EAAE;QACjC,YAAY,EAAE,EAAE;QAChB,WAAW,EAAE,UAAU;QACvB,WAAW,EAAE,YAAY;QACzB,aAAa,EAAE,MAAM;QACrB,SAAS,EAAE,IAAI;QACf,iBAAiB,EAAE,IAAI;QACvB,cAAc,EAAE,IAAI;QACpB,aAAa,EAAE,IAAI;QACnB,YAAY,EAAE,KAAK;KACpB;IACD,yBAAyB,EAAE;QACzB,OAAO,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE;QAC5C,OAAO,EAAE,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE;QAC/B,YAAY,EAAE,EAAE;QAChB,WAAW,EAAE,UAAU;QACvB,WAAW,EAAE,YAAY;QACzB,aAAa,EAAE,MAAM;QACrB,SAAS,EAAE,IAAI;QACf,iBAAiB,EAAE,IAAI;QACvB,cAAc,EAAE,IAAI;QACpB,aAAa,EAAE,IAAI;QACnB,YAAY,EAAE,KAAK;KACpB;IACD,2BAA2B,EAAE;QAC3B,OAAO,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE;QACxC,OAAO,EAAE,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,IAAI,EAAE;QAChC,YAAY,EAAE,EAAE;QAChB,WAAW,EAAE,UAAU;QACvB,WAAW,EAAE,YAAY;QACzB,aAAa,EAAE,MAAM;QACrB,SAAS,EAAE,IAAI;QACf,iBAAiB,EAAE,IAAI;QACvB,cAAc,EAAE,KAAK;QACrB,aAAa,EAAE,IAAI;QACnB,YAAY,EAAE,KAAK;KACpB;IAED,uBAAuB;IACvB,gBAAgB,EAAE;QAChB,OAAO,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE;QAC1C,OAAO,EAAE,EAAE,GAAG,EAAE,IAAI,EAAE,GAAG,EAAE,IAAI,EAAE;QACjC,YAAY,EAAE,EAAE;QAChB,WAAW,EAAE,YAAY;QACzB,WAAW,EAAE,YAAY;QACzB,aAAa,EAAE,OAAO;QACtB,SAAS,EAAE,IAAI;QACf,iBAAiB,EAAE,IAAI;QACvB,cAAc,EAAE,IAAI;QACpB,aAAa,EAAE,IAAI;QACnB,YAAY,EAAE,IAAI;KACnB;IACD,kBAAkB,EAAE;QAClB,OAAO,EAAE,EAAE,KAAK,EAAE,QAAQ,EAAE,MAAM,EAAE,MAAM,EAAE;QAC5C,OAAO,EAAE,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,IAAI,EAAE;QAChC,YAAY,EAAE,EAAE;QAChB,WAAW,EAAE,YAAY;QACzB,WAAW,EAAE,YAAY;QACzB,aAAa,EAAE,OAAO;QACtB,SAAS,EAAE,IAAI;QACf,iBAAiB,EAAE,IAAI;QACvB,cAAc,EAAE,IAAI;QACpB,aAAa,EAAE,IAAI;QACnB,YAAY,EAAE,IAAI;KACnB;IACD,qBAAqB,EAAE;QACrB,OAAO,EAAE,EAAE,KAAK,EAAE,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE;QAC9C,OAAO,EAAE,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE;QAC/B,YAAY,EAAE,EAAE;QAChB,WAAW,EAAE,YAAY;QACzB,WAAW,EAAE,YAAY;QACzB,aAAa,EAAE,OAAO;QACtB,SAAS,EAAE,IAAI;QACf,iBAAiB,EAAE,IAAI;QACvB,cAAc,EAAE,IAAI;QACpB,aAAa,EAAE,IAAI;QACnB,YAAY,EAAE,IAAI;KACnB;IAED,oBAAoB;IACpB,sBAAsB,EAAE;QACtB,OAAO,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE;QACxC,OAAO,EAAE,EAAE,GAAG,EAAE,IAAI,EAAE,GAAG,EAAE,IAAI,EAAE;QACjC,YAAY,EAAE,EAAE;QAChB,WAAW,EAAE,SAAS;QACtB,WAAW,EAAE,YAAY;QACzB,aAAa,EAAE,MAAM;QACrB,SAAS,EAAE,IAAI;QACf,iBAAiB,EAAE,IAAI;QACvB,cAAc,EAAE,KAAK;QACrB,aAAa,EAAE,IAAI;QACnB,YAAY,EAAE,IAAI;KACnB;IACD,uBAAuB,EAAE;QACvB,OAAO,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE;QAC1C,OAAO,EAAE,EAAE,GAAG,EAAE,IAAI,EAAE,GAAG,EAAE,IAAI,EAAE;QACjC,YAAY,EAAE,EAAE;QAChB,WAAW,EAAE,SAAS;QACtB,WAAW,EAAE,YAAY;QACzB,aAAa,EAAE,KAAK;QACpB,SAAS,EAAE,IAAI;QACf,iBAAiB,EAAE,IAAI;QACvB,cAAc,EAAE,KAAK;QACrB,aAAa,EAAE,IAAI;QACnB,YAAY,EAAE,IAAI;KACnB;IACD,sBAAsB,EAAE;QACtB,OAAO,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE;QACxC,OAAO,EAAE,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,IAAI,EAAE;QAChC,YAAY,EAAE,EAAE;QAChB,WAAW,EAAE,SAAS;QACtB,WAAW,EAAE,YAAY;QACzB,aAAa,EAAE,KAAK;QACpB,SAAS,EAAE,IAAI;QACf,iBAAiB,EAAE,IAAI;QACvB,cAAc,EAAE,KAAK;QACrB,aAAa,EAAE,IAAI;QACnB,YAAY,EAAE,IAAI;KACnB;IAED,kBAAkB;IAClB,eAAe,EAAE;QACf,OAAO,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE;QAC5C,OAAO,EAAE,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,IAAI,EAAE;QAChC,YAAY,EAAE,EAAE;QAChB,WAAW,EAAE,UAAU;QACvB,WAAW,EAAE,YAAY;QACzB,aAAa,EAAE,KAAK;QACpB,SAAS,EAAE,IAAI;QACf,iBAAiB,EAAE,IAAI;QACvB,cAAc,EAAE,KAAK;QACrB,aAAa,EAAE,IAAI;QACnB,YAAY,EAAE,IAAI;KACnB;IACD,gBAAgB,EAAE;QAChB,OAAO,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE;QAC5C,OAAO,EAAE,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,IAAI,EAAE;QAChC,YAAY,EAAE,EAAE;QAChB,WAAW,EAAE,UAAU;QACvB,WAAW,EAAE,YAAY;QACzB,aAAa,EAAE,KAAK;QACpB,SAAS,EAAE,IAAI;QACf,iBAAiB,EAAE,IAAI;QACvB,cAAc,EAAE,KAAK;QACrB,aAAa,EAAE,IAAI;QACnB,YAAY,EAAE,IAAI;KACnB;CACF,CAAC;AAEF;;;;;GAKG;AACH,SAAgB,eAAe,CAAC,OAAe;IAC7C,MAAM,IAAI,GAAG,sBAAsB,CAAC,OAAO,CAAC,CAAC;IAC7C,OAAO,IAAI,EAAE,OAAO,IAAI,IAAI,CAAC;AAC/B,CAAC;AAED;;;;;GAKG;AACH,SAAgB,oBAAoB,CAAC,OAAe;IAClD,OAAO,sBAAsB,CAAC,OAAO,CAAC,IAAI,IAAI,CAAC;AACjD,CAAC;AAED;;;;GAIG;AACH,SAAgB,kBAAkB;IAChC,OAAO,MAAM,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC;AAC7C,CAAC;AAED;;;;;GAKG;AACH,SAAgB,iBAAiB,CAAC,MAAc;IAC9C,OAAO,MAAM,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC,MAAM,CAC/C,CAAC,OAAO,EAAE,EAAE,CAAC,sBAAsB,CAAC,OAAO,CAAC,EAAE,WAAW,KAAK,MAAM,CACrE,CAAC;AACJ,CAAC;AAED;;;GAGG;AACH,MAAM,gBAAgB,GAAG,IAAI,GAAG,EAAwB,CAAC;AAEzD;;;;;GAKG;AACH,SAAgB,kBAAkB,CAAC,OAAe,EAAE,OAAqB;IACvE,gBAAgB,CAAC,GAAG,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;AACzC,CAAC;AAED;;;;GAIG;AACH,SAAgB,oBAAoB,CAAC,OAAe;IAClD,gBAAgB,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;AACnC,CAAC;AAED;;GAEG;AACH,SAAgB,wBAAwB;IACtC,gBAAgB,CAAC,KAAK,EAAE,CAAC;AAC3B,CAAC;AAED;;;;;GAKG;AACH,SAAgB,uBAAuB,CAAC,OAAe;IACrD,OAAO,gBAAgB,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,eAAe,CAAC,OAAO,CAAC,CAAC;AACnE,CAAC"}