@revenium/perplexity 2.0.3 → 2.0.5

package/dist/cjs/core/wrapper/perplexity-client.js

@@ -1,177 +0,0 @@
-"use strict";
-/**
- * Perplexity Client Wrapper
- * Wraps Perplexity API calls with Revenium metering
- */
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.createChatCompletion = createChatCompletion;
-exports.createStreamingChatCompletion = createStreamingChatCompletion;
-const openai_1 = __importDefault(require("openai"));
-const perplexity_config_1 = require("../config/perplexity-config");
-const logger_1 = require("../../utils/logger");
-const metering_1 = require("../tracking/metering");
-let perplexityClient = null;
-/**
- * Get or create Perplexity client
- */
-function getClient() {
-    if (!perplexityClient) {
-        const config = (0, perplexity_config_1.getPerplexityConfig)();
-        if (!config) {
-            throw new Error("Perplexity configuration not initialized. Call initializePerplexityFromEnv() or initializePerplexity() first.");
-        }
-        perplexityClient = new openai_1.default({
-            apiKey: config.apiKey,
-            baseURL: config.baseUrl,
-        });
-        logger_1.logger.info("Perplexity client initialized successfully");
-    }
-    return perplexityClient;
-}
-/**
- * Create a chat completion
- */
-async function createChatCompletion(params) {
-    const client = getClient();
-    const transactionId = (0, metering_1.generateTransactionId)();
-    const startTime = new Date();
-    try {
-        const { usageMetadata, ...requestParams } = params;
-        logger_1.logger.info(`Creating chat completion with model: ${params.model}`);
-        const response = (await client.chat.completions.create({
-            model: requestParams.model,
-            messages: requestParams.messages,
-            max_tokens: requestParams.maxTokens,
-            temperature: requestParams.temperature,
-            top_p: requestParams.topP,
-            presence_penalty: requestParams.presencePenalty,
-            frequency_penalty: requestParams.frequencyPenalty,
-            stream: false,
-        }));
-        const endTime = new Date();
-        // Extract response data
-        const content = response.choices[0]?.message?.content || "";
-        const role = response.choices[0]?.message?.role || "assistant";
-        const finishReason = response.choices[0]?.finish_reason || null;
-        const usage = response.usage || {
-            prompt_tokens: 0,
-            completion_tokens: 0,
-            total_tokens: 0,
-        };
-        // Build and send metering data
-        const meteringData = (0, metering_1.buildMeteringData)({
-            model: response.model,
-            startTime,
-            endTime,
-            inputTokens: usage.prompt_tokens,
-            outputTokens: usage.completion_tokens,
-            totalTokens: usage.total_tokens,
-            transactionId,
-            isStreamed: false,
-            stopReason: finishReason || "END",
-            usageMetadata,
-        });
-        // Send metering data asynchronously (don't wait)
-        (0, metering_1.sendMeteringData)(meteringData).catch((error) => {
-            logger_1.logger.error("Failed to send metering data:", error);
-        });
-        return {
-            content,
-            role,
-            finishReason,
-            usage: {
-                promptTokens: usage.prompt_tokens,
-                completionTokens: usage.completion_tokens,
-                totalTokens: usage.total_tokens,
-            },
-            model: response.model,
-            transactionId,
-            rawResponse: response,
-        };
-    }
-    catch (error) {
-        logger_1.logger.error("Error in chat completion:", error.message);
-        throw error;
-    }
-}
-/**
- * Create a streaming chat completion
- */
-async function createStreamingChatCompletion(params) {
-    const client = getClient();
-    const transactionId = (0, metering_1.generateTransactionId)();
-    const startTime = new Date();
-    try {
-        const { usageMetadata, ...requestParams } = params;
-        logger_1.logger.info(`Creating streaming chat completion with model: ${params.model}`);
-        const stream = await client.chat.completions.create({
-            model: requestParams.model,
-            messages: requestParams.messages,
-            max_tokens: requestParams.maxTokens,
-            temperature: requestParams.temperature,
-            top_p: requestParams.topP,
-            presence_penalty: requestParams.presencePenalty,
-            frequency_penalty: requestParams.frequencyPenalty,
-            stream: true,
-        });
-        // Wrap the stream to track metering
-        const wrappedStream = wrapStream(stream, transactionId, params.model, startTime, usageMetadata);
-        return {
-            stream: wrappedStream,
-            transactionId,
-            model: params.model,
-        };
-    }
-    catch (error) {
-        logger_1.logger.error("Error in streaming chat completion:", error.message);
-        throw error;
-    }
-}
-/**
- * Wrap stream to track metering data
- */
-async function* wrapStream(stream, transactionId, model, startTime, usageMetadata) {
-    let lastChunk = null;
-    let inputTokens = 0;
-    let outputTokens = 0;
-    let totalTokens = 0;
-    try {
-        for await (const chunk of stream) {
-            lastChunk = chunk;
-            // Track usage if available in chunk
-            if (chunk.usage) {
-                inputTokens = chunk.usage.prompt_tokens || 0;
-                outputTokens = chunk.usage.completion_tokens || 0;
-                totalTokens = chunk.usage.total_tokens || 0;
-            }
-            yield chunk;
-        }
-        // Send metering data when stream completes
-        const endTime = new Date();
-        const finishReason = lastChunk?.choices?.[0]?.finish_reason || "END";
-        const meteringData = (0, metering_1.buildMeteringData)({
-            model,
-            startTime,
-            endTime,
-            inputTokens,
-            outputTokens,
-            totalTokens,
-            transactionId,
-            isStreamed: true,
-            stopReason: finishReason,
-            usageMetadata,
-        });
-        // Send metering data asynchronously (don't wait)
-        (0, metering_1.sendMeteringData)(meteringData).catch((error) => {
-            logger_1.logger.error("Failed to send metering data:", error);
-        });
-    }
-    catch (error) {
-        logger_1.logger.error("Error in stream processing:", error.message);
-        throw error;
-    }
-}
-//# sourceMappingURL=perplexity-client.js.map

