@revenium/perplexity 1.0.23 → 1.0.25

package/examples/chat-completions.ts

@@ -1,22 +0,0 @@
-import { PerplexityReveniumMiddleware } from "../src";
-
-const chatCompletionsExample = async () => {
-  console.log("=".repeat(50));
-  try {
-    const middleware = new PerplexityReveniumMiddleware();
-    const model = middleware.getGenerativeModel("sonar-pro");
-    const result = await model.createChatCompletion({
-      messages: [
-        {
-          role: "user",
-          content: "Hello world",
-        },
-      ],
-    });
-    console.log(result.choices[0].message.content);
-  } catch (error) {
-    console.log(error);
-  }
-};
-
-chatCompletionsExample();

package/examples/enhanced.ts

@@ -1,20 +0,0 @@
-import { PerplexityReveniumMiddleware } from "../src";
-
-async function enhancedRequest() {
-  console.log("=".repeat(50));
-  try {
-    const middleware = new PerplexityReveniumMiddleware();
-    const model = middleware.getGenerativeModel("sonar-pro");
-    const result = await model.createChatCompletion({
-      messages: [
-        { role: "user", content: "Explain quantum computing in simple terms" },
-      ],
-      max_tokens: 100,
-      temperature: 0.7,
-    });
-    console.log("[ENHANCED REQUEST]", result.choices[0].message.content);
-  } catch (error) {
-    console.log(error);
-  }
-}
-enhancedRequest();

package/examples/metadata.ts

@@ -1,43 +0,0 @@
-import { PerplexityReveniumMiddleware } from "../src";
-
-async function metadataRequest() {
-  console.log("=".repeat(50));
-  try {
-    const middleware = new PerplexityReveniumMiddleware();
-    const model = middleware.getGenerativeModel("sonar-pro");
-    const result = await model.createChatCompletion({
-      model: "sonar-pro",
-      messages: [{ role: "user", content: "What is the capital of France?" }],
-      usageMetadata: {
-        taskType: "test",
-        subscriberEmail: "test@revenium.ai",
-        subscriberId: "123456",
-        subscriberCredentialName: "apiKey",
-        subscriberCredential: "keyValue",
-        organizationId: "123456",
-        subscriptionId: "123456",
-        productId: "free-trial",
-        agent: "perplexity",
-        responseQualityScore: 100,
-        transactionId: "123456",
-        timeToFirstToken: 1000,
-        requestTime: new Date(),
-        completionStartTime: new Date(),
-        operationType: "CHAT",
-        inputTokenCount: 10,
-        outputTokenCount: 10,
-        reasoningTokenCount: 20,
-        cacheCreationTokenCount: 0,
-        cacheReadTokenCount: 0,
-        totalTokenCount: 40,
-        responseTime: new Date(),
-        requestDuration: 1000,
-        stopReason: "END",
-      },
-    });
-    console.log("[METADATA REQUEST]", result.choices[0].message);
-  } catch (error) {
-    console.log(error);
-  }
-}
-metadataRequest();

package/examples/streaming.ts

@@ -1,24 +0,0 @@
-import { PerplexityReveniumMiddleware } from "../src";
-
-const chatCompletionStreamingExample = async () => {
-  console.log("=".repeat(50));
-  try {
-    const middleware = new PerplexityReveniumMiddleware();
-    const model = middleware.getGenerativeModel("sonar-pro");
-    const result = await model.createChatCompletionStream({
-      messages: [
-        {
-          role: "user",
-          content: "Hello world",
-        },
-      ],
-    });
-    for await (const chunk of result) {
-      process.stdout.write(chunk.choices[0]?.delta?.content || "");
-    }
-  } catch (error) {
-    console.log(error);
-  }
-};
-
-chatCompletionStreamingExample();

package/playground/basic.js

@@ -1,17 +0,0 @@
-import { PerplexityReveniumMiddleware } from "@revenium/perplexity";
-
-async function basicRequest() {
-  console.log("=".repeat(50));
-  try {
-    const middleware = new PerplexityReveniumMiddleware();
-    const model = middleware.getGenerativeModel("sonar-pro");
-    const result = await model.createChatCompletion({
-      model: "sonar-pro",
-      messages: [{ role: "user", content: "What is the capital of France?" }],
-    });
-    console.log("[BASIC REQUEST]", result.choices[0].message);
-  } catch (error) {
-    console.log(error);
-  }
-}
-basicRequest();

package/playground/chat-completions.js

