opik 0.1.1 → 0.1.2

package/dist/index.d.cts

@@ -1,5 +1,5 @@
-import { O as OpikClient, S as Span, T as Trace, a as SpanType } from './Client-VaP_If_d.cjs';
-export { b as OpikConfig } from './Client-VaP_If_d.cjs';
+import { O as OpikClient, S as Span, T as Trace, a as SpanType } from './Client-DilyWTdT.cjs';
+export { b as OpikConfig } from './Client-DilyWTdT.cjs';
 import 'stream';
 
 type TrackContext = {

package/dist/index.d.ts

@@ -1,5 +1,5 @@
-import { O as OpikClient, S as Span, T as Trace, a as SpanType } from './Client-VaP_If_d.js';
-export { b as OpikConfig } from './Client-VaP_If_d.js';
+import { O as OpikClient, S as Span, T as Trace, a as SpanType } from './Client-DilyWTdT.js';
+export { b as OpikConfig } from './Client-DilyWTdT.js';
 import 'stream';
 
 type TrackContext = {

package/dist/index.js

@@ -1,4 +1,4 @@
-export { OpikClient as Opik, flushAll, getTrackContext, track, trackOpikClient } from './chunk-OC5DO255.js';
+export { OpikClient as Opik, flushAll, getTrackContext, track, trackOpikClient } from './chunk-ZB4ZJRLY.js';
 import './chunk-PZ5AY32C.js';
 //# sourceMappingURL=index.js.map
 //# sourceMappingURL=index.js.map

package/dist/vercel/index.cjs

@@ -12483,6 +12483,15 @@ var Span2 = class {
     this.end = () => {
       return this.update({ endTime: /* @__PURE__ */ new Date() });
     };
+    this.score = (score) => {
+      var _a;
+      this.opik.spanFeedbackScoresBatchQueue.create({
+        ...score,
+        projectName: (_a = this.data.projectName) != null ? _a : this.opik.config.projectName,
+        id: this.data.id,
+        source: "sdk"
+      });
+    };
     this.update = (updates) => {
       var _a;
       const spanUpdates = {
@@ -12507,6 +12516,15 @@ var Trace2 = class {
     this.end = () => {
       return this.update({ endTime: /* @__PURE__ */ new Date() });
     };
+    this.score = (score) => {
+      var _a;
+      this.opik.traceFeedbackScoresBatchQueue.create({
+        ...score,
+        projectName: (_a = this.data.projectName) != null ? _a : this.opik.config.projectName,
+        id: this.data.id,
+        source: "sdk"
+      });
+    };
     this.span = (spanData) => {
       var _a, _b;
       const projectName = (_b = (_a = this.data.projectName) != null ? _a : spanData.projectName) != null ? _b : this.opik.config.projectName;
@@ -12598,7 +12616,8 @@ var BatchQueue = class {
     name = "BatchQueue"
   }) {
     this.create = (entity) => {
-      this.createQueue.add(entity.id, entity);
+      const id = this.getId(entity);
+      this.createQueue.add(id, entity);
     };
     this.get = async (id) => {
       const entity = this.createQueue.queue.get(id);
@@ -12667,6 +12686,9 @@ var SpanBatchQueue = class extends BatchQueue {
     super({ delay, enableDeleteBatch: false, name: "SpanBatchQueue" });
     this.api = api;
   }
+  getId(entity) {
+    return entity.id;
+  }
   async createEntities(spans) {
     await this.api.spans.createSpans({ spans });
   }
@@ -12683,12 +12705,46 @@ var SpanBatchQueue = class extends BatchQueue {
   }
 };
 
+// src/opik/client/SpanFeedbackScoresBatchQueue.ts
+var SpanFeedbackScoresBatchQueue = class extends BatchQueue {
+  constructor(api, delay) {
+    super({
+      delay,
+      enableDeleteBatch: false,
+      name: "SpanFeedbackScoresBatchQueue"
+    });
+    this.api = api;
+  }
+  getId(entity) {
+    return { id: entity.id, name: entity.name };
+  }
+  async createEntities(scores) {
+    await this.api.spans.scoreBatchOfSpans({ scores });
+  }
+  async getEntity() {
+    throw new Error("Not implemented");
+  }
+  async updateEntity() {
+    throw new Error("Not implemented");
+  }
+  async deleteEntities(scoreIds) {
+    for (const scoreId of scoreIds) {
+      await this.api.spans.deleteSpanFeedbackScore(scoreId.id, {
+        name: scoreId.name
+      });
+    }
+  }
+};
+
 // src/opik/client/TraceBatchQueue.ts
 var TraceBatchQueue = class extends BatchQueue {
   constructor(api, delay) {
     super({ delay, name: "TraceBatchQueue" });
     this.api = api;
   }
+  getId(entity) {
+    return entity.id;
+  }
   async createEntities(traces) {
     await this.api.traces.createTraces({ traces });
   }
@@ -12702,6 +12758,37 @@ var TraceBatchQueue = class extends BatchQueue {
     await this.api.traces.deleteTraces({ ids });
   }
 };
+
+// src/opik/client/TraceFeedbackScoresBatchQueue.ts
+var TraceFeedbackScoresBatchQueue = class extends BatchQueue {
+  constructor(api, delay) {
+    super({
+      delay,
+      enableDeleteBatch: false,
+      name: "TraceFeedbackScoresBatchQueue"
+    });
+    this.api = api;
+  }
+  getId(entity) {
+    return { id: entity.id, name: entity.name };
+  }
+  async createEntities(scores) {
+    await this.api.traces.scoreBatchOfTraces({ scores });
+  }
+  async getEntity() {
+    throw new Error("Not implemented");
+  }
+  async updateEntity() {
+    throw new Error("Not implemented");
+  }
+  async deleteEntities(scoreIds) {
+    for (const scoreId of scoreIds) {
+      await this.api.traces.deleteTraceFeedbackScore(scoreId.id, {
+        name: scoreId.name
+      });
+    }
+  }
+};
 var OpikClient = class {
   constructor(explicitConfig) {
     this.trace = (traceData) => {
@@ -12722,6 +12809,8 @@ var OpikClient = class {
     this.flush = async () => {
       await this.traceBatchQueue.flush();
       await this.spanBatchQueue.flush();
+      await this.traceFeedbackScoresBatchQueue.flush();
+      await this.spanFeedbackScoresBatchQueue.flush();
     };
     this.config = loadConfig(explicitConfig);
     this.api = new OpikApiClient({
@@ -12731,6 +12820,12 @@ var OpikClient = class {
     });
     this.spanBatchQueue = new SpanBatchQueue(this.api);
     this.traceBatchQueue = new TraceBatchQueue(this.api);
+    this.spanFeedbackScoresBatchQueue = new SpanFeedbackScoresBatchQueue(
+      this.api
+    );
+    this.traceFeedbackScoresBatchQueue = new TraceFeedbackScoresBatchQueue(
+      this.api
+    );
   }
 };
 new async_hooks.AsyncLocalStorage();
@@ -12918,7 +13013,7 @@ function hrTimeToMilliseconds(hrTime) {
 function safeParseJson(value) {
   try {
     return JSON.parse(value);
-  } catch (e) {
+  } catch {
     return value;
   }
 }
@@ -12931,7 +13026,7 @@ function tryParseJSON(input) {
     if (parsed !== null && typeof parsed === "object" && !Array.isArray(parsed)) {
       return parsed;
     }
-  } catch (error) {
+  } catch {
     return void 0;
   }
   return void 0;