@revenium/openai 1.0.8

package/dist/cjs/core/tracking/payload-builder.js

@@ -0,0 +1,107 @@
+"use strict";
+/**
+ * Payload Builder Module
+ *
+ * Handles construction of Revenium API payloads.
+ * Extracted from tracking.ts for single responsibility.
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.buildPayload = buildPayload;
+const crypto_1 = require("crypto");
+const index_js_1 = require("../config/index.js");
+const stop_reason_mapper_js_1 = require("../../utils/stop-reason-mapper.js");
+const metadata_builder_js_1 = require("../../utils/metadata-builder.js");
+const azure_model_resolver_js_1 = require("../../utils/azure-model-resolver.js");
+const index_js_2 = require("../providers/index.js");
+// Global logger
+const logger = (0, index_js_1.getLogger)();
+/**
+ * Build payload for Revenium API
+ *
+ * This shared payload builder eliminates payload duplication between
+ * chat completions and embeddings. Handles both CHAT and EMBED operation types.
+ *
+ * @param operationType - Type of operation (CHAT or EMBED)
+ * @param response - API response from OpenAI/Azure
+ * @param request - Original request parameters
+ * @param startTime - Request start timestamp
+ * @param duration - Request duration in milliseconds
+ * @param providerInfo - Provider information for Azure support
+ * @returns Constructed payload for Revenium API
+ */
+function buildPayload(operationType, response, request, startTime, duration, providerInfo) {
+    const now = new Date().toISOString();
+    const requestTime = new Date(startTime).toISOString();
+    const usage = response.usage;
+    // Resolve model name for Azure deployments
+    const originalModel = response.model;
+    const resolvedModel = providerInfo?.isAzure
+        ? (0, azure_model_resolver_js_1.resolveAzureModelName)(originalModel)
+        : originalModel;
+    // Get provider metadata
+    const providerMetadata = providerInfo
+        ? (0, index_js_2.getProviderMetadata)(providerInfo)
+        : { provider: 'OPENAI', modelSource: 'OPENAI' };
+    if (providerInfo?.isAzure && resolvedModel !== originalModel) {
+        logger.debug('Azure model name resolved for pricing', {
+            deployment: originalModel,
+            resolved: resolvedModel,
+            provider: providerMetadata.provider,
+        });
+    }
+    // Build metadata fields using utility (eliminates repetitive spreading)
+    const metadataFields = (0, metadata_builder_js_1.buildMetadataFields)(request.usageMetadata);
+    // Common fields for all operations
+    const commonPayload = {
+        costType: 'AI',
+        model: resolvedModel, // Use resolved model name for accurate pricing
+        responseTime: now,
+        requestDuration: duration,
+        provider: providerMetadata.provider,
+        modelSource: providerMetadata.modelSource,
+        requestTime,
+        completionStartTime: now,
+        // Common token counts
+        inputTokenCount: usage.prompt_tokens,
+        totalTokenCount: usage.total_tokens,
+        // Metadata fields (processed by utility)
+        ...metadataFields,
+        // Fixed middleware source identifier
+        middlewareSource: 'nodejs',
+        // Backend calculates costs
+        inputTokenCost: undefined,
+        outputTokenCost: undefined,
+        totalCost: undefined,
+    };
+    // Operation-specific fields
+    if (operationType !== 'CHAT') {
+        // For embeddings, we don't need the response cast since we use commonPayload
+        return {
+            ...commonPayload,
+            operationType: 'EMBED',
+            transactionId: `embed-${(0, crypto_1.randomUUID)()}`,
+            outputTokenCount: 0,
+            reasoningTokenCount: 0,
+            cacheCreationTokenCount: 0,
+            cacheReadTokenCount: 0,
+            stopReason: 'END',
+            isStreamed: false,
+            timeToFirstToken: 0,
+        };
+    }
+    const chatResponse = response;
+    const chatUsage = chatResponse.usage;
+    return {
+        ...commonPayload,
+        operationType: 'CHAT',
+        transactionId: chatResponse.id || `chat-${(0, crypto_1.randomUUID)()}`,
+        outputTokenCount: chatUsage.completion_tokens || 0,
+        reasoningTokenCount: chatUsage.reasoning_tokens || 0,
+        cacheCreationTokenCount: 0,
+        cacheReadTokenCount: chatUsage.cached_tokens || 0,
+        stopReason: (0, stop_reason_mapper_js_1.mapStopReason)(chatResponse.choices?.[0]?.finish_reason, logger),
+        isStreamed: Boolean(request.stream),
+        timeToFirstToken: request.stream ? 0 : duration,
+    };
+}
+//# sourceMappingURL=payload-builder.js.map

