@blaxel/core 0.2.49-dev.210 → 0.2.49-dev.212

package/dist/cjs/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/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/sandbox/filesystem/filesystem.js

@@ -8,7 +8,7 @@ const index_js_1 = require("../client/index.js");
 // Multipart upload constants
 const MULTIPART_THRESHOLD = 5 * 1024 * 1024; // 5MB
 const CHUNK_SIZE = 5 * 1024 * 1024; // 5MB per part
-const MAX_PARALLEL_UPLOADS = 3; // Number of parallel part uploads
+const MAX_PARALLEL_UPLOADS = 20; // Number of parallel part uploads
 class SandboxFileSystem extends action_js_1.SandboxAction {
     process;
     constructor(sandbox, process) {
@@ -304,7 +304,6 @@ class SandboxFileSystem extends action_js_1.SandboxAction {
     // Multipart upload helper methods
     async initiateMultipartUpload(path, permissions = "0644") {
         path = this.formatPath(path);
-        // eslint-disable-next-line @typescript-eslint/no-unsafe-assignment, @typescript-eslint/no-unsafe-call
         const { data } = await (0, index_js_1.postFilesystemMultipartInitiateByPath)({
             path: { path },
             body: { permissions },
@@ -315,7 +314,6 @@ class SandboxFileSystem extends action_js_1.SandboxAction {
         return data;
     }
     async uploadPart(uploadId, partNumber, fileBlob) {
-        // eslint-disable-next-line @typescript-eslint/no-unsafe-assignment, @typescript-eslint/no-unsafe-call
         const { data } = await (0, index_js_1.putFilesystemMultipartByUploadIdPart)({
             path: { uploadId },
             query: { partNumber },
@@ -327,7 +325,6 @@ class SandboxFileSystem extends action_js_1.SandboxAction {
         return data;
     }
     async completeMultipartUpload(uploadId, parts) {
-        // eslint-disable-next-line @typescript-eslint/no-unsafe-assignment, @typescript-eslint/no-unsafe-call
         const { data } = await (0, index_js_1.postFilesystemMultipartByUploadIdComplete)({
             path: { uploadId },
             body: { parts },
@@ -338,19 +335,17 @@ class SandboxFileSystem extends action_js_1.SandboxAction {
         return data;
     }
     async abortMultipartUpload(uploadId) {
-        // eslint-disable-next-line @typescript-eslint/no-unsafe-call
-        await (0, index_js_1.deleteFilesystemMultipartByUploadIdAbort)({
+        const { data } = await (0, index_js_1.deleteFilesystemMultipartByUploadIdAbort)({
             path: { uploadId },
             baseUrl: this.url,
             client: this.client,
             throwOnError: true,
         });
+        return data;
     }
     async uploadWithMultipart(path, blob, permissions = "0644") {
         // Initiate multipart upload
-        // eslint-disable-next-line @typescript-eslint/no-unsafe-assignment
         const initResponse = await this.initiateMultipartUpload(path, permissions);
-        // eslint-disable-next-line @typescript-eslint/no-unsafe-assignment, @typescript-eslint/no-unsafe-member-access
         const uploadId = initResponse.uploadId;
         if (!uploadId) {
             throw new Error("Failed to get upload ID from initiate response");
@@ -367,26 +362,20 @@ class SandboxFileSystem extends action_js_1.SandboxAction {
                     const start = (partNumber - 1) * CHUNK_SIZE;
                     const end = Math.min(start + CHUNK_SIZE, size);
                     const chunk = blob.slice(start, end);
-                    // eslint-disable-next-line @typescript-eslint/no-unsafe-argument
                     batch.push(this.uploadPart(uploadId, partNumber, chunk));
                 }
                 // Wait for batch to complete
-                // eslint-disable-next-line @typescript-eslint/no-unsafe-assignment
                 const batchResults = await Promise.all(batch);
-                // eslint-disable-next-line @typescript-eslint/no-unsafe-call, @typescript-eslint/no-unsafe-member-access, @typescript-eslint/no-unsafe-assignment, @typescript-eslint/no-unsafe-argument
-                parts.push(...batchResults.map((r) => ({ partNumber: r.partNumber, etag: r.etag })));
+                parts.push(...batchResults);
             }
             // Sort parts by partNumber to ensure correct order
-            // eslint-disable-next-line @typescript-eslint/no-unsafe-member-access
             parts.sort((a, b) => (a.partNumber ?? 0) - (b.partNumber ?? 0));
             // Complete the upload
-            // eslint-disable-next-line @typescript-eslint/no-unsafe-argument
             return await this.completeMultipartUpload(uploadId, parts);
         }
         catch (error) {
             // Abort the upload on failure
             try {
-                // eslint-disable-next-line @typescript-eslint/no-unsafe-argument
                 await this.abortMultipartUpload(uploadId);
             }
             catch (abortError) {

package/dist/cjs/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/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;
+    }
+    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/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