cassian-cli 0.3.5 → 0.3.7

package/dist/commands/init.d.ts

@@ -3,7 +3,6 @@ export interface InitOptions {
     gpu?: string;
     gpuCount?: string;
     disk?: string;
-    storage?: string;
     yes?: boolean;
 }
 export declare function init(options?: InitOptions): Promise<void>;

package/dist/commands/init.js

@@ -1,18 +1,17 @@
-import { createInterface } from "readline";
-import { writeFileSync, existsSync } from "fs";
-import { green, dim, bold } from "../lib/output.js";
-import { fatal } from "../lib/errors.js";
+import { existsSync } from "fs";
+import { writeFileSync } from "fs";
+import * as p from "@clack/prompts";
 import { AGENT_URL } from "../lib/constants.js";
 const ALL_GPUS = [
-    { label: "H100 SXM", value: "h100-sxm", memory: "80GB", hint: "best for large training" },
-    { label: "H100 PCIe", value: "h100-pcie", memory: "80GB" },
-    { label: "A100 SXM", value: "a100-sxm", memory: "80GB", hint: "reliable workhorse" },
-    { label: "A100 PCIe", value: "a100-pcie", memory: "80GB" },
-    { label: "L40S", value: "l40s", memory: "48GB", hint: "great for inference" },
-    { label: "L4", value: "l4", memory: "24GB", hint: "efficient inference" },
-    { label: "A10G", value: "a10g", memory: "24GB", hint: "good all-rounder" },
-    { label: "RTX 4090", value: "rtx4090", memory: "24GB", hint: "fast, affordable" },
-    { label: "RTX 3090", value: "rtx3090", memory: "24GB" },
+    { value: "h100-sxm", label: "H100 SXM", memory: "80GB", hint: "best for large training" },
+    { value: "h100-pcie", label: "H100 PCIe", memory: "80GB" },
+    { value: "a100-sxm", label: "A100 SXM", memory: "80GB", hint: "reliable workhorse" },
+    { value: "a100-pcie", label: "A100 PCIe", memory: "80GB" },
+    { value: "l40s", label: "L40S", memory: "48GB", hint: "great for inference" },
+    { value: "l4", label: "L4", memory: "24GB", hint: "efficient inference" },
+    { value: "a10g", label: "A10G", memory: "24GB", hint: "good all-rounder" },
+    { value: "rtx4090", label: "RTX 4090", memory: "24GB", hint: "fast, affordable" },
+    { value: "rtx3090", label: "RTX 3090", memory: "24GB" },
 ];
 async function fetchAvailability() {
     try {
@@ -29,43 +28,8 @@ async function fetchAvailability() {
         return {};
     }
 }
-function buildGpuOptions(availability) {
-    return ALL_GPUS.map(gpu => {
-        const count = availability[gpu.value] ?? 0;
-        const parts = [gpu.memory];
-        if (gpu.hint)
-            parts.push(gpu.hint);
-        if (count > 0) {
-            parts.push(`${count} available`);
-        }
-        else {
-            parts.push("unavailable");
-        }
-        return { label: gpu.label, value: gpu.value, hint: parts.join(" — ") };
-    });
-}
-function prompt(rl, question, fallback) {
-    return new Promise((resolve) => {
-        rl.question(question, (answer) => resolve(answer.trim() || fallback));
-    });
-}
-function promptChoice(rl, question, choices, fallback) {
-    return new Promise((resolve) => {
-        console.log(question);
-        choices.forEach((c, i) => {
-            const marker = c.value === fallback ? green("›") : " ";
-            const hint = c.hint ? dim(` — ${c.hint}`) : "";
-            console.log(`  ${marker} ${i + 1}) ${c.label}${hint}`);
-        });
-        rl.question(dim(`  [1-${choices.length}, default ${choices.findIndex(c => c.value === fallback) + 1}]: `), (answer) => {
-            const idx = parseInt(answer.trim()) - 1;
-            resolve(idx >= 0 && idx < choices.length ? choices[idx].value : fallback);
-        });
-    });
-}
 function buildYaml(opts) {
-    const storageLines = opts.storage.length
-        ? `\nstorage:\n${opts.storage.map(p => `  - ${p}`).join("\n")}\n` : "";
+    const storageLine = opts.storage ? "\nstorage: true\n" : "";
     const excludeLines = opts.exclude.length
         ? `\nworkspace:\n  exclude:\n${opts.exclude.map(p => `    - "${p}"`).join("\n")}\n` : "";
     return `name: ${opts.name}
@@ -75,76 +39,113 @@ gpu:
   type: ${opts.gpuType}
 
 disk: ${opts.diskSize}
-${storageLines}${excludeLines}`;
+${storageLine}${excludeLines}`;
 }
 export async function init(options = {}) {
     if (existsSync("cassian.yaml")) {
-        fatal("cassian.yaml already exists in this directory.");
+        p.log.error("cassian.yaml already exists in this directory.");
+        process.exit(1);
     }
     const dirName = process.cwd().split("/").pop().toLowerCase().replace(/[^a-z0-9-]/g, "-");
-    const skipPrompts = options.yes ?? false;
-    if (skipPrompts) {
+    if (options.yes) {
         const yaml = buildYaml({
             name: options.name || dirName,
             gpuType: options.gpu || "rtx3090",
             gpuCount: parseInt(options.gpuCount || "1"),
             diskSize: options.disk || "50G",
-            storage: [],
+            storage: true,
             exclude: ["node_modules/", ".venv/", "__pycache__/", "*.pyc", "wandb/"],
         });
         writeFileSync("cassian.yaml", yaml);
-        console.log();
-        console.log(`  ${green("✓")} Created cassian.yaml`);
-        console.log();
-        console.log(`  Run ${green("cassian up")} to start your instance.`);
-        console.log();
+        p.log.success("Created cassian.yaml");
+        p.log.info("Run \x1b[32mcassian up\x1b[0m to start your instance.");
         return;
     }
-    // Interactive wizard
-    const rl = createInterface({ input: process.stdin, output: process.stdout });
-    console.log();
-    console.log(bold("  Let's set up your Cassian project."));
-    console.log(dim("  Press enter to accept defaults.\n"));
-    const name = options.name || await prompt(rl, `  Instance name ${dim(`[${dirName}]`)}: `, dirName);
-    console.log();
+    p.intro("\x1b[36m⚡ Cassian Setup\x1b[0m");
+    const name = options.name || await p.text({
+        message: "Instance name",
+        placeholder: dirName,
+        defaultValue: dirName,
+        validate: (v) => {
+            if (!v || !/^[a-z0-9][a-z0-9-]*$/.test(v))
+                return "Lowercase letters, numbers, and hyphens only";
+        },
+    });
+    if (p.isCancel(name)) {
+        p.cancel("Setup cancelled.");
+        process.exit(0);
+    }
     const availability = await fetchAvailability();
-    const gpuOptions = buildGpuOptions(availability);
-    const availableTypes = Object.keys(availability).filter(k => availability[k] > 0);
-    const defaultGpu = availableTypes[0] || "rtx3090";
-    const gpuType = options.gpu || await promptChoice(rl, "  GPU type:", gpuOptions, defaultGpu);
-    console.log();
-    const gpuCountStr = options.gpuCount || await promptChoice(rl, "  Number of GPUs:", [
-        { label: "1", value: "1" }, { label: "2", value: "2" },
-        { label: "4", value: "4" }, { label: "8", value: "8" },
-    ], "1");
+    const gpuType = options.gpu || await p.select({
+        message: "GPU type",
+        options: ALL_GPUS.map(gpu => {
+            const count = availability[gpu.value] ?? 0;
+            const parts = [gpu.memory];
+            if (gpu.hint)
+                parts.push(gpu.hint);
+            if (count > 0) {
+                parts.push(`\x1b[32m${count} available\x1b[0m`);
+            }
+            else {
+                parts.push(`\x1b[90munavailable\x1b[0m`);
+            }
+            return { value: gpu.value, label: gpu.label, hint: parts.join(" · ") };
+        }),
+        initialValue: Object.keys(availability).find(k => availability[k] > 0) || "rtx3090",
+    });
+    if (p.isCancel(gpuType)) {
+        p.cancel("Setup cancelled.");
+        process.exit(0);
+    }
+    const gpuCountStr = options.gpuCount || await p.select({
+        message: "Number of GPUs",
+        options: [
+            { value: "1", label: "1" },
+            { value: "2", label: "2" },
+            { value: "4", label: "4" },
+            { value: "8", label: "8" },
+        ],
+        initialValue: "1",
+    });
+    if (p.isCancel(gpuCountStr)) {
+        p.cancel("Setup cancelled.");
+        process.exit(0);
+    }
     const gpuCount = parseInt(gpuCountStr);
-    console.log();
-    const diskSize = options.disk || await promptChoice(rl, "  Disk size (fast storage for code + training):", [
-        { label: "20 GB", value: "20G" },
-        { label: "50 GB", value: "50G", hint: "recommended" },
-        { label: "100 GB", value: "100G" },
-        { label: "200 GB", value: "200G" },
-        { label: "500 GB", value: "500G" },
-    ], "50G");
-    console.log();
-    console.log(dim("  Large folders that need more storage (datasets, pretrained models)"));
-    console.log(dim("  These get unlimited storage, accessible at /workspace/<folder>"));
-    const storageRaw = options.storage ?? await prompt(rl, `  Storage folders ${dim("[none]")}: `, "");
-    const storage = storageRaw
-        ? storageRaw.split(",").map(s => s.trim().replace(/\/+$/, "/")).filter(Boolean)
-        : [];
-    console.log();
-    console.log(dim("  Folders to exclude (build artifacts, can be recreated)"));
-    const excludeRaw = await prompt(rl, `  Exclude ${dim("[node_modules/, .venv/, __pycache__/, *.pyc, wandb/]")}: `, "");
-    const exclude = excludeRaw
-        ? excludeRaw.split(",").map(s => s.trim()).filter(Boolean)
-        : ["node_modules/", ".venv/", "__pycache__/", "*.pyc", "wandb/"];
-    rl.close();
-    const yaml = buildYaml({ name, gpuType, gpuCount, diskSize, storage, exclude });
+    const diskSize = options.disk || await p.select({
+        message: "Disk size (fast NVMe storage)",
+        options: [
+            { value: "20G", label: "20 GB" },
+            { value: "50G", label: "50 GB", hint: "recommended" },
+            { value: "100G", label: "100 GB" },
+            { value: "200G", label: "200 GB" },
+            { value: "500G", label: "500 GB" },
+        ],
+        initialValue: "50G",
+    });
+    if (p.isCancel(diskSize)) {
+        p.cancel("Setup cancelled.");
+        process.exit(0);
+    }
+    const storageEnabled = await p.confirm({
+        message: "Enable cloud storage? \x1b[90m(unlimited, persists at /workspace/storage)\x1b[0m",
+        initialValue: true,
+    });
+    if (p.isCancel(storageEnabled)) {
+        p.cancel("Setup cancelled.");
+        process.exit(0);
+    }
+    const excludeRaw = await p.text({
+        message: "Exclude patterns \x1b[90m(comma-separated, not persisted)\x1b[0m",
+        placeholder: "node_modules/, .venv/, __pycache__/, *.pyc, wandb/",
+        defaultValue: "node_modules/, .venv/, __pycache__/, *.pyc, wandb/",
+    });
+    if (p.isCancel(excludeRaw)) {
+        p.cancel("Setup cancelled.");
+        process.exit(0);
+    }
+    const exclude = excludeRaw.split(",").map(s => s.trim()).filter(Boolean);
+    const yaml = buildYaml({ name, gpuType, gpuCount, diskSize, storage: storageEnabled, exclude });
     writeFileSync("cassian.yaml", yaml);
-    console.log();
-    console.log(`  ${green("✓")} Created cassian.yaml`);
-    console.log();
-    console.log(`  Run ${green("cassian up")} to start your instance.`);
-    console.log();
+    p.outro(`\x1b[32m✓\x1b[0m Created cassian.yaml — run \x1b[32mcassian up\x1b[0m to start`);
 }

package/dist/commands/up.js

@@ -58,10 +58,10 @@ export async function up() {
         info("Disk", config.disk);
     }
     else if (config.volumes?.length) {
-        info("Storage", config.volumes.map((v) => `${v.size} at ${v.mount}`).join(", "));
+        info("Disk", config.volumes.map((v) => `${v.size} at ${v.mount}`).join(", "));
     }
-    if (config.storage?.length) {
-        info("Storage", config.storage.join(", "));
+    if (config.storage) {
+        info("Storage", "enabled (/workspace/storage)");
     }
     console.log();
     const vm = dev.vm;
@@ -125,7 +125,7 @@ export async function up() {
             },
             workspace: {
                 no_sync: config.workspace?.no_sync || [],
-                storage: config.storage || config.workspace?.storage || [],
+                storage: config.storage === true ? ["storage"] : (Array.isArray(config.storage) ? config.storage : config.workspace?.storage || []),
                 exclude: config.workspace?.exclude || config.workspace?.syncignore || [],
             },
         });

package/dist/index.js

@@ -88,14 +88,12 @@ program
     .option("--gpu <type>", "GPU type (e.g. h100-sxm, a100-sxm, rtx3090)")
     .option("--gpu-count <n>", "Number of GPUs")
     .option("--disk <size>", "Disk size (e.g. 50G, 200G)")
-    .option("--storage <folders>", "Comma-separated storage folders (e.g. datasets/,models/)")
     .option("-y, --yes", "Skip all prompts, use defaults")
     .action((opts) => init({
     name: opts.name,
     gpu: opts.gpu,
     gpuCount: opts.gpuCount,
     disk: opts.disk,
-    storage: opts.storage,
     yes: opts.yes,
 }));
 program.parse();

package/dist/types.d.ts

@@ -12,7 +12,7 @@ export interface CassianConfig {
     };
     image?: string;
     disk?: string;
-    storage?: string[];
+    storage?: boolean | string[];
     volumes?: Array<{
         name: string;
         size: string;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "cassian-cli",
-  "version": "0.3.5",
+  "version": "0.3.7",
   "description": "The Cassian GPU cloud CLI — provision GPUs, sync files, and run workloads from your terminal.",
   "type": "module",
   "bin": {
@@ -28,6 +28,7 @@
     "node": ">=18"
   },
   "dependencies": {
+    "@clack/prompts": "^1.4.0",
     "@types/tar": "^6.1.13",
     "chalk": "^5.3.0",
     "chokidar": "^5.0.0",