@revenium/openai 1.0.13 → 1.0.15

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

@@ -1,202 +0,0 @@
-"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

@@ -1 +0,0 @@
-{"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"}

package/dist/cjs/core/wrapper/request-handler.js

@@ -1,317 +0,0 @@
-"use strict";
-/**
- * Request Handler Module
- *
- * Handles different types of OpenAI requests (streaming, non-streaming, embeddings).
- * Extracted from wrapper.ts for better organization.
- */
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.handleNonStreamingRequest = handleNonStreamingRequest;
-exports.handleStreamingRequest = handleStreamingRequest;
-exports.handleEmbeddingsRequest = handleEmbeddingsRequest;
-exports.handleResponsesNonStreamingRequest = handleResponsesNonStreamingRequest;
-exports.handleResponsesStreamingRequest = handleResponsesStreamingRequest;
-const crypto_1 = require("crypto");
-const type_guards_js_1 = require("../../utils/type-guards.js");
-const error_handler_js_1 = require("../../utils/error-handler.js");
-const metadata_builder_js_1 = require("../../utils/metadata-builder.js");
-const index_js_1 = require("../tracking/index.js");
-const index_js_2 = require("../config/index.js");
-const instance_patcher_js_1 = require("./instance-patcher.js");
-const stream_wrapper_js_1 = require("./stream-wrapper.js");
-// Global logger
-const logger = (0, index_js_2.getLogger)();
-/**
- * Handle non-streaming OpenAI requests
- */
-async function handleNonStreamingRequest(originalCreate, params, options, usageMetadata, requestStartTime, instance) {
-    const loggingContext = (0, metadata_builder_js_1.createLoggingContext)(usageMetadata);
-    const result = await (0, error_handler_js_1.safeAsyncOperation)(async () => {
-        // Call the original OpenAI method (cast params back to original type since usageMetadata is removed)
-        const response = await originalCreate(params, options);
-        // Validate response structure
-        if (!(0, type_guards_js_1.hasValidUsage)(response)) {
-            logger.warn('Invalid response structure from OpenAI API', {
-                ...loggingContext,
-                response,
-            });
-            return response;
-        }
-        // Calculate duration
-        const duration = Date.now() - requestStartTime;
-        // Get provider info for this instance
-        const providerInfo = instance_patcher_js_1.instanceProviders.get(instance);
-        // Track usage for chat completions
-        if ((0, type_guards_js_1.isOpenAIChatResponse)(response)) {
-            (0, index_js_1.trackUsageAsync)({
-                requestId: response.id,
-                model: response.model,
-                promptTokens: response.usage.prompt_tokens,
-                completionTokens: response.usage.completion_tokens || 0,
-                totalTokens: response.usage.total_tokens,
-                reasoningTokens: response.usage.reasoning_tokens,
-                cachedTokens: response.usage.cached_tokens,
-                duration,
-                finishReason: response.choices?.[0]?.finish_reason || null,
-                usageMetadata,
-                isStreamed: false,
-                providerInfo,
-            });
-        }
-        logger.debug('Chat completion request completed', {
-            ...loggingContext,
-            model: response.model,
-            duration,
-            totalTokens: response.usage.total_tokens,
-        });
-        return response;
-    }, 'Non-streaming OpenAI request', {
-        logError: true,
-        rethrow: true,
-        messagePrefix: 'Chat completion request failed: ',
-        transformError: error => {
-            const classified = (0, error_handler_js_1.classifyError)(error);
-            if (classified.type === 'network') {
-                return new error_handler_js_1.NetworkError(classified.message, {
-                    ...loggingContext,
-                    duration: Date.now() - requestStartTime,
-                });
-            }
-            return error instanceof Error ? error : new Error(String(error));
-        },
-    }, logger);
-    if (!result)
-        throw new Error('OpenAI request failed without specific error');
-    return result;
-}
-/**
- * Handle streaming OpenAI requests
- */
-async function handleStreamingRequest(originalCreate, params, options, usageMetadata, requestStartTime, instance) {
-    try {
-        // Ensure stream_options includes usage data for token tracking
-        const enhancedParams = {
-            ...params,
-            stream_options: {
-                include_usage: true,
-                ...(params.stream_options || {}),
-            },
-        };
-        logger.debug('Enhanced streaming params with usage tracking', {
-            originalStreamOptions: params.stream_options,
-            enhancedStreamOptions: enhancedParams.stream_options,
-        });
-        // Call the original OpenAI method to get the stream (cast params back to original type since usageMetadata is removed)
-        const originalStream = await originalCreate(enhancedParams, options);
-        logger.debug('Chat completion streaming request initiated', {
-            model: params.model,
-        });
-        // Return a wrapped stream that tracks usage when complete
-        return (0, stream_wrapper_js_1.createTrackingStreamWrapper)(originalStream, usageMetadata, requestStartTime, instance);
-    }
-    catch (error) {
-        const duration = Date.now() - requestStartTime;
-        logger.error('Chat completion streaming request failed', {
-            error: error instanceof Error ? error.message : String(error),
-            duration,
-        });
-        // Re-throw the error to maintain original behavior
-        throw error;
-    }
-}
-/**
- * Handle embeddings requests
- */
-async function handleEmbeddingsRequest(originalCreate, params, options, usageMetadata, requestStartTime, instance) {
-    try {
-        // Call the original OpenAI method (cast params back to original type since usageMetadata is removed)
-        const response = await originalCreate(params, options);
-        // Validate response structure
-        if (!(0, type_guards_js_1.isOpenAIEmbeddingResponse)(response)) {
-            logger.warn('Invalid embeddings response structure from OpenAI API', { response });
-            return response;
-        }
-        // Calculate duration
-        const duration = Date.now() - requestStartTime;
-        // Get provider info for this instance
-        const providerInfo = instance_patcher_js_1.instanceProviders.get(instance);
-        // Track embeddings usage
-        (0, index_js_1.trackEmbeddingsUsageAsync)({
-            transactionId: `embed-${(0, crypto_1.randomUUID)()}`,
-            model: response.model,
-            promptTokens: response.usage.prompt_tokens,
-            totalTokens: response.usage.total_tokens,
-            duration,
-            usageMetadata,
-            requestStartTime,
-            providerInfo,
-        });
-        logger.debug('Embeddings request completed', {
-            model: response.model,
-            duration,
-            totalTokens: response.usage.total_tokens,
-        });
-        return response;
-    }
-    catch (error) {
-        const duration = Date.now() - requestStartTime;
-        logger.error('Embeddings request failed', {
-            error: error instanceof Error ? error.message : String(error),
-            duration,
-        });
-        // Re-throw the error to maintain original behavior
-        throw error;
-    }
-}
-/**
- * Handle non-streaming OpenAI Responses API requests
- */
-async function handleResponsesNonStreamingRequest(originalCreate, params, options, usageMetadata, requestStartTime, instance) {
-    const loggingContext = (0, metadata_builder_js_1.createLoggingContext)(usageMetadata);
-    const result = await (0, error_handler_js_1.safeAsyncOperation)(async () => {
-        // Call the original OpenAI method (cast params back to original type since usageMetadata is removed)
-        const response = await originalCreate(params, options);
-        // Validate response structure
-        if (!response || typeof response !== 'object') {
-            throw new Error('Invalid response from OpenAI Responses API');
-        }
-        const duration = Date.now() - requestStartTime;
-        // Extract usage information (Responses API may have different structure)
-        const usage = response.usage;
-        if (usage) {
-            // Track usage asynchronously using similar pattern to chat completions
-            (0, index_js_1.trackUsageAsync)({
-                requestId: response.id || (0, crypto_1.randomUUID)(),
-                model: response.model || params.model,
-                promptTokens: usage.input_tokens || 0,
-                completionTokens: usage.output_tokens || 0,
-                totalTokens: usage.total_tokens || 0,
-                reasoningTokens: usage.reasoning_tokens,
-                cachedTokens: usage.cached_tokens,
-                duration,
-                finishReason: response.finish_reason || 'completed',
-                usageMetadata,
-                isStreamed: false,
-                providerInfo: instance_patcher_js_1.instanceProviders.get(instance),
-            });
-        }
-        logger.debug('Responses API request completed', {
-            ...loggingContext,
-            model: response.model,
-            duration,
-            totalTokens: usage?.total_tokens,
-        });
-        return response;
-    }, 'Non-streaming OpenAI Responses API request', {
-        logError: true,
-        rethrow: true,
-        messagePrefix: 'Responses API request failed: ',
-        transformError: error => {
-            const classified = (0, error_handler_js_1.classifyError)(error);
-            if (classified.type === 'network') {
-                return new error_handler_js_1.NetworkError(classified.message, {
-                    ...loggingContext,
-                    duration: Date.now() - requestStartTime,
-                });
-            }
-            return error instanceof Error ? error : new Error(String(error));
-        },
-    }, logger);
-    if (!result)
-        throw new Error('OpenAI Responses API request failed without specific error');
-    return result;
-}
-/**
- * Handle streaming OpenAI Responses API requests
- */
-async function handleResponsesStreamingRequest(originalCreate, params, options, usageMetadata, requestStartTime, instance) {
-    try {
-        // Call the original OpenAI method to get the stream (cast params back to original type since usageMetadata is removed)
-        const originalStream = await originalCreate(params, options);
-        logger.debug('Responses API streaming request initiated', {
-            model: params.model,
-        });
-        // Return a wrapped stream that tracks usage when complete
-        // We'll use a similar pattern to chat completions but adapted for Responses API
-        return createResponsesTrackingStreamWrapper(originalStream, usageMetadata, requestStartTime, instance);
-    }
-    catch (error) {
-        const duration = Date.now() - requestStartTime;
-        logger.error('Responses API streaming request failed', {
-            error: error instanceof Error ? error.message : String(error),
-            duration,
-        });
-        // Re-throw the error to maintain original behavior
-        throw error;
-    }
-}
-/**
- * Create a tracking stream wrapper for Responses API
- * Similar to createTrackingStreamWrapper but adapted for Responses API structure
- */
-async function* createResponsesTrackingStreamWrapper(originalStream, usageMetadata, requestStartTime, instance) {
-    let firstChunkTime;
-    let finalUsage = null;
-    let model = '';
-    let requestId = '';
-    let finishReason = null;
-    try {
-        for await (const chunk of originalStream) {
-            // Record time to first token
-            if (!firstChunkTime) {
-                firstChunkTime = Date.now();
-            }
-            // Extract information from chunk (Responses API structure may differ)
-            if (chunk && typeof chunk === 'object') {
-                const chunkObj = chunk;
-                // Extract model and ID from chunk
-                if (chunkObj.model)
-                    model = chunkObj.model;
-                if (chunkObj.id)
-                    requestId = chunkObj.id;
-                // Check for final usage information
-                if (chunkObj.usage) {
-                    finalUsage = chunkObj.usage;
-                }
-                // Check for finish reason
-                if (chunkObj.finish_reason) {
-                    finishReason = chunkObj.finish_reason;
-                }
-            }
-            yield chunk;
-        }
-        // Track usage after stream completes
-        if (finalUsage) {
-            const duration = Date.now() - requestStartTime;
-            const timeToFirstToken = firstChunkTime ? firstChunkTime - requestStartTime : undefined;
-            (0, index_js_1.trackUsageAsync)({
-                requestId: requestId || (0, crypto_1.randomUUID)(),
-                model: model,
-                promptTokens: finalUsage.input_tokens || 0,
-                completionTokens: finalUsage.output_tokens || 0,
-                totalTokens: finalUsage.total_tokens || 0,
-                reasoningTokens: finalUsage.reasoning_tokens,
-                cachedTokens: finalUsage.cached_tokens,
-                duration,
-                finishReason: finishReason || 'completed',
-                usageMetadata,
-                isStreamed: true,
-                timeToFirstToken,
-                providerInfo: instance_patcher_js_1.instanceProviders.get(instance),
-            });
-            logger.debug('Responses API streaming completed', {
-                model,
-                duration,
-                timeToFirstToken,
-                totalTokens: finalUsage.total_tokens,
-            });
-        }
-    }
-    catch (error) {
-        logger.error('Error in Responses API stream wrapper', {
-            error: error instanceof Error ? error.message : String(error),
-        });
-        throw error;
-    }
-}
-//# sourceMappingURL=request-handler.js.map

