@revenium/openai 1.0.8

package/dist/esm/index.js

@@ -0,0 +1,175 @@
+/**
+ * Revenium OpenAI Middleware for TypeScript
+ *
+ * This middleware tracks OpenAI usage and sends metrics to Revenium.
+ * Uses hybrid initialization: auto-initializes on import with graceful fallback to manual init.
+ *
+ * Environment Variables:
+ * REVENIUM_METERING_API_KEY=hak_your_api_key
+ * REVENIUM_METERING_BASE_URL=https://api.revenium.io/meter (optional)
+ * OPENAI_API_KEY=sk_your_openai_key
+ *
+ * Simple Usage (auto-initialization):
+ * import { patchOpenAIInstance } from 'revenium-middleware-openai-node';
+ * import OpenAI from 'openai';
+ *
+ * const openai = patchOpenAIInstance(new OpenAI());
+ * // Auto-initializes from environment variables
+ *
+ * Advanced Usage (explicit initialization):
+ * import { initializeReveniumFromEnv, patchOpenAIInstance } from 'revenium-middleware-openai-node';
+ * import OpenAI from 'openai';
+ *
+ * const result = initializeReveniumFromEnv();
+ * if (!result.success) {
+ *   throw new Error(result.message);
+ * }
+ * const openai = patchOpenAIInstance(new OpenAI());
+ *
+ * const response = await openai.chat.completions.create({
+ *   model: 'gpt-4',
+ *   messages: [{ role: 'user', content: 'Hello!' }],
+ *   usageMetadata: {
+ *     subscriber: {
+ *       id: 'user-123',
+ *       email: 'user@my-org.com'
+ *     },
+ *     organizationId: 'my-org',
+ *     productId: 'my-app'
+ *   }
+ * });
+ */
+// Import type augmentations to extend OpenAI types with usageMetadata
+import './types/openai-augmentation.js';
+// Import from new modular structure
+import { setConfig, validateConfig, initializeConfig, getConfig, getLogger, } from './core/config/index.js';
+import { patchOpenAI } from './core/wrapper/index.js';
+// Track initialization state
+let isInitialized = false;
+let autoInitAttempted = false;
+// Note: ExtendedChatCompletionCreateParams and ExtendedEmbeddingCreateParams are no longer exported
+// as they have been replaced with seamless TypeScript module augmentation. The usageMetadata field
+// is now natively available on OpenAI's types without requiring additional imports.
+// Export tracking functions
+export { trackUsageAsync, trackEmbeddingsUsageAsync } from './core/tracking/index.js';
+// Export provider detection functions
+export { detectProvider, hasAzureConfig, validateAzureConfig, getProviderMetadata, } from './core/providers/index.js';
+export { getProviderInfo } from './core/wrapper/index.js';
+// Export Azure model resolution functions
+export { resolveAzureModelName, clearModelNameCache, getModelNameCacheStats, batchResolveModelNames, wouldTransformDeploymentName, } from './utils/azure-model-resolver.js';
+// Global logger
+const logger = getLogger();
+/**
+ * Initialize Revenium middleware with configuration
+ */
+export function initializeRevenium(config) {
+    // Check if already initialized to prevent duplicate initialization
+    if (isInitialized) {
+        return {
+            success: true,
+            message: 'Revenium middleware already initialized',
+        };
+    }
+    try {
+        // Apply default base URL if not provided
+        const configWithDefaults = {
+            ...config,
+            reveniumBaseUrl: config.reveniumBaseUrl || 'https://api.revenium.io/meter',
+        };
+        validateConfig(configWithDefaults);
+        setConfig(configWithDefaults);
+        // Mark as initialized
+        isInitialized = true;
+        // Patch OpenAI prototype methods
+        patchOpenAI();
+        return {
+            success: true,
+            message: 'Revenium middleware initialized successfully',
+        };
+    }
+    catch (error) {
+        isInitialized = false;
+        return {
+            success: false,
+            message: error instanceof Error ? error.message : 'Unknown initialization error',
+        };
+    }
+}
+/**
+ * Initialize Revenium middleware from environment variables
+ */
+export function initializeReveniumFromEnv() {
+    // Check if already initialized to prevent duplicate initialization
+    if (isInitialized) {
+        return {
+            success: true,
+            message: 'Revenium middleware already initialized',
+        };
+    }
+    const envSuccess = initializeConfig();
+    if (!envSuccess) {
+        isInitialized = false;
+        return {
+            success: false,
+            message: 'Failed to load configuration from environment variables. Check REVENIUM_METERING_API_KEY and REVENIUM_METERING_BASE_URL.',
+        };
+    }
+    // Mark as initialized
+    isInitialized = true;
+    // Patch OpenAI prototype methods
+    patchOpenAI();
+    return {
+        success: true,
+        message: 'Revenium middleware initialized from environment',
+    };
+}
+/**
+ * Manually patch an OpenAI instance (for advanced use cases)
+ */
+export { patchOpenAIInstance } from './core/wrapper/index.js';
+/**
+ * Auto-initialization with graceful fallback
+ * Attempts to initialize from environment variables on module load.
+ * If it fails, logs a debug message but doesn't throw - allows manual configuration later.
+ */
+function attemptAutoInitialization() {
+    if (autoInitAttempted || isInitialized)
+        return;
+    autoInitAttempted = true;
+    try {
+        const result = initializeReveniumFromEnv();
+        if (result.success) {
+            // Auto-init succeeded - log debug message
+            logger.debug('Revenium middleware auto-initialized from environment variables');
+        }
+        else {
+            // Auto-init failed - log debug message but don't throw
+            logger.debug('Auto-initialization failed, manual initialization required:', result.message);
+        }
+    }
+    catch (error) {
+        // Unexpected error during auto-init - log but don't throw
+        logger.debug('Auto-initialization encountered error:', error instanceof Error ? error.message : String(error));
+    }
+}
+/**
+ * Check if middleware has been initialized
+ */
+export function isReveniumInitialized() {
+    return isInitialized;
+}
+/**
+ * Get detailed initialization status
+ */
+export function getInitializationStatus() {
+    const config = getConfig();
+    return {
+        initialized: isInitialized,
+        hasConfig: !!config,
+        hasApiKey: !!config?.reveniumApiKey,
+        autoInitAttempted,
+    };
+}
+// Perform auto-initialization when module is imported
+attemptAutoInitialization();
+//# sourceMappingURL=index.js.map

