6-track-requests-from-js-library

src/index.ts

@@ -1,26 +1 @@
-import { Configuration, OpenAIApi } from "openai";
-
-const configuration = new Configuration({
-  apiKey: process.env.OPENAI_API_KEY,
-});
-const target = new OpenAIApi(configuration);
-
-export const openai = new Proxy(target, {
-  get: (target, prop, receiver) => {
-    const value = target[prop as keyof OpenAIApi];
-    if (typeof value === "function") {
-      return (...args: any[]) => {
-        let return_pl_id = false;
-        const newArgs = args.map((arg) => {
-          if (arg["return_pl_id"] !== undefined) {
-            return_pl_id = arg["return_pl_id"];
-            delete arg["return_pl_id"];
-          }
-          return arg;
-        });
-        return (value as any).apply(target, args);
-      };
-    }
-    return Reflect.get(target, prop, receiver);
-  },
-});
+export * as utils from "@/utils";

src/types/index.ts

@@ -0,0 +1 @@
+export { TrackRequest } from "@/types/track-request";

src/types/track-request.ts

@@ -0,0 +1,40 @@
+export type ApiKey = string;
+export type ProviderType = "openai" | "anthropic" | "langchain";
+export type FunctionName = string;
+export type Args = unknown[];
+export type RequestEndTime = string;
+export type RequestStartTime = string;
+export type PromptId = number;
+export type PromptVersion = number;
+export type Tags = string[];
+export type PromptInputVariables =
+  | {
+      [k: string]: string;
+    }
+  | string[];
+
+export interface TrackRequest {
+  api_key: ApiKey;
+  provider_type?: ProviderType;
+  function_name: FunctionName;
+  args?: Args;
+  kwargs?: Kwargs;
+  request_end_time: RequestEndTime;
+  request_start_time: RequestStartTime;
+  prompt_id?: PromptId;
+  prompt_version?: PromptVersion;
+  metadata?: Metadata;
+  tags?: Tags;
+  request_response?: RequestResponse;
+  prompt_input_variables?: PromptInputVariables;
+  [k: string]: unknown;
+}
+export interface Kwargs {
+  [k: string]: unknown;
+}
+export interface Metadata {
+  [k: string]: string;
+}
+export interface RequestResponse {
+  [k: string]: unknown;
+}

src/utils/index.ts

@@ -0,0 +1,33 @@
+import { TrackRequest } from "@/types";
+
+const URL_API_PROMPTLAYER =
+  process.env.URL_API_PROMPTLAYER || "https://api.promptlayer.com";
+
+const promptlayerApiRequest = async (body: TrackRequest) => {
+  try {
+    const response = await fetch(`${URL_API_PROMPTLAYER}/track-request`, {
+      method: "POST",
+      headers: {
+        "Content-Type": "application/json",
+      },
+      body: JSON.stringify(body),
+    });
+    const data = await response.json();
+    if (response.status !== 200) {
+      console.warn(
+        `WARNING: While logging your request PromptLayer had the following error: ${JSON.stringify(
+          data
+        )}`
+      );
+    }
+    if (data && body.return_pl_id) {
+      return data.request_id;
+    }
+  } catch (e) {
+    console.warn(
+      `WARNING: While logging your request PromptLayer had the following error: ${e}`
+    );
+  }
+};
+
+export { promptlayerApiRequest };

tsconfig.json

@@ -105,6 +105,10 @@
     /* Completeness */
     // "skipDefaultLibCheck": true,                      /* Skip type checking .d.ts files that are included with TypeScript. */
     "skipLibCheck": true /* Skip type checking all .d.ts files. */,
-    "noEmit": true
+    "noEmit": true,
+    "baseUrl": "./src",
+    "paths": {
+      "@/*": ["*"]
+    }
   }
 }