package/dist/cjs/utils/logger.js

@@ -1,23 +0,0 @@
-"use strict";
-/**
- * Simple logger utility
- */
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.logger = void 0;
-exports.logger = {
-    info: (...args) => {
-        console.log("[Revenium]", ...args);
-    },
-    warn: (...args) => {
-        console.warn("[Revenium Warning]", ...args);
-    },
-    error: (...args) => {
-        console.error("[Revenium Error]", ...args);
-    },
-    debug: (...args) => {
-        if (process.env.DEBUG === "true") {
-            console.log("[Revenium Debug]", ...args);
-        }
-    },
-};
-//# sourceMappingURL=logger.js.map

package/dist/esm/core/config/perplexity-config.js

@@ -1,40 +0,0 @@
-/**
- * Perplexity Configuration Module
- * Manages Perplexity API configuration
- */
-import { logger } from "../../utils/logger";
-let perplexityConfig = null;
-/**
- * Initialize Perplexity configuration from environment variables
- */
-export function initializePerplexityFromEnv() {
-    const apiKey = process.env.PERPLEXITY_API_KEY;
-    const baseUrl = process.env.PERPLEXITY_API_BASE_URL || "https://api.perplexity.ai";
-    if (!apiKey) {
-        throw new Error("PERPLEXITY_API_KEY environment variable is required but not set");
-    }
-    perplexityConfig = {
-        apiKey,
-        baseUrl,
-    };
-    logger.info("Perplexity configuration initialized successfully");
-    return perplexityConfig;
-}
-/**
- * Initialize Perplexity configuration with custom values
- */
-export function initializePerplexity(config) {
-    perplexityConfig = {
-        apiKey: config.apiKey,
-        baseUrl: config.baseUrl || "https://api.perplexity.ai",
-    };
-    logger.info("Perplexity configuration initialized with custom config");
-    return perplexityConfig;
-}
-/**
- * Get current Perplexity configuration
- */
-export function getPerplexityConfig() {
-    return perplexityConfig;
-}
-//# sourceMappingURL=perplexity-config.js.map