package/dist/esm/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAwCG;AAEH,sEAAsE;AACtE,OAAO,gCAAgC,CAAC;AAExC,oCAAoC;AACpC,OAAO,EACL,SAAS,EACT,cAAc,EACd,gBAAgB,EAChB,SAAS,EACT,SAAS,GACV,MAAM,wBAAwB,CAAC;AAChC,OAAO,EAAE,WAAW,EAAuB,MAAM,yBAAyB,CAAC;AAG3E,6BAA6B;AAC7B,IAAI,aAAa,GAAG,KAAK,CAAC;AAC1B,IAAI,iBAAiB,GAAG,KAAK,CAAC;AA6G9B,oGAAoG;AACpG,mGAAmG;AACnG,oFAAoF;AAEpF,4BAA4B;AAC5B,OAAO,EAAE,eAAe,EAAE,yBAAyB,EAAE,MAAM,0BAA0B,CAAC;AAEtF,sCAAsC;AACtC,OAAO,EACL,cAAc,EACd,cAAc,EACd,mBAAmB,EACnB,mBAAmB,GACpB,MAAM,2BAA2B,CAAC;AACnC,OAAO,EAAE,eAAe,EAAE,MAAM,yBAAyB,CAAC;AAE1D,0CAA0C;AAC1C,OAAO,EACL,qBAAqB,EACrB,mBAAmB,EACnB,sBAAsB,EACtB,sBAAsB,EACtB,4BAA4B,GAC7B,MAAM,iCAAiC,CAAC;AAEzC,gBAAgB;AAChB,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;AAE3B;;GAEG;AACH,MAAM,UAAU,kBAAkB,CAAC,MAAsB;IAIvD,mEAAmE;IACnE,IAAI,aAAa,EAAE,CAAC;QAClB,OAAO;YACL,OAAO,EAAE,IAAI;YACb,OAAO,EAAE,yCAAyC;SACnD,CAAC;IACJ,CAAC;IAED,IAAI,CAAC;QACH,yCAAyC;QACzC,MAAM,kBAAkB,GAAG;YACzB,GAAG,MAAM;YACT,eAAe,EAAE,MAAM,CAAC,eAAe,IAAI,+BAA+B;SAC3E,CAAC;QAEF,cAAc,CAAC,kBAAkB,CAAC,CAAC;QACnC,SAAS,CAAC,kBAAkB,CAAC,CAAC;QAE9B,sBAAsB;QACtB,aAAa,GAAG,IAAI,CAAC;QAErB,iCAAiC;QACjC,WAAW,EAAE,CAAC;QAEd,OAAO;YACL,OAAO,EAAE,IAAI;YACb,OAAO,EAAE,8CAA8C;SACxD,CAAC;IACJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,aAAa,GAAG,KAAK,CAAC;QACtB,OAAO;YACL,OAAO,EAAE,KAAK;YACd,OAAO,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,8BAA8B;SACjF,CAAC;IACJ,CAAC;AACH,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,yBAAyB;IAIvC,mEAAmE;IACnE,IAAI,aAAa,EAAE,CAAC;QAClB,OAAO;YACL,OAAO,EAAE,IAAI;YACb,OAAO,EAAE,yCAAyC;SACnD,CAAC;IACJ,CAAC;IAED,MAAM,UAAU,GAAG,gBAAgB,EAAE,CAAC;IAEtC,IAAI,CAAC,UAAU,EAAE,CAAC;QAChB,aAAa,GAAG,KAAK,CAAC;QACtB,OAAO;YACL,OAAO,EAAE,KAAK;YACd,OAAO,EACL,0HAA0H;SAC7H,CAAC;IACJ,CAAC;IAED,sBAAsB;IACtB,aAAa,GAAG,IAAI,CAAC;IAErB,iCAAiC;IACjC,WAAW,EAAE,CAAC;IAEd,OAAO;QACL,OAAO,EAAE,IAAI;QACb,OAAO,EAAE,kDAAkD;KAC5D,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,OAAO,EAAE,mBAAmB,EAAE,MAAM,yBAAyB,CAAC;AAE9D;;;;GAIG;AACH,SAAS,yBAAyB;IAChC,IAAI,iBAAiB,IAAI,aAAa;QAAE,OAAO;IAE/C,iBAAiB,GAAG,IAAI,CAAC;IACzB,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,yBAAyB,EAAE,CAAC;QAC3C,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;YACnB,0CAA0C;YAC1C,MAAM,CAAC,KAAK,CAAC,iEAAiE,CAAC,CAAC;QAClF,CAAC;aAAM,CAAC;YACN,uDAAuD;YACvD,MAAM,CAAC,KAAK,CAAC,6DAA6D,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC;QAC9F,CAAC;IACH,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,0DAA0D;QAC1D,MAAM,CAAC,KAAK,CACV,wCAAwC,EACxC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CACvD,CAAC;IACJ,CAAC;AACH,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,qBAAqB;IACnC,OAAO,aAAa,CAAC;AACvB,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,uBAAuB;IAMrC,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;IAC3B,OAAO;QACL,WAAW,EAAE,aAAa;QAC1B,SAAS,EAAE,CAAC,CAAC,MAAM;QACnB,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,cAAc;QACnC,iBAAiB;KAClB,CAAC;AACJ,CAAC;AAED,sDAAsD;AACtD,yBAAyB,EAAE,CAAC"}

