@blaxel/core 0.2.71-dev.101 → 0.2.71-preview.102

package/dist/cjs/agents/index.js

@@ -1,11 +1,12 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.getAgentMetadata = exports.blAgent = exports.BlAgent = void 0;
+exports.getAgentMetadata = exports.blAgent = void 0;
 const index_js_1 = require("../cache/index.js");
 const index_js_2 = require("../client/index.js");
 const internal_js_1 = require("../common/internal.js");
 const logger_js_1 = require("../common/logger.js");
 const settings_js_1 = require("../common/settings.js");
+const telemetry_js_1 = require("../telemetry/telemetry.js");
 class BlAgent {
     agentName;
     constructor(agentName) {
@@ -56,28 +57,44 @@ class BlAgent {
     }
     async run(input, headers = {}, params = {}) {
         logger_js_1.logger.debug(`Agent Calling: ${this.agentName}`);
+        const span = (0, telemetry_js_1.startSpan)(this.agentName, {
+            attributes: {
+                "agent.name": this.agentName,
+                "agent.args": JSON.stringify(input),
+                "span.type": "agent.run",
+            },
+            isRoot: false,
+        });
         try {
             const response = await this.call(this.url, input, headers, params);
+            span.setAttribute("agent.run.result", await response.text());
             return await response.text();
         }
         catch (err) {
             if (err instanceof Error) {
                 if (!this.fallbackUrl) {
+                    span.setAttribute("agent.run.error", err.stack);
                     throw err;
                 }
                 try {
                     const response = await this.call(this.fallbackUrl, input, headers, params);
+                    span.setAttribute("agent.run.result", await response.text());
                     return await response.text();
                 }
                 catch (err) {
+                    if (err instanceof Error) {
+                        span.setAttribute("agent.run.error", err.stack);
+                    }
                     throw err;
                 }
             }
             throw err;
         }
+        finally {
+            span.end();
+        }
     }
 }
-exports.BlAgent = BlAgent;
 const blAgent = (agentName) => {
     return new BlAgent(agentName);
 };

package/dist/cjs/common/settings.js

@@ -9,8 +9,8 @@ const index_js_1 = require("../authentication/index.js");
 const env_js_1 = require("../common/env.js");
 const node_js_1 = require("../common/node.js");
 // Build info - these placeholders are replaced at build time by build:replace-imports
-const BUILD_VERSION = "0.2.71-dev.101";
-const BUILD_COMMIT = "dccf33c2f51602cd1cd06116b55d326f4645465a";
+const BUILD_VERSION = "0.2.71-preview.102";
+const BUILD_COMMIT = "0e2cce73a323c7322baf2e2fd878f0819b40b862";
 const BUILD_SENTRY_DSN = "https://fd5e60e1c9820e1eef5ccebb84a07127@o4508714045276160.ingest.us.sentry.io/4510465864564736";
 // Cache for config.yaml tracking value
 let configTrackingValue = null;

package/dist/cjs/jobs/jobs.js

@@ -1,9 +1,10 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.blJob = exports.BlJob = void 0;
+exports.blJob = void 0;
 const index_js_1 = require("../client/index.js");
 const logger_js_1 = require("../common/logger.js");
 const settings_js_1 = require("../common/settings.js");
+const telemetry_js_1 = require("../telemetry/telemetry.js");
 class BlJob {
     jobName;
     constructor(jobName) {
@@ -11,14 +12,32 @@ class BlJob {
     }
     async run(tasks, options) {
         logger_js_1.logger.debug(`Job Calling: ${this.jobName}`);
-        const request = {
-            tasks,
-            ...(options?.env && { env: options.env }),
-            ...(options?.memory && { memory: options.memory }),
-            ...(options?.executionId && { executionId: options.executionId }),
-        };
-        const executionId = await this.createExecution(request);
-        return executionId;
+        const span = (0, telemetry_js_1.startSpan)(this.jobName, {
+            attributes: {
+                "job.name": this.jobName,
+                "span.type": "job.run",
+            },
+            isRoot: false,
+        });
+        try {
+            const request = {
+                tasks,
+                ...(options?.env && { env: options.env }),
+                ...(options?.memory && { memory: options.memory }),
+                ...(options?.executionId && { executionId: options.executionId }),
+            };
+            const executionId = await this.createExecution(request);
+            return executionId;
+        }
+        catch (err) {
+            if (err instanceof Error) {
+                span.setAttribute("job.run.error", err.stack);
+            }
+            throw err;
+        }
+        finally {
+            span.end();
+        }
     }
     /**
      * Create a new execution for this job and return the execution ID
@@ -120,7 +139,6 @@ class BlJob {
         throw new Error(`Execution ${executionId} did not complete within ${maxWait}ms`);
     }
 }
-exports.BlJob = BlJob;
 const blJob = (jobName) => {
     return new BlJob(jobName);
 };

package/dist/cjs/mcp/server.js

@@ -38,6 +38,8 @@ const uuid_1 = require("uuid");
 const ws_1 = __importStar(require("ws"));
 const env_js_1 = require("../common/env.js");
 const logger_js_1 = require("../common/logger.js");
+const telemetry_js_1 = require("../telemetry/telemetry.js");
+const spans = new Map();
 class BlaxelMcpServerTransport {
     port;
     wss;
@@ -72,10 +74,26 @@ class BlaxelMcpServerTransport {
                 ws,
             });
             this.onconnection?.(clientId);
-            ws.on("message", async (data) => {
+            ws.on("message", (data) => {
+                const span = (0, telemetry_js_1.startSpan)("message", {
+                    attributes: {
+                        "mcp.client.id": clientId,
+                        "span.type": "mcp.message",
+                    },
+                    isRoot: false,
+                });
                 try {
                     const msg = JSON.parse(data.toString());
-                    await this.messageHandler?.(msg, clientId);
+                    this.messageHandler?.(msg, clientId);
+                    if ("method" in msg && "id" in msg && "params" in msg) {
+                        span.setAttributes({
+                            "mcp.message.parsed": true,
+                            "mcp.method": msg.method,
+                            "mcp.messageId": msg.id,
+                            "mcp.toolName": msg.params?.name,
+                        });
+                        spans.set(clientId + ":" + msg.id, span);
+                    }
                     // Handle msg.id safely
                     const msgId = msg.id ? String(msg.id) : "";
                     const [cId, parsedMsgId] = msgId.split(":");
@@ -83,7 +101,27 @@ class BlaxelMcpServerTransport {
                     // Use optional chaining for safe access
                     const client = this.clients.get(cId ?? "");
                     if (client?.ws?.readyState === ws_1.default.OPEN) {
-                        client.ws.send(JSON.stringify(msg));
+                        const msgSpan = spans.get(cId + ":" + (msg.id ?? ""));
+                        try {
+                            client.ws.send(JSON.stringify(msg));
+                            if (msgSpan) {
+                                msgSpan.setAttributes({
+                                    "mcp.message.response_sent": true,
+                                });
+                            }
+                        }
+                        catch (err) {
+                            if (msgSpan) {
+                                msgSpan.setStatus("error"); // Error status
+                                msgSpan.recordException(err);
+                            }
+                            throw err;
+                        }
+                        finally {
+                            if (msgSpan) {
+                                msgSpan.end();
+                            }
+                        }
                     }
                     else {
                         this.clients.delete(cId);
@@ -92,11 +130,14 @@ class BlaxelMcpServerTransport {
                 }
                 catch (err) {
                     if (err instanceof Error) {
+                        span.setStatus("error"); // Error status
+                        span.recordException(err);
                         this.onerror?.(err);
                     }
                     else {
                         this.onerror?.(new Error(`Failed to parse message: ${String(err)}`));
                     }
+                    span.end();
                 }
             });
             ws.on("close", () => {
@@ -118,7 +159,27 @@ class BlaxelMcpServerTransport {
             // Send to specific client
             const client = this.clients.get(cId);
             if (client?.ws?.readyState === ws_1.default.OPEN) {
-                client.ws.send(data);
+                const msgSpan = spans.get(cId + ":" + msg.id);
+                try {
+                    client.ws.send(data);
+                    if (msgSpan) {
+                        msgSpan.setAttributes({
+                            "mcp.message.response_sent": true,
+                        });
+                    }
+                }
+                catch (err) {
+                    if (msgSpan) {
+                        msgSpan.setStatus("error"); // Error status
+                        msgSpan.recordException(err);
+                    }
+                    throw err;
+                }
+                finally {
+                    if (msgSpan) {
+                        msgSpan.end();
+                    }
+                }
             }
             else {
                 this.clients.delete(cId);

package/dist/cjs/tools/index.js

@@ -1,12 +1,10 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.getToolMetadata = exports.blTool = exports.blTools = exports.BLTools = exports.getTool = exports.McpTool = void 0;
+exports.getToolMetadata = exports.blTool = exports.blTools = exports.BLTools = exports.getTool = void 0;
 const index_js_1 = require("../cache/index.js");
 const client_js_1 = require("../client/client.js");
 const internal_js_1 = require("../common/internal.js");
 const mcpTool_js_1 = require("./mcpTool.js");
-var mcpTool_js_2 = require("./mcpTool.js");
-Object.defineProperty(exports, "McpTool", { enumerable: true, get: function () { return mcpTool_js_2.McpTool; } });
 const getTool = async (name, options) => {
     return await (0, mcpTool_js_1.getMcpTool)(name, options);
 };

package/dist/cjs/tools/mcpTool.js

@@ -9,6 +9,7 @@ const logger_js_1 = require("../common/logger.js");
 const settings_js_1 = require("../common/settings.js");
 const index_js_2 = require("../index.js");
 const client_js_1 = require("../mcp/client.js");
+const telemetry_js_1 = require("../telemetry/telemetry.js");
 const zodSchema_js_1 = require("./zodSchema.js");
 const McpToolCache = new Map();
 class McpTool {
@@ -140,24 +141,47 @@ class McpTool {
         }
     }
     async listTools() {
-        logger_js_1.logger.debug(`MCP:${this.name}:Listing tools`);
-        await this.start();
-        const { tools } = (await this.client.listTools());
-        await this.close();
-        const result = tools.map((tool) => {
-            return {
-                name: tool.name,
-                description: tool.description,
-                inputSchema: (0, zodSchema_js_1.schemaToZodSchema)(tool.inputSchema),
-                originalSchema: tool.inputSchema,
-                call: (input) => {
-                    return this.call(tool.name, input);
-                },
-            };
+        const span = (0, telemetry_js_1.startSpan)(this.name, {
+            attributes: {
+                "span.type": "tool.list",
+            },
         });
-        return result;
+        try {
+            logger_js_1.logger.debug(`MCP:${this.name}:Listing tools`);
+            await this.start();
+            const { tools } = (await this.client.listTools());
+            await this.close();
+            const result = tools.map((tool) => {
+                return {
+                    name: tool.name,
+                    description: tool.description,
+                    inputSchema: (0, zodSchema_js_1.schemaToZodSchema)(tool.inputSchema),
+                    originalSchema: tool.inputSchema,
+                    call: (input) => {
+                        return this.call(tool.name, input);
+                    },
+                };
+            });
+            span.setAttribute("tool.list.result", JSON.stringify(result));
+            return result;
+        }
+        catch (err) {
+            span.setStatus("error");
+            span.recordException(err);
+            throw err;
+        }
+        finally {
+            span.end();
+        }
     }
     async call(toolName, args) {
+        const span = (0, telemetry_js_1.startSpan)(this.name + "." + toolName, {
+            attributes: {
+                "span.type": "tool.call",
+                "tool.name": toolName,
+                "tool.args": JSON.stringify(args),
+            },
+        });
         try {
             logger_js_1.logger.debug(`MCP:${this.name}:Tool calling`, toolName, JSON.stringify(args));
             logger_js_1.logger.debug(`MCP:${this.name}:Tool calling:start`);
@@ -171,6 +195,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));
+            span.setAttribute("tool.call.result", JSON.stringify(result));
             return result;
         }
         catch (err) {
@@ -185,6 +210,9 @@ class McpTool {
             }
             throw err;
         }
+        finally {
+            span.end();
+        }
     }
     async getTransport(forcedUrl) {
         if (!this.transportName) {

package/dist/cjs/types/agents/index.d.ts

@@ -1,5 +1,5 @@
 import { Agent } from "../client/index.js";
-export declare class BlAgent {
+declare class BlAgent {
     agentName: string;
     constructor(agentName: string);
     get fallbackUrl(): import("url").URL | null;
@@ -12,3 +12,4 @@ export declare class BlAgent {
 }
 export declare const blAgent: (agentName: string) => BlAgent;
 export declare const getAgentMetadata: (agent: string) => Promise<Agent | null>;
+export {};

package/dist/cjs/types/jobs/jobs.d.ts

@@ -1,5 +1,5 @@
 import { CreateJobExecutionRequest, JobExecution } from "../client/index.js";
-export declare class BlJob {
+declare class BlJob {
     jobName: string;
     constructor(jobName: string);
     run(tasks: Record<string, unknown>[], options?: {
@@ -39,3 +39,4 @@ export declare class BlJob {
     }): Promise<JobExecution>;
 }
 export declare const blJob: (jobName: string) => BlJob;
+export {};

package/dist/cjs/types/sandbox/client/types.gen.d.ts

@@ -137,16 +137,9 @@ export type ProcessRequest = {
     env?: {
         [key: string]: string;
     };
-    /**
-     * Disable scale-to-zero while process runs. Default timeout is 600s (10 minutes). Set timeout to 0 for infinite.
-     */
-    keepAlive?: boolean;
     maxRestarts?: number;
     name?: string;
     restartOnFailure?: boolean;
-    /**
-     * Timeout in seconds. When keepAlive is true, defaults to 600s (10 minutes). Set to 0 for infinite (no auto-kill).
-     */
     timeout?: number;
     waitForCompletion?: boolean;
     waitForPorts?: Array<number>;
@@ -156,10 +149,6 @@ export type ProcessResponse = {
     command: string;
     completedAt: string;
     exitCode: number;
-    /**
-     * Whether scale-to-zero is disabled for this process
-     */
-    keepAlive?: boolean;
     logs: string;
     maxRestarts?: number;
     name: string;

package/dist/cjs/types/tools/index.d.ts

@@ -1,7 +1,6 @@
 import { Function } from "../client/client.js";
 import { ToolOptions } from "./mcpTool.js";
 import { Tool } from "./types.js";
-export { McpTool } from "./mcpTool.js";
 export type { ToolOptions };
 export declare const getTool: (name: string, options?: number | ToolOptions) => Promise<Tool[]>;
 export declare class BLTools {