@tensorlakeai/tensorlake 0.5.157 → 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.157"}.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.157"}.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.157";
+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.157"}.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.157";
+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.157"}.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.157";
+    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.157"}.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`
         );
       }
       /**
@@ -5475,10 +5509,9 @@ var init_client = __esm({
       /**
        * Create a sandbox, wait for it to reach `Running`, and return a connected handle.
        *
-       * This is `create({ wait: false })` followed by `PendingSandbox.ready()`
-       * polling every `pollInterval` seconds, with `requestTimeout` as the wait
-       * budget, in one call. The returned `Sandbox` auto-terminates when
-       * `terminate()` is called.
+       * The create waits on the server, which answers as soon as the sandbox
+       * runs, with `requestTimeout` as the wait budget. The returned `Sandbox`
+       * auto-terminates when `terminate()` is called.
        *
        * **When the wait runs out the sandbox is not deleted.** A `SandboxPending`
        * is thrown with the sandbox id, and the sandbox keeps its place in the
@@ -5490,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);
@@ -5500,11 +5538,13 @@ var init_client = __esm({
           pool_id: options?.poolId
         });
         const requestedName = options?.poolId != null ? null : options?.name ?? null;
-        let sandboxId;
-        let traceId;
         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) {
-          const result = await requestClient.claim(options.poolId, {
+          result = await requestClient.claim(options.poolId, {
             fileSystems: options.fileSystems
           });
           logSdkTiming("sandbox.create", "claim_response", createStart, {
@@ -5512,84 +5552,55 @@ var init_client = __esm({
             status: result.status,
             server_trace_id: result.traceId
           });
-          if (result.status === "running" /* RUNNING */) {
-            const explicitProxyUrl = options?.proxyUrl ?? explicitProxyUrlOverride() ?? void 0;
-            const selectedProxyUrl = await this.native.selectSandboxProxyUrl(
-              result.sandboxId,
-              result.sandboxUrl ?? null,
-              result.ingressEndpoint ?? null,
-              explicitProxyUrl ?? null
-            );
-            const sandbox2 = requestClient.connect(
-              result.sandboxId,
-              selectedProxyUrl,
-              result.routingHint,
-              requestTimeout,
-              explicitProxyUrl
-            );
-            sandbox2._setOwner(requestClient);
-            sandbox2.traceId = result.traceId;
-            sandbox2._setLifecycleIdentifier(result.sandboxId);
-            sandbox2._setName(result.name ?? requestedName);
-            logSdkTiming("sandbox.create", "complete", opStart, {
-              sandbox_id: result.sandboxId,
-              status: "running" /* RUNNING */,
-              server_trace_id: result.traceId
-            });
-            return sandbox2;
-          }
-          if (result.status === "suspended" /* SUSPENDED */ || result.status === "terminated" /* TERMINATED */ || result.status === "failed" /* FAILED */) {
-            throw startupFailure(result.sandboxId, result.status, {
-              errorDetails: result.errorDetails,
-              terminationReason: result.terminationReason ?? result.reason
-            });
-          }
-          sandboxId = result.sandboxId;
-          traceId = result.traceId;
         } else {
-          const pending = await requestClient.requestPending(options, {
-            ownsSandbox: true
+          const body = _SandboxClient.buildCreateRequestBody({
+            ...options,
+            wait: void 0
           });
+          result = await requestClient.tracedJson(
+            () => requestClient.native.createSandboxAndWait(JSON.stringify(body)),
+            "sandboxId"
+          );
           logSdkTiming("sandbox.create", "create_response", createStart, {
-            sandbox_id: pending.sandboxId,
-            status: pending.state,
-            server_trace_id: pending.traceId
+            sandbox_id: result.sandboxId,
+            status: result.status,
+            server_trace_id: result.traceId
+          });
+        }
+        if (result.status === "running" /* RUNNING */) {
+          const explicitProxyUrl = options?.proxyUrl ?? explicitProxyUrlOverride() ?? void 0;
+          const selectedProxyUrl = await this.native.selectSandboxProxyUrl(
+            result.sandboxId,
+            result.sandboxUrl ?? null,
+            result.ingressEndpoint ?? null,
+            explicitProxyUrl ?? null
+          );
+          const sandbox2 = requestClient.connect(
+            result.sandboxId,
+            selectedProxyUrl,
+            result.routingHint,
+            requestTimeout,
+            explicitProxyUrl
+          );
+          sandbox2._setOwner(requestClient);
+          sandbox2.traceId = result.traceId;
+          sandbox2._setLifecycleIdentifier(result.sandboxId);
+          sandbox2._setName(result.name ?? requestedName);
+          logSdkTiming("sandbox.create", "complete", opStart, {
+            sandbox_id: result.sandboxId,
+            status: "running" /* RUNNING */,
+            server_trace_id: result.traceId
+          });
+          return sandbox2;
+        }
+        if (result.status === "suspended" /* SUSPENDED */ || result.status === "terminated" /* TERMINATED */ || result.status === "failed" /* FAILED */) {
+          throw startupFailure(result.sandboxId, result.status, {
+            errorDetails: result.errorDetails,
+            terminationReason: result.terminationReason ?? result.reason
           });
-          sandboxId = pending.sandboxId;
-          traceId = pending.traceId;
-          if (pending.state === "running" /* RUNNING */) {
-            const explicitProxyUrl = options?.proxyUrl ?? explicitProxyUrlOverride() ?? void 0;
-            const selectedProxyUrl = await this.native.selectSandboxProxyUrl(
-              sandboxId,
-              pending.sandboxUrl ?? null,
-              pending.ingressEndpoint ?? null,
-              explicitProxyUrl ?? null
-            );
-            const sandbox2 = requestClient.connect(
-              sandboxId,
-              selectedProxyUrl,
-              pending.routingHint,
-              requestTimeout,
-              explicitProxyUrl
-            );
-            sandbox2._setOwner(requestClient);
-            sandbox2.traceId = traceId;
-            sandbox2._setLifecycleIdentifier(sandboxId);
-            sandbox2._setName(pending.name ?? requestedName);
-            logSdkTiming("sandbox.create", "complete", opStart, {
-              sandbox_id: sandboxId,
-              status: "running" /* RUNNING */,
-              server_trace_id: traceId
-            });
-            return sandbox2;
-          }
-          if (pending.state === "terminated" /* TERMINATED */ || pending.state === "failed" /* FAILED */) {
-            throw startupFailure(sandboxId, pending.state, {
-              errorDetails: pending.errorDetails,
-              terminationReason: pending.terminationReason ?? pending.reason
-            });
-          }
         }
+        const sandboxId = result.sandboxId;
+        const traceId = result.traceId;
         const remaining = Math.max(0, (deadline - Date.now()) / 1e3);
         const observed = await requestClient.waitForSandbox(
           sandboxId,