@blaxel/telemetry 0.2.18-dev.157 → 0.2.18

package/dist/telemetry.js

@@ -195,7 +195,6 @@ class TelemetryManager {
         core_1.telemetryRegistry.registerProvider(new telemetry_provider_1.OtelTelemetryProvider());
         const httpInstrumentation = new instrumentation_http_1.HttpInstrumentation({
             requireParentforOutgoingSpans: true,
-            requireParentforIncomingSpans: true,
         });
         (0, instrumentation_1.registerInstrumentations)({
             instrumentations: [httpInstrumentation],

package/dist/telemetry_provider.d.ts

@@ -1,5 +1,7 @@
 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,6 +5,7 @@ const api_1 = require("@opentelemetry/api");
 const telemetry_1 = require("./telemetry");
 class OtelSpan {
     span;
+    closed = false;
     constructor(span) {
         this.span = span;
     }
@@ -20,11 +21,12 @@ 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() {
@@ -32,22 +34,26 @@ 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,
         };
-        // Handle parent context if provided
-        let ctx = api_1.context.active();
-        if (options?.parentContext) {
-            ctx = options.parentContext;
-        }
-        // Start the span
-        const span = tracer.startSpan(name, otelOptions, ctx);
-        return new OtelSpan(span);
+        const ctx = this.retrieveActiveSpanContext();
+        const span = new OtelSpan(tracer.startSpan(name, otelOptions, ctx));
+        this.spans.push(span);
+        return span;
     }
     async flush() {
         await telemetry_1.blaxelTelemetry.flush();

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@blaxel/telemetry",
-  "version": "0.2.18-dev.157",
+  "version": "0.2.18",
   "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.18-dev.157"
+    "@blaxel/core": "0.2.18"
   },
   "devDependencies": {
     "@eslint/js": "^9.26.0",