@blaxel/telemetry 0.2.6 → 0.2.7-preview.6

package/dist/index.js

@@ -1,9 +1,15 @@
 "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 (settings.loggerType === "http") {
-(0, legacy_logger_1.setLegacyLogger)();
+if (core_1.settings.loggerType === "http") {
+    (0, legacy_logger_1.setLegacyLogger)();
+}
+else if (core_1.settings.loggerType === "json") {
+    (0, json_logger_1.setJsonLogger)();
+}

package/dist/telemetry.d.ts

@@ -38,6 +38,7 @@ 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,6 +112,17 @@ 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,4 +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

@@ -2,8 +2,10 @@
 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;
     }
@@ -24,6 +26,7 @@ class OtelSpan {
         });
     }
     end() {
+        this.closed = true;
         this.span.end();
     }
     getContext() {
@@ -31,22 +34,29 @@ 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);
+        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();
     }
 }
 exports.OtelTelemetryProvider = OtelTelemetryProvider;

package/package.json

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