package/dist/cjs/core/wrapper/request-handler.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"request-handler.js","sourceRoot":"","sources":["../../../../src/core/wrapper/request-handler.ts"],"names":[],"mappings":";AAAA;;;;;GAKG;;AAgCH,8DA8EC;AAKD,wDA+CC;AAKD,0DAqDC;AAKD,gFAwEC;AAKD,0EAkCC;AA9UD,mCAAoC;AAYpC,+DAIoC;AACpC,mEAA+F;AAC/F,yEAAuE;AACvE,mDAAkF;AAClF,iDAA+C;AAC/C,+DAA0D;AAC1D,2DAAkE;AAElE,gBAAgB;AAChB,MAAM,MAAM,GAAG,IAAA,oBAAS,GAAE,CAAC;AAE3B;;GAEG;AACI,KAAK,UAAU,yBAAyB,CAC7C,cAAsC,EACtC,MAAgG,EAChG,OAAyC,EACzC,aAAwC,EACxC,gBAAwB,EACxB,QAA8B;IAE9B,MAAM,cAAc,GAAG,IAAA,0CAAoB,EAAC,aAAa,CAAC,CAAC;IAE3D,MAAM,MAAM,GAAG,MAAM,IAAA,qCAAkB,EACrC,KAAK,IAAI,EAAE;QACT,qGAAqG;QACrG,MAAM,QAAQ,GAAG,MAAM,cAAc,CAAC,MAAa,EAAE,OAAO,CAAC,CAAC;QAE9D,8BAA8B;QAC9B,IAAI,CAAC,IAAA,8BAAa,EAAC,QAAQ,CAAC,EAAE,CAAC;YAC7B,MAAM,CAAC,IAAI,CAAC,4CAA4C,EAAE;gBACxD,GAAG,cAAc;gBACjB,QAAQ;aACT,CAAC,CAAC;YACH,OAAO,QAAQ,CAAC;QAClB,CAAC;QAED,qBAAqB;QACrB,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,gBAAgB,CAAC;QAE/C,sCAAsC;QACtC,MAAM,YAAY,GAAG,uCAAiB,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAErD,mCAAmC;QACnC,IAAI,IAAA,qCAAoB,EAAC,QAAQ,CAAC,EAAE,CAAC;YACnC,IAAA,0BAAe,EAAC;gBACd,SAAS,EAAE,QAAQ,CAAC,EAAE;gBACtB,KAAK,EAAE,QAAQ,CAAC,KAAK;gBACrB,YAAY,EAAE,QAAQ,CAAC,KAAK,CAAC,aAAa;gBAC1C,gBAAgB,EAAE,QAAQ,CAAC,KAAK,CAAC,iBAAiB,IAAI,CAAC;gBACvD,WAAW,EAAE,QAAQ,CAAC,KAAK,CAAC,YAAY;gBACxC,eAAe,EAAE,QAAQ,CAAC,KAAK,CAAC,gBAAgB;gBAChD,YAAY,EAAE,QAAQ,CAAC,KAAK,CAAC,aAAa;gBAC1C,QAAQ;gBACR,YAAY,EAAE,QAAQ,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,aAAa,IAAI,IAAI;gBAC1D,aAAa;gBACb,UAAU,EAAE,KAAK;gBACjB,YAAY;aACb,CAAC,CAAC;QACL,CAAC;QAED,MAAM,CAAC,KAAK,CAAC,mCAAmC,EAAE;YAChD,GAAG,cAAc;YACjB,KAAK,EAAE,QAAQ,CAAC,KAAK;YACrB,QAAQ;YACR,WAAW,EAAE,QAAQ,CAAC,KAAK,CAAC,YAAY;SACzC,CAAC,CAAC;QAEH,OAAO,QAAQ,CAAC;IAClB,CAAC,EACD,8BAA8B,EAC9B;QACE,QAAQ,EAAE,IAAI;QACd,OAAO,EAAE,IAAI;QACb,aAAa,EAAE,kCAAkC;QACjD,cAAc,EAAE,KAAK,CAAC,EAAE;YACtB,MAAM,UAAU,GAAG,IAAA,gCAAa,EAAC,KAAK,CAAC,CAAC;YACxC,IAAI,UAAU,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;gBAClC,OAAO,IAAI,+BAAY,CAAC,UAAU,CAAC,OAAO,EAAE;oBAC1C,GAAG,cAAc;oBACjB,QAAQ,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,gBAAgB;iBACxC,CAAC,CAAC;YACL,CAAC;YACD,OAAO,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;QACnE,CAAC;KACF,EACD,MAAM,CACP,CAAC;IAEF,IAAI,CAAC,MAAM;QAAE,MAAM,IAAI,KAAK,CAAC,8CAA8C,CAAC,CAAC;IAC7E,OAAO,MAAM,CAAC;AAChB,CAAC;AAED;;GAEG;AACI,KAAK,UAAU,sBAAsB,CAC1C,cAAsC,EACtC,MAAgD,EAChD,OAAyC,EACzC,aAAwC,EACxC,gBAAwB,EACxB,QAA8B;IAE9B,IAAI,CAAC;QACH,+DAA+D;QAC/D,MAAM,cAAc,GAAG;YACrB,GAAG,MAAM;YACT,cAAc,EAAE;gBACd,aAAa,EAAE,IAAI;gBACnB,GAAG,CAAC,MAAM,CAAC,cAAc,IAAI,EAAE,CAAC;aACjC;SACF,CAAC;QAEF,MAAM,CAAC,KAAK,CAAC,+CAA+C,EAAE;YAC5D,qBAAqB,EAAE,MAAM,CAAC,cAAc;YAC5C,qBAAqB,EAAE,cAAc,CAAC,cAAc;SACrD,CAAC,CAAC;QAEH,uHAAuH;QACvH,MAAM,cAAc,GAAG,MAAM,cAAc,CAAC,cAAqB,EAAE,OAAO,CAAC,CAAC;QAE5E,MAAM,CAAC,KAAK,CAAC,6CAA6C,EAAE;YAC1D,KAAK,EAAE,MAAM,CAAC,KAAK;SACpB,CAAC,CAAC;QAEH,0DAA0D;QAC1D,OAAO,IAAA,+CAA2B,EAChC,cAAmD,EACnD,aAAa,EACb,gBAAgB,EAChB,QAAQ,CACT,CAAC;IACJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,gBAAgB,CAAC;QAC/C,MAAM,CAAC,KAAK,CAAC,0CAA0C,EAAE;YACvD,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;YAC7D,QAAQ;SACT,CAAC,CAAC;QAEH,mDAAmD;QACnD,MAAM,KAAK,CAAC;IACd,CAAC;AACH,CAAC;AAED;;GAEG;AACI,KAAK,UAAU,uBAAuB,CAC3C,cAAsC,EACtC,MAAqD,EACrD,OAAyC,EACzC,aAAwC,EACxC,gBAAwB,EACxB,QAA8B;IAE9B,IAAI,CAAC;QACH,qGAAqG;QACrG,MAAM,QAAQ,GAAG,MAAM,cAAc,CAAC,MAAa,EAAE,OAAO,CAAC,CAAC;QAE9D,8BAA8B;QAC9B,IAAI,CAAC,IAAA,0CAAyB,EAAC,QAAQ,CAAC,EAAE,CAAC;YACzC,MAAM,CAAC,IAAI,CAAC,uDAAuD,EAAE,EAAE,QAAQ,EAAE,CAAC,CAAC;YACnF,OAAO,QAA8C,CAAC;QACxD,CAAC;QAED,qBAAqB;QACrB,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,gBAAgB,CAAC;QAE/C,sCAAsC;QACtC,MAAM,YAAY,GAAG,uCAAiB,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAErD,yBAAyB;QACzB,IAAA,oCAAyB,EAAC;YACxB,aAAa,EAAE,SAAS,IAAA,mBAAU,GAAE,EAAE;YACtC,KAAK,EAAE,QAAQ,CAAC,KAAK;YACrB,YAAY,EAAE,QAAQ,CAAC,KAAK,CAAC,aAAa;YAC1C,WAAW,EAAE,QAAQ,CAAC,KAAK,CAAC,YAAY;YACxC,QAAQ;YACR,aAAa;YACb,gBAAgB;YAChB,YAAY;SACb,CAAC,CAAC;QAEH,MAAM,CAAC,KAAK,CAAC,8BAA8B,EAAE;YAC3C,KAAK,EAAE,QAAQ,CAAC,KAAK;YACrB,QAAQ;YACR,WAAW,EAAE,QAAQ,CAAC,KAAK,CAAC,YAAY;SACzC,CAAC,CAAC;QAEH,OAAO,QAAQ,CAAC;IAClB,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,gBAAgB,CAAC;QAC/C,MAAM,CAAC,KAAK,CAAC,2BAA2B,EAAE;YACxC,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;YAC7D,QAAQ;SACT,CAAC,CAAC;QAEH,mDAAmD;QACnD,MAAM,KAAK,CAAC;IACd,CAAC;AACH,CAAC;AAED;;GAEG;AACI,KAAK,UAAU,kCAAkC,CACtD,cAA+C,EAC/C,MAAqD,EACrD,OAAyC,EACzC,aAAwC,EACxC,gBAAwB,EACxB,QAA8B;IAE9B,MAAM,cAAc,GAAG,IAAA,0CAAoB,EAAC,aAAa,CAAC,CAAC;IAE3D,MAAM,MAAM,GAAG,MAAM,IAAA,qCAAkB,EACrC,KAAK,IAAI,EAAE;QACT,qGAAqG;QACrG,MAAM,QAAQ,GAAG,MAAM,cAAc,CAAC,MAAa,EAAE,OAAO,CAAC,CAAC;QAE9D,8BAA8B;QAC9B,IAAI,CAAC,QAAQ,IAAI,OAAO,QAAQ,KAAK,QAAQ,EAAE,CAAC;YAC9C,MAAM,IAAI,KAAK,CAAC,4CAA4C,CAAC,CAAC;QAChE,CAAC;QAED,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,gBAAgB,CAAC;QAE/C,yEAAyE;QACzE,MAAM,KAAK,GAAI,QAAgB,CAAC,KAAK,CAAC;QACtC,IAAI,KAAK,EAAE,CAAC;YACV,uEAAuE;YACvE,IAAA,0BAAe,EAAC;gBACd,SAAS,EAAG,QAAgB,CAAC,EAAE,IAAI,IAAA,mBAAU,GAAE;gBAC/C,KAAK,EAAG,QAAgB,CAAC,KAAK,IAAI,MAAM,CAAC,KAAK;gBAC9C,YAAY,EAAE,KAAK,CAAC,YAAY,IAAI,CAAC;gBACrC,gBAAgB,EAAE,KAAK,CAAC,aAAa,IAAI,CAAC;gBAC1C,WAAW,EAAE,KAAK,CAAC,YAAY,IAAI,CAAC;gBACpC,eAAe,EAAE,KAAK,CAAC,gBAAgB;gBACvC,YAAY,EAAE,KAAK,CAAC,aAAa;gBACjC,QAAQ;gBACR,YAAY,EAAG,QAAgB,CAAC,aAAa,IAAI,WAAW;gBAC5D,aAAa;gBACb,UAAU,EAAE,KAAK;gBACjB,YAAY,EAAE,uCAAiB,CAAC,GAAG,CAAC,QAAQ,CAAC;aAC9C,CAAC,CAAC;QACL,CAAC;QAED,MAAM,CAAC,KAAK,CAAC,iCAAiC,EAAE;YAC9C,GAAG,cAAc;YACjB,KAAK,EAAG,QAAgB,CAAC,KAAK;YAC9B,QAAQ;YACR,WAAW,EAAE,KAAK,EAAE,YAAY;SACjC,CAAC,CAAC;QAEH,OAAO,QAAQ,CAAC;IAClB,CAAC,EACD,4CAA4C,EAC5C;QACE,QAAQ,EAAE,IAAI;QACd,OAAO,EAAE,IAAI;QACb,aAAa,EAAE,gCAAgC;QAC/C,cAAc,EAAE,KAAK,CAAC,EAAE;YACtB,MAAM,UAAU,GAAG,IAAA,gCAAa,EAAC,KAAK,CAAC,CAAC;YACxC,IAAI,UAAU,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;gBAClC,OAAO,IAAI,+BAAY,CAAC,UAAU,CAAC,OAAO,EAAE;oBAC1C,GAAG,cAAc;oBACjB,QAAQ,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,gBAAgB;iBACxC,CAAC,CAAC;YACL,CAAC;YACD,OAAO,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;QACnE,CAAC;KACF,EACD,MAAM,CACP,CAAC;IAEF,IAAI,CAAC,MAAM;QAAE,MAAM,IAAI,KAAK,CAAC,4DAA4D,CAAC,CAAC;IAC3F,OAAO,MAAM,CAAC;AAChB,CAAC;AAED;;GAEG;AACI,KAAK,UAAU,+BAA+B,CACnD,cAA+C,EAC/C,MAAqD,EACrD,OAAyC,EACzC,aAAwC,EACxC,gBAAwB,EACxB,QAA8B;IAE9B,IAAI,CAAC;QACH,uHAAuH;QACvH,MAAM,cAAc,GAAG,MAAM,cAAc,CAAC,MAAa,EAAE,OAAO,CAAC,CAAC;QAEpE,MAAM,CAAC,KAAK,CAAC,2CAA2C,EAAE;YACxD,KAAK,EAAE,MAAM,CAAC,KAAK;SACpB,CAAC,CAAC;QAEH,0DAA0D;QAC1D,gFAAgF;QAChF,OAAO,oCAAoC,CACzC,cAAmD,EACnD,aAAa,EACb,gBAAgB,EAChB,QAAQ,CACT,CAAC;IACJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,gBAAgB,CAAC;QAC/C,MAAM,CAAC,KAAK,CAAC,wCAAwC,EAAE;YACrD,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;YAC7D,QAAQ;SACT,CAAC,CAAC;QAEH,mDAAmD;QACnD,MAAM,KAAK,CAAC;IACd,CAAC;AACH,CAAC;AAED;;;GAGG;AACH,KAAK,SAAS,CAAC,CAAC,oCAAoC,CAClD,cAAsC,EACtC,aAAwC,EACxC,gBAAwB,EACxB,QAA8B;IAE9B,IAAI,cAAkC,CAAC;IACvC,IAAI,UAAU,GAAQ,IAAI,CAAC;IAC3B,IAAI,KAAK,GAAG,EAAE,CAAC;IACf,IAAI,SAAS,GAAG,EAAE,CAAC;IACnB,IAAI,YAAY,GAAkB,IAAI,CAAC;IAEvC,IAAI,CAAC;QACH,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,cAAc,EAAE,CAAC;YACzC,6BAA6B;YAC7B,IAAI,CAAC,cAAc,EAAE,CAAC;gBACpB,cAAc,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAC9B,CAAC;YAED,sEAAsE;YACtE,IAAI,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;gBACvC,MAAM,QAAQ,GAAG,KAAY,CAAC;gBAE9B,kCAAkC;gBAClC,IAAI,QAAQ,CAAC,KAAK;oBAAE,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC;gBAC3C,IAAI,QAAQ,CAAC,EAAE;oBAAE,SAAS,GAAG,QAAQ,CAAC,EAAE,CAAC;gBAEzC,oCAAoC;gBACpC,IAAI,QAAQ,CAAC,KAAK,EAAE,CAAC;oBACnB,UAAU,GAAG,QAAQ,CAAC,KAAK,CAAC;gBAC9B,CAAC;gBAED,0BAA0B;gBAC1B,IAAI,QAAQ,CAAC,aAAa,EAAE,CAAC;oBAC3B,YAAY,GAAG,QAAQ,CAAC,aAAa,CAAC;gBACxC,CAAC;YACH,CAAC;YAED,MAAM,KAAK,CAAC;QACd,CAAC;QAED,qCAAqC;QACrC,IAAI,UAAU,EAAE,CAAC;YACf,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,gBAAgB,CAAC;YAC/C,MAAM,gBAAgB,GAAG,cAAc,CAAC,CAAC,CAAC,cAAc,GAAG,gBAAgB,CAAC,CAAC,CAAC,SAAS,CAAC;YAExF,IAAA,0BAAe,EAAC;gBACd,SAAS,EAAE,SAAS,IAAI,IAAA,mBAAU,GAAE;gBACpC,KAAK,EAAE,KAAK;gBACZ,YAAY,EAAE,UAAU,CAAC,YAAY,IAAI,CAAC;gBAC1C,gBAAgB,EAAE,UAAU,CAAC,aAAa,IAAI,CAAC;gBAC/C,WAAW,EAAE,UAAU,CAAC,YAAY,IAAI,CAAC;gBACzC,eAAe,EAAE,UAAU,CAAC,gBAAgB;gBAC5C,YAAY,EAAE,UAAU,CAAC,aAAa;gBACtC,QAAQ;gBACR,YAAY,EAAE,YAAY,IAAI,WAAW;gBACzC,aAAa;gBACb,UAAU,EAAE,IAAI;gBAChB,gBAAgB;gBAChB,YAAY,EAAE,uCAAiB,CAAC,GAAG,CAAC,QAAQ,CAAC;aAC9C,CAAC,CAAC;YAEH,MAAM,CAAC,KAAK,CAAC,mCAAmC,EAAE;gBAChD,KAAK;gBACL,QAAQ;gBACR,gBAAgB;gBAChB,WAAW,EAAE,UAAU,CAAC,YAAY;aACrC,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,CAAC,KAAK,CAAC,uCAAuC,EAAE;YACpD,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;SAC9D,CAAC,CAAC;QACH,MAAM,KAAK,CAAC;IACd,CAAC;AACH,CAAC"}

