openlayer 0.1.9 → 0.1.11

package/dist/index.d.ts

@@ -127,8 +127,9 @@ type OpenlayerSampleVolumeGraph = {
     weekly: OpenlayerSampleVolumeGraphBucket;
 };
 type OpenlayerTaskType = 'llm-base' | 'tabular-classification' | 'tabular-regression' | 'text-classification';
-declare class OpenlayerClient {
+export declare class OpenlayerClient {
     private openlayerApiKey?;
+    private openlayerDefaultDataConfig;
     private openlayerServerUrl;
     private version;
     constructor({ openlayerApiKey, openlayerServerUrl, }: OpenlayerClientConstructorProps);
@@ -141,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.
      */
-    streamData: (data: ChatCompletionData, config: ChatCompletionConfig, inferencePipelineId: string) => Promise<void>;
+    streamData: (data: ChatCompletionData, inferencePipelineId: 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.
@@ -172,7 +173,6 @@ declare class OpenlayerClient {
 export declare class OpenAIMonitor {
     private openlayerClient;
     private openAIClient;
-    private openlayerDefaultDataConfig;
     private openlayerProjectName;
     private openlayerInferencePipelineName;
     private monitoringOn;
@@ -205,4 +205,4 @@ export declare class OpenAIMonitor {
      */
     stopMonitoring(): void;
 }
-export default OpenlayerClient;
+export {};

package/dist/index.js

@@ -16,12 +16,21 @@ var __asyncValues = (this && this.__asyncValues) || function (o) {
     function settle(resolve, reject, d, v) { Promise.resolve(v).then(function(v) { resolve({ value: v, done: d }); }, reject); }
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.OpenAIMonitor = void 0;
+exports.OpenAIMonitor = exports.OpenlayerClient = void 0;
 const openai_1 = require("openai");
 const uuid_1 = require("uuid");
 const request_1 = require("./utils/request");
 class OpenlayerClient {
     constructor({ openlayerApiKey, openlayerServerUrl, }) {
+        this.openlayerDefaultDataConfig = {
+            groundTruthColumnName: null,
+            inferenceIdColumnName: 'id',
+            inputVariableNames: ['input'],
+            latencyColumnName: 'latency',
+            numOfTokenColumnName: 'tokens',
+            outputColumnName: 'output',
+            timestampColumnName: 'timestamp',
+        };
         this.openlayerServerUrl = 'https://api.openlayer.com/v1';
         this.version = '0.1.0a16';
         this.resolvedQuery = (endpoint, args = {}) => (0, request_1.resolvedQuery)(this.openlayerServerUrl, endpoint, args);
@@ -33,7 +42,7 @@ 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.streamData = (data, config, inferencePipelineId) => __awaiter(this, void 0, void 0, function* () {
+        this.streamData = (data, inferencePipelineId) => __awaiter(this, void 0, void 0, function* () {
             var _a;
             if (!this.openlayerApiKey) {
                 throw new Error('Openlayer API key are required for streaming data.');
@@ -43,7 +52,7 @@ class OpenlayerClient {
                 const dataStreamQuery = this.resolvedQuery(dataStreamEndpoint);
                 const response = yield fetch(dataStreamQuery, {
                     body: JSON.stringify({
-                        config,
+                        config: this.openlayerDefaultDataConfig,
                         rows: [
                             Object.assign(Object.assign({}, data), { id: (0, uuid_1.v4)(), timestamp: Math.round(((_a = data.timestamp) !== null && _a !== void 0 ? _a : Date.now()) / 1000) }),
                         ],
@@ -74,7 +83,7 @@ class OpenlayerClient {
          */
         this.createInferencePipeline = (projectId, name = 'production') => __awaiter(this, void 0, void 0, function* () {
             try {
-                return yield this.loadInferencePipeline(name, projectId);
+                return yield this.loadInferencePipeline(projectId, name);
             }
             catch (_b) {
                 const createInferencePipelineEndpoint = `/projects/${projectId}/inference-pipelines`;
@@ -198,17 +207,9 @@ class OpenlayerClient {
         }
     }
 }
+exports.OpenlayerClient = OpenlayerClient;
 class OpenAIMonitor {
     constructor({ openAiApiKey, openlayerApiKey, openlayerProjectName, openlayerInferencePipelineName, openlayerServerUrl, }) {
-        this.openlayerDefaultDataConfig = {
-            groundTruthColumnName: null,
-            inferenceIdColumnName: 'id',
-            inputVariableNames: ['input'],
-            latencyColumnName: 'latency',
-            numOfTokenColumnName: 'tokens',
-            outputColumnName: 'output',
-            timestampColumnName: 'timestamp',
-        };
         this.openlayerInferencePipelineName = 'production';
         this.monitoringOn = false;
         this.formatChatCompletionInput = (messages) => messages
@@ -262,7 +263,7 @@ class OpenAIMonitor {
                     latency,
                     output: outputData,
                     timestamp: startTime,
-                }, this.openlayerDefaultDataConfig, inferencePipeline.id);
+                }, inferencePipeline.id);
             }
             else {
                 const nonStreamedResponse = response;
@@ -279,7 +280,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, inferencePipeline.id);
+                }, inferencePipeline.id);
             }
             return response;
         });
@@ -335,7 +336,7 @@ class OpenAIMonitor {
                     output: outputData,
                     timestamp: startTime,
                     tokens: tokensData,
-                }, this.openlayerDefaultDataConfig, inferencePipeline.id);
+                }, inferencePipeline.id);
             }
             else {
                 const nonStreamedResponse = response;
@@ -348,7 +349,7 @@ 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, inferencePipeline.id);
+                }, inferencePipeline.id);
             }
             return response;
         });
@@ -389,4 +390,3 @@ class OpenAIMonitor {
     }
 }
 exports.OpenAIMonitor = OpenAIMonitor;
-exports.default = OpenlayerClient;

package/package.json

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