@tensorlakeai/tensorlake 0.5.153 → 0.5.154

package/dist/index.js

@@ -46,7 +46,7 @@ var init_defaults = __esm({
   "src/defaults.ts"() {
     "use strict";
     init_esm_shims();
-    SDK_VERSION = "0.5.153";
+    SDK_VERSION = "0.5.154";
     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";
@@ -57,6 +57,20 @@ var init_defaults = __esm({
   }
 });
 
+// src/enum.ts
+function enumValue(values, value) {
+  return typeof value === "string" && Object.values(values).includes(value) ? value : values.UNKNOWN;
+}
+function withRaw(value, raw) {
+  return Object.defineProperty(value, "raw", { value: raw, enumerable: false, configurable: true });
+}
+var init_enum = __esm({
+  "src/enum.ts"() {
+    "use strict";
+    init_esm_shims();
+  }
+});
+
 // src/errors.ts
 function formatErrorDetails(errorDetails) {
   if (errorDetails == null) return void 0;
@@ -495,7 +509,7 @@ var init_native_worker_client = __esm({
       }
     };
     runtimeKey = /* @__PURE__ */ Symbol.for(
-      `tensorlake.native.worker.${false ? "source" : "0.5.153"}.v1`
+      `tensorlake.native.worker.${false ? "source" : "0.5.154"}.v1`
     );
     runtimes = globalThis;
   }
@@ -597,6 +611,7 @@ var init_models = __esm({
       SnapshotStatus2["FAILED"] = "failed";
       SnapshotStatus2["DELETING"] = "deleting";
       SnapshotStatus2["REPLICATING"] = "replicating";
+      SnapshotStatus2["UNKNOWN"] = "unknown";
       return SnapshotStatus2;
     })(SnapshotStatus || {});
     ProcessStatus = /* @__PURE__ */ ((ProcessStatus2) => {
@@ -4493,6 +4508,18 @@ __export(client_exports, {
   DEFAULT_WAIT_POLL_INTERVAL_SEC: () => DEFAULT_WAIT_POLL_INTERVAL_SEC,
   SandboxClient: () => SandboxClient
 });
+function parseSnapshot(raw) {
+  if (raw === null || typeof raw !== "object" || Array.isArray(raw)) {
+    throw new SandboxError("Invalid snapshot response: expected an object");
+  }
+  if (raw.status == null) {
+    throw new SandboxError("Invalid snapshot response: missing status");
+  }
+  return withRaw({
+    ...fromSnakeKeys(raw, "snapshotId"),
+    status: enumValue(SnapshotStatus, raw.status)
+  }, raw);
+}
 function requireReadOnlySnapshotPin(fileSystemId, mountPath, readOnly, snapshotId) {
   if (snapshotId != null && readOnly !== true) {
     throw new SandboxError(
@@ -4614,6 +4641,7 @@ var init_client = __esm({
     "use strict";
     init_esm_shims();
     init_defaults();
+    init_enum();
     init_errors();
     init_native_worker_client();
     init_native_sandbox();
@@ -5261,18 +5289,18 @@ var init_client = __esm({
        * block automatically.
        */
       async snapshot(sandboxId, options) {
-        return this.plainJson(
+        const { json } = await callNative(
           () => this.native.createSnapshot(sandboxId, options?.snapshotType ?? null),
-          "snapshotId",
           { sandboxId, notFoundKind: "sandbox" }
         );
+        return parseSnapshot(JSON.parse(json));
       }
       /** Get current status and metadata for a snapshot by ID. */
       async getSnapshot(snapshotId) {
-        return this.tracedJson(
-          () => this.native.getSnapshot(snapshotId),
-          "snapshotId"
+        const { traceId, json } = await callNative(
+          () => this.native.getSnapshot(snapshotId)
         );
+        return Object.assign(parseSnapshot(JSON.parse(json)), { traceId });
       }
       /** List all snapshots in the namespace. */
       async listSnapshots() {
@@ -5280,9 +5308,14 @@ var init_client = __esm({
           () => this.native.listSnapshots()
         );
         const parsed = JSON.parse(json);
-        const snapshots = (parsed.snapshots ?? []).map(
-          (s) => fromSnakeKeys(s, "snapshotId")
-        );
+        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. */
@@ -5850,6 +5883,9 @@ var init_native_binding = __esm({
 import path3 from "path";
 import os from "os";
 import { existsSync as existsSync2, mkdtempSync, rmSync } from "fs";
+function isConfirmedCancellation(outcome) {
+  return outcome === "cancelled" /* CANCELLED */ || outcome === "already_cancelled" /* ALREADY_CANCELLED */ || outcome === "already_failed" /* ALREADY_FAILED */;
+}
 function camelize(key) {
   return key.replace(/_([a-z])/g, (_, c) => c.toUpperCase());
 }
@@ -5867,62 +5903,87 @@ function toImageBuildError(error) {
     if (key === "category") continue;
     details[camelize(key)] = value;
   }
+  details.code = enumValue(ImageBuildErrorCode, details.code || "internal" /* INTERNAL */);
+  if ("activeKind" in details) details.activeKind = details.activeKind ? enumValue(ImageBuildKind, details.activeKind) : null;
+  if ("activeStatus" in details) details.activeStatus = details.activeStatus ? enumValue(ImageBuildStatus, details.activeStatus) : null;
+  if ("outcome" in details) details.outcome = details.outcome ? enumValue(ImageBuildCancellationOutcome, details.outcome) : null;
+  if ("info" in details) details.info = buildInfoOrNull(details.info);
   const typed = details;
-  switch (typed.code) {
-    case "not_found":
-      return new ImageBuildNotFoundError(typed);
-    case "image_build_in_progress":
-      return new ImageBuildInProgressError(typed);
-    case "failed":
-      return new ImageBuildFailedError(typed);
-    case "cancelled":
-      return new ImageBuildCancelledError(typed);
-    case "timeout":
-      return new ImageBuildTimeoutError(typed);
-    case "interrupted":
-      return new ImageBuildInterruptedError(typed);
-    default:
-      return new ImageBuildError(typed);
-  }
+  const errors = {
+    not_found: ImageBuildNotFoundError,
+    image_build_in_progress: ImageBuildInProgressError,
+    failed: ImageBuildFailedError,
+    cancelled: ImageBuildCancelledError,
+    timeout: ImageBuildTimeoutError,
+    interrupted: ImageBuildInterruptedError
+  };
+  const ErrorClass = errors[typed.code] ?? ImageBuildError;
+  return withRaw(new ErrorClass(typed), payload);
 }
 function camelizeRecord(value) {
   const out = {};
   for (const [key, inner] of Object.entries(value)) out[camelize(key)] = inner;
   return out;
 }
-function parseBuildInfo(raw) {
+function buildInfoOrNull(value) {
+  if (!value || typeof value !== "object" || Array.isArray(value)) return null;
+  try {
+    return parseBuildInfo(value);
+  } catch (error) {
+    if (error instanceof ImageBuildError) return null;
+    throw error;
+  }
+}
+function decodeResponse(value) {
+  try {
+    const raw = typeof value === "string" ? JSON.parse(value) : value;
+    if (!raw || typeof raw !== "object" || Array.isArray(raw)) throw new TypeError("expected a response object");
+    return raw;
+  } catch (error) {
+    throw withRaw(new ImageBuildError({ code: "service" /* SERVICE */, message: `Invalid image-build response: ${String(error)}` }), { response: value });
+  }
+}
+function parseBuildInfo(value) {
+  const raw = decodeResponse(value);
+  if (raw.status == null) {
+    throw withRaw(new ImageBuildError({ code: "service" /* SERVICE */, message: "Invalid image-build response: missing status" }), { response: value });
+  }
+  const attempt = raw.attempt_no ?? 0;
+  const attemptNo = Number(attempt);
+  if (typeof attempt !== "number" && typeof attempt !== "string" || typeof attempt === "string" && attempt.trim() === "" || !Number.isSafeInteger(attemptNo)) {
+    throw withRaw(new ImageBuildError({ code: "service" /* SERVICE */, message: "Invalid image-build attempt_no" }), { response: value });
+  }
   const camel = camelizeRecord(raw);
   const result = raw.result && typeof raw.result === "object" ? camelizeRecord(raw.result) : null;
-  return {
+  return withRaw({
     buildId: String(raw.build_id ?? ""),
     clientRequestId: camel.clientRequestId ?? null,
-    kind: String(raw.kind ?? ""),
+    kind: raw.kind ? enumValue(ImageBuildKind, raw.kind) : null,
     name: raw.name ?? null,
-    status: String(raw.status ?? ""),
-    phase: raw.phase ?? null,
-    attemptNo: Number(raw.attempt_no ?? 0),
+    status: enumValue(ImageBuildStatus, raw.status),
+    phase: raw.phase ? enumValue(ImageBuildPhase, raw.phase) : null,
+    attemptNo,
     builderSandboxId: camel.builderSandboxId ?? null,
     error: raw.error ?? null,
     result,
     operationDeadlineMs: camel.operationDeadlineMs ?? null,
     createdAtMs: camel.createdAtMs ?? null,
-    updatedAtMs: camel.updatedAtMs ?? null,
-    raw
-  };
+    updatedAtMs: camel.updatedAtMs ?? null
+  }, raw);
 }
 function parseResult(json) {
-  return camelizeRecord(JSON.parse(json));
+  return camelizeRecord(decodeResponse(json));
 }
 function credentialsFromEnv() {
   const context = __imageBuildInternals.buildContextFromEnv();
   if (!context.apiKey) {
     if (context.personalAccessToken) {
       throw new ImageBuildError({
-        code: "auth",
+        code: "auth" /* AUTH */,
         message: "Sandbox image SDKs require TENSORLAKE_API_KEY. Personal access tokens are CLI-only."
       });
     }
-    throw new ImageBuildError({ code: "auth", message: "Missing TENSORLAKE_API_KEY credentials." });
+    throw new ImageBuildError({ code: "auth" /* AUTH */, message: "Missing TENSORLAKE_API_KEY credentials." });
   }
   return { apiUrl: context.apiUrl, bearerToken: context.apiKey, namespace: context.namespace };
 }
@@ -5936,18 +5997,18 @@ function clientOptions(credentials) {
 }
 function rejectLegacyReference(buildId) {
   if (typeof buildId !== "string" || buildId.trim().length === 0) {
-    throw new ImageBuildError({ code: "usage", message: "buildId must be a non-empty string" });
+    throw new ImageBuildError({ code: "usage" /* USAGE */, message: "buildId must be a non-empty string" });
   }
   if (buildId.startsWith("sandbox_template_")) {
     throw new ImageBuildError({
-      code: "usage",
+      code: "usage" /* USAGE */,
       message: `'${buildId}' is a legacy sandbox template id; legacy images have no build operation to get, wait for or cancel`,
       buildId
     });
   }
   if (buildId.startsWith("image_")) {
     throw new ImageBuildError({
-      code: "usage",
+      code: "usage" /* USAGE */,
       message: `'${buildId}' is an image id, not a build id; look the image up with findSandboxImageByName or use the build id admission returned`,
       buildId
     });
@@ -5994,8 +6055,15 @@ function makeEmitter(options) {
   };
 }
 function decodeStarted(json, credentials) {
-  const payload = JSON.parse(json);
-  return new ImageBuildHandle(parseBuildInfo(payload.build ?? {}), credentials, payload.client_request_id);
+  const payload = decodeResponse(json);
+  const build = payload.build;
+  if (!build || typeof build !== "object" || Array.isArray(build) || typeof build.build_id !== "string" || !build.build_id.trim()) {
+    throw withRaw(new ImageBuildError({
+      code: "service" /* SERVICE */,
+      message: "Invalid image-build admission: expected a build with a non-empty build_id"
+    }), { response: payload });
+  }
+  return new ImageBuildHandle(parseBuildInfo(build), credentials, payload.client_request_id);
 }
 async function startImageBuild(source, options = {}, deps = {}) {
   const emitter = makeEmitter({ emit: deps.emit ?? options.emit, verbose: options.verbose });
@@ -6107,7 +6175,7 @@ async function getImageBuild(buildId) {
   } catch (error) {
     throw toImageBuildError(error);
   }
-  return new ImageBuildHandle(parseBuildInfo(JSON.parse(json)), credentials);
+  return new ImageBuildHandle(parseBuildInfo(json), credentials);
 }
 async function waitImageBuild(buildId, options = {}) {
   rejectLegacyReference(buildId);
@@ -6138,33 +6206,81 @@ async function cancelImageBuildWith(credentials, selector) {
   } catch (error) {
     throw toImageBuildError(error);
   }
-  const payload = JSON.parse(json);
-  const outcome = String(payload.outcome ?? "");
-  return {
-    outcome,
-    build: payload.build ? parseBuildInfo(payload.build) : null,
-    confirmed: ["cancelled", "already_cancelled", "already_failed"].includes(outcome)
-  };
+  const payload = decodeResponse(json);
+  const outcome = enumValue(ImageBuildCancellationOutcome, payload.outcome);
+  return withRaw({ outcome, build: buildInfoOrNull(payload.build), confirmed: isConfirmedCancellation(outcome) }, payload);
 }
 async function cancelImageBuild(selector) {
   const hasBuild = selector.buildId != null;
   const hasRequest = selector.clientRequestId != null;
   if (hasBuild === hasRequest) {
     throw new ImageBuildError({
-      code: "usage",
+      code: "usage" /* USAGE */,
       message: "pass exactly one of buildId or clientRequestId"
     });
   }
   if (selector.buildId != null) rejectLegacyReference(selector.buildId);
   return cancelImageBuildWith(credentialsFromEnv(), selector);
 }
-var ImageBuildError, ImageBuildNotFoundError, ImageBuildInProgressError, ImageBuildFailedError, ImageBuildCancelledError, ImageBuildTimeoutError, ImageBuildInterruptedError, ImageBuildHandle;
+var ImageBuildStatus, ImageBuildPhase, ImageBuildKind, ImageBuildCancellationOutcome, ImageBuildErrorCode, ImageBuildError, ImageBuildNotFoundError, ImageBuildInProgressError, ImageBuildFailedError, ImageBuildCancelledError, ImageBuildTimeoutError, ImageBuildInterruptedError, ImageBuildHandle;
 var init_image_builds = __esm({
   "src/image-builds.ts"() {
     "use strict";
     init_esm_shims();
+    init_enum();
     init_image();
     init_sandbox_image();
+    ImageBuildStatus = /* @__PURE__ */ ((ImageBuildStatus2) => {
+      ImageBuildStatus2["PENDING"] = "pending";
+      ImageBuildStatus2["RUNNING"] = "running";
+      ImageBuildStatus2["SUCCEEDED"] = "succeeded";
+      ImageBuildStatus2["FAILED"] = "failed";
+      ImageBuildStatus2["CANCELLED"] = "cancelled";
+      ImageBuildStatus2["UNKNOWN"] = "unknown";
+      return ImageBuildStatus2;
+    })(ImageBuildStatus || {});
+    ImageBuildPhase = /* @__PURE__ */ ((ImageBuildPhase2) => {
+      ImageBuildPhase2["AWAITING_CONTEXT"] = "awaiting_context";
+      ImageBuildPhase2["STARTING_BUILDER"] = "starting_builder";
+      ImageBuildPhase2["EXECUTING"] = "executing";
+      ImageBuildPhase2["SNAPSHOTTING"] = "snapshotting";
+      ImageBuildPhase2["PUBLISHING"] = "publishing";
+      ImageBuildPhase2["FINISHED"] = "finished";
+      ImageBuildPhase2["UNKNOWN"] = "unknown";
+      return ImageBuildPhase2;
+    })(ImageBuildPhase || {});
+    ImageBuildKind = /* @__PURE__ */ ((ImageBuildKind2) => {
+      ImageBuildKind2["DOCKERFILE"] = "dockerfile";
+      ImageBuildKind2["IMPORT"] = "import";
+      ImageBuildKind2["DIRECT"] = "direct";
+      ImageBuildKind2["UNKNOWN"] = "unknown";
+      return ImageBuildKind2;
+    })(ImageBuildKind || {});
+    ImageBuildCancellationOutcome = /* @__PURE__ */ ((ImageBuildCancellationOutcome2) => {
+      ImageBuildCancellationOutcome2["CANCELLED"] = "cancelled";
+      ImageBuildCancellationOutcome2["ALREADY_CANCELLED"] = "already_cancelled";
+      ImageBuildCancellationOutcome2["ALREADY_COMPLETED"] = "already_completed";
+      ImageBuildCancellationOutcome2["ALREADY_FAILED"] = "already_failed";
+      ImageBuildCancellationOutcome2["NOT_ADMITTED"] = "not_admitted";
+      ImageBuildCancellationOutcome2["UNCONFIRMED"] = "unconfirmed";
+      ImageBuildCancellationOutcome2["UNKNOWN"] = "unknown";
+      return ImageBuildCancellationOutcome2;
+    })(ImageBuildCancellationOutcome || {});
+    ImageBuildErrorCode = /* @__PURE__ */ ((ImageBuildErrorCode2) => {
+      ImageBuildErrorCode2["USAGE"] = "usage";
+      ImageBuildErrorCode2["AUTH"] = "auth";
+      ImageBuildErrorCode2["NOT_FOUND"] = "not_found";
+      ImageBuildErrorCode2["IMAGE_BUILD_IN_PROGRESS"] = "image_build_in_progress";
+      ImageBuildErrorCode2["CLIENT_REQUEST_ID_MISMATCH"] = "client_request_id_mismatch";
+      ImageBuildErrorCode2["FAILED"] = "failed";
+      ImageBuildErrorCode2["CANCELLED"] = "cancelled";
+      ImageBuildErrorCode2["TIMEOUT"] = "timeout";
+      ImageBuildErrorCode2["INTERRUPTED"] = "interrupted";
+      ImageBuildErrorCode2["SERVICE"] = "service";
+      ImageBuildErrorCode2["INTERNAL"] = "internal";
+      ImageBuildErrorCode2["UNKNOWN"] = "unknown";
+      return ImageBuildErrorCode2;
+    })(ImageBuildErrorCode || {});
     ImageBuildError = class extends Error {
       code;
       buildId;
@@ -6174,7 +6290,11 @@ var init_image_builds = __esm({
         super(details.message);
         this.name = "ImageBuildError";
         this.code = details.code;
-        this.details = details;
+        this.details = { ...details };
+        withRaw(this, {});
+        if ("confirmed" in details) {
+          this.details.confirmed = details.confirmed === true && isConfirmedCancellation(details.outcome);
+        }
         if (details.buildId) this.buildId = details.buildId;
         if (details.clientRequestId) this.clientRequestId = details.clientRequestId;
       }
@@ -6206,7 +6326,7 @@ var init_image_builds = __esm({
         super(details);
         this.name = "ImageBuildFailedError";
         this.error = details.error ?? "no error recorded";
-        if (details.info) this.info = parseBuildInfo(details.info);
+        if (details.info) this.info = details.info;
       }
     };
     ImageBuildCancelledError = class extends ImageBuildError {
@@ -6214,7 +6334,7 @@ var init_image_builds = __esm({
       constructor(details) {
         super(details);
         this.name = "ImageBuildCancelledError";
-        if (details.info) this.info = parseBuildInfo(details.info);
+        if (details.info) this.info = details.info;
       }
     };
     ImageBuildTimeoutError = class extends ImageBuildError {
@@ -6230,8 +6350,8 @@ var init_image_builds = __esm({
       constructor(details) {
         super(details);
         this.name = "ImageBuildInterruptedError";
-        this.outcome = details.outcome ?? "unconfirmed";
-        this.confirmed = details.confirmed === true;
+        this.outcome = details.outcome ?? "unconfirmed" /* UNCONFIRMED */;
+        this.confirmed = this.details.confirmed === true;
         if (details.result) this.result = details.result;
       }
     };
@@ -6264,7 +6384,7 @@ var init_image_builds = __esm({
         } catch (error) {
           throw toImageBuildError(error);
         }
-        this.#info = parseBuildInfo(JSON.parse(json));
+        this.#info = parseBuildInfo(json);
         return this.#info;
       }
       /**
@@ -6285,8 +6405,12 @@ var init_image_builds = __esm({
           )
         );
         emitter.rethrow();
-        const result = parseResult(json);
-        this.#info = { ...this.#info, status: "succeeded", result };
+        const raw = decodeResponse(json);
+        const result = camelizeRecord(raw);
+        this.#info = withRaw(
+          { ...this.#info, status: "succeeded" /* SUCCEEDED */, result },
+          { ...this.#info.raw, status: "succeeded", result: raw }
+        );
         return result;
       }
       /** Cancel this build by its id (the sufficient selector). */
@@ -10413,14 +10537,19 @@ export {
   Headers,
   HttpBody,
   Image,
+  ImageBuildCancellationOutcome,
   ImageBuildCancelledError,
   ImageBuildError,
+  ImageBuildErrorCode,
   ImageBuildFailedError,
   ImageBuildHandle,
   ImageBuildInProgressError,
   ImageBuildInterruptedError,
+  ImageBuildKind,
   ImageBuildNotFoundError,
   ImageBuildOperationType,
+  ImageBuildPhase,
+  ImageBuildStatus,
   ImageBuildTimeoutError,
   MountError,
   OutputMode,