@blaxel/telemetry 0.2.4-dev.70 → 0.2.4-dev.71

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,5 @@
 import { BlaxelSpan, BlaxelSpanOptions, BlaxelTelemetryProvider } from "@blaxel/core";
 export declare class OtelTelemetryProvider implements BlaxelTelemetryProvider {
     startSpan(name: string, options?: BlaxelSpanOptions): BlaxelSpan;
+    flush(): Promise<void>;
 }

package/dist/telemetry_provider.js

@@ -2,6 +2,7 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.OtelTelemetryProvider = void 0;
 const api_1 = require("@opentelemetry/api");
+const telemetry_1 = require("./telemetry");
 class OtelSpan {
     span;
     constructor(span) {
@@ -48,5 +49,8 @@ class OtelTelemetryProvider {
         const span = tracer.startSpan(name, otelOptions, ctx);
         return new OtelSpan(span);
     }
+    async flush() {
+        await telemetry_1.blaxelTelemetry.flush();
+    }
 }
 exports.OtelTelemetryProvider = OtelTelemetryProvider;

package/package.json

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