@revenium/perplexity 1.0.22 → 1.0.25

package/dist/interfaces/meteringResponse.d.ts

@@ -1,27 +1,28 @@
-import { ISubscriber } from "./subscriber";
-export interface IMeteringResponse {
-    stopReason: string;
-    costType: string;
-    isStreamed: boolean;
-    taskType: string;
-    agent: string;
-    operationType: string;
-    inputTokenCount: number;
-    outputTokenCount: number;
-    reasoningTokenCount: number;
-    cacheCreationTokenCount: number;
-    cacheReadTokenCount: number;
-    totalTokenCount: number;
-    organizationId: string;
-    productId: string;
-    subscriber: ISubscriber;
-    model: string;
-    transactionId: string;
-    responseTime: string;
-    requestDuration: number;
-    provider: string;
-    requestTime: string;
-    completionStartTime: string;
-    timeToFirstToken: number;
-    middleware_source: string;
-}
+import { ISubscriber } from "./subscriber";
+export interface IMeteringResponse {
+    stopReason: string;
+    costType: string;
+    isStreamed: boolean;
+    taskType: string;
+    agent: string;
+    operationType: string;
+    inputTokenCount: number;
+    outputTokenCount: number;
+    reasoningTokenCount: number;
+    cacheCreationTokenCount: number;
+    cacheReadTokenCount: number;
+    totalTokenCount: number;
+    organizationId: string;
+    productId: string;
+    subscriber: ISubscriber;
+    model: string;
+    transactionId: string;
+    responseTime: string;
+    requestDuration: number;
+    provider: string;
+    requestTime: string;
+    completionStartTime: string;
+    timeToFirstToken: number;
+    middleware_source: string;
+    traceId?: string;
+}

package/dist/interfaces/meteringResponse.js

@@ -1,2 +1,2 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });

package/dist/models/Metering.js

@@ -3,7 +3,7 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.Metering = void 0;
 const utils_1 = require("../utils");
 const calculateDurationMs_1 = require("../utils/calculateDurationMs");
