@lmnr-ai/lmnr 0.7.11 → 0.7.12

package/dist/cli.js

@@ -27,7 +27,7 @@ var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__ge
 var import_commander = require("commander");
 
 // package.json
-var version = "0.7.11";
+var version = "0.7.12";
 
 // src/utils.ts
 var import_api = require("@opentelemetry/api");
@@ -143,6 +143,125 @@ var import_api3 = require("@opentelemetry/api");
 // src/opentelemetry-lib/tracing/context.ts
 var import_api2 = require("@opentelemetry/api");
 var import_async_hooks = require("async_hooks");
+
+// src/opentelemetry-lib/tracing/compat.ts
+var import_resources = require("@opentelemetry/resources");
+var makeSpanOtelV2Compatible = (span) => {
+  if (span.instrumentationScope && !span.instrumentationLibrary) {
+    Object.assign(span, {
+      instrumentationLibrary: span.instrumentationScope
+    });
+  } else if (span.instrumentationLibrary && !span.instrumentationScope) {
+    Object.assign(span, {
+      instrumentationScope: span.instrumentationLibrary
+    });
+  }
+};
+var getParentSpanId = (span) => span.parentSpanContext?.spanId ?? span.parentSpanId;
+
+// src/opentelemetry-lib/tracing/span.ts
+var LaminarSpan = class {
+  constructor(span, activated) {
+    this._activated = activated ?? false;
+    this._span = span;
+    this.name = this._span.name;
+    this.kind = this._span.kind;
+    this.startTime = this._span.startTime;
+    this.endTime = this._span.endTime;
+    this.status = this._span.status;
+    this.attributes = this._span.attributes;
+    this.links = this._span.links;
+    this.events = this._span.events;
+    this.duration = this._span.duration;
+    this.ended = this._span.ended;
+    this.resource = this._span.resource;
+    this.instrumentationLibrary = this._span.instrumentationLibrary;
+    this.droppedAttributesCount = this._span.droppedAttributesCount;
+    this.droppedEventsCount = this._span.droppedEventsCount;
+    this.droppedLinksCount = this._span.droppedLinksCount;
+    this.makeOtelV2Compatible();
+    LaminarContextManager.addActiveSpan(this.spanContext().spanId);
+  }
+  spanContext() {
+    return this._span.spanContext();
+  }
+  setAttribute(key, value) {
+    this._span.setAttribute(key, value);
+    return this;
+  }
+  setAttributes(attributes) {
+    this._span.setAttributes(attributes);
+    return this;
+  }
+  addEvent(name, attributesOrStartTime, startTime) {
+    this._span.addEvent(name, attributesOrStartTime, startTime);
+    return this;
+  }
+  addLink(link) {
+    this._span.addLink(link);
+    return this;
+  }
+  addLinks(links) {
+    this._span.addLinks(links);
+    return this;
+  }
+  setStatus(status) {
+    this._span.setStatus(status);
+    return this;
+  }
+  updateName(name) {
+    this._span.updateName(name);
+    return this;
+  }
+  end(endTime) {
+    if (!this._span.isRecording()) {
+      return this._span.end(endTime);
+    }
+    LaminarContextManager.removeActiveSpan(this.spanContext().spanId);
+    if (this._activated) {
+      LaminarContextManager.popContext();
+    }
+    return this._span.end(endTime);
+  }
+  set activated(activated) {
+    this._activated = activated;
+  }
+  isRecording() {
+    return this._span.isRecording();
+  }
+  recordException(exception, time) {
+    return this._span.recordException(exception, time);
+  }
+  // ================================
+  // OTel V2 compatibility
+  // ================================
+  makeOtelV2Compatible() {
+    makeSpanOtelV2Compatible(this._span);
+  }
+  get instrumentationScope() {
+    return this._span.instrumentationLibrary;
+  }
+  get parentSpanContext() {
+    const parentSpanId = this._span.parentSpanId;
+    if (!parentSpanId) {
+      return void 0;
+    }
+    return {
+      spanId: parentSpanId,
+      traceId: this.spanContext().traceId,
+      // TODO: somehow get the traceFlags from the parent span
+      traceFlags: this.spanContext().traceFlags
+    };
+  }
+  getParentSpanId() {
+    return getParentSpanId(this._span);
+  }
+  get isActivated() {
+    return this._activated;
+  }
+};
+
+// src/opentelemetry-lib/tracing/context.ts
 var LaminarContextManager = class {
   static getContext() {
     const contexts = this._asyncLocalStorage.getStore() || [];
@@ -155,6 +274,12 @@ var LaminarContextManager = class {
       if (!span.isRecording() && span.spanContext().isRemote) {
         return context;
       }
+      if (!(span instanceof LaminarSpan)) {
+        return context;
+      }
+      if (!span.isActivated) {
+        return context;
+      }
       try {
         const isActive = this._activeSpans.has(span.spanContext().spanId);
         if (isActive) {