@revenium/perplexity 1.0.23 → 2.0.1

package/dist/types/index.d.ts

@@ -0,0 +1,34 @@
+/**
+ * Revenium Middleware for Perplexity
+ *
+ * A lightweight middleware that adds Revenium metering to Perplexity API calls.
+ *
+ * @example
+ * ```typescript
+ * import {
+ *   initializeReveniumFromEnv,
+ *   initializePerplexityFromEnv,
+ *   createChatCompletion,
+ * } from '@revenium/perplexity';
+ *
+ * // Initialize configurations
+ * initializeReveniumFromEnv();
+ * initializePerplexityFromEnv();
+ *
+ * // Create a chat completion
+ * const result = await createChatCompletion({
+ *   messages: [{ role: 'user', content: 'Hello!' }],
+ *   model: 'sonar-pro',
+ * });
+ *
+ * console.log(result.content);
+ * ```
+ */
+export { initializeReveniumFromEnv, initializeRevenium, getReveniumConfig, isReveniumEnabled, disableRevenium, enableRevenium, type ReveniumConfig, } from "./core/config/revenium-config";
+export { initializePerplexityFromEnv, initializePerplexity, getPerplexityConfig, type PerplexityConfig, } from "./core/config/perplexity-config";
+export { createChatCompletion, createStreamingChatCompletion, } from "./core/wrapper/perplexity-client";
+export { generateTransactionId, buildMeteringData, sendMeteringData, } from "./core/tracking/metering";
+export type { PerplexityMessage, PerplexityChatRequest, PerplexityUsage, PerplexityChoice, PerplexityResponse, PerplexityStreamChunk, UsageMetadata, MeteringData, ChatCompletionResult, StreamingResult, PerplexityModel, } from "./types";
+export { PERPLEXITY_MODELS, DEFAULT_MODEL } from "./types";
+export { logger } from "./utils/logger";
+//# sourceMappingURL=index.d.ts.map

package/dist/types/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;;;;;;GAyBG;AASH,OAAO,EACL,yBAAyB,EACzB,kBAAkB,EAClB,iBAAiB,EACjB,iBAAiB,EACjB,eAAe,EACf,cAAc,EACd,KAAK,cAAc,GACpB,MAAM,+BAA+B,CAAC;AAEvC,OAAO,EACL,2BAA2B,EAC3B,oBAAoB,EACpB,mBAAmB,EACnB,KAAK,gBAAgB,GACtB,MAAM,iCAAiC,CAAC;AAMzC,OAAO,EACL,oBAAoB,EACpB,6BAA6B,GAC9B,MAAM,kCAAkC,CAAC;AAE1C,OAAO,EACL,qBAAqB,EACrB,iBAAiB,EACjB,gBAAgB,GACjB,MAAM,0BAA0B,CAAC;AAMlC,YAAY,EACV,iBAAiB,EACjB,qBAAqB,EACrB,eAAe,EACf,gBAAgB,EAChB,kBAAkB,EAClB,qBAAqB,EACrB,aAAa,EACb,YAAY,EACZ,oBAAoB,EACpB,eAAe,EACf,eAAe,GAChB,MAAM,SAAS,CAAC;AAEjB,OAAO,EAAE,iBAAiB,EAAE,aAAa,EAAE,MAAM,SAAS,CAAC;AAM3D,OAAO,EAAE,MAAM,EAAE,MAAM,gBAAgB,CAAC"}

package/dist/types/types/index.d.ts

