@vellumai/cli 0.1.2 → 0.1.3

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@vellumai/cli",
-  "version": "0.1.2",
+  "version": "0.1.3",
   "description": "CLI tools for vellum-assistant",
   "type": "module",
   "bin": {

package/src/commands/hatch.ts

@@ -2,7 +2,7 @@ import { spawn } from "child_process";
 import { randomBytes } from "crypto";
 import { existsSync, unlinkSync, writeFileSync } from "fs";
 import { tmpdir, userInfo } from "os";
-import { join } from "path";
+import { dirname, join } from "path";
 
 import { buildOpenclawStartupScript } from "../adapters/openclaw";
 import { saveAssistantEntry } from "../lib/assistant-config";
@@ -722,21 +722,48 @@ async function hatchLocal(species: Species, name: string | null): Promise<void>
   console.log("");
 
   console.log("🔨 Starting local daemon...");
-  const child = spawn("bunx", ["vellum", "daemon", "start"], {
-    stdio: "inherit",
-    env: { ...process.env },
-  });
+  const daemonBinary = join(dirname(process.execPath), "vellum-daemon");
 
-  await new Promise<void>((resolve, reject) => {
-    child.on("close", (code) => {
-      if (code === 0) {
-        resolve();
-      } else {
-        reject(new Error(`Daemon start exited with code ${code}`));
+  if (existsSync(daemonBinary)) {
+    const child = spawn(daemonBinary, [], {
+      detached: true,
+      stdio: "ignore",
+      env: { ...process.env },
+    });
+    child.unref();
+
+    const homeDir = process.env.HOME ?? userInfo().homedir;
+    const socketPath = join(homeDir, ".vellum", "vellum.sock");
+    const maxWait = 10000;
+    const pollInterval = 100;
+    let waited = 0;
+    while (waited < maxWait) {
+      if (existsSync(socketPath)) {
+        break;
       }
+      await new Promise((r) => setTimeout(r, pollInterval));
+      waited += pollInterval;
+    }
+    if (!existsSync(socketPath)) {
+      console.warn("⚠️  Daemon socket did not appear within 10s — continuing anyway");
+    }
+  } else {
+    const child = spawn("bunx", ["vellum", "daemon", "start"], {
+      stdio: "inherit",
+      env: { ...process.env },
     });
-    child.on("error", reject);
-  });
+
+    await new Promise<void>((resolve, reject) => {
+      child.on("close", (code) => {
+        if (code === 0) {
+          resolve();
+        } else {
+          reject(new Error(`Daemon start exited with code ${code}`));
+        }
+      });
+      child.on("error", reject);
+    });
+  }
 
   console.log("🌐 Starting gateway...");
   const gatewayDir = join(import.meta.dir, "..", "..", "..", "gateway");

package/src/commands/retire.ts

@@ -92,12 +92,22 @@ async function retireCustom(entry: AssistantEntry): Promise<void> {
   console.log(`\u2705 Custom instance retired.`);
 }
 
+function parseSource(): string | undefined {
+  const args = process.argv.slice(4);
+  for (let i = 0; i < args.length; i++) {
+    if (args[i] === "--source" && args[i + 1]) {
+      return args[i + 1];
+    }
+  }
+  return undefined;
+}
+
 export async function retire(): Promise<void> {
   const name = process.argv[3];
 
   if (!name) {
     console.error("Error: Instance name is required.");
-    console.error("Usage: vellum-cli retire <name>");
+    console.error("Usage: vellum-cli retire <name> [--source <source>]");
     process.exit(1);
   }
 
@@ -108,6 +118,7 @@ export async function retire(): Promise<void> {
     process.exit(1);
   }
 
+  const source = parseSource();
   const cloud = resolveCloud(entry);
 
   if (cloud === "gcp") {
@@ -117,14 +128,14 @@ export async function retire(): Promise<void> {
       console.error("Error: GCP project and zone not found in assistant config.");
       process.exit(1);
     }
-    await retireGcpInstance(name, project, zone);
+    await retireGcpInstance(name, project, zone, source);
   } else if (cloud === "aws") {
     const region = entry.region;
     if (!region) {
       console.error("Error: AWS region not found in assistant config.");
       process.exit(1);
     }
-    await retireAwsInstance(name, region);
+    await retireAwsInstance(name, region, source);
   } else if (cloud === "local") {
     await retireLocal();
   } else if (cloud === "custom") {

package/src/lib/aws.ts

@@ -547,6 +547,7 @@ async function getInstanceIdByName(
 export async function retireInstance(
   name: string,
   region: string,
+  source?: string,
 ): Promise<void> {
   const instanceId = await getInstanceIdByName(name, region);
   if (!instanceId) {
@@ -556,6 +557,23 @@ export async function retireInstance(
     return;
   }
 
+  if (source) {
+    try {
+      await exec("aws", [
+        "ec2",
+        "create-tags",
+        "--resources",
+        instanceId,
+        "--tags",
+        `Key=retired-by,Value=${source}`,
+        "--region",
+        region,
+      ]);
+    } catch {
+      console.warn(`\u26a0\ufe0f  Could not tag instance before termination`);
+    }
+  }
+
   console.log(`\u{1F5D1}\ufe0f  Terminating AWS instance ${name} (${instanceId})\n`);
 
   const child = spawnChild(

package/src/lib/gcp.ts

@@ -278,6 +278,7 @@ export async function retireInstance(
   name: string,
   project: string,
   zone: string,
+  source?: string,
 ): Promise<void> {
   const exists = await instanceExists(name, project, zone);
   if (!exists) {
@@ -287,6 +288,22 @@ export async function retireInstance(
     return;
   }
 
+  if (source) {
+    try {
+      await exec("gcloud", [
+        "compute",
+        "instances",
+        "add-labels",
+        name,
+        `--project=${project}`,
+        `--zone=${zone}`,
+        `--labels=retired-by=${source}`,
+      ]);
+    } catch {
+      console.warn(`\u26a0\ufe0f  Could not label instance before deletion`);
+    }
+  }
+
   console.log(`\u{1F5D1}\ufe0f  Deleting GCP instance ${name}\n`);
 
   const child = spawn(