package/dist/cjs/core/tracking/payload-builder.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"payload-builder.js","sourceRoot":"","sources":["../../../../src/core/tracking/payload-builder.ts"],"names":[],"mappings":";AAAA;;;;;GAKG;;AAiCH,oCA6FC;AA5HD,mCAAoC;AAQpC,iDAA+C;AAC/C,6EAAkE;AAClE,yEAAsE;AACtE,iFAA4E;AAC5E,oDAA4D;AAE5D,gBAAgB;AAChB,MAAM,MAAM,GAAG,IAAA,oBAAS,GAAE,CAAC;AAE3B;;;;;;;;;;;;;GAaG;AACH,SAAgB,YAAY,CAC1B,aAA+B,EAC/B,QAAsD,EACtD,OAAmD,EACnD,SAAiB,EACjB,QAAgB,EAChB,YAA2B;IAE3B,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;IACrC,MAAM,WAAW,GAAG,IAAI,IAAI,CAAC,SAAS,CAAC,CAAC,WAAW,EAAE,CAAC;IACtD,MAAM,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC;IAE7B,2CAA2C;IAC3C,MAAM,aAAa,GAAG,QAAQ,CAAC,KAAK,CAAC;IACrC,MAAM,aAAa,GAAG,YAAY,EAAE,OAAO;QACzC,CAAC,CAAC,IAAA,+CAAqB,EAAC,aAAa,CAAC;QACtC,CAAC,CAAC,aAAa,CAAC;IAElB,wBAAwB;IACxB,MAAM,gBAAgB,GAAG,YAAY;QACnC,CAAC,CAAC,IAAA,8BAAmB,EAAC,YAAY,CAAC;QACnC,CAAC,CAAC,EAAE,QAAQ,EAAE,QAAQ,EAAE,WAAW,EAAE,QAAQ,EAAE,CAAC;IAElD,IAAI,YAAY,EAAE,OAAO,IAAI,aAAa,KAAK,aAAa,EAAE,CAAC;QAC7D,MAAM,CAAC,KAAK,CAAC,uCAAuC,EAAE;YACpD,UAAU,EAAE,aAAa;YACzB,QAAQ,EAAE,aAAa;YACvB,QAAQ,EAAE,gBAAgB,CAAC,QAAQ;SACpC,CAAC,CAAC;IACL,CAAC;IAED,wEAAwE;IACxE,MAAM,cAAc,GAAG,IAAA,yCAAmB,EAAC,OAAO,CAAC,aAAa,CAAC,CAAC;IAElE,mCAAmC;IACnC,MAAM,aAAa,GAAG;QACpB,QAAQ,EAAE,IAAa;QACvB,KAAK,EAAE,aAAa,EAAE,+CAA+C;QACrE,YAAY,EAAE,GAAG;QACjB,eAAe,EAAE,QAAQ;QACzB,QAAQ,EAAE,gBAAgB,CAAC,QAAQ;QACnC,WAAW,EAAE,gBAAgB,CAAC,WAAW;QACzC,WAAW;QACX,mBAAmB,EAAE,GAAG;QAExB,sBAAsB;QACtB,eAAe,EAAE,KAAK,CAAC,aAAa;QACpC,eAAe,EAAE,KAAK,CAAC,YAAY;QAEnC,yCAAyC;QACzC,GAAG,cAAc;QAEjB,qCAAqC;QACrC,gBAAgB,EAAE,QAAQ;QAE1B,2BAA2B;QAC3B,cAAc,EAAE,SAAS;QACzB,eAAe,EAAE,SAAS;QAC1B,SAAS,EAAE,SAAS;KACrB,CAAC;IAEF,4BAA4B;IAE5B,IAAI,aAAa,KAAK,MAAM,EAAE,CAAC;QAC7B,6EAA6E;QAC7E,OAAO;YACL,GAAG,aAAa;YAChB,aAAa,EAAE,OAAO;YACtB,aAAa,EAAE,SAAS,IAAA,mBAAU,GAAE,EAAE;YACtC,gBAAgB,EAAE,CAAC;YACnB,mBAAmB,EAAE,CAAC;YACtB,uBAAuB,EAAE,CAAC;YAC1B,mBAAmB,EAAE,CAAC;YACtB,UAAU,EAAE,KAAK;YACjB,UAAU,EAAE,KAAK;YACjB,gBAAgB,EAAE,CAAC;SACpB,CAAC;IACJ,CAAC;IACD,MAAM,YAAY,GAAG,QAA8B,CAAC;IACpD,MAAM,SAAS,GAAG,YAAY,CAAC,KAAK,CAAC;IAErC,OAAO;QACL,GAAG,aAAa;QAChB,aAAa,EAAE,MAAM;QACrB,aAAa,EAAE,YAAY,CAAC,EAAE,IAAI,QAAQ,IAAA,mBAAU,GAAE,EAAE;QACxD,gBAAgB,EAAE,SAAS,CAAC,iBAAiB,IAAI,CAAC;QAClD,mBAAmB,EAAE,SAAS,CAAC,gBAAgB,IAAI,CAAC;QACpD,uBAAuB,EAAE,CAAC;QAC1B,mBAAmB,EAAE,SAAS,CAAC,aAAa,IAAI,CAAC;QACjD,UAAU,EAAE,IAAA,qCAAa,EAAC,YAAY,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,aAAa,EAAE,MAAM,CAAC;QAC3E,UAAU,EAAE,OAAO,CAAE,OAA6B,CAAC,MAAM,CAAC;QAC1D,gBAAgB,EAAG,OAA6B,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,QAAQ;KACvE,CAAC;AACJ,CAAC"}

package/dist/cjs/core/tracking/usage-tracker.js