@@ -1,22 +0,0 @@
-import { PerplexityReveniumMiddleware } from "@revenium/perplexity";
-
-const chatCompletionsExample = async () => {
-  console.log("=".repeat(50));
-  try {
-    const middleware = new PerplexityReveniumMiddleware();
-    const model = middleware.getGenerativeModel("sonar-pro");
-    const result = await model.createChatCompletion({
-      messages: [
-        {
-          role: "user",
-          content: "Hello world",
-        },
-      ],
-    });
-    console.log(result.choices[0].message.content);
-  } catch (error) {
-    console.log(error);
-  }
-};
-
-chatCompletionsExample();

package/playground/enhanced.js

@@ -1,20 +0,0 @@
-import { PerplexityReveniumMiddleware } from "@revenium/perplexity";
-
-async function enhancedRequest() {
-  console.log("=".repeat(50));
-  try {
-    const middleware = new PerplexityReveniumMiddleware();
-    const model = middleware.getGenerativeModel("sonar-pro");
-    const result = await model.createChatCompletion({
-      messages: [
-        { role: "user", content: "Explain quantum computing in simple terms" },
-      ],
-      max_tokens: 100,
-      temperature: 0.7,
-    });
-    console.log("[ENHANCED REQUEST]", result.choices[0].message.content);
-  } catch (error) {
-    console.log(error);
-  }
-}
-enhancedRequest();

package/playground/metadata.js

@@ -1,43 +0,0 @@
-import { PerplexityReveniumMiddleware } from "@revenium/perplexity";
-
-async function metadataRequest() {
-  console.log("=".repeat(50));
-  try {
-    const middleware = new PerplexityReveniumMiddleware();
-    const model = middleware.getGenerativeModel("sonar-pro");
-    const result = await model.createChatCompletion({
-      model: "sonar-pro",
-      messages: [{ role: "user", content: "What is the capital of France?" }],
-      usageMetadata: {
-        taskType: "test",
-        subscriberEmail: "test@revenium.ai",
-        subscriberId: "123456",
-        subscriberCredentialName: "apiKey",
-        subscriberCredential: "keyValue",
-        organizationId: "123456",
-        subscriptionId: "123456",
-        productId: "free-trial",
-        agent: "perplexity",
-        responseQualityScore: 100,
-        transactionId: "123456",
-        timeToFirstToken: 1000,
-        requestTime: new Date(),
-        completionStartTime: new Date(),
-        operationType: "CHAT",
-        inputTokenCount: 10,
-        outputTokenCount: 10,
-        reasoningTokenCount: 20,
-        cacheCreationTokenCount: 0,
-        cacheReadTokenCount: 0,
-        totalTokenCount: 40,
-        responseTime: new Date(),
-        requestDuration: 1000,
-        stopReason: "END",
-      },
-    });
-    console.log("[METADATA REQUEST]", result.choices[0].message);
-  } catch (error) {
-    console.log(error);
-  }
-}
-metadataRequest();

package/playground/streaming.js

@@ -1,24 +0,0 @@
-import { PerplexityReveniumMiddleware } from "@revenium/perplexity";
-
-const chatCompletionStreamingExample = async () => {
-  console.log("=".repeat(50));
-  try {
-    const middleware = new PerplexityReveniumMiddleware();
-    const model = middleware.getGenerativeModel("sonar-pro");
-    const result = await model.createChatCompletionStream({
-      messages: [
-        {
-          role: "user",
-          content: "Hello world",
-        },
-      ],
-    });
-    for await (const chunk of result) {
-      process.stdout.write(chunk.choices[0]?.delta?.content || "");
-    }
-  } catch (error) {
-    console.log(error);
-  }
-};
-
-chatCompletionStreamingExample();

package/src/middleware.ts

