@ornncompute/cli 0.2.23 → 1.0.25

package/vendor/capabilities/capabilities/clusters.js

@@ -1,5 +1,5 @@
 import { z } from "zod";
-import { defineCapability, preview } from "../capability.js";
+import { defineCapability, preview, } from "../capability.js";
 const reservationInput = z.object({
     reservation_id: z.string().min(1),
     reservationId: z.string().optional(),
@@ -9,12 +9,66 @@ const launchInput = reservationInput.extend({
     node_count: z.number().int().positive().optional(),
     node_ids: z.array(z.string()).optional(),
 });
+const clusterCliInput = z
+    .object({
+    $positionals: z.array(z.string()).optional(),
+    cursor: z.string().optional(),
+    guided: z.boolean().optional(),
+    idempotencyKey: z.string().optional(),
+    limit: z.union([z.string(), z.number()]).optional(),
+    machine: z.string().optional(),
+    machines: z.string().optional(),
+    network: z.enum(["public", "private"]).optional(),
+    noWait: z.boolean().optional(),
+    nodes: z.union([z.string(), z.number()]).optional(),
+    output: z.string().optional(),
+    overwrite: z.boolean().optional(),
+    timeout: z.string().optional(),
+    wait: z.boolean().optional(),
+    yes: z.boolean().optional(),
+})
+    .strict();
 function reservationId(input) {
     return (input.reservation_id || input.reservationId || "").trim();
 }
 function clusterPath(kind, id, suffix) {
     return `/${kind}/reservations/${encodeURIComponent(id)}/${suffix}`;
 }
+function clusterExecutor(context) {
+    const runtime = context.cli?.interaction;
+    if (!runtime?.cluster) {
+        throw new Error("This capability must execute through the Ornn CLI cluster runtime.");
+    }
+    return runtime.cluster.bind(runtime);
+}
+function clusterCliCapability({ acceptedReferences = [], automation, command, description, human, id, mutation, outcome, related = [], review, }) {
+    return defineCapability({
+        id,
+        domain: "clusters",
+        roles: ["user"],
+        description,
+        mutation,
+        input: clusterCliInput,
+        cli: {
+            command,
+            vnext: {
+                arguments: "semantic",
+                audience: "customer",
+                category: "resource",
+                outcome,
+                human,
+                automation,
+                acceptedReferences,
+                ...(review ? { review } : {}),
+                resumableStates: ["working", "ready", "partial", "failed", "ended"],
+                related,
+            },
+        },
+        mcp: false,
+        slack: false,
+        execute: async (context, input) => clusterExecutor(context)(id, input),
+    });
+}
 function clusterPair(kind, domain) {
     const label = kind === "slurm" ? "Slurm" : "Kubernetes";
     return [
@@ -25,6 +79,7 @@ function clusterPair(kind, domain) {
             description: `${label} launch is blocked only when the reservation's nodes already belong to a different controller. Returns eligible=false with reason other_cluster in that case.`,
             mutation: "read",
             http: { method: "GET", path: `/${kind}/reservations/{id}/cluster/eligible` },
+            slack: false,
             input: reservationInput,
             execute: async (ctx, input) => {
                 const parsed = reservationInput.parse(input);
@@ -40,6 +95,7 @@ function clusterPair(kind, domain) {
             description: `Launch a ${label} controller on a reservation. Blocked when ${kind}.eligible reports other_cluster.`,
             mutation: "preview-confirm",
             http: { method: "POST", path: `/${kind}/reservations/{id}/cluster/launch` },
+            slack: false,
             input: launchInput,
             execute: async (ctx, input) => {
                 const parsed = launchInput.parse(input);
@@ -70,6 +126,7 @@ function clusterPair(kind, domain) {
             description: `Tear down the ${label} controller for a reservation.`,
             mutation: "preview-confirm",
             http: { method: "POST", path: `/${kind}/reservations/{id}/cluster/teardown` },
+            slack: false,
             input: reservationInput,
             execute: async (ctx, input) => {
                 const parsed = reservationInput.parse(input);
@@ -87,6 +144,7 @@ function clusterPair(kind, domain) {
             description: `Show the ${label} controller for a reservation.`,
             mutation: "read",
             http: { method: "GET", path: `/${kind}/reservations/{id}/cluster` },
+            slack: false,
             input: reservationInput,
             execute: async (ctx, input) => {
                 const parsed = reservationInput.parse(input);
@@ -95,5 +153,96 @@ function clusterPair(kind, domain) {
         }),
     ];
 }
+export const clusterCliCapabilities = [
+    clusterCliCapability({
+        id: "clusters.list",
+        command: ["clusters", "list"],
+        description: "List customer-safe Kubernetes and Slurm cluster facts in one bounded page.",
+        mutation: "read",
+        outcome: "List clusters without generated names, credentials, or controller details.",
+        human: "ornn clusters list [--limit <1-500>] [--cursor <cursor>]",
+        automation: "ornn clusters list --limit 50 --format json",
+        related: ["clusters show", "clusters wait"],
+    }),
+    clusterCliCapability({
+        id: "clusters.show",
+        command: ["clusters", "show"],
+        description: "Inspect one exact cluster through safe customer facts.",
+        mutation: "read",
+        outcome: "Show lifecycle and membership without exposing credentials or controller internals.",
+        human: "ornn clusters show [<cluster-id>] [--guided]",
+        automation: "ornn clusters show <cluster-id> --format json",
+        acceptedReferences: ["canonical cluster ID", "invocation-local guided choice"],
+        related: ["clusters wait", "connect"],
+    }),
+    clusterCliCapability({
+        id: "clusters.wait",
+        command: ["clusters", "wait"],
+        description: "Wait locally for one cluster to become usable without cancelling remote work.",
+        mutation: "read",
+        outcome: "Wait for usable native access with bounded timeout and interruption results.",
+        human: "ornn clusters wait [<cluster-id>] [--guided] [--timeout <duration>]",
+        automation: "ornn clusters wait <cluster-id> --timeout <duration> --format json",
+        acceptedReferences: ["canonical cluster ID", "invocation-local guided choice"],
+        related: ["clusters show", "connect"],
+    }),
+    clusterCliCapability({
+        id: "clusters.eligible_machines",
+        command: ["clusters", "eligible-machines"],
+        description: "Read the authoritative same-reservation eligible-machine projection.",
+        mutation: "read",
+        outcome: "List only canonical machine IDs and hostnames currently eligible for this cluster.",
+        human: "ornn clusters eligible-machines [<cluster-id>] [--guided] [--limit <1-500>] [--cursor <cursor>]",
+        automation: "ornn clusters eligible-machines <cluster-id> --limit 50 --format json",
+        acceptedReferences: ["canonical cluster ID", "invocation-local guided choice"],
+        related: ["clusters add-node", "clusters remove-node"],
+    }),
+    clusterCliCapability({
+        id: "clusters.add_node",
+        command: ["clusters", "add-node"],
+        description: "Reconcile one exact cluster membership-increase goal.",
+        mutation: "preview-confirm",
+        outcome: "Add one currently eligible machine, or rediscover that exact membership goal.",
+        human: "ornn clusters add-node [<cluster-id>] [<machine-id-or-hostname>] [--guided] [--no-wait]",
+        automation: "ornn clusters add-node <cluster-id> <machine-id> --yes --idempotency-key <key> --format json",
+        acceptedReferences: [
+            "canonical cluster ID",
+            "canonical machine ID",
+            "exact unique eligible hostname",
+            "invocation-local guided choice",
+        ],
+        review: "Review the exact cluster, selected machine, and resulting membership count.",
+        related: ["clusters eligible-machines", "clusters remove-node"],
+    }),
+    clusterCliCapability({
+        id: "clusters.remove_node",
+        command: ["clusters", "remove-node"],
+        description: "Reconcile one exact cluster membership-reduction goal.",
+        mutation: "preview-confirm",
+        outcome: "Remove one exact member, or rediscover that exact membership goal.",
+        human: "ornn clusters remove-node [<cluster-id>] [<machine-id-or-hostname>] [--guided] [--no-wait]",
+        automation: "ornn clusters remove-node <cluster-id> <machine-id> --yes --idempotency-key <key> --format json",
+        acceptedReferences: [
+            "canonical cluster ID",
+            "canonical machine ID",
+            "exact unique member hostname",
+            "invocation-local guided choice",
+        ],
+        review: "Review the exact cluster, removed machine, access reduction, and resulting membership count.",
+        related: ["clusters eligible-machines", "clusters add-node"],
+    }),
+    clusterCliCapability({
+        id: "clusters.teardown",
+        command: ["clusters", "teardown"],
+        description: "Tear down one cluster without surrendering its capacity reservation.",
+        mutation: "preview-confirm",
+        outcome: "End cluster lifecycle while reporting partial member failures and continuing reservation charges.",
+        human: "ornn clusters teardown [<cluster-id>] [--guided] [--no-wait]",
+        automation: "ornn clusters teardown <cluster-id> --yes --idempotency-key <key> --format json",
+        acceptedReferences: ["canonical cluster ID", "invocation-local guided choice"],
+        review: "The cluster and native access end; the capacity reservation and its charges remain.",
+        related: ["clusters show", "reservations show"],
+    }),
+];
 export const slurmCapabilities = clusterPair("slurm", "slurm");
 export const kubernetesCapabilities = clusterPair("kubernetes", "kubernetes");