package/dist/esm/types/function-parameters.js

@@ -0,0 +1,13 @@
+/**
+ * Function Parameter Types
+ *
+ * Comprehensive type definitions for function parameters throughout the middleware.
+ * These interfaces provide type safety by replacing 'any' types with proper,
+ * well-documented interfaces that match OpenAI API structures and internal requirements.
+ *
+ * @fileoverview Type-safe function parameter definitions
+ * @author Revenium
+ * @since 1.0.0
+ */
+export {};
+//# sourceMappingURL=function-parameters.js.map

package/dist/esm/types/function-parameters.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"function-parameters.js","sourceRoot":"","sources":["../../../src/types/function-parameters.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;GAUG"}

package/dist/esm/types/index.js

@@ -0,0 +1,32 @@
+/**
+ * Core Types Module
+ *
+ * Central type definitions for the Revenium OpenAI middleware.
+ * This module exports all core types used throughout the application.
+ */
+// Re-export function parameter types
+export * from './function-parameters.js';
+// Re-export Responses API types
+export * from './responses-api.js';
+/**
+ * Supported AI providers
+ *
+ * Enumeration of AI providers supported by the Revenium middleware.
+ * Used for automatic detection, routing, and provider-specific handling.
+ *
+ * @public
+ * @example
+ * ```typescript
+ * if (providerInfo.provider === Provider.AZURE_OPENAI) {
+ *   console.log('Using Azure OpenAI');
+ * }
+ * ```
+ */
+export var Provider;
+(function (Provider) {
+    /** Standard OpenAI API */
+    Provider["OPENAI"] = "OPENAI";
+    /** Azure OpenAI Service */
+    Provider["AZURE_OPENAI"] = "AZURE_OPENAI";
+})(Provider || (Provider = {}));
+//# sourceMappingURL=index.js.map

