@tensorlakeai/tensorlake 0.5.158 → 0.5.159

package/dist/function-agent/main.cjs

@@ -402,7 +402,7 @@ var NativeWorkerClient = class {
   }
 };
 var runtimeKey = /* @__PURE__ */ Symbol.for(
-  `tensorlake.native.worker.${false ? "source" : "0.5.158"}.v1`
+  `tensorlake.native.worker.${false ? "source" : "0.5.159"}.v1`
 );
 var runtimes = globalThis;
 function configureNativeWorker(workerPath) {

package/dist/function-agent/main.js

@@ -390,7 +390,7 @@ var NativeWorkerClient = class {
   }
 };
 var runtimeKey = /* @__PURE__ */ Symbol.for(
-  `tensorlake.native.worker.${false ? "source" : "0.5.158"}.v1`
+  `tensorlake.native.worker.${false ? "source" : "0.5.159"}.v1`
 );
 var runtimes = globalThis;
 function configureNativeWorker(workerPath) {

package/dist/function-executor/main.cjs

@@ -51,7 +51,7 @@ var cancelNativeCall = /* @__PURE__ */ Symbol.for("tensorlake.native.cancel.v1")
 var closeNativeHandle = /* @__PURE__ */ Symbol.for("tensorlake.native.close.v1");
 
 // src/defaults.ts
-var SDK_VERSION = "0.5.158";
+var SDK_VERSION = "0.5.159";
 var API_URL = process.env.TENSORLAKE_API_URL ?? "https://api.tensorlake.ai";
 var API_KEY = process.env.TENSORLAKE_API_KEY ?? void 0;
 var NAMESPACE = process.env.INDEXIFY_NAMESPACE ?? "default";
@@ -247,7 +247,7 @@ var NativeWorkerClient = class {
   }
 };
 var runtimeKey = /* @__PURE__ */ Symbol.for(
-  `tensorlake.native.worker.${false ? "source" : "0.5.158"}.v1`
+  `tensorlake.native.worker.${false ? "source" : "0.5.159"}.v1`
 );
 var runtimes = globalThis;
 function configureNativeWorker(workerPath) {

package/dist/function-executor/main.js

@@ -10,7 +10,7 @@ var cancelNativeCall = /* @__PURE__ */ Symbol.for("tensorlake.native.cancel.v1")
 var closeNativeHandle = /* @__PURE__ */ Symbol.for("tensorlake.native.close.v1");
 
 // src/defaults.ts
-var SDK_VERSION = "0.5.158";
+var SDK_VERSION = "0.5.159";
 var API_URL = process.env.TENSORLAKE_API_URL ?? "https://api.tensorlake.ai";
 var API_KEY = process.env.TENSORLAKE_API_KEY ?? void 0;
 var NAMESPACE = process.env.INDEXIFY_NAMESPACE ?? "default";
@@ -206,7 +206,7 @@ var NativeWorkerClient = class {
   }
 };
 var runtimeKey = /* @__PURE__ */ Symbol.for(
-  `tensorlake.native.worker.${false ? "source" : "0.5.158"}.v1`
+  `tensorlake.native.worker.${false ? "source" : "0.5.159"}.v1`
 );
 var runtimes = globalThis;
 function configureNativeWorker(workerPath) {

package/dist/index.cjs

@@ -49,7 +49,7 @@ var init_defaults = __esm({
   "src/defaults.ts"() {
     "use strict";
     init_cjs_shims();
-    SDK_VERSION = "0.5.158";
+    SDK_VERSION = "0.5.159";
     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";
@@ -184,13 +184,17 @@ var init_errors = __esm({
       statusCode;
       /** Original response body, retained for compatibility and diagnostics. */
       responseMessage;
+      /** Machine-readable API error code, including proxy policy failures. */
+      code;
       errorDetails;
       constructor(statusCode, message) {
         let failure;
+        let code;
         try {
           const payload = JSON.parse(message);
           if (payload && typeof payload === "object" && !Array.isArray(payload)) {
             const record = payload;
+            if (typeof record.code === "string") code = record.code;
             if (typeof record.sandbox_id === "string" && record.sandbox_id.trim() && (record.status === "failed" || record.status === "terminated"))
               failure = record;
           }
@@ -216,6 +220,7 @@ var init_errors = __esm({
         this.name = "RemoteAPIError";
         this.statusCode = statusCode;
         this.responseMessage = message;
+        this.code = code;
       }
     };
     RequestNotFinishedError = class extends Error {
@@ -512,7 +517,7 @@ var init_native_worker_client = __esm({
       }
     };
     runtimeKey = /* @__PURE__ */ Symbol.for(
-      `tensorlake.native.worker.${false ? "source" : "0.5.158"}.v1`
+      `tensorlake.native.worker.${false ? "source" : "0.5.159"}.v1`
     );
     runtimes = globalThis;
   }
@@ -4933,6 +4938,12 @@ var init_client = __esm({
         if (options?.timeoutSecs != null) body.timeout_secs = options.timeoutSecs;
         if (options?.entrypoint != null) body.entrypoint = options.entrypoint;
         if (options?.snapshotId != null) body.snapshot_id = options.snapshotId;
+        if (options?.allowUnauthenticatedAccess != null) {
+          body.allow_unauthenticated_access = options.allowUnauthenticatedAccess;
+        }
+        if (options?.exposedPorts != null) {
+          body.exposed_ports = normalizeUserPorts(options.exposedPorts);
+        }
         if (options?.name != null) body.name = options.name;
         if (options?.fileSystems != null && options.fileSystems.length > 0) {
           body.file_systems = options.fileSystems.map(fileSystemMountToWire);
@@ -4944,6 +4955,7 @@ var init_client = __esm({
             deny_out: options?.denyOut ?? []
           };
         }
+        if (options?.autoResume != null) body.auto_resume = options.autoResume;
         if (options?.maxPendingSecs != null) {
           if (!Number.isInteger(options.maxPendingSecs) || options.maxPendingSecs < 0) {
             throw new SandboxError("maxPendingSecs must be a non-negative integer");
@@ -5060,13 +5072,14 @@ var init_client = __esm({
         const resizing = Object.keys(resources).length > 0;
         if (resizing) {
           validateResizeWait(options);
-          if (options.name != null || options.allowUnauthenticatedAccess != null || options.exposedPorts != null || options.network !== void 0) {
+          if (options.name != null || options.allowUnauthenticatedAccess != null || options.exposedPorts != null || options.autoResume != null || options.network !== void 0) {
             throw new SandboxError(
               "resources cannot be combined with other sandbox update fields"
             );
           }
           body.resources = resources;
         }
+        if (options.autoResume != null) body.auto_resume = options.autoResume;
         if (options.name != null) body.name = options.name;
         if (options.allowUnauthenticatedAccess != null) {
           body.allow_unauthenticated_access = options.allowUnauthenticatedAccess;
@@ -5161,28 +5174,19 @@ var init_client = __esm({
        * cannot. By default blocks until the sandbox is fully `Suspended`. Pass
        * `{ wait: false }` to return immediately after the request is sent
        * (fire-and-return); the server processes the suspend asynchronously.
+       *
+       * The server answers as soon as the suspend completes. When its wait runs
+       * out first, or the server does not support waiting, the rest of `timeout`
+       * is spent polling every `pollInterval` seconds.
        */
       async suspend(sandboxId, options) {
-        await callNative(() => this.native.suspendSandbox(sandboxId), {
+        await this.changeLifecycle(
           sandboxId,
-          notFoundKind: "sandbox"
-        });
-        if (options?.wait === false) return;
-        const timeout = options?.timeout ?? 300;
-        const pollInterval = options?.pollInterval ?? 1;
-        const deadline = Date.now() + timeout * 1e3;
-        while (Date.now() < deadline) {
-          const info = await this.get(sandboxId);
-          if (info.status === "suspended" /* SUSPENDED */) return;
-          if (info.status === "terminated" /* TERMINATED */) {
-            throw new SandboxError(
-              `Sandbox ${sandboxId} terminated while waiting for suspend`
-            );
-          }
-          await sleep2(pollInterval * 1e3);
-        }
-        throw new SandboxError(
-          `Sandbox ${sandboxId} did not suspend within ${timeout}s`
+          "suspend",
+          "suspended" /* SUSPENDED */,
+          () => this.native.suspendSandbox(sandboxId),
+          (waitSec) => this.native.suspendSandboxAndWait(sandboxId, waitSec),
+          options
         );
       }
       /**
@@ -5191,28 +5195,58 @@ var init_client = __esm({
        * By default blocks until the sandbox is `Running` and routable. Pass
        * `{ wait: false }` to return immediately after the request is sent
        * (fire-and-return); the server processes the resume asynchronously.
+       *
+       * The server answers as soon as the sandbox runs. When its wait runs out
+       * first, or the server does not support waiting, the rest of `timeout` is
+       * spent polling every `pollInterval` seconds.
        */
       async resume(sandboxId, options) {
-        await callNative(() => this.native.resumeSandbox(sandboxId), {
+        await this.changeLifecycle(
           sandboxId,
-          notFoundKind: "sandbox"
-        });
-        if (options?.wait === false) return;
+          "resume",
+          "running" /* RUNNING */,
+          () => this.native.resumeSandbox(sandboxId),
+          (waitSec) => this.native.resumeSandboxAndWait(sandboxId, waitSec),
+          options
+        );
+      }
+      /**
+       * Send a suspend or resume. With `wait`, the server waits for `target` and
+       * answers whether it was reached; when its wait runs out first, the rest of
+       * the timeout is spent polling.
+       */
+      async changeLifecycle(sandboxId, action, target, send, sendAndWait, options) {
+        const context = { sandboxId, notFoundKind: "sandbox" };
+        if (options?.wait === false) {
+          await callNative(send, context);
+          return;
+        }
         const timeout = options?.timeout ?? 300;
         const pollInterval = options?.pollInterval ?? 1;
         const deadline = Date.now() + timeout * 1e3;
+        let settled;
+        try {
+          settled = await callNative(() => sendAndWait(timeout), context);
+        } catch (error) {
+          if (!(error instanceof RemoteAPIError) || error.statusCode !== 409) {
+            throw error;
+          }
+          if ((await this.get(sandboxId)).status === target) return;
+          throw error;
+        }
+        if (settled) return;
         while (Date.now() < deadline) {
           const info = await this.get(sandboxId);
-          if (info.status === "running" /* RUNNING */) return;
+          if (info.status === target) return;
           if (info.status === "terminated" /* TERMINATED */) {
             throw new SandboxError(
-              `Sandbox ${sandboxId} terminated while waiting for resume`
+              `Sandbox ${sandboxId} terminated while waiting for ${action}`
             );
           }
           await sleep2(pollInterval * 1e3);
         }
         throw new SandboxError(
-          `Sandbox ${sandboxId} did not resume within ${timeout}s`
+          `Sandbox ${sandboxId} did not ${action} within ${timeout}s`
         );
       }
       /**
@@ -5489,6 +5523,11 @@ var init_client = __esm({
        * claim that is still starting is then waited on the same way.
        */
       async createAndConnect(options) {
+        if (options?.poolId != null && (options.allowUnauthenticatedAccess != null || options.exposedPorts != null)) {
+          throw new SandboxError(
+            "allowUnauthenticatedAccess and exposedPorts cannot be combined with poolId: pool claims inherit access settings from the pool."
+          );
+        }
         const opStart = nowMs();
         const requestTimeout = options?.requestTimeout ?? options?.startupTimeout ?? this.requestTimeoutMs / 1e3;
         const requestClient = this.withRequestTimeout(requestTimeout);
@@ -5501,6 +5540,9 @@ var init_client = __esm({
         const requestedName = options?.poolId != null ? null : options?.name ?? null;
         const createStart = nowMs();
         let result;
+        if (options?.poolId != null && options.autoResume != null) {
+          throw new SandboxError("autoResume cannot be set when claiming from a pool");
+        }
         if (options?.poolId != null) {
           result = await requestClient.claim(options.poolId, {
             fileSystems: options.fileSystems