trainfabric 0.1.39 → 0.1.41

package/dist/index.cjs

@@ -9180,6 +9180,7 @@ var PRECISION_MODES = /* @__PURE__ */ new Set(["fp16_bf16", "fp8", "fp32"]);
 var INTERCONNECT_TYPES = /* @__PURE__ */ new Set(["pcie", "nvlink"]);
 var BASE_MODELS = new Set(baseModels);
 var TRAINING_MODES = new Set(trainingModes);
+var MAX_MIN_GPU_MEMORY_GB = 192;
 function normalizePositiveInteger(value, label, fallback, max) {
   const parsed = Number(value ?? String(fallback));
   if (!Number.isInteger(parsed) || parsed <= 0) {
@@ -9266,6 +9267,9 @@ function buildComputeSpec(options) {
     compute.acceleratorClass = accelerator;
   }
   if (minMemory !== void 0) {
+    if (minMemory > MAX_MIN_GPU_MEMORY_GB) {
+      throw new Error(`Minimum GPU memory must be <= ${MAX_MIN_GPU_MEMORY_GB} GB.`);
+    }
     compute.minGpuMemoryGb = minMemory;
   }
   if (precision) {
@@ -9279,7 +9283,7 @@ function buildComputeSpec(options) {
 
 // src/index.ts
 var DEFAULT_TRAINFABRIC_API_URL2 = "https://api.trainfabric.com";
-var CLI_VERSION = "0.1.39";
+var CLI_VERSION = "0.1.41";
 var CONFIG_DIR = import_node_path3.default.join(import_node_os2.default.homedir(), ".trainfabric");
 var CONFIG_PATH = import_node_path3.default.join(CONFIG_DIR, "config.json");
 var FALLBACK_SECRET_PATH = import_node_path3.default.join(CONFIG_DIR, "session.enc");
@@ -9618,6 +9622,20 @@ function printUsageSummary(usage) {
     console.log(`  ${item.label}: ${formatUsd(item.amountUsd)} - ${item.description}`);
   }
 }
+function summarizeRunDetail(detail) {
+  return {
+    runId: detail.run.id,
+    status: detail.run.status,
+    dispatchStatus: detail.run.dispatchStatus,
+    durationSeconds: detail.run.durationSeconds,
+    estimatedCostUsd: detail.run.estimatedCostUsd,
+    customerChargeUsd: detail.usageSummary?.customerChargeUsd,
+    reservedGpuSeconds: detail.usageSummary?.reservedGpuSeconds,
+    artifactCount: detail.artifacts?.length ?? 0,
+    metricCount: detail.metrics?.length ?? 0,
+    modelId: detail.run.modelId
+  };
+}
 function updateConfig(mutate) {
   const config = loadConfig();
   mutate(config);
@@ -9776,7 +9794,14 @@ async function watchRun(runId, json = false, timeoutMs, pollMs) {
     });
   }
   const detail = await handle.wait({ timeoutMs, pollMs });
-  printJson(detail);
+  const summary = summarizeRunDetail(detail);
+  if (json) {
+    printJson({ type: "terminal", payload: summary });
+  } else {
+    console.log(
+      `[terminal] status=${summary.status} charge=${formatUsd(summary.customerChargeUsd)} artifacts=${summary.artifactCount}`
+    );
+  }
 }
 function parseDurationMs(value) {
   if (!value) {
@@ -9963,7 +9988,14 @@ program2.command("datasets:get").argument("<datasetId>").description("Fetch data
 });
 program2.command("datasets:delete").argument("<datasetId>").description("Delete a dataset").action(async (datasetId) => {
   const normalizedDatasetId = normalizeId(datasetId, "dataset", "Dataset ID");
-  await createClient(loadConfig()).datasets.delete(normalizedDatasetId);
+  const client = createClient(loadConfig());
+  try {
+    await client.datasets.get(normalizedDatasetId);
+  } catch {
+    printJson({ deleted: false, alreadyDeleted: true, id: normalizedDatasetId });
+    return;
+  }
+  await client.datasets.delete(normalizedDatasetId);
   printJson({ deleted: true, id: normalizedDatasetId });
 });
 program2.command("datasets:upload").argument("<file>").option("--project <projectId>").option("--name <name>").description("Upload a dataset through upload sessions").action(async (file, options) => {
@@ -10048,14 +10080,13 @@ program2.command("runs:status").argument("<runId>").description("Fetch full run
 program2.command("runs:logs").argument("<runId>").description("Fetch run logs").action(async (runId) => {
   printJson(await createClient(loadConfig()).runs.logs(normalizeId(runId, "run", "Run ID")));
 });
-program2.command("runs:wait").argument("<runId>").option("--timeout <duration>", "maximum wait duration, for example 30s, 10m, or 1h").option("--poll <duration>", "poll interval, for example 1s or 500ms").description("Wait for a run to reach a terminal state").action(async (runId, options) => {
+program2.command("runs:wait").argument("<runId>").option("--timeout <duration>", "maximum wait duration, for example 30s, 10m, or 1h").option("--poll <duration>", "poll interval, for example 1s or 500ms").option("--summary", "print a compact terminal run summary").description("Wait for a run to reach a terminal state").action(async (runId, options) => {
   const normalizedRunId = normalizeId(runId, "run", "Run ID");
-  printJson(
-    await createClient(loadConfig()).runs.wait(normalizedRunId, {
-      timeoutMs: parseDurationMs(options.timeout),
-      pollMs: parsePositiveMs(options.poll, 1e3)
-    })
-  );
+  const detail = await createClient(loadConfig()).runs.wait(normalizedRunId, {
+    timeoutMs: parseDurationMs(options.timeout),
+    pollMs: parsePositiveMs(options.poll, 1e3)
+  });
+  printJson(options.summary ? summarizeRunDetail(detail) : detail);
 });
 program2.command("runs:watch").argument("<runId>").option("--json", "emit event output as JSON").option("--timeout <duration>", "maximum watch duration, for example 30s, 10m, or 1h").option("--poll <duration>", "poll interval, for example 1s or 500ms").description("Stream a run until it completes").action(async (runId, options) => {
   await watchRun(normalizeId(runId, "run", "Run ID"), Boolean(options.json), parseDurationMs(options.timeout), parsePositiveMs(options.poll, 1e3));

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "trainfabric",
-  "version": "0.1.39",
+  "version": "0.1.41",
   "description": "Trainfabric CLI for launching GPU training jobs on the hosted Trainfabric backend.",
   "type": "module",
   "main": "dist/index.cjs",