@blaxel/core 0.2.49-preview.112 → 0.2.50-dev.215

package/dist/cjs-browser/authentication/clientcredentials.js

@@ -54,10 +54,13 @@ class ClientCredentials extends credentials_js_1.Credentials {
         }
         catch (error) {
             if (retry > 0) {
+                console.error(`Authentication failed, retrying... (${4 - retry}/3)`, error);
                 await this.sleep(1000);
                 return this.processWithRetry(retry - 1);
             }
-            throw error;
+            this.currentPromise = null;
+            const originalMessage = error instanceof Error ? error.message : String(error);
+            throw new Error(`Authentication failed after 4 attempts: ${originalMessage}`, { cause: error });
         }
     }
     async process() {

package/dist/cjs-browser/common/internal.js

@@ -5,6 +5,8 @@ exports.getAlphanumericLimitedHash = getAlphanumericLimitedHash;
 exports.getGlobalUniqueHash = getGlobalUniqueHash;
 exports.pluralize = pluralize;
 exports.getForcedUrl = getForcedUrl;
+exports.getWorkloadTypeShort = getWorkloadTypeShort;
+exports.generateInternalUrl = generateInternalUrl;
 const env_js_1 = require("./env.js");
 // Pure JS MD5 implementation that matches standard crypto MD5
 function md5(input) {
@@ -186,3 +188,41 @@ function getForcedUrl(type, name) {
     }
     return null;
 }
+function getWorkloadTypeShort(type) {
+    const lowerType = type.toLowerCase();
+    switch (lowerType) {
+        case 'agent':
+        case 'agents':
+            return 'agt';
+        case 'mcp':
+        case 'mcps':
+        case 'function':
+        case 'functions':
+            return 'mcp';
+        case 'sandbox':
+        case 'sandboxes':
+            return 'sbx';
+        case 'job':
+        case 'jobs':
+            return 'job';
+        case 'model':
+        case 'models':
+            return 'mdl';
+        default:
+            // fallback to first 3 letters of type
+            return lowerType.substring(0, 3);
+    }
+}
+function generateInternalUrl(workspace, type, name, env, protocol, hostname, blCloud, workspaceId) {
+    if (blCloud && workspaceId) {
+        // New cloud format: bl-ENV-WORKLOAD_CALLED_NAME-WORKLOAD_TYPE_SHORT-WORKSPACE_ID
+        const workloadTypeShort = getWorkloadTypeShort(type);
+        const subdomain = `bl-${env}-${name}-${workloadTypeShort}-${workspaceId}`;
+        return `${protocol}://${subdomain}.${hostname}`;
+    }
+    else {
+        // Legacy format: bl-ENV-HASH.internalhostname
+        const hash = getGlobalUniqueHash(workspace, type, name);
+        return `${protocol}://bl-${env}-${hash}.${hostname}`;
+    }
+}

package/dist/cjs-browser/common/settings.js

@@ -10,7 +10,7 @@ function getPackageVersion() {
         if (typeof require !== "undefined") {
             // Try to require package.json (Node.js only, gracefully fails in browser)
             // eslint-disable-next-line @typescript-eslint/no-require-imports
-            const packageJson = {"version":"0.2.49-preview.112","commit":"d50dddc5a1963abd22bad3b294e9f6096c3e8287"};
+            const packageJson = {"version":"0.2.50-dev.215","commit":"833c34cbba3735060a62bdd3594fa917160f3d3c"};
             return packageJson.version || "unknown";
         }
         else {
@@ -62,7 +62,7 @@ function getCommitHash() {
         if (typeof require !== "undefined") {
             // Try to require package.json and look for commit field (set during build)
             // eslint-disable-next-line @typescript-eslint/no-require-imports
-            const packageJson = {"version":"0.2.49-preview.112","commit":"d50dddc5a1963abd22bad3b294e9f6096c3e8287"};
+            const packageJson = {"version":"0.2.50-dev.215","commit":"833c34cbba3735060a62bdd3594fa917160f3d3c"};
             // Check for commit in various possible locations
             const commit = packageJson.commit || packageJson.buildInfo?.commit;
             if (commit) {
@@ -165,6 +165,9 @@ class Settings {
     get blCloud() {
         return env_js_1.env.BL_CLOUD === "true";
     }
+    get workspaceId() {
+        return env_js_1.env.BL_WORKSPACE_ID || "";
+    }
     get generation() {
         return env_js_1.env.BL_GENERATION || "";
     }

package/dist/cjs-browser/jobs/jobs.js

@@ -20,8 +20,8 @@ class BlJob {
         return new URL(`${settings_js_1.settings.runUrl}/${settings_js_1.settings.workspace}/jobs/${this.jobName}`);
     }
     get internalUrl() {
-        const hash = (0, internal_js_1.getGlobalUniqueHash)(settings_js_1.settings.workspace, "job", this.jobName);
-        return new URL(`${settings_js_1.settings.runInternalProtocol}://bl-${settings_js_1.settings.env}-${hash}.${settings_js_1.settings.runInternalHostname}`);
+        const url = (0, internal_js_1.generateInternalUrl)(settings_js_1.settings.workspace, "job", this.jobName, settings_js_1.settings.env, settings_js_1.settings.runInternalProtocol, settings_js_1.settings.runInternalHostname, settings_js_1.settings.blCloud, settings_js_1.settings.workspaceId);
+        return new URL(url);
     }
     get forcedUrl() {
         return (0, internal_js_1.getForcedUrl)('job', this.jobName);

package/dist/cjs-browser/jobs/start.js

@@ -5,12 +5,43 @@ const autoload_js_1 = require("../common/autoload.js");
 const env_js_1 = require("../common/env.js");
 const telemetry_js_1 = require("../telemetry/telemetry.js");
 class BlJobWrapper {
+    async fetchWithRetry(url, maxRetries = 3) {
+        let lastError;
+        for (let attempt = 0; attempt <= maxRetries; attempt++) {
+            try {
+                const response = await fetch(url);
+                // If the response is successful, return it
+                if (response.ok) {
+                    return response;
+                }
+                // If it's not the last attempt and the status is retriable, retry
+                if (attempt < maxRetries && (response.status >= 500 || response.status === 429)) {
+                    lastError = new Error(`HTTP ${response.status}: ${response.statusText}`);
+                }
+                else {
+                    // For non-retriable errors or last attempt, return the response
+                    return response;
+                }
+            }
+            catch (error) {
+                lastError = error instanceof Error ? error : new Error(String(error));
+                // If this is the last attempt, throw the error
+                if (attempt === maxRetries) {
+                    throw lastError;
+                }
+            }
+            // Calculate exponential backoff delay: 2^attempt * 1000ms (1s, 2s, 4s)
+            const delay = Math.pow(2, attempt) * 1000;
+            await new Promise(resolve => setTimeout(resolve, delay));
+        }
+        throw lastError || new Error('Failed to fetch after retries');
+    }
     async getArguments() {
         if (!env_js_1.env.BL_EXECUTION_DATA_URL) {
             const args = this.parseCommandLineArgs();
             return args;
         }
-        const response = await fetch(env_js_1.env.BL_EXECUTION_DATA_URL);
+        const response = await this.fetchWithRetry(env_js_1.env.BL_EXECUTION_DATA_URL);
         const data = await response.json();
         return data.tasks[this.index] ?? {};
     }

package/dist/cjs-browser/sandbox/action.js

@@ -48,8 +48,7 @@ class SandboxAction {
         return this.sandbox.metadata?.url ?? `${settings_js_1.settings.runUrl}/${settings_js_1.settings.workspace}/sandboxes/${this.name}`;
     }
     get internalUrl() {
-        const hash = (0, internal_js_1.getGlobalUniqueHash)(settings_js_1.settings.workspace, "sandbox", this.name);
-        return `${settings_js_1.settings.runInternalProtocol}://bl-${settings_js_1.settings.env}-${hash}.${settings_js_1.settings.runInternalHostname}`;
+        return (0, internal_js_1.generateInternalUrl)(settings_js_1.settings.workspace, "sandbox", this.name, settings_js_1.settings.env, settings_js_1.settings.runInternalProtocol, settings_js_1.settings.runInternalHostname, settings_js_1.settings.blCloud, settings_js_1.settings.workspaceId);
     }
     get client() {
         if (this.sandbox.forceUrl) {

package/dist/cjs-browser/sandbox/codegen/codegen-ws.js

@@ -0,0 +1,30 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.SandboxCodegenWebSocket = void 0;
+const action_js_1 = require("../action.js");
+class SandboxCodegenWebSocket extends action_js_1.SandboxAction {
+    wsClient;
+    constructor(sandbox, wsClient) {
+        super(sandbox);
+        this.wsClient = wsClient;
+    }
+    async fastapply(path, codeEdit, model) {
+        const data = await this.wsClient.send("codegen:fastapply", {
+            path,
+            codeEdit,
+            model,
+        });
+        return data;
+    }
+    async reranking(path, query, scoreThreshold, tokenLimit, filePattern) {
+        const data = await this.wsClient.send("codegen:reranking", {
+            path,
+            query,
+            scoreThreshold,
+            tokenLimit,
+            filePattern,
+        });
+        return data;
+    }
+}
+exports.SandboxCodegenWebSocket = SandboxCodegenWebSocket;

package/dist/cjs-browser/sandbox/filesystem/filesystem-ws.js

@@ -0,0 +1,106 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.SandboxFileSystemWebSocket = void 0;
+const action_js_1 = require("../action.js");
+const filesystem_js_1 = require("./filesystem.js");
+class SandboxFileSystemWebSocket extends action_js_1.SandboxAction {
+    process;
+    wsClient;
+    httpClient;
+    constructor(sandbox, process, wsClient) {
+        super(sandbox);
+        this.process = process;
+        this.wsClient = wsClient;
+        // Create HTTP client for fallback operations
+        this.httpClient = new filesystem_js_1.SandboxFileSystem(sandbox, process);
+    }
+    async mkdir(path, permissions = "0755") {
+        path = this.formatPath(path);
+        const data = await this.wsClient.send("filesystem:create", {
+            path,
+            isDirectory: true,
+            permissions,
+        });
+        return data;
+    }
+    async write(path, content) {
+        path = this.formatPath(path);
+        const data = await this.wsClient.send("filesystem:create", {
+            path,
+            content,
+            isDirectory: false,
+        });
+        return data;
+    }
+    async writeBinary(path, content) {
+        return this.httpClient.writeBinary(path, content);
+    }
+    async writeTree(files, destinationPath = null) {
+        const path = this.formatPath(destinationPath ?? "");
+        const filesMap = files.reduce((acc, file) => {
+            acc[file.path] = file.content;
+            return acc;
+        }, {});
+        const data = await this.wsClient.send("filesystem:tree:create", {
+            path,
+            files: filesMap,
+        });
+        return data;
+    }
+    async read(path) {
+        path = this.formatPath(path);
+        const data = await this.wsClient.send("filesystem:get", { path });
+        return data.content;
+    }
+    async readBinary(path) {
+        // Binary downloads are better suited for HTTP
+        // Fall back to HTTP client for binary operations
+        return this.httpClient.readBinary(path);
+    }
+    async download(src, destinationPath, options = {}) {
+        // File downloads are better suited for HTTP
+        // Fall back to HTTP client
+        return this.httpClient.download(src, destinationPath, options);
+    }
+    async rm(path, recursive = false) {
+        path = this.formatPath(path);
+        const data = await this.wsClient.send("filesystem:delete", {
+            path,
+            recursive,
+        });
+        return data;
+    }
+    async ls(path) {
+        path = this.formatPath(path);
+        const data = await this.wsClient.send("filesystem:get", { path });
+        return data;
+    }
+    async cp(source, destination, { maxWait = 180000 } = {}) {
+        // Copy operation is typically done via process execution
+        // eslint-disable-next-line @typescript-eslint/no-unsafe-assignment
+        let process = await this.process.exec({
+            command: `cp -r ${source} ${destination}`,
+        });
+        // eslint-disable-next-line @typescript-eslint/no-unsafe-assignment, @typescript-eslint/no-unsafe-member-access, @typescript-eslint/no-unsafe-argument
+        process = await this.process.wait(process.pid, { maxWait, interval: 100 });
+        // eslint-disable-next-line @typescript-eslint/no-unsafe-member-access
+        if (process.status === "failed") {
+            // eslint-disable-next-line @typescript-eslint/no-unsafe-member-access
+            throw new Error(`Could not copy ${source} to ${destination} cause: ${process.logs}`);
+        }
+        return {
+            message: "Files copied",
+            source,
+            destination,
+        };
+    }
+    watch(path, callback, options) {
+        // File watching uses HTTP streaming which is already optimized
+        // Fall back to HTTP client
+        return this.httpClient.watch(path, callback, options);
+    }
+    formatPath(path) {
+        return path;
+    }
+}
+exports.SandboxFileSystemWebSocket = SandboxFileSystemWebSocket;

package/dist/cjs-browser/sandbox/network/network-ws.js

@@ -0,0 +1,12 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.SandboxNetworkWebSocket = void 0;
+const action_js_1 = require("../action.js");
+class SandboxNetworkWebSocket extends action_js_1.SandboxAction {
+    wsClient;
+    constructor(sandbox, wsClient) {
+        super(sandbox);
+        this.wsClient = wsClient;
+    }
+}
+exports.SandboxNetworkWebSocket = SandboxNetworkWebSocket;

package/dist/cjs-browser/sandbox/process/process-ws.js

@@ -0,0 +1,139 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.SandboxProcessWebSocket = void 0;
+const action_js_1 = require("../action.js");
+const process_js_1 = require("./process.js");
+class SandboxProcessWebSocket extends action_js_1.SandboxAction {
+    wsClient;
+    httpClient;
+    constructor(sandbox, wsClient) {
+        super(sandbox);
+        this.wsClient = wsClient;
+        // Create HTTP client for fallback operations
+        this.httpClient = new process_js_1.SandboxProcess(sandbox);
+    }
+    streamLogs(identifier, options) {
+        const streamId = this.wsClient.sendStream("process:logs:stream:start", { identifier }, (data) => {
+            // Handle streaming log data
+            // eslint-disable-next-line @typescript-eslint/no-unsafe-member-access
+            if (data && data.log) {
+                // eslint-disable-next-line @typescript-eslint/no-unsafe-member-access
+                const log = String(data.log);
+                // Parse log format: "stdout:" or "stderr:" prefix
+                if (log.startsWith('stdout:')) {
+                    const stdout = log.slice(7);
+                    options.onStdout?.(stdout);
+                    options.onLog?.(stdout);
+                }
+                else if (log.startsWith('stderr:')) {
+                    const stderr = log.slice(7);
+                    options.onStderr?.(stderr);
+                    options.onLog?.(stderr);
+                }
+                else {
+                    options.onLog?.(log);
+                }
+            }
+        }, () => {
+            // Stream ended
+        });
+        return {
+            close: () => this.wsClient.cancelStream(streamId),
+        };
+    }
+    async exec(process) {
+        let onLog;
+        if ('onLog' in process && process.onLog) {
+            onLog = process.onLog;
+            delete process.onLog;
+        }
+        // Store original wait_for_completion setting
+        const shouldWaitForCompletion = process.waitForCompletion;
+        // Always start process without wait_for_completion to avoid server-side blocking
+        if (shouldWaitForCompletion && onLog) {
+            process.waitForCompletion = false;
+        }
+        const data = await this.wsClient.send("process:execute", process);
+        let result = data;
+        // Handle wait_for_completion with parallel log streaming
+        if (shouldWaitForCompletion && onLog) {
+            const streamControl = this.streamLogs(result.pid, { onLog });
+            try {
+                // Wait for process completion
+                result = await this.wait(result.pid, { interval: 500, maxWait: 1000 * 60 * 60 });
+            }
+            finally {
+                // Clean up log streaming
+                if (streamControl) {
+                    streamControl.close();
+                }
+            }
+        }
+        else {
+            // For non-blocking execution, set up log streaming immediately if requested
+            if (onLog) {
+                const streamControl = this.streamLogs(result.pid, { onLog });
+                return {
+                    ...result,
+                    close() {
+                        if (streamControl) {
+                            streamControl.close();
+                        }
+                    },
+                };
+            }
+        }
+        return { ...result, close: () => { } };
+    }
+    async wait(identifier, { maxWait = 60000, interval = 1000 } = {}) {
+        const startTime = Date.now();
+        let status = "running";
+        let data = await this.get(identifier);
+        while (status === "running") {
+            await new Promise((resolve) => setTimeout(resolve, interval));
+            try {
+                data = await this.get(identifier);
+                status = data.status ?? "running";
+            }
+            catch {
+                break;
+            }
+            if (Date.now() - startTime > maxWait) {
+                throw new Error("Process did not finish in time");
+            }
+        }
+        return data;
+    }
+    async get(identifier) {
+        const data = await this.wsClient.send("process:get", { identifier });
+        return data;
+    }
+    async list() {
+        const data = await this.wsClient.send("process:list", {});
+        return data.processes || [];
+    }
+    async stop(identifier) {
+        const data = await this.wsClient.send("process:stop", { identifier });
+        return data;
+    }
+    async kill(identifier) {
+        const data = await this.wsClient.send("process:kill", { identifier });
+        return data;
+    }
+    async logs(identifier, type = "all") {
+        const data = await this.wsClient.send("process:logs", {
+            identifier,
+        });
+        if (type === "all") {
+            return data.logs || "";
+        }
+        else if (type === "stdout") {
+            return data.stdout || "";
+        }
+        else if (type === "stderr") {
+            return data.stderr || "";
+        }
+        throw new Error("Unsupported log type");
+    }
+}
+exports.SandboxProcessWebSocket = SandboxProcessWebSocket;

package/dist/cjs-browser/sandbox/sandbox.js

@@ -5,11 +5,16 @@ const uuid_1 = require("uuid");
 const index_js_1 = require("../client/index.js");
 const logger_js_1 = require("../common/logger.js");
 const index_js_2 = require("./filesystem/index.js");
+const filesystem_ws_js_1 = require("./filesystem/filesystem-ws.js");
 const index_js_3 = require("./network/index.js");
+const network_ws_js_1 = require("./network/network-ws.js");
 const preview_js_1 = require("./preview.js");
 const index_js_4 = require("./process/index.js");
+const process_ws_js_1 = require("./process/process-ws.js");
 const index_js_5 = require("./codegen/index.js");
+const codegen_ws_js_1 = require("./codegen/codegen-ws.js");
 const session_js_1 = require("./session.js");
+const index_js_6 = require("./websocket/index.js");
 const types_js_1 = require("./types.js");
 class SandboxInstance {
     sandbox;
@@ -19,14 +24,32 @@ class SandboxInstance {
     previews;
     sessions;
     codegen;
+    wsClient;
     constructor(sandbox) {
         this.sandbox = sandbox;
-        this.process = new index_js_4.SandboxProcess(sandbox);
-        this.fs = new index_js_2.SandboxFileSystem(sandbox, this.process);
-        this.network = new index_js_3.SandboxNetwork(sandbox);
+        // If connection type is websocket, initialize WebSocket client and use WebSocket transport layers
+        if (sandbox.connectionType === "websocket") {
+            const url = sandbox.forceUrl || sandbox.metadata?.url || "";
+            this.wsClient = new index_js_6.WebSocketClient({
+                url,
+                headers: sandbox.headers,
+            });
+            // Initialize WebSocket-based action handlers
+            this.process = new process_ws_js_1.SandboxProcessWebSocket(sandbox, this.wsClient);
+            this.fs = new filesystem_ws_js_1.SandboxFileSystemWebSocket(sandbox, this.process, this.wsClient);
+            this.network = new network_ws_js_1.SandboxNetworkWebSocket(sandbox, this.wsClient);
+            this.codegen = new codegen_ws_js_1.SandboxCodegenWebSocket(sandbox, this.wsClient);
+        }
+        else {
+            // Default to HTTP-based action handlers
+            this.process = new index_js_4.SandboxProcess(sandbox);
+            this.fs = new index_js_2.SandboxFileSystem(sandbox, this.process);
+            this.network = new index_js_3.SandboxNetwork(sandbox);
+            this.codegen = new index_js_5.SandboxCodegen(sandbox);
+        }
+        // These are always HTTP-based
         this.previews = new preview_js_1.SandboxPreviews(sandbox);
         this.sessions = new session_js_1.SandboxSessions(sandbox);
-        this.codegen = new index_js_5.SandboxCodegen(sandbox);
     }
     get metadata() {
         return this.sandbox.metadata;
@@ -45,10 +68,17 @@ class SandboxInstance {
         logger_js_1.logger.warn("⚠️  Warning: sandbox.wait() is deprecated. You don't need to wait for the sandbox to be deployed anymore.");
         return this;
     }
+    closeConnection() {
+        if (this.wsClient) {
+            this.wsClient.close();
+        }
+    }
     static async create(sandbox, { safe = true } = {}) {
         const defaultName = `sandbox-${(0, uuid_1.v4)().replace(/-/g, '').substring(0, 8)}`;
         const defaultImage = `blaxel/base-image:latest`;
         const defaultMemory = 4096;
+        // Store connection type if provided
+        let connectionType;
         // Handle SandboxCreateConfiguration or simple dict with name/image/memory/ports/envs/volumes keys
         if (!sandbox ||
             'name' in sandbox ||
@@ -58,7 +88,8 @@ class SandboxInstance {
             'envs' in sandbox ||
             'volumes' in sandbox ||
             'lifecycle' in sandbox ||
-            'snapshotEnabled' in sandbox) {
+            'snapshotEnabled' in sandbox ||
+            'connectionType' in sandbox) {
             if (!sandbox)
                 sandbox = {};
             if (!sandbox.name)
@@ -67,6 +98,7 @@ class SandboxInstance {
                 sandbox.image = defaultImage;
             if (!sandbox.memory)
                 sandbox.memory = defaultMemory;
+            connectionType = sandbox.connectionType;
             const ports = (0, types_js_1.normalizePorts)(sandbox.ports);
             const envs = (0, types_js_1.normalizeEnvs)(sandbox.envs);
             const volumes = (0, types_js_1.normalizeVolumes)(sandbox.volumes);
@@ -115,7 +147,16 @@ class SandboxInstance {
             body: sandbox,
             throwOnError: true,
         });
-        const instance = new SandboxInstance(data);
+        // Add connection type to configuration
+        const config = {
+            ...data,
+            connectionType: connectionType || "http",
+        };
+        const instance = new SandboxInstance(config);
+        // Connect WebSocket if needed
+        if (connectionType === "websocket" && instance.wsClient) {
+            await instance.wsClient.connect();
+        }
         // TODO remove this part once we have a better way to handle this
         if (safe) {
             try {
@@ -125,20 +166,34 @@ class SandboxInstance {
         }
         return instance;
     }
-    static async get(sandboxName) {
+    static async get(sandboxName, connectionType) {
         const { data } = await (0, index_js_1.getSandbox)({
             path: {
                 sandboxName,
             },
             throwOnError: true,
         });
-        return new SandboxInstance(data);
+        // Add connection type to configuration
+        const config = {
+            ...data,
+            connectionType: connectionType || "http",
+        };
+        const instance = new SandboxInstance(config);
+        // Connect WebSocket if needed
+        if (connectionType === "websocket" && instance.wsClient) {
+            await instance.wsClient.connect();
+        }
+        return instance;
     }
     static async list() {
         const { data } = await (0, index_js_1.listSandboxes)({ throwOnError: true });
         return data.map((sandbox) => new SandboxInstance(sandbox));
     }
-    static async delete(sandboxName) {
+    static async delete(sandboxName, instance) {
+        // Close WebSocket connection if instance is provided
+        if (instance && instance.wsClient) {
+            instance.closeConnection();
+        }
         const { data } = await (0, index_js_1.deleteSandbox)({
             path: {
                 sandboxName,
@@ -168,8 +223,10 @@ class SandboxInstance {
                 if (!name) {
                     throw new Error("Sandbox name is required");
                 }
+                // Get connection type if specified
+                const connectionType = 'connectionType' in sandbox ? sandbox.connectionType : undefined;
                 // Get the existing sandbox to check its status
-                const sandboxInstance = await SandboxInstance.get(name);
+                const sandboxInstance = await SandboxInstance.get(name, connectionType);
                 // If the sandbox is TERMINATED, treat it as not existing
                 if (sandboxInstance.status === "TERMINATED") {
                     // Create a new sandbox - backend will handle cleanup of the terminated one