@ornncompute/cli 1.0.120 → 1.0.122

package/vendor/capabilities/cli.js

@@ -1339,6 +1339,102 @@ function geolocate(request, query3, signal) {
   });
 }
 
+// ../../packages/api-client/dist/orchestrator/registry.js
+function getContainerRegistry(request) {
+  return request("/v1/orchestrator/registry", { method: "GET" });
+}
+function listRegistryRepositories(request) {
+  return request("/v1/orchestrator/registry/repositories", { method: "GET" });
+}
+function listRegistryTags(request, repository) {
+  return request(`/v1/orchestrator/registry/tags?${new URLSearchParams({ repository })}`, {
+    method: "GET"
+  });
+}
+function getRegistryUsage(request) {
+  return request("/v1/orchestrator/registry/usage", { method: "GET" });
+}
+function listRegistryCaches(request, query3 = {}) {
+  const params = new URLSearchParams();
+  if (query3.controller_id !== void 0)
+    params.set("controller_id", query3.controller_id);
+  return request(`/v1/orchestrator/registry/caches${params.size ? `?${params}` : ""}`, {
+    method: "GET"
+  });
+}
+function deleteRegistryTag(request, repository, tag) {
+  return request(`/v1/orchestrator/registry/tags?${new URLSearchParams({ repository, tag })}`, {
+    method: "DELETE"
+  });
+}
+
+// ../../packages/api-client/dist/orchestrator/drives.js
+function queryPath3(path, query3) {
+  const params = new URLSearchParams();
+  for (const [key, value] of Object.entries(query3)) {
+    if (value !== void 0 && value !== null)
+      params.set(key, String(value));
+  }
+  return `${path}${params.size ? `?${params}` : ""}`;
+}
+function listDrives(request, query3 = {}, signal) {
+  return request(queryPath3("/v1/orchestrator/drives", query3), {
+    method: "GET",
+    ...signal ? { signal } : {}
+  });
+}
+function getDrive(request, id, query3 = {}, options = {}) {
+  return request(queryPath3(`/v1/orchestrator/drives/${encodeURIComponent(id)}`, query3), {
+    method: "GET",
+    ...options
+  });
+}
+function createDrive(request, body2, options = {}) {
+  return request("/v1/orchestrator/drives", {
+    method: "POST",
+    body: body2,
+    ...options
+  });
+}
+function updateDrive(request, id, body2, options = {}) {
+  return request(`/v1/orchestrator/drives/${encodeURIComponent(id)}`, {
+    method: "PATCH",
+    body: body2,
+    ...options
+  });
+}
+function deleteDrive(request, id, query3 = {}, options = {}) {
+  return request(queryPath3(`/v1/orchestrator/drives/${encodeURIComponent(id)}`, query3), {
+    method: "DELETE",
+    ...options
+  });
+}
+function listStorageTargets(request, signal) {
+  return request("/v1/orchestrator/storage-targets", {
+    method: "GET",
+    ...signal ? { signal } : {}
+  });
+}
+
+// ../../packages/api-client/dist/orchestrator/drive-files.js
+function drivePath(id) {
+  return `/v1/orchestrator/drives/${encodeURIComponent(id)}`;
+}
+function listDriveContents(request, id, query3 = {}, signal) {
+  const params = new URLSearchParams();
+  for (const [key, value] of Object.entries(query3)) {
+    if (value !== void 0)
+      params.set(key, String(value));
+  }
+  return request(`${drivePath(id)}/contents${params.size ? `?${params}` : ""}`, {
+    method: "GET",
+    ...signal ? { signal } : {}
+  });
+}
+function clearDriveContents(request, id) {
+  return request(`${drivePath(id)}/contents`, { method: "DELETE" });
+}
+
 // ../../packages/capabilities/dist/capabilities/staff-shared.js
 function statusOf(error) {
   if (error && typeof error === "object" && "status" in error) {
@@ -6955,9 +7051,6 @@ function formatBytes(bytes) {
   }
   return `${value >= 10 || unit === 0 ? Math.round(value) : value.toFixed(1)} ${units[unit]}`;
 }
-async function registryOf(context) {
-  return await context.fetch(REGISTRY_PATH, { method: "GET" });
-}
 function repositoryArgument(input, interaction2) {
   const raw = input.repository ?? input.$positionals[0];
   if (!raw) {
@@ -6996,7 +7089,7 @@ var registryShowCapability = defineCapability({
   },
   slack: false,
   execute: async (context) => {
-    const registry = await registryOf(context);
+    const registry = await getContainerRegistry(context.fetch);
     const interaction2 = cliInteraction(context);
     if (!interaction2)
       return registry;
@@ -7031,11 +7124,11 @@ var registryRepositoriesListCapability = defineCapability({
   },
   slack: false,
   execute: async (context) => {
-    const listed = await context.fetch(`${REGISTRY_PATH}/repositories`, { method: "GET" });
+    const listed = await listRegistryRepositories(context.fetch);
     const interaction2 = cliInteraction(context);
     if (!interaction2)
       return listed;
-    const registry = await registryOf(context);
+    const registry = await getContainerRegistry(context.fetch);
     return interaction2.result({
       data: listed,
       presentation: {
@@ -7069,9 +7162,9 @@ var registryTagsListCapability = defineCapability({
   slack: false,
   execute: async (context, input) => {
     const interaction2 = cliInteraction(context);
-    const registry = await registryOf(context);
+    const registry = await getContainerRegistry(context.fetch);
     const repository = qualifiedRepository(repositoryArgument(input, interaction2).repository, registry);
-    const listed = await context.fetch(`${REGISTRY_PATH}/tags?${new URLSearchParams({ repository })}`, { method: "GET" });
+    const listed = await listRegistryTags(context.fetch, repository);
     if (!interaction2)
       return listed;
     return interaction2.result({
@@ -7108,7 +7201,7 @@ var registryTagsDeleteCapability = defineCapability({
   slack: false,
   execute: async (context, input) => {
     const interaction2 = cliInteraction(context);
-    const registry = await registryOf(context);
+    const registry = await getContainerRegistry(context.fetch);
     const argument = repositoryArgument(input, interaction2);
     const repository = qualifiedRepository(argument.repository, registry);
     if (!argument.tag) {
@@ -7142,9 +7235,7 @@ var registryTagsDeleteCapability = defineCapability({
     } else if (!context.confirmed) {
       return preview("registry.tags.delete", target);
     }
-    await context.fetch(`${REGISTRY_PATH}/tags?${new URLSearchParams(target)}`, {
-      method: "DELETE"
-    });
+    await deleteRegistryTag(context.fetch, target.repository, target.tag);
     const deleted = { deleted: true, ...target };
     if (!interaction2)
       return deleted;
@@ -7176,7 +7267,7 @@ var registryUsageCapability = defineCapability({
   },
   slack: false,
   execute: async (context) => {
-    const usage = await context.fetch(`${REGISTRY_PATH}/usage`, { method: "GET" });
+    const usage = await getRegistryUsage(context.fetch);
     const interaction2 = cliInteraction(context);
     if (!interaction2)
       return usage;
@@ -7207,7 +7298,7 @@ var registryCopiesListCapability = defineCapability({
     }
   },
   slack: false,
-  execute: async (context) => context.fetch(`${REGISTRY_PATH}/caches`, { method: "GET" })
+  execute: (context) => listRegistryCaches(context.fetch)
 });
 var registryLoginCapability = defineCapability({
   id: "registry.login",
@@ -7484,13 +7575,6 @@ var volumeCapabilities = [
   })
 ];
 var DRIVES_ROUTE = "/v1/orchestrator/drives";
-function drivePath(driveId) {
-  return `${DRIVES_ROUTE}/${encodeURIComponent(driveId)}`;
-}
-var STORAGE_DRIVES_ROUTE = "/v1/workflows/compute/nodes/storage-drives";
-function storageDrivePath(driveId, action) {
-  return `${STORAGE_DRIVES_ROUTE}/${encodeURIComponent(driveId)}/${action}`;
-}
 var reservationInput = z30.object({ reservationId: z30.string() });
 var reservationWrite = reservationInput.extend({ confirm: z30.boolean().optional() });
 var driveInput = z30.object({ driveId: z30.string() });
@@ -7569,7 +7653,7 @@ var storageVerbCapabilities = [
     http: { method: "GET", path: "/v1/orchestrator/storage-targets" },
     input: z30.object({}),
     slack: false,
-    execute: (ctx) => ctx.fetch("/v1/orchestrator/storage-targets", { method: "GET" })
+    execute: (ctx) => listStorageTargets(ctx.fetch)
   }),
   defineCapability({
     id: "volume.list",
@@ -7584,12 +7668,10 @@ var storageVerbCapabilities = [
     }),
     mcp: { name: "storage_volumes_list" },
     slack: false,
-    execute: (ctx, input) => {
-      const query3 = new URLSearchParams({ limit: String(input.limit ?? 25) });
-      if (input.cursor)
-        query3.set("cursor", input.cursor);
-      return ctx.fetch(`${DRIVES_ROUTE}?${query3}`, { method: "GET" });
-    }
+    execute: (ctx, input) => listDrives(ctx.fetch, {
+      limit: input.limit ?? 25,
+      ...input.cursor ? { cursor: input.cursor } : {}
+    })
   }),
   defineCapability({
     id: "volume.show",
@@ -7601,7 +7683,7 @@ var storageVerbCapabilities = [
     input: driveInput,
     mcp: { name: "storage_volumes_show" },
     slack: false,
-    execute: (ctx, input) => ctx.fetch(drivePath(input.driveId), { method: "GET" })
+    execute: (ctx, input) => getDrive(ctx.fetch, input.driveId)
   }),
   defineCapability({
     id: "volume.create",
@@ -7617,9 +7699,9 @@ var storageVerbCapabilities = [
     }),
     mcp: { name: "storage_volumes_create" },
     slack: false,
-    execute: (ctx, input) => ctx.fetch(DRIVES_ROUTE, {
-      method: "POST",
-      body: { name: input.name, source_drive_id: input.sourceDriveId ?? null }
+    execute: (ctx, input) => createDrive(ctx.fetch, {
+      name: input.name,
+      source_drive_id: input.sourceDriveId ?? null
     })
   }),
   defineCapability({
@@ -7628,11 +7710,11 @@ var storageVerbCapabilities = [
     roles: ["user"],
     description: "Refresh a storage volume's (drive's) reported usage/state.",
     mutation: "preview-confirm",
-    http: { method: "POST", path: `${STORAGE_DRIVES_ROUTE}/{driveId}/refresh` },
+    http: { method: "GET", path: `${DRIVES_ROUTE}/{driveId}` },
     input: driveWrite,
     mcp: { name: "storage_volumes_refresh" },
     slack: false,
-    execute: (ctx, input) => ctx.fetch(storageDrivePath(input.driveId, "refresh"), { method: "POST" })
+    execute: (ctx, input) => getDrive(ctx.fetch, input.driveId, { refresh: true })
   }),
   defineCapability({
     id: "volume.clear",
@@ -7646,7 +7728,7 @@ var storageVerbCapabilities = [
     slack: false,
     // The route answers 204, so report which volume was cleared.
     execute: async (ctx, input) => {
-      await ctx.fetch(`${drivePath(input.driveId)}/contents`, { method: "DELETE" });
+      await clearDriveContents(ctx.fetch, input.driveId);
       return { cleared: true, drive_id: input.driveId };
     }
   }),
@@ -7661,7 +7743,7 @@ var storageVerbCapabilities = [
     mcp: { name: "storage_volumes_delete" },
     slack: false,
     execute: async (ctx, input) => {
-      await ctx.fetch(drivePath(input.driveId), { method: "DELETE" });
+      await deleteDrive(ctx.fetch, input.driveId);
       return { deleted: true, drive_id: input.driveId };
     }
   }),
@@ -7677,22 +7759,19 @@ var storageVerbCapabilities = [
     slack: false,
     execute: (ctx, input) => {
       const endpoint = input.endpointUrl ?? (input.provider === "r2" && input.accountId ? `https://${input.accountId}.r2.cloudflarestorage.com` : void 0);
-      return ctx.fetch(DRIVES_ROUTE, {
-        method: "POST",
-        body: {
-          name: input.name ?? input.bucket,
-          provider: "ornn_volume",
-          source_provider: input.provider === "gcs" ? "external_gcs" : "s3",
-          external_bucket: input.bucket,
-          external_prefix: input.prefix ?? null,
-          external_region: input.region ?? (input.provider === "r2" ? "auto" : null),
-          external_read_only: input.readOnly !== false,
-          ...input.provider === "r2" ? { external_s3_provider: "cloudflare_r2" } : {},
-          ...input.provider === "s3" ? { external_s3_provider: "aws_s3" } : {},
-          ...endpoint ? { external_endpoint_url: endpoint } : {},
-          ...input.accessKeyId ? { external_access_key_id: input.accessKeyId } : {},
-          ...input.secretAccessKey ? { external_secret_access_key: input.secretAccessKey } : {}
-        }
+      return createDrive(ctx.fetch, {
+        name: input.name ?? input.bucket,
+        provider: "ornn_volume",
+        source_provider: input.provider === "gcs" ? "external_gcs" : "s3",
+        external_bucket: input.bucket,
+        external_prefix: input.prefix ?? null,
+        external_region: input.region ?? (input.provider === "r2" ? "auto" : null),
+        external_read_only: input.readOnly !== false,
+        ...input.provider === "r2" ? { external_s3_provider: "cloudflare_r2" } : {},
+        ...input.provider === "s3" ? { external_s3_provider: "aws_s3" } : {},
+        ...endpoint ? { external_endpoint_url: endpoint } : {},
+        ...input.accessKeyId ? { external_access_key_id: input.accessKeyId } : {},
+        ...input.secretAccessKey ? { external_secret_access_key: input.secretAccessKey } : {}
       });
     }
   }),
@@ -7702,11 +7781,11 @@ var storageVerbCapabilities = [
     roles: ["user"],
     description: "Verify a connected S3-compatible bucket's source credentials/connectivity.",
     mutation: "preview-confirm",
-    http: { method: "POST", path: `${STORAGE_DRIVES_ROUTE}/{driveId}/verify-source` },
+    http: { method: "PATCH", path: `${DRIVES_ROUTE}/{driveId}` },
     input: driveWrite,
     mcp: { name: "storage_buckets_verify" },
     slack: false,
-    execute: (ctx, input) => ctx.fetch(storageDrivePath(input.driveId, "verify-source"), { method: "POST" })
+    execute: (ctx, input) => updateDrive(ctx.fetch, input.driveId, { verify_source: true })
   }),
   defineCapability({
     id: "bucket.update_credentials",
@@ -7714,13 +7793,12 @@ var storageVerbCapabilities = [
     roles: ["user"],
     description: "Update a connected S3-compatible bucket's access key credentials.",
     mutation: "preview-confirm",
-    http: { method: "PATCH", path: `${STORAGE_DRIVES_ROUTE}/{driveId}/source-credentials` },
+    http: { method: "PATCH", path: `${DRIVES_ROUTE}/{driveId}` },
     input: driveWrite.extend({ accessKeyId: z30.string(), secretAccessKey: z30.string() }),
     mcp: { name: "storage_buckets_update_credentials" },
     slack: false,
-    execute: (ctx, input) => ctx.fetch(storageDrivePath(input.driveId, "source-credentials"), {
-      method: "PATCH",
-      body: {
+    execute: (ctx, input) => updateDrive(ctx.fetch, input.driveId, {
+      source_credentials: {
         external_access_key_id: input.accessKeyId,
         external_secret_access_key: input.secretAccessKey
       }
@@ -7741,12 +7819,11 @@ var storageCompositionCapabilities = [
     }),
     mcp: false,
     slack: false,
-    execute: (ctx, input) => {
-      const query3 = new URLSearchParams({ limit: String(input.limit ?? 25), scope: "external" });
-      if (input.cursor)
-        query3.set("cursor", input.cursor);
-      return ctx.fetch(`${DRIVES_ROUTE}?${query3}`, { method: "GET" });
-    }
+    execute: (ctx, input) => listDrives(ctx.fetch, {
+      limit: input.limit ?? 25,
+      scope: "external",
+      ...input.cursor ? { cursor: input.cursor } : {}
+    })
   }),
   defineCapability({
     id: "volume.contents",
@@ -7762,14 +7839,11 @@ var storageCompositionCapabilities = [
     }),
     mcp: false,
     slack: false,
-    execute: (ctx, input) => {
-      const query3 = new URLSearchParams({ recursive: "true" });
-      if (input.path)
-        query3.set("path", input.path);
-      if (input.cursor)
-        query3.set("cursor", String(input.cursor));
-      return ctx.fetch(`${drivePath(input.driveId)}/contents?${query3}`, { method: "GET" });
-    }
+    execute: (ctx, input) => listDriveContents(ctx.fetch, input.driveId, {
+      recursive: true,
+      ...input.path ? { path: input.path } : {},
+      ...input.cursor ? { cursor: input.cursor } : {}
+    })
   })
 ];