@@ -0,0 +1,159 @@
+/**
+ * TypeScript Type Definitions for Perplexity Middleware
+ */
+export interface PerplexityMessage {
+    role: "system" | "user" | "assistant";
+    content: string;
+}
+export interface PerplexityChatRequest {
+    model: string;
+    messages: PerplexityMessage[];
+    max_tokens?: number;
+    temperature?: number;
+    top_p?: number;
+    stream?: boolean;
+    presence_penalty?: number;
+    frequency_penalty?: number;
+}
+export interface PerplexityUsage {
+    prompt_tokens: number;
+    completion_tokens: number;
+    total_tokens: number;
+}
+export interface PerplexityChoice {
+    index: number;
+    message?: {
+        role: string;
+        content: string;
+    };
+    delta?: {
+        role?: string;
+        content?: string;
+    };
+    finish_reason: string | null;
+}
+export interface PerplexityResponse {
+    id: string;
+    object: string;
+    created: number;
+    model: string;
+    choices: PerplexityChoice[];
+    usage?: PerplexityUsage;
+}
+export interface PerplexityStreamChunk {
+    id: string;
+    object: string;
+    created: number;
+    model: string;
+    choices: PerplexityChoice[];
+}
+export interface UsageMetadata {
+    subscriber?: {
+        id: string;
+        email?: string;
+        credential?: {
+            name: string;
+            value: string;
+        };
+    };
+    organizationId?: string;
+    productId?: string;
+    transactionId?: string;
+    traceId?: string;
+    inputTokenCount?: number;
+    outputTokenCount?: number;
+    totalTokenCount?: number;
+    reasoningTokenCount?: number;
+    cacheCreationTokenCount?: number;
+    cacheReadTokenCount?: number;
+    requestTime?: string;
+    responseTime?: string;
+    completionStartTime?: string;
+    timeToFirstToken?: number;
+    operationType?: string;
+    taskType?: string;
+    agent?: string;
+    stopReason?: string;
+    costType?: string;
+    subscriberId?: string;
+    subscriberEmail?: string;
+    subscriberCredentialName?: string;
+    subscriberCredential?: string;
+}
+/**
+ * Metering data sent to Revenium
+ * Field order matches the expected API format
+ */
+export interface MeteringData {
+    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: {
+        id: string;
+        email: string;
+        credential: {
+            name: string;
+            value: string;
+        };
+    };
+    model: string;
+    transactionId: string;
+    responseTime: string;
+    requestDuration: number;
+    provider: string;
+    requestTime: string;
+    completionStartTime: string;
+    timeToFirstToken: number;
+    middleware_source: string;
+    traceId?: string;
+}
+export interface ChatCompletionResult {
+    content: string;
+    role: string;
+    finishReason: string | null;
+    usage: {
+        promptTokens: number;
+        completionTokens: number;
+        totalTokens: number;
+    };
+    model: string;
+    transactionId: string;
+    rawResponse: PerplexityResponse;
+}
+export interface StreamingResult {
+    stream: AsyncGenerator<PerplexityStreamChunk>;
+    transactionId: string;
+    model: string;
+}
+export interface ReveniumConfig {
+    meteringApiKey: string;
+    meteringBaseUrl: string;
+    teamId?: string;
+    enabled: boolean;
+}
+export interface PerplexityConfig {
+    apiKey: string;
+    baseUrl: string;
+}
+export declare const PERPLEXITY_MODELS: {
+    readonly SONAR: "sonar";
+    readonly SONAR_PRO: "sonar-pro";
+    readonly SONAR_REASONING: "sonar-reasoning";
+    readonly LLAMA_3_1_SONAR_SMALL_128K_CHAT: "llama-3.1-sonar-small-128k-chat";
+    readonly LLAMA_3_1_SONAR_LARGE_128K_CHAT: "llama-3.1-sonar-large-128k-chat";
+    readonly LLAMA_3_1_SONAR_HUGE_128K_CHAT: "llama-3.1-sonar-huge-128k-chat";
+};
+export type PerplexityModel = (typeof PERPLEXITY_MODELS)[keyof typeof PERPLEXITY_MODELS];
+export declare const DEFAULT_MODEL: PerplexityModel;
+//# sourceMappingURL=index.d.ts.map

