@blaxel/core 0.2.49-dev.213 → 0.2.49-dev1

package/dist/jobs/index.d.ts

@@ -0,0 +1,3 @@
+export * from "./jobs.js";
+export * from "./start.js";
+export * from "./types.js";

package/dist/jobs/index.js

@@ -0,0 +1,19 @@
+"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 });
+__exportStar(require("./jobs.js"), exports);
+__exportStar(require("./start.js"), exports);
+__exportStar(require("./types.js"), exports);

package/dist/jobs/jobs.d.ts

@@ -0,0 +1,13 @@
+declare class BlJob {
+    jobName: string;
+    constructor(jobName: string);
+    get fallbackUrl(): import("url").URL | null;
+    get externalUrl(): import("url").URL;
+    get internalUrl(): import("url").URL;
+    get forcedUrl(): import("url").URL | null;
+    get url(): import("url").URL;
+    call(url: URL, tasks: Record<string, unknown>[]): Promise<Response>;
+    run(tasks: Record<string, unknown>[]): Promise<string>;
+}
+export declare const blJob: (jobName: string) => BlJob;
+export {};

package/dist/jobs/jobs.js

@@ -0,0 +1,90 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.blJob = void 0;
+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 BlJob {
+    jobName;
+    constructor(jobName) {
+        this.jobName = jobName;
+    }
+    get fallbackUrl() {
+        if (this.externalUrl != this.url) {
+            return this.externalUrl;
+        }
+        return null;
+    }
+    get externalUrl() {
+        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}`);
+    }
+    get forcedUrl() {
+        return (0, internal_js_1.getForcedUrl)('job', this.jobName);
+    }
+    get url() {
+        if (this.forcedUrl)
+            return this.forcedUrl;
+        if (settings_js_1.settings.runInternalHostname)
+            return this.internalUrl;
+        return this.externalUrl;
+    }
+    async call(url, tasks) {
+        const body = {
+            tasks: tasks
+        };
+        const response = await fetch(url.toString() + "/executions", {
+            method: "POST",
+            headers: {
+                ...settings_js_1.settings.headers,
+                "Content-Type": "application/json",
+            },
+            body: JSON.stringify(body),
+        });
+        return response;
+    }
+    async run(tasks) {
+        logger_js_1.logger.debug(`Job Calling: ${this.jobName}`);
+        const span = (0, telemetry_js_1.startSpan)(this.jobName, {
+            attributes: {
+                "job.name": this.jobName,
+                "span.type": "job.run",
+            },
+            isRoot: false,
+        });
+        try {
+            const response = await this.call(this.url, tasks);
+            return await response.text();
+        }
+        catch (err) {
+            if (err instanceof Error) {
+                if (!this.fallbackUrl) {
+                    span.setAttribute("job.run.error", err.stack);
+                    throw err;
+                }
+                try {
+                    const response = await this.call(this.fallbackUrl, tasks);
+                    return await response.text();
+                }
+                catch (err) {
+                    if (err instanceof Error) {
+                        span.setAttribute("job.run.error", err.stack);
+                    }
+                    throw err;
+                }
+            }
+            throw err;
+        }
+        finally {
+            span.end();
+        }
+    }
+}
+const blJob = (jobName) => {
+    return new BlJob(jobName);
+};
+exports.blJob = blJob;

package/dist/jobs/start.d.ts

@@ -0,0 +1 @@
+export declare const blStartJob: (func: (args: any) => Promise<void>) => void;

package/dist/jobs/start.js

@@ -0,0 +1,66 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.blStartJob = void 0;
+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 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 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_TASK_KEY ?? "TASK_INDEX";
+    }
+    get index() {
+        return env_js_1.env[this.indexKey] ? Number(env_js_1.env[this.indexKey]) : 0;
+    }
+    /*
+      Run a job defined in a function, it's run in the current process
+    */
+    async start(func) {
+        await (0, autoload_js_1.authenticate)();
+        const parsedArgs = await this.getArguments();
+        await func(parsedArgs);
+    }
+}
+const blStartJob = (func) => {
+    const job = new BlJobWrapper();
+    job.start(func).then(async () => {
+        try {
+            await (0, telemetry_js_1.flush)();
+        }
+        catch (error) {
+            console.error('Error flushing telemetry:', error);
+        }
+        process.exit(0);
+    }).catch((error) => {
+        console.error('Job execution failed:', error);
+        process.exit(1);
+    });
+};
+exports.blStartJob = blStartJob;

package/dist/jobs/types.d.ts

@@ -0,0 +1,3 @@
+export type ExecutionArgs = {
+    [key: number]: any;
+}[];

package/dist/jobs/types.js

@@ -0,0 +1,2 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });

package/dist/mcp/client.d.ts

@@ -0,0 +1,28 @@
+import { Transport } from "@modelcontextprotocol/sdk/shared/transport.js";
+import { JSONRPCMessage } from "@modelcontextprotocol/sdk/types.js";
+/**
+ * Client transport for WebSocket: this will connect to a server over the WebSocket protocol.
+ * Works in both browser and Node.js environments.
+ */
+export declare class BlaxelMcpClientTransport implements Transport {
+    private _socket?;
+    private _url;
+    private _headers;
+    private _isBrowser;
+    private _retry_max;
+    private _retry_delay;
+    onclose?: () => void;
+    onerror?: (error: Error) => void;
+    onmessage?: (message: JSONRPCMessage) => void;
+    constructor(url: string, headers?: Record<string, string>, options?: {
+        retry: {
+            max?: number;
+            delay?: number;
+        };
+    });
+    start(): Promise<void>;
+    private _connect;
+    get isConnected(): boolean;
+    close(): Promise<void>;
+    send(message: JSONRPCMessage): Promise<void>;
+}

package/dist/mcp/client.js

@@ -0,0 +1,247 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.BlaxelMcpClientTransport = void 0;
+const types_js_1 = require("@modelcontextprotocol/sdk/types.js");
+const logger_js_1 = require("../common/logger.js");
+const settings_js_1 = require("../common/settings.js");
+// Detect environment
+// eslint-disable-next-line @typescript-eslint/no-unsafe-member-access
+const isBrowser = typeof globalThis !== "undefined" && globalThis.window !== undefined;
+// Conditional import for Node.js WebSocket
+let NodeWebSocket;
+if (!isBrowser) {
+    try {
+        // Dynamic import for Node.js environment
+        // eslint-disable-next-line @typescript-eslint/no-unsafe-assignment, @typescript-eslint/no-require-imports
+        NodeWebSocket = require("ws");
+    }
+    catch {
+        console.warn("ws is not available in this environment");
+        // ws is not available
+    }
+}
+//const SUBPROTOCOL = "mcp";
+// Helper function to wait
+const delay = (ms) => new Promise((resolve) => setTimeout(resolve, ms));
+/**
+ * Client transport for WebSocket: this will connect to a server over the WebSocket protocol.
+ * Works in both browser and Node.js environments.
+ */
+class BlaxelMcpClientTransport {
+    _socket;
+    _url;
+    _headers;
+    _isBrowser;
+    _retry_max;
+    _retry_delay;
+    onclose;
+    onerror;
+    onmessage;
+    constructor(url, headers, options) {
+        this._url = new URL(url.replace("http", "ws"));
+        this._retry_max = options?.retry?.max ?? 3;
+        this._retry_delay = options?.retry?.delay ?? 1000;
+        this._headers = headers ?? {};
+        this._isBrowser = isBrowser;
+    }
+    async start() {
+        if (this._socket) {
+            throw new Error("Blaxel already started! If using Client class, note that connect() calls start() automatically.");
+        }
+        let attempts = 0;
+        const maxAttempts = Math.max(1, this._retry_max + 1); // Ensure at least 1 attempt
+        while (attempts < maxAttempts) {
+            try {
+                await this._connect();
+                return;
+            }
+            catch (error) {
+                attempts++;
+                if (error instanceof Error) {
+                    logger_js_1.logger.warn(error.stack ?? error.message);
+                }
+                if (attempts >= maxAttempts) {
+                    throw error;
+                }
+                logger_js_1.logger.debug(`WebSocket connection attempt ${attempts} failed, retrying in ${this._retry_delay}ms...`);
+                await delay(this._retry_delay);
+            }
+        }
+    }
+    _connect() {
+        // eslint-disable-next-line @typescript-eslint/no-misused-promises
+        return new Promise((resolve, reject) => {
+            try {
+                if (this._isBrowser) {
+                    // Use native browser WebSocket
+                    const url = `${this._url.toString()}?token=${settings_js_1.settings.token}`;
+                    this._socket = new WebSocket(url);
+                }
+                else {
+                    // Use Node.js WebSocket
+                    if (!NodeWebSocket) {
+                        throw new Error("WebSocket library not available in Node.js environment");
+                    }
+                    // eslint-disable-next-line @typescript-eslint/no-unsafe-call
+                    this._socket = new NodeWebSocket(this._url, {
+                        //protocols: SUBPROTOCOL,
+                        headers: this._headers,
+                    });
+                }
+                this._socket.onerror = (event) => {
+                    // Log websocket error with meaningful information instead of raw event
+                    const errorInfo = {
+                        type: 'WebSocket Error',
+                        url: this._url.toString(),
+                        readyState: this._socket?.readyState,
+                        browser: this._isBrowser,
+                        // Extract any available error details from the event
+                        eventType: event && typeof event === 'object' && 'type' in event
+                            ? String(event.type)
+                            : 'unknown',
+                        // Browser events might have different properties than Node.js
+                        message: this._isBrowser && event && typeof event === 'object' && 'message' in event
+                            ? String(event.message)
+                            : undefined,
+                        error: !this._isBrowser && event && typeof event === 'object' && 'error' in event
+                            ? String(event.error)
+                            : undefined
+                    };
+                    logger_js_1.logger.error('WebSocket connection error', errorInfo);
+                    const error = this._isBrowser
+                        // eslint-disable-next-line @typescript-eslint/no-unsafe-member-access
+                        ? new Error(`WebSocket error: ${event.message}`)
+                        : "error" in event
+                            // eslint-disable-next-line @typescript-eslint/no-unsafe-member-access
+                            ? event.error
+                            // eslint-disable-next-line @typescript-eslint/no-unsafe-member-access
+                            : new Error(`WebSocket error: ${event.message}`);
+                    reject(error);
+                    this.onerror?.(error);
+                };
+                this._socket.onopen = () => {
+                    resolve();
+                };
+                this._socket.onclose = () => {
+                    this.onclose?.();
+                    this._socket = undefined;
+                };
+                this._socket.onmessage = (event) => {
+                    let message;
+                    try {
+                        let dataString;
+                        if (this._isBrowser) {
+                            // Browser WebSocket MessageEvent
+                            const browserEvent = event;
+                            // eslint-disable-next-line @typescript-eslint/no-unsafe-assignment
+                            dataString = typeof browserEvent.data === "string"
+                                ? browserEvent.data
+                                // eslint-disable-next-line @typescript-eslint/no-unsafe-member-access, @typescript-eslint/no-unsafe-call
+                                : browserEvent.data.toString();
+                        }
+                        else {
+                            // Node.js WebSocket MessageEvent
+                            const nodeEvent = event;
+                            if (typeof nodeEvent.data === "string") {
+                                dataString = nodeEvent.data;
+                            }
+                            else if (nodeEvent.data instanceof Buffer) {
+                                dataString = nodeEvent.data.toString("utf-8");
+                            }
+                            else {
+                                throw new Error("Unsupported data type for event.data");
+                            }
+                        }
+                        message = types_js_1.JSONRPCMessageSchema.parse(JSON.parse(dataString));
+                    }
+                    catch (error) {
+                        logger_js_1.logger.error(`Error parsing message: ${
+                        // eslint-disable-next-line @typescript-eslint/no-unsafe-member-access
+                        typeof event.data === "object"
+                            // eslint-disable-next-line @typescript-eslint/no-unsafe-member-access
+                            ? JSON.stringify(event.data)
+                            // eslint-disable-next-line @typescript-eslint/no-unsafe-member-access
+                            : event.data}`);
+                        this.onerror?.(error);
+                        return;
+                    }
+                    this.onmessage?.(message);
+                };
+            }
+            catch (error) {
+                if (error instanceof Error && error.message.includes("ws does not work in the browser")) {
+                    this._isBrowser = true;
+                    return this._connect().then(resolve).catch(reject);
+                }
+                reject(error);
+            }
+        });
+    }
+    get isConnected() {
+        if (!this._socket)
+            return false;
+        if (this._isBrowser) {
+            return this._socket.readyState === 1; // WebSocket.OPEN = 1
+        }
+        else {
+            return this._socket.readyState === 1; // WebSocket.OPEN = 1
+        }
+    }
+    async close() {
+        this._socket?.close();
+        this._socket = undefined;
+        this.onclose?.();
+        return Promise.resolve();
+    }
+    async send(message) {
+        let attempts = 0;
+        const maxAttempts = Math.max(1, this._retry_max + 1); // Ensure at least 1 attempt
+        while (attempts < maxAttempts) {
+            try {
+                if (!this._socket || !this.isConnected) {
+                    if (!this._socket) {
+                        // Only try to start if socket doesn't exist
+                        await this.start();
+                    }
+                    else {
+                        throw new Error("WebSocket is not in OPEN state");
+                    }
+                }
+                await new Promise((resolve, reject) => {
+                    try {
+                        const messageStr = JSON.stringify(message);
+                        if (this._isBrowser) {
+                            // Browser WebSocket
+                            this._socket?.send(messageStr);
+                            resolve();
+                        }
+                        else {
+                            // Node.js WebSocket
+                            this._socket?.send(messageStr, (error) => {
+                                if (error) {
+                                    reject(error);
+                                }
+                                else {
+                                    resolve();
+                                }
+                            });
+                        }
+                    }
+                    catch (error) {
+                        reject(error);
+                    }
+                });
+                return;
+            }
+            catch (error) {
+                attempts++;
+                if (attempts >= maxAttempts) {
+                    throw error;
+                }
+                logger_js_1.logger.warn(`WebSocket send attempt ${attempts} failed, retrying in ${this._retry_delay}ms...`);
+                await delay(this._retry_delay);
+            }
+        }
+    }
+}
+exports.BlaxelMcpClientTransport = BlaxelMcpClientTransport;

package/dist/{cjs-browser/types/sandbox/websocket → mcp}/index.d.ts

@@ -1 +1,2 @@
 export * from "./client.js";
+export * from "./server.js";

package/dist/{cjs-browser/sandbox/websocket → mcp}/index.js

@@ -15,3 +15,4 @@ var __exportStar = (this && this.__exportStar) || function(m, exports) {
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 __exportStar(require("./client.js"), exports);
+__exportStar(require("./server.js"), exports);

package/dist/mcp/server.d.ts

@@ -0,0 +1,24 @@
+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 BlaxelMcpServerTransport 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 {};