cassian-cli 0.3.0 → 0.3.2

package/dist/commands/init.js

@@ -2,15 +2,48 @@ 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" },
+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" },
 ];
+async function fetchAvailability() {
+    try {
+        const resp = await fetch(`${AGENT_URL}/v1/gpu-types`);
+        if (!resp.ok)
+            return {};
+        const { gpu_types } = await resp.json();
+        const map = {};
+        for (const g of gpu_types)
+            map[g.type.toLowerCase()] = g.available;
+        return map;
+    }
+    catch {
+        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));
@@ -95,7 +128,11 @@ export async function init(options = {}) {
     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 = options.gpu || await promptChoice(rl, "  GPU type:", GPU_OPTIONS, "rtx3090");
+    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" },

package/package.json

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