@blaxel/core 0.2.6-dev.74 → 0.2.6-preview.5

package/dist/authentication/clientcredentials.d.ts

@@ -9,8 +9,6 @@ export declare class ClientCredentials extends Credentials {
     get workspace(): string;
     needRefresh(): boolean;
     authenticate(): Promise<void>;
-    private sleep;
-    processWithRetry(retry?: number): Promise<void>;
     process(): Promise<void>;
     get authorization(): string;
     get token(): string;

package/dist/authentication/clientcredentials.js

@@ -42,24 +42,9 @@ class ClientCredentials extends credentials_js_1.Credentials {
         if (!this.needRefresh()) {
             return this.currentPromise || Promise.resolve();
         }
-        this.currentPromise = this.processWithRetry();
+        this.currentPromise = this.process();
         return this.currentPromise;
     }
-    sleep(ms) {
-        return new Promise((resolve) => setTimeout(resolve, ms));
-    }
-    async processWithRetry(retry = 3) {
-        try {
-            return await this.process();
-        }
-        catch (error) {
-            if (retry > 0) {
-                await this.sleep(1000);
-                return this.processWithRetry(retry - 1);
-            }
-            throw error;
-        }
-    }
     async process() {
         const response = await (0, authentication_js_1.oauthToken)({
             headers: {

package/dist/client/sdk.gen.d.ts

@@ -1,5 +1,5 @@
 import type { Options as ClientOptions, TDataShape, Client } from '@hey-api/client-fetch';
-import type { ListAgentsData, CreateAgentData, DeleteAgentData, GetAgentData, UpdateAgentData, ListAgentRevisionsData, GetConfigurationData, ListFunctionsData, CreateFunctionData, DeleteFunctionData, GetFunctionData, UpdateFunctionData, ListFunctionRevisionsData, GetIntegrationData, ListIntegrationConnectionsData, CreateIntegrationConnectionData, DeleteIntegrationConnectionData, GetIntegrationConnectionData, UpdateIntegrationConnectionData, GetIntegrationConnectionModelEndpointConfigurationsData, ListIntegrationConnectionModelsData, GetIntegrationConnectionModelData, ListJobsData, CreateJobData, DeleteJobData, GetJobData, UpdateJobData, ListJobRevisionsData, ListKnowledgebasesData, CreateKnowledgebaseData, DeleteKnowledgebaseData, GetKnowledgebaseData, UpdateKnowledgebaseData, ListKnowledgebaseRevisionsData, ListLocationsData, ListMcpHubDefinitionsData, ListModelsData, CreateModelData, DeleteModelData, GetModelData, UpdateModelData, ListModelRevisionsData, ListPoliciesData, CreatePolicyData, DeletePolicyData, GetPolicyData, UpdatePolicyData, ListPrivateClustersData, CreatePrivateClusterData, DeletePrivateClusterData, GetPrivateClusterData, UpdatePrivateClusterData, GetPrivateClusterHealthData, UpdatePrivateClusterHealthData, ListAllPendingInvitationsData, ListSandboxHubDefinitionsData, ListSandboxesData, CreateSandboxData, DeleteSandboxData, GetSandboxData, UpdateSandboxData, ListSandboxPreviewsData, CreateSandboxPreviewData, DeleteSandboxPreviewData, GetSandboxPreviewData, UpdateSandboxPreviewData, ListSandboxPreviewTokensData, CreateSandboxPreviewTokenData, DeleteSandboxPreviewTokenData, StartSandboxData, StopSandboxData, GetWorkspaceServiceAccountsData, CreateWorkspaceServiceAccountData, DeleteWorkspaceServiceAccountData, UpdateWorkspaceServiceAccountData, ListApiKeysForServiceAccountData, CreateApiKeyForServiceAccountData, DeleteApiKeyForServiceAccountData, ListTemplatesData, GetTemplateData, GetTemplateContentsData, GetTemplateFileContentsData, ListWorkspaceUsersData, InviteWorkspaceUserData, RemoveWorkspaceUserData, UpdateWorkspaceUserRoleData, ListWorkspacesData, CreateWorspaceData, DeleteWorkspaceData, GetWorkspaceData, UpdateWorkspaceData, DeclineWorkspaceInvitationData, AcceptWorkspaceInvitationData, LeaveWorkspaceData, CheckWorkspaceAvailabilityData } from './types.gen.js';
+import type { ListAgentsData, CreateAgentData, DeleteAgentData, GetAgentData, UpdateAgentData, ListAgentRevisionsData, GetConfigurationData, ListFunctionsData, CreateFunctionData, DeleteFunctionData, GetFunctionData, UpdateFunctionData, ListFunctionRevisionsData, GetIntegrationData, ListIntegrationConnectionsData, CreateIntegrationConnectionData, DeleteIntegrationConnectionData, GetIntegrationConnectionData, UpdateIntegrationConnectionData, GetIntegrationConnectionModelEndpointConfigurationsData, ListIntegrationConnectionModelsData, GetIntegrationConnectionModelData, ListJobsData, CreateJobData, DeleteJobData, GetJobData, UpdateJobData, ListJobRevisionsData, ListKnowledgebasesData, CreateKnowledgebaseData, DeleteKnowledgebaseData, GetKnowledgebaseData, UpdateKnowledgebaseData, ListKnowledgebaseRevisionsData, ListLocationsData, ListMcpHubDefinitionsData, ListModelsData, CreateModelData, DeleteModelData, GetModelData, UpdateModelData, ListModelRevisionsData, ListPoliciesData, CreatePolicyData, DeletePolicyData, GetPolicyData, UpdatePolicyData, ListPrivateClustersData, CreatePrivateClusterData, DeletePrivateClusterData, GetPrivateClusterData, UpdatePrivateClusterData, GetPrivateClusterHealthData, UpdatePrivateClusterHealthData, ListAllPendingInvitationsData, ListSandboxHubDefinitionsData, ListSandboxesData, CreateSandboxData, DeleteSandboxData, GetSandboxData, UpdateSandboxData, ListSandboxPreviewsData, CreateSandboxPreviewData, DeleteSandboxPreviewData, GetSandboxPreviewData, UpdateSandboxPreviewData, ListSandboxPreviewTokensData, CreateSandboxPreviewTokenData, DeleteSandboxPreviewTokenData, StartSandboxData, StopSandboxData, GetWorkspaceServiceAccountsData, CreateWorkspaceServiceAccountData, DeleteWorkspaceServiceAccountData, UpdateWorkspaceServiceAccountData, ListApiKeysForServiceAccountData, CreateApiKeyForServiceAccountData, DeleteApiKeyForServiceAccountData, ListTemplatesData, GetTemplateData, ListWorkspaceUsersData, InviteWorkspaceUserData, RemoveWorkspaceUserData, UpdateWorkspaceUserRoleData, ListWorkspacesData, CreateWorspaceData, DeleteWorkspaceData, GetWorkspaceData, UpdateWorkspaceData, DeclineWorkspaceInvitationData, AcceptWorkspaceInvitationData, LeaveWorkspaceData, CheckWorkspaceAvailabilityData } from './types.gen.js';
 export type Options<TData extends TDataShape = TDataShape, ThrowOnError extends boolean = boolean> = ClientOptions<TData, ThrowOnError> & {
     /**
      * You can provide a client instance returned by `createClient()` instead of
@@ -413,8 +413,6 @@ export declare const listTemplates: <ThrowOnError extends boolean = false>(optio
  * Returns a template by name.
  */
 export declare const getTemplate: <ThrowOnError extends boolean = false>(options: Options<GetTemplateData, ThrowOnError>) => import("@hey-api/client-fetch").RequestResult<import("./types.gen.js").Template, unknown, ThrowOnError>;
-export declare const getTemplateContents: <ThrowOnError extends boolean = false>(options: Options<GetTemplateContentsData, ThrowOnError>) => import("@hey-api/client-fetch").RequestResult<string[], unknown, ThrowOnError>;
-export declare const getTemplateFileContents: <ThrowOnError extends boolean = false>(options: Options<GetTemplateFileContentsData, ThrowOnError>) => import("@hey-api/client-fetch").RequestResult<string, unknown, ThrowOnError>;
 /**
  * List users in workspace
  * Returns a list of all users in the workspace.

package/dist/client/sdk.gen.js

@@ -2,7 +2,7 @@
 // This file is auto-generated by @hey-api/openapi-ts
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.deletePrivateCluster = exports.createPrivateCluster = exports.listPrivateClusters = exports.updatePolicy = exports.getPolicy = exports.deletePolicy = exports.createPolicy = exports.listPolicies = exports.listModelRevisions = exports.updateModel = exports.getModel = exports.deleteModel = exports.createModel = exports.listModels = exports.listMcpHubDefinitions = exports.listLocations = exports.listKnowledgebaseRevisions = exports.updateKnowledgebase = exports.getKnowledgebase = exports.deleteKnowledgebase = exports.createKnowledgebase = exports.listKnowledgebases = exports.listJobRevisions = exports.updateJob = exports.getJob = exports.deleteJob = exports.createJob = exports.listJobs = exports.getIntegrationConnectionModel = exports.listIntegrationConnectionModels = exports.getIntegrationConnectionModelEndpointConfigurations = exports.updateIntegrationConnection = exports.getIntegrationConnection = exports.deleteIntegrationConnection = exports.createIntegrationConnection = exports.listIntegrationConnections = exports.getIntegration = exports.listFunctionRevisions = exports.updateFunction = exports.getFunction = exports.deleteFunction = exports.createFunction = exports.listFunctions = exports.getConfiguration = exports.listAgentRevisions = exports.updateAgent = exports.getAgent = exports.deleteAgent = exports.createAgent = exports.listAgents = void 0;
-exports.checkWorkspaceAvailability = exports.leaveWorkspace = exports.acceptWorkspaceInvitation = exports.declineWorkspaceInvitation = exports.updateWorkspace = exports.getWorkspace = exports.deleteWorkspace = exports.createWorspace = exports.listWorkspaces = exports.updateWorkspaceUserRole = exports.removeWorkspaceUser = exports.inviteWorkspaceUser = exports.listWorkspaceUsers = exports.getTemplateFileContents = exports.getTemplateContents = exports.getTemplate = exports.listTemplates = exports.deleteApiKeyForServiceAccount = exports.createApiKeyForServiceAccount = exports.listApiKeysForServiceAccount = exports.updateWorkspaceServiceAccount = exports.deleteWorkspaceServiceAccount = exports.createWorkspaceServiceAccount = exports.getWorkspaceServiceAccounts = exports.stopSandbox = exports.startSandbox = exports.deleteSandboxPreviewToken = exports.createSandboxPreviewToken = exports.listSandboxPreviewTokens = exports.updateSandboxPreview = exports.getSandboxPreview = exports.deleteSandboxPreview = exports.createSandboxPreview = exports.listSandboxPreviews = exports.updateSandbox = exports.getSandbox = exports.deleteSandbox = exports.createSandbox = exports.listSandboxes = exports.listSandboxHubDefinitions = exports.listAllPendingInvitations = exports.updatePrivateClusterHealth = exports.getPrivateClusterHealth = exports.updatePrivateCluster = exports.getPrivateCluster = void 0;
+exports.checkWorkspaceAvailability = exports.leaveWorkspace = exports.acceptWorkspaceInvitation = exports.declineWorkspaceInvitation = exports.updateWorkspace = exports.getWorkspace = exports.deleteWorkspace = exports.createWorspace = exports.listWorkspaces = exports.updateWorkspaceUserRole = exports.removeWorkspaceUser = exports.inviteWorkspaceUser = exports.listWorkspaceUsers = exports.getTemplate = exports.listTemplates = exports.deleteApiKeyForServiceAccount = exports.createApiKeyForServiceAccount = exports.listApiKeysForServiceAccount = exports.updateWorkspaceServiceAccount = exports.deleteWorkspaceServiceAccount = exports.createWorkspaceServiceAccount = exports.getWorkspaceServiceAccounts = exports.stopSandbox = exports.startSandbox = exports.deleteSandboxPreviewToken = exports.createSandboxPreviewToken = exports.listSandboxPreviewTokens = exports.updateSandboxPreview = exports.getSandboxPreview = exports.deleteSandboxPreview = exports.createSandboxPreview = exports.listSandboxPreviews = exports.updateSandbox = exports.getSandbox = exports.deleteSandbox = exports.createSandbox = exports.listSandboxes = exports.listSandboxHubDefinitions = exports.listAllPendingInvitations = exports.updatePrivateClusterHealth = exports.getPrivateClusterHealth = exports.updatePrivateCluster = exports.getPrivateCluster = void 0;
 const client_gen_1 = require("./client.gen");
 /**
  * List all agents
@@ -1425,32 +1425,6 @@ const getTemplate = (options) => {
     });
 };
 exports.getTemplate = getTemplate;
-const getTemplateContents = (options) => {
-    return (options.client ?? client_gen_1.client).get({
-        security: [
-            {
-                scheme: 'bearer',
-                type: 'http'
-            }
-        ],
-        url: '/templates/{templateName}/contents',
-        ...options
-    });
-};
-exports.getTemplateContents = getTemplateContents;
-const getTemplateFileContents = (options) => {
-    return (options.client ?? client_gen_1.client).get({
-        security: [
-            {
-                scheme: 'bearer',
-                type: 'http'
-            }
-        ],
-        url: '/templates/{templateName}/contents/{fileName}',
-        ...options
-    });
-};
-exports.getTemplateFileContents = getTemplateFileContents;
 /**
  * List users in workspace
  * Returns a list of all users in the workspace.

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

@@ -651,10 +651,183 @@ export type Job = {
      */
     status?: string;
 };
+/**
+ * Configuration for a job execution
+ */
+export type JobExecutionConfig = {
+    /**
+     * The maximum number of concurrent tasks for an execution
+     */
+    maxConcurrentTasks?: number;
+    /**
+     * The maximum number of retries for the job execution
+     */
+    maxRetries?: number;
+    /**
+     * The timeout for the job execution in seconds
+     */
+    timeout?: number;
+};
+/**
+ * Metrics for job
+ */
+export type JobMetrics = {
+    /**
+     * Billable time
+     */
+    billableTime?: Array<unknown>;
+    /**
+     * CPU usage
+     */
+    cpuUsage?: Array<unknown>;
+    /**
+     * Executions chart
+     */
+    executionsChart?: {
+        [key: string]: unknown;
+    };
+    /**
+     * Executions running
+     */
+    executionsRunning?: Array<unknown>;
+    /**
+     * Total executions
+     */
+    executionsTotal?: {
+        [key: string]: unknown;
+    };
+    /**
+     * RAM usage
+     */
+    ramUsage?: Array<unknown>;
+    /**
+     * Tasks chart
+     */
+    tasksChart?: {
+        [key: string]: unknown;
+    };
+    /**
+     * Tasks running
+     */
+    tasksRunning?: Array<unknown>;
+    /**
+     * Total tasks
+     */
+    tasksTotal?: {
+        [key: string]: unknown;
+    };
+};
 /**
  * Job specification
  */
 export type JobSpec = CoreSpec & unknown;
+/**
+ * Jobs chart
+ */
+export type JobsChart = {
+    failed?: JobsChartValue;
+    success?: JobsChartValue;
+};
+/**
+ * Jobs CPU usage
+ */
+export type JobsChartValue = {
+    /**
+     * Metric timestamp
+     */
+    timestamp?: string;
+    /**
+     * Metric value
+     */
+    value?: number;
+};
+/**
+ * Jobs executions
+ */
+export type JobsExecutions = {
+    /**
+     * Failed executions
+     */
+    failed?: number;
+    /**
+     * Running executions
+     */
+    running?: number;
+    /**
+     * Success executions
+     */
+    success?: number;
+    /**
+     * Total executions
+     */
+    total?: number;
+};
+/**
+ * Jobs chart
+ */
+export type JobsNetworkChart = {
+    received?: JobsChartValue;
+    sent?: JobsChartValue;
+};
+/**
+ * Jobs chart
+ */
+export type JobsSuccessFailedChart = {
+    failed?: JobsChartValue;
+    retried?: JobsChartValue;
+    success?: JobsChartValue;
+    /**
+     * Metric timestamp
+     */
+    timestamp?: string;
+    total?: JobsChartValue;
+};
+/**
+ * Jobs tasks
+ */
+export type JobsTasks = {
+    /**
+     * Failed executions
+     */
+    failed?: number;
+    /**
+     * Running executions
+     */
+    running?: number;
+    /**
+     * Success executions
+     */
+    success?: number;
+    /**
+     * Total executions
+     */
+    total?: number;
+};
+/**
+ * Jobs executions
+ */
+export type JobsTotal = {
+    /**
+     * Failed executions
+     */
+    failed?: number;
+    /**
+     * Retried executions
+     */
+    retried?: number;
+    /**
+     * Running executions
+     */
+    running?: number;
+    /**
+     * Success executions
+     */
+    success?: number;
+    /**
+     * Total executions
+     */
+    total?: number;
+};
 /**
  * Knowledgebase
  */
@@ -1242,6 +1415,18 @@ export type PreviewSpec = {
      * Whether the preview is public
      */
     public?: boolean;
+    /**
+     * Those headers will be set in all requests to your preview. This is especially useful to set the Authorization header.
+     */
+    requestHeaders?: {
+        [key: string]: string;
+    };
+    /**
+     * Those headers will be set in all responses of your preview. This is especially useful to set the CORS headers.
+     */
+    responseHeaders?: {
+        [key: string]: string;
+    };
     /**
      * URL of the preview
      */
@@ -1640,6 +1825,10 @@ export type Runtime = {
      * The Docker image for the deployment
      */
     image?: string;
+    /**
+     * The maximum number of concurrent task for an execution
+     */
+    maxConcurrentTasks?: number;
     /**
      * The maximum number of retries for the deployment
      */
@@ -3743,37 +3932,6 @@ export type GetTemplateResponses = {
     200: Template;
 };
 export type GetTemplateResponse = GetTemplateResponses[keyof GetTemplateResponses];
-export type GetTemplateContentsData = {
-    body?: never;
-    path: {
-        templateName: string;
-    };
-    query?: never;
-    url: '/templates/{templateName}/contents';
-};
-export type GetTemplateContentsResponses = {
-    /**
-     * List of files in the template
-     */
-    200: Array<string>;
-};
-export type GetTemplateContentsResponse = GetTemplateContentsResponses[keyof GetTemplateContentsResponses];
-export type GetTemplateFileContentsData = {
-    body?: never;
-    path: {
-        templateName: string;
-        fileName: string;
-    };
-    query?: never;
-    url: '/templates/{templateName}/contents/{fileName}';
-};
-export type GetTemplateFileContentsResponses = {
-    /**
-     * Content of the file
-     */
-    200: string;
-};
-export type GetTemplateFileContentsResponse = GetTemplateFileContentsResponses[keyof GetTemplateFileContentsResponses];
 export type ListWorkspaceUsersData = {
     body?: never;
     path?: never;

package/dist/jobs/index.d.ts

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

package/dist/jobs/index.js

@@ -15,5 +15,3 @@ var __exportStar = (this && this.__exportStar) || function(m, exports) {
 };
 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

@@ -1,13 +1,11 @@
 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>;
+    getArguments(): Promise<{
+        [key: number]: any;
+    }>;
+    private parseCommandLineArgs;
+    get indexKey(): string;
+    get index(): number;
+    start(func: (args: any) => Promise<void>): Promise<void>;
 }
-export declare const blJob: (jobName: string) => BlJob;
+export declare const blJob: BlJob;
 export {};

package/dist/jobs/jobs.js

@@ -2,94 +2,54 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.blJob = void 0;
 const env_js_1 = require("../common/env.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 BlJob {
-    jobName;
-    constructor(jobName) {
-        this.jobName = jobName;
-    }
-    get fallbackUrl() {
-        if (this.externalUrl != this.url) {
-            return this.externalUrl;
+    async getArguments() {
+        if (!env_js_1.env.BL_EXECUTION_DATA_URL) {
+            const args = this.parseCommandLineArgs();
+            return args;
         }
-        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() {
-        const envVar = this.jobName.replace(/-/g, "_").toUpperCase();
-        if (env_js_1.env[`BL_JOB_${envVar}_URL`]) {
-            return new URL(env_js_1.env[`BL_JOB_${envVar}_URL`]);
+        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 null;
+        return result;
     }
-    get url() {
-        if (this.forcedUrl)
-            return this.forcedUrl;
-        if (settings_js_1.settings.runInternalHostname)
-            return this.internalUrl;
-        return this.externalUrl;
+    get indexKey() {
+        return env_js_1.env.BL_TASK_KEY ?? "TASK_INDEX";
     }
-    async call(url, tasks) {
-        let body = {
-            tasks: tasks
-        };
-        const response = await fetch(url + "/executions", {
-            method: "POST",
-            headers: {
-                ...settings_js_1.settings.headers,
-                "Content-Type": "application/json",
-            },
-            body: JSON.stringify(body),
-        });
-        return response;
+    get index() {
+        return env_js_1.env[this.indexKey] ? Number(env_js_1.env[this.indexKey]) ?? 0 : 0;
     }
-    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,
-        });
+    /*
+      Run a job defined in a function, it's run in the current process
+    */
+    async start(func) {
         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;
+            const parsedArgs = await this.getArguments();
+            await func(parsedArgs);
+            process.exit(0);
         }
-        finally {
-            span.end();
+        catch (error) {
+            console.error('Job execution failed:', error);
+            process.exit(1);
         }
     }
 }
-const blJob = (jobName) => {
-    return new BlJob(jobName);
-};
-exports.blJob = blJob;
+exports.blJob = new BlJob();

package/dist/sandbox/client/sdk.gen.d.ts

@@ -1,5 +1,5 @@
 import type { Options as ClientOptions, TDataShape, Client } from '@hey-api/client-fetch';
-import type { DeleteFilesystemByPathData, GetFilesystemByPathData, PutFilesystemByPathData, DeleteNetworkProcessByPidMonitorData, PostNetworkProcessByPidMonitorData, GetNetworkProcessByPidPortsData, GetProcessData, PostProcessData, DeleteProcessByIdentifierData, GetProcessByIdentifierData, DeleteProcessByIdentifierKillData, GetProcessByIdentifierLogsData, GetProcessByIdentifierLogsStreamData, GetWatchFilesystemByPathData } from './types.gen';
+import type { DeleteFilesystemByPathData, GetFilesystemByPathData, PutFilesystemByPathData, DeleteNetworkProcessByPidMonitorData, PostNetworkProcessByPidMonitorData, GetNetworkProcessByPidPortsData, GetProcessData, PostProcessData, DeleteProcessByIdentifierData, GetProcessByIdentifierData, DeleteProcessByIdentifierKillData, GetProcessByIdentifierLogsData, GetProcessByIdentifierLogsStreamData, GetWatchFilesystemByPathData, GetWsProcessByIdentifierLogsStreamData, GetWsWatchFilesystemByPathData } from './types.gen';
 export type Options<TData extends TDataShape = TDataShape, ThrowOnError extends boolean = boolean> = ClientOptions<TData, ThrowOnError> & {
     /**
      * You can provide a client instance returned by `createClient()` instead of
@@ -22,7 +22,7 @@ export declare const deleteFilesystemByPath: <ThrowOnError extends boolean = fal
  * Get file or directory information
  * Get content of a file or listing of a directory
  */
-export declare const getFilesystemByPath: <ThrowOnError extends boolean = false>(options: Options<GetFilesystemByPathData, ThrowOnError>) => import("@hey-api/client-fetch").RequestResult<import("./types.gen").FileWithContent | import("./types.gen").Directory, import("./types.gen").ErrorResponse, ThrowOnError>;
+export declare const getFilesystemByPath: <ThrowOnError extends boolean = false>(options: Options<GetFilesystemByPathData, ThrowOnError>) => import("@hey-api/client-fetch").RequestResult<import("./types.gen").Directory | import("./types.gen").FileWithContent, import("./types.gen").ErrorResponse, ThrowOnError>;
 /**
  * Create or update file or directory
  * Create or update a file or directory
@@ -91,3 +91,13 @@ export declare const getProcessByIdentifierLogsStream: <ThrowOnError extends boo
  * Streams the path of modified files (one per line) in the given directory. Closes when the client disconnects.
  */
 export declare const getWatchFilesystemByPath: <ThrowOnError extends boolean = false>(options: Options<GetWatchFilesystemByPathData, ThrowOnError>) => import("@hey-api/client-fetch").RequestResult<string, import("./types.gen").ErrorResponse, ThrowOnError>;
+/**
+ * Stream process logs in real time via WebSocket
+ * Streams the stdout and stderr output of a process in real time as JSON messages. Closes when the process exits or the client disconnects.
+ */
+export declare const getWsProcessByIdentifierLogsStream: <ThrowOnError extends boolean = false>(options: Options<GetWsProcessByIdentifierLogsStreamData, ThrowOnError>) => import("@hey-api/client-fetch").RequestResult<unknown, import("./types.gen").ErrorResponse, ThrowOnError>;
+/**
+ * Stream file modification events in a directory via WebSocket
+ * Streams JSON events of modified files in the given directory. Closes when the client disconnects.
+ */
+export declare const getWsWatchFilesystemByPath: <ThrowOnError extends boolean = false>(options: Options<GetWsWatchFilesystemByPathData, ThrowOnError>) => import("@hey-api/client-fetch").RequestResult<unknown, import("./types.gen").ErrorResponse, ThrowOnError>;

package/dist/sandbox/client/sdk.gen.js

@@ -1,7 +1,7 @@
 "use strict";
 // This file is auto-generated by @hey-api/openapi-ts
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.getWatchFilesystemByPath = exports.getProcessByIdentifierLogsStream = exports.getProcessByIdentifierLogs = exports.deleteProcessByIdentifierKill = exports.getProcessByIdentifier = exports.deleteProcessByIdentifier = exports.postProcess = exports.getProcess = exports.getNetworkProcessByPidPorts = exports.postNetworkProcessByPidMonitor = exports.deleteNetworkProcessByPidMonitor = exports.putFilesystemByPath = exports.getFilesystemByPath = exports.deleteFilesystemByPath = void 0;
+exports.getWsWatchFilesystemByPath = exports.getWsProcessByIdentifierLogsStream = exports.getWatchFilesystemByPath = exports.getProcessByIdentifierLogsStream = exports.getProcessByIdentifierLogs = exports.deleteProcessByIdentifierKill = exports.getProcessByIdentifier = exports.deleteProcessByIdentifier = exports.postProcess = exports.getProcess = exports.getNetworkProcessByPidPorts = exports.postNetworkProcessByPidMonitor = exports.deleteNetworkProcessByPidMonitor = exports.putFilesystemByPath = exports.getFilesystemByPath = exports.deleteFilesystemByPath = void 0;
 const client_gen_1 = require("./client.gen");
 /**
  * Delete file or directory
@@ -173,3 +173,25 @@ const getWatchFilesystemByPath = (options) => {
     });
 };
 exports.getWatchFilesystemByPath = getWatchFilesystemByPath;
+/**
+ * Stream process logs in real time via WebSocket
+ * Streams the stdout and stderr output of a process in real time as JSON messages. Closes when the process exits or the client disconnects.
+ */
+const getWsProcessByIdentifierLogsStream = (options) => {
+    return (options.client ?? client_gen_1.client).get({
+        url: '/ws/process/{identifier}/logs/stream',
+        ...options
+    });
+};
+exports.getWsProcessByIdentifierLogsStream = getWsProcessByIdentifierLogsStream;
+/**
+ * Stream file modification events in a directory via WebSocket
+ * Streams JSON events of modified files in the given directory. Closes when the client disconnects.
+ */
+const getWsWatchFilesystemByPath = (options) => {
+    return (options.client ?? client_gen_1.client).get({
+        url: '/ws/watch/filesystem/{path}',
+        ...options
+    });
+};
+exports.getWsWatchFilesystemByPath = getWsWatchFilesystemByPath;

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

@@ -1,3 +1,11 @@
+export type Directory = {
+    files?: Array<File>;
+    path?: string;
+    /**
+     * @name Subdirectories
+     */
+    subdirectories?: Array<Subdirectory>;
+};
 export type ErrorResponse = {
     error?: string;
 };
@@ -6,9 +14,6 @@ export type File = {
     lastModified?: string;
     owner?: string;
     path?: string;
-    /**
-     * swagger:strfmt string
-     */
     permissions?: string;
     size?: number;
 };
@@ -23,9 +28,6 @@ export type FileWithContent = {
     lastModified?: string;
     owner?: string;
     path?: string;
-    /**
-     * swagger:strfmt string
-     */
     permissions?: string;
     size?: number;
 };
@@ -56,17 +58,12 @@ export type ProcessResponse = {
     status?: string;
     workingDir?: string;
 };
-export type SuccessResponse = {
-    message?: string;
+export type Subdirectory = {
     path?: string;
 };
-export type Directory = {
-    files?: Array<File>;
+export type SuccessResponse = {
+    message?: string;
     path?: string;
-    /**
-     * @name Subdirectories
-     */
-    subdirectories?: Array<Directory>;
 };
 export type DeleteFilesystemByPathData = {
     body?: never;
@@ -474,6 +471,50 @@ export type GetWatchFilesystemByPathResponses = {
     200: string;
 };
 export type GetWatchFilesystemByPathResponse = GetWatchFilesystemByPathResponses[keyof GetWatchFilesystemByPathResponses];
+export type GetWsProcessByIdentifierLogsStreamData = {
+    body?: never;
+    path: {
+        /**
+         * Process identifier (PID or name)
+         */
+        identifier: string;
+    };
+    query?: never;
+    url: '/ws/process/{identifier}/logs/stream';
+};
+export type GetWsProcessByIdentifierLogsStreamErrors = {
+    /**
+     * Process not found
+     */
+    404: ErrorResponse;
+    /**
+     * Internal server error
+     */
+    500: ErrorResponse;
+};
+export type GetWsProcessByIdentifierLogsStreamError = GetWsProcessByIdentifierLogsStreamErrors[keyof GetWsProcessByIdentifierLogsStreamErrors];
+export type GetWsWatchFilesystemByPathData = {
+    body?: never;
+    path: {
+        /**
+         * Directory path to watch
+         */
+        path: string;
+    };
+    query?: never;
+    url: '/ws/watch/filesystem/{path}';
+};
+export type GetWsWatchFilesystemByPathErrors = {
+    /**
+     * Invalid path
+     */
+    400: ErrorResponse;
+    /**
+     * Internal server error
+     */
+    500: ErrorResponse;
+};
+export type GetWsWatchFilesystemByPathError = GetWsWatchFilesystemByPathErrors[keyof GetWsWatchFilesystemByPathErrors];
 export type ClientOptions = {
     baseUrl: `${string}://localhost:8080` | (string & {});
 };

package/dist/sandbox/session.js

@@ -20,6 +20,8 @@ class SandboxSessions {
             spec: {
                 port: 443,
                 public: false,
+                requestHeaders: options.requestHeaders,
+                responseHeaders: options.responseHeaders,
             },
         };
         const { data } = await (0, index_js_1.createSandboxPreview)({

package/dist/sandbox/types.d.ts

@@ -1,6 +1,8 @@
 import { Sandbox } from "../client/types.gen";
 export interface SessionCreateOptions {
     expiresAt?: Date;
+    responseHeaders?: Record<string, string>;
+    requestHeaders?: Record<string, string>;
 }
 export interface SessionWithToken {
     name: string;

package/dist/telemetry/telemetry.d.ts

@@ -32,8 +32,6 @@ export interface BlaxelSpan {
 export interface BlaxelTelemetryProvider {
     /** Create a new span */
     startSpan(name: string, options?: BlaxelSpanOptions): BlaxelSpan;
-    /** Flush the telemetry provider */
-    flush(): Promise<void>;
 }
 /**
  * Registry for managing the global telemetry provider
@@ -57,6 +55,5 @@ export declare const telemetryRegistry: TelemetryRegistry;
  * Create a span with the registered provider
  */
 export declare function startSpan(name: string, options?: BlaxelSpanOptions): BlaxelSpan;
-export declare function withSpan<T>(name: string, fn: () => Promise<T>, options?: BlaxelSpanOptions): Promise<T>;
-export declare function flush(): Promise<void>;
+export declare function withSpan<T>(name: string, fn: () => T, options?: BlaxelSpanOptions): T;
 export {};

package/dist/telemetry/telemetry.js

@@ -4,7 +4,6 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.telemetryRegistry = void 0;
 exports.startSpan = startSpan;
 exports.withSpan = withSpan;
-exports.flush = flush;
 /**
  * No-operation implementation of Span
  */
@@ -23,9 +22,6 @@ class NoopTelemetryProvider {
     startSpan() {
         return new NoopSpan();
     }
-    flush() {
-        return Promise.resolve();
-    }
 }
 /**
  * Registry for managing the global telemetry provider
@@ -62,10 +58,10 @@ exports.telemetryRegistry = TelemetryRegistry.getInstance();
 function startSpan(name, options) {
     return exports.telemetryRegistry.getProvider().startSpan(name, options);
 }
-async function withSpan(name, fn, options) {
+function withSpan(name, fn, options) {
     const span = startSpan(name, options);
     try {
-        const result = await fn();
+        const result = fn();
         span.end();
         return result;
     }
@@ -75,6 +71,3 @@ async function withSpan(name, fn, options) {
         throw error;
     }
 }
-async function flush() {
-    await exports.telemetryRegistry.getProvider().flush();
-}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@blaxel/core",
-  "version": "0.2.6-dev.74",
+  "version": "0.2.6-preview.5",
   "description": "Blaxel Core SDK for TypeScript",
   "license": "MIT",
   "author": "Blaxel, INC (https://blaxel.ai)",

package/dist/jobs/start.d.ts

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

package/dist/jobs/start.js

@@ -1,83 +0,0 @@
-"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 : 0;
-    }
-    /*
-      Run a job defined in a function, it's run in the current process
-    */
-    async start(func) {
-        await (0, autoload_js_1.authenticate)();
-        let span = (0, telemetry_js_1.startSpan)(`blStartJob-${func.name}`, {
-            attributes: {
-                'job.name': func.name,
-                'job.index': this.index,
-                'job.args': JSON.stringify(this.getArguments()),
-            },
-            isRoot: true,
-        });
-        try {
-            const parsedArgs = await this.getArguments();
-            await func(parsedArgs);
-            span.setStatus('ok');
-            span.end();
-        }
-        catch (error) {
-            span.recordException(error);
-            span.setStatus('error', 'Job execution failed');
-            span.end();
-            throw error;
-        }
-    }
-}
-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);
-        }
-    }).catch((error) => {
-        console.error('Job execution failed:', error);
-        process.exit(1);
-    });
-};
-exports.blStartJob = blStartJob;

package/dist/jobs/types.d.ts

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

package/dist/jobs/types.js

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