@ornncompute/cli 1.0.35 → 1.0.48

package/vendor/capabilities/software-catalog.js

@@ -1,6 +1,8 @@
 /** Live orchestrator software dictionary: `GET /v1/orchestrator/catalog/software`. */
 export const SOFTWARE_CATALOG_PATH = "/v1/orchestrator/catalog/software";
-export const SOFTWARE_KINDS = ["python", "package", "cuda", "nvidia_smi", "dcgmi"];
+export const SOFTWARE_KINDS = ["python", "package", "cuda", "nvidia_driver", "dcgmi"];
+/** Kinds a disk build installs per OS release; Python packages install anywhere. */
+const RELEASE_KINDS = ["python", "cuda", "nvidia_driver", "dcgmi"];
 function stringMap(value) {
     if (!value || typeof value !== "object" || Array.isArray(value)) {
         return {};
@@ -13,22 +15,66 @@ function stringMap(value) {
     }
     return out;
 }
+function stringList(value) {
+    return Array.isArray(value)
+        ? value.filter((entry) => typeof entry === "string")
+        : [];
+}
+function releaseSupport(value) {
+    if (!value || typeof value !== "object" || Array.isArray(value)) {
+        return {};
+    }
+    const out = {};
+    for (const [release, kinds] of Object.entries(value)) {
+        if (!kinds || typeof kinds !== "object")
+            continue;
+        const row = kinds;
+        out[release] = Object.fromEntries(RELEASE_KINDS.map((kind) => [kind, stringList(row[kind])]));
+    }
+    return out;
+}
 export function parseSoftwareCatalog(payload) {
     const row = payload && typeof payload === "object" ? payload : {};
-    const python = Array.isArray(row.python)
-        ? row.python.filter((value) => typeof value === "string")
-        : [];
     return {
-        python,
+        python: stringList(row.python),
         packages: stringMap(row.packages),
         cuda: stringMap(row.cuda),
-        nvidia_smi: stringMap(row.nvidia_smi),
+        nvidia_driver: stringMap(row.nvidia_driver),
         dcgmi: stringMap(row.dcgmi),
+        os_releases: releaseSupport(row.os_releases),
     };
 }
 export async function loadSoftwareCatalog(fetchImpl) {
     return parseSoftwareCatalog(await fetchImpl(SOFTWARE_CATALOG_PATH, { method: "GET" }));
 }
+/**
+ * The catalog a disk built on an image of `osRelease` can install: Python,
+ * CUDA, driver, and DCGM choices narrowed to that release (none when it is
+ * unset or unknown); packages unchanged.
+ */
+export function softwareCatalogForOsRelease(catalog, osRelease) {
+    const supported = catalog.os_releases[osRelease ?? ""] ?? {};
+    const allows = (kind, slug) => supported[kind]?.includes(slug) ?? false;
+    const narrow = (kind) => Object.fromEntries(Object.entries(catalog[kind]).filter(([, slug]) => allows(kind, slug)));
+    return {
+        ...catalog,
+        python: catalog.python.filter((version) => allows("python", version)),
+        cuda: narrow("cuda"),
+        nvidia_driver: narrow("nvidia_driver"),
+        dcgmi: narrow("dcgmi"),
+    };
+}
+/** `stack` with every version `catalog` does not offer cleared. */
+export function keepOfferedChoices(stack, catalog) {
+    const offered = (map, slug) => slug && Object.values(map).includes(slug) ? slug : "";
+    return {
+        ...stack,
+        python_version: catalog.python.includes(stack.python_version) ? stack.python_version : "",
+        cuda_version: offered(catalog.cuda, stack.cuda_version),
+        nvidia_driver_version: offered(catalog.nvidia_driver, stack.nvidia_driver_version),
+        dcgmi_version: offered(catalog.dcgmi, stack.dcgmi_version),
+    };
+}
 export function displayForSoftwareSlug(catalog, kind, slug) {
     const map = catalog[kind];
     const hit = Object.entries(map).find(([, value]) => value === slug);