package/dist/esm/types/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/types/index.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,qCAAqC;AACrC,cAAc,0BAA0B,CAAC;AAEzC,gCAAgC;AAChC,cAAc,oBAAoB,CAAC;AA0InC;;;;;;;;;;;;;GAaG;AACH,MAAM,CAAN,IAAY,QAKX;AALD,WAAY,QAAQ;IAClB,0BAA0B;IAC1B,6BAAiB,CAAA;IACjB,2BAA2B;IAC3B,yCAA6B,CAAA;AAC/B,CAAC,EALW,QAAQ,KAAR,QAAQ,QAKnB"}

package/dist/esm/types/openai-augmentation.js

@@ -0,0 +1,54 @@
+/**
+ * TypeScript module augmentation for OpenAI SDK
+ *
+ * This file extends OpenAI's existing types to include the usageMetadata field
+ * through TypeScript's declaration merging feature. This provides seamless
+ * integration with the OpenAI SDK, allowing developers to use usageMetadata
+ * directly in OpenAI API calls without type casting or additional imports.
+ *
+ * The augmentation covers all major OpenAI API endpoints including:
+ * - Chat completions (streaming and non-streaming)
+ * - Embeddings
+ * - Future API endpoints as they become available
+ *
+ * @fileoverview OpenAI SDK type augmentation for Revenium middleware
+ * @author Revenium
+ * @since 1.0.0
+ *
+ * @example Basic usage with chat completions
+ * ```typescript
+ * import 'revenium-middleware-openai-node';
+ * import OpenAI from 'openai';
+ *
+ * const openai = new OpenAI();
+ *
+ * const response = await openai.chat.completions.create({
+ *   model: 'gpt-4o-mini',
+ *   messages: [{ role: 'user', content: 'Hello!' }],
+ *   usageMetadata: {  // TypeScript recognizes this natively
+ *     subscriber: {
+ *       id: 'user-123',
+ *       email: 'user@my-company.com'
+ *     },
+ *     organizationId: 'my-company',
+ *     productId: 'chat-app',
+ *     taskType: 'customer-support'
+ *   }
+ * });
+ * ```
+ *
+ * @example Usage with embeddings
+ * ```typescript
+ * const embedding = await openai.embeddings.create({
+ *   model: 'text-embedding-ada-002',
+ *   input: 'Text to embed',
+ *   usageMetadata: {
+ *     subscriber: { id: 'user-456' },
+ *     productId: 'search-engine',
+ *     taskType: 'document-indexing'
+ *   }
+ * });
+ * ```
+ */
+export {};
+//# sourceMappingURL=openai-augmentation.js.map

package/dist/esm/types/openai-augmentation.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"openai-augmentation.js","sourceRoot":"","sources":["../../../src/types/openai-augmentation.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAmDG"}

package/dist/esm/types/responses-api.js

@@ -0,0 +1,26 @@
+/**
+ * Types for OpenAI Responses API support
+ *
+ * This module defines types for the new OpenAI Responses API that replaces
+ * the traditional Chat Completions API. The Responses API provides a unified
+ * interface for building agent-like applications with built-in tools and capabilities.
+ *
+ * Reference: https://platform.openai.com/docs/guides/migrate-to-responses
+ */
+/**
+ * Type guard to check if a request is for Responses API
+ */
+export function isResponsesRequest(params) {
+    return typeof params === 'object' && params !== null && 'input' in params && 'model' in params;
+}
+/**
+ * Type guard to check if a response is from Responses API
+ */
+export function isResponsesResponse(response) {
+    return (typeof response === 'object' &&
+        response !== null &&
+        'id' in response &&
+        'model' in response &&
+        ('output' in response || 'usage' in response));
+}
+//# sourceMappingURL=responses-api.js.map

package/dist/esm/types/responses-api.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"responses-api.js","sourceRoot":"","sources":["../../../src/types/responses-api.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAwOH;;GAEG;AACH,MAAM,UAAU,kBAAkB,CAAC,MAAe;IAChD,OAAO,OAAO,MAAM,KAAK,QAAQ,IAAI,MAAM,KAAK,IAAI,IAAI,OAAO,IAAI,MAAM,IAAI,OAAO,IAAI,MAAM,CAAC;AACjG,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,mBAAmB,CAAC,QAAiB;IACnD,OAAO,CACL,OAAO,QAAQ,KAAK,QAAQ;QAC5B,QAAQ,KAAK,IAAI;QACjB,IAAI,IAAI,QAAQ;QAChB,OAAO,IAAI,QAAQ;QACnB,CAAC,QAAQ,IAAI,QAAQ,IAAI,OAAO,IAAI,QAAQ,CAAC,CAC9C,CAAC;AACJ,CAAC"}

