tensorlake 0.5.33 → 0.5.35

package/dist/index.cjs

@@ -35,7 +35,7 @@ var SDK_VERSION, API_URL, API_KEY, NAMESPACE, SANDBOX_PROXY_URL, DEFAULT_HTTP_TI
 var init_defaults = __esm({
   "src/defaults.ts"() {
     "use strict";
-    SDK_VERSION = "0.5.33";
+    SDK_VERSION = "0.5.35";
     API_URL = process.env.TENSORLAKE_API_URL ?? "https://api.tensorlake.ai";
     API_KEY = process.env.TENSORLAKE_API_KEY ?? void 0;
     NAMESPACE = process.env.INDEXIFY_NAMESPACE ?? "default";
@@ -411,9 +411,23 @@ var init_http = __esm({
 });
 
 // src/models.ts
+function camelToSnake(str) {
+  return str.replace(CAMEL_TO_SNAKE_RE, (ch) => "_" + ch.toLowerCase());
+}
 function snakeToCamel(str) {
   return str.replace(/_([a-z])/g, (_, ch) => ch.toUpperCase());
 }
+function toSnakeKeys(obj) {
+  if (Array.isArray(obj)) return obj.map(toSnakeKeys);
+  if (obj !== null && typeof obj === "object" && !(obj instanceof Date)) {
+    const result = {};
+    for (const [k, v] of Object.entries(obj)) {
+      result[camelToSnake(k)] = toSnakeKeys(v);
+    }
+    return result;
+  }
+  return obj;
+}
 function parseTimestamp(v) {
   if (v == null) return void 0;
   if (v instanceof Date) return v;
@@ -452,7 +466,7 @@ function fromSnakeKeys(obj, idField) {
   }
   return obj;
 }
-var SandboxStatus, SnapshotStatus, ProcessStatus, StdinMode, OutputMode, ContainerState;
+var SandboxStatus, SnapshotStatus, ProcessStatus, StdinMode, OutputMode, ContainerState, CAMEL_TO_SNAKE_RE;
 var init_models = __esm({
   "src/models.ts"() {
     "use strict";
@@ -477,6 +491,7 @@ var init_models = __esm({
       ProcessStatus2["RUNNING"] = "running";
       ProcessStatus2["EXITED"] = "exited";
       ProcessStatus2["SIGNALED"] = "signaled";
+      ProcessStatus2["OOM_KILLED"] = "oom_killed";
       return ProcessStatus2;
     })(ProcessStatus || {});
     StdinMode = /* @__PURE__ */ ((StdinMode2) => {
@@ -494,6 +509,7 @@ var init_models = __esm({
       ContainerState2["RUNNING"] = "Running";
       return ContainerState2;
     })(ContainerState || {});
+    CAMEL_TO_SNAKE_RE = /[A-Z]/g;
   }
 });
 
@@ -3651,6 +3667,11 @@ var init_sandbox = __esm({
         if (options?.stderrMode != null && options.stderrMode !== "capture" /* CAPTURE */) {
           payload.stderr_mode = options.stderrMode;
         }
+        if (options?.name != null) payload.name = options.name;
+        if (options?.restart != null) payload.restart = toSnakeKeys(options.restart);
+        if (options?.healthCheck != null) {
+          payload.health_check = toSnakeKeys(options.healthCheck);
+        }
         const raw = await this.http.requestJson(
           "POST",
           "/api/v1/processes",
@@ -3679,6 +3700,14 @@ var init_sandbox = __esm({
       async killProcess(pid) {
         await this.http.requestJson("DELETE", `/api/v1/processes/${pid}`);
       }
+      /** Restart a managed process by PID. */
+      async restartProcess(pid) {
+        const raw = await this.http.requestJson(
+          "POST",
+          `/api/v1/processes/${pid}/restart`
+        );
+        return Object.assign(fromSnakeKeys(raw), { traceId: raw.traceId });
+      }
       /** Send an arbitrary signal to a process (e.g. `15` for SIGTERM, `9` for SIGKILL). */
       async sendSignal(pid, signal) {
         const raw = await this.http.requestJson(
@@ -3955,6 +3984,49 @@ var client_exports = {};
 __export(client_exports, {
   SandboxClient: () => SandboxClient
 });
+function cloudInitIncludeData(source) {
+  const isWindowsPath = /^[A-Za-z]:[\\/]/.test(source);
+  try {
+    const url = new URL(source);
+    if ((url.protocol === "http:" || url.protocol === "https:") && url.hostname.length > 0) {
+      return Buffer.from(`#include
+${url.toString()}
+`, "utf8");
+    }
+    if (!isWindowsPath) {
+      throw new SandboxError(
+        "cloud-init URL must be an absolute HTTP(S) URL with a host"
+      );
+    }
+  } catch (error) {
+    if (error instanceof SandboxError) throw error;
+    if (!isWindowsPath && /^[A-Za-z][A-Za-z0-9+.-]*:/.test(source)) {
+      throw new SandboxError(
+        "cloud-init URL must be an absolute HTTP(S) URL with a host"
+      );
+    }
+  }
+  return void 0;
+}
+async function readCloudInitBase64(cloudInit) {
+  if (cloudInit == null) return void 0;
+  const sourceText = cloudInit instanceof URL ? cloudInit.toString() : cloudInit;
+  const includeData = cloudInitIncludeData(sourceText);
+  const data = includeData ?? await (0, import_promises.readFile)(sourceText).catch((error) => {
+    throw new SandboxError(
+      `failed to read cloud-init file ${JSON.stringify(sourceText)}: ${error}`
+    );
+  });
+  if (data.length === 0) {
+    throw new SandboxError("cloud-init user data must not be empty");
+  }
+  if (data.length > MAX_CLOUD_INIT_USER_DATA_BYTES) {
+    throw new SandboxError(
+      `cloud-init user data exceeds ${MAX_CLOUD_INIT_USER_DATA_BYTES} byte limit`
+    );
+  }
+  return data.toString("base64");
+}
 function resolveRequestTimeoutMs(options) {
   if (options?.requestTimeout != null) {
     return secondsToMillis4(options.requestTimeout);
@@ -4032,10 +4104,11 @@ function validateUserPort(port) {
 function dedupeAndSortPorts(ports) {
   return [...new Set(ports)].sort((a, b) => a - b);
 }
-var CREATE_SANDBOX_RETRYABLE_STATUS_CODES, CREATE_SANDBOX_ALLOWED_ERROR_STATUS_CODES, COPY_SANDBOX_ALLOWED_ERROR_STATUS_CODES, SandboxClient, RESERVED_SANDBOX_MANAGEMENT_PORT;
+var import_promises, CREATE_SANDBOX_RETRYABLE_STATUS_CODES, CREATE_SANDBOX_ALLOWED_ERROR_STATUS_CODES, COPY_SANDBOX_ALLOWED_ERROR_STATUS_CODES, MAX_CLOUD_INIT_USER_DATA_BYTES, SandboxClient, RESERVED_SANDBOX_MANAGEMENT_PORT;
 var init_client = __esm({
   "src/client.ts"() {
     "use strict";
+    import_promises = require("fs/promises");
     init_defaults();
     init_errors();
     init_http();
@@ -4045,6 +4118,7 @@ var init_client = __esm({
     CREATE_SANDBOX_RETRYABLE_STATUS_CODES = /* @__PURE__ */ new Set([502, 503]);
     CREATE_SANDBOX_ALLOWED_ERROR_STATUS_CODES = /* @__PURE__ */ new Set([504]);
     COPY_SANDBOX_ALLOWED_ERROR_STATUS_CODES = /* @__PURE__ */ new Set([422, 504]);
+    MAX_CLOUD_INIT_USER_DATA_BYTES = 16 * 1024;
     SandboxClient = class _SandboxClient {
       http;
       apiUrl;
@@ -4135,6 +4209,7 @@ var init_client = __esm({
       // --- Sandbox CRUD ---
       /** Create a new sandbox. Returns immediately; the sandbox may still be starting. Use `createAndConnect()` for a blocking, ready-to-use handle. */
       async create(options) {
+        const cloudInitBase64 = await readCloudInitBase64(options?.cloudInit);
         const body = {
           resources: {
             cpus: options?.cpus ?? 1,
@@ -4147,6 +4222,7 @@ var init_client = __esm({
         if (options?.entrypoint != null) body.entrypoint = options.entrypoint;
         if (options?.snapshotId != null) body.snapshot_id = options.snapshotId;
         if (options?.name != null) body.name = options.name;
+        if (cloudInitBase64 != null) body.cloud_init_base64 = cloudInitBase64;
         if (options?.allowInternetAccess === false || options?.allowOut != null || options?.denyOut != null) {
           body.network = {
             allow_internet_access: options?.allowInternetAccess ?? true,
@@ -4379,7 +4455,7 @@ var init_client = __esm({
         const client = this.withRequestTimeout(options?.requestTimeout);
         const raw = await client.http.requestJson(
           "POST",
-          client.path(`sandbox/${sandboxId}/copy?times=${encodeURIComponent(String(times))}`),
+          client.path(`sandboxes/${sandboxId}/copy?times=${encodeURIComponent(String(times))}`),
           { allowedErrorStatusCodes: COPY_SANDBOX_ALLOWED_ERROR_STATUS_CODES }
         );
         const result = fromSnakeKeys(raw, "sandboxId");
@@ -4564,6 +4640,9 @@ var init_client = __esm({
        * @throws {SandboxError} If the sandbox terminates during startup or the timeout elapses.
        */
       async createAndConnect(options) {
+        if (options?.poolId != null && options?.cloudInit != null) {
+          throw new SandboxError("cloud-init cannot be used with poolId");
+        }
         const requestTimeout = options?.requestTimeout ?? options?.startupTimeout ?? this.requestTimeoutMs / 1e3;
         const requestClient = this.withRequestTimeout(requestTimeout);
         const result = options?.poolId != null ? await requestClient.claim(options.poolId) : await requestClient.create(options);