@vm0/cli 1.5.0 → 1.8.1

package/index.js

@@ -736,15 +736,15 @@ function mergeDefs(...defs) {
 function cloneDef(schema) {
   return mergeDefs(schema._zod.def);
 }
-function getElementAtPath(obj, path8) {
-  if (!path8)
+function getElementAtPath(obj, path9) {
+  if (!path9)
     return obj;
-  return path8.reduce((acc, key) => acc?.[key], obj);
+  return path9.reduce((acc, key) => acc?.[key], obj);
 }
 function promiseAllObject(promisesObj) {
   const keys = Object.keys(promisesObj);
-  const promises5 = keys.map((key) => promisesObj[key]);
-  return Promise.all(promises5).then((results) => {
+  const promises6 = keys.map((key) => promisesObj[key]);
+  return Promise.all(promises6).then((results) => {
     const resolvedObj = {};
     for (let i = 0; i < keys.length; i++) {
       resolvedObj[keys[i]] = results[i];
@@ -1098,11 +1098,11 @@ function aborted(x, startIndex = 0) {
   }
   return false;
 }
-function prefixIssues(path8, issues) {
+function prefixIssues(path9, issues) {
   return issues.map((iss) => {
     var _a;
     (_a = iss).path ?? (_a.path = []);
-    iss.path.unshift(path8);
+    iss.path.unshift(path9);
     return iss;
   });
 }
@@ -1270,7 +1270,7 @@ function treeifyError(error43, _mapper) {
     return issue2.message;
   };
   const result = { errors: [] };
-  const processError = (error44, path8 = []) => {
+  const processError = (error44, path9 = []) => {
     var _a, _b;
     for (const issue2 of error44.issues) {
       if (issue2.code === "invalid_union" && issue2.errors.length) {
@@ -1280,7 +1280,7 @@ function treeifyError(error43, _mapper) {
       } else if (issue2.code === "invalid_element") {
         processError({ issues: issue2.issues }, issue2.path);
       } else {
-        const fullpath = [...path8, ...issue2.path];
+        const fullpath = [...path9, ...issue2.path];
         if (fullpath.length === 0) {
           result.errors.push(mapper(issue2));
           continue;
@@ -1312,8 +1312,8 @@ function treeifyError(error43, _mapper) {
 }
 function toDotPath(_path) {
   const segs = [];
-  const path8 = _path.map((seg) => typeof seg === "object" ? seg.key : seg);
-  for (const seg of path8) {
+  const path9 = _path.map((seg) => typeof seg === "object" ? seg.key : seg);
+  for (const seg of path9) {
     if (typeof seg === "number")
       segs.push(`[${seg}]`);
     else if (typeof seg === "symbol")
@@ -12413,10 +12413,37 @@ var ApiClient = class {
     }
     return await response.json();
   }
+  async continueSession(body) {
+    const baseUrl = await this.getBaseUrl();
+    const headers = await this.getHeaders();
+    const response = await fetch(`${baseUrl}/api/agent/runs/continue`, {
+      method: "POST",
+      headers,
+      body: JSON.stringify(body)
+    });
+    if (!response.ok) {
+      const error43 = await response.json();
+      throw new Error(error43.error?.message || "Failed to continue session");
+    }
+    return await response.json();
+  }
+  async getAgentSession(id) {
+    const baseUrl = await this.getBaseUrl();
+    const headers = await this.getHeaders();
+    const response = await fetch(`${baseUrl}/api/agent/sessions/${id}`, {
+      method: "GET",
+      headers
+    });
+    if (!response.ok) {
+      const error43 = await response.json();
+      throw new Error(error43.error?.message || "Failed to get agent session");
+    }
+    return await response.json();
+  }
   /**
    * Generic GET request
    */
-  async get(path8) {
+  async get(path9) {
     const baseUrl = await this.getBaseUrl();
     const token = await getToken();
     if (!token) {
@@ -12429,7 +12456,7 @@ var ApiClient = class {
     if (bypassSecret) {
       headers["x-vercel-protection-bypass"] = bypassSecret;
     }
-    return fetch(`${baseUrl}${path8}`, {
+    return fetch(`${baseUrl}${path9}`, {
       method: "GET",
       headers
     });
@@ -12437,7 +12464,7 @@ var ApiClient = class {
   /**
    * Generic POST request
    */
-  async post(path8, options) {
+  async post(path9, options) {
     const baseUrl = await this.getBaseUrl();
     const token = await getToken();
     if (!token) {
@@ -12450,7 +12477,7 @@ var ApiClient = class {
     if (bypassSecret) {
       headers["x-vercel-protection-bypass"] = bypassSecret;
     }
-    return fetch(`${baseUrl}${path8}`, {
+    return fetch(`${baseUrl}${path9}`, {
       method: "POST",
       headers,
       body: options?.body
@@ -12464,6 +12491,19 @@ function validateAgentName(name) {
   const nameRegex = /^[a-zA-Z0-9]([a-zA-Z0-9-]{1,62}[a-zA-Z0-9])?$/;
   return nameRegex.test(name);
 }
+function validateVolumeConfig(volumeKey, volumeConfig) {
+  if (!volumeConfig || typeof volumeConfig !== "object") {
+    return `Volume "${volumeKey}" must be an object`;
+  }
+  const vol = volumeConfig;
+  if (!vol.name || typeof vol.name !== "string") {
+    return `Volume "${volumeKey}" must have a 'name' field (string)`;
+  }
+  if (!vol.version || typeof vol.version !== "string") {
+    return `Volume "${volumeKey}" must have a 'version' field (string)`;
+  }
+  return null;
+}
 function validateAgentConfig(config2) {
   if (!config2 || typeof config2 !== "object") {
     return { valid: false, error: "Config must be an object" };
@@ -12472,22 +12512,83 @@ function validateAgentConfig(config2) {
   if (!cfg.version) {
     return { valid: false, error: "Missing config.version" };
   }
-  if (!cfg.agent || typeof cfg.agent !== "object") {
-    return { valid: false, error: "Missing config.agent" };
+  if (!cfg.agents || !Array.isArray(cfg.agents)) {
+    return { valid: false, error: "Missing config.agents array" };
+  }
+  if (cfg.agents.length === 0) {
+    return { valid: false, error: "config.agents array must not be empty" };
+  }
+  const agent = cfg.agents[0];
+  if (!agent || typeof agent !== "object") {
+    return { valid: false, error: "First agent must be an object" };
   }
-  const agent = cfg.agent;
   if (!agent.name) {
-    return { valid: false, error: "Missing agent.name" };
+    return { valid: false, error: "Missing agents[0].name" };
   }
   if (typeof agent.name !== "string") {
-    return { valid: false, error: "agent.name must be a string" };
+    return { valid: false, error: "agents[0].name must be a string" };
   }
   if (!validateAgentName(agent.name)) {
     return {
       valid: false,
-      error: "Invalid agent.name format. Must be 3-64 characters, letters, numbers, and hyphens only. Must start and end with letter or number."
+      error: "Invalid agents[0].name format. Must be 3-64 characters, letters, numbers, and hyphens only. Must start and end with letter or number."
     };
   }
+  if (!agent.working_dir || typeof agent.working_dir !== "string") {
+    return {
+      valid: false,
+      error: "Missing or invalid agents[0].working_dir (must be a string)"
+    };
+  }
+  if (!agent.image || typeof agent.image !== "string") {
+    return {
+      valid: false,
+      error: "Missing or invalid agents[0].image (must be a string)"
+    };
+  }
+  if (!agent.provider || typeof agent.provider !== "string") {
+    return {
+      valid: false,
+      error: "Missing or invalid agents[0].provider (must be a string)"
+    };
+  }
+  const agentVolumes = agent.volumes;
+  if (agentVolumes && Array.isArray(agentVolumes) && agentVolumes.length > 0) {
+    const volumesSection = cfg.volumes;
+    if (!volumesSection || typeof volumesSection !== "object") {
+      return {
+        valid: false,
+        error: "Agent references volumes but no volumes section defined. Each volume must have explicit name and version."
+      };
+    }
+    for (const volDeclaration of agentVolumes) {
+      if (typeof volDeclaration !== "string") {
+        return {
+          valid: false,
+          error: "Volume declaration must be a string in format 'key:/path'"
+        };
+      }
+      const parts = volDeclaration.split(":");
+      if (parts.length !== 2) {
+        return {
+          valid: false,
+          error: `Invalid volume declaration: ${volDeclaration}. Expected format: volume-key:/mount/path`
+        };
+      }
+      const volumeKey = parts[0].trim();
+      const volumeConfig = volumesSection[volumeKey];
+      if (!volumeConfig) {
+        return {
+          valid: false,
+          error: `Volume "${volumeKey}" is not defined in volumes section. Each volume must have explicit name and version.`
+        };
+      }
+      const volError = validateVolumeConfig(volumeKey, volumeConfig);
+      if (volError) {
+        return { valid: false, error: volError };
+      }
+    }
+  }
   return { valid: true };
 }
 
@@ -12598,7 +12699,11 @@ var buildCommand = new Command().name("build").description("Create or update age
     console.log(chalk2.gray(`  Config ID: ${response.configId}`));
     console.log();
     console.log("  Run your agent:");
-    console.log(chalk2.cyan(`    vm0 run ${response.name} "your prompt"`));
+    console.log(
+      chalk2.cyan(
+        `    vm0 run ${response.name} --artifact-name <artifact> "your prompt"`
+      )
+    );
   } catch (error43) {
     if (error43 instanceof Error) {
       if (error43.message.includes("Not authenticated")) {
@@ -12737,7 +12842,7 @@ var ClaudeEventParser = class {
         agentConfigId: event.agentConfigId,
         agentName: event.agentName,
         prompt: event.prompt,
-        dynamicVars: event.dynamicVars
+        templateVars: event.templateVars
       }
     };
   }
@@ -12748,7 +12853,8 @@ var ClaudeEventParser = class {
       data: {
         runId: event.runId,
         checkpointId: event.checkpointId,
-        volumeSnapshots: event.volumeSnapshots
+        agentSessionId: event.agentSessionId,
+        hasArtifact: event.hasArtifact
       }
     };
   }
@@ -12821,9 +12927,7 @@ var EventRenderer = class {
     if (input && typeof input === "object") {
       for (const [key, value] of Object.entries(input)) {
         if (value !== void 0 && value !== null) {
-          const valueStr = String(value);
-          const displayValue = valueStr.length > 100 ? valueStr.substring(0, 100) + "..." : valueStr;
-          console.log(`  ${key}: ${chalk3.gray(displayValue)}`);
+          console.log(`  ${key}: ${chalk3.gray(String(value))}`);
         }
       }
     }
@@ -12834,8 +12938,7 @@ var EventRenderer = class {
     const color = isError ? chalk3.red : chalk3.green;
     console.log(color("[tool_result]") + " " + status);
     const result = String(event.data.result || "");
-    const displayResult = result.length > 200 ? result.substring(0, 200) + "..." : result;
-    console.log(`  ${chalk3.gray(displayResult)}`);
+    console.log(`  ${chalk3.gray(result)}`);
   }
   static renderResult(event) {
     const success2 = Boolean(event.data.success);
@@ -12872,8 +12975,7 @@ var EventRenderer = class {
       console.log(`  Run ID: ${chalk3.gray(String(event.data.runId))}`);
     }
     const prompt = String(event.data.prompt || "");
-    const displayPrompt = prompt.length > 100 ? prompt.substring(0, 100) + "..." : prompt;
-    console.log(`  Prompt: ${chalk3.gray(displayPrompt)}`);
+    console.log(`  Prompt: ${chalk3.gray(prompt)}`);
     if (event.data.agentName) {
       console.log(`  Agent: ${chalk3.gray(String(event.data.agentName))}`);
     }
@@ -12883,9 +12985,11 @@ var EventRenderer = class {
     console.log(
       `  Checkpoint: ${chalk3.gray(String(event.data.checkpointId || ""))}`
     );
-    const volumeSnapshots = Number(event.data.volumeSnapshots || 0);
-    if (volumeSnapshots > 0) {
-      console.log(`  Volume snapshots: ${chalk3.gray(String(volumeSnapshots))}`);
+    console.log(
+      `  Session: ${chalk3.gray(String(event.data.agentSessionId || ""))}`
+    );
+    if (event.data.hasArtifact) {
+      console.log(`  Artifact: ${chalk3.gray("saved")}`);
     }
   }
   static renderVm0Error(event) {
@@ -12909,29 +13013,31 @@ var EventRenderer = class {
 };
 
 // src/commands/run.ts
-function collectEnvVars(value, previous) {
+function collectVars(value, previous) {
   const [key, ...valueParts] = value.split("=");
   const val = valueParts.join("=");
   if (!key || val === void 0 || val === "") {
-    throw new Error(`Invalid env var format: ${value} (expected key=value)`);
+    throw new Error(`Invalid variable format: ${value} (expected key=value)`);
   }
   return { ...previous, [key]: val };
 }
 function isUUID(str) {
   return /^[0-9a-f-]{36}$/i.test(str);
 }
-async function pollEvents(runId) {
+var DEFAULT_TIMEOUT_SECONDS = 60;
+async function pollEvents(runId, timeoutSeconds) {
   let nextSequence = -1;
   let complete = false;
   const pollIntervalMs = 500;
-  const timeoutMs = 3 * 60 * 1e3;
+  const timeoutMs = timeoutSeconds * 1e3;
   const startTime = Date.now();
   while (!complete) {
     const elapsed = Date.now() - startTime;
     if (elapsed > timeoutMs) {
       console.error(
         chalk4.red(
-          "\n\u2717 Agent execution timed out after 3 minutes without receiving events"
+          `
+\u2717 Agent execution timed out after ${timeoutSeconds} seconds without receiving events`
         )
       );
       throw new Error("Agent execution timed out");
@@ -12960,15 +13066,35 @@ var runCmd = new Command2().name("run").description("Execute an agent").argument
   "<identifier>",
   "Agent name or config ID (e.g., 'my-agent' or 'cfg-abc-123')"
 ).argument("<prompt>", "Prompt for the agent").option(
-  "-e, --env <key=value>",
-  "Environment variables (repeatable)",
-  collectEnvVars,
+  "--vars <KEY=value>",
+  "Template variables for config placeholders (repeatable)",
+  collectVars,
   {}
+).option("--artifact-name <name>", "Artifact storage name (required for run)").option(
+  "--artifact-version <hash>",
+  "Artifact version hash (defaults to latest)"
 ).option(
-  "-a, --artifact <key>",
-  "Artifact key to mount (for VM0 driver artifacts)"
+  "-t, --timeout <seconds>",
+  "Polling timeout in seconds (default: 60)",
+  String(DEFAULT_TIMEOUT_SECONDS)
 ).action(
   async (identifier, prompt, options) => {
+    const timeoutSeconds = parseInt(options.timeout, 10);
+    if (isNaN(timeoutSeconds) || timeoutSeconds <= 0) {
+      console.error(
+        chalk4.red("\u2717 Invalid timeout value. Must be a positive number.")
+      );
+      process.exit(1);
+    }
+    if (!options.artifactName) {
+      console.error(
+        chalk4.red("\u2717 Missing required option: --artifact-name <name>")
+      );
+      console.error(
+        chalk4.gray("  The artifact-name is required for new agent runs.")
+      );
+      process.exit(1);
+    }
     try {
       let configId;
       if (isUUID(identifier)) {
@@ -12994,13 +13120,16 @@ var runCmd = new Command2().name("run").description("Execute an agent").argument
       }
       console.log(chalk4.blue("\nCreating agent run..."));
       console.log(chalk4.gray(`  Prompt: ${prompt}`));
-      if (Object.keys(options.env).length > 0) {
+      if (Object.keys(options.vars).length > 0) {
         console.log(
-          chalk4.gray(`  Variables: ${JSON.stringify(options.env)}`)
+          chalk4.gray(`  Variables: ${JSON.stringify(options.vars)}`)
         );
       }
-      if (options.artifact) {
-        console.log(chalk4.gray(`  Artifact: ${options.artifact}`));
+      console.log(chalk4.gray(`  Artifact: ${options.artifactName}`));
+      if (options.artifactVersion) {
+        console.log(
+          chalk4.gray(`  Artifact version: ${options.artifactVersion}`)
+        );
       }
       console.log();
       console.log(chalk4.blue("Executing in sandbox..."));
@@ -13008,10 +13137,11 @@ var runCmd = new Command2().name("run").description("Execute an agent").argument
       const response = await apiClient.createRun({
         agentConfigId: configId,
         prompt,
-        dynamicVars: Object.keys(options.env).length > 0 ? options.env : void 0,
-        artifactKey: options.artifact
+        templateVars: Object.keys(options.vars).length > 0 ? options.vars : void 0,
+        artifactName: options.artifactName,
+        artifactVersion: options.artifactVersion
       });
-      await pollEvents(response.runId);
+      await pollEvents(response.runId, timeoutSeconds);
     } catch (error43) {
       if (error43 instanceof Error) {
         if (error43.message.includes("Not authenticated")) {
@@ -13034,42 +13164,113 @@ var runCmd = new Command2().name("run").description("Execute an agent").argument
     }
   }
 );
-runCmd.command("resume").description("Resume an agent run from a checkpoint").argument("<checkpointId>", "Checkpoint ID to resume from").argument("<prompt>", "Prompt for the resumed agent").action(async (checkpointId, prompt) => {
-  try {
-    if (!isUUID(checkpointId)) {
+runCmd.command("resume").description("Resume an agent run from a checkpoint (uses all snapshot data)").argument("<checkpointId>", "Checkpoint ID to resume from").argument("<prompt>", "Prompt for the resumed agent").option(
+  "-t, --timeout <seconds>",
+  "Polling timeout in seconds (default: 60)",
+  String(DEFAULT_TIMEOUT_SECONDS)
+).action(
+  async (checkpointId, prompt, options) => {
+    const timeoutSeconds = parseInt(options.timeout, 10);
+    if (isNaN(timeoutSeconds) || timeoutSeconds <= 0) {
       console.error(
-        chalk4.red(`\u2717 Invalid checkpoint ID format: ${checkpointId}`)
+        chalk4.red("\u2717 Invalid timeout value. Must be a positive number.")
       );
-      console.error(chalk4.gray("  Checkpoint ID must be a valid UUID"));
       process.exit(1);
     }
-    console.log(chalk4.blue("\nResuming agent run from checkpoint..."));
-    console.log(chalk4.gray(`  Checkpoint ID: ${checkpointId}`));
-    console.log(chalk4.gray(`  Prompt: ${prompt}`));
-    console.log();
-    console.log(chalk4.blue("Executing in sandbox..."));
-    console.log();
-    const response = await apiClient.resumeRun({
-      checkpointId,
-      prompt
-    });
-    await pollEvents(response.runId);
-  } catch (error43) {
-    if (error43 instanceof Error) {
-      if (error43.message.includes("Not authenticated")) {
-        console.error(chalk4.red("\u2717 Not authenticated. Run: vm0 auth login"));
-      } else if (error43.message.includes("not found")) {
-        console.error(chalk4.red(`\u2717 Checkpoint not found: ${checkpointId}`));
+    try {
+      if (!isUUID(checkpointId)) {
+        console.error(
+          chalk4.red(`\u2717 Invalid checkpoint ID format: ${checkpointId}`)
+        );
+        console.error(chalk4.gray("  Checkpoint ID must be a valid UUID"));
+        process.exit(1);
+      }
+      console.log(chalk4.blue("\nResuming agent run from checkpoint..."));
+      console.log(chalk4.gray(`  Checkpoint ID: ${checkpointId}`));
+      console.log(chalk4.gray(`  Prompt: ${prompt}`));
+      console.log();
+      console.log(chalk4.blue("Executing in sandbox..."));
+      console.log();
+      const response = await apiClient.resumeRun({
+        checkpointId,
+        prompt
+      });
+      await pollEvents(response.runId, timeoutSeconds);
+    } catch (error43) {
+      if (error43 instanceof Error) {
+        if (error43.message.includes("Not authenticated")) {
+          console.error(
+            chalk4.red("\u2717 Not authenticated. Run: vm0 auth login")
+          );
+        } else if (error43.message.includes("not found")) {
+          console.error(chalk4.red(`\u2717 Checkpoint not found: ${checkpointId}`));
+        } else {
+          console.error(chalk4.red("\u2717 Resume failed"));
+          console.error(chalk4.gray(`  ${error43.message}`));
+        }
       } else {
-        console.error(chalk4.red("\u2717 Resume failed"));
-        console.error(chalk4.gray(`  ${error43.message}`));
+        console.error(chalk4.red("\u2717 An unexpected error occurred"));
       }
-    } else {
-      console.error(chalk4.red("\u2717 An unexpected error occurred"));
+      process.exit(1);
     }
-    process.exit(1);
   }
-});
+);
+runCmd.command("continue").description(
+  "Continue an agent run from a session (uses latest artifact version)"
+).argument("<agentSessionId>", "Agent session ID to continue from").argument("<prompt>", "Prompt for the continued agent").option(
+  "-t, --timeout <seconds>",
+  "Polling timeout in seconds (default: 60)",
+  String(DEFAULT_TIMEOUT_SECONDS)
+).action(
+  async (agentSessionId, prompt, options) => {
+    const timeoutSeconds = parseInt(options.timeout, 10);
+    if (isNaN(timeoutSeconds) || timeoutSeconds <= 0) {
+      console.error(
+        chalk4.red("\u2717 Invalid timeout value. Must be a positive number.")
+      );
+      process.exit(1);
+    }
+    try {
+      if (!isUUID(agentSessionId)) {
+        console.error(
+          chalk4.red(`\u2717 Invalid agent session ID format: ${agentSessionId}`)
+        );
+        console.error(chalk4.gray("  Agent session ID must be a valid UUID"));
+        process.exit(1);
+      }
+      console.log(chalk4.blue("\nContinuing agent run from session..."));
+      console.log(chalk4.gray(`  Session ID: ${agentSessionId}`));
+      console.log(chalk4.gray(`  Prompt: ${prompt}`));
+      console.log(chalk4.gray(`  Note: Using latest artifact version`));
+      console.log();
+      console.log(chalk4.blue("Executing in sandbox..."));
+      console.log();
+      const response = await apiClient.continueSession({
+        agentSessionId,
+        prompt
+      });
+      await pollEvents(response.runId, timeoutSeconds);
+    } catch (error43) {
+      if (error43 instanceof Error) {
+        if (error43.message.includes("Not authenticated")) {
+          console.error(
+            chalk4.red("\u2717 Not authenticated. Run: vm0 auth login")
+          );
+        } else if (error43.message.includes("not found")) {
+          console.error(
+            chalk4.red(`\u2717 Agent session not found: ${agentSessionId}`)
+          );
+        } else {
+          console.error(chalk4.red("\u2717 Continue failed"));
+          console.error(chalk4.gray(`  ${error43.message}`));
+        }
+      } else {
+        console.error(chalk4.red("\u2717 An unexpected error occurred"));
+      }
+      process.exit(1);
+    }
+  }
+);
 var runCommand = runCmd;
 
 // src/commands/volume/index.ts
@@ -13269,9 +13470,80 @@ var pushCommand = new Command4().name("push").description("Push local files to c
 // src/commands/volume/pull.ts
 import { Command as Command5 } from "commander";
 import chalk7 from "chalk";
-import path4 from "path";
-import * as fs2 from "fs";
+import path5 from "path";
+import * as fs3 from "fs";
 import AdmZip2 from "adm-zip";
+
+// src/lib/file-utils.ts
+import * as fs2 from "fs";
+import * as path4 from "path";
+function getRemoteFilesFromZip(zipEntries) {
+  const remoteFiles = /* @__PURE__ */ new Set();
+  for (const entry of zipEntries) {
+    if (!entry.isDirectory) {
+      remoteFiles.add(entry.entryName.replace(/\\/g, "/"));
+    }
+  }
+  return remoteFiles;
+}
+async function listLocalFiles(dir, excludeDirs = [".vm0"]) {
+  const files = [];
+  async function walkDir(currentDir, relativePath = "") {
+    const entries = await fs2.promises.readdir(currentDir, {
+      withFileTypes: true
+    });
+    for (const entry of entries) {
+      const entryRelativePath = relativePath ? path4.join(relativePath, entry.name) : entry.name;
+      if (entry.isDirectory()) {
+        if (!excludeDirs.includes(entry.name)) {
+          await walkDir(path4.join(currentDir, entry.name), entryRelativePath);
+        }
+      } else {
+        files.push(entryRelativePath);
+      }
+    }
+  }
+  await walkDir(dir);
+  return files;
+}
+async function removeExtraFiles(dir, remoteFiles, excludeDirs = [".vm0"]) {
+  const localFiles = await listLocalFiles(dir, excludeDirs);
+  let removedCount = 0;
+  for (const localFile of localFiles) {
+    const normalizedPath = localFile.replace(/\\/g, "/");
+    if (!remoteFiles.has(normalizedPath)) {
+      const fullPath = path4.join(dir, localFile);
+      await fs2.promises.unlink(fullPath);
+      removedCount++;
+    }
+  }
+  await removeEmptyDirs(dir, excludeDirs);
+  return removedCount;
+}
+async function removeEmptyDirs(dir, excludeDirs = [".vm0"]) {
+  const entries = await fs2.promises.readdir(dir, { withFileTypes: true });
+  let isEmpty = true;
+  for (const entry of entries) {
+    const fullPath = path4.join(dir, entry.name);
+    if (entry.isDirectory()) {
+      if (excludeDirs.includes(entry.name)) {
+        isEmpty = false;
+      } else {
+        const subDirEmpty = await removeEmptyDirs(fullPath, excludeDirs);
+        if (subDirEmpty) {
+          await fs2.promises.rmdir(fullPath);
+        } else {
+          isEmpty = false;
+        }
+      }
+    } else {
+      isEmpty = false;
+    }
+  }
+  return isEmpty;
+}
+
+// src/commands/volume/pull.ts
 function formatBytes2(bytes) {
   if (bytes === 0) return "0 B";
   const k = 1024;
@@ -13279,7 +13551,7 @@ function formatBytes2(bytes) {
   const i = Math.floor(Math.log(bytes) / Math.log(k));
   return `${(bytes / Math.pow(k, i)).toFixed(2)} ${sizes[i]}`;
 }
-var pullCommand = new Command5().name("pull").description("Pull cloud files to local directory").action(async () => {
+var pullCommand = new Command5().name("pull").description("Pull cloud files to local directory").argument("[versionId]", "Version ID to pull (default: latest)").action(async (versionId) => {
   try {
     const cwd = process.cwd();
     const config2 = await readStorageConfig(cwd);
@@ -13288,11 +13560,19 @@ var pullCommand = new Command5().name("pull").description("Pull cloud files to l
       console.error(chalk7.gray("  Run: vm0 volume init"));
       process.exit(1);
     }
-    console.log(chalk7.cyan(`Pulling volume: ${config2.name}`));
+    if (versionId) {
+      console.log(
+        chalk7.cyan(`Pulling volume: ${config2.name} (version: ${versionId})`)
+      );
+    } else {
+      console.log(chalk7.cyan(`Pulling volume: ${config2.name}`));
+    }
     console.log(chalk7.gray("Downloading..."));
-    const response = await apiClient.get(
-      `/api/storages?name=${encodeURIComponent(config2.name)}`
-    );
+    let url2 = `/api/storages?name=${encodeURIComponent(config2.name)}`;
+    if (versionId) {
+      url2 += `&version=${encodeURIComponent(versionId)}`;
+    }
+    const response = await apiClient.get(url2);
     if (!response.ok) {
       if (response.status === 404) {
         console.error(chalk7.red(`\u2717 Volume "${config2.name}" not found`));
@@ -13316,14 +13596,22 @@ var pullCommand = new Command5().name("pull").description("Pull cloud files to l
     console.log(chalk7.gray("Extracting files..."));
     const zip = new AdmZip2(zipBuffer);
     const zipEntries = zip.getEntries();
+    const remoteFiles = getRemoteFilesFromZip(zipEntries);
+    console.log(chalk7.gray("Syncing local files..."));
+    const removedCount = await removeExtraFiles(cwd, remoteFiles);
+    if (removedCount > 0) {
+      console.log(
+        chalk7.green(`\u2713 Removed ${removedCount} files not in remote`)
+      );
+    }
     let extractedCount = 0;
     for (const entry of zipEntries) {
       if (!entry.isDirectory) {
-        const targetPath = path4.join(cwd, entry.entryName);
-        const dir = path4.dirname(targetPath);
-        await fs2.promises.mkdir(dir, { recursive: true });
+        const targetPath = path5.join(cwd, entry.entryName);
+        const dir = path5.dirname(targetPath);
+        await fs3.promises.mkdir(dir, { recursive: true });
         const data = entry.getData();
-        await fs2.promises.writeFile(targetPath, data);
+        await fs3.promises.writeFile(targetPath, data);
         extractedCount++;
       }
     }
@@ -13346,11 +13634,11 @@ import { Command as Command10 } from "commander";
 // src/commands/artifact/init.ts
 import { Command as Command7 } from "commander";
 import chalk8 from "chalk";
-import path5 from "path";
+import path6 from "path";
 var initCommand2 = new Command7().name("init").description("Initialize an artifact in the current directory").action(async () => {
   try {
     const cwd = process.cwd();
-    const dirName = path5.basename(cwd);
+    const dirName = path6.basename(cwd);
     const existingConfig = await readStorageConfig(cwd);
     if (existingConfig) {
       if (existingConfig.type === "artifact") {
@@ -13372,7 +13660,7 @@ var initCommand2 = new Command7().name("init").description("Initialize an artifa
         );
       }
       console.log(
-        chalk8.gray(`Config file: ${path5.join(cwd, ".vm0", "storage.yaml")}`)
+        chalk8.gray(`Config file: ${path6.join(cwd, ".vm0", "storage.yaml")}`)
       );
       return;
     }
@@ -13393,7 +13681,7 @@ var initCommand2 = new Command7().name("init").description("Initialize an artifa
     console.log(chalk8.green(`\u2713 Initialized artifact: ${artifactName}`));
     console.log(
       chalk8.gray(
-        `\u2713 Config saved to ${path5.join(cwd, ".vm0", "storage.yaml")}`
+        `\u2713 Config saved to ${path6.join(cwd, ".vm0", "storage.yaml")}`
       )
     );
   } catch (error43) {
@@ -13408,15 +13696,15 @@ var initCommand2 = new Command7().name("init").description("Initialize an artifa
 // src/commands/artifact/push.ts
 import { Command as Command8 } from "commander";
 import chalk9 from "chalk";
-import path6 from "path";
-import * as fs3 from "fs";
+import path7 from "path";
+import * as fs4 from "fs";
 import AdmZip3 from "adm-zip";
 async function getAllFiles2(dirPath, baseDir = dirPath) {
   const files = [];
-  const entries = await fs3.promises.readdir(dirPath, { withFileTypes: true });
+  const entries = await fs4.promises.readdir(dirPath, { withFileTypes: true });
   for (const entry of entries) {
-    const fullPath = path6.join(dirPath, entry.name);
-    const relativePath = path6.relative(baseDir, fullPath);
+    const fullPath = path7.join(dirPath, entry.name);
+    const relativePath = path7.relative(baseDir, fullPath);
     if (relativePath.startsWith(".vm0")) {
       continue;
     }
@@ -13463,7 +13751,7 @@ var pushCommand2 = new Command8().name("push").description("Push local files to
     }
     let totalSize = 0;
     for (const file2 of files) {
-      const stats = await fs3.promises.stat(file2);
+      const stats = await fs4.promises.stat(file2);
       totalSize += stats.size;
     }
     console.log(
@@ -13472,8 +13760,8 @@ var pushCommand2 = new Command8().name("push").description("Push local files to
     console.log(chalk9.gray("Compressing files..."));
     const zip = new AdmZip3();
     for (const file2 of files) {
-      const relativePath = path6.relative(cwd, file2);
-      zip.addLocalFile(file2, path6.dirname(relativePath));
+      const relativePath = path7.relative(cwd, file2);
+      zip.addLocalFile(file2, path7.dirname(relativePath));
     }
     const zipBuffer = zip.toBuffer();
     console.log(
@@ -13512,8 +13800,8 @@ var pushCommand2 = new Command8().name("push").description("Push local files to
 // src/commands/artifact/pull.ts
 import { Command as Command9 } from "commander";
 import chalk10 from "chalk";
-import path7 from "path";
-import * as fs4 from "fs";
+import path8 from "path";
+import * as fs5 from "fs";
 import AdmZip4 from "adm-zip";
 function formatBytes4(bytes) {
   if (bytes === 0) return "0 B";
@@ -13522,7 +13810,7 @@ function formatBytes4(bytes) {
   const i = Math.floor(Math.log(bytes) / Math.log(k));
   return `${(bytes / Math.pow(k, i)).toFixed(2)} ${sizes[i]}`;
 }
-var pullCommand2 = new Command9().name("pull").description("Pull cloud artifact to local directory").action(async () => {
+var pullCommand2 = new Command9().name("pull").description("Pull cloud artifact to local directory").argument("[versionId]", "Version ID to pull (default: latest)").action(async (versionId) => {
   try {
     const cwd = process.cwd();
     const config2 = await readStorageConfig(cwd);
@@ -13540,11 +13828,21 @@ var pullCommand2 = new Command9().name("pull").description("Pull cloud artifact
       console.error(chalk10.gray("  Use: vm0 volume pull"));
       process.exit(1);
     }
-    console.log(chalk10.cyan(`Pulling artifact: ${config2.name}`));
+    if (versionId) {
+      console.log(
+        chalk10.cyan(
+          `Pulling artifact: ${config2.name} (version: ${versionId})`
+        )
+      );
+    } else {
+      console.log(chalk10.cyan(`Pulling artifact: ${config2.name}`));
+    }
     console.log(chalk10.gray("Downloading..."));
-    const response = await apiClient.get(
-      `/api/storages?name=${encodeURIComponent(config2.name)}`
-    );
+    let url2 = `/api/storages?name=${encodeURIComponent(config2.name)}`;
+    if (versionId) {
+      url2 += `&version=${encodeURIComponent(versionId)}`;
+    }
+    const response = await apiClient.get(url2);
     if (!response.ok) {
       if (response.status === 404) {
         console.error(chalk10.red(`\u2717 Artifact "${config2.name}" not found`));
@@ -13568,14 +13866,22 @@ var pullCommand2 = new Command9().name("pull").description("Pull cloud artifact
     console.log(chalk10.gray("Extracting files..."));
     const zip = new AdmZip4(zipBuffer);
     const zipEntries = zip.getEntries();
+    const remoteFiles = getRemoteFilesFromZip(zipEntries);
+    console.log(chalk10.gray("Syncing local files..."));
+    const removedCount = await removeExtraFiles(cwd, remoteFiles);
+    if (removedCount > 0) {
+      console.log(
+        chalk10.green(`\u2713 Removed ${removedCount} files not in remote`)
+      );
+    }
     let extractedCount = 0;
     for (const entry of zipEntries) {
       if (!entry.isDirectory) {
-        const targetPath = path7.join(cwd, entry.entryName);
-        const dir = path7.dirname(targetPath);
-        await fs4.promises.mkdir(dir, { recursive: true });
+        const targetPath = path8.join(cwd, entry.entryName);
+        const dir = path8.dirname(targetPath);
+        await fs5.promises.mkdir(dir, { recursive: true });
         const data = entry.getData();
-        await fs4.promises.writeFile(targetPath, data);
+        await fs5.promises.writeFile(targetPath, data);
         extractedCount++;
       }
     }

package/package.json

@@ -1,7 +1,12 @@
 {
   "name": "@vm0/cli",
-  "version": "1.5.0",
+  "version": "1.8.1",
   "description": "CLI application",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/vm0-ai/vm0",
+    "directory": "turbo/apps/cli"
+  },
   "type": "module",
   "bin": {
     "vm0": "index.js"