@blaxel/core 0.2.17-dev.130 → 0.2.17-dev.132

package/dist/common/logger.js

@@ -31,7 +31,7 @@ function stringify(obj, maxDepth = 1, depth = 0) {
 class Logger {
     logger;
     constructor() {
-        this.logger = console;
+        this.logger = null;
     }
     setLogger(logger) {
         this.logger = logger;
@@ -45,16 +45,24 @@ class Logger {
         }).join(" ");
     }
     info(...message) {
-        this.logger.info(this.parseArgs(message));
+        if (this.logger) {
+            this.logger.info(this.parseArgs(message));
+        }
     }
     debug(...message) {
-        this.logger.debug(this.parseArgs(message));
+        if (this.logger) {
+            this.logger.debug(this.parseArgs(message));
+        }
     }
     warn(...message) {
-        this.logger.warn(this.parseArgs(message));
+        if (this.logger) {
+            this.logger.warn(this.parseArgs(message));
+        }
     }
     error(...message) {
-        this.logger.error(this.parseArgs(message));
+        if (this.logger) {
+            this.logger.error(this.parseArgs(message));
+        }
     }
 }
 const logger = new Logger();

package/dist/sandbox/sandbox.js

@@ -46,7 +46,7 @@ class SandboxInstance {
                     },
                     throwOnError: true,
                 });
-                logger_js_1.logger.info(`Waiting for sandbox to be deployed, status: ${data.status}`);
+                logger_js_1.logger.debug(`Waiting for sandbox to be deployed, status: ${data.status}`);
                 this.sandbox = data;
             }
             catch (e) {

package/dist/telemetry/telemetry.d.ts

@@ -20,7 +20,7 @@ export interface BlaxelSpan {
     /** Record an error on the span */
     recordException(error: Error): void;
     /** Set the status of the span */
-    setStatus(status: "ok" | "error", message?: string): void;
+    setStatus(status: 'ok' | 'error', message?: string): void;
     /** End the span */
     end(): void;
     /** Get the span context (for passing to child spans) */

package/dist/telemetry/telemetry.js

@@ -14,9 +14,7 @@ class NoopSpan {
     recordException() { }
     setStatus() { }
     end() { }
-    getContext() {
-        return null;
-    }
+    getContext() { return null; }
 }
 /**
  * No-operation implementation of TelemetryProvider
@@ -68,16 +66,13 @@ async function withSpan(name, fn, options) {
     const span = startSpan(name, options);
     try {
         const result = await fn();
-        span.setStatus("ok");
+        span.end();
         return result;
     }
     catch (error) {
-        span.setStatus("error", error.message);
         span.recordException(error);
-        throw error;
-    }
-    finally {
         span.end();
+        throw error;
     }
 }
 async function flush() {

package/dist/tools/mcpTool.d.ts

@@ -15,7 +15,7 @@ export declare class McpTool {
     get forcedUrl(): import("url").URL | null;
     get url(): import("url").URL;
     start(): Promise<void>;
-    close(): Promise<void>;
+    close(now?: boolean): Promise<void>;
     stopCloseTimer(): void;
     listTools(): Promise<Tool[]>;
     call(toolName: string, args: Record<string, unknown> | undefined): Promise<unknown>;

package/dist/tools/mcpTool.js

@@ -70,38 +70,38 @@ class McpTool {
     async start() {
         logger_js_1.logger.debug(`MCP:${this.name}:start`);
         this.stopCloseTimer();
-        this.startPromise =
-            this.startPromise ||
-                (async () => {
-                    await (0, index_js_2.authenticate)();
-                    try {
-                        logger_js_1.logger.debug(`MCP:${this.name}:Connecting::${this.url.toString()}`);
-                        this.transport = new client_js_1.BlaxelMcpClientTransport(this.url.toString(), settings_js_1.settings.headers);
-                        await this.client.connect(this.transport);
-                        logger_js_1.logger.debug(`MCP:${this.name}:Connected`);
-                    }
-                    catch (err) {
-                        if (err instanceof Error) {
-                            logger_js_1.logger.error(err.stack);
-                        }
-                        if (!this.fallbackUrl) {
-                            throw err;
-                        }
-                        logger_js_1.logger.debug(`MCP:${this.name}:Connecting to fallback`);
-                        this.transport = new client_js_1.BlaxelMcpClientTransport(this.fallbackUrl.toString(), settings_js_1.settings.headers);
-                        await this.client.connect(this.transport);
-                        logger_js_1.logger.debug(`MCP:${this.name}:Connected to fallback`);
-                    }
-                })();
+        this.startPromise = this.startPromise || (async () => {
+            await (0, index_js_2.authenticate)();
+            try {
+                logger_js_1.logger.debug(`MCP:${this.name}:Connecting::${this.url.toString()}`);
+                this.transport = new client_js_1.BlaxelMcpClientTransport(this.url.toString(), settings_js_1.settings.headers);
+                await this.client.connect(this.transport);
+                logger_js_1.logger.debug(`MCP:${this.name}:Connected`);
+            }
+            catch (err) {
+                if (err instanceof Error) {
+                    logger_js_1.logger.error(err.stack);
+                }
+                if (!this.fallbackUrl) {
+                    throw err;
+                }
+                logger_js_1.logger.debug(`MCP:${this.name}:Connecting to fallback`);
+                this.transport = new client_js_1.BlaxelMcpClientTransport(this.fallbackUrl.toString(), settings_js_1.settings.headers);
+                await this.client.connect(this.transport);
+                logger_js_1.logger.debug(`MCP:${this.name}:Connected to fallback`);
+            }
+        })();
         return await this.startPromise;
     }
-    async close() {
-        logger_js_1.logger.debug(`MCP:${this.name}:Close in ${this.ms}ms`);
-        if (!this.ms) {
+    async close(now = false) {
+        logger_js_1.logger.debug(`MCP:${this.name}:Close in ${now ? 0 : this.ms}ms`);
+        if (now || !this.ms) {
+            if (this.timer) {
+                clearTimeout(this.timer);
+            }
             delete this.startPromise;
             return this.client.close();
         }
-        // Use setTimeout with proper error handling
         this.timer = setTimeout(() => {
             logger_js_1.logger.debug(`MCP:${this.name}:CloseTimer`);
             delete this.startPromise;
@@ -110,7 +110,7 @@ class McpTool {
                     logger_js_1.logger.error(err.stack);
                 }
             });
-        }, this.ms);
+        }, now ? 0 : this.ms);
     }
     stopCloseTimer() {
         logger_js_1.logger.debug(`MCP:${this.name}:StopCloseTimer`);
@@ -172,9 +172,7 @@ class McpTool {
             });
             logger_js_1.logger.debug(`MCP:${this.name}:Tool calling:result`);
             await this.close();
-            logger_js_1.logger.debug(`MCP:${this.name}:Tool result`, toolName, JSON.stringify(args)
-            // result
-            );
+            logger_js_1.logger.debug(`MCP:${this.name}:Tool result`, toolName, JSON.stringify(args));
             span.setAttribute("tool.call.result", JSON.stringify(result));
             return result;
         }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@blaxel/core",
-  "version": "0.2.17-dev.130",
+  "version": "0.2.17-dev.132",
   "description": "Blaxel Core SDK for TypeScript",
   "license": "MIT",
   "author": "Blaxel, INC (https://blaxel.ai)",