@blaxel/core 0.2.2-preview1 → 0.2.2-preview3

package/dist/common/node.js

@@ -10,6 +10,11 @@ exports.fs = fs;
 let os = null;
 exports.os = os;
 if (isNode) {
-    exports.fs = fs = eval("require")("fs");
-    exports.os = os = eval("require")("os");
+    try {
+        exports.fs = fs = eval("require")("fs");
+        exports.os = os = eval("require")("os");
+    }
+    catch (e) {
+        console.warn("fs and os are not available in this environment");
+    }
 }

package/dist/common/settings.js

@@ -75,7 +75,7 @@ class Settings {
         if (!this.generation) {
             return "";
         }
-        return env_js_1.env.BL_RUN_INTERNAL_HOSTNAME || "";
+        return env_js_1.env.BL_RUN_INTERNAL_HOST || "";
     }
     get runInternalProtocol() {
         return env_js_1.env.BL_RUN_INTERNAL_PROTOCOL || "https";

package/dist/jobs/batches.d.ts

@@ -0,0 +1,32 @@
+declare class BlBatch {
+    /**
+     * Lists all remote batch jobs for a given name
+     * @param name - The name of the job to list
+     */
+    list(name: string): void;
+    /**
+     * Creates a new remote batch job configuration.
+     * This is used for remote execution management, not for local execution.
+     * @param name - The name of the job
+     * @param args - Array of argument arrays for each job in the batch
+     * @returns The batch configuration object
+     */
+    create(name: string, args: any[][]): {
+        name: string;
+        args: any[][];
+    };
+    /**
+     * Kills a remote batch job
+     * @param name - The name of the job
+     * @param batchId - The ID of the batch to kill
+     */
+    kill(name: string, batchId: string): void;
+    /**
+     * Retrieves information about a remote batch job
+     * @param name - The name of the job
+     * @param batchId - The ID of the batch to get information for
+     */
+    get(name: string, batchId: string): void;
+}
+export declare const blBatch: BlBatch;
+export {};

package/dist/jobs/batches.js

@@ -0,0 +1,40 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.blBatch = void 0;
+class BlBatch {
+    /**
+     * Lists all remote batch jobs for a given name
+     * @param name - The name of the job to list
+     */
+    list(name) {
+    }
+    /**
+     * Creates a new remote batch job configuration.
+     * This is used for remote execution management, not for local execution.
+     * @param name - The name of the job
+     * @param args - Array of argument arrays for each job in the batch
+     * @returns The batch configuration object
+     */
+    create(name, args) {
+        const batch = {
+            name,
+            args,
+        };
+        return batch;
+    }
+    /**
+     * Kills a remote batch job
+     * @param name - The name of the job
+     * @param batchId - The ID of the batch to kill
+     */
+    kill(name, batchId) {
+    }
+    /**
+     * Retrieves information about a remote batch job
+     * @param name - The name of the job
+     * @param batchId - The ID of the batch to get information for
+     */
+    get(name, batchId) {
+    }
+}
+exports.blBatch = new BlBatch();

package/dist/jobs/job.d.ts

@@ -0,0 +1,33 @@
+declare class BlJob {
+    start(func: (...args: any[]) => Promise<void>): void;
+    /**
+     * Lists all remote batch jobs for a given name
+     * @param name - The name of the job to list
+     */
+    list(name: string): void;
+    /**
+     * Creates a new remote batch job configuration.
+     * This is used for remote execution management, not for local execution.
+     * @param name - The name of the job
+     * @param args - Array of argument arrays for each job in the batch
+     * @returns The batch configuration object
+     */
+    create(name: string, args: any[][]): {
+        name: string;
+        args: any[][];
+    };
+    /**
+     * Kills a remote batch job
+     * @param name - The name of the job
+     * @param batchId - The ID of the batch to kill
+     */
+    kill(name: string, batchId: string): void;
+    /**
+     * Retrieves information about a remote batch job
+     * @param name - The name of the job
+     * @param batchId - The ID of the batch to get information for
+     */
+    get(name: string, batchId: string): void;
+}
+export declare const blJob: BlJob;
+export {};

package/dist/jobs/job.js

@@ -0,0 +1,51 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.blJob = void 0;
+function retrieveArguments() {
+    const args = process.argv.slice(2);
+    return args;
+}
+class BlJob {
+    /*
+      Run a job defined in a function, it's run in the current process
+    */
+    start(func) {
+        const args = retrieveArguments();
+        func(...args);
+    }
+    /**
+     * Lists all remote batch jobs for a given name
+     * @param name - The name of the job to list
+     */
+    list(name) {
+    }
+    /**
+     * Creates a new remote batch job configuration.
+     * This is used for remote execution management, not for local execution.
+     * @param name - The name of the job
+     * @param args - Array of argument arrays for each job in the batch
+     * @returns The batch configuration object
+     */
+    create(name, args) {
+        const batch = {
+            name,
+            args,
+        };
+        return batch;
+    }
+    /**
+     * Kills a remote batch job
+     * @param name - The name of the job
+     * @param batchId - The ID of the batch to kill
+     */
+    kill(name, batchId) {
+    }
+    /**
+     * Retrieves information about a remote batch job
+     * @param name - The name of the job
+     * @param batchId - The ID of the batch to get information for
+     */
+    get(name, batchId) {
+    }
+}
+exports.blJob = new BlJob();

package/dist/jobs/jobs.d.ts

@@ -2,6 +2,7 @@ declare class BlJob {
     getArguments(): Promise<{
         [key: number]: any;
     }>;
+    private parseCommandLineArgs;
     get indexKey(): string;
     get index(): number;
     start(func: (args: any) => Promise<void>): Promise<void>;

package/dist/jobs/jobs.js

@@ -1,25 +1,38 @@
 "use strict";
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.blJob = void 0;
-const yargs_1 = __importDefault(require("yargs"));
-const helpers_1 = require("yargs/helpers");
 const env_js_1 = require("../common/env.js");
 class BlJob {
     async getArguments() {
         if (!env_js_1.env.BL_EXECUTION_DATA_URL) {
-            const argv = await (0, yargs_1.default)((0, helpers_1.hideBin)(process.argv))
-                .parseAsync();
-            return argv;
+            const args = this.parseCommandLineArgs();
+            return args;
         }
         const response = await fetch(env_js_1.env.BL_EXECUTION_DATA_URL);
         const data = await response.json();
         return data.tasks[this.index] ?? {};
     }
+    parseCommandLineArgs() {
+        const args = process.argv.slice(2);
+        const result = {};
+        for (let i = 0; i < args.length; i++) {
+            const arg = args[i];
+            if (arg.startsWith('--')) {
+                const key = arg.slice(2);
+                const value = args[i + 1];
+                if (value && !value.startsWith('--')) {
+                    result[key] = value;
+                    i++;
+                }
+                else {
+                    result[key] = 'true';
+                }
+            }
+        }
+        return result;
+    }
     get indexKey() {
-        return env_js_1.env.BL_EXECUTION_INDEX_KEY ?? "TASK_INDEX";
+        return env_js_1.env.BL_TASK_KEY ?? "TASK_INDEX";
     }
     get index() {
         return env_js_1.env[this.indexKey] ? Number(env_js_1.env[this.indexKey]) ?? 0 : 0;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@blaxel/core",
-  "version": "0.2.2-preview1",
+  "version": "0.2.2-preview3",
   "description": "Blaxel Core SDK for TypeScript",
   "license": "MIT",
   "author": "Blaxel, INC (https://blaxel.ai)",
@@ -62,13 +62,11 @@
     "uuid": "^11.1.0",
     "ws": "^8.18.2",
     "yaml": "^2.7.1",
-    "yargs": "^17.7.2",
     "zod": "^3.24.3"
   },
   "devDependencies": {
     "@eslint/js": "^9.26.0",
     "@types/ws": "^8.18.1",
-    "@types/yargs": "^17.0.33",
     "typescript": "^5.0.0",
     "typescript-eslint": "^8.31.1"
   },

package/dist/mcp/server/http.d.ts

@@ -1,8 +0,0 @@
-import { Transport } from "@modelcontextprotocol/sdk/shared/transport.js";
-import { JSONRPCMessage } from "@modelcontextprotocol/sdk/types.js";
-export declare class BlaxelHttpMcpServerTransport implements Transport {
-    private server;
-    constructor(port: number);
-    start(): Promise<void>;
-    send(msg: JSONRPCMessage): Promise<void>;
-}

package/dist/mcp/server/http.js

@@ -1,16 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.BlaxelHttpMcpServerTransport = void 0;
-const streamableHttp_js_1 = require("@modelcontextprotocol/sdk/server/streamableHttp.js");
-class BlaxelHttpMcpServerTransport {
-    server;
-    constructor(port) {
-        this.server = new streamableHttp_js_1.StreamableHTTPServerTransport({ port });
-    }
-    async start() {
-        await this.server.start();
-    }
-    async send(msg) {
-    }
-}
-exports.BlaxelHttpMcpServerTransport = BlaxelHttpMcpServerTransport;

package/dist/mcp/server/index.d.ts

@@ -1,4 +0,0 @@
-export * from "./http.js";
-export * from "./websocket.js";
-import { BlaxelWebsocketMcpServerTransport } from "./websocket.js";
-export declare const BlaxelMcpServerTransport: typeof BlaxelWebsocketMcpServerTransport;

package/dist/mcp/server/index.js

@@ -1,21 +0,0 @@
-"use strict";
-var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    var desc = Object.getOwnPropertyDescriptor(m, k);
-    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
-      desc = { enumerable: true, get: function() { return m[k]; } };
-    }
-    Object.defineProperty(o, k2, desc);
-}) : (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    o[k2] = m[k];
-}));
-var __exportStar = (this && this.__exportStar) || function(m, exports) {
-    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.BlaxelMcpServerTransport = void 0;
-__exportStar(require("./http.js"), exports);
-__exportStar(require("./websocket.js"), exports);
-const websocket_js_1 = require("./websocket.js");
-exports.BlaxelMcpServerTransport = websocket_js_1.BlaxelWebsocketMcpServerTransport;

package/dist/mcp/server/websocket.d.ts

@@ -1,24 +0,0 @@
-import { Transport } from "@modelcontextprotocol/sdk/shared/transport.js";
-interface JSONRPCMessage {
-    jsonrpc: "2.0";
-    id?: string | number;
-    method?: string;
-    params?: Record<string, unknown>;
-}
-export declare class BlaxelWebsocketMcpServerTransport implements Transport {
-    private port;
-    private wss;
-    private clients;
-    onclose?: () => void;
-    onerror?: (err: Error) => void;
-    private messageHandler?;
-    onconnection?: (clientId: string) => void;
-    ondisconnection?: (clientId: string) => void;
-    set onmessage(handler: ((message: JSONRPCMessage) => void) | undefined);
-    constructor(port?: number);
-    start(): Promise<void>;
-    send(msg: JSONRPCMessage): Promise<void>;
-    broadcast(msg: JSONRPCMessage): Promise<void>;
-    close(): Promise<void>;
-}
-export {};

package/dist/mcp/server/websocket.js

@@ -1,213 +0,0 @@
-"use strict";
-var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    var desc = Object.getOwnPropertyDescriptor(m, k);
-    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
-      desc = { enumerable: true, get: function() { return m[k]; } };
-    }
-    Object.defineProperty(o, k2, desc);
-}) : (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    o[k2] = m[k];
-}));
-var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
-    Object.defineProperty(o, "default", { enumerable: true, value: v });
-}) : function(o, v) {
-    o["default"] = v;
-});
-var __importStar = (this && this.__importStar) || (function () {
-    var ownKeys = function(o) {
-        ownKeys = Object.getOwnPropertyNames || function (o) {
-            var ar = [];
-            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
-            return ar;
-        };
-        return ownKeys(o);
-    };
-    return function (mod) {
-        if (mod && mod.__esModule) return mod;
-        var result = {};
-        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
-        __setModuleDefault(result, mod);
-        return result;
-    };
-})();
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.BlaxelWebsocketMcpServerTransport = void 0;
-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 BlaxelWebsocketMcpServerTransport {
-    port;
-    wss;
-    clients = new Map();
-    onclose;
-    onerror;
-    messageHandler;
-    onconnection;
-    ondisconnection;
-    set onmessage(handler) {
-        this.messageHandler = handler
-            ? (msg, clientId) => {
-                if (!("id" in msg)) {
-                    return handler(msg);
-                }
-                return handler({
-                    ...msg,
-                    id: clientId + ":" + msg.id,
-                });
-            }
-            : undefined;
-    }
-    constructor(port) {
-        this.port = port ?? parseInt(env_js_1.env.BL_SERVER_PORT ?? "8080", 10);
-        this.wss = new ws_1.WebSocketServer({ port: this.port });
-    }
-    async start() {
-        logger_js_1.logger.info("Starting WebSocket Server on port " + this.port);
-        this.wss.on("connection", (ws) => {
-            const clientId = (0, uuid_1.v4)();
-            this.clients.set(clientId, {
-                ws,
-            });
-            this.onconnection?.(clientId);
-            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());
-                    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(":");
-                    msg.id = parsedMsgId ? parseInt(parsedMsgId) : undefined;
-                    // Use optional chaining for safe access
-                    const client = this.clients.get(cId ?? "");
-                    if (client?.ws?.readyState === ws_1.default.OPEN) {
-                        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);
-                        this.ondisconnection?.(cId);
-                    }
-                }
-                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", () => {
-                this.clients.delete(clientId);
-                this.ondisconnection?.(clientId);
-            });
-            ws.on("error", (err) => {
-                this.onerror?.(err);
-            });
-        });
-        return Promise.resolve();
-    }
-    async send(msg) {
-        const [cId, msgId] = msg.id ? String(msg.id).split(":") : [];
-        msg.id = parseInt(msgId);
-        const data = JSON.stringify(msg);
-        const deadClients = [];
-        if (cId) {
-            // Send to specific client
-            const client = this.clients.get(cId);
-            if (client?.ws?.readyState === ws_1.default.OPEN) {
-                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);
-                this.ondisconnection?.(cId);
-            }
-        }
-        for (const [id, client] of this.clients.entries()) {
-            if (client.ws.readyState !== ws_1.default.OPEN) {
-                deadClients.push(id);
-            }
-        }
-        // Cleanup dead clients
-        deadClients.forEach((id) => {
-            this.clients.delete(id);
-            this.ondisconnection?.(id);
-        });
-        return Promise.resolve();
-    }
-    async broadcast(msg) {
-        return this.send(msg);
-    }
-    async close() {
-        return new Promise((resolve) => {
-            this.wss.close(() => {
-                this.clients.clear();
-                resolve();
-            });
-        });
-    }
-}
-exports.BlaxelWebsocketMcpServerTransport = BlaxelWebsocketMcpServerTransport;