cassian-cli 0.1.1 → 0.1.3

package/dist/commands/init.d.ts

@@ -1 +1,12 @@
-export declare function init(): Promise<void>;
+export interface InitOptions {
+    name?: string;
+    gpu?: string;
+    gpuCount?: string;
+    memory?: string;
+    shm?: string;
+    disk?: string;
+    setup?: string;
+    syncignore?: string;
+    yes?: boolean;
+}
+export declare function init(options?: InitOptions): Promise<void>;

package/dist/commands/init.js

@@ -2,11 +2,18 @@ import { createInterface } from "readline";
 import { writeFileSync, existsSync } from "fs";
 import { green, dim, bold } from "../lib/output.js";
 import { fatal } from "../lib/errors.js";
+const GPU_OPTIONS = [
+    { label: "H100 SXM", value: "h100-sxm", hint: "80GB — best for large training" },
+    { label: "H100 PCIe", value: "h100-pcie", hint: "80GB" },
+    { label: "A100 SXM", value: "a100-sxm", hint: "80GB — reliable workhorse" },
+    { label: "A100 PCIe", value: "a100-pcie", hint: "80GB" },
+    { label: "L40S", value: "l40s", hint: "48GB — great for inference" },
+    { label: "RTX 4090", value: "rtx4090", hint: "24GB — fast, affordable" },
+    { label: "RTX 3090", value: "rtx3090", hint: "24GB" },
+];
 function prompt(rl, question, fallback) {
     return new Promise((resolve) => {
-        rl.question(question, (answer) => {
-            resolve(answer.trim() || fallback);
-        });
+        rl.question(question, (answer) => resolve(answer.trim() || fallback));
     });
 }
 function promptChoice(rl, question, choices, fallback) {
@@ -19,116 +26,109 @@ function promptChoice(rl, question, choices, fallback) {
         });
         rl.question(dim(`  [1-${choices.length}, default ${choices.findIndex(c => c.value === fallback) + 1}]: `), (answer) => {
             const idx = parseInt(answer.trim()) - 1;
-            if (idx >= 0 && idx < choices.length)
-                resolve(choices[idx].value);
-            else
-                resolve(fallback);
+            resolve(idx >= 0 && idx < choices.length ? choices[idx].value : fallback);
         });
     });
 }