package/dist/types/types/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/types/index.ts"],"names":[],"mappings":"AAAA;;GAEG;AAMH,MAAM,WAAW,iBAAiB;IAChC,IAAI,EAAE,QAAQ,GAAG,MAAM,GAAG,WAAW,CAAC;IACtC,OAAO,EAAE,MAAM,CAAC;CACjB;AAED,MAAM,WAAW,qBAAqB;IACpC,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,EAAE,iBAAiB,EAAE,CAAC;IAC9B,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,MAAM,CAAC,EAAE,OAAO,CAAC;IACjB,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,iBAAiB,CAAC,EAAE,MAAM,CAAC;CAC5B;AAED,MAAM,WAAW,eAAe;IAC9B,aAAa,EAAE,MAAM,CAAC;IACtB,iBAAiB,EAAE,MAAM,CAAC;IAC1B,YAAY,EAAE,MAAM,CAAC;CACtB;AAED,MAAM,WAAW,gBAAgB;IAC/B,KAAK,EAAE,MAAM,CAAC;IACd,OAAO,CAAC,EAAE;QACR,IAAI,EAAE,MAAM,CAAC;QACb,OAAO,EAAE,MAAM,CAAC;KACjB,CAAC;IACF,KAAK,CAAC,EAAE;QACN,IAAI,CAAC,EAAE,MAAM,CAAC;QACd,OAAO,CAAC,EAAE,MAAM,CAAC;KAClB,CAAC;IACF,aAAa,EAAE,MAAM,GAAG,IAAI,CAAC;CAC9B;AAED,MAAM,WAAW,kBAAkB;IACjC,EAAE,EAAE,MAAM,CAAC;IACX,MAAM,EAAE,MAAM,CAAC;IACf,OAAO,EAAE,MAAM,CAAC;IAChB,KAAK,EAAE,MAAM,CAAC;IACd,OAAO,EAAE,gBAAgB,EAAE,CAAC;IAC5B,KAAK,CAAC,EAAE,eAAe,CAAC;CACzB;AAED,MAAM,WAAW,qBAAqB;IACpC,EAAE,EAAE,MAAM,CAAC;IACX,MAAM,EAAE,MAAM,CAAC;IACf,OAAO,EAAE,MAAM,CAAC;IAChB,KAAK,EAAE,MAAM,CAAC;IACd,OAAO,EAAE,gBAAgB,EAAE,CAAC;CAC7B;AAMD,MAAM,WAAW,aAAa;IAE5B,UAAU,CAAC,EAAE;QACX,EAAE,EAAE,MAAM,CAAC;QACX,KAAK,CAAC,EAAE,MAAM,CAAC;QACf,UAAU,CAAC,EAAE;YACX,IAAI,EAAE,MAAM,CAAC;YACb,KAAK,EAAE,MAAM,CAAC;SACf,CAAC;KACH,CAAC;IAGF,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,SAAS,CAAC,EAAE,MAAM,CAAC;IAGnB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,OAAO,CAAC,EAAE,MAAM,CAAC;IAGjB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,mBAAmB,CAAC,EAAE,MAAM,CAAC;IAC7B,uBAAuB,CAAC,EAAE,MAAM,CAAC;IACjC,mBAAmB,CAAC,EAAE,MAAM,CAAC;IAG7B,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,mBAAmB,CAAC,EAAE,MAAM,CAAC;IAC7B,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAG1B,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAGlB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,wBAAwB,CAAC,EAAE,MAAM,CAAC;IAClC,oBAAoB,CAAC,EAAE,MAAM,CAAC;CAC/B;AAED;;;GAGG;AACH,MAAM,WAAW,YAAY;IAC3B,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,EAAE,MAAM,CAAC;IACjB,UAAU,EAAE,OAAO,CAAC;IACpB,QAAQ,EAAE,MAAM,CAAC;IACjB,KAAK,EAAE,MAAM,CAAC;IACd,aAAa,EAAE,MAAM,CAAC;IACtB,eAAe,EAAE,MAAM,CAAC;IACxB,gBAAgB,EAAE,MAAM,CAAC;IACzB,mBAAmB,EAAE,MAAM,CAAC;IAC5B,uBAAuB,EAAE,MAAM,CAAC;IAChC,mBAAmB,EAAE,MAAM,CAAC;IAC5B,eAAe,EAAE,MAAM,CAAC;IACxB,cAAc,EAAE,MAAM,CAAC;IACvB,SAAS,EAAE,MAAM,CAAC;IAClB,UAAU,EAAE;QACV,EAAE,EAAE,MAAM,CAAC;QACX,KAAK,EAAE,MAAM,CAAC;QACd,UAAU,EAAE;YACV,IAAI,EAAE,MAAM,CAAC;YACb,KAAK,EAAE,MAAM,CAAC;SACf,CAAC;KACH,CAAC;IACF,KAAK,EAAE,MAAM,CAAC;IACd,aAAa,EAAE,MAAM,CAAC;IACtB,YAAY,EAAE,MAAM,CAAC;IACrB,eAAe,EAAE,MAAM,CAAC;IACxB,QAAQ,EAAE,MAAM,CAAC;IACjB,WAAW,EAAE,MAAM,CAAC;IACpB,mBAAmB,EAAE,MAAM,CAAC;IAC5B,gBAAgB,EAAE,MAAM,CAAC;IACzB,iBAAiB,EAAE,MAAM,CAAC;IAC1B,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB;AAMD,MAAM,WAAW,oBAAoB;IACnC,OAAO,EAAE,MAAM,CAAC;IAChB,IAAI,EAAE,MAAM,CAAC;IACb,YAAY,EAAE,MAAM,GAAG,IAAI,CAAC;IAC5B,KAAK,EAAE;QACL,YAAY,EAAE,MAAM,CAAC;QACrB,gBAAgB,EAAE,MAAM,CAAC;QACzB,WAAW,EAAE,MAAM,CAAC;KACrB,CAAC;IACF,KAAK,EAAE,MAAM,CAAC;IACd,aAAa,EAAE,MAAM,CAAC;IACtB,WAAW,EAAE,kBAAkB,CAAC;CACjC;AAED,MAAM,WAAW,eAAe;IAC9B,MAAM,EAAE,cAAc,CAAC,qBAAqB,CAAC,CAAC;IAC9C,aAAa,EAAE,MAAM,CAAC;IACtB,KAAK,EAAE,MAAM,CAAC;CACf;AAMD,MAAM,WAAW,cAAc;IAC7B,cAAc,EAAE,MAAM,CAAC;IACvB,eAAe,EAAE,MAAM,CAAC;IACxB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,OAAO,EAAE,OAAO,CAAC;CAClB;AAED,MAAM,WAAW,gBAAgB;IAC/B,MAAM,EAAE,MAAM,CAAC;IACf,OAAO,EAAE,MAAM,CAAC;CACjB;AAMD,eAAO,MAAM,iBAAiB;;;;;;;CAUpB,CAAC;AAEX,MAAM,MAAM,eAAe,GACzB,CAAC,OAAO,iBAAiB,CAAC,CAAC,MAAM,OAAO,iBAAiB,CAAC,CAAC;AAE7D,eAAO,MAAM,aAAa,EAAE,eAA6C,CAAC"}