package/dist/cjs/core/wrapper/stream-wrapper.js

@@ -1,82 +0,0 @@
-"use strict";
-/**
- * Stream Wrapper Module
- *
- * Handles wrapping of streaming responses for usage tracking.
- * Extracted from wrapper.ts for better organization.
- */
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.createTrackingStreamWrapper = createTrackingStreamWrapper;
-const type_guards_js_1 = require("../../utils/type-guards.js");
-const index_js_1 = require("../tracking/index.js");
-const index_js_2 = require("../config/index.js");
-const instance_patcher_js_1 = require("./instance-patcher.js");
-// Global logger
-const logger = (0, index_js_2.getLogger)();
-/**
- * Create a simple stream wrapper that tracks usage when complete
- */
-function createTrackingStreamWrapper(originalStream, usageMetadata, requestStartTime, instance) {
-    // For streaming, we need to collect the final response data
-    let accumulatedResponse = null;
-    // Create async iterator
-    const wrappedIterator = {
-        async *[Symbol.asyncIterator]() {
-            try {
-                for await (const chunk of originalStream) {
-                    // Validate and accumulate response data for tracking
-                    if ((0, type_guards_js_1.isStreamChunk)(chunk)) {
-                        if (!accumulatedResponse) {
-                            accumulatedResponse = {
-                                id: chunk.id,
-                                model: chunk.model,
-                                usage: { prompt_tokens: 0, completion_tokens: 0, total_tokens: 0 },
-                            };
-                        }
-                        // Update usage if available in chunk
-                        if (chunk.usage) {
-                            accumulatedResponse.usage = chunk.usage;
-                        }
-                    }
-                    // Forward the chunk to the client
-                    yield chunk;
-                }
-                // Stream completed - track usage
-                if (accumulatedResponse && accumulatedResponse.usage) {
-                    const duration = Date.now() - requestStartTime;
-                    // Get provider info for this instance
-                    const providerInfo = instance_patcher_js_1.instanceProviders.get(instance);
-                    // Safely access extended usage fields
-                    const usage = accumulatedResponse.usage;
-                    (0, index_js_1.trackUsageAsync)({
-                        requestId: accumulatedResponse.id,
-                        model: accumulatedResponse.model,
-                        promptTokens: usage.prompt_tokens,
-                        completionTokens: usage.completion_tokens || 0,
-                        totalTokens: usage.total_tokens,
-                        reasoningTokens: usage.reasoning_tokens,
-                        cachedTokens: usage.cached_tokens,
-                        duration,
-                        finishReason: null, // Will be determined from final chunk
-                        usageMetadata,
-                        isStreamed: true,
-                        providerInfo,
-                    });
-                    logger.debug('Chat completion streaming completed', {
-                        model: accumulatedResponse.model,
-                        duration,
-                        totalTokens: accumulatedResponse.usage.total_tokens,
-                    });
-                }
-            }
-            catch (error) {
-                logger.error('Chat completion streaming error', {
-                    error: error instanceof Error ? error.message : String(error),
-                });
-                throw error;
-            }
-        },
-    };
-    return wrappedIterator;
-}
-//# sourceMappingURL=stream-wrapper.js.map

