@blaxel/telemetry 0.2.16 → 0.2.17-dev.107

package/dist/telemetry_provider.d.ts

@@ -1,7 +1,5 @@
 import { BlaxelSpan, BlaxelSpanOptions, BlaxelTelemetryProvider } from "@blaxel/core";
 export declare class OtelTelemetryProvider implements BlaxelTelemetryProvider {
-    private spans;
-    retrieveActiveSpanContext(): import("@opentelemetry/api").Context;
     startSpan(name: string, options?: BlaxelSpanOptions): BlaxelSpan;
     flush(): Promise<void>;
 }

package/dist/telemetry_provider.js

@@ -5,7 +5,6 @@ const api_1 = require("@opentelemetry/api");
 const telemetry_1 = require("./telemetry");
 class OtelSpan {
     span;
-    closed = false;
     constructor(span) {
         this.span = span;
     }
@@ -21,12 +20,11 @@ class OtelSpan {
     }
     setStatus(status, message) {
         this.span.setStatus({
-            code: status === 'ok' ? api_1.SpanStatusCode.OK : api_1.SpanStatusCode.ERROR,
+            code: status === "ok" ? api_1.SpanStatusCode.OK : api_1.SpanStatusCode.ERROR,
             message,
         });
     }
     end() {
-        this.closed = true;
         this.span.end();
     }
     getContext() {
@@ -34,26 +32,28 @@ class OtelSpan {
     }
 }
 class OtelTelemetryProvider {
-    spans = [];
-    retrieveActiveSpanContext() {
-        for (let i = this.spans.length - 1; i >= 0; i--) {
-            const span = this.spans[i];
-            if (!span.closed) {
-                return api_1.trace.setSpanContext(api_1.ROOT_CONTEXT, span.getContext());
-            }
-        }
-        return api_1.context.active();
-    }
     startSpan(name, options) {
+        // Use the tracer from the registered NodeTracerProvider
         const tracer = api_1.trace.getTracer("blaxel");
+        // Prepare OpenTelemetry span options
         const otelOptions = {
             attributes: options?.attributes,
             root: options?.isRoot,
         };
-        const ctx = this.retrieveActiveSpanContext();
-        const span = new OtelSpan(tracer.startSpan(name, otelOptions, ctx));
-        this.spans.push(span);
-        return span;
+        // Handle parent context properly
+        let ctx = api_1.context.active();
+        if (options?.parentContext) {
+            // If explicit parent context is provided, use it
+            ctx = api_1.trace.setSpanContext(api_1.ROOT_CONTEXT, options.parentContext);
+        }
+        else if (options?.isRoot) {
+            // If explicitly marked as root, use ROOT_CONTEXT
+            ctx = api_1.ROOT_CONTEXT;
+        }
+        // Otherwise, use the active context (default behavior)
+        // Start the span with proper context
+        const span = tracer.startSpan(name, otelOptions, ctx);
+        return new OtelSpan(span);
     }
     async flush() {
         await telemetry_1.blaxelTelemetry.flush();

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@blaxel/telemetry",
-  "version": "0.2.16",
+  "version": "0.2.17-dev.107",
   "description": "Blaxel SDK for TypeScript",
   "license": "MIT",
   "author": "Blaxel, INC (https://blaxel.ai)",
@@ -71,7 +71,7 @@
     "@opentelemetry/sdk-trace-base": "^2.0.0",
     "@opentelemetry/sdk-trace-node": "^2.0.0",
     "ai": "^4.3.13",
-    "@blaxel/core": "0.2.16"
+    "@blaxel/core": "0.2.17-dev.107"
   },
   "devDependencies": {
     "@eslint/js": "^9.26.0",