@@ -0,0 +1,120 @@
+"use strict";
+/**
+ * Usage Tracker Module
+ *
+ * High-level tracking functions that combine payload building and API communication.
+ * Extracted from tracking.ts for better organization.
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.sendReveniumMetrics = sendReveniumMetrics;
+exports.sendReveniumEmbeddingsMetrics = sendReveniumEmbeddingsMetrics;
+exports.trackUsageAsync = trackUsageAsync;
+exports.trackEmbeddingsUsageAsync = trackEmbeddingsUsageAsync;
+const index_js_1 = require("../config/index.js");
+const api_client_js_1 = require("./api-client.js");
+const payload_builder_js_1 = require("./payload-builder.js");
+const error_handler_js_1 = require("../../utils/error-handler.js");
+// Global logger
+const logger = (0, index_js_1.getLogger)();
+/**
+ * Chat completions tracking - now a thin wrapper with Azure support
+ */
+async function sendReveniumMetrics(response, request, startTime, duration, providerInfo) {
+    await (0, error_handler_js_1.safeAsyncOperation)(async () => {
+        const payload = (0, payload_builder_js_1.buildPayload)('CHAT', response, request, startTime, duration, providerInfo);
+        await (0, api_client_js_1.sendToRevenium)(payload);
+    }, 'Chat completion tracking', {
+        logError: true,
+        rethrow: false, // Don't rethrow to maintain fire-and-forget behavior
+        messagePrefix: 'Chat completion tracking failed: ',
+    }, logger);
+}
+/**
+ * Embeddings tracking - now a thin wrapper with Azure support
+ */
+async function sendReveniumEmbeddingsMetrics(response, request, startTime, duration, providerInfo) {
+    await (0, error_handler_js_1.safeAsyncOperation)(async () => {
+        const payload = (0, payload_builder_js_1.buildPayload)('EMBED', response, request, startTime, duration, providerInfo);
+        await (0, api_client_js_1.sendToRevenium)(payload);
+    }, 'Embeddings tracking', {
+        logError: true,
+        rethrow: false, // Don't rethrow to maintain fire-and-forget behavior
+        messagePrefix: 'Embeddings tracking failed: ',
+    }, logger);
+}
+/**
+ * Fire-and-forget wrapper for chat completions with Azure support
+ */
+function trackUsageAsync(trackingData) {
+    const mockResponse = {
+        id: trackingData.requestId,
+        model: trackingData.model,
+        usage: {
+            prompt_tokens: trackingData.promptTokens,
+            completion_tokens: trackingData.completionTokens,
+            total_tokens: trackingData.totalTokens,
+            ...(trackingData.reasoningTokens && { reasoning_tokens: trackingData.reasoningTokens }),
+            ...(trackingData.cachedTokens && { cached_tokens: trackingData.cachedTokens }),
+        },
+        choices: [
+            {
+                finish_reason: trackingData.finishReason,
+            },
+        ],
+    };
+    const mockRequest = {
+        model: trackingData.model,
+        messages: [], // Mock empty messages array for type compliance
+        usageMetadata: trackingData.usageMetadata,
+        stream: trackingData.isStreamed,
+    };
+    const startTime = Date.now() - trackingData.duration;
+    sendReveniumMetrics(mockResponse, mockRequest, startTime, trackingData.duration, trackingData.providerInfo)
+        .then(() => {
+        logger.debug('Usage tracking completed successfully', {
+            requestId: trackingData.requestId,
+            model: trackingData.model,
+            totalTokens: trackingData.totalTokens,
+            isStreamed: trackingData.isStreamed,
+        });
+    })
+        .catch(error => {
+        logger.warn('Usage tracking failed', {
+            error: error instanceof Error ? error.message : String(error),
+            requestId: trackingData.requestId,
+            model: trackingData.model,
+        });
+    });
+}
+/**
+ * Fire-and-forget wrapper for embeddings with Azure support
+ */
+function trackEmbeddingsUsageAsync(trackingData) {
+    const mockResponse = {
+        model: trackingData.model,
+        usage: {
+            prompt_tokens: trackingData.promptTokens,
+            total_tokens: trackingData.totalTokens,
+        },
+        data: [], // Mock empty data array for type compliance
+        object: 'list',
+    };
+    const mockRequest = {
+        model: trackingData.model,
+        input: '', // Mock empty input for type compliance
+        usageMetadata: trackingData.usageMetadata,
+    };
+    sendReveniumEmbeddingsMetrics(mockResponse, mockRequest, trackingData.requestStartTime, trackingData.duration, trackingData.providerInfo)
+        .then(() => {
+        logger.debug('Embeddings tracking completed successfully', {
+            transactionId: trackingData.transactionId,
+        });
+    })
+        .catch(error => {
+        logger.warn('Embeddings tracking failed', {
+            error: error instanceof Error ? error.message : String(error),
+            transactionId: trackingData.transactionId,
+        });
+    });
+}
+//# sourceMappingURL=usage-tracker.js.map

