@tensorlakeai/tensorlake 0.5.156 → 0.5.158

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.156"}.v1`
+  `tensorlake.native.worker.${false ? "source" : "0.5.158"}.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.156"}.v1`
+  `tensorlake.native.worker.${false ? "source" : "0.5.158"}.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.156";
+var SDK_VERSION = "0.5.158";
 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.156"}.v1`
+  `tensorlake.native.worker.${false ? "source" : "0.5.158"}.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.156";
+var SDK_VERSION = "0.5.158";
 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.156"}.v1`
+  `tensorlake.native.worker.${false ? "source" : "0.5.158"}.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.156";
+    SDK_VERSION = "0.5.158";
     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";
@@ -512,7 +512,7 @@ var init_native_worker_client = __esm({
       }
     };
     runtimeKey = /* @__PURE__ */ Symbol.for(
-      `tensorlake.native.worker.${false ? "source" : "0.5.156"}.v1`
+      `tensorlake.native.worker.${false ? "source" : "0.5.158"}.v1`
     );
     runtimes = globalThis;
   }
@@ -4088,9 +4088,7 @@ var init_sandbox = __esm({
        */
       async listSnapshots() {
         const client = this.requireLifecycleClient("listSnapshots");
-        const all = await client.listSnapshots();
-        const filtered = all.filter((s) => s.sandboxId === this.lifecycleIdentifier);
-        return Object.assign(filtered, { traceId: all.traceId });
+        return client.listSandboxSnapshots(this.lifecycleIdentifier);
       }
       /** Close the proxy connection. The sandbox keeps running. */
       close() {
@@ -4638,6 +4636,18 @@ function validateResizeWait(options) {
     throw new SandboxError("pollInterval must be finite and positive");
   }
 }
+function parseSnapshotList({ traceId, json }) {
+  const parsed = JSON.parse(json);
+  if (parsed !== null && (typeof parsed !== "object" || Array.isArray(parsed))) {
+    throw new SandboxError("Invalid snapshot list response: expected an object");
+  }
+  const records = parsed?.snapshots ?? [];
+  if (!Array.isArray(records)) {
+    throw new SandboxError("Invalid snapshot list response: snapshots must be an array");
+  }
+  const snapshots = records.map((s) => parseSnapshot(s));
+  return Object.assign(snapshots, { traceId });
+}
 var DEFAULT_WAIT_POLL_INTERVAL_SEC, GPU_MODELS, SandboxClient, RESERVED_SANDBOX_MANAGEMENT_PORT;
 var init_client = __esm({
   "src/client.ts"() {
@@ -5307,19 +5317,18 @@ var init_client = __esm({
       }
       /** List all snapshots in the namespace. */
       async listSnapshots() {
-        const { traceId, json } = await callNative(
-          () => this.native.listSnapshots()
+        return parseSnapshotList(
+          await callNative(() => this.native.listSnapshots())
+        );
+      }
+      /**
+       * List snapshots taken from one sandbox (its ID or name), from the
+       * sandbox's own listing rather than the whole namespace's.
+       */
+      async listSandboxSnapshots(sandboxId) {
+        return parseSnapshotList(
+          await callNative(() => this.native.listSandboxSnapshots(sandboxId))
         );
-        const parsed = JSON.parse(json);
-        if (parsed !== null && (typeof parsed !== "object" || Array.isArray(parsed))) {
-          throw new SandboxError("Invalid snapshot list response: expected an object");
-        }
-        const records = parsed?.snapshots ?? [];
-        if (!Array.isArray(records)) {
-          throw new SandboxError("Invalid snapshot list response: snapshots must be an array");
-        }
-        const snapshots = records.map((s) => parseSnapshot(s));
-        return Object.assign(snapshots, { traceId });
       }
       /** Delete a snapshot by ID. */
       async deleteSnapshot(snapshotId) {
@@ -5466,10 +5475,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
@@ -5491,11 +5499,10 @@ 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) {
-          const result = await requestClient.claim(options.poolId, {
+          result = await requestClient.claim(options.poolId, {
             fileSystems: options.fileSystems
           });
           logSdkTiming("sandbox.create", "claim_response", createStart, {
@@ -5503,84 +5510,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,