opik 0.1.1 → 0.1.2

package/dist/{Client-VaP_If_d.d.cts → Client-DilyWTdT.d.cts}

@@ -4666,6 +4666,12 @@ declare class Span {
     private opik;
     constructor(data: SavedSpan, opik: OpikClient);
     end: () => this;
+    score: (score: {
+        name: string;
+        categoryName?: string;
+        value: number;
+        reason?: string;
+    }) => void;
     update: (updates: Omit<SpanUpdate$1, "traceId" | "parentSpanId" | "projectId" | "projectName">) => this;
 }
 
@@ -4681,14 +4687,17 @@ declare class Trace {
     private spans;
     constructor(data: SavedTrace, opik: OpikClient);
     end: () => this;
+    score: (score: {
+        name: string;
+        categoryName?: string;
+        value: number;
+        reason?: string;
+    }) => void;
     span: (spanData: SpanData) => Span;
     update: (updates: Omit<TraceUpdate, "projectId">) => this;
 }
 
-type CreateEntity = {
-    id: string;
-};
-declare abstract class BatchQueue<EntityData = {}> {
+declare abstract class BatchQueue<EntityData = object, EntityId = string> {
     private readonly createQueue;
     private readonly updateQueue;
     private readonly deleteQueue;
@@ -4700,13 +4709,14 @@ declare abstract class BatchQueue<EntityData = {}> {
         name?: string;
     });
     protected abstract createEntities(entities: EntityData[]): Promise<void>;
-    protected abstract getEntity(id: string): Promise<EntityData | undefined>;
-    protected abstract updateEntity(id: string, updates: Partial<EntityData>): Promise<void>;
-    protected abstract deleteEntities(ids: string[]): Promise<void>;
-    create: (entity: CreateEntity & EntityData) => void;
-    get: (id: string) => Promise<EntityData | undefined>;
-    update: (id: string, updates: Partial<EntityData>) => void;
-    delete: (id: string) => void;
+    protected abstract getEntity(id: EntityId): Promise<EntityData | undefined>;
+    protected abstract updateEntity(id: EntityId, updates: Partial<EntityData>): Promise<void>;
+    protected abstract deleteEntities(ids: EntityId[]): Promise<void>;
+    protected abstract getId(entity: EntityData): EntityId;
+    create: (entity: EntityData) => void;
+    get: (id: EntityId) => Promise<EntityData | undefined>;
+    update: (id: EntityId, updates: Partial<EntityData>) => void;
+    delete: (id: EntityId) => void;
     flush: () => Promise<void>;
 }
 
