opik 0.0.7 → 0.1.0

package/dist/vercel/index.d.cts

@@ -0,0 +1,42 @@
+import { O as OpikClient, b as Span, T as Trace } from '../Client-Y3fv8TIb.cjs';
+import { AttributeValue, Tracer } from '@opentelemetry/api';
+import { NodeSDKConfiguration } from '@opentelemetry/sdk-node';
+import 'stream';
+
+type SpanExporter = NodeSDKConfiguration["traceExporter"];
+type ExportFunction = SpanExporter["export"];
+type ReadableSpan = Parameters<ExportFunction>[0][0];
+type TelemetrySettings = {
+    isEnabled?: boolean;
+    recordInputs?: boolean;
+    recordOutputs?: boolean;
+    functionId?: string;
+    metadata?: Record<string, AttributeValue>;
+    tracer?: Tracer;
+};
+type OpikExporterSettings = TelemetrySettings & {
+    name?: string;
+};
+declare class OpikExporter implements SpanExporter {
+    private traces;
+    private spans;
+    private client;
+    constructor({ client }?: {
+        client?: OpikClient;
+    });
+    private getSpanInput;
+    private getSpanOutput;
+    private getSpanMetadata;
+    private getSpanUsage;
+    processSpan: ({ otelSpan, parentSpan, trace, }: {
+        otelSpan: ReadableSpan;
+        parentSpan?: Span;
+        trace: Trace;
+    }) => Span;
+    shutdown: () => Promise<void>;
+    forceFlush: () => Promise<void>;
+    export: ExportFunction;
+    static getSettings(settings: OpikExporterSettings): TelemetrySettings;
+}
+
+export { OpikExporter };

package/dist/vercel/index.d.ts

@@ -0,0 +1,42 @@
+import { O as OpikClient, b as Span, T as Trace } from '../Client-Y3fv8TIb.js';
+import { AttributeValue, Tracer } from '@opentelemetry/api';
+import { NodeSDKConfiguration } from '@opentelemetry/sdk-node';
+import 'stream';
+
+type SpanExporter = NodeSDKConfiguration["traceExporter"];
+type ExportFunction = SpanExporter["export"];
+type ReadableSpan = Parameters<ExportFunction>[0][0];
+type TelemetrySettings = {
+    isEnabled?: boolean;
+    recordInputs?: boolean;
+    recordOutputs?: boolean;
+    functionId?: string;
+    metadata?: Record<string, AttributeValue>;
+    tracer?: Tracer;
+};
+type OpikExporterSettings = TelemetrySettings & {
+    name?: string;
+};
+declare class OpikExporter implements SpanExporter {
+    private traces;
+    private spans;
+    private client;
+    constructor({ client }?: {
+        client?: OpikClient;
+    });
+    private getSpanInput;
+    private getSpanOutput;
+    private getSpanMetadata;
+    private getSpanUsage;
+    processSpan: ({ otelSpan, parentSpan, trace, }: {
+        otelSpan: ReadableSpan;
+        parentSpan?: Span;
+        trace: Trace;
+    }) => Span;
+    shutdown: () => Promise<void>;
+    forceFlush: () => Promise<void>;
+    export: ExportFunction;
+    static getSettings(settings: OpikExporterSettings): TelemetrySettings;
+}
+
+export { OpikExporter };

package/dist/vercel/index.js