package/dist/types/utils/logger.d.ts

@@ -0,0 +1,10 @@
+/**
+ * Simple logger utility
+ */
+export declare const logger: {
+    info: (...args: any[]) => void;
+    warn: (...args: any[]) => void;
+    error: (...args: any[]) => void;
+    debug: (...args: any[]) => void;
+};
+//# sourceMappingURL=logger.d.ts.map

package/dist/types/utils/logger.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"logger.d.ts","sourceRoot":"","sources":["../../../src/utils/logger.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,eAAO,MAAM,MAAM;oBACD,GAAG,EAAE;oBAGL,GAAG,EAAE;qBAGJ,GAAG,EAAE;qBAGL,GAAG,EAAE;CAKvB,CAAC"}

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@revenium/perplexity",
-  "version": "1.0.23",
-  "description": "NodeJS middleware for perplexity's AI API",
+  "version": "2.0.1",
+  "description": "NodeJS middleware for Perplexity AI API with Revenium metering",
   "homepage": "https://github.com/revenium/revenium-middleware-perplexity-node#readme",
   "bugs": {
     "url": "https://github.com/revenium/revenium-middleware-perplexity-node/issues"
@@ -10,20 +10,37 @@
     "type": "git",
     "url": "git+https://github.com/revenium/revenium-middleware-perplexity-node.git"
   },
