openlayer 0.0.4 → 0.0.5

package/dist/index.js

@@ -8,17 +8,6 @@ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, ge
         step((generator = generator.apply(thisArg, _arguments || [])).next());
     });
 };
-var __rest = (this && this.__rest) || function (s, e) {
-    var t = {};
-    for (var p in s) if (Object.prototype.hasOwnProperty.call(s, p) && e.indexOf(p) < 0)
-        t[p] = s[p];
-    if (s != null && typeof Object.getOwnPropertySymbols === "function")
-        for (var i = 0, p = Object.getOwnPropertySymbols(s); i < p.length; i++) {
-            if (e.indexOf(p[i]) < 0 && Object.prototype.propertyIsEnumerable.call(s, p[i]))
-                t[p[i]] = s[p[i]];
-        }
-    return t;
-};
 var __asyncValues = (this && this.__asyncValues) || function (o) {
     if (!Symbol.asyncIterator) throw new TypeError("Symbol.asyncIterator is not defined.");
     var m = o[Symbol.asyncIterator], i;
@@ -27,6 +16,7 @@ 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 });
+const node_fetch_1 = require("node-fetch");
 const openai_1 = require("openai");
 const request_1 = require("./utils/request");
 class OpenAIMonitor {
@@ -44,7 +34,7 @@ class OpenAIMonitor {
             const uploadToInferencePipeline = (id) => __awaiter(this, void 0, void 0, function* () {
                 const dataStreamEndpoint = `/inference-pipelines/${id}/data-stream`;
                 const dataStreamQuery = this.resolvedQuery(dataStreamEndpoint);
-                const response = yield fetch(dataStreamQuery, {
+                const response = yield (0, node_fetch_1.default)(dataStreamQuery, {
                     body: JSON.stringify({
                         config: {
                             inferenceIdColumnName: 'inference_id',
@@ -62,7 +52,6 @@ class OpenAIMonitor {
                     },
                     method: 'POST',
                 });
-                console.log('bruh', response, dataStreamQuery);
                 if (!response.ok) {
                     console.error('Error making POST request:', response.status);
                     throw new Error(`Error: ${response.status}`);
@@ -78,7 +67,7 @@ class OpenAIMonitor {
                     name: this.openlayerProjectName,
                 };
                 const projectsQuery = this.resolvedQuery(projectsEndpoint, projectsQueryParameters);
-                const projectsResponse = yield fetch(projectsQuery, {
+                const projectsResponse = yield (0, node_fetch_1.default)(projectsQuery, {
                     headers: {
                         Authorization: `Bearer ${this.openlayerApiKey}`,
                         'Content-Type': 'application/json',
@@ -98,7 +87,7 @@ class OpenAIMonitor {
                     name: this.openlayerInferencePipelineName,
                 };
                 const inferencePipelineQuery = this.resolvedQuery(inferencePipelineEndpoint, inferencePipelineQueryParameters);
-                const inferencePipelineResponse = yield fetch(inferencePipelineQuery, {
+                const inferencePipelineResponse = yield (0, node_fetch_1.default)(inferencePipelineQuery, {
                     headers: {
                         Authorization: `Bearer ${this.openlayerApiKey}`,
                         'Content-Type': 'application/json',
@@ -117,7 +106,7 @@ class OpenAIMonitor {
                 else {
                     const createInferencePipelineEndpoint = `/projects/${project.id}/inference-pipelines`;
                     const createInferencePipelineQuery = this.resolvedQuery(createInferencePipelineEndpoint);
-                    const createInferencePipelineResponse = yield fetch(createInferencePipelineQuery, {
+                    const createInferencePipelineResponse = yield (0, node_fetch_1.default)(createInferencePipelineQuery, {
                         body: JSON.stringify({
                             description: '',
                             name: typeof this.openlayerInferencePipelineName === 'undefined'
@@ -130,8 +119,7 @@ class OpenAIMonitor {
                         },
                         method: 'POST',
                     });
-                    const _a = yield createInferencePipelineResponse.json(), { id: inferencePipelineId } = _a, response = __rest(_a, ["id"]);
-                    console.log(createInferencePipelineResponse, response, inferencePipelineId);
+                    const { id: inferencePipelineId } = yield createInferencePipelineResponse.json();
                     if (!inferencePipelineId) {
                         throw new Error('Error creating inference pipeline');
                     }
@@ -144,8 +132,8 @@ class OpenAIMonitor {
             }
         });
         this.createChatCompletion = (body, options) => __awaiter(this, void 0, void 0, function* () {
-            var _b, e_1, _c, _d;
-            var _e, _f;
+            var _a, e_1, _b, _c;
+            var _d, _e;
             if (!this.monitoringOn) {
                 throw new Error('Monitoring is not active.');
             }
@@ -157,10 +145,10 @@ class OpenAIMonitor {
             if (body.stream) {
                 const streamedResponse = response;
                 try {
-                    for (var _g = true, streamedResponse_1 = __asyncValues(streamedResponse), streamedResponse_1_1; streamedResponse_1_1 = yield streamedResponse_1.next(), _b = streamedResponse_1_1.done, !_b; _g = true) {
-                        _d = streamedResponse_1_1.value;
-                        _g = false;
-                        const chunk = _d;
+                    for (var _f = true, streamedResponse_1 = __asyncValues(streamedResponse), streamedResponse_1_1; streamedResponse_1_1 = yield streamedResponse_1.next(), _a = streamedResponse_1_1.done, !_a; _f = true) {
+                        _c = streamedResponse_1_1.value;
+                        _f = false;
+                        const chunk = _c;
                         // Process each chunk - for example, accumulate input data
                         outputData += chunk.choices[0].delta.content;
                     }
@@ -168,7 +156,7 @@ class OpenAIMonitor {
                 catch (e_1_1) { e_1 = { error: e_1_1 }; }
                 finally {
                     try {
-                        if (!_g && !_b && (_c = streamedResponse_1.return)) yield _c.call(streamedResponse_1);
+                        if (!_f && !_a && (_b = streamedResponse_1.return)) yield _b.call(streamedResponse_1);
                     }
                     finally { if (e_1) throw e_1.error; }
                 }
@@ -189,14 +177,14 @@ class OpenAIMonitor {
                     input: this.formatChatCompletionInput(body.messages),
                     latency,
                     output: nonStreamedResponse.choices[0].message.content,
-                    tokens: (_f = (_e = nonStreamedResponse.usage) === null || _e === void 0 ? void 0 : _e.total_tokens) !== null && _f !== void 0 ? _f : 0,
+                    tokens: (_e = (_d = nonStreamedResponse.usage) === null || _d === void 0 ? void 0 : _d.total_tokens) !== null && _e !== void 0 ? _e : 0,
                 });
             }
             return response;
         });
         this.createCompletion = (body, options) => __awaiter(this, void 0, void 0, function* () {
-            var _h, e_2, _j, _k;
-            var _l, _m, _o, _p;
+            var _g, e_2, _h, _j;
+            var _k, _l, _m, _o;
             if (!this.monitoringOn) {
                 throw new Error('Monitoring is not active.');
             }
@@ -209,19 +197,19 @@ class OpenAIMonitor {
             if (body.stream) {
                 const streamedResponse = response;
                 try {
-                    for (var _q = true, streamedResponse_2 = __asyncValues(streamedResponse), streamedResponse_2_1; streamedResponse_2_1 = yield streamedResponse_2.next(), _h = streamedResponse_2_1.done, !_h; _q = true) {
-                        _k = streamedResponse_2_1.value;
-                        _q = false;
-                        const chunk = _k;
+                    for (var _p = true, streamedResponse_2 = __asyncValues(streamedResponse), streamedResponse_2_1; streamedResponse_2_1 = yield streamedResponse_2.next(), _g = streamedResponse_2_1.done, !_g; _p = true) {
+                        _j = streamedResponse_2_1.value;
+                        _p = false;
+                        const chunk = _j;
                         // Process each chunk - for example, accumulate input data
                         outputData += chunk.choices[0].text.trim();
-                        tokensData += (_m = (_l = chunk.usage) === null || _l === void 0 ? void 0 : _l.total_tokens) !== null && _m !== void 0 ? _m : 0;
+                        tokensData += (_l = (_k = chunk.usage) === null || _k === void 0 ? void 0 : _k.total_tokens) !== null && _l !== void 0 ? _l : 0;
                     }
                 }
                 catch (e_2_1) { e_2 = { error: e_2_1 }; }
                 finally {
                     try {
-                        if (!_q && !_h && (_j = streamedResponse_2.return)) yield _j.call(streamedResponse_2);
+                        if (!_p && !_g && (_h = streamedResponse_2.return)) yield _h.call(streamedResponse_2);
                     }
                     finally { if (e_2) throw e_2.error; }
                 }
@@ -243,7 +231,7 @@ class OpenAIMonitor {
                     input: body.prompt,
                     latency,
                     output: nonStreamedResponse.choices[0].text,
-                    tokens: (_p = (_o = nonStreamedResponse.usage) === null || _o === void 0 ? void 0 : _o.total_tokens) !== null && _p !== void 0 ? _p : 0,
+                    tokens: (_o = (_m = nonStreamedResponse.usage) === null || _m === void 0 ? void 0 : _m.total_tokens) !== null && _o !== void 0 ? _o : 0,
                 });
             }
             return response;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "openlayer",
-  "version": "0.0.4",
+  "version": "0.0.5",
   "description": "The Openlayer TypeScript client",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
@@ -18,6 +18,7 @@
     "eslint-config-prettier": "^9.0.0",
     "eslint-plugin-prettier": "^5.0.1",
     "eslint-plugin-typescript-sort-keys": "^3.1.0",
+    "node-fetch": "^3.3.2",
     "openai": "^4.19.0"
   },
   "devDependencies": {