@@ -4716,21 +4726,57 @@ type SpanUpdate = Partial<SavedSpan> & {
 declare class SpanBatchQueue extends BatchQueue<SavedSpan> {
     private readonly api;
     constructor(api: OpikApiClient, delay?: number);
+    protected getId(entity: SavedSpan): string;
     protected createEntities(spans: SavedSpan[]): Promise<void>;
     protected getEntity(id: string): Promise<SavedSpan>;
     protected updateEntity(id: string, updates: SpanUpdate): Promise<void>;
     protected deleteEntities(ids: string[]): Promise<void>;
 }
 
+type FeedbackScoreId$1 = {
+    id: string;
+    name: string;
+};
+declare class SpanFeedbackScoresBatchQueue extends BatchQueue<FeedbackScoreBatchItem, FeedbackScoreId$1> {
+    private readonly api;
+    constructor(api: OpikApiClient, delay?: number);
+    protected getId(entity: FeedbackScoreBatchItem): {
+        id: string;
+        name: string;
+    };
+    protected createEntities(scores: FeedbackScoreBatchItem[]): Promise<void>;
+    protected getEntity(): Promise<FeedbackScoreBatchItem | undefined>;
+    protected updateEntity(): Promise<void>;
+    protected deleteEntities(scoreIds: FeedbackScoreId$1[]): Promise<void>;
+}
+
 declare class TraceBatchQueue extends BatchQueue<SavedTrace> {
     private readonly api;
     constructor(api: OpikApiClient, delay?: number);
+    protected getId(entity: SavedTrace): string;
     protected createEntities(traces: SavedTrace[]): Promise<void>;
     protected getEntity(id: string): Promise<SavedTrace>;
     protected updateEntity(id: string, updates: Partial<SavedTrace>): Promise<void>;
     protected deleteEntities(ids: string[]): Promise<void>;
 }
 
+type FeedbackScoreId = {
+    id: string;
+    name: string;
+};
+declare class TraceFeedbackScoresBatchQueue extends BatchQueue<FeedbackScoreBatchItem, FeedbackScoreId> {
+    private readonly api;
+    constructor(api: OpikApiClient, delay?: number);
+    protected getId(entity: FeedbackScoreBatchItem): {
+        id: string;
+        name: string;
+    };
+    protected createEntities(scores: FeedbackScoreBatchItem[]): Promise<void>;
+    protected getEntity(): Promise<FeedbackScoreBatchItem | undefined>;
+    protected updateEntity(): Promise<void>;
+    protected deleteEntities(scoreIds: FeedbackScoreId[]): Promise<void>;
+}
+
 interface TraceData extends Omit<Trace$1, "startTime"> {
     startTime?: Date;
 }
@@ -4739,6 +4785,8 @@ declare class OpikClient {
     config: OpikConfig;
     spanBatchQueue: SpanBatchQueue;
     traceBatchQueue: TraceBatchQueue;
+    spanFeedbackScoresBatchQueue: SpanFeedbackScoresBatchQueue;
+    traceFeedbackScoresBatchQueue: TraceFeedbackScoresBatchQueue;
     constructor(explicitConfig?: Partial<OpikConfig>);
     trace: (traceData: TraceData) => Trace;
     flush: () => Promise<void>;

package/dist/{Client-VaP_If_d.d.ts → Client-DilyWTdT.d.ts}

@@ -4666,6 +4666,12 @@ declare class Span {
     private opik;
     constructor(data: SavedSpan, opik: OpikClient);
     end: () => this;
+    score: (score: {
+        name: string;
+        categoryName?: string;
+        value: number;
+        reason?: string;
+    }) => void;
     update: (updates: Omit<SpanUpdate$1, "traceId" | "parentSpanId" | "projectId" | "projectName">) => this;
 }
 
@@ -4681,14 +4687,17 @@ declare class Trace {
     private spans;
     constructor(data: SavedTrace, opik: OpikClient);
     end: () => this;
+    score: (score: {
+        name: string;
+        categoryName?: string;
+        value: number;
+        reason?: string;
+    }) => void;
     span: (spanData: SpanData) => Span;
     update: (updates: Omit<TraceUpdate, "projectId">) => this;
 }
 
-type CreateEntity = {
-    id: string;
-};
-declare abstract class BatchQueue<EntityData = {}> {
+declare abstract class BatchQueue<EntityData = object, EntityId = string> {
     private readonly createQueue;
     private readonly updateQueue;
     private readonly deleteQueue;
@@ -4700,13 +4709,14 @@ declare abstract class BatchQueue<EntityData = {}> {
         name?: string;
     });
     protected abstract createEntities(entities: EntityData[]): Promise<void>;
-    protected abstract getEntity(id: string): Promise<EntityData | undefined>;
-    protected abstract updateEntity(id: string, updates: Partial<EntityData>): Promise<void>;
-    protected abstract deleteEntities(ids: string[]): Promise<void>;
-    create: (entity: CreateEntity & EntityData) => void;
-    get: (id: string) => Promise<EntityData | undefined>;
-    update: (id: string, updates: Partial<EntityData>) => void;
-    delete: (id: string) => void;
+    protected abstract getEntity(id: EntityId): Promise<EntityData | undefined>;
+    protected abstract updateEntity(id: EntityId, updates: Partial<EntityData>): Promise<void>;
+    protected abstract deleteEntities(ids: EntityId[]): Promise<void>;
+    protected abstract getId(entity: EntityData): EntityId;
+    create: (entity: EntityData) => void;
+    get: (id: EntityId) => Promise<EntityData | undefined>;
+    update: (id: EntityId, updates: Partial<EntityData>) => void;
+    delete: (id: EntityId) => void;
     flush: () => Promise<void>;
 }
 
@@ -4716,21 +4726,57 @@ type SpanUpdate = Partial<SavedSpan> & {
 declare class SpanBatchQueue extends BatchQueue<SavedSpan> {
     private readonly api;
     constructor(api: OpikApiClient, delay?: number);
+    protected getId(entity: SavedSpan): string;
     protected createEntities(spans: SavedSpan[]): Promise<void>;
     protected getEntity(id: string): Promise<SavedSpan>;
     protected updateEntity(id: string, updates: SpanUpdate): Promise<void>;
     protected deleteEntities(ids: string[]): Promise<void>;
 }
 
+type FeedbackScoreId$1 = {
+    id: string;
+    name: string;
+};
+declare class SpanFeedbackScoresBatchQueue extends BatchQueue<FeedbackScoreBatchItem, FeedbackScoreId$1> {
+    private readonly api;
+    constructor(api: OpikApiClient, delay?: number);
+    protected getId(entity: FeedbackScoreBatchItem): {
+        id: string;
+        name: string;
+    };
+    protected createEntities(scores: FeedbackScoreBatchItem[]): Promise<void>;
+    protected getEntity(): Promise<FeedbackScoreBatchItem | undefined>;
+    protected updateEntity(): Promise<void>;
+    protected deleteEntities(scoreIds: FeedbackScoreId$1[]): Promise<void>;
+}
+
 declare class TraceBatchQueue extends BatchQueue<SavedTrace> {
     private readonly api;
     constructor(api: OpikApiClient, delay?: number);
+    protected getId(entity: SavedTrace): string;
     protected createEntities(traces: SavedTrace[]): Promise<void>;
     protected getEntity(id: string): Promise<SavedTrace>;
     protected updateEntity(id: string, updates: Partial<SavedTrace>): Promise<void>;
     protected deleteEntities(ids: string[]): Promise<void>;
 }
 
+type FeedbackScoreId = {
+    id: string;
+    name: string;
+};
+declare class TraceFeedbackScoresBatchQueue extends BatchQueue<FeedbackScoreBatchItem, FeedbackScoreId> {
+    private readonly api;
+    constructor(api: OpikApiClient, delay?: number);
+    protected getId(entity: FeedbackScoreBatchItem): {
+        id: string;
+        name: string;
+    };
+    protected createEntities(scores: FeedbackScoreBatchItem[]): Promise<void>;
+    protected getEntity(): Promise<FeedbackScoreBatchItem | undefined>;
+    protected updateEntity(): Promise<void>;
+    protected deleteEntities(scoreIds: FeedbackScoreId[]): Promise<void>;
+}
+
 interface TraceData extends Omit<Trace$1, "startTime"> {
     startTime?: Date;
 }
@@ -4739,6 +4785,8 @@ declare class OpikClient {
     config: OpikConfig;
     spanBatchQueue: SpanBatchQueue;
     traceBatchQueue: TraceBatchQueue;
+    spanFeedbackScoresBatchQueue: SpanFeedbackScoresBatchQueue;
+    traceFeedbackScoresBatchQueue: TraceFeedbackScoresBatchQueue;
     constructor(explicitConfig?: Partial<OpikConfig>);
     trace: (traceData: TraceData) => Trace;
     flush: () => Promise<void>;

package/dist/{chunk-OC5DO255.js → chunk-ZB4ZJRLY.js}

@@ -11945,6 +11945,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 = {
@@ -11969,6 +11978,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;
@@ -12060,7 +12078,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);
@@ -12129,6 +12148,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 });
   }
@@ -12145,12 +12167,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 });
   }
@@ -12165,6 +12221,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 {
@@ -12187,6 +12274,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({
@@ -12196,6 +12285,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);
   }
 };
@@ -12360,7 +12455,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");
     }
@@ -12380,5 +12475,5 @@ var flushAll = async () => {
 };
 
 export { OpikClient, flushAll, getTrackContext, track, trackOpikClient };
-//# sourceMappingURL=chunk-OC5DO255.js.map
-//# sourceMappingURL=chunk-OC5DO255.js.map
+//# sourceMappingURL=chunk-ZB4ZJRLY.js.map
+//# sourceMappingURL=chunk-ZB4ZJRLY.js.map