opik 0.1.0 → 0.1.2

package/dist/{Client-Y3fv8TIb.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,9 +4785,11 @@ declare class OpikClient {
     config: OpikConfig;
     spanBatchQueue: SpanBatchQueue;
     traceBatchQueue: TraceBatchQueue;
+    spanFeedbackScoresBatchQueue: SpanFeedbackScoresBatchQueue;
+    traceFeedbackScoresBatchQueue: TraceFeedbackScoresBatchQueue;
     constructor(explicitConfig?: Partial<OpikConfig>);
     trace: (traceData: TraceData) => Trace;
     flush: () => Promise<void>;
 }
 
-export { OpikClient as O, SpanType as S, Trace as T, type OpikConfig as a, Span as b };
+export { OpikClient as O, Span as S, Trace as T, SpanType as a, type OpikConfig as b };

package/dist/{Client-Y3fv8TIb.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,9 +4785,11 @@ declare class OpikClient {
     config: OpikConfig;
     spanBatchQueue: SpanBatchQueue;
     traceBatchQueue: TraceBatchQueue;
+    spanFeedbackScoresBatchQueue: SpanFeedbackScoresBatchQueue;
+    traceFeedbackScoresBatchQueue: TraceFeedbackScoresBatchQueue;
     constructor(explicitConfig?: Partial<OpikConfig>);
     trace: (traceData: TraceData) => Trace;
     flush: () => Promise<void>;
 }
 
-export { OpikClient as O, SpanType as S, Trace as T, type OpikConfig as a, Span as b };
+export { OpikClient as O, Span as S, Trace as T, SpanType as a, type OpikConfig as b };

package/dist/{chunk-MCPL44VT.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,10 +12285,24 @@ 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);
   }
 };
+var DEFAULT_TRACK_NAME = "track.decorator";
 var trackStorage = new AsyncLocalStorage();
+var getTrackContext = () => {
+  const { span, trace } = trackStorage.getStore() || {};
+  if (!span || !trace) {
+    return void 0;
+  }
+  return { span, trace };
+};
 function isPromise(obj) {
   return !!obj && (typeof obj === "object" || typeof obj === "function") && typeof obj.then === "function";
 }
@@ -12225,76 +12328,139 @@ function logSpan({
   });
   return { span, trace: spanTrace };
 }
-function withTrack({
+function logStart({
+  args,
+  span,
+  trace
+}) {
+  if (args.length === 0) {
+    return;
+  }
+  const input = { arguments: args };
+  span.update({ input });
+  if (trace) {
+    trace.update({ input });
+  }
+}
+function logSuccess({
+  result,
+  span,
+  trace
+}) {
+  const output = typeof result === "object" ? result : { result };
+  const endTime = /* @__PURE__ */ new Date();
+  span.update({ endTime, output });
+  if (trace) {
+    trace.update({ endTime, output });
+  }
+}
+function logError({
+  span,
+  error,
+  trace
+}) {
+  var _a, _b;
+  if (error instanceof Error) {
+    span.update({
+      errorInfo: {
+        message: error.message,
+        exceptionType: error.name,
+        traceback: (_a = error.stack) != null ? _a : ""
+      }
+    });
+  }
+  span.end();
+  if (trace) {
+    trace.update({
+      errorInfo: {
+        message: error.message,
+        exceptionType: error.name,
+        traceback: (_b = error.stack) != null ? _b : ""
+      }
+    });
+    trace.end();
+  }
+}
+function executeTrack({
   name,
   projectName,
   type
-} = {}) {
-  return function trackDecorator(originalFn) {
-    const wrappedFn = function(...args) {
-      const context = trackStorage.getStore();
-      const { span, trace } = logSpan({
-        name: name != null ? name : originalFn.name,
-        parentSpan: context == null ? void 0 : context.span,
-        projectName,
-        trace: context == null ? void 0 : context.trace,
-        type
-      });
-      const isRootSpan = !context;
-      const fnThis = this;
-      return trackStorage.run({ span, trace }, () => {
-        try {
-          const result = originalFn.apply(fnThis, args);
-          if (isPromise(result)) {
-            return result.then(
-              (res) => {
-                span.end();
-                if (isRootSpan) {
-                  trace.end();
-                }
-                return res;
-              },
-              (err) => {
-                span.end();
-                if (isRootSpan) {
-                  trace.end();
-                }
-                throw err;
-              }
-            );
-          }
-          span.end();
-          if (isRootSpan) {
-            trace.end();
-          }
-          return result;
-        } catch (e) {
-          if (isRootSpan) {
-            trace.end();
-          }
-          span.end();
-          throw e;
+} = {}, originalFn) {
+  const wrappedFn = function(...args) {
+    const context = trackStorage.getStore();
+    const { span, trace } = logSpan({
+      name: name != null ? name : originalFn.name || DEFAULT_TRACK_NAME,
+      parentSpan: context == null ? void 0 : context.span,
+      projectName,
+      trace: context == null ? void 0 : context.trace,
+      type
+    });
+    const isRootSpan = !context;
+    const fnThis = this;
+    return trackStorage.run({ span, trace }, () => {
+      try {
+        logStart({ args, span, trace: isRootSpan ? trace : void 0 });
+        const result = originalFn.apply(fnThis, args);
+        if (isPromise(result)) {
+          return result.then(
+            (res) => {
+              logSuccess({
+                span,
+                result: res,
+                trace: isRootSpan ? trace : void 0
+              });
+              return res;
+            },
+            (err) => {
+              logError({
+                span,
+                error: err,
+                trace: isRootSpan ? trace : void 0
+              });
+              throw err;
+            }
+          );
         }
-      });
-    };
-    return wrappedFn;
+        logSuccess({
+          span,
+          result,
+          trace: isRootSpan ? trace : void 0
+        });
+        return result;
+      } catch (error) {
+        logError({
+          span,
+          error,
+          trace: isRootSpan ? trace : void 0
+        });
+        throw error;
+      }
+    });
   };
+  return wrappedFn;
 }
-function track(options = {}) {
+function track(optionsOrOriginalFunction, originalFunction) {
+  if (typeof optionsOrOriginalFunction === "function") {
+    return executeTrack({}, optionsOrOriginalFunction);
+  }
+  const options = optionsOrOriginalFunction;
+  if (originalFunction) {
+    return executeTrack(options, originalFunction);
+  }
   return function(...args) {
     if (args.length === 2 && typeof args[1] === "object" && args[1] !== null && "kind" in args[1]) {
       const [originalMethod2, context] = args;
       if (context.kind !== "method") {
         throw new Error("track decorator is only applicable to methods");
       }
-      return withTrack(options)(originalMethod2);
+      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");
     }
     const originalMethod = descriptor.value;
-    descriptor.value = withTrack(options)(originalMethod);
+    descriptor.value = executeTrack(options, originalMethod);
     return descriptor;
   };
 }
@@ -12308,6 +12474,6 @@ var flushAll = async () => {
   ]);
 };
 
-export { OpikClient, flushAll, track, trackOpikClient, withTrack };
-//# sourceMappingURL=chunk-MCPL44VT.js.map
-//# sourceMappingURL=chunk-MCPL44VT.js.map
+export { OpikClient, flushAll, getTrackContext, track, trackOpikClient };
+//# sourceMappingURL=chunk-ZB4ZJRLY.js.map
+//# sourceMappingURL=chunk-ZB4ZJRLY.js.map