package/dist/cjs/core/tracking/usage-tracker.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"usage-tracker.js","sourceRoot":"","sources":["../../../../src/core/tracking/usage-tracker.ts"],"names":[],"mappings":";AAAA;;;;;GAKG;;AAqBH,kDAoBC;AAKD,sEAoBC;AAKD,0CA+DC;AAKD,8DA4CC;AA7KD,iDAA+C;AAC/C,mDAAiD;AACjD,6DAAoD;AACpD,mEAAkE;AAElE,gBAAgB;AAChB,MAAM,MAAM,GAAG,IAAA,oBAAS,GAAE,CAAC;AAE3B;;GAEG;AACI,KAAK,UAAU,mBAAmB,CACvC,QAA4B,EAC5B,OAA0B,EAC1B,SAAiB,EACjB,QAAgB,EAChB,YAA2B;IAE3B,MAAM,IAAA,qCAAkB,EACtB,KAAK,IAAI,EAAE;QACT,MAAM,OAAO,GAAG,IAAA,iCAAY,EAAC,MAAM,EAAE,QAAQ,EAAE,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,YAAY,CAAC,CAAC;QAC3F,MAAM,IAAA,8BAAc,EAAC,OAAO,CAAC,CAAC;IAChC,CAAC,EACD,0BAA0B,EAC1B;QACE,QAAQ,EAAE,IAAI;QACd,OAAO,EAAE,KAAK,EAAE,qDAAqD;QACrE,aAAa,EAAE,mCAAmC;KACnD,EACD,MAAM,CACP,CAAC;AACJ,CAAC;AAED;;GAEG;AACI,KAAK,UAAU,6BAA6B,CACjD,QAAiC,EACjC,OAA+B,EAC/B,SAAiB,EACjB,QAAgB,EAChB,YAA2B;IAE3B,MAAM,IAAA,qCAAkB,EACtB,KAAK,IAAI,EAAE;QACT,MAAM,OAAO,GAAG,IAAA,iCAAY,EAAC,OAAO,EAAE,QAAQ,EAAE,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,YAAY,CAAC,CAAC;QAC5F,MAAM,IAAA,8BAAc,EAAC,OAAO,CAAC,CAAC;IAChC,CAAC,EACD,qBAAqB,EACrB;QACE,QAAQ,EAAE,IAAI;QACd,OAAO,EAAE,KAAK,EAAE,qDAAqD;QACrE,aAAa,EAAE,8BAA8B;KAC9C,EACD,MAAM,CACP,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,SAAgB,eAAe,CAAC,YAc/B;IACC,MAAM,YAAY,GAAG;QACnB,EAAE,EAAE,YAAY,CAAC,SAAS;QAC1B,KAAK,EAAE,YAAY,CAAC,KAAK;QACzB,KAAK,EAAE;YACL,aAAa,EAAE,YAAY,CAAC,YAAY;YACxC,iBAAiB,EAAE,YAAY,CAAC,gBAAgB;YAChD,YAAY,EAAE,YAAY,CAAC,WAAW;YACtC,GAAG,CAAC,YAAY,CAAC,eAAe,IAAI,EAAE,gBAAgB,EAAE,YAAY,CAAC,eAAe,EAAE,CAAC;YACvF,GAAG,CAAC,YAAY,CAAC,YAAY,IAAI,EAAE,aAAa,EAAE,YAAY,CAAC,YAAY,EAAE,CAAC;SAC/E;QACD,OAAO,EAAE;YACP;gBACE,aAAa,EAAE,YAAY,CAAC,YAAY;aACzC;SACF;KACF,CAAC;IAEF,MAAM,WAAW,GAAsB;QACrC,KAAK,EAAE,YAAY,CAAC,KAAK;QACzB,QAAQ,EAAE,EAAE,EAAE,gDAAgD;QAC9D,aAAa,EAAE,YAAY,CAAC,aAAa;QACzC,MAAM,EAAE,YAAY,CAAC,UAAU;KAChC,CAAC;IAEF,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,YAAY,CAAC,QAAQ,CAAC;IAErD,mBAAmB,CACjB,YAAY,EACZ,WAAW,EACX,SAAS,EACT,YAAY,CAAC,QAAQ,EACrB,YAAY,CAAC,YAAY,CAC1B;SACE,IAAI,CAAC,GAAG,EAAE;QACT,MAAM,CAAC,KAAK,CAAC,uCAAuC,EAAE;YACpD,SAAS,EAAE,YAAY,CAAC,SAAS;YACjC,KAAK,EAAE,YAAY,CAAC,KAAK;YACzB,WAAW,EAAE,YAAY,CAAC,WAAW;YACrC,UAAU,EAAE,YAAY,CAAC,UAAU;SACpC,CAAC,CAAC;IACL,CAAC,CAAC;SACD,KAAK,CAAC,KAAK,CAAC,EAAE;QACb,MAAM,CAAC,IAAI,CAAC,uBAAuB,EAAE;YACnC,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;YAC7D,SAAS,EAAE,YAAY,CAAC,SAAS;YACjC,KAAK,EAAE,YAAY,CAAC,KAAK;SAC1B,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACP,CAAC;AAED;;GAEG;AACH,SAAgB,yBAAyB,CAAC,YASzC;IACC,MAAM,YAAY,GAA4B;QAC5C,KAAK,EAAE,YAAY,CAAC,KAAK;QACzB,KAAK,EAAE;YACL,aAAa,EAAE,YAAY,CAAC,YAAY;YACxC,YAAY,EAAE,YAAY,CAAC,WAAW;SACvC;QACD,IAAI,EAAE,EAAE,EAAE,4CAA4C;QACtD,MAAM,EAAE,MAAM;KACf,CAAC;IAEF,MAAM,WAAW,GAA2B;QAC1C,KAAK,EAAE,YAAY,CAAC,KAAK;QACzB,KAAK,EAAE,EAAE,EAAE,uCAAuC;QAClD,aAAa,EAAE,YAAY,CAAC,aAAa;KAC1C,CAAC;IAEF,6BAA6B,CAC3B,YAAY,EACZ,WAAW,EACX,YAAY,CAAC,gBAAgB,EAC7B,YAAY,CAAC,QAAQ,EACrB,YAAY,CAAC,YAAY,CAC1B;SACE,IAAI,CAAC,GAAG,EAAE;QACT,MAAM,CAAC,KAAK,CAAC,4CAA4C,EAAE;YACzD,aAAa,EAAE,YAAY,CAAC,aAAa;SAC1C,CAAC,CAAC;IACL,CAAC,CAAC;SACD,KAAK,CAAC,KAAK,CAAC,EAAE;QACb,MAAM,CAAC,IAAI,CAAC,4BAA4B,EAAE;YACxC,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;YAC7D,aAAa,EAAE,YAAY,CAAC,aAAa;SAC1C,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACP,CAAC"}