-  "main": "dist/index.js",
-  "types": "dist/index.d.ts",
+  "main": "dist/cjs/index.js",
+  "module": "dist/esm/index.js",
+  "types": "dist/types/index.d.ts",
+  "exports": {
+    ".": {
+      "import": "./dist/esm/index.js",
+      "require": "./dist/cjs/index.js",
+      "types": "./dist/types/index.d.ts"
+    }
+  },
+  "files": [
+    "dist/**/*",
+    "README.md",
+    "LICENSE"
+  ],
   "scripts": {
-    "build": "tsc -b",
-    "p-basic": "npx ts-node playground/basic.js",
-    "p-streaming": "npx ts-node playground/streaming.js",
-    "p-enhanced": "npx ts-node playground/enhanced.js",
-    "p-chat-completions": "npx ts-node playground/chat-completions.js",
-    "p-metadata": "npx ts-node playground/metadata.js",
-    "e-basic": "npx ts-node examples/basic.ts",
-    "e-streaming": "npx ts-node examples/streaming.ts",
-    "e-chat-completions": "npx ts-node examples/chat-completions.ts",
-    "e-enhanced": "npx ts-node examples/enhanced.ts",
-    "e-metadata": "npx ts-node examples/metadata.ts"
+    "clean": "if exist dist rmdir /s /q dist",
+    "build:esm": "tsc -p tsconfig.esm.json",
+    "build:cjs": "tsc -p tsconfig.cjs.json",
+    "build:types": "tsc -p tsconfig.types.json",
+    "build": "npm run clean && npm run build:esm && npm run build:cjs && npm run build:types",
+    "prepublishOnly": "npm run build",
+    "example:basic": "tsx examples/basic.ts",
+    "example:streaming": "tsx examples/streaming.ts",
+    "example:chat": "tsx examples/chat.ts",
+    "example:metadata": "tsx examples/metadata.ts",
+    "playground:basic": "node playground/basic.js",
+    "playground:streaming": "node playground/streaming.js",
+    "playground:chat": "node playground/chat.js",
+    "playground:metadata": "node playground/metadata.js",
+    "test": "npm run example:basic && npm run example:streaming && npm run example:chat && npm run example:metadata"
   },
   "keywords": [
     "revenium",
@@ -42,8 +59,11 @@
   "dependencies": {
     "dotenv": "^17.2.2",
     "openai": "^5.23.1",
-    "ts-node": "^10.9.2",
+    "tsx": "^4.19.2",
     "typescript": "^5.9.2",
     "uuid": "^13.0.0"
+  },
+  "devDependencies": {
+    "@types/node": "^24.7.1"
   }
 }

package/.env.example

@@ -1,3 +0,0 @@
-PERPLEXITY_API_KEY="your_perplexity_api_key_here"
-REVENIUM_METERING_API_KEY="your_revenium_api_key_here"
-REVENIUM_METERING_BASE_URL="https://api.revenium.io/meter"

package/dist/index.js

@@ -1,19 +0,0 @@
-"use strict";
-var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    var desc = Object.getOwnPropertyDescriptor(m, k);
-    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
-      desc = { enumerable: true, get: function() { return m[k]; } };
-    }
-    Object.defineProperty(o, k2, desc);
-}) : (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    o[k2] = m[k];
-}));
-var __exportStar = (this && this.__exportStar) || function(m, exports) {
-    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-const dotenv_1 = require("dotenv");
-(0, dotenv_1.config)();
-__exportStar(require("./middleware"), exports);

package/dist/interfaces/chatCompletionRequest.d.ts

@@ -1,9 +0,0 @@
-import { ChatCompletionMessageParam } from "openai/resources/index";
-import { IUsageMetadata } from "./usageMetadata";
-export interface IChatCompletionRequest {
-    model?: string;
-    messages: ChatCompletionMessageParam[];
-    stream?: boolean;
-    usageMetadata?: IUsageMetadata;
-    [key: string]: any;
-}

package/dist/interfaces/chatCompletionRequest.js

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

package/dist/interfaces/credential.d.ts

@@ -1,4 +0,0 @@
-export interface ICredential {
-    name: string;
-    value: string;
-}

package/dist/interfaces/credential.js

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

package/dist/interfaces/meteringRequest.d.ts

@@ -1,13 +0,0 @@
-import { IOperationType } from "./operation";
-import { ITokenCounts } from "./tokenCounts";
-import { IUsageMetadata } from "./usageMetadata";
-export interface IMeteringRequest {
-    transactionId?: string;
-    startTime: Date;
-    endTime: Date;
-    modelName: string;
-    tokenCounts: ITokenCounts;
-    stopReason: string;
-    operationType: IOperationType;
-    usageMetadata?: IUsageMetadata;
-}

package/dist/interfaces/meteringRequest.js

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

package/dist/interfaces/meteringResponse.d.ts

@@ -1,27 +0,0 @@
-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;
-}

package/dist/interfaces/meteringResponse.js

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

package/dist/interfaces/operation.d.ts

@@ -1,4 +0,0 @@
-export declare enum IOperationType {
-    CHAT = "CHAT",
-    EMBED = "EMBED"
-}

package/dist/interfaces/operation.js

@@ -1,8 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.IOperationType = void 0;
-var IOperationType;
-(function (IOperationType) {
-    IOperationType["CHAT"] = "CHAT";
-    IOperationType["EMBED"] = "EMBED";
-})(IOperationType || (exports.IOperationType = IOperationType = {}));

package/dist/interfaces/subscriber.d.ts