@@ -0,0 +1,207 @@
+import { OpikClient } from '../chunk-MCPL44VT.js';
+import '../chunk-PZ5AY32C.js';
+
+// node_modules/@opentelemetry/core/build/esm/ExportResult.js
+var ExportResultCode;
+(function(ExportResultCode2) {
+  ExportResultCode2[ExportResultCode2["SUCCESS"] = 0] = "SUCCESS";
+  ExportResultCode2[ExportResultCode2["FAILED"] = 1] = "FAILED";
+})(ExportResultCode || (ExportResultCode = {}));
+
+// src/opik/integrations/vercel/exporter.ts
+var aiSDKClient = new OpikClient();
+var OpikExporter = class {
+  constructor({ client = aiSDKClient } = {}) {
+    // <otelTraceId, opikTrace>
+    this.traces = /* @__PURE__ */ new Map();
+    // <otelSpanId, opikSpan>
+    this.spans = /* @__PURE__ */ new Map();
+    this.getSpanInput = (otelSpan) => {
+      let input = {};
+      const { attributes } = otelSpan;
+      const attributeKeys = Object.keys(attributes);
+      attributeKeys.forEach((key) => {
+        if (key === "ai.prompt" || key === "gen_ai.request") {
+          const parsedValue = tryParseJSON(attributes[key]);
+          if (parsedValue) {
+            input = { ...input, ...parsedValue };
+          }
+        }
+        if (key.startsWith("ai.prompt.")) {
+          const promptKey = key.replace("ai.prompt.", "");
+          input[promptKey] = safeParseJson(attributes[key]);
+        }
+        if (key.startsWith("gen_ai.request.")) {
+          const promptKey = key.replace("gen_ai.request.", "");
+          input[promptKey] = safeParseJson(attributes[key]);
+        }
+      });
+      if (Object.keys(input).length > 0) {
+        return input;
+      }
+      if ("ai.toolCall.name" in attributes) {
+        input.toolName = attributes["ai.toolCall.name"];
+      }
+      if ("ai.toolCall.args" in attributes) {
+        input.args = attributes["ai.toolCall.args"];
+      }
+      return input;
+    };
+    this.getSpanOutput = (otelSpan) => {
+      const { attributes } = otelSpan;
+      if (attributes["ai.response.text"]) {
+        return { text: attributes["ai.response.text"] };
+      }
+      if (attributes["ai.toolCall.result"]) {
+        return { result: attributes["ai.toolCall.result"] };
+      }
+      if (attributes["ai.response.toolCalls"]) {
+        return { toolCalls: safeParseJson(attributes["ai.response.toolCalls"]) };
+      }
+      return {};
+    };
+    this.getSpanMetadata = (otelSpan) => {
+      const { attributes } = otelSpan;
+      const metadata = {};
+      if (attributes["gen_ai.response.model"]) {
+        metadata.model = attributes["gen_ai.response.model"];
+      }
+      if (attributes["gen_ai.system"]) {
+        metadata.system = attributes["gen_ai.system"];
+      }
+      return metadata;
+    };
+    this.getSpanUsage = (otelSpan) => {
+      const { attributes } = otelSpan;
+      const usage = {};
+      if ("ai.usage.promptTokens" in attributes) {
+        usage.prompt_tokens = attributes["ai.usage.promptTokens"];
+      }
+      if ("gen_ai.usage.input_tokens" in attributes) {
+        usage.prompt_tokens = attributes["gen_ai.usage.input_tokens"];
+      }
+      if ("ai.usage.completionTokens" in attributes) {
+        usage.completion_tokens = attributes["ai.usage.completionTokens"];
+      }
+      if ("gen_ai.usage.output_tokens" in attributes) {
+        usage.completion_tokens = attributes["gen_ai.usage.output_tokens"];
+      }
+      if ("prompt_tokens" in usage || "completion_tokens" in usage) {
+        usage.total_tokens = (usage.prompt_tokens || 0) + (usage.completion_tokens || 0);
+      }
+      return usage;
+    };
+    this.processSpan = ({
+      otelSpan,
+      parentSpan,
+      trace
+    }) => {
+      return trace.span({
+        name: otelSpan.name,
+        startTime: new Date(hrTimeToMilliseconds(otelSpan.startTime)),
+        endTime: new Date(hrTimeToMilliseconds(otelSpan.endTime)),
+        parentSpanId: parentSpan == null ? void 0 : parentSpan.data.id,
+        input: this.getSpanInput(otelSpan),
+        output: this.getSpanOutput(otelSpan),
+        metadata: this.getSpanMetadata(otelSpan),
+        usage: this.getSpanUsage(otelSpan),
+        type: "llm"
+      });
+    };
+    this.shutdown = async () => {
+      await this.client.flush();
+    };
+    this.forceFlush = async () => {
+      await this.client.flush();
+    };
+    this.export = (otelSpans, resultCallback) => {
+      const spanGroups = groupAndSortOtelSpans(otelSpans);
+      Object.entries(spanGroups).forEach(([otelTraceId, otelSpans2]) => {
+        var _a, _b;
+        const [rootOtelSpan, ...otherOtelSpans] = otelSpans2;
+        const trace = this.client.trace({
+          startTime: new Date(hrTimeToMilliseconds(rootOtelSpan.startTime)),
+          endTime: new Date(hrTimeToMilliseconds(rootOtelSpan.endTime)),
+          name: (_b = (_a = rootOtelSpan.attributes["ai.telemetry.metadata.traceName"]) == null ? void 0 : _a.toString()) != null ? _b : rootOtelSpan.name,
+          input: this.getSpanInput(rootOtelSpan),
+          output: this.getSpanOutput(rootOtelSpan),
+          metadata: this.getSpanMetadata(rootOtelSpan),
+          usage: this.getSpanUsage(rootOtelSpan)
+        });
+        this.traces.set(otelTraceId, trace);
+        otherOtelSpans.forEach((otelSpan) => {
+          var _a2;
+          const parentSpan = this.spans.get((_a2 = otelSpan.parentSpanId) != null ? _a2 : "");
+          const span = this.processSpan({ parentSpan, otelSpan, trace });
+          this.spans.set(otelSpan.spanContext().spanId, span);
+        });
+      });
+      resultCallback({ code: ExportResultCode.SUCCESS });
+    };
+    this.client = client;
+  }
+  static getSettings(settings) {
+    var _a, _b, _c;
+    const metadata = { ...settings.metadata };
+    if (settings.name) {
+      metadata.traceName = settings.name;
+    }
+    return {
+      isEnabled: (_a = settings.isEnabled) != null ? _a : true,
+      recordInputs: (_b = settings.recordInputs) != null ? _b : true,
+      recordOutputs: (_c = settings.recordOutputs) != null ? _c : true,
+      functionId: settings.functionId,
+      metadata
+    };
+  }
+};
+function groupAndSortOtelSpans(otelSpans) {
+  const spanGroupsByTraceId = {};
+  otelSpans.forEach((otelSpan) => {
+    const context = otelSpan.spanContext();
+    if (!spanGroupsByTraceId[context.traceId]) {
+      spanGroupsByTraceId[context.traceId] = [];
+    }
+    spanGroupsByTraceId[context.traceId].push(otelSpan);
+  });
+  Object.values(spanGroupsByTraceId).forEach((otelSpans2) => {
+    otelSpans2.sort((a, b) => {
+      if (!a.parentSpanId && b.parentSpanId) return -1;
+      if (a.parentSpanId && !b.parentSpanId) return 1;
+      if (a.parentSpanId && b.parentSpanId) {
+        if (a.parentSpanId === b.spanContext().spanId) return 1;
+        if (b.parentSpanId === a.spanContext().spanId) return -1;
+      }
+      return 0;
+    });
+  });
+  return spanGroupsByTraceId;
+}
+function hrTimeToMilliseconds(hrTime) {
+  return hrTime[0] * 1e3 + hrTime[1] / 1e6;
+}
+function safeParseJson(value) {
+  try {
+    return JSON.parse(value);
+  } catch (e) {
+    return value;
+  }
+}
+function tryParseJSON(input) {
+  if (typeof input !== "string") {
+    return void 0;
+  }
+  try {
+    const parsed = JSON.parse(input);
+    if (parsed !== null && typeof parsed === "object" && !Array.isArray(parsed)) {
+      return parsed;
+    }
+  } catch (error) {
+    return void 0;
+  }
+  return void 0;
+}
+
+export { OpikExporter };
+//# sourceMappingURL=index.js.map
+//# sourceMappingURL=index.js.map

