openlayer 0.1.6 → 0.1.8

package/dist/index.d.ts

@@ -69,6 +69,8 @@ type OpenlayerClientConstructorProps = {
 };
 type OpenAIMonitorConstructorProps = OpenlayerClientConstructorProps & {
     openAiApiKey: string;
+    openlayerInferencePipelineName?: string;
+    openlayerProjectName: string;
 };
 type OpenlayerInferencePipeline = {
     dataVolumeGraphs?: OpenlayerSampleVolumeGraph;
@@ -127,11 +129,9 @@ type OpenlayerSampleVolumeGraph = {
 type OpenlayerTaskType = 'llm-base' | 'tabular-classification' | 'tabular-regression' | 'text-classification';
 declare class OpenlayerClient {
     private openlayerApiKey?;
-    private openlayerProjectName?;
-    private openlayerInferencePipelineName?;
     private openlayerServerUrl;
     private version;
-    constructor({ openlayerApiKey, openlayerInferencePipelineName, openlayerProjectName, openlayerServerUrl, }: OpenlayerClientConstructorProps);
+    constructor({ openlayerApiKey, openlayerServerUrl, }: OpenlayerClientConstructorProps);
     private resolvedQuery;
     private uploadToInferencePipeline;
     /**
@@ -142,7 +142,7 @@ declare class OpenlayerClient {
      * @throws Throws an error if Openlayer API key or project name are not set.
      * @returns A promise that resolves when the data has been successfully uploaded.
      */
-    uploadDataToOpenlayer: (data: ChatCompletionData, config: ChatCompletionConfig) => Promise<void>;
+    uploadDataToOpenlayer: (data: ChatCompletionData, config: ChatCompletionConfig, projectName: string, inferencePipelineName?: string) => Promise<void>;
     /**
      * Creates a new inference pipeline in Openlayer, or loads an existing one if it already exists.
      * @param name The name of the inference pipeline.
@@ -150,7 +150,7 @@ declare class OpenlayerClient {
      * @throws Throws an error if the inference pipeline cannot be created or found.
      * @returns A promise that resolves to an OpenlayerInferencePipeline object.
      */
-    createInferencePipeline: (name: string, projectId: string) => Promise<OpenlayerInferencePipeline>;
+    createInferencePipeline: (projectId: string, name?: string) => Promise<OpenlayerInferencePipeline>;
     /**
      * Creates a new project in Openlayer, or loads an existing one if it already exists.
      * @param name The name of the project.
@@ -167,15 +167,17 @@ declare class OpenlayerClient {
      * @throws Throws an error if the inference pipeline is not found.
      * @returns A promise that resolves to an OpenlayerInferencePipeline object.
      */
-    loadInferencePipeline: (name: string, projectId: string) => Promise<OpenlayerInferencePipeline>;
+    loadInferencePipeline: (projectId: string, name?: string) => Promise<OpenlayerInferencePipeline>;
     loadProject: (name: string) => Promise<OpenlayerProject>;
 }
 export declare class OpenAIMonitor {
     private openlayerClient;
     private openAIClient;
     private openlayerDefaultDataConfig;
+    private openlayerProjectName;
+    private openlayerInferencePipelineName;
     private monitoringOn;
-    constructor({ openAiApiKey, openlayerApiKey, openlayerInferencePipelineName, openlayerProjectName, openlayerServerUrl, }: OpenAIMonitorConstructorProps);
+    constructor({ openAiApiKey, openlayerApiKey, openlayerProjectName, openlayerInferencePipelineName, openlayerServerUrl, }: OpenAIMonitorConstructorProps);
     private formatChatCompletionInput;
     /**
      * Creates a new ChatCompletion instance. If monitoring is not active, an error is thrown.

package/dist/index.js

@@ -21,7 +21,7 @@ const openai_1 = require("openai");
 const uuid_1 = require("uuid");
 const request_1 = require("./utils/request");
 class OpenlayerClient {
-    constructor({ openlayerApiKey, openlayerInferencePipelineName, openlayerProjectName, openlayerServerUrl, }) {
+    constructor({ openlayerApiKey, openlayerServerUrl, }) {
         this.openlayerServerUrl = 'https://api.openlayer.com/v1';
         this.version = '0.1.0a16';
         this.resolvedQuery = (endpoint, args = {}) => (0, request_1.resolvedQuery)(this.openlayerServerUrl, endpoint, args);
@@ -56,13 +56,13 @@ class OpenlayerClient {
          * @throws Throws an error if Openlayer API key or project name are not set.
          * @returns A promise that resolves when the data has been successfully uploaded.
          */
-        this.uploadDataToOpenlayer = (data, config) => __awaiter(this, void 0, void 0, function* () {
-            if (!this.openlayerApiKey || !this.openlayerProjectName) {
-                throw new Error('Openlayer API key and project name are required for publishing.');
+        this.uploadDataToOpenlayer = (data, config, projectName, inferencePipelineName = 'production') => __awaiter(this, void 0, void 0, function* () {
+            if (!this.openlayerApiKey) {
+                throw new Error('Openlayer API key are required for publishing.');
             }
             try {
-                const project = yield this.createProject(this.openlayerProjectName, 'llm-base');
-                const inferencePipeline = yield this.createInferencePipeline(this.openlayerProjectName, project.id);
+                const project = yield this.createProject(projectName, 'llm-base');
+                const inferencePipeline = yield this.createInferencePipeline(inferencePipelineName, project.id);
                 yield this.uploadToInferencePipeline(inferencePipeline.id, data, config);
             }
             catch (error) {
@@ -77,7 +77,7 @@ class OpenlayerClient {
          * @throws Throws an error if the inference pipeline cannot be created or found.
          * @returns A promise that resolves to an OpenlayerInferencePipeline object.
          */
-        this.createInferencePipeline = (name, projectId) => __awaiter(this, void 0, void 0, function* () {
+        this.createInferencePipeline = (projectId, name = 'production') => __awaiter(this, void 0, void 0, function* () {
             try {
                 return yield this.loadInferencePipeline(name, projectId);
             }
@@ -87,9 +87,7 @@ class OpenlayerClient {
                 const createInferencePipelineResponse = yield fetch(createInferencePipelineQuery, {
                     body: JSON.stringify({
                         description: '',
-                        name: typeof this.openlayerInferencePipelineName === 'undefined'
-                            ? 'production'
-                            : this.openlayerInferencePipelineName,
+                        name,
                     }),
                     headers: {
                         Authorization: `Bearer ${this.openlayerApiKey}`,
@@ -149,10 +147,10 @@ class OpenlayerClient {
          * @throws Throws an error if the inference pipeline is not found.
          * @returns A promise that resolves to an OpenlayerInferencePipeline object.
          */
-        this.loadInferencePipeline = (name, projectId) => __awaiter(this, void 0, void 0, function* () {
+        this.loadInferencePipeline = (projectId, name = 'production') => __awaiter(this, void 0, void 0, function* () {
             const inferencePipelineEndpoint = `/projects/${projectId}/inference-pipelines`;
             const inferencePipelineQueryParameters = {
-                name: this.openlayerInferencePipelineName,
+                name,
                 version: this.version,
             };
             const inferencePipelineQuery = this.resolvedQuery(inferencePipelineEndpoint, inferencePipelineQueryParameters);
@@ -197,18 +195,16 @@ class OpenlayerClient {
             return project;
         });
         this.openlayerApiKey = openlayerApiKey;
-        this.openlayerInferencePipelineName = openlayerInferencePipelineName;
-        this.openlayerProjectName = openlayerProjectName;
         if (openlayerServerUrl) {
             this.openlayerServerUrl = openlayerServerUrl;
         }
-        if (!this.openlayerApiKey || !this.openlayerProjectName) {
-            throw new Error('Openlayer API key and project name are required for publishing.');
+        if (!this.openlayerApiKey) {
+            throw new Error('Openlayer API key are required for publishing.');
         }
     }
 }
 class OpenAIMonitor {
-    constructor({ openAiApiKey, openlayerApiKey, openlayerInferencePipelineName, openlayerProjectName, openlayerServerUrl, }) {
+    constructor({ openAiApiKey, openlayerApiKey, openlayerProjectName, openlayerInferencePipelineName, openlayerServerUrl, }) {
         this.openlayerDefaultDataConfig = {
             groundTruthColumnName: null,
             inferenceIdColumnName: 'id',
@@ -218,6 +214,7 @@ class OpenAIMonitor {
             outputColumnName: 'output',
             timestampColumnName: 'timestamp',
         };
+        this.openlayerInferencePipelineName = 'production';
         this.monitoringOn = false;
         this.formatChatCompletionInput = (messages) => messages
             .filter(({ role }) => role === 'user')
@@ -268,7 +265,7 @@ class OpenAIMonitor {
                     latency,
                     output: outputData,
                     timestamp: startTime,
-                }, this.openlayerDefaultDataConfig);
+                }, this.openlayerDefaultDataConfig, this.openlayerProjectName, this.openlayerInferencePipelineName);
             }
             else {
                 const nonStreamedResponse = response;
@@ -285,7 +282,7 @@ class OpenAIMonitor {
                     output,
                     timestamp: startTime,
                     tokens: (_e = (_d = nonStreamedResponse.usage) === null || _d === void 0 ? void 0 : _d.total_tokens) !== null && _e !== void 0 ? _e : 0,
-                }, this.openlayerDefaultDataConfig);
+                }, this.openlayerDefaultDataConfig, this.openlayerProjectName, this.openlayerInferencePipelineName);
             }
             return response;
         });
@@ -339,7 +336,7 @@ class OpenAIMonitor {
                     output: outputData,
                     timestamp: startTime,
                     tokens: tokensData,
-                }, this.openlayerDefaultDataConfig);
+                }, this.openlayerDefaultDataConfig, this.openlayerProjectName, this.openlayerInferencePipelineName);
             }
             else {
                 const nonStreamedResponse = response;
@@ -352,14 +349,16 @@ class OpenAIMonitor {
                     output: nonStreamedResponse.choices[0].text,
                     timestamp: startTime,
                     tokens: (_o = (_m = nonStreamedResponse.usage) === null || _m === void 0 ? void 0 : _m.total_tokens) !== null && _o !== void 0 ? _o : 0,
-                }, this.openlayerDefaultDataConfig);
+                }, this.openlayerDefaultDataConfig, this.openlayerProjectName, this.openlayerInferencePipelineName);
             }
             return response;
         });
+        this.openlayerProjectName = openlayerProjectName;
+        if (openlayerInferencePipelineName) {
+            this.openlayerInferencePipelineName = openlayerInferencePipelineName;
+        }
         this.openlayerClient = new OpenlayerClient({
             openlayerApiKey,
-            openlayerInferencePipelineName,
-            openlayerProjectName,
             openlayerServerUrl,
         });
         this.openAIClient = new openai_1.default({

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "openlayer",
-  "version": "0.1.6",
+  "version": "0.1.8",
   "description": "The Openlayer TypeScript client",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",