package/dist/esm/utils/azure-model-resolver.js

@@ -0,0 +1,204 @@
+import { getLogger } from '../core/config/index.js';
+import { knownModels } from './constants.js';
+/**
+ * Azure Model Name Resolution Module
+ *
+ * This module maps Azure deployment names to LiteLLM-compatible model names for accurate pricing.
+ * Based on learnings from the Python implementation, it uses heuristic pattern matching
+ * with fallback strategies to ensure reliable model name resolution.
+ *
+ * Key patterns observed in real Azure deployments:
+ * - "gpt-4o-2024-11-20" → "gpt-4o"
+ * - "text-embedding-3-large" → "text-embedding-3-large" (exact match)
+ * - "o4-mini" → "gpt-4o-mini"
+ * - "gpt4o-prod" → "gpt-4o"
+ * - "gpt-35-turbo-dev" → "gpt-3.5-turbo"
+ */
+/**
+ * In-memory cache for resolved model names
+ * Using Map for thread-safe operations in Node.js
+ */
+const modelNameCache = new Map();
+/**
+ * Cache for failed resolution attempts to avoid repeated warnings
+ */
+const failedResolutionCache = new Set();
+// Global logger
+const logger = getLogger();
+/**
+ * Resolve Azure deployment name to LiteLLM-compatible model name
+ *
+ * @param deploymentName - Azure deployment name
+ * @param useCache - Whether to use cached results (default: true)
+ * @returns LiteLLM-compatible model name
+ */
+export function resolveAzureModelName(deploymentName, useCache = true) {
+    if (!deploymentName) {
+        logger.warn('Empty deployment name provided to model resolver');
+        return deploymentName;
+    }
+    // Check cache first
+    if (useCache && modelNameCache.has(deploymentName)) {
+        const cachedResult = modelNameCache.get(deploymentName);
+        logger.debug('Model name resolved from cache', {
+            deployment: deploymentName,
+            resolved: cachedResult,
+        });
+        return cachedResult;
+    }
+    try {
+        const resolved = resolveModelNameHeuristic(deploymentName);
+        // Cache the result
+        if (useCache) {
+            modelNameCache.set(deploymentName, resolved);
+        }
+        // Log successful resolution
+        if (resolved !== deploymentName) {
+            logger.debug('Model name resolved via heuristics', {
+                deployment: deploymentName,
+                resolved,
+            });
+        }
+        return resolved;
+    }
+    catch (error) {
+        logger.error('Error during model name resolution', {
+            deployment: deploymentName,
+            error: error instanceof Error ? error.message : String(error),
+        });
+        // Fallback to deployment name
+        return deploymentName;
+    }
+}
+/**
+ * Heuristic pattern matching for Azure deployment names
+ * Based on real-world patterns observed in the Python implementation
+ *
+ * @param deploymentName - Azure deployment name
+ * @returns LiteLLM-compatible model name
+ */
+function resolveModelNameHeuristic(deploymentName) {
+    const nameLower = deploymentName.toLowerCase();
+    // GPT-4o family - handle both "gpt-4o" and "o4" patterns
+    if (/gpt-?4o/.test(nameLower) || /o4/.test(nameLower)) {
+        if (/mini/.test(nameLower))
+            return 'gpt-4o-mini';
+        return 'gpt-4o';
+    }
+    // GPT-4 family (non-omni)
+    if (/gpt-?4(?!o)/.test(nameLower)) {
+        if (/turbo/.test(nameLower))
+            return 'gpt-4-turbo';
+        if (/vision/.test(nameLower) || /v/.test(nameLower))
+            return 'gpt-4-vision-preview';
+        return 'gpt-4';
+    }
+    // GPT-3.5 family
+    if (/gpt-?3\.?5/.test(nameLower) || /35-turbo/.test(nameLower) || /gpt-35/.test(nameLower)) {
+        if (/instruct/.test(nameLower))
+            return 'gpt-3.5-turbo-instruct';
+        return 'gpt-3.5-turbo';
+    }
+    // Embedding models - exact matches work well
+    if (/embed/.test(nameLower)) {
+        if (/text-embedding-3-large/.test(nameLower))
+            return 'text-embedding-3-large';
+        if (/text-embedding-3-small/.test(nameLower))
+            return 'text-embedding-3-small';
+        if (/text-embedding-ada-002/.test(nameLower) || /ada-002/.test(nameLower))
+            return 'text-embedding-ada-002';
+        if (/3-large/.test(nameLower))
+            return 'text-embedding-3-large';
+        if (/3-small/.test(nameLower))
+            return 'text-embedding-3-small';
+    }
+    // Ada-002 pattern (can appear without "embed" in deployment name)
+    if (/ada-002/.test(nameLower))
+        return 'text-embedding-ada-002';
+    // DALL-E models
+    if (/dall-?e/.test(nameLower)) {
+        if (/3/.test(nameLower))
+            return 'dall-e-3';
+        if (/2/.test(nameLower))
+            return 'dall-e-2';
+    }
+    // Whisper models
+    if (/whisper/.test(nameLower))
+        return 'whisper-1';
+    // TTS models
+    if (/tts/.test(nameLower)) {
+        if (/hd/.test(nameLower))
+            return 'tts-1-hd';
+        return 'tts-1';
+    }
+    if (knownModels.includes(nameLower))
+        return nameLower;
+    // No heuristic match found - log warning and use deployment name
+    if (!failedResolutionCache.has(deploymentName)) {
+        logger.warn(`⚠️ No heuristic match for Azure deployment: ${deploymentName}. Using deployment name for pricing. Consider adding pattern to azure-model-resolver.ts`);
+        failedResolutionCache.add(deploymentName);
+    }
+    return deploymentName;
+}
+/**
+ * Clear the model name cache
+ * Useful for testing or when deployment configurations change
+ */
+export function clearModelNameCache() {
+    modelNameCache.clear();
+    failedResolutionCache.clear();
+    getLogger().debug('Model name cache cleared');
+}
+/**
+ * Get cache statistics for monitoring
+ */
+export function getModelNameCacheStats() {
+    return {
+        cacheSize: modelNameCache.size,
+        failedResolutionCount: failedResolutionCache.size,
+        cacheEntries: Array.from(modelNameCache.entries()).map(([deployment, resolved]) => ({
+            deployment,
+            resolved,
+        })),
+    };
+}
+/**
+ * Batch resolve multiple deployment names
+ * Useful for pre-warming cache or bulk operations
+ *
+ * @param deploymentNames - Array of deployment names to resolve
+ * @returns Map of deployment name to resolved model name
+ */
+export function batchResolveModelNames(deploymentNames) {
+    const results = new Map();
+    logger.debug('Batch resolving model names', {
+        count: deploymentNames.length,
+        deployments: deploymentNames,
+    });
+    for (const deployment of deploymentNames) {
+        try {
+            const resolved = resolveAzureModelName(deployment);
+            results.set(deployment, resolved);
+        }
+        catch (error) {
+            logger.error('Error in batch resolution', {
+                deployment,
+                error: error instanceof Error ? error.message : String(error),
+            });
+            results.set(deployment, deployment); // Fallback to original name
+        }
+    }
+    return results;
+}
+/**
+ * Check if a deployment name would be resolved to a different model name
+ * Useful for validation and testing
+ *
+ * @param deploymentName - Azure deployment name
+ * @returns true if the deployment name would be transformed
+ */
+export function wouldTransformDeploymentName(deploymentName) {
+    const resolved = resolveAzureModelName(deploymentName, false); // Don't use cache for this check
+    return resolved !== deploymentName;
+}
+//# sourceMappingURL=azure-model-resolver.js.map