package/dist/vercel/index.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../node_modules/@opentelemetry/core/src/ExportResult.ts","../../src/opik/integrations/vercel/exporter.ts"],"names":["ExportResultCode","otelSpans","_a"],"mappings":";;;;AAqBA,IAAY,gBAAA;AAAZ,CAAA,SAAYA,iBAAgB,EAAA;AAC1B,EAAAA,iBAAAA,CAAAA,iBAAAA,CAAA,SAAA,CAAA,GAAA,CAAA,CAAA,GAAA,SAAA;AACA,EAAAA,iBAAAA,CAAAA,iBAAAA,CAAA,QAAA,CAAA,GAAA,CAAA,CAAA,GAAA,QAAA;AACF,CAHY,EAAA,gBAAA,KAAA,gBAAgB,GAAA,EAAA,CAAA,CAAA;;;ACE5B,IAAM,WAAA,GAAc,IAAI,UAAK,EAAA;AAEtB,IAAM,eAAN,MAA2C;AAAA,EAOhD,YAAY,EAAE,MAAA,GAAS,WAAY,EAAA,GAAuB,EAAI,EAAA;AAL9D;AAAA,IAAQ,IAAA,CAAA,MAAA,uBAAa,GAAmB,EAAA;AAExC;AAAA,IAAQ,IAAA,CAAA,KAAA,uBAAY,GAAkB,EAAA;AAOtC,IAAQ,IAAA,CAAA,YAAA,GAAe,CAAC,QAAoD,KAAA;AAC1E,MAAA,IAAI,QAAiC,EAAC;AACtC,MAAM,MAAA,EAAE,YAAe,GAAA,QAAA;AACvB,MAAM,MAAA,aAAA,GAAgB,MAAO,CAAA,IAAA,CAAK,UAAU,CAAA;AAE5C,MAAc,aAAA,CAAA,OAAA,CAAQ,CAAC,GAAQ,KAAA;AAC7B,QAAI,IAAA,GAAA,KAAQ,WAAe,IAAA,GAAA,KAAQ,gBAAkB,EAAA;AACnD,UAAA,MAAM,WAAc,GAAA,YAAA,CAAa,UAAW,CAAA,GAAG,CAAC,CAAA;AAEhD,UAAA,IAAI,WAAa,EAAA;AACf,YAAA,KAAA,GAAQ,EAAE,GAAG,KAAO,EAAA,GAAG,WAAY,EAAA;AAAA;AACrC;AAGF,QAAI,IAAA,GAAA,CAAI,UAAW,CAAA,YAAY,CAAG,EAAA;AAChC,UAAA,MAAM,SAAY,GAAA,GAAA,CAAI,OAAQ,CAAA,YAAA,EAAc,EAAE,CAAA;AAE9C,UAAA,KAAA,CAAM,SAAS,CAAA,GAAI,aAAc,CAAA,UAAA,CAAW,GAAG,CAAC,CAAA;AAAA;AAGlD,QAAI,IAAA,GAAA,CAAI,UAAW,CAAA,iBAAiB,CAAG,EAAA;AACrC,UAAA,MAAM,SAAY,GAAA,GAAA,CAAI,OAAQ,CAAA,iBAAA,EAAmB,EAAE,CAAA;AAEnD,UAAA,KAAA,CAAM,SAAS,CAAA,GAAI,aAAc,CAAA,UAAA,CAAW,GAAG,CAAC,CAAA;AAAA;AAClD,OACD,CAAA;AAED,MAAA,IAAI,MAAO,CAAA,IAAA,CAAK,KAAK,CAAA,CAAE,SAAS,CAAG,EAAA;AACjC,QAAO,OAAA,KAAA;AAAA;AAGT,MAAA,IAAI,sBAAsB,UAAY,EAAA;AACpC,QAAM,KAAA,CAAA,QAAA,GAAW,WAAW,kBAAkB,CAAA;AAAA;AAGhD,MAAA,IAAI,sBAAsB,UAAY,EAAA;AACpC,QAAM,KAAA,CAAA,IAAA,GAAO,WAAW,kBAAkB,CAAA;AAAA;AAG5C,MAAO,OAAA,KAAA;AAAA,KACT;AAEA,IAAQ,IAAA,CAAA,aAAA,GAAgB,CAAC,QAAoD,KAAA;AAC3E,MAAM,MAAA,EAAE,YAAe,GAAA,QAAA;AAEvB,MAAI,IAAA,UAAA,CAAW,kBAAkB,CAAG,EAAA;AAClC,QAAA,OAAO,EAAE,IAAA,EAAM,UAAW,CAAA,kBAAkB,CAAE,EAAA;AAAA;AAGhD,MAAI,IAAA,UAAA,CAAW,oBAAoB,CAAG,EAAA;AACpC,QAAA,OAAO,EAAE,MAAA,EAAQ,UAAW,CAAA,oBAAoB,CAAE,EAAA;AAAA;AAGpD,MAAI,IAAA,UAAA,CAAW,uBAAuB,CAAG,EAAA;AACvC,QAAA,OAAO,EAAE,SAAW,EAAA,aAAA,CAAc,UAAW,CAAA,uBAAuB,CAAC,CAAE,EAAA;AAAA;AAGzE,MAAA,OAAO,EAAC;AAAA,KACV;AAEA,IAAQ,IAAA,CAAA,eAAA,GAAkB,CACxB,QAC4B,KAAA;AAC5B,MAAM,MAAA,EAAE,YAAe,GAAA,QAAA;AACvB,MAAA,MAAM,WAAoC,EAAC;AAE3C,MAAI,IAAA,UAAA,CAAW,uBAAuB,CAAG,EAAA;AACvC,QAAS,QAAA,CAAA,KAAA,GAAQ,WAAW,uBAAuB,CAAA;AAAA;AAGrD,MAAI,IAAA,UAAA,CAAW,eAAe,CAAG,EAAA;AAC/B,QAAS,QAAA,CAAA,MAAA,GAAS,WAAW,eAAe,CAAA;AAAA;AAG9C,MAAO,OAAA,QAAA;AAAA,KACT;AAEA,IAAQ,IAAA,CAAA,YAAA,GAAe,CAAC,QAAmD,KAAA;AACzE,MAAM,MAAA,EAAE,YAAe,GAAA,QAAA;AACvB,MAAA,MAAM,QAAgC,EAAC;AAGvC,MAAA,IAAI,2BAA2B,UAAY,EAAA;AACzC,QAAM,KAAA,CAAA,aAAA,GAAgB,WAAW,uBAAuB,CAAA;AAAA;AAE1D,MAAA,IAAI,+BAA+B,UAAY,EAAA;AAC7C,QAAM,KAAA,CAAA,aAAA,GAAgB,WAAW,2BAA2B,CAAA;AAAA;AAI9D,MAAA,IAAI,+BAA+B,UAAY,EAAA;AAC7C,QAAM,KAAA,CAAA,iBAAA,GAAoB,WACxB,2BACF,CAAA;AAAA;AAEF,MAAA,IAAI,gCAAgC,UAAY,EAAA;AAC9C,QAAM,KAAA,CAAA,iBAAA,GAAoB,WACxB,4BACF,CAAA;AAAA;AAGF,MAAI,IAAA,eAAA,IAAmB,KAAS,IAAA,mBAAA,IAAuB,KAAO,EAAA;AAC5D,QAAA,KAAA,CAAM,YACH,GAAA,CAAA,KAAA,CAAM,aAAiB,IAAA,CAAA,KAAM,MAAM,iBAAqB,IAAA,CAAA,CAAA;AAAA;AAG7D,MAAO,OAAA,KAAA;AAAA,KACT;AAEA,IAAA,IAAA,CAAA,WAAA,GAAc,CAAC;AAAA,MACb,QAAA;AAAA,MACA,UAAA;AAAA,MACA;AAAA,KAKU,KAAA;AACV,MAAA,OAAO,MAAM,IAAK,CAAA;AAAA,QAChB,MAAM,QAAS,CAAA,IAAA;AAAA,QACf,WAAW,IAAI,IAAA,CAAK,oBAAqB,CAAA,QAAA,CAAS,SAAS,CAAC,CAAA;AAAA,QAC5D,SAAS,IAAI,IAAA,CAAK,oBAAqB,CAAA,QAAA,CAAS,OAAO,CAAC,CAAA;AAAA,QACxD,YAAA,EAAc,yCAAY,IAAK,CAAA,EAAA;AAAA,QAC/B,KAAA,EAAO,IAAK,CAAA,YAAA,CAAa,QAAQ,CAAA;AAAA,QACjC,MAAA,EAAQ,IAAK,CAAA,aAAA,CAAc,QAAQ,CAAA;AAAA,QACnC,QAAA,EAAU,IAAK,CAAA,eAAA,CAAgB,QAAQ,CAAA;AAAA,QACvC,KAAA,EAAO,IAAK,CAAA,YAAA,CAAa,QAAQ,CAAA;AAAA,QACjC,IAAM,EAAA;AAAA,OACP,CAAA;AAAA,KACH;AAEA,IAAA,IAAA,CAAA,QAAA,GAAW,YAA2B;AACpC,MAAM,MAAA,IAAA,CAAK,OAAO,KAAM,EAAA;AAAA,KAC1B;AAEA,IAAA,IAAA,CAAA,UAAA,GAAa,YAA2B;AACtC,MAAM,MAAA,IAAA,CAAK,OAAO,KAAM,EAAA;AAAA,KAC1B;AAEA,IAAyB,IAAA,CAAA,MAAA,GAAA,CAAC,WAAW,cAAmB,KAAA;AACtD,MAAM,MAAA,UAAA,GAAa,sBAAsB,SAAS,CAAA;AAElD,MAAO,MAAA,CAAA,OAAA,CAAQ,UAAU,CAAE,CAAA,OAAA,CAAQ,CAAC,CAAC,WAAA,EAAaC,UAAS,CAAM,KAAA;AAlLrE,QAAA,IAAA,EAAA,EAAA,EAAA;AAmLM,QAAA,MAAM,CAAC,YAAA,EAAc,GAAG,cAAc,CAAIA,GAAAA,UAAAA;AAC1C,QAAM,MAAA,KAAA,GAAQ,IAAK,CAAA,MAAA,CAAO,KAAM,CAAA;AAAA,UAC9B,WAAW,IAAI,IAAA,CAAK,oBAAqB,CAAA,YAAA,CAAa,SAAS,CAAC,CAAA;AAAA,UAChE,SAAS,IAAI,IAAA,CAAK,oBAAqB,CAAA,YAAA,CAAa,OAAO,CAAC,CAAA;AAAA,UAC5D,IAAA,EAAA,CACE,wBAAa,UACX,CAAA,iCACF,MAFA,IAEG,GAAA,KAAA,CAAA,GAAA,EAAA,CAAA,QAAA,EAAA,KAFH,YAEiB,YAAa,CAAA,IAAA;AAAA,UAChC,KAAA,EAAO,IAAK,CAAA,YAAA,CAAa,YAAY,CAAA;AAAA,UACrC,MAAA,EAAQ,IAAK,CAAA,aAAA,CAAc,YAAY,CAAA;AAAA,UACvC,QAAA,EAAU,IAAK,CAAA,eAAA,CAAgB,YAAY,CAAA;AAAA,UAC3C,KAAA,EAAO,IAAK,CAAA,YAAA,CAAa,YAAY;AAAA,SACtC,CAAA;AAED,QAAK,IAAA,CAAA,MAAA,CAAO,GAAI,CAAA,WAAA,EAAa,KAAK,CAAA;AAElC,QAAe,cAAA,CAAA,OAAA,CAAQ,CAAC,QAAa,KAAA;AAnM3C,UAAAC,IAAAA,GAAAA;AAoMQ,UAAM,MAAA,UAAA,GAAa,KAAK,KAAM,CAAA,GAAA,CAAA,CAAIA,MAAA,QAAS,CAAA,YAAA,KAAT,IAAAA,GAAAA,GAAAA,GAAyB,EAAE,CAAA;AAC7D,UAAA,MAAM,OAAO,IAAK,CAAA,WAAA,CAAY,EAAE,UAAY,EAAA,QAAA,EAAU,OAAO,CAAA;AAE7D,UAAA,IAAA,CAAK,MAAM,GAAI,CAAA,QAAA,CAAS,WAAY,EAAA,CAAE,QAAQ,IAAI,CAAA;AAAA,SACnD,CAAA;AAAA,OACF,CAAA;AAED,MAAA,cAAA,CAAe,EAAE,IAAA,EAAM,gBAAiB,CAAA,OAAA,EAAS,CAAA;AAAA,KACnD;AA3KE,IAAA,IAAA,CAAK,MAAS,GAAA,MAAA;AAAA;AAChB,EA4KA,OAAO,YAAY,QAAmD,EAAA;AA9MxE,IAAA,IAAA,EAAA,EAAA,EAAA,EAAA,EAAA;AA+MI,IAAA,MAAM,QAAW,GAAA,EAAE,GAAG,QAAA,CAAS,QAAS,EAAA;AAExC,IAAA,IAAI,SAAS,IAAM,EAAA;AACjB,MAAA,QAAA,CAAS,YAAY,QAAS,CAAA,IAAA;AAAA;AAGhC,IAAO,OAAA;AAAA,MACL,SAAA,EAAA,CAAW,EAAS,GAAA,QAAA,CAAA,SAAA,KAAT,IAAsB,GAAA,EAAA,GAAA,IAAA;AAAA,MACjC,YAAA,EAAA,CAAc,EAAS,GAAA,QAAA,CAAA,YAAA,KAAT,IAAyB,GAAA,EAAA,GAAA,IAAA;AAAA,MACvC,aAAA,EAAA,CAAe,EAAS,GAAA,QAAA,CAAA,aAAA,KAAT,IAA0B,GAAA,EAAA,GAAA,IAAA;AAAA,MACzC,YAAY,QAAS,CAAA,UAAA;AAAA,MACrB;AAAA,KACF;AAAA;AAEJ;AAEA,SAAS,sBACP,SACgC,EAAA;AAChC,EAAA,MAAM,sBAAsD,EAAC;AAE7D,EAAU,SAAA,CAAA,OAAA,CAAQ,CAAC,QAAa,KAAA;AAC9B,IAAM,MAAA,OAAA,GAAU,SAAS,WAAY,EAAA;AAErC,IAAA,IAAI,CAAC,mBAAA,CAAoB,OAAQ,CAAA,OAAO,CAAG,EAAA;AACzC,MAAoB,mBAAA,CAAA,OAAA,CAAQ,OAAO,CAAA,GAAI,EAAC;AAAA;AAG1C,IAAA,mBAAA,CAAoB,OAAQ,CAAA,OAAO,CAAE,CAAA,IAAA,CAAK,QAAQ,CAAA;AAAA,GACnD,CAAA;AAED,EAAA,MAAA,CAAO,MAAO,CAAA,mBAAmB,CAAE,CAAA,OAAA,CAAQ,CAACD,UAAc,KAAA;AACxD,IAAAA,UAAU,CAAA,IAAA,CAAK,CAAC,CAAA,EAAG,CAAM,KAAA;AAEvB,MAAA,IAAI,CAAC,CAAA,CAAE,YAAgB,IAAA,CAAA,CAAE,cAAqB,OAAA,CAAA,CAAA;AAC9C,MAAA,IAAI,CAAE,CAAA,YAAA,IAAgB,CAAC,CAAA,CAAE,cAAqB,OAAA,CAAA;AAG9C,MAAI,IAAA,CAAA,CAAE,YAAgB,IAAA,CAAA,CAAE,YAAc,EAAA;AACpC,QAAA,IAAI,EAAE,YAAiB,KAAA,CAAA,CAAE,WAAY,EAAA,CAAE,QAAe,OAAA,CAAA;AACtD,QAAA,IAAI,EAAE,YAAiB,KAAA,CAAA,CAAE,WAAY,EAAA,CAAE,QAAe,OAAA,CAAA,CAAA;AAAA;AAIxD,MAAO,OAAA,CAAA;AAAA,KACR,CAAA;AAAA,GACF,CAAA;AAED,EAAO,OAAA,mBAAA;AACT;AAGA,SAAS,qBAAqB,MAA0B,EAAA;AACtD,EAAA,OAAO,OAAO,CAAC,CAAA,GAAI,GAAM,GAAA,MAAA,CAAO,CAAC,CAAI,GAAA,GAAA;AACvC;AAEA,SAAS,cAAc,KAAyB,EAAA;AAC9C,EAAI,IAAA;AACF,IAAO,OAAA,IAAA,CAAK,MAAM,KAAe,CAAA;AAAA,WAC1B,CAAG,EAAA;AACV,IAAO,OAAA,KAAA;AAAA;AAEX;AAEA,SAAS,aAAa,KAAqD,EAAA;AACzE,EAAI,IAAA,OAAO,UAAU,QAAU,EAAA;AAC7B,IAAO,OAAA,KAAA,CAAA;AAAA;AAGT,EAAI,IAAA;AACF,IAAM,MAAA,MAAA,GAAS,IAAK,CAAA,KAAA,CAAM,KAAK,CAAA;AAE/B,IACE,IAAA,MAAA,KAAW,QACX,OAAO,MAAA,KAAW,YAClB,CAAC,KAAA,CAAM,OAAQ,CAAA,MAAM,CACrB,EAAA;AACA,MAAO,OAAA,MAAA;AAAA;AACT,WACO,KAAO,EAAA;AACd,IAAO,OAAA,KAAA,CAAA;AAAA;AAGT,EAAO,OAAA,KAAA,CAAA;AACT","file":"index.js","sourcesContent":["/*\n * Copyright The OpenTelemetry Authors\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *      https://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nexport interface ExportResult {\n  code: ExportResultCode;\n  error?: Error;\n}\n\nexport enum ExportResultCode {\n  SUCCESS,\n  FAILED,\n}\n","import { Opik } from \"opik\";\nimport type { Span, Trace } from \"opik\";\nimport { AttributeValue, Tracer } from \"@opentelemetry/api\";\nimport { ExportResultCode } from \"@opentelemetry/core\";\nimport { NodeSDKConfiguration } from \"@opentelemetry/sdk-node\";\n\ntype SpanExporter = NodeSDKConfiguration[\"traceExporter\"];\ntype ExportFunction = SpanExporter[\"export\"];\ntype ReadableSpan = Parameters<ExportFunction>[0][0];\n\ntype TelemetrySettings = {\n  isEnabled?: boolean;\n  recordInputs?: boolean;\n  recordOutputs?: boolean;\n  functionId?: string;\n  metadata?: Record<string, AttributeValue>;\n  tracer?: Tracer;\n};\n\ntype OpikExporterSettings = TelemetrySettings & {\n  name?: string;\n};\n\nconst aiSDKClient = new Opik();\n\nexport class OpikExporter implements SpanExporter {\n  // <otelTraceId, opikTrace>\n  private traces = new Map<string, Trace>();\n  // <otelSpanId, opikSpan>\n  private spans = new Map<string, Span>();\n  private client: Opik;\n\n  constructor({ client = aiSDKClient }: { client?: Opik } = {}) {\n    this.client = client;\n  }\n\n  private getSpanInput = (otelSpan: ReadableSpan): Record<string, unknown> => {\n    let input: Record<string, unknown> = {};\n    const { attributes } = otelSpan;\n    const attributeKeys = Object.keys(attributes);\n\n    attributeKeys.forEach((key) => {\n      if (key === \"ai.prompt\" || key === \"gen_ai.request\") {\n        const parsedValue = tryParseJSON(attributes[key]);\n\n        if (parsedValue) {\n          input = { ...input, ...parsedValue };\n        }\n      }\n\n      if (key.startsWith(\"ai.prompt.\")) {\n        const promptKey = key.replace(\"ai.prompt.\", \"\");\n\n        input[promptKey] = safeParseJson(attributes[key]);\n      }\n\n      if (key.startsWith(\"gen_ai.request.\")) {\n        const promptKey = key.replace(\"gen_ai.request.\", \"\");\n\n        input[promptKey] = safeParseJson(attributes[key]);\n      }\n    });\n\n    if (Object.keys(input).length > 0) {\n      return input;\n    }\n\n    if (\"ai.toolCall.name\" in attributes) {\n      input.toolName = attributes[\"ai.toolCall.name\"];\n    }\n\n    if (\"ai.toolCall.args\" in attributes) {\n      input.args = attributes[\"ai.toolCall.args\"];\n    }\n\n    return input;\n  };\n\n  private getSpanOutput = (otelSpan: ReadableSpan): Record<string, unknown> => {\n    const { attributes } = otelSpan;\n\n    if (attributes[\"ai.response.text\"]) {\n      return { text: attributes[\"ai.response.text\"] };\n    }\n\n    if (attributes[\"ai.toolCall.result\"]) {\n      return { result: attributes[\"ai.toolCall.result\"] };\n    }\n\n    if (attributes[\"ai.response.toolCalls\"]) {\n      return { toolCalls: safeParseJson(attributes[\"ai.response.toolCalls\"]) };\n    }\n\n    return {};\n  };\n\n  private getSpanMetadata = (\n    otelSpan: ReadableSpan\n  ): Record<string, unknown> => {\n    const { attributes } = otelSpan;\n    const metadata: Record<string, unknown> = {};\n\n    if (attributes[\"gen_ai.response.model\"]) {\n      metadata.model = attributes[\"gen_ai.response.model\"];\n    }\n\n    if (attributes[\"gen_ai.system\"]) {\n      metadata.system = attributes[\"gen_ai.system\"];\n    }\n\n    return metadata;\n  };\n\n  private getSpanUsage = (otelSpan: ReadableSpan): Record<string, number> => {\n    const { attributes } = otelSpan;\n    const usage: Record<string, number> = {};\n\n    // prompt tokens\n    if (\"ai.usage.promptTokens\" in attributes) {\n      usage.prompt_tokens = attributes[\"ai.usage.promptTokens\"] as number;\n    }\n    if (\"gen_ai.usage.input_tokens\" in attributes) {\n      usage.prompt_tokens = attributes[\"gen_ai.usage.input_tokens\"] as number;\n    }\n\n    // completion tokens\n    if (\"ai.usage.completionTokens\" in attributes) {\n      usage.completion_tokens = attributes[\n        \"ai.usage.completionTokens\"\n      ] as number;\n    }\n    if (\"gen_ai.usage.output_tokens\" in attributes) {\n      usage.completion_tokens = attributes[\n        \"gen_ai.usage.output_tokens\"\n      ] as number;\n    }\n\n    if (\"prompt_tokens\" in usage || \"completion_tokens\" in usage) {\n      usage.total_tokens =\n        (usage.prompt_tokens || 0) + (usage.completion_tokens || 0);\n    }\n\n    return usage;\n  };\n\n  processSpan = ({\n    otelSpan,\n    parentSpan,\n    trace,\n  }: {\n    otelSpan: ReadableSpan;\n    parentSpan?: Span;\n    trace: Trace;\n  }): Span => {\n    return trace.span({\n      name: otelSpan.name,\n      startTime: new Date(hrTimeToMilliseconds(otelSpan.startTime)),\n      endTime: new Date(hrTimeToMilliseconds(otelSpan.endTime)),\n      parentSpanId: parentSpan?.data.id,\n      input: this.getSpanInput(otelSpan),\n      output: this.getSpanOutput(otelSpan),\n      metadata: this.getSpanMetadata(otelSpan),\n      usage: this.getSpanUsage(otelSpan),\n      type: \"llm\",\n    });\n  };\n\n  shutdown = async (): Promise<void> => {\n    await this.client.flush();\n  };\n\n  forceFlush = async (): Promise<void> => {\n    await this.client.flush();\n  };\n\n  export: ExportFunction = (otelSpans, resultCallback) => {\n    const spanGroups = groupAndSortOtelSpans(otelSpans);\n\n    Object.entries(spanGroups).forEach(([otelTraceId, otelSpans]) => {\n      const [rootOtelSpan, ...otherOtelSpans] = otelSpans;\n      const trace = this.client.trace({\n        startTime: new Date(hrTimeToMilliseconds(rootOtelSpan.startTime)),\n        endTime: new Date(hrTimeToMilliseconds(rootOtelSpan.endTime)),\n        name:\n          rootOtelSpan.attributes[\n            \"ai.telemetry.metadata.traceName\"\n          ]?.toString() ?? rootOtelSpan.name,\n        input: this.getSpanInput(rootOtelSpan),\n        output: this.getSpanOutput(rootOtelSpan),\n        metadata: this.getSpanMetadata(rootOtelSpan),\n        usage: this.getSpanUsage(rootOtelSpan),\n      });\n\n      this.traces.set(otelTraceId, trace);\n\n      otherOtelSpans.forEach((otelSpan) => {\n        const parentSpan = this.spans.get(otelSpan.parentSpanId ?? \"\");\n        const span = this.processSpan({ parentSpan, otelSpan, trace });\n\n        this.spans.set(otelSpan.spanContext().spanId, span);\n      });\n    });\n\n    resultCallback({ code: ExportResultCode.SUCCESS });\n  };\n\n  static getSettings(settings: OpikExporterSettings): TelemetrySettings {\n    const metadata = { ...settings.metadata };\n\n    if (settings.name) {\n      metadata.traceName = settings.name;\n    }\n\n    return {\n      isEnabled: settings.isEnabled ?? true,\n      recordInputs: settings.recordInputs ?? true,\n      recordOutputs: settings.recordOutputs ?? true,\n      functionId: settings.functionId,\n      metadata,\n    };\n  }\n}\n\nfunction groupAndSortOtelSpans(\n  otelSpans: ReadableSpan[]\n): Record<string, ReadableSpan[]> {\n  const spanGroupsByTraceId: Record<string, ReadableSpan[]> = {};\n\n  otelSpans.forEach((otelSpan) => {\n    const context = otelSpan.spanContext();\n\n    if (!spanGroupsByTraceId[context.traceId]) {\n      spanGroupsByTraceId[context.traceId] = [];\n    }\n\n    spanGroupsByTraceId[context.traceId].push(otelSpan);\n  });\n\n  Object.values(spanGroupsByTraceId).forEach((otelSpans) => {\n    otelSpans.sort((a, b) => {\n      // Put spans without parent first\n      if (!a.parentSpanId && b.parentSpanId) return -1;\n      if (a.parentSpanId && !b.parentSpanId) return 1;\n\n      // If both have parents, check if one is parent of the other\n      if (a.parentSpanId && b.parentSpanId) {\n        if (a.parentSpanId === b.spanContext().spanId) return 1;\n        if (b.parentSpanId === a.spanContext().spanId) return -1;\n      }\n\n      // If no parent relationship, maintain relative order\n      return 0;\n    });\n  });\n\n  return spanGroupsByTraceId;\n}\n\n// Convert hrTime ([seconds, nanoseconds]) to milliseconds\nfunction hrTimeToMilliseconds(hrTime: [number, number]) {\n  return hrTime[0] * 1e3 + hrTime[1] / 1e6;\n}\n\nfunction safeParseJson(value: unknown): unknown {\n  try {\n    return JSON.parse(value as string) as Record<string, unknown>;\n  } catch (e) {\n    return value;\n  }\n}\n\nfunction tryParseJSON(input: unknown): Record<string, unknown> | undefined {\n  if (typeof input !== \"string\") {\n    return undefined;\n  }\n\n  try {\n    const parsed = JSON.parse(input);\n\n    if (\n      parsed !== null &&\n      typeof parsed === \"object\" &&\n      !Array.isArray(parsed)\n    ) {\n      return parsed as Record<string, unknown>;\n    }\n  } catch (error) {\n    return undefined;\n  }\n\n  return undefined;\n}\n"]}

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "opik",
   "description": "Opik TypeScript and JavaScript SDK",
-  "version": "0.0.7",
+  "version": "0.1.0",
   "repository": {
     "type": "git",
     "url": "git+https://github.com/comet-ml/opik.git",
@@ -37,6 +37,11 @@
       "types": "./dist/index.d.ts",
       "import": "./dist/index.js",
       "require": "./dist/index.cjs"
+    },
+    "./vercel": {
+      "types": "./dist/vercel/index.d.ts",
+      "import": "./dist/vercel/index.js",
+      "require": "./dist/vercel/index.cjs"
     }
   },
   "main": "dist/index.cjs",
@@ -51,18 +56,24 @@
     "test": "vitest"
   },
   "devDependencies": {
+    "@ai-sdk/openai": "^1.1.9",
+    "@opentelemetry/api": "^1.9.0",
+    "@opentelemetry/auto-instrumentations-node": "^0.56.0",
+    "@opentelemetry/sdk-node": "^0.57.1",
     "@types/node-fetch": "^2.6.12",
     "@types/qs": "^6.9.17",
     "@types/url-join": "^4.0.3",
     "@types/uuid": "^10.0.0",
     "@typescript-eslint/eslint-plugin": "^8.16.0",
     "@typescript-eslint/parser": "^8.16.0",
+    "ai": "^4.1.34",
     "eslint": "^9.15.0",
     "prettier": "^3.4.1",
     "tsup": "^8.3.5",
     "typescript": "^5.7.2",
     "vite-tsconfig-paths": "^5.1.3",
-    "vitest": "^2.1.6"
+    "vitest": "^3.0.5",
+    "zod": "^3.24.1"
   },
   "engines": {
     "node": ">=18"

package/dist/Node18UniversalStreamWrapper-RYU62PI4.js

@@ -1,220 +0,0 @@
-import "./chunk-MLKGABMK.js";
-
-// src/opik/rest_api/core/fetcher/stream-wrappers/Node18UniversalStreamWrapper.ts
-var Node18UniversalStreamWrapper = class _Node18UniversalStreamWrapper {
-  readableStream;
-  reader;
-  events;
-  paused;
-  resumeCallback;
-  encoding;
-  constructor(readableStream) {
-    this.readableStream = readableStream;
-    this.reader = this.readableStream.getReader();
-    this.events = {
-      data: [],
-      end: [],
-      error: [],
-      readable: [],
-      close: [],
-      pause: [],
-      resume: []
-    };
-    this.paused = false;
-    this.resumeCallback = null;
-    this.encoding = null;
-  }
-  on(event, callback) {
-    this.events[event]?.push(callback);
-  }
-  off(event, callback) {
-    this.events[event] = this.events[event]?.filter((cb) => cb !== callback);
-  }
-  pipe(dest) {
-    this.on("data", async (chunk) => {
-      if (dest instanceof _Node18UniversalStreamWrapper) {
-        dest._write(chunk);
-      } else if (dest instanceof WritableStream) {
-        const writer = dest.getWriter();
-        writer.write(chunk).then(() => writer.releaseLock());
-      } else {
-        dest.write(chunk);
-      }
-    });
-    this.on("end", async () => {
-      if (dest instanceof _Node18UniversalStreamWrapper) {
-        dest._end();
-      } else if (dest instanceof WritableStream) {
-        const writer = dest.getWriter();
-        writer.close();
-      } else {
-        dest.end();
-      }
-    });
-    this.on("error", async (error) => {
-      if (dest instanceof _Node18UniversalStreamWrapper) {
-        dest._error(error);
-      } else if (dest instanceof WritableStream) {
-        const writer = dest.getWriter();
-        writer.abort(error);
-      } else {
-        dest.destroy(error);
-      }
-    });
-    this._startReading();
-    return dest;
-  }
-  pipeTo(dest) {
-    return this.pipe(dest);
-  }
-  unpipe(dest) {
-    this.off("data", async (chunk) => {
-      if (dest instanceof _Node18UniversalStreamWrapper) {
-        dest._write(chunk);
-      } else if (dest instanceof WritableStream) {
-        const writer = dest.getWriter();
-        writer.write(chunk).then(() => writer.releaseLock());
-      } else {
-        dest.write(chunk);
-      }
-    });
-    this.off("end", async () => {
-      if (dest instanceof _Node18UniversalStreamWrapper) {
-        dest._end();
-      } else if (dest instanceof WritableStream) {
-        const writer = dest.getWriter();
-        writer.close();
-      } else {
-        dest.end();
-      }
-    });
-    this.off("error", async (error) => {
-      if (dest instanceof _Node18UniversalStreamWrapper) {
-        dest._error(error);
-      } else if (dest instanceof WritableStream) {
-        const writer = dest.getWriter();
-        writer.abort(error);
-      } else {
-        dest.destroy(error);
-      }
-    });
-  }
-  destroy(error) {
-    this.reader.cancel(error).then(() => {
-      this._emit("close");
-    }).catch((err) => {
-      this._emit("error", err);
-    });
-  }
-  pause() {
-    this.paused = true;
-    this._emit("pause");
-  }
-  resume() {
-    if (this.paused) {
-      this.paused = false;
-      this._emit("resume");
-      if (this.resumeCallback) {
-        this.resumeCallback();
-        this.resumeCallback = null;
-      }
-    }
-  }
-  get isPaused() {
-    return this.paused;
-  }
-  async read() {
-    if (this.paused) {
-      await new Promise((resolve) => {
-        this.resumeCallback = resolve;
-      });
-    }
-    const { done, value } = await this.reader.read();
-    if (done) {
-      return void 0;
-    }
-    return value;
-  }
-  setEncoding(encoding) {
-    this.encoding = encoding;
-  }
-  async text() {
-    const chunks = [];
-    while (true) {
-      const { done, value } = await this.reader.read();
-      if (done) {
-        break;
-      }
-      if (value) {
-        chunks.push(value);
-      }
-    }
-    const decoder = new TextDecoder(this.encoding || "utf-8");
-    return decoder.decode(await new Blob(chunks).arrayBuffer());
-  }
-  async json() {
-    const text = await this.text();
-    return JSON.parse(text);
-  }
-  _write(chunk) {
-    this._emit("data", chunk);
-  }
-  _end() {
-    this._emit("end");
-  }
-  _error(error) {
-    this._emit("error", error);
-  }
-  _emit(event, data) {
-    if (this.events[event]) {
-      for (const callback of this.events[event] || []) {
-        callback(data);
-      }
-    }
-  }
-  async _startReading() {
-    try {
-      this._emit("readable");
-      while (true) {
-        if (this.paused) {
-          await new Promise((resolve) => {
-            this.resumeCallback = resolve;
-          });
-        }
-        const { done, value } = await this.reader.read();
-        if (done) {
-          this._emit("end");
-          this._emit("close");
-          break;
-        }
-        if (value) {
-          this._emit("data", value);
-        }
-      }
-    } catch (error) {
-      this._emit("error", error);
-    }
-  }
-  [Symbol.asyncIterator]() {
-    return {
-      next: async () => {
-        if (this.paused) {
-          await new Promise((resolve) => {
-            this.resumeCallback = resolve;
-          });
-        }
-        const { done, value } = await this.reader.read();
-        if (done) {
-          return { done: true, value: void 0 };
-        }
-        return { done: false, value };
-      },
-      [Symbol.asyncIterator]() {
-        return this;
-      }
-    };
-  }
-};
-export {
-  Node18UniversalStreamWrapper
-};

package/dist/NodePre18StreamWrapper-U7T7ZEM6.js

@@ -1,90 +0,0 @@
-import "./chunk-MLKGABMK.js";
-
-// src/opik/rest_api/core/fetcher/stream-wrappers/NodePre18StreamWrapper.ts
-var NodePre18StreamWrapper = class {
-  readableStream;
-  encoding;
-  constructor(readableStream) {
-    this.readableStream = readableStream;
-  }
-  on(event, callback) {
-    this.readableStream.on(event, callback);
-  }
-  off(event, callback) {
-    this.readableStream.off(event, callback);
-  }
-  pipe(dest) {
-    this.readableStream.pipe(dest);
-    return dest;
-  }
-  pipeTo(dest) {
-    return this.pipe(dest);
-  }
-  unpipe(dest) {
-    if (dest) {
-      this.readableStream.unpipe(dest);
-    } else {
-      this.readableStream.unpipe();
-    }
-  }
-  destroy(error) {
-    this.readableStream.destroy(error);
-  }
-  pause() {
-    this.readableStream.pause();
-  }
-  resume() {
-    this.readableStream.resume();
-  }
-  get isPaused() {
-    return this.readableStream.isPaused();
-  }
-  async read() {
-    return new Promise((resolve, reject) => {
-      const chunk = this.readableStream.read();
-      if (chunk) {
-        resolve(chunk);
-      } else {
-        this.readableStream.once("readable", () => {
-          const chunk2 = this.readableStream.read();
-          resolve(chunk2);
-        });
-        this.readableStream.once("error", reject);
-      }
-    });
-  }
-  setEncoding(encoding) {
-    this.readableStream.setEncoding(encoding);
-    this.encoding = encoding;
-  }
-  async text() {
-    const chunks = [];
-    const encoder = new TextEncoder();
-    this.readableStream.setEncoding(this.encoding || "utf-8");
-    for await (const chunk of this.readableStream) {
-      chunks.push(encoder.encode(chunk));
-    }
-    const decoder = new TextDecoder(this.encoding || "utf-8");
-    return decoder.decode(Buffer.concat(chunks));
-  }
-  async json() {
-    const text = await this.text();
-    return JSON.parse(text);
-  }
-  [Symbol.asyncIterator]() {
-    const readableStream = this.readableStream;
-    const iterator = readableStream[Symbol.asyncIterator]();
-    return {
-      async next() {
-        const { value, done } = await iterator.next();
-        return { value, done };
-      },
-      [Symbol.asyncIterator]() {
-        return this;
-      }
-    };
-  }
-};
-export {
-  NodePre18StreamWrapper
-};