cassian-cli 0.3.1 → 0.3.2

package/dist/commands/init.js

@@ -3,42 +3,47 @@ import { writeFileSync, existsSync } from "fs";
 import { green, dim, bold } from "../lib/output.js";
 import { fatal } from "../lib/errors.js";
 import { AGENT_URL } from "../lib/constants.js";
-const GPU_METADATA = {
-    "h100": { memory: "80GB", hint: "best for large training" },
-    "a100": { memory: "80GB", hint: "reliable workhorse" },
-    "l40s": { memory: "48GB", hint: "great for inference" },
-    "l4": { memory: "24GB", hint: "efficient inference" },
-    "a10g": { memory: "24GB", hint: "good all-rounder" },
-    "rtx4090": { memory: "24GB", hint: "fast, affordable" },
-    "rtx3090": { memory: "24GB" },
-};
-async function fetchAvailableGpus() {
+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 [];
+            return {};
         const { gpu_types } = await resp.json();
-        const options = [];
-        for (const gpu of gpu_types) {
-            const type = gpu.type.toLowerCase();
-            const meta = GPU_METADATA[type];
-            const label = type.toUpperCase().replace("RTX", "RTX ");
-            const hint = meta
-                ? `${meta.memory}${meta.hint ? ` — ${meta.hint}` : ""} — ${gpu.available} available`
-                : `${gpu.available} available`;
-            options.push({ label, value: type, hint });
-        }
-        return options;
+        const map = {};
+        for (const g of gpu_types)
+            map[g.type.toLowerCase()] = g.available;
+        return map;
     }
     catch {
-        return [];
+        return {};
     }
 }
-const FALLBACK_GPU_OPTIONS = [
-    { label: "RTX 3090", value: "rtx3090", hint: "24GB" },
-    { label: "L4", value: "l4", hint: "24GB — efficient inference" },
-    { label: "A10G", value: "a10g", hint: "24GB — good all-rounder" },
-];
+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));
@@ -123,9 +128,10 @@ 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 liveGpus = await fetchAvailableGpus();
-    const gpuOptions = liveGpus.length > 0 ? liveGpus : FALLBACK_GPU_OPTIONS;
-    const defaultGpu = gpuOptions[0].value;
+    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:", [

package/package.json

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