-export async function init() {
+function buildYaml(opts) {
+    const volName = opts.name.replace(/[^a-z0-9-]/g, "-") + "-data";
+    const syncignoreLines = opts.syncignore.map(p => `    - "${p}"`).join("\n");
+    const setupLine = opts.setup ? `  setup: ${opts.setup}\n` : "";
+    return `name: ${opts.name}
+
+gpu:
+  count: ${opts.gpuCount}
+  type: ${opts.gpuType}
+
+image: default
+
+volumes:
+  - name: ${volName}
+    size: ${opts.diskSize}
+    mount: /workspace
+
+resources:
+  memory: ${opts.memory}
+  shm_size: ${opts.shmSize}
+
+workspace:
+${setupLine}  syncignore:
+${syncignoreLines}
+`;
+}
+export async function init(options = {}) {
     if (existsSync("cassian.yaml")) {
         fatal("cassian.yaml already exists in this directory.");
     }
+    const dirName = process.cwd().split("/").pop().toLowerCase().replace(/[^a-z0-9-]/g, "-");
+    const skipPrompts = options.yes ?? false;
+    // If all required flags are provided, skip the wizard entirely
+    const allFlagsProvided = !!(options.name && options.gpu && options.gpuCount && options.memory && options.disk);
+    if (allFlagsProvided || skipPrompts) {
+        const syncignore = options.syncignore
+            ? options.syncignore.split(",").map(s => s.trim()).filter(Boolean)
+            : ["outputs/**", "*.ckpt", "*.safetensors", "**/.cache/**"];
+        const yaml = buildYaml({
+            name: options.name || dirName,
+            gpuType: options.gpu || "rtx3090",
+            gpuCount: parseInt(options.gpuCount || "1"),
+            memory: options.memory || "32G",
+            shmSize: options.shm || "16G",
+            diskSize: options.disk || "50G",
+            setup: options.setup || (existsSync("requirements.txt") ? "pip install -r /workspace/requirements.txt" : ""),
+            syncignore,
+        });
+        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();
+        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"));
-    // Instance name
-    const dirName = process.cwd().split("/").pop().toLowerCase().replace(/[^a-z0-9-]/g, "-");
-    const name = await prompt(rl, `  Instance name ${dim(`[${dirName}]`)}: `, dirName);
-    // GPU type
+    console.log(dim("  Press enter to accept defaults. Or pass flags to skip: cassian init --help\n"));
+    const name = options.name || await prompt(rl, `  Instance name ${dim(`[${dirName}]`)}: `, dirName);
     console.log();
-    const gpuType = await promptChoice(rl, "  GPU type:", [
-        { label: "H100 SXM", value: "h100-sxm", hint: "80GB — best for large training runs" },
-        { label: "H100 PCIe", value: "h100-pcie", hint: "80GB" },
-        { label: "A100 SXM", value: "a100-sxm", hint: "80GB — reliable workhorse" },
-        { label: "A100 PCIe", value: "a100-pcie", hint: "80GB" },
-        { label: "L40S", value: "l40s", hint: "48GB — great for inference" },
-        { label: "RTX 4090", value: "rtx4090", hint: "24GB — fast, affordable" },
-        { label: "RTX 3090", value: "rtx3090", hint: "24GB" },
-    ], "rtx3090");
-    // GPU count
+    const gpuType = options.gpu || await promptChoice(rl, "  GPU type:", GPU_OPTIONS, "rtx3090");
     console.log();
-    const gpuCountStr = await promptChoice(rl, "  Number of GPUs:", [
-        { label: "1", value: "1" },
-        { label: "2", value: "2" },
-        { label: "4", value: "4" },
-        { label: "8", value: "8" },
+    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 gpuCount = parseInt(gpuCountStr);
-    // RAM
     console.log();
-    const memory = await promptChoice(rl, "  RAM:", [
-        { label: "16 GB", value: "16G" },
-        { label: "32 GB", value: "32G", hint: "recommended" },
-        { label: "64 GB", value: "64G" },
-        { label: "128 GB", value: "128G" },
-        { label: "256 GB", value: "256G" },
+    const memory = options.memory || await promptChoice(rl, "  RAM:", [
+        { label: "16 GB", value: "16G" }, { label: "32 GB", value: "32G", hint: "recommended" },
+        { label: "64 GB", value: "64G" }, { label: "128 GB", value: "128G" }, { label: "256 GB", value: "256G" },
     ], "32G");
-    // Shared memory (for PyTorch DataLoader)
     console.log();
-    const shmSize = await promptChoice(rl, "  Shared memory (for PyTorch DataLoader):", [
-        { label: "4 GB", value: "4G" },
-        { label: "8 GB", value: "8G" },
-        { label: "16 GB", value: "16G", hint: "recommended" },
-        { label: "32 GB", value: "32G" },
+    const shmSize = options.shm || await promptChoice(rl, "  Shared memory (PyTorch DataLoader):", [
+        { label: "4 GB", value: "4G" }, { label: "8 GB", value: "8G" },
+        { label: "16 GB", value: "16G", hint: "recommended" }, { label: "32 GB", value: "32G" },
     ], "16G");
-    // Disk size
     console.log();
-    const diskSize = await promptChoice(rl, "  Workspace disk size:", [
-        { 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", hint: "for large datasets / checkpoints" },
-        { label: "1 TB", value: "1T" },
+    const diskSize = options.disk || await promptChoice(rl, "  Workspace disk size:", [
+        { 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", hint: "for large datasets / checkpoints" }, { label: "1 TB", value: "1T" },
     ], "50G");
-    // Setup command
     console.log();
     const hasReqs = existsSync("requirements.txt");
-    const hasPyproject = existsSync("pyproject.toml");
-    let defaultSetup = "";
-    if (hasReqs)
-        defaultSetup = "pip install -r /workspace/requirements.txt";
-    else if (hasPyproject)
-        defaultSetup = "pip install -e /workspace";
-    const setupRaw = await prompt(rl, `  Setup command ${dim(`[${defaultSetup || "skip"}]`)}: `, defaultSetup);
+    const defaultSetup = hasReqs ? "pip install -r /workspace/requirements.txt" : existsSync("pyproject.toml") ? "pip install -e /workspace" : "";
+    const setupRaw = options.setup ?? await prompt(rl, `  Setup command ${dim(`[${defaultSetup || "skip"}]`)}: `, defaultSetup);
     const setup = setupRaw === "skip" ? "" : setupRaw;
-    // Syncignore — what not to sync
     console.log();
-    console.log(dim("  Patterns to exclude from sync (like .gitignore)."));
-    console.log(dim("  Separate with commas, or press enter for defaults."));
-    const syncignoreRaw = await prompt(rl, `  Syncignore ${dim("[outputs/**, *.ckpt, *.safetensors, **/.cache/**]")}: `, "");
-    const customIgnore = syncignoreRaw
+    console.log(dim("  Patterns to exclude from sync (comma-separated, or enter for defaults)."));
+    const syncignoreRaw = options.syncignore ?? await prompt(rl, `  Syncignore ${dim("[outputs/**, *.ckpt, *.safetensors, **/.cache/**]")}: `, "");
+    const syncignore = syncignoreRaw
         ? syncignoreRaw.split(",").map(s => s.trim()).filter(Boolean)
         : ["outputs/**", "*.ckpt", "*.safetensors", "**/.cache/**"];
     rl.close();
-    // Build yaml
-    const volName = name.replace(/[^a-z0-9-]/g, "-") + "-data";
-    const syncignoreLines = customIgnore.map(p => `    - "${p}"`).join("\n");
-    const setupLine = setup ? `  setup: ${setup}\n` : "";
-    const yaml = `name: ${name}
-
-gpu:
-  count: ${gpuCount}
-  type: ${gpuType}
-
-image: default
-
-volumes:
-  - name: ${volName}
-    size: ${diskSize}
-    mount: /workspace
-
-resources:
-  memory: ${memory}
-  shm_size: ${shmSize}
-
-workspace:
-${setupLine}  syncignore:
-${syncignoreLines}
-`;
+    const yaml = buildYaml({ name, gpuType, gpuCount, memory, shmSize, diskSize, setup, syncignore });
     writeFileSync("cassian.yaml", yaml);
     console.log();
     console.log(`  ${green("✓")} Created cassian.yaml`);

package/dist/commands/up.js

@@ -100,6 +100,20 @@ export async function up() {
             await new Promise((r) => setTimeout(r, 1000));
             spinner.text = "Starting instance...";
         }
+        // Validate GPU type against what the host offers before provisioning
+        if (config.gpu.type) {
+            const gpus = await client.get("/v1/gpus").catch(() => null);
+            if (gpus && gpus.gpus.length > 0) {
+                const available = gpus.gpus[0].name.toLowerCase().replace(/[\s_]/g, "-");
+                const requested = config.gpu.type.toLowerCase().replace(/[\s_]/g, "-");
+                // Check if the requested type is a substring match of what's available
+                if (!available.includes(requested) && !requested.includes(available.split("-").pop())) {
+                    spinner.fail("GPU type not available");
+                    const names = [...new Set(gpus.gpus.map((g) => g.name))];
+                    fatal(`GPU type '${config.gpu.type}' is not available on this host.`, `Available: ${names.join(", ")} — update gpu.type in cassian.yaml`);
+                }
+            }
+        }
         const instance = await client.post("/v1/instances", {
             name: config.name,
             gpu_count: config.gpu.count,

package/dist/index.js

@@ -73,6 +73,25 @@ volume
     .action((opts) => volumeDelete(opts.name));
 program
     .command("init")
-    .description("Set up a new Cassian project interactively")
-    .action(init);
+    .description("Set up a new Cassian project (interactive or via flags)")
+    .option("--name <name>", "Instance name")
+    .option("--gpu <type>", "GPU type (e.g. h100-sxm, a100-sxm, rtx3090)")
+    .option("--gpu-count <n>", "Number of GPUs")
+    .option("--memory <size>", "RAM (e.g. 32G, 128G)")
+    .option("--shm <size>", "Shared memory for PyTorch (e.g. 16G)")
+    .option("--disk <size>", "Workspace disk size (e.g. 50G, 1T)")
+    .option("--setup <cmd>", "Setup command to run after sync")
+    .option("--syncignore <list>", "Comma-separated patterns to exclude from sync")
+    .option("-y, --yes", "Skip all prompts, use defaults or provided flags")
+    .action((opts) => init({
+    name: opts.name,
+    gpu: opts.gpu,
+    gpuCount: opts.gpuCount,
+    memory: opts.memory,
+    shm: opts.shm,
+    disk: opts.disk,
+    setup: opts.setup,
+    syncignore: opts.syncignore,
+    yes: opts.yes,
+}));
 program.parse();

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "cassian-cli",
-  "version": "0.1.1",
+  "version": "0.1.3",
   "description": "The Cassian GPU cloud CLI — provision GPUs, sync files, and run workloads from your terminal.",
   "type": "module",
   "bin": {