-const formatTimeStamp_1 = require("../utils/formatTimeStamp");
+const formatTimestamp_1 = require("../utils/formatTimestamp");
 const generateTransactionId_1 = require("../utils/generateTransactionId");
 const Logger_1 = require("./Logger");
 class Metering {
@@ -12,6 +12,8 @@ class Metering {
         this.apiKey = "";
         this.sendMeteringData = async (metering) => {
             try {
+                // Log the metering data being sent for debugging
+                Logger_1.logger.info("Sending metering data to Revenium:", JSON.stringify(metering, null, 2));
                 const response = await fetch(`${this.endpoint}/v2/ai/completions`, {
                     method: "POST",
                     headers: {
@@ -36,27 +38,27 @@ class Metering {
         this.endpoint = clientEndpoint;
     }
     createMetering(metering, isStreamed) {
-        var _a, _b, _c, _d, _e, _f, _g, _h, _j, _k, _l, _m, _o, _p, _q, _r, _s, _t, _u, _v, _w;
+        var _a, _b, _c, _d, _e, _f, _g, _h, _j, _k, _l, _m, _o, _p, _q, _r, _s, _t, _u, _v, _w, _x;
         const usageMetadata = metering.usageMetadata;
         const agent = "perplexity";
         return {
             stopReason: (_a = usageMetadata === null || usageMetadata === void 0 ? void 0 : usageMetadata.stopReason) !== null && _a !== void 0 ? _a : metering.stopReason,
             costType: utils_1.COST_TYPE,
             isStreamed,
-            taskType: utils_1.COST_TYPE,
-            agent: (_b = usageMetadata === null || usageMetadata === void 0 ? void 0 : usageMetadata.agent) !== null && _b !== void 0 ? _b : agent,
-            operationType: (_c = usageMetadata === null || usageMetadata === void 0 ? void 0 : usageMetadata.operationType) !== null && _c !== void 0 ? _c : metering.operationType.toString(),
-            inputTokenCount: (_d = usageMetadata === null || usageMetadata === void 0 ? void 0 : usageMetadata.inputTokenCount) !== null && _d !== void 0 ? _d : metering.tokenCounts.inputTokens,
-            outputTokenCount: (_e = usageMetadata === null || usageMetadata === void 0 ? void 0 : usageMetadata.outputTokenCount) !== null && _e !== void 0 ? _e : metering.tokenCounts.outputTokens,
-            reasoningTokenCount: (_g = (_f = usageMetadata === null || usageMetadata === void 0 ? void 0 : usageMetadata.reasoningTokenCount) !== null && _f !== void 0 ? _f : metering.tokenCounts.reasoningTokens) !== null && _g !== void 0 ? _g : 0,
-            cacheCreationTokenCount: (_j = (_h = usageMetadata === null || usageMetadata === void 0 ? void 0 : usageMetadata.cacheCreationTokenCount) !== null && _h !== void 0 ? _h : metering.tokenCounts.cachedTokens) !== null && _j !== void 0 ? _j : 0,
-            cacheReadTokenCount: (_k = usageMetadata === null || usageMetadata === void 0 ? void 0 : usageMetadata.cacheReadTokenCount) !== null && _k !== void 0 ? _k : 0,
-            totalTokenCount: (_l = usageMetadata === null || usageMetadata === void 0 ? void 0 : usageMetadata.totalTokenCount) !== null && _l !== void 0 ? _l : metering.tokenCounts.totalTokens,
-            organizationId: (_m = usageMetadata === null || usageMetadata === void 0 ? void 0 : usageMetadata.organizationId) !== null && _m !== void 0 ? _m : `my-customer-name-${(0, generateTransactionId_1.generateTransactionId)()}`,
-            productId: (_o = usageMetadata === null || usageMetadata === void 0 ? void 0 : usageMetadata.productId) !== null && _o !== void 0 ? _o : utils_1.PRODUCT_ID_FREE,
+            taskType: (_b = usageMetadata === null || usageMetadata === void 0 ? void 0 : usageMetadata.taskType) !== null && _b !== void 0 ? _b : utils_1.COST_TYPE,
+            agent: (_c = usageMetadata === null || usageMetadata === void 0 ? void 0 : usageMetadata.agent) !== null && _c !== void 0 ? _c : agent,
+            operationType: (_d = usageMetadata === null || usageMetadata === void 0 ? void 0 : usageMetadata.operationType) !== null && _d !== void 0 ? _d : metering.operationType.toString(),
+            inputTokenCount: (_e = usageMetadata === null || usageMetadata === void 0 ? void 0 : usageMetadata.inputTokenCount) !== null && _e !== void 0 ? _e : metering.tokenCounts.inputTokens,
+            outputTokenCount: (_f = usageMetadata === null || usageMetadata === void 0 ? void 0 : usageMetadata.outputTokenCount) !== null && _f !== void 0 ? _f : metering.tokenCounts.outputTokens,
+            reasoningTokenCount: (_h = (_g = usageMetadata === null || usageMetadata === void 0 ? void 0 : usageMetadata.reasoningTokenCount) !== null && _g !== void 0 ? _g : metering.tokenCounts.reasoningTokens) !== null && _h !== void 0 ? _h : 0,
+            cacheCreationTokenCount: (_k = (_j = usageMetadata === null || usageMetadata === void 0 ? void 0 : usageMetadata.cacheCreationTokenCount) !== null && _j !== void 0 ? _j : metering.tokenCounts.cachedTokens) !== null && _k !== void 0 ? _k : 0,
+            cacheReadTokenCount: (_l = usageMetadata === null || usageMetadata === void 0 ? void 0 : usageMetadata.cacheReadTokenCount) !== null && _l !== void 0 ? _l : 0,
+            totalTokenCount: (_m = usageMetadata === null || usageMetadata === void 0 ? void 0 : usageMetadata.totalTokenCount) !== null && _m !== void 0 ? _m : metering.tokenCounts.totalTokens,
+            organizationId: (_o = usageMetadata === null || usageMetadata === void 0 ? void 0 : usageMetadata.organizationId) !== null && _o !== void 0 ? _o : `my-customer-name-${(0, generateTransactionId_1.generateTransactionId)()}`,
+            productId: (_p = usageMetadata === null || usageMetadata === void 0 ? void 0 : usageMetadata.productId) !== null && _p !== void 0 ? _p : utils_1.PRODUCT_ID_FREE,
             subscriber: {
-                id: (_p = usageMetadata === null || usageMetadata === void 0 ? void 0 : usageMetadata.subscriberId) !== null && _p !== void 0 ? _p : `user-${(0, generateTransactionId_1.generateTransactionId)()}`,
-                email: (_q = usageMetadata === null || usageMetadata === void 0 ? void 0 : usageMetadata.subscriberEmail) !== null && _q !== void 0 ? _q : `user-@${agent.toLowerCase()}.ai`,
+                id: (_q = usageMetadata === null || usageMetadata === void 0 ? void 0 : usageMetadata.subscriberId) !== null && _q !== void 0 ? _q : `user-${(0, generateTransactionId_1.generateTransactionId)()}`,
+                email: (_r = usageMetadata === null || usageMetadata === void 0 ? void 0 : usageMetadata.subscriberEmail) !== null && _r !== void 0 ? _r : `user-@${agent.toLowerCase()}.ai`,
                 credential: (usageMetadata === null || usageMetadata === void 0 ? void 0 : usageMetadata.subscriberCredentialName) &&
                     (usageMetadata === null || usageMetadata === void 0 ? void 0 : usageMetadata.subscriberCredential)
                     ? {
@@ -66,14 +68,15 @@ class Metering {
                     : utils_1.CURRENT_CREDENTIAL,
             },
             model: metering.modelName,
-            transactionId: (_r = usageMetadata === null || usageMetadata === void 0 ? void 0 : usageMetadata.transactionId) !== null && _r !== void 0 ? _r : (0, generateTransactionId_1.generateTransactionId)(),
-            responseTime: (0, formatTimeStamp_1.formatTimestamp)(metering.endTime),
+            transactionId: (_s = usageMetadata === null || usageMetadata === void 0 ? void 0 : usageMetadata.transactionId) !== null && _s !== void 0 ? _s : (0, generateTransactionId_1.generateTransactionId)(),
+            responseTime: (0, formatTimestamp_1.formatTimestamp)(metering.endTime),
             requestDuration: (0, calculateDurationMs_1.calculateDurationMs)(metering.startTime, metering.endTime),
             provider: agent,
-            requestTime: (_t = (_s = usageMetadata === null || usageMetadata === void 0 ? void 0 : usageMetadata.requestTime) === null || _s === void 0 ? void 0 : _s.toString()) !== null && _t !== void 0 ? _t : (0, formatTimeStamp_1.formatTimestamp)(metering.startTime),
-            completionStartTime: (_v = (_u = usageMetadata === null || usageMetadata === void 0 ? void 0 : usageMetadata.completionStartTime) === null || _u === void 0 ? void 0 : _u.toString()) !== null && _v !== void 0 ? _v : (0, formatTimeStamp_1.formatTimestamp)(metering.endTime),
-            timeToFirstToken: (_w = usageMetadata === null || usageMetadata === void 0 ? void 0 : usageMetadata.timeToFirstToken) !== null && _w !== void 0 ? _w : 0,
+            requestTime: (_u = (_t = usageMetadata === null || usageMetadata === void 0 ? void 0 : usageMetadata.requestTime) === null || _t === void 0 ? void 0 : _t.toString()) !== null && _u !== void 0 ? _u : (0, formatTimestamp_1.formatTimestamp)(metering.startTime),
+            completionStartTime: (_w = (_v = usageMetadata === null || usageMetadata === void 0 ? void 0 : usageMetadata.completionStartTime) === null || _v === void 0 ? void 0 : _v.toString()) !== null && _w !== void 0 ? _w : (0, formatTimestamp_1.formatTimestamp)(metering.endTime),
+            timeToFirstToken: (_x = usageMetadata === null || usageMetadata === void 0 ? void 0 : usageMetadata.timeToFirstToken) !== null && _x !== void 0 ? _x : 0,
             middleware_source: utils_1.MIDDLEWARE_SOURCE,
+            traceId: usageMetadata === null || usageMetadata === void 0 ? void 0 : usageMetadata.traceId,
         };
     }
 }

package/dist/v1/perplexityV1.service.js

@@ -0,0 +1,166 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.PerplexityV1Service = void 0;
+const openai_1 = __importDefault(require("openai"));
+const Logger_1 = require("../models/Logger");
+const Metering_1 = require("../models/Metering");
+const generateTransactionId_1 = require("../utils/generateTransactionId");
+const operation_1 = require("../interfaces/operation");
+const utils_1 = require("../utils");
+class PerplexityV1Service {
+    constructor() {
+        this.working = true;
+        this.client = new openai_1.default({
+            apiKey: utils_1.PERPLEXITY_API_KEY,
+            baseURL: utils_1.PERPLEXITY_API_BASE_URL,
+        });
+        Logger_1.logger.info(utils_1.PERPLEXITY_CLIENT_INITIALIZED_MESSAGE);
+    }
+    async createChatCompletion(params, model) {
+        var _a, _b, _c, _d, _e, _f, _g, _h, _j;
+        const transactionId = (0, generateTransactionId_1.generateTransactionId)();
+        const startTime = new Date();
+        try {
+            const { usageMetadata, ...openaiParams } = params;
+            const requestParams = {
+                ...openaiParams,
+                model,
+            };
+            const result = (await this.client.chat.completions.create({
+                ...requestParams,
+                stream: false,
+            }));
+            // Build V1 response format
+            const perplexityResponse = {
+                responses: [
+                    {
+                        text: ((_b = (_a = result.choices[0]) === null || _a === void 0 ? void 0 : _a.message) === null || _b === void 0 ? void 0 : _b.content) || "",
+                        role: "assistant",
+                        finishReason: (_c = result.choices[0]) === null || _c === void 0 ? void 0 : _c.finish_reason,
+                    },
+                ],
+                usageMetadata: {
+                    inputTokenCount: ((_d = result.usage) === null || _d === void 0 ? void 0 : _d.prompt_tokens) || 0,
+                    outputTokenCount: ((_e = result.usage) === null || _e === void 0 ? void 0 : _e.completion_tokens) || 0,
+                    totalTokenCount: ((_f = result.usage) === null || _f === void 0 ? void 0 : _f.total_tokens) || 0,
+                },
+                modelVersion: result.model,
+                transactionId,
+            };
+            // Send metering data (using current middleware logic)
+            if (this.working) {
+                const tokenCounts = {
+                    inputTokens: ((_g = result.usage) === null || _g === void 0 ? void 0 : _g.prompt_tokens) || 0,
+                    outputTokens: ((_h = result.usage) === null || _h === void 0 ? void 0 : _h.completion_tokens) || 0,
+                    totalTokens: ((_j = result.usage) === null || _j === void 0 ? void 0 : _j.total_tokens) || 0,
+                };
+                const endTime = new Date();
+                Logger_1.logger.info("Metering is working.");
+                const metering = new Metering_1.Metering(utils_1.REVENIUM_METERING_API_KEY !== null && utils_1.REVENIUM_METERING_API_KEY !== void 0 ? utils_1.REVENIUM_METERING_API_KEY : "", utils_1.REVENIUM_METERING_BASE_URL !== null && utils_1.REVENIUM_METERING_BASE_URL !== void 0 ? utils_1.REVENIUM_METERING_BASE_URL : "");
+                // Combine user metadata with transaction ID
+                const combinedMetadata = {
+                    ...usageMetadata,
+                    transactionId,
+                };
+                const getMetering = metering.createMetering({
+                    modelName: model,
+                    endTime,
+                    startTime,
+                    operationType: operation_1.IOperationType.CHAT,
+                    stopReason: "END",
+                    tokenCounts,
+                    usageMetadata: combinedMetadata,
+                }, false);
+                await metering.sendMeteringData(getMetering);
+            }
+            else {
+                Logger_1.logger.warning("Metering is not working. Check your configuration.");
+            }
+            return perplexityResponse;
+        }
+        catch (error) {
+            Logger_1.logger.error("Error in Perplexity V1 chat completion:", error);
+            throw error;
+        }
+    }
+    async createStreamingCompletion(params, model) {
+        const transactionId = (0, generateTransactionId_1.generateTransactionId)();
+        const startTime = new Date();
+        try {
+            const { usageMetadata, ...openaiParams } = params;
+            const requestParams = {
+                ...openaiParams,
+                model,
+            };
+            const stream = await this.client.chat.completions.create({
+                ...requestParams,
+                stream: true,
+            });
+            const wrappedStream = this.wrapStream(stream, transactionId, model, startTime, usageMetadata);
+            return {
+                stream: wrappedStream,
+                usageMetadata: {
+                    inputTokenCount: 0, // Will be updated when stream completes
+                    outputTokenCount: 0,
+                    totalTokenCount: 0,
+                },
+                modelVersion: model,
+                transactionId,
+            };
+        }
+        catch (error) {
+            Logger_1.logger.error("Error in Perplexity V1 streaming:", error);
+            throw error;
+        }
+    }
+    async *wrapStream(stream, transactionId, model, startTime, usageMetadata) {
+        let lastChunk = null;
+        try {
+            for await (const chunk of stream) {
+                lastChunk = chunk;
+                yield chunk;
+            }
+            // Send metering data when stream completes (using current middleware logic)
+            if (this.working) {
+                // For streaming, we'll use the last chunk's usage data if available
+                const tokenCounts = {
+                    inputTokens: 0, // Will be updated if lastChunk has usage data
+                    outputTokens: 0,
+                    totalTokens: 0,
+                };
+                const endTime = new Date();
+                Logger_1.logger.info("Metering is working.");
+                const metering = new Metering_1.Metering(utils_1.REVENIUM_METERING_API_KEY !== null && utils_1.REVENIUM_METERING_API_KEY !== void 0 ? utils_1.REVENIUM_METERING_API_KEY : "", utils_1.REVENIUM_METERING_BASE_URL !== null && utils_1.REVENIUM_METERING_BASE_URL !== void 0 ? utils_1.REVENIUM_METERING_BASE_URL : "");
+                // Combine user metadata with transaction ID
+                const combinedMetadata = {
+                    ...usageMetadata,
+                    transactionId,
+                };
+                const getMetering = metering.createMetering({
+                    modelName: model,
+                    endTime,
+                    startTime,
+                    operationType: operation_1.IOperationType.CHAT,
+                    stopReason: "END",
+                    tokenCounts,
+                    usageMetadata: combinedMetadata,
+                }, true);
+                await metering.sendMeteringData(getMetering);
+            }
+            else {
+                Logger_1.logger.warning("Metering is not working. Check your configuration.");
+            }
+        }
+        catch (error) {
+            Logger_1.logger.error("Error in Perplexity V1 stream processing:", error);
+            throw error;
+        }
+    }
+    setWorking(working) {
+        this.working = working;
+    }
+}
+exports.PerplexityV1Service = PerplexityV1Service;

package/dist/v2/perplexityV2.service.js

@@ -0,0 +1,178 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.PerplexityV2Service = void 0;
+const openai_1 = __importDefault(require("openai"));
+const Logger_1 = require("../models/Logger");
+const Metering_1 = require("../models/Metering");
+const generateTransactionId_1 = require("../utils/generateTransactionId");
+const operation_1 = require("../interfaces/operation");
+const utils_1 = require("../utils");
+class PerplexityV2Service {
+    constructor() {
+        this.working = true;
+        this.client = new openai_1.default({
+            apiKey: utils_1.PERPLEXITY_API_KEY,
+            baseURL: utils_1.PERPLEXITY_API_BASE_URL,
+        });
+        Logger_1.logger.info(`${utils_1.PERPLEXITY_CLIENT_INITIALIZED_MESSAGE} (V2)`);
+    }
+    async createChatCompletion(params, model) {
+        var _a, _b, _c;
+        const transactionId = (0, generateTransactionId_1.generateTransactionId)();
+        const startTime = Date.now();
+        try {
+            const { usageMetadata, ...openaiParams } = params;
+            const requestParams = {
+                ...openaiParams,
+                model,
+            };
+            const result = (await this.client.chat.completions.create({
+                ...requestParams,
+                stream: false,
+            }));
+            const endTime = Date.now();
+            const processingTime = endTime - startTime;
+            // Build V2 enhanced response format (following OpenAI Responses API)
+            const perplexityV2Response = {
+                id: result.id,
+                object: "chat.completion",
+                created: result.created,
+                model: result.model,
+                choices: result.choices,
+                usage: {
+                    prompt_tokens: ((_a = result.usage) === null || _a === void 0 ? void 0 : _a.prompt_tokens) || 0,
+                    completion_tokens: ((_b = result.usage) === null || _b === void 0 ? void 0 : _b.completion_tokens) || 0,
+                    total_tokens: ((_c = result.usage) === null || _c === void 0 ? void 0 : _c.total_tokens) || 0,
+                },
+                metadata: {
+                    transactionId,
+                    processingTime,
+                    version: "v2",
+                },
+            };
+            // Send metering data (modernized approach)
+            await this.sendMeteringData(transactionId, model, operation_1.IOperationType.CHAT, {
+                inputTokenCount: perplexityV2Response.usage.prompt_tokens,
+                outputTokenCount: perplexityV2Response.usage.completion_tokens,
+                totalTokenCount: perplexityV2Response.usage.total_tokens,
+            }, processingTime, usageMetadata);
+            return perplexityV2Response;
+        }
+        catch (error) {
+            Logger_1.logger.error("Error in Perplexity V2 chat completion:", error);
+            throw error;
+        }
+    }
+    async createStreamingCompletion(params, model) {
+        const transactionId = (0, generateTransactionId_1.generateTransactionId)();
+        const startTime = Date.now();
+        try {
+            const { usageMetadata, ...openaiParams } = params;
+            const requestParams = {
+                ...openaiParams,
+                model,
+            };
+            const stream = await this.client.chat.completions.create({
+                ...requestParams,
+                stream: true,
+            });
+            const wrappedStream = this.wrapStreamV2(stream, transactionId, model, startTime, usageMetadata);
+            return {
+                id: transactionId,
+                object: "chat.completion.chunk",
+                stream: wrappedStream,
+                metadata: {
+                    transactionId,
+                    startTime,
+                    version: "v2",
+                },
+            };
+        }
+        catch (error) {
+            Logger_1.logger.error("Error in Perplexity V2 streaming:", error);
+            throw error;
+        }
+    }
+    async *wrapStreamV2(stream, transactionId, model, startTime, usageMetadata) {
+        let lastChunk = null;
+        try {
+            for await (const chunk of stream) {
+                lastChunk = chunk;
+                // Transform to V2 format (following OpenAI Responses API)
+                const v2Chunk = {
+                    id: chunk.id,
+                    object: "chat.completion.chunk",
+                    created: chunk.created,
+                    model: chunk.model,
+                    choices: chunk.choices,
+                    usage: chunk.usage
+                        ? {
+                            prompt_tokens: chunk.usage.prompt_tokens || 0,
+                            completion_tokens: chunk.usage.completion_tokens || 0,
+                            total_tokens: chunk.usage.total_tokens || 0,
+                        }
+                        : undefined,
+                };
+                yield v2Chunk;
+            }
+            // Send metering data when stream completes
+            const endTime = Date.now();
+            const processingTime = endTime - startTime;
+            if (lastChunk === null || lastChunk === void 0 ? void 0 : lastChunk.usage) {
+                await this.sendMeteringData(transactionId, model, operation_1.IOperationType.CHAT, {
+                    inputTokenCount: lastChunk.usage.prompt_tokens || 0,
+                    outputTokenCount: lastChunk.usage.completion_tokens || 0,
+                    totalTokenCount: lastChunk.usage.total_tokens || 0,
+                }, processingTime, usageMetadata);
+            }
+        }
+        catch (error) {
+            Logger_1.logger.error("Error in Perplexity V2 stream processing:", error);
+            throw error;
+        }
+    }
+    async sendMeteringData(transactionId, model, operationType, usageData, processingTime, customMetadata) {
+        if (!this.working) {
+            Logger_1.logger.warning("Metering is not working. Check your configuration.");
+            return;
+        }
+        try {
+            const startTime = new Date(Date.now() - processingTime);
+            const endTime = new Date();
+            const metering = new Metering_1.Metering(utils_1.REVENIUM_METERING_API_KEY !== null && utils_1.REVENIUM_METERING_API_KEY !== void 0 ? utils_1.REVENIUM_METERING_API_KEY : "", utils_1.REVENIUM_METERING_BASE_URL !== null && utils_1.REVENIUM_METERING_BASE_URL !== void 0 ? utils_1.REVENIUM_METERING_BASE_URL : "");
+            // Convert IUsageMetadata to ITokenCounts for metering
+            const tokenCounts = {
+                inputTokens: usageData.inputTokenCount || 0,
+                outputTokens: usageData.outputTokenCount || 0,
+                totalTokens: usageData.totalTokenCount || 0,
+            };
+            // Combine custom metadata with transaction ID (token counts are handled separately)
+            const combinedMetadata = {
+                ...customMetadata,
+                transactionId,
+            };
+            // Use the same approach as V1 but with V2 enhancements
+            const meteringData = metering.createMetering({
+                modelName: model,
+                endTime,
+                startTime,
+                operationType,
+                stopReason: "END",
+                tokenCounts,
+                usageMetadata: combinedMetadata,
+            }, false);
+            await metering.sendMeteringData(meteringData);
+            Logger_1.logger.info("V2 Metering data sent successfully.");
+        }
+        catch (error) {
+            Logger_1.logger.error("Error sending V2 metering data:", error);
+        }
+    }
+    setWorking(working) {
+        this.working = working;
+    }
+}
+exports.PerplexityV2Service = PerplexityV2Service;

package/examples/v1/basic.ts

@@ -0,0 +1,50 @@
+import { PerplexityV1Controller } from "../../src/index";
+
+const basicV1Example = async () => {
+  console.log("=".repeat(50));
+  console.log(
+    "🔥 Perplexity V1 Basic Example (Compatible with current middleware)"
+  );
+  console.log("=".repeat(50));
+
+  try {
+    const controller = new PerplexityV1Controller();
+
+    // Basic chat completion
+    const result = await controller.createChat([
+      "What is artificial intelligence and how does it work?",
+    ]);
+
+    console.log("📝 Response:");
+    console.log(result.responses[0].text);
+
+    console.log("\n📊 Usage Metadata:");
+    console.log(`Input tokens: ${result.usageMetadata.inputTokenCount}`);
+    console.log(`Output tokens: ${result.usageMetadata.outputTokenCount}`);
+    console.log(`Total tokens: ${result.usageMetadata.totalTokenCount}`);
+
+    console.log("\n🔧 Model Info:");
+    console.log(`Model: ${result.modelVersion}`);
+    console.log(`Transaction ID: ${result.transactionId}`);
+
+    // Test model capabilities
+    console.log("\n🎯 Model Capabilities:");
+    const models = controller.getAvailableModels();
+    console.log(`Available models: ${models.slice(0, 3).join(", ")}...`);
+
+    const supportsChat = controller.supportsCapability("sonar-pro", "chat");
+    const supportsStreaming = controller.supportsCapability(
+      "sonar-pro",
+      "streaming"
+    );
+    const supportsOnline = controller.supportsCapability("sonar-pro", "online");
+
+    console.log(
+      `Sonar-pro capabilities: chat=${supportsChat}, streaming=${supportsStreaming}, online=${supportsOnline}`
+    );
+  } catch (error) {
+    console.error("❌ Error:", error);
+  }
+};
+
+basicV1Example();

package/examples/v1/chat.ts

@@ -0,0 +1,40 @@
+import { PerplexityV1Controller } from "../../src/index";
+
+const chatV1Example = async () => {
+  console.log("=".repeat(50));
+  console.log("💬 Perplexity V1 Multi-turn Chat Example");
+  console.log("=".repeat(50));
+
+  try {
+    const controller = new PerplexityV1Controller();
+
+    // Multi-turn conversation using message objects
+    const messages = [
+      { role: "user", content: "What are the benefits of renewable energy?" },
+      { role: "assistant", content: "Renewable energy offers several key benefits including environmental sustainability, reduced greenhouse gas emissions, and long-term cost savings." },
+      { role: "user", content: "Can you elaborate on the cost savings aspect?" }
+    ];
+
+    const result = await controller.createChat(messages, "sonar-pro");
+
+    console.log("📝 Multi-turn Chat Response:");
+    console.log(result.responses[0].text);
+    
+    console.log("\n📊 Usage Statistics:");
+    console.log(`Input tokens: ${result.usageMetadata.inputTokenCount}`);
+    console.log(`Output tokens: ${result.usageMetadata.outputTokenCount}`);
+    console.log(`Total tokens: ${result.usageMetadata.totalTokenCount}`);
+    
+    console.log("\n🔧 Model Information:");
+    const modelInfo = controller.getModelInfo("sonar-pro");
+    console.log(`Model: ${modelInfo.model}`);
+    console.log(`Version: ${modelInfo.version}`);
+    console.log(`Context Window: ${modelInfo.contextWindow} tokens`);
+    console.log(`Capabilities:`, modelInfo.capabilities);
+
+  } catch (error) {
+    console.error("❌ Error:", error);
+  }
+};
+
+chatV1Example();

package/examples/v1/metadata.ts

@@ -0,0 +1,49 @@
+import { PerplexityV1Controller, IUsageMetadata } from "../../src/index";
+
+const metadataV1Example = async () => {
+  console.log("=".repeat(50));
+  console.log("📊 Perplexity V1 Custom Metadata Example");
+  console.log("=".repeat(50));
+
+  try {
+    const controller = new PerplexityV1Controller();
+
+    // Custom metadata for tracking
+    const customMetadata: IUsageMetadata = {
+      traceId: "trace-12345",
+      taskType: "research-query",
+      subscriberEmail: "user@example.com",
+      subscriberId: "user-456",
+      organizationId: "org-789",
+      agent: "perplexity-v1-demo",
+    };
+
+    const result = await controller.createChat([
+      "Explain quantum computing in simple terms"
+    ], "sonar-pro", customMetadata);
+
+    console.log("📝 Response:");
+    console.log(result.responses[0].text);
+    
+    console.log("\n📊 Usage Metadata:");
+    console.log(`Input tokens: ${result.usageMetadata.inputTokenCount}`);
+    console.log(`Output tokens: ${result.usageMetadata.outputTokenCount}`);
+    console.log(`Total tokens: ${result.usageMetadata.totalTokenCount}`);
+    
+    console.log("\n🏷️ Custom Metadata Sent:");
+    console.log(`Trace ID: ${customMetadata.traceId}`);
+    console.log(`Task Type: ${customMetadata.taskType}`);
+    console.log(`Subscriber: ${customMetadata.subscriberEmail}`);
+    console.log(`Organization: ${customMetadata.organizationId}`);
+    console.log(`Agent: ${customMetadata.agent}`);
+    
+    console.log("\n🔧 Transaction Info:");
+    console.log(`Model: ${result.modelVersion}`);
+    console.log(`Transaction ID: ${result.transactionId}`);
+
+  } catch (error) {
+    console.error("❌ Error:", error);
+  }
+};
+
+metadataV1Example();