package/dist/esm/utils/azure-model-resolver.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"azure-model-resolver.js","sourceRoot":"","sources":["../../../src/utils/azure-model-resolver.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,yBAAyB,CAAC;AACpD,OAAO,EAAE,WAAW,EAAE,MAAM,gBAAgB,CAAC;AAE7C;;;;;;;;;;;;;GAaG;AAEH;;;GAGG;AACH,MAAM,cAAc,GAAG,IAAI,GAAG,EAAkB,CAAC;AAEjD;;GAEG;AACH,MAAM,qBAAqB,GAAG,IAAI,GAAG,EAAU,CAAC;AAEhD,gBAAgB;AAChB,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;AAE3B;;;;;;GAMG;AACH,MAAM,UAAU,qBAAqB,CAAC,cAAsB,EAAE,WAAoB,IAAI;IACpF,IAAI,CAAC,cAAc,EAAE,CAAC;QACpB,MAAM,CAAC,IAAI,CAAC,kDAAkD,CAAC,CAAC;QAChE,OAAO,cAAc,CAAC;IACxB,CAAC;IAED,oBAAoB;IACpB,IAAI,QAAQ,IAAI,cAAc,CAAC,GAAG,CAAC,cAAc,CAAC,EAAE,CAAC;QACnD,MAAM,YAAY,GAAG,cAAc,CAAC,GAAG,CAAC,cAAc,CAAE,CAAC;QACzD,MAAM,CAAC,KAAK,CAAC,gCAAgC,EAAE;YAC7C,UAAU,EAAE,cAAc;YAC1B,QAAQ,EAAE,YAAY;SACvB,CAAC,CAAC;QACH,OAAO,YAAY,CAAC;IACtB,CAAC;IAED,IAAI,CAAC;QACH,MAAM,QAAQ,GAAG,yBAAyB,CAAC,cAAc,CAAC,CAAC;QAE3D,mBAAmB;QACnB,IAAI,QAAQ,EAAE,CAAC;YACb,cAAc,CAAC,GAAG,CAAC,cAAc,EAAE,QAAQ,CAAC,CAAC;QAC/C,CAAC;QAED,4BAA4B;QAC5B,IAAI,QAAQ,KAAK,cAAc,EAAE,CAAC;YAChC,MAAM,CAAC,KAAK,CAAC,oCAAoC,EAAE;gBACjD,UAAU,EAAE,cAAc;gBAC1B,QAAQ;aACT,CAAC,CAAC;QACL,CAAC;QAED,OAAO,QAAQ,CAAC;IAClB,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,CAAC,KAAK,CAAC,oCAAoC,EAAE;YACjD,UAAU,EAAE,cAAc;YAC1B,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;SAC9D,CAAC,CAAC;QAEH,8BAA8B;QAC9B,OAAO,cAAc,CAAC;IACxB,CAAC;AACH,CAAC;AAED;;;;;;GAMG;AACH,SAAS,yBAAyB,CAAC,cAAsB;IACvD,MAAM,SAAS,GAAG,cAAc,CAAC,WAAW,EAAE,CAAC;IAE/C,yDAAyD;IACzD,IAAI,SAAS,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC;QACtD,IAAI,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC;YAAE,OAAO,aAAa,CAAC;QACjD,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,0BAA0B;IAC1B,IAAI,aAAa,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC;QAClC,IAAI,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC;YAAE,OAAO,aAAa,CAAC;QAClD,IAAI,QAAQ,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC;YAAE,OAAO,sBAAsB,CAAC;QACnF,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,iBAAiB;IACjB,IAAI,YAAY,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,UAAU,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,QAAQ,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC;QAC3F,IAAI,UAAU,CAAC,IAAI,CAAC,SAAS,CAAC;YAAE,OAAO,wBAAwB,CAAC;QAChE,OAAO,eAAe,CAAC;IACzB,CAAC;IAED,6CAA6C;IAC7C,IAAI,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC;QAC5B,IAAI,wBAAwB,CAAC,IAAI,CAAC,SAAS,CAAC;YAAE,OAAO,wBAAwB,CAAC;QAC9E,IAAI,wBAAwB,CAAC,IAAI,CAAC,SAAS,CAAC;YAAE,OAAO,wBAAwB,CAAC;QAC9E,IAAI,wBAAwB,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,SAAS,CAAC,IAAI,CAAC,SAAS,CAAC;YACvE,OAAO,wBAAwB,CAAC;QAClC,IAAI,SAAS,CAAC,IAAI,CAAC,SAAS,CAAC;YAAE,OAAO,wBAAwB,CAAC;QAC/D,IAAI,SAAS,CAAC,IAAI,CAAC,SAAS,CAAC;YAAE,OAAO,wBAAwB,CAAC;IACjE,CAAC;IAED,kEAAkE;IAClE,IAAI,SAAS,CAAC,IAAI,CAAC,SAAS,CAAC;QAAE,OAAO,wBAAwB,CAAC;IAE/D,gBAAgB;IAChB,IAAI,SAAS,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC;QAC9B,IAAI,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC;YAAE,OAAO,UAAU,CAAC;QAC3C,IAAI,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC;YAAE,OAAO,UAAU,CAAC;IAC7C,CAAC;IAED,iBAAiB;IACjB,IAAI,SAAS,CAAC,IAAI,CAAC,SAAS,CAAC;QAAE,OAAO,WAAW,CAAC;IAElD,aAAa;IACb,IAAI,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC;QAC1B,IAAI,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC;YAAE,OAAO,UAAU,CAAC;QAC5C,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,IAAI,WAAW,CAAC,QAAQ,CAAC,SAAS,CAAC;QAAE,OAAO,SAAS,CAAC;IAEtD,iEAAiE;IACjE,IAAI,CAAC,qBAAqB,CAAC,GAAG,CAAC,cAAc,CAAC,EAAE,CAAC;QAC/C,MAAM,CAAC,IAAI,CACT,+CAA+C,cAAc,yFAAyF,CACvJ,CAAC;QACF,qBAAqB,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;IAC5C,CAAC;IACD,OAAO,cAAc,CAAC;AACxB,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,mBAAmB;IACjC,cAAc,CAAC,KAAK,EAAE,CAAC;IACvB,qBAAqB,CAAC,KAAK,EAAE,CAAC;IAC9B,SAAS,EAAE,CAAC,KAAK,CAAC,0BAA0B,CAAC,CAAC;AAChD,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,sBAAsB;IAKpC,OAAO;QACL,SAAS,EAAE,cAAc,CAAC,IAAI;QAC9B,qBAAqB,EAAE,qBAAqB,CAAC,IAAI;QACjD,YAAY,EAAE,KAAK,CAAC,IAAI,CAAC,cAAc,CAAC,OAAO,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,UAAU,EAAE,QAAQ,CAAC,EAAE,EAAE,CAAC,CAAC;YAClF,UAAU;YACV,QAAQ;SACT,CAAC,CAAC;KACJ,CAAC;AACJ,CAAC;AAED;;;;;;GAMG;AACH,MAAM,UAAU,sBAAsB,CAAC,eAAyB;IAC9D,MAAM,OAAO,GAAG,IAAI,GAAG,EAAkB,CAAC;IAC1C,MAAM,CAAC,KAAK,CAAC,6BAA6B,EAAE;QAC1C,KAAK,EAAE,eAAe,CAAC,MAAM;QAC7B,WAAW,EAAE,eAAe;KAC7B,CAAC,CAAC;IAEH,KAAK,MAAM,UAAU,IAAI,eAAe,EAAE,CAAC;QACzC,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,qBAAqB,CAAC,UAAU,CAAC,CAAC;YACnD,OAAO,CAAC,GAAG,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;QACpC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC,KAAK,CAAC,2BAA2B,EAAE;gBACxC,UAAU;gBACV,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;aAC9D,CAAC,CAAC;YACH,OAAO,CAAC,GAAG,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC,CAAC,4BAA4B;QACnE,CAAC;IACH,CAAC;IACD,OAAO,OAAO,CAAC;AACjB,CAAC;AAED;;;;;;GAMG;AACH,MAAM,UAAU,4BAA4B,CAAC,cAAsB;IACjE,MAAM,QAAQ,GAAG,qBAAqB,CAAC,cAAc,EAAE,KAAK,CAAC,CAAC,CAAC,iCAAiC;IAChG,OAAO,QAAQ,KAAK,cAAc,CAAC;AACrC,CAAC"}

