cassian-cli 0.3.0 → 0.3.1

package/dist/commands/init.js

@@ -2,14 +2,42 @@ 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" },
+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() {
+    try {
+        const resp = await fetch(`${AGENT_URL}/v1/gpu-types`);
+        if (!resp.ok)
+            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;
+    }
+    catch {
+        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 prompt(rl, question, fallback) {
     return new Promise((resolve) => {
@@ -95,7 +123,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 gpuType = options.gpu || await promptChoice(rl, "  GPU type:", GPU_OPTIONS, "rtx3090");
+    const liveGpus = await fetchAvailableGpus();
+    const gpuOptions = liveGpus.length > 0 ? liveGpus : FALLBACK_GPU_OPTIONS;
+    const defaultGpu = gpuOptions[0].value;
+    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.1",
   "description": "The Cassian GPU cloud CLI — provision GPUs, sync files, and run workloads from your terminal.",
   "type": "module",
   "bin": {