opik 0.1.1 → 0.1.2

package/dist/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 });
   }
@@ -12703,6 +12759,37 @@ var TraceBatchQueue = class extends BatchQueue {
   }
 };
 
+// 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
+      });
+    }
+  }
+};
+
 // src/opik/client/Client.ts
 var clients = [];
 var OpikClient = class {
@@ -12725,6 +12812,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({
@@ -12734,6 +12823,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
+    );
     clients.push(this);
   }
 };
@@ -12898,7 +12993,7 @@ function track(optionsOrOriginalFunction, originalFunction) {
       }
       return executeTrack(options, originalMethod2);
     }
-    const [target, propertyKey, descriptor] = args;
+    const [, , descriptor] = args;
     if (!descriptor || typeof descriptor.value !== "function") {
       throw new Error("track decorator can only be applied to methods");
     }