package/dist/esm/utils/constants.js

@@ -0,0 +1,21 @@
+// Direct match check for known LiteLLM model names
+export const knownModels = [
+    'gpt-4o',
+    'gpt-4o-mini',
+    'gpt-4',
+    'gpt-4-turbo',
+    'gpt-4-vision-preview',
+    'gpt-3.5-turbo',
+    'gpt-3.5-turbo-instruct',
+    'text-embedding-3-large',
+    'text-embedding-3-small',
+    'text-embedding-ada-002',
+    'dall-e-3',
+    'dall-e-2',
+    'whisper-1',
+    'tts-1',
+    'tts-1-hd'
+];
+export const MESSAGE_PATTERNS_TYPE_NETWORK = ["network", "timeout", "ECONNRESET"];
+export const ERROR_MESSAGE_PATTERNS_TYPE_CONFIG = ["config", "key", "unauthorized"];
+//# sourceMappingURL=constants.js.map

package/dist/esm/utils/constants.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"constants.js","sourceRoot":"","sources":["../../../src/utils/constants.ts"],"names":[],"mappings":"AAAA,mDAAmD;AACnD,MAAM,CAAC,MAAM,WAAW,GAAG;IAC3B,QAAQ;IACR,aAAa;IACb,OAAO;IACP,aAAa;IACb,sBAAsB;IACtB,eAAe;IACf,wBAAwB;IACxB,wBAAwB;IACxB,wBAAwB;IACxB,wBAAwB;IACxB,UAAU;IACV,UAAU;IACV,WAAW;IACX,OAAO;IACP,UAAU;CACT,CAAC;AAEF,MAAM,CAAC,MAAM,6BAA6B,GAAG,CAAC,SAAS,EAAE,SAAS,EAAE,YAAY,CAAC,CAAC;AAClF,MAAM,CAAC,MAAM,kCAAkC,GAAG,CAAC,QAAQ,EAAE,KAAK,EAAE,cAAc,CAAC,CAAC"}