package/dist/esm/core/config/revenium-config.js

@@ -1,72 +0,0 @@
-/**
- * Revenium Configuration Module
- * Manages Revenium metering configuration and initialization
- */
-import { logger } from "../../utils/logger";
-let reveniumConfig = null;
-/**
- * Initialize Revenium configuration from environment variables
- */
-export function initializeReveniumFromEnv() {
-    const meteringApiKey = process.env.REVENIUM_METERING_API_KEY?.trim().replace(/^["']|["']$/g, "");
-    const meteringBaseUrl = process.env.REVENIUM_METERING_BASE_URL?.trim().replace(/^["']|["']$/g, "");
-    if (!meteringApiKey || !meteringBaseUrl) {
-        logger.warn("Revenium metering disabled: Missing REVENIUM_METERING_API_KEY or REVENIUM_METERING_BASE_URL");
-        reveniumConfig = {
-            meteringApiKey: "",
-            meteringBaseUrl: "",
-            enabled: false,
-        };
-        return reveniumConfig;
-    }
-    reveniumConfig = {
-        meteringApiKey,
-        meteringBaseUrl,
-        enabled: true,
-    };
-    logger.info("Revenium configuration initialized successfully");
-    logger.debug(`Metering Base URL: ${meteringBaseUrl}`);
-    return reveniumConfig;
-}
-/**
- * Initialize Revenium configuration with custom values
- */
-export function initializeRevenium(config) {
-    reveniumConfig = {
-        ...config,
-        enabled: true,
-    };
-    logger.info("Revenium configuration initialized with custom config");
-    return reveniumConfig;
-}
-/**
- * Get current Revenium configuration
- */
-export function getReveniumConfig() {
-    return reveniumConfig;
-}
-/**
- * Check if Revenium is enabled
- */
-export function isReveniumEnabled() {
-    return reveniumConfig?.enabled ?? false;
-}
-/**
- * Disable Revenium metering
- */
-export function disableRevenium() {
-    if (reveniumConfig) {
-        reveniumConfig.enabled = false;
-        logger.info("Revenium metering disabled");
-    }
-}
-/**
- * Enable Revenium metering
- */
-export function enableRevenium() {
-    if (reveniumConfig) {
-        reveniumConfig.enabled = true;
-        logger.info("Revenium metering enabled");
-    }
-}
-//# sourceMappingURL=revenium-config.js.map

package/dist/esm/core/tracking/metering.js

@@ -1,126 +0,0 @@
-/**
- * Revenium Metering Module
- * Handles tracking and sending metering data to Revenium
- */
-import { getReveniumConfig, isReveniumEnabled, } from "../config/revenium-config";
-import { logger } from "../../utils/logger";
-/**
- * Generate a unique transaction ID
- */
-export function generateTransactionId() {
-    return `txn_${Date.now()}_${Math.random().toString(36).substring(2, 15)}`;
-}
-/**
- * Format timestamp to ISO string
- */
-function formatTimestamp(date) {
-    return date.toISOString();
-}
-/**
- * Calculate duration in milliseconds
- */
-function calculateDuration(startTime, endTime) {
-    return endTime.getTime() - startTime.getTime();
-}
-/**
- * Build metering data from request information
- */
-export function buildMeteringData(params) {
-    const { model, startTime, endTime, inputTokens, outputTokens, totalTokens, reasoningTokens = 0, cachedTokens = 0, transactionId, isStreamed, stopReason, usageMetadata = {}, } = params;
-    const config = getReveniumConfig();
-    const agent = "perplexity";
-    const costType = "TOKEN";
-    const operationType = "CHAT";
-    // Build subscriber information
-    const subscriberId = usageMetadata.subscriber?.id ||
-        usageMetadata.subscriberId ||
-        `user-${generateTransactionId()}`;
-    const subscriberEmail = usageMetadata.subscriber?.email ||
-        usageMetadata.subscriberEmail ||
-        `user@${agent}.ai`;
-    const subscriberCredential = usageMetadata.subscriber?.credential ||
-        (usageMetadata.subscriberCredentialName &&
-            usageMetadata.subscriberCredential
-            ? {
-                name: usageMetadata.subscriberCredentialName,
-                value: usageMetadata.subscriberCredential,
-            }
-            : {
-                name: "default",
-                value: "default-credential",
-            });
-    const subscriber = {
-        id: subscriberId,
-        email: subscriberEmail,
-        credential: subscriberCredential,
-    };
-    // Build the payload in the exact order expected by Revenium API v2
-    const payload = {
-        stopReason: "END",
-        costType: "AI", // Fixed value as per Google middleware
-        isStreamed,
-        taskType: "AI", // Fixed value as per Google middleware
-        agent: usageMetadata.agent || agent,
-        operationType: usageMetadata.operationType || operationType,
-        inputTokenCount: usageMetadata.inputTokenCount ?? inputTokens,
-        outputTokenCount: usageMetadata.outputTokenCount ?? outputTokens,
-        reasoningTokenCount: usageMetadata.reasoningTokenCount ?? reasoningTokens,
-        cacheCreationTokenCount: usageMetadata.cacheCreationTokenCount ?? cachedTokens,
-        cacheReadTokenCount: usageMetadata.cacheReadTokenCount ?? 0,
-        totalTokenCount: usageMetadata.totalTokenCount ?? totalTokens,
-        organizationId: usageMetadata.organizationId || "my-customers-name",
-        productId: usageMetadata.productId || "free-trial",
-        subscriber,
-        model,
-        transactionId: usageMetadata.transactionId || transactionId,
-        responseTime: usageMetadata.responseTime || formatTimestamp(endTime),
-        requestDuration: calculateDuration(startTime, endTime),
-        provider: agent,
-        requestTime: usageMetadata.requestTime || formatTimestamp(startTime),
-        completionStartTime: usageMetadata.completionStartTime || formatTimestamp(endTime),
-        timeToFirstToken: usageMetadata.timeToFirstToken || 0,
-        middleware_source: "node",
-    };
-    // Only add traceId if it's provided
-    if (usageMetadata.traceId) {
-        payload.traceId = usageMetadata.traceId;
-    }
-    return payload;
-}
-/**
- * Send metering data to Revenium
- */
-export async function sendMeteringData(meteringData) {
-    if (!isReveniumEnabled()) {
-        logger.debug("Revenium metering is disabled, skipping metering data");
-        return;
-    }
-    const config = getReveniumConfig();
-    if (!config) {
-        logger.warn("Revenium configuration not initialized");
-        return;
-    }
-    try {
-        const url = `${config.meteringBaseUrl}/v2/ai/completions`;
-        logger.debug("Sending metering data to Revenium:", JSON.stringify(meteringData, null, 2));
-        const response = await fetch(url, {
-            method: "POST",
-            headers: {
-                "Content-Type": "application/json",
-                "x-api-key": config.meteringApiKey,
-                accept: "application/json",
-            },
-            body: JSON.stringify(meteringData),
-        });
-        if (!response.ok) {
-            const errorData = await response.text();
-            logger.error(`Metering API request failed with status ${response.status}: ${errorData}`);
-            return;
-        }
-        logger.info("Metering data sent successfully to Revenium");
-    }
-    catch (error) {
-        logger.error("Error sending metering data:", error.message);
-    }
-}
-//# sourceMappingURL=metering.js.map

package/dist/esm/core/wrapper/perplexity-client.js

@@ -1,170 +0,0 @@
-/**
- * Perplexity Client Wrapper
- * Wraps Perplexity API calls with Revenium metering
- */
-import OpenAI from "openai";
-import { getPerplexityConfig } from "../config/perplexity-config";
-import { logger } from "../../utils/logger";
-import { buildMeteringData, generateTransactionId, sendMeteringData, } from "../tracking/metering";
-let perplexityClient = null;
-/**
- * Get or create Perplexity client
- */
-function getClient() {
-    if (!perplexityClient) {
-        const config = getPerplexityConfig();
-        if (!config) {
-            throw new Error("Perplexity configuration not initialized. Call initializePerplexityFromEnv() or initializePerplexity() first.");
-        }
-        perplexityClient = new OpenAI({
-            apiKey: config.apiKey,
-            baseURL: config.baseUrl,
-        });
-        logger.info("Perplexity client initialized successfully");
-    }
-    return perplexityClient;
-}
-/**
- * Create a chat completion
- */
-export async function createChatCompletion(params) {
-    const client = getClient();
-    const transactionId = generateTransactionId();
-    const startTime = new Date();
-    try {
-        const { usageMetadata, ...requestParams } = params;
-        logger.info(`Creating chat completion with model: ${params.model}`);
-        const response = (await client.chat.completions.create({
-            model: requestParams.model,
-            messages: requestParams.messages,
-            max_tokens: requestParams.maxTokens,
-            temperature: requestParams.temperature,
-            top_p: requestParams.topP,
-            presence_penalty: requestParams.presencePenalty,
-            frequency_penalty: requestParams.frequencyPenalty,
-            stream: false,
-        }));
-        const endTime = new Date();
-        // Extract response data
-        const content = response.choices[0]?.message?.content || "";
-        const role = response.choices[0]?.message?.role || "assistant";
-        const finishReason = response.choices[0]?.finish_reason || null;
-        const usage = response.usage || {
-            prompt_tokens: 0,
-            completion_tokens: 0,
-            total_tokens: 0,
-        };
-        // Build and send metering data
-        const meteringData = buildMeteringData({
-            model: response.model,
-            startTime,
-            endTime,
-            inputTokens: usage.prompt_tokens,
-            outputTokens: usage.completion_tokens,
-            totalTokens: usage.total_tokens,
-            transactionId,
-            isStreamed: false,
-            stopReason: finishReason || "END",
-            usageMetadata,
-        });
-        // Send metering data asynchronously (don't wait)
-        sendMeteringData(meteringData).catch((error) => {
-            logger.error("Failed to send metering data:", error);
-        });
-        return {
-            content,
-            role,
-            finishReason,
-            usage: {
-                promptTokens: usage.prompt_tokens,
-                completionTokens: usage.completion_tokens,
-                totalTokens: usage.total_tokens,
-            },
-            model: response.model,
-            transactionId,
-            rawResponse: response,
-        };
-    }
-    catch (error) {
-        logger.error("Error in chat completion:", error.message);
-        throw error;
-    }
-}
-/**
- * Create a streaming chat completion
- */
-export async function createStreamingChatCompletion(params) {
-    const client = getClient();
-    const transactionId = generateTransactionId();
-    const startTime = new Date();
-    try {
-        const { usageMetadata, ...requestParams } = params;
-        logger.info(`Creating streaming chat completion with model: ${params.model}`);
-        const stream = await client.chat.completions.create({
-            model: requestParams.model,
-            messages: requestParams.messages,
-            max_tokens: requestParams.maxTokens,
-            temperature: requestParams.temperature,
-            top_p: requestParams.topP,
-            presence_penalty: requestParams.presencePenalty,
-            frequency_penalty: requestParams.frequencyPenalty,
-            stream: true,
-        });
-        // Wrap the stream to track metering
-        const wrappedStream = wrapStream(stream, transactionId, params.model, startTime, usageMetadata);
-        return {
-            stream: wrappedStream,
-            transactionId,
-            model: params.model,
-        };
-    }
-    catch (error) {
-        logger.error("Error in streaming chat completion:", error.message);
-        throw error;
-    }
-}
-/**
- * Wrap stream to track metering data
- */
-async function* wrapStream(stream, transactionId, model, startTime, usageMetadata) {
-    let lastChunk = null;
-    let inputTokens = 0;
-    let outputTokens = 0;
-    let totalTokens = 0;
-    try {
-        for await (const chunk of stream) {
-            lastChunk = chunk;
-            // Track usage if available in chunk
-            if (chunk.usage) {
-                inputTokens = chunk.usage.prompt_tokens || 0;
-                outputTokens = chunk.usage.completion_tokens || 0;
-                totalTokens = chunk.usage.total_tokens || 0;
-            }
-            yield chunk;
-        }
-        // Send metering data when stream completes
-        const endTime = new Date();
-        const finishReason = lastChunk?.choices?.[0]?.finish_reason || "END";
-        const meteringData = buildMeteringData({
-            model,
-            startTime,
-            endTime,
-            inputTokens,
-            outputTokens,
-            totalTokens,
-            transactionId,
-            isStreamed: true,
-            stopReason: finishReason,
-            usageMetadata,
-        });
-        // Send metering data asynchronously (don't wait)
-        sendMeteringData(meteringData).catch((error) => {
-            logger.error("Failed to send metering data:", error);
-        });
-    }
-    catch (error) {
-        logger.error("Error in stream processing:", error.message);
-        throw error;
-    }
-}
-//# sourceMappingURL=perplexity-client.js.map

package/dist/esm/utils/logger.js

@@ -1,20 +0,0 @@
-/**
- * Simple logger utility
- */
-export const logger = {
-    info: (...args) => {
-        console.log("[Revenium]", ...args);
-    },
-    warn: (...args) => {
-        console.warn("[Revenium Warning]", ...args);
-    },
-    error: (...args) => {
-        console.error("[Revenium Error]", ...args);
-    },
-    debug: (...args) => {
-        if (process.env.DEBUG === "true") {
-            console.log("[Revenium Debug]", ...args);
-        }
-    },
-};
-//# sourceMappingURL=logger.js.map

package/dist/types/core/config/perplexity-config.d.ts

@@ -1,24 +0,0 @@
-/**
- * Perplexity Configuration Module
- * Manages Perplexity API configuration
- */
-export interface PerplexityConfig {
-    apiKey: string;
-    baseUrl: string;
-}
-/**
- * Initialize Perplexity configuration from environment variables
- */
-export declare function initializePerplexityFromEnv(): PerplexityConfig;
-/**
- * Initialize Perplexity configuration with custom values
- */
-export declare function initializePerplexity(config: {
-    apiKey: string;
-    baseUrl?: string;
-}): PerplexityConfig;
-/**
- * Get current Perplexity configuration
- */
-export declare function getPerplexityConfig(): PerplexityConfig | null;
-//# sourceMappingURL=perplexity-config.d.ts.map

package/dist/types/core/config/revenium-config.d.ts

@@ -1,37 +0,0 @@
-/**
- * Revenium Configuration Module
- * Manages Revenium metering configuration and initialization
- */
-export interface ReveniumConfig {
-    meteringApiKey: string;
-    meteringBaseUrl: string;
-    enabled: boolean;
-}
-/**
- * Initialize Revenium configuration from environment variables
- */
-export declare function initializeReveniumFromEnv(): ReveniumConfig;
-/**
- * Initialize Revenium configuration with custom values
- */
-export declare function initializeRevenium(config: {
-    meteringApiKey: string;
-    meteringBaseUrl: string;
-}): ReveniumConfig;
-/**
- * Get current Revenium configuration
- */
-export declare function getReveniumConfig(): ReveniumConfig | null;
-/**
- * Check if Revenium is enabled
- */
-export declare function isReveniumEnabled(): boolean;
-/**
- * Disable Revenium metering
- */
-export declare function disableRevenium(): void;
-/**
- * Enable Revenium metering
- */
-export declare function enableRevenium(): void;
-//# sourceMappingURL=revenium-config.d.ts.map