@tensorlakeai/tensorlake 0.5.158 → 0.5.160

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.160";
     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 {
@@ -328,6 +333,11 @@ function createWorkerBinding(runtime) {
         return runtime.handle("NativeRepositoryClient", args);
       }
     },
+    NativeUsageClient: class {
+      constructor(...args) {
+        return runtime.handle("NativeUsageClient", args);
+      }
+    },
     NativeTcpTunnel: class {
       constructor() {
         return runtime.handle("NativeTcpTunnel", []);
@@ -512,7 +522,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.160"}.v1`
     );
     runtimes = globalThis;
   }
@@ -4933,6 +4943,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 +4960,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 +5077,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 +5179,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 +5200,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 +5528,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 +5545,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
@@ -7123,6 +7170,7 @@ __export(index_exports, {
   TERMINATION_REASON_CANCELLED: () => TERMINATION_REASON_CANCELLED,
   TERMINATION_REASON_NO_CAPACITY: () => TERMINATION_REASON_NO_CAPACITY,
   TcpTunnel: () => TcpTunnel,
+  UsageClient: () => UsageClient,
   cancelImageBuild: () => cancelImageBuild,
   createFileSystem: () => createFileSystem,
   createSandboxImage: () => createSandboxImage,
@@ -8847,6 +8895,59 @@ function validateTimeoutMs2(timeoutMs) {
   }
 }
 
+// src/usage.ts
+init_cjs_shims();
+init_defaults();
+init_errors();
+init_native_sandbox();
+init_native_worker_client();
+var UsageClient = class {
+  native;
+  constructor(options) {
+    const apiKey = options?.apiKey ?? API_KEY;
+    if (!apiKey) {
+      throw new SandboxError("Missing TENSORLAKE_API_KEY credentials.");
+    }
+    const organizationId = options?.organizationId ?? process.env.TENSORLAKE_ORGANIZATION_ID;
+    if (!organizationId) {
+      throw new SandboxError(
+        "An organization ID is required. Pass organizationId or set TENSORLAKE_ORGANIZATION_ID."
+      );
+    }
+    const binding = loadNativeSandboxBinding();
+    if (typeof binding.NativeUsageClient !== "function") {
+      throw new SandboxError(
+        "native binding does not export the usage client; rebuild with 'npm run build:native'"
+      );
+    }
+    this.native = new binding.NativeUsageClient(
+      options?.apiUrl ?? API_URL,
+      apiKey,
+      organizationId,
+      `tensorlake-typescript-sdk/${SDK_VERSION}`,
+      DEFAULT_HTTP_TIMEOUT_MS / 1e3
+    );
+  }
+  close() {
+    releaseNativeHandle(this.native);
+  }
+  /** Organization usage totals for the window. */
+  async summary(options) {
+    const { json } = await callNative(() => this.native.summary(options?.last ?? null));
+    return JSON.parse(json);
+  }
+  /**
+   * Usage and attributed cost of every sandbox that ran in the window.
+   *
+   * The server computes the attribution once and returns every sandbox in one
+   * response.
+   */
+  async sandboxes(options) {
+    const { json } = await callNative(() => this.native.sandboxes(options?.last ?? null));
+    return JSON.parse(json);
+  }
+};
+
 // src/index.ts
 init_sandbox_image();
 init_image_builds();
@@ -10760,6 +10861,7 @@ init_models();
   TERMINATION_REASON_CANCELLED,
   TERMINATION_REASON_NO_CAPACITY,
   TcpTunnel,
+  UsageClient,
   cancelImageBuild,
   createFileSystem,
   createSandboxImage,