package/dist/cjs/core/wrapper/index.js

@@ -0,0 +1,15 @@
+"use strict";
+/**
+ * Wrapper module - Main exports
+ *
+ * This module provides a clean interface for OpenAI client wrapping,
+ * separating concerns into focused sub-modules.
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getProviderInfo = exports.patchOpenAIInstance = exports.patchOpenAI = void 0;
+// Re-export all wrapper functionality
+var instance_patcher_js_1 = require("./instance-patcher.js");
+Object.defineProperty(exports, "patchOpenAI", { enumerable: true, get: function () { return instance_patcher_js_1.patchOpenAI; } });
+Object.defineProperty(exports, "patchOpenAIInstance", { enumerable: true, get: function () { return instance_patcher_js_1.patchOpenAIInstance; } });
+Object.defineProperty(exports, "getProviderInfo", { enumerable: true, get: function () { return instance_patcher_js_1.getProviderInfo; } });
+//# sourceMappingURL=index.js.map

package/dist/cjs/core/wrapper/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/core/wrapper/index.ts"],"names":[],"mappings":";AAAA;;;;;GAKG;;;AAEH,sCAAsC;AACtC,6DAA0F;AAAjF,kHAAA,WAAW,OAAA;AAAE,0HAAA,mBAAmB,OAAA;AAAE,sHAAA,eAAe,OAAA"}

package/dist/cjs/core/wrapper/instance-patcher.js

@@ -0,0 +1,202 @@
+"use strict";
+/**
+ * Instance Patcher Module
+ *
+ * Handles patching of OpenAI client instances.
+ * Extracted from wrapper.ts for better organization.
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.instanceProviders = void 0;
+exports.getProviderInfo = getProviderInfo;
+exports.patchOpenAI = patchOpenAI;
+exports.patchOpenAIInstance = patchOpenAIInstance;
+const type_guards_js_1 = require("../../utils/type-guards.js");
+const metadata_builder_js_1 = require("../../utils/metadata-builder.js");
+const request_handler_factory_js_1 = require("../../utils/request-handler-factory.js");
+const index_js_1 = require("../config/index.js");
+const index_js_2 = require("../providers/index.js");
+// Import the type augmentations to ensure they're available
+require("../../types/openai-augmentation.js");
+/**
+ * Set to track patched instances
+ */
+const patchedInstances = new WeakSet();
+/**
+ * WeakMap to store provider information for each client instance
+ */
+const instanceProviders = new WeakMap();
+exports.instanceProviders = instanceProviders;
+// Global logger
+const logger = (0, index_js_1.getLogger)();
+/**
+ * Get provider information for a client instance
+ */
+function getProviderInfo(instance) {
+    return instanceProviders.get(instance);
+}
+/**
+ * Simple approach: Only patch instances when users create them
+ * No global patching, no dummy keys - just clean instance patching
+ */
+function patchOpenAI() {
+    logger.info('Revenium OpenAI middleware loaded and ready');
+    logger.debug('Use patchOpenAIInstance() to patch specific OpenAI instances');
+}
+/**
+ * Manually patch an existing OpenAI instance
+ * This is the main function users should call
+ */
+function patchOpenAIInstance(instance) {
+    // Check if middleware is initialized
+    const config = (0, index_js_1.getConfig)();
+    if (!config) {
+        logger.warn('Revenium middleware not initialized.');
+        logger.warn('Auto-initialization may have failed. Try calling initializeReveniumFromEnv() explicitly.');
+        logger.warn('Check that REVENIUM_METERING_API_KEY environment variable is set.');
+        logger.warn('OpenAI instance will be patched but tracking may not work without proper configuration.');
+    }
+    else {
+        logger.debug('Revenium middleware is properly configured');
+    }
+    if (patchedInstances.has(instance)) {
+        logger.debug('OpenAI instance already patched, skipping');
+        return instance;
+    }
+    patchInstance(instance);
+    logger.debug('OpenAI instance patched successfully');
+    return instance;
+}
+/**
+ * Patch an individual OpenAI instance
+ */
+function patchInstance(instance) {
+    try {
+        // Validate instance
+        if (!(0, type_guards_js_1.isOpenAIClientInstance)(instance)) {
+            logger.error('Invalid OpenAI client instance provided to patchInstance');
+            return;
+        }
+        // Detect provider type for this instance
+        const providerInfo = (0, index_js_2.detectProvider)(instance);
+        instanceProviders.set(instance, providerInfo);
+        logger.debug('Provider detection completed for instance', {
+            provider: providerInfo.provider,
+            isAzure: providerInfo.isAzure,
+            hasAzureConfig: !!providerInfo.azureConfig,
+        });
+        // Patch chat completions
+        patchChatCompletions(instance);
+        // Patch embeddings
+        patchEmbeddings(instance);
+        // Patch responses API (new OpenAI Responses API)
+        patchResponses(instance);
+        // Mark as patched
+        patchedInstances.add(instance);
+    }
+    catch (error) {
+        logger.error('Failed to patch OpenAI instance', {
+            error: error instanceof Error ? error.message : String(error),
+        });
+    }
+}
+/**
+ * Patch chat completions endpoint
+ */
+function patchChatCompletions(instance) {
+    if (!instance.chat || !instance.chat.completions || !instance.chat.completions.create) {
+        return logger.warn('OpenAI instance missing chat.completions.create, skipping chat patch');
+    }
+    // Store the original create method
+    const originalCreate = instance.chat.completions.create.bind(instance.chat.completions);
+    // Replace the create method with our wrapped version
+    instance.chat.completions.create = async function (params, options) {
+        // Extract metadata using utility
+        const { metadata, cleanParams } = (0, metadata_builder_js_1.extractMetadata)(params);
+        const typedParams = params;
+        logger.debug('OpenAI chat.completions.create intercepted', {
+            ...(0, metadata_builder_js_1.createLoggingContext)(metadata),
+            model: typedParams.model,
+            stream: !!typedParams.stream,
+        });
+        // Record request start time
+        const requestStartTime = Date.now();
+        // Ensure factory is initialized and route request
+        try {
+            await (0, request_handler_factory_js_1.ensureFactoryInitialized)();
+        }
+        catch (error) {
+            logger.error('Failed to initialize request handler factory', { error });
+            throw new Error('Middleware initialization failed - cannot process request');
+        }
+        return request_handler_factory_js_1.requestHandlerFactory.routeChatRequest(originalCreate, cleanParams, options, metadata, requestStartTime, instance);
+    };
+}
+/**
+ * Patch embeddings endpoint
+ */
+function patchEmbeddings(instance) {
+    if (!instance.embeddings || !instance.embeddings.create) {
+        return logger.warn('OpenAI instance missing embeddings.create, skipping embeddings patch');
+    }
+    // Store the original embeddings create method
+    const originalEmbeddingsCreate = instance.embeddings.create.bind(instance.embeddings);
+    // Replace the embeddings create method with our wrapped version
+    instance.embeddings.create = async function (params, options) {
+        // Extract metadata using utility
+        const { metadata, cleanParams } = (0, metadata_builder_js_1.extractMetadata)(params);
+        const typedParams = params;
+        logger.debug('OpenAI embeddings.create intercepted', {
+            ...(0, metadata_builder_js_1.createLoggingContext)(metadata),
+            model: typedParams.model,
+            inputType: typeof typedParams.input,
+        });
+        // Record request start time
+        const requestStartTime = Date.now();
+        // Ensure factory is initialized and route request
+        try {
+            await (0, request_handler_factory_js_1.ensureFactoryInitialized)();
+        }
+        catch (error) {
+            logger.error('Failed to initialize request handler factory', { error });
+            throw new Error('Middleware initialization failed - cannot process request');
+        }
+        return request_handler_factory_js_1.requestHandlerFactory.routeEmbeddingsRequest(originalEmbeddingsCreate, cleanParams, options, metadata, requestStartTime, instance);
+    };
+}
+/**
+ * Patch responses endpoint (new OpenAI Responses API)
+ */
+function patchResponses(instance) {
+    // Type assertion for new Responses API (not yet in OpenAI types)
+    const responsesAPI = instance;
+    // Check if the instance has the responses API (it's a newer feature)
+    if (!responsesAPI.responses || !responsesAPI.responses.create) {
+        logger.debug('OpenAI instance missing responses.create, skipping responses patch (this is normal for older SDK versions)');
+        return;
+    }
+    // Store the original responses create method
+    const originalResponsesCreate = responsesAPI.responses.create.bind(responsesAPI.responses);
+    // Replace the responses create method with our wrapped version
+    responsesAPI.responses.create = async function (params, options) {
+        // Extract metadata using utility (similar to chat completions)
+        const { metadata, cleanParams } = (0, metadata_builder_js_1.extractMetadata)(params);
+        logger.debug('OpenAI responses.create intercepted', {
+            ...(0, metadata_builder_js_1.createLoggingContext)(metadata),
+            model: params.model,
+            stream: !!params.stream,
+            inputType: typeof params.input,
+        });
+        // Record request start time
+        const requestStartTime = Date.now();
+        // Ensure factory is initialized and route request
+        try {
+            await (0, request_handler_factory_js_1.ensureFactoryInitialized)();
+        }
+        catch (error) {
+            logger.error('Failed to initialize request handler factory', { error });
+            throw new Error('Middleware initialization failed - cannot process request');
+        }
+        return request_handler_factory_js_1.requestHandlerFactory.routeResponsesRequest(originalResponsesCreate, cleanParams, options, metadata, requestStartTime, instance);
+    };
+}
+//# sourceMappingURL=instance-patcher.js.map

