tensorlake 0.4.41 → 0.4.43

package/dist/index.cjs

@@ -156,7 +156,8 @@ var HttpClient = class {
   timeoutMs;
   abortController = null;
   constructor(options) {
-    this.baseUrl = options.baseUrl.replace(/\/+$/, "");
+    const url = options.baseUrl;
+    this.baseUrl = url.endsWith("/") ? url.slice(0, -1) : url;
     this.maxRetries = options.maxRetries ?? MAX_RETRIES;
     this.retryBackoffMs = options.retryBackoffMs ?? RETRY_BACKOFF_MS;
     this.timeoutMs = options.timeoutMs ?? DEFAULT_HTTP_TIMEOUT_MS;
@@ -484,6 +485,9 @@ async function* parseSSEStream(stream, signal) {
 }
 
 // src/url.ts
+function trimTrailingSlashes(url) {
+  return url.endsWith("/") ? url.slice(0, -1) : url;
+}
 function isLocalhost(apiUrl) {
   try {
     const parsed = new URL(apiUrl);
@@ -513,7 +517,7 @@ function resolveProxyTarget(proxyUrl, sandboxId) {
     const host = parsed.hostname;
     if (host === "localhost" || host === "127.0.0.1") {
       return {
-        baseUrl: proxyUrl.replace(/\/+$/, ""),
+        baseUrl: trimTrailingSlashes(proxyUrl),
         hostHeader: `${sandboxId}.local`
       };
     }
@@ -524,7 +528,7 @@ function resolveProxyTarget(proxyUrl, sandboxId) {
     };
   } catch {
     return {
-      baseUrl: `${proxyUrl.replace(/\/+$/, "")}/${sandboxId}`,
+      baseUrl: `${trimTrailingSlashes(proxyUrl)}/${sandboxId}`,
       hostHeader: void 0
     };
   }
@@ -1205,10 +1209,12 @@ var SandboxClient = class _SandboxClient {
     return fromSnakeKeys(raw, "sandboxId");
   }
   // --- Snapshots ---
-  async snapshot(sandboxId) {
+  async snapshot(sandboxId, options) {
+    const requestOptions = options?.contentMode != null ? { body: { snapshot_content_mode: options.contentMode } } : void 0;
     const raw = await this.http.requestJson(
       "POST",
-      this.path(`sandboxes/${sandboxId}/snapshot`)
+      this.path(`sandboxes/${sandboxId}/snapshot`),
+      requestOptions
     );
     return fromSnakeKeys(raw, "snapshotId");
   }
@@ -1237,7 +1243,9 @@ var SandboxClient = class _SandboxClient {
   async snapshotAndWait(sandboxId, options) {
     const timeout = options?.timeout ?? 300;
     const pollInterval = options?.pollInterval ?? 1;
-    const result = await this.snapshot(sandboxId);
+    const result = await this.snapshot(sandboxId, {
+      contentMode: options?.contentMode
+    });
     const deadline = Date.now() + timeout * 1e3;
     while (Date.now() < deadline) {
       const info = await this.getSnapshot(result.snapshotId);
@@ -1319,10 +1327,10 @@ var SandboxClient = class _SandboxClient {
     );
   }
   // --- Connect ---
-  connect(sandboxId, proxyUrl) {
+  connect(identifier, proxyUrl) {
     const resolvedProxy = proxyUrl ?? resolveProxyUrl(this.apiUrl);
     return new Sandbox({
-      sandboxId,
+      sandboxId: identifier,
       proxyUrl: resolvedProxy,
       apiKey: this.apiKey,
       organizationId: this.organizationId,
@@ -1559,7 +1567,7 @@ var CloudClient = class _CloudClient {
     );
     const response = await this.http.requestResponse(
       "PUT",
-      `${trimTrailingSlashes(buildServicePath)}/builds`,
+      `${trimTrailingSlashes2(buildServicePath)}/builds`,
       { body: form }
     );
     const raw = await parseJsonResponse(response);
@@ -1569,7 +1577,7 @@ var CloudClient = class _CloudClient {
     const form = createApplicationBuildForm(request, imageContexts);
     const response = await this.http.requestResponse(
       "POST",
-      trimTrailingSlashes(buildServicePath),
+      trimTrailingSlashes2(buildServicePath),
       { body: form }
     );
     const raw = await parseJsonResponse(response);
@@ -1578,34 +1586,34 @@ var CloudClient = class _CloudClient {
   async applicationBuildInfo(buildServicePath, applicationBuildId) {
     const raw = await this.http.requestJson(
       "GET",
-      `${trimTrailingSlashes(buildServicePath)}/${encodeURIComponent(applicationBuildId)}`
+      `${trimTrailingSlashes2(buildServicePath)}/${encodeURIComponent(applicationBuildId)}`
     );
     return fromSnakeKeys(raw);
   }
   async cancelApplicationBuild(buildServicePath, applicationBuildId) {
     const raw = await this.http.requestJson(
       "POST",
-      `${trimTrailingSlashes(buildServicePath)}/${encodeURIComponent(applicationBuildId)}/cancel`
+      `${trimTrailingSlashes2(buildServicePath)}/${encodeURIComponent(applicationBuildId)}/cancel`
     );
     return fromSnakeKeys(raw);
   }
   async buildInfo(buildServicePath, buildId) {
     const raw = await this.http.requestJson(
       "GET",
-      `${trimTrailingSlashes(buildServicePath)}/builds/${encodeURIComponent(buildId)}`
+      `${trimTrailingSlashes2(buildServicePath)}/builds/${encodeURIComponent(buildId)}`
     );
     return fromSnakeKeys(raw);
   }
   async cancelBuild(buildServicePath, buildId) {
     await this.http.requestResponse(
       "POST",
-      `${trimTrailingSlashes(buildServicePath)}/builds/${encodeURIComponent(buildId)}/cancel`
+      `${trimTrailingSlashes2(buildServicePath)}/builds/${encodeURIComponent(buildId)}/cancel`
     );
   }
   async *streamBuildLogs(buildServicePath, buildId, signal) {
     const stream = await this.http.requestStream(
       "GET",
-      `${trimTrailingSlashes(buildServicePath)}/builds/${encodeURIComponent(buildId)}/logs`,
+      `${trimTrailingSlashes2(buildServicePath)}/builds/${encodeURIComponent(buildId)}/logs`,
       { signal }
     );
     for await (const event of parseSSEStream(stream, signal)) {
@@ -1681,8 +1689,8 @@ function createApplicationBuildForm(request, imageContexts) {
   }
   return form;
 }
-function trimTrailingSlashes(value) {
-  return value.replace(/\/+$/, "");
+function trimTrailingSlashes2(value) {
+  return value.endsWith("/") ? value.slice(0, -1) : value;
 }
 function toBlobPart(data) {
   if (typeof data === "string" || data instanceof Blob) {
@@ -2558,7 +2566,9 @@ async function createSandboxImage(source, options = {}, deps = {}) {
     });
     await executeDockerfilePlan(sandbox, plan, emit, sleep4);
     emit({ type: "status", message: "Creating snapshot..." });
-    const snapshot = await client.snapshotAndWait(sandbox.sandboxId);
+    const snapshot = await client.snapshotAndWait(sandbox.sandboxId, {
+      contentMode: "filesystem_only"
+    });
     emit({
       type: "snapshot_created",
       snapshot_id: snapshot.snapshotId,