@@ -1,157 +0,0 @@
-import OpenAI from "openai";
-import {
-  models,
-  PERPLEXITY_API_BASE_URL,
-  PERPLEXITY_API_KEY,
-  PERPLEXITY_API_KEY_INVALID_MESSAGE,
-  PERPLEXITY_CLIENT_INITIALIZED_MESSAGE,
-  PERPLEXITY_METERING_API_KEY_IS_NOT_SET_MESSAGE,
-  PERPLEXITY_METERING_BASE_URL_IS_NOT_SET_MESSAGE,
-  PERPLEXITY_REQUIRED_API_KEY_MESSAGE,
-  REVENIUM_METERING_API_KEY,
-  REVENIUM_METERING_BASE_URL,
-} from "./utils";
-import { logger } from "./models/Logger";
-import { IChatCompletionRequest } from "./interfaces/chatCompletionRequest";
-import { Metering } from "./models/Metering";
-import { IOperationType } from "./interfaces/operation";
-import { extractGoogleAITokenCounts } from "./utils/extractTokenCount";
-import { RequestOptions } from "openai/internal/request-options";
-
-export class PerplexityReveniumMiddleware {
-  private client: OpenAI;
-  private working: boolean = true;
-  private modelName: string = "";
-  constructor() {
-    this.verifyEnv();
-    this.client = new OpenAI({
-      apiKey: PERPLEXITY_API_KEY,
-      baseURL: PERPLEXITY_API_BASE_URL,
-    });
-    logger.info(PERPLEXITY_CLIENT_INITIALIZED_MESSAGE);
-  }
-  public getGenerativeModel(model: string) {
-    this.modelName = model;
-    if (!models.includes(model)) {
-      throw new Error(`Model ${model} is not supported`);
-    }
-    return {
-      createChatCompletion: this.chatCompletion,
-      createChatCompletionStream: this.streamingChatCompletion,
-      createEmbeddings: this.embeddings,
-    };
-  }
-
-  private embeddings = async (
-    _body: Omit<OpenAI.Embeddings.EmbeddingCreateParams, "model">,
-    _options?: RequestOptions
-  ) => {};
-
-  private streamingChatCompletion = async (params: IChatCompletionRequest) => {
-    const startTime: Date = new Date();
-    try {
-      const { usageMetadata, ...openaiParams } = params;
-      const requestParams = {
-        ...openaiParams,
-        model: this.modelName,
-      };
-      const result = await this.client?.chat.completions.create({
-        ...requestParams,
-        stream: true,
-      });
-      if (!this.working) {
-        logger.warning("Metering is not working. Check your configuration.");
-        return result;
-      }
-      const tokenCounts = extractGoogleAITokenCounts(result);
-      const endTime: Date = new Date();
-      logger.info("Metering is working.");
-      const metering = new Metering(
-        REVENIUM_METERING_API_KEY ?? "",
-        REVENIUM_METERING_BASE_URL ?? ""
-      );
-      const getMetering = metering.createMetering(
-        {
-          modelName: this.modelName,
-          endTime,
-          startTime,
-          operationType: IOperationType.CHAT,
-          stopReason: "END",
-          tokenCounts,
-          usageMetadata,
-        },
-        true
-      );
-      await metering.sendMeteringData(getMetering);
-      return result;
-    } catch (error) {
-      throw error;
-    }
-  };
-
-  private chatCompletion = async (params: IChatCompletionRequest) => {
-    const startTime: Date = new Date();
-    try {
-      const { usageMetadata, ...openaiParams } = params;
-      const requestParams = {
-        ...openaiParams,
-        model: this.modelName,
-      };
-      const result: OpenAI.Chat.Completions.ChatCompletion & {
-        _request_id?: string | null | undefined;
-      } = await this.client?.chat.completions.create({
-        ...requestParams,
-        stream: false,
-      });
-      if (!this.working) {
-        logger.warning("Metering is not working. Check your configuration.");
-        return result;
-      }
-      const tokenCounts = extractGoogleAITokenCounts(result);
-      const endTime: Date = new Date();
-      logger.info(" Metering is working.");
-      const metering = new Metering(
-        REVENIUM_METERING_API_KEY ?? "",
-        REVENIUM_METERING_BASE_URL ?? ""
-      );
-      const getMetering = metering.createMetering(
-        {
-          modelName: this.modelName,
-          endTime,
-          startTime,
-          operationType: IOperationType.CHAT,
-          stopReason: "END",
-          tokenCounts,
-          usageMetadata,
-        },
-        false
-      );
-      await metering.sendMeteringData(getMetering);
-      return result;
-    } catch (error) {
-      throw error;
-    }
-  };
-
-  private verifyEnv(): void {
-    if (!PERPLEXITY_API_KEY) {
-      logger.error(PERPLEXITY_REQUIRED_API_KEY_MESSAGE);
-      throw new Error(PERPLEXITY_REQUIRED_API_KEY_MESSAGE);
-    }
-
-    if (!PERPLEXITY_API_KEY.includes("pplx-")) {
-      logger.error(PERPLEXITY_API_KEY_INVALID_MESSAGE);
-      throw new Error(PERPLEXITY_API_KEY_INVALID_MESSAGE);
-    }
-
-    if (!REVENIUM_METERING_API_KEY) {
-      logger.warning(PERPLEXITY_METERING_API_KEY_IS_NOT_SET_MESSAGE);
-      this.working = false;
-    }
-
-    if (!REVENIUM_METERING_BASE_URL) {
-      logger.warning(PERPLEXITY_METERING_BASE_URL_IS_NOT_SET_MESSAGE);
-      this.working = false;
-    }
-  }
-}