@blaxel/telemetry 0.2.6-dev.74 → 0.2.6-preview.4

package/dist/index.js

@@ -1,15 +1,9 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.blaxelTelemetry = void 0;
-const core_1 = require("@blaxel/core");
-const json_logger_1 = require("./json_logger");
 const legacy_logger_1 = require("./legacy_logger");
 const telemetry_1 = require("./telemetry");
 Object.defineProperty(exports, "blaxelTelemetry", { enumerable: true, get: function () { return telemetry_1.blaxelTelemetry; } });
 telemetry_1.blaxelTelemetry.initialize();
-if (core_1.settings.loggerType === "http") {
-    (0, legacy_logger_1.setLegacyLogger)();
-}
-else if (core_1.settings.loggerType === "json") {
-    (0, json_logger_1.setJsonLogger)();
-}
+// if (settings.loggerType === "http") {
+(0, legacy_logger_1.setLegacyLogger)();

package/dist/telemetry.d.ts

@@ -38,7 +38,6 @@ declare class TelemetryManager {
     get enabled(): boolean;
     get authHeaders(): Record<string, string>;
     sleep(ms: number): Promise<unknown>;
-    flush(): Promise<void>;
     getLogger(): Promise<Logger>;
     setupSignalHandler(): void;
     /**

package/dist/telemetry.js

@@ -112,17 +112,6 @@ class TelemetryManager {
     async sleep(ms) {
         return new Promise((resolve) => setTimeout(resolve, ms));
     }
-    async flush() {
-        if (this.nodeTracerProvider) {
-            await this.nodeTracerProvider.shutdown();
-        }
-        if (this.meterProvider) {
-            await this.meterProvider.shutdown();
-        }
-        if (this.loggerProvider) {
-            await this.loggerProvider.shutdown();
-        }
-    }
     async getLogger() {
         if (!this.otelLogger) {
             await this.sleep(100);

package/dist/telemetry_provider.d.ts

@@ -1,7 +1,4 @@
 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

@@ -2,10 +2,8 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.OtelTelemetryProvider = void 0;
 const api_1 = require("@opentelemetry/api");
-const telemetry_1 = require("./telemetry");
 class OtelSpan {
     span;
-    closed = false;
     constructor(span) {
         this.span = span;
     }
@@ -26,7 +24,6 @@ class OtelSpan {
         });
     }
     end() {
-        this.closed = true;
         this.span.end();
     }
     getContext() {
@@ -34,29 +31,22 @@ 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,
         };
-        let ctx = this.retrieveActiveSpanContext();
-        let span = new OtelSpan(tracer.startSpan(name, otelOptions, ctx));
-        this.spans.push(span);
-        return span;
-    }
-    async flush() {
-        await telemetry_1.blaxelTelemetry.flush();
+        // 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);
     }
 }
 exports.OtelTelemetryProvider = OtelTelemetryProvider;

package/package.json

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