tensorlake 0.5.34 → 0.5.35

package/dist/index.cjs

@@ -35,7 +35,7 @@ var SDK_VERSION, API_URL, API_KEY, NAMESPACE, SANDBOX_PROXY_URL, DEFAULT_HTTP_TI
 var init_defaults = __esm({
   "src/defaults.ts"() {
     "use strict";
-    SDK_VERSION = "0.5.34";
+    SDK_VERSION = "0.5.35";
     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";
@@ -3984,6 +3984,49 @@ var client_exports = {};
 __export(client_exports, {
   SandboxClient: () => SandboxClient
 });
+function cloudInitIncludeData(source) {
+  const isWindowsPath = /^[A-Za-z]:[\\/]/.test(source);
+  try {
+    const url = new URL(source);
+    if ((url.protocol === "http:" || url.protocol === "https:") && url.hostname.length > 0) {
+      return Buffer.from(`#include
+${url.toString()}
+`, "utf8");
+    }
+    if (!isWindowsPath) {
+      throw new SandboxError(
+        "cloud-init URL must be an absolute HTTP(S) URL with a host"
+      );
+    }
+  } catch (error) {
+    if (error instanceof SandboxError) throw error;
+    if (!isWindowsPath && /^[A-Za-z][A-Za-z0-9+.-]*:/.test(source)) {
+      throw new SandboxError(
+        "cloud-init URL must be an absolute HTTP(S) URL with a host"
+      );
+    }
+  }
+  return void 0;
+}
+async function readCloudInitBase64(cloudInit) {
+  if (cloudInit == null) return void 0;
+  const sourceText = cloudInit instanceof URL ? cloudInit.toString() : cloudInit;
+  const includeData = cloudInitIncludeData(sourceText);
+  const data = includeData ?? await (0, import_promises.readFile)(sourceText).catch((error) => {
+    throw new SandboxError(
+      `failed to read cloud-init file ${JSON.stringify(sourceText)}: ${error}`
+    );
+  });
+  if (data.length === 0) {
+    throw new SandboxError("cloud-init user data must not be empty");
+  }
+  if (data.length > MAX_CLOUD_INIT_USER_DATA_BYTES) {
+    throw new SandboxError(
+      `cloud-init user data exceeds ${MAX_CLOUD_INIT_USER_DATA_BYTES} byte limit`
+    );
+  }
+  return data.toString("base64");
+}
 function resolveRequestTimeoutMs(options) {
   if (options?.requestTimeout != null) {
     return secondsToMillis4(options.requestTimeout);
@@ -4061,10 +4104,11 @@ function validateUserPort(port) {
 function dedupeAndSortPorts(ports) {
   return [...new Set(ports)].sort((a, b) => a - b);
 }
-var CREATE_SANDBOX_RETRYABLE_STATUS_CODES, CREATE_SANDBOX_ALLOWED_ERROR_STATUS_CODES, COPY_SANDBOX_ALLOWED_ERROR_STATUS_CODES, SandboxClient, RESERVED_SANDBOX_MANAGEMENT_PORT;
+var import_promises, CREATE_SANDBOX_RETRYABLE_STATUS_CODES, CREATE_SANDBOX_ALLOWED_ERROR_STATUS_CODES, COPY_SANDBOX_ALLOWED_ERROR_STATUS_CODES, MAX_CLOUD_INIT_USER_DATA_BYTES, SandboxClient, RESERVED_SANDBOX_MANAGEMENT_PORT;
 var init_client = __esm({
   "src/client.ts"() {
     "use strict";
+    import_promises = require("fs/promises");
     init_defaults();
     init_errors();
     init_http();
@@ -4074,6 +4118,7 @@ var init_client = __esm({
     CREATE_SANDBOX_RETRYABLE_STATUS_CODES = /* @__PURE__ */ new Set([502, 503]);
     CREATE_SANDBOX_ALLOWED_ERROR_STATUS_CODES = /* @__PURE__ */ new Set([504]);
     COPY_SANDBOX_ALLOWED_ERROR_STATUS_CODES = /* @__PURE__ */ new Set([422, 504]);
+    MAX_CLOUD_INIT_USER_DATA_BYTES = 16 * 1024;
     SandboxClient = class _SandboxClient {
       http;
       apiUrl;
@@ -4164,6 +4209,7 @@ var init_client = __esm({
       // --- Sandbox CRUD ---
       /** Create a new sandbox. Returns immediately; the sandbox may still be starting. Use `createAndConnect()` for a blocking, ready-to-use handle. */
       async create(options) {
+        const cloudInitBase64 = await readCloudInitBase64(options?.cloudInit);
         const body = {
           resources: {
             cpus: options?.cpus ?? 1,
@@ -4176,6 +4222,7 @@ var init_client = __esm({
         if (options?.entrypoint != null) body.entrypoint = options.entrypoint;
         if (options?.snapshotId != null) body.snapshot_id = options.snapshotId;
         if (options?.name != null) body.name = options.name;
+        if (cloudInitBase64 != null) body.cloud_init_base64 = cloudInitBase64;
         if (options?.allowInternetAccess === false || options?.allowOut != null || options?.denyOut != null) {
           body.network = {
             allow_internet_access: options?.allowInternetAccess ?? true,
@@ -4593,6 +4640,9 @@ var init_client = __esm({
        * @throws {SandboxError} If the sandbox terminates during startup or the timeout elapses.
        */
       async createAndConnect(options) {
+        if (options?.poolId != null && options?.cloudInit != null) {
+          throw new SandboxError("cloud-init cannot be used with poolId");
+        }
         const requestTimeout = options?.requestTimeout ?? options?.startupTimeout ?? this.requestTimeoutMs / 1e3;
         const requestClient = this.withRequestTimeout(requestTimeout);
         const result = options?.poolId != null ? await requestClient.claim(options.poolId) : await requestClient.create(options);