package/dist/cjs/core/wrapper/stream-wrapper.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"stream-wrapper.js","sourceRoot":"","sources":["../../../../src/core/wrapper/stream-wrapper.ts"],"names":[],"mappings":";AAAA;;;;;GAKG;;AAmBH,kEA2EC;AAtFD,+DAA2D;AAC3D,mDAAuD;AACvD,iDAA+C;AAC/C,+DAA0D;AAE1D,gBAAgB;AAChB,MAAM,MAAM,GAAG,IAAA,oBAAS,GAAE,CAAC;AAE3B;;GAEG;AACH,SAAgB,2BAA2B,CACzC,cAAsC,EACtC,aAAwC,EACxC,gBAAwB,EACxB,QAA8B;IAE9B,4DAA4D;IAC5D,IAAI,mBAAmB,GAAuB,IAAI,CAAC;IAEnD,wBAAwB;IACxB,MAAM,eAAe,GAAG;QACtB,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,aAAa,CAAC;YAC3B,IAAI,CAAC;gBACH,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,cAAc,EAAE,CAAC;oBACzC,qDAAqD;oBACrD,IAAI,IAAA,8BAAa,EAAC,KAAK,CAAC,EAAE,CAAC;wBACzB,IAAI,CAAC,mBAAmB,EAAE,CAAC;4BACzB,mBAAmB,GAAG;gCACpB,EAAE,EAAE,KAAK,CAAC,EAAE;gCACZ,KAAK,EAAE,KAAK,CAAC,KAAK;gCAClB,KAAK,EAAE,EAAE,aAAa,EAAE,CAAC,EAAE,iBAAiB,EAAE,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE;6BACnE,CAAC;wBACJ,CAAC;wBAED,qCAAqC;wBACrC,IAAI,KAAK,CAAC,KAAK,EAAE,CAAC;4BAChB,mBAAmB,CAAC,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC;wBAC1C,CAAC;oBACH,CAAC;oBAED,kCAAkC;oBAClC,MAAM,KAAK,CAAC;gBACd,CAAC;gBAED,iCAAiC;gBACjC,IAAI,mBAAmB,IAAI,mBAAmB,CAAC,KAAK,EAAE,CAAC;oBACrD,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,gBAAgB,CAAC;oBAE/C,sCAAsC;oBACtC,MAAM,YAAY,GAAG,uCAAiB,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;oBAErD,sCAAsC;oBACtC,MAAM,KAAK,GAAG,mBAAmB,CAAC,KAAsB,CAAC;oBAEzD,IAAA,0BAAe,EAAC;wBACd,SAAS,EAAE,mBAAmB,CAAC,EAAE;wBACjC,KAAK,EAAE,mBAAmB,CAAC,KAAK;wBAChC,YAAY,EAAE,KAAK,CAAC,aAAa;wBACjC,gBAAgB,EAAE,KAAK,CAAC,iBAAiB,IAAI,CAAC;wBAC9C,WAAW,EAAE,KAAK,CAAC,YAAY;wBAC/B,eAAe,EAAE,KAAK,CAAC,gBAAgB;wBACvC,YAAY,EAAE,KAAK,CAAC,aAAa;wBACjC,QAAQ;wBACR,YAAY,EAAE,IAAI,EAAE,sCAAsC;wBAC1D,aAAa;wBACb,UAAU,EAAE,IAAI;wBAChB,YAAY;qBACb,CAAC,CAAC;oBAEH,MAAM,CAAC,KAAK,CAAC,qCAAqC,EAAE;wBAClD,KAAK,EAAE,mBAAmB,CAAC,KAAK;wBAChC,QAAQ;wBACR,WAAW,EAAE,mBAAmB,CAAC,KAAK,CAAC,YAAY;qBACpD,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,MAAM,CAAC,KAAK,CAAC,iCAAiC,EAAE;oBAC9C,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;iBAC9D,CAAC,CAAC;gBACH,MAAM,KAAK,CAAC;YACd,CAAC;QACH,CAAC;KACF,CAAC;IAEF,OAAO,eAAe,CAAC;AACzB,CAAC"}