@@ -1,8 +0,0 @@
-export interface ISubscriber {
-    id: string;
-    email: string;
-    credential: {
-        name: string;
-        value: string;
-    };
-}

package/dist/interfaces/subscriber.js

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

package/dist/interfaces/tokenCounts.d.ts

@@ -1,7 +0,0 @@
-export interface ITokenCounts {
-    inputTokens: number;
-    outputTokens: number;
-    totalTokens: number;
-    cachedTokens?: number;
-    reasoningTokens?: number;
-}

package/dist/interfaces/tokenCounts.js

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

package/dist/interfaces/usageMetadata.d.ts

@@ -1,27 +0,0 @@
-export interface IUsageMetadata {
-    traceId?: string;
-    taskType?: string;
-    subscriberEmail?: string;
-    subscriberId?: string;
-    subscriberCredentialName?: string;
-    subscriberCredential?: string;
-    organizationId?: string;
-    subscriptionId?: string;
-    productId?: string;
-    agent?: string;
-    responseQualityScore?: number;
-    transactionId?: string;
-    timeToFirstToken?: number;
-    requestTime?: Date;
-    completionStartTime?: Date;
-    operationType?: string;
-    inputTokenCount?: number;
-    outputTokenCount?: number;
-    reasoningTokenCount?: number;
-    cacheCreationTokenCount?: number;
-    cacheReadTokenCount?: number;
-    totalTokenCount?: number;
-    responseTime?: Date;
-    requestDuration?: number;
-    stopReason?: string;
-}

package/dist/interfaces/usageMetadata.js

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

package/dist/middleware.d.ts

@@ -1,22 +0,0 @@
-import OpenAI from "openai";
-import { IChatCompletionRequest } from "./interfaces/chatCompletionRequest";
-import { RequestOptions } from "openai/internal/request-options";
-export declare class PerplexityReveniumMiddleware {
-    private client;
-    private working;
-    private modelName;
-    constructor();
-    getGenerativeModel(model: string): {
-        createChatCompletion: (params: IChatCompletionRequest) => Promise<OpenAI.Chat.Completions.ChatCompletion & {
-            _request_id?: string | null | undefined;
-        }>;
-        createChatCompletionStream: (params: IChatCompletionRequest) => Promise<import("openai/core/streaming").Stream<OpenAI.Chat.Completions.ChatCompletionChunk> & {
-            _request_id?: string | null;
-        }>;
-        createEmbeddings: (_body: Omit<OpenAI.Embeddings.EmbeddingCreateParams, "model">, _options?: RequestOptions) => Promise<void>;
-    };
-    private embeddings;
-    private streamingChatCompletion;
-    private chatCompletion;
-    private verifyEnv;
-}

package/dist/middleware.js