package/dist/cjs/core/wrapper/instance-patcher.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"instance-patcher.js","sourceRoot":"","sources":["../../../../src/core/wrapper/instance-patcher.ts"],"names":[],"mappings":";AAAA;;;;;GAKG;;;AAiDH,0CAEC;AAMD,kCAGC;AAMD,kDAyBC;AA9ED,+DAAoE;AACpE,yEAAwF;AACxF,uFAGgD;AAChD,iDAA0D;AAC1D,oDAAuD;AAGvD,4DAA4D;AAC5D,8CAA4C;AAS5C;;GAEG;AACH,MAAM,gBAAgB,GAAG,IAAI,OAAO,EAAE,CAAC;AAEvC;;GAEG;AACH,MAAM,iBAAiB,GAAG,IAAI,OAAO,EAAsC,CAAC;AAiPnE,8CAAiB;AA/O1B,gBAAgB;AAChB,MAAM,MAAM,GAAG,IAAA,oBAAS,GAAE,CAAC;AAE3B;;GAEG;AACH,SAAgB,eAAe,CAAC,QAA8B;IAC5D,OAAO,iBAAiB,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;AACzC,CAAC;AAED;;;GAGG;AACH,SAAgB,WAAW;IACzB,MAAM,CAAC,IAAI,CAAC,6CAA6C,CAAC,CAAC;IAC3D,MAAM,CAAC,KAAK,CAAC,8DAA8D,CAAC,CAAC;AAC/E,CAAC;AAED;;;GAGG;AACH,SAAgB,mBAAmB,CAAC,QAAgB;IAClD,qCAAqC;IACrC,MAAM,MAAM,GAAG,IAAA,oBAAS,GAAE,CAAC;IAC3B,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,MAAM,CAAC,IAAI,CAAC,sCAAsC,CAAC,CAAC;QACpD,MAAM,CAAC,IAAI,CACT,0FAA0F,CAC3F,CAAC;QACF,MAAM,CAAC,IAAI,CAAC,mEAAmE,CAAC,CAAC;QACjF,MAAM,CAAC,IAAI,CACT,yFAAyF,CAC1F,CAAC;IACJ,CAAC;SAAM,CAAC;QACN,MAAM,CAAC,KAAK,CAAC,4CAA4C,CAAC,CAAC;IAC7D,CAAC;IAED,IAAI,gBAAgB,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC;QACnC,MAAM,CAAC,KAAK,CAAC,2CAA2C,CAAC,CAAC;QAC1D,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,aAAa,CAAC,QAA2C,CAAC,CAAC;IAC3D,MAAM,CAAC,KAAK,CAAC,sCAAsC,CAAC,CAAC;IAErD,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED;;GAEG;AACH,SAAS,aAAa,CAAC,QAA8B;IACnD,IAAI,CAAC;QACH,oBAAoB;QACpB,IAAI,CAAC,IAAA,uCAAsB,EAAC,QAAQ,CAAC,EAAE,CAAC;YACtC,MAAM,CAAC,KAAK,CAAC,0DAA0D,CAAC,CAAC;YACzE,OAAO;QACT,CAAC;QAED,yCAAyC;QACzC,MAAM,YAAY,GAAG,IAAA,yBAAc,EAAC,QAAQ,CAAC,CAAC;QAC9C,iBAAiB,CAAC,GAAG,CAAC,QAAQ,EAAE,YAAY,CAAC,CAAC;QAE9C,MAAM,CAAC,KAAK,CAAC,2CAA2C,EAAE;YACxD,QAAQ,EAAE,YAAY,CAAC,QAAQ;YAC/B,OAAO,EAAE,YAAY,CAAC,OAAO;YAC7B,cAAc,EAAE,CAAC,CAAC,YAAY,CAAC,WAAW;SAC3C,CAAC,CAAC;QAEH,yBAAyB;QACzB,oBAAoB,CAAC,QAAQ,CAAC,CAAC;QAE/B,mBAAmB;QACnB,eAAe,CAAC,QAAQ,CAAC,CAAC;QAE1B,iDAAiD;QACjD,cAAc,CAAC,QAAQ,CAAC,CAAC;QAEzB,kBAAkB;QAClB,gBAAgB,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;IACjC,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,CAAC,KAAK,CAAC,iCAAiC,EAAE;YAC9C,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;SAC9D,CAAC,CAAC;IACL,CAAC;AACH,CAAC;AAED;;GAEG;AACH,SAAS,oBAAoB,CAAC,QAA8B;IAC1D,IAAI,CAAC,QAAQ,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,WAAW,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC;QACtF,OAAO,MAAM,CAAC,IAAI,CAAC,sEAAsE,CAAC,CAAC;IAC7F,CAAC;IAED,mCAAmC;IACnC,MAAM,cAAc,GAAG,QAAQ,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IAExF,qDAAqD;IACrD,QAAQ,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,GAAG,KAAK,WACtC,MAAkC,EAClC,OAA8B;QAE9B,iCAAiC;QACjC,MAAM,EAAE,QAAQ,EAAE,WAAW,EAAE,GAAG,IAAA,qCAAe,EAAC,MAA2B,CAAC,CAAC;QAC/E,MAAM,WAAW,GAAG,MAA2B,CAAC;QAEhD,MAAM,CAAC,KAAK,CAAC,4CAA4C,EAAE;YACzD,GAAG,IAAA,0CAAoB,EAAC,QAAQ,CAAC;YACjC,KAAK,EAAE,WAAW,CAAC,KAAK;YACxB,MAAM,EAAE,CAAC,CAAC,WAAW,CAAC,MAAM;SAC7B,CAAC,CAAC;QAEH,4BAA4B;QAC5B,MAAM,gBAAgB,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAEpC,kDAAkD;QAClD,IAAI,CAAC;YACH,MAAM,IAAA,qDAAwB,GAAE,CAAC;QACnC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC,KAAK,CAAC,8CAA8C,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;YACxE,MAAM,IAAI,KAAK,CAAC,2DAA2D,CAAC,CAAC;QAC/E,CAAC;QACD,OAAO,kDAAqB,CAAC,gBAAgB,CAC3C,cAAc,EACd,WAAW,EACX,OAAO,EACP,QAAQ,EACR,gBAAgB,EAChB,QAAQ,CACT,CAAC;IACJ,CAAC,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,SAAS,eAAe,CAAC,QAA8B;IACrD,IAAI,CAAC,QAAQ,CAAC,UAAU,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,MAAM,EAAE,CAAC;QACxD,OAAO,MAAM,CAAC,IAAI,CAAC,sEAAsE,CAAC,CAAC;IAC7F,CAAC;IACD,8CAA8C;IAC9C,MAAM,wBAAwB,GAAG,QAAQ,CAAC,UAAU,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC;IAEtF,gEAAgE;IAChE,QAAQ,CAAC,UAAU,CAAC,MAAM,GAAG,KAAK,WAChC,MAA6B,EAC7B,OAA8B;QAE9B,iCAAiC;QACjC,MAAM,EAAE,QAAQ,EAAE,WAAW,EAAE,GAAG,IAAA,qCAAe,EAAC,MAAgC,CAAC,CAAC;QACpF,MAAM,WAAW,GAAG,MAAgC,CAAC;QAErD,MAAM,CAAC,KAAK,CAAC,sCAAsC,EAAE;YACnD,GAAG,IAAA,0CAAoB,EAAC,QAAQ,CAAC;YACjC,KAAK,EAAE,WAAW,CAAC,KAAK;YACxB,SAAS,EAAE,OAAO,WAAW,CAAC,KAAK;SACpC,CAAC,CAAC;QAEH,4BAA4B;QAC5B,MAAM,gBAAgB,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAEpC,kDAAkD;QAClD,IAAI,CAAC;YACH,MAAM,IAAA,qDAAwB,GAAE,CAAC;QACnC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC,KAAK,CAAC,8CAA8C,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;YACxE,MAAM,IAAI,KAAK,CAAC,2DAA2D,CAAC,CAAC;QAC/E,CAAC;QACD,OAAO,kDAAqB,CAAC,sBAAsB,CACjD,wBAAwB,EACxB,WAAW,EACX,OAAO,EACP,QAAQ,EACR,gBAAgB,EAChB,QAAQ,CACT,CAAC;IACJ,CAAC,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,SAAS,cAAc,CAAC,QAA8B;IACpD,iEAAiE;IACjE,MAAM,YAAY,GAAG,QAA+B,CAAC;IAErD,qEAAqE;IACrE,IAAI,CAAC,YAAY,CAAC,SAAS,IAAI,CAAC,YAAY,CAAC,SAAS,CAAC,MAAM,EAAE,CAAC;QAC9D,MAAM,CAAC,KAAK,CACV,4GAA4G,CAC7G,CAAC;QACF,OAAO;IACT,CAAC;IAED,6CAA6C;IAC7C,MAAM,uBAAuB,GAAG,YAAY,CAAC,SAAS,CAAC,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC;IAE3F,+DAA+D;IAC/D,YAAY,CAAC,SAAS,CAAC,MAAM,GAAG,KAAK,WACnC,MAA8B,EAC9B,OAA8B;QAE9B,+DAA+D;QAC/D,MAAM,EAAE,QAAQ,EAAE,WAAW,EAAE,GAAG,IAAA,qCAAe,EAAC,MAAM,CAAC,CAAC;QAE1D,MAAM,CAAC,KAAK,CAAC,qCAAqC,EAAE;YAClD,GAAG,IAAA,0CAAoB,EAAC,QAAQ,CAAC;YACjC,KAAK,EAAE,MAAM,CAAC,KAAK;YACnB,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,MAAM;YACvB,SAAS,EAAE,OAAO,MAAM,CAAC,KAAK;SAC/B,CAAC,CAAC;QAEH,4BAA4B;QAC5B,MAAM,gBAAgB,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAEpC,kDAAkD;QAClD,IAAI,CAAC;YACH,MAAM,IAAA,qDAAwB,GAAE,CAAC;QACnC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC,KAAK,CAAC,8CAA8C,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;YACxE,MAAM,IAAI,KAAK,CAAC,2DAA2D,CAAC,CAAC;QAC/E,CAAC;QACD,OAAO,kDAAqB,CAAC,qBAAqB,CAChD,uBAA0D,EAC1D,WAAW,EACX,OAAO,EACP,QAAQ,EACR,gBAAgB,EAChB,QAAQ,CACT,CAAC;IACJ,CAAC,CAAC;AACJ,CAAC"}