@@ -1,129 +0,0 @@
-"use strict";
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.PerplexityReveniumMiddleware = void 0;
-const openai_1 = __importDefault(require("openai"));
-const utils_1 = require("./utils");
-const Logger_1 = require("./models/Logger");
-const Metering_1 = require("./models/Metering");
-const operation_1 = require("./interfaces/operation");
-const extractTokenCount_1 = require("./utils/extractTokenCount");
-class PerplexityReveniumMiddleware {
-    constructor() {
-        this.working = true;
-        this.modelName = "";
-        this.embeddings = async (_body, _options) => { };
-        this.streamingChatCompletion = async (params) => {
-            var _a;
-            const startTime = new Date();
-            try {
-                const { usageMetadata, ...openaiParams } = params;
-                const requestParams = {
-                    ...openaiParams,
-                    model: this.modelName,
-                };
-                const result = await ((_a = this.client) === null || _a === void 0 ? void 0 : _a.chat.completions.create({
-                    ...requestParams,
-                    stream: true,
-                }));
-                if (!this.working) {
-                    Logger_1.logger.warning("Metering is not working. Check your configuration.");
-                    return result;
-                }
-                const tokenCounts = (0, extractTokenCount_1.extractGoogleAITokenCounts)(result);
-                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 : "");
-                const getMetering = metering.createMetering({
-                    modelName: this.modelName,
-                    endTime,
-                    startTime,
-                    operationType: operation_1.IOperationType.CHAT,
-                    stopReason: "END",
-                    tokenCounts,
-                    usageMetadata,
-                }, true);
-                await metering.sendMeteringData(getMetering);
-                return result;
-            }
-            catch (error) {
-                throw error;
-            }
-        };
-        this.chatCompletion = async (params) => {
-            var _a;
-            const startTime = new Date();
-            try {
-                const { usageMetadata, ...openaiParams } = params;
-                const requestParams = {
-                    ...openaiParams,
-                    model: this.modelName,
-                };
-                const result = await ((_a = this.client) === null || _a === void 0 ? void 0 : _a.chat.completions.create({
-                    ...requestParams,
-                    stream: false,
-                }));
-                if (!this.working) {
-                    Logger_1.logger.warning("Metering is not working. Check your configuration.");
-                    return result;
-                }
-                const tokenCounts = (0, extractTokenCount_1.extractGoogleAITokenCounts)(result);
-                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 : "");
-                const getMetering = metering.createMetering({
-                    modelName: this.modelName,
-                    endTime,
-                    startTime,
-                    operationType: operation_1.IOperationType.CHAT,
-                    stopReason: "END",
-                    tokenCounts,
-                    usageMetadata,
-                }, false);
-                await metering.sendMeteringData(getMetering);
-                return result;
-            }
-            catch (error) {
-                throw error;
-            }
-        };
-        this.verifyEnv();
-        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);
-    }
-    getGenerativeModel(model) {
-        this.modelName = model;
-        if (!utils_1.models.includes(model)) {
-            throw new Error(`Model ${model} is not supported`);
-        }
-        return {
-            createChatCompletion: this.chatCompletion,
-            createChatCompletionStream: this.streamingChatCompletion,
-            createEmbeddings: this.embeddings,
-        };
-    }
-    verifyEnv() {
-        if (!utils_1.PERPLEXITY_API_KEY) {
-            Logger_1.logger.error(utils_1.PERPLEXITY_REQUIRED_API_KEY_MESSAGE);
-            throw new Error(utils_1.PERPLEXITY_REQUIRED_API_KEY_MESSAGE);
-        }
-        if (!utils_1.PERPLEXITY_API_KEY.includes("pplx-")) {
-            Logger_1.logger.error(utils_1.PERPLEXITY_API_KEY_INVALID_MESSAGE);
-            throw new Error(utils_1.PERPLEXITY_API_KEY_INVALID_MESSAGE);
-        }
-        if (!utils_1.REVENIUM_METERING_API_KEY) {
-            Logger_1.logger.warning(utils_1.PERPLEXITY_METERING_API_KEY_IS_NOT_SET_MESSAGE);
-            this.working = false;
-        }
-        if (!utils_1.REVENIUM_METERING_BASE_URL) {
-            Logger_1.logger.warning(utils_1.PERPLEXITY_METERING_BASE_URL_IS_NOT_SET_MESSAGE);
-            this.working = false;
-        }
-    }
-}
-exports.PerplexityReveniumMiddleware = PerplexityReveniumMiddleware;

package/dist/models/Logger.js

@@ -1,35 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.logger = exports.Logger = void 0;
-const utils_1 = require("../utils");
-class Logger {
-    static debug(message, data) {
-        if (this.shouldLog(utils_1.LOG_LEVELS[0])) {
-            console.log(`[${utils_1.LOG_LEVELS[0]}] ${message}`, data || "");
-        }
-    }
-    static info(message, data) {
-        if (this.shouldLog(utils_1.LOG_LEVELS[1])) {
-            console.log(`[${utils_1.LOG_LEVELS[1]}] ${message}`, data || "");
-        }
-    }
-    static warning(message, data) {
-        if (this.shouldLog(utils_1.LOG_LEVELS[2])) {
-            console.warn(`[${utils_1.LOG_LEVELS[2]}] ${message}`, data || "");
-        }
-    }
-    static error(message, data) {
-        if (this.shouldLog(utils_1.LOG_LEVELS[3])) {
-            console.error(`[${utils_1.LOG_LEVELS[3]}] ${message}`, data || "");
-        }
-    }
-    static shouldLog(level) {
-        const levels = utils_1.LOG_LEVELS;
-        const currentLevel = levels.indexOf(this.logLevel.toUpperCase());
-        const messageLevel = levels.indexOf(level);
-        return messageLevel >= currentLevel;
-    }
-}
-exports.Logger = Logger;
-Logger.logLevel = process.env.REVENIUM_LOG_LEVEL || "INFO";
-exports.logger = Logger;