trainfabric 0.1.30 → 0.1.32

package/README.md

@@ -30,7 +30,7 @@ trainfabric projects:create --name <name>
 trainfabric projects:delete <projectId>
 trainfabric datasets:upload ./train.jsonl --project <projectId>
 trainfabric runs:quote --summary --project <projectId> --dataset <datasetId> --model llama-3-8b
-trainfabric runs:create --yes --quote <quoteId> --project <projectId> --dataset <datasetId> --model llama-3-8b
+trainfabric runs:create --yes --quote <quoteId>
 trainfabric runs:watch <runId> --timeout 30m
 trainfabric runs:cost-breakdown <runId> --summary
 ```

package/dist/index.cjs

@@ -8478,15 +8478,6 @@ var RunHandle = class {
       this.snapshot = detail.run;
       if (["completed", "failed", "canceled", "terminated"].includes(detail.run.status)) {
         this.close();
-        if (detail.run.status === "failed") {
-          throw new Error(detail.run.failureReason ?? "Run failed.");
-        }
-        if (detail.run.status === "canceled") {
-          throw new Error("Run was canceled.");
-        }
-        if (detail.run.status === "terminated") {
-          throw new Error("Run was terminated.");
-        }
         return detail;
       }
       if (timeoutMs !== void 0 && Date.now() - startedAt > timeoutMs) {
@@ -8677,7 +8668,7 @@ var RunsClient = class extends ResourceClient {
         mode: selectedModeQuote.mode
       };
     }
-    const parsed = runCreateSchema.parse(payload);
+    const parsed = input.pricingQuoteId ? payload : runCreateSchema.parse(payload);
     const run = await this.requestPost("/v1/runs", parsed);
     return new RunHandle(this.parent, run);
   }
@@ -8737,7 +8728,12 @@ var RuntimeClient = class extends ResourceClient {
       projectId: input.projectId ?? this.parent.projectId,
       source: input.source
     });
-    return this.requestPost("/v1/runtime/detect", payload);
+    return this.requestPost("/v1/runtime/detect", payload).catch((error) => {
+      if (input.source.kind === "git" && error instanceof Error && /Request failed with status 404/.test(error.message)) {
+        throw new Error("Git repository could not be resolved. Check the repository URL, branch, and access permissions.");
+      }
+      throw error;
+    });
   }
   build(input) {
     const payload = runtimeBuildCreateSchema.parse({
@@ -8899,6 +8895,7 @@ var idPatterns = {
   org: /^org_[A-Za-z0-9_-]+$/,
   pool: /^pool_[A-Za-z0-9_-]+$/,
   project: /^proj_[A-Za-z0-9_-]+$/,
+  quote: /^quote_[A-Za-z0-9_-]+$/,
   run: /^run_[A-Za-z0-9_-]+$/,
   serviceAccount: /^svc_[A-Za-z0-9_-]+$/,
   supplier: /^[A-Za-z][A-Za-z0-9_-]*$/
@@ -8926,7 +8923,17 @@ function collectRuntimeFiles(repoPath) {
     if (found.length >= maxFiles) {
       return;
     }
-    for (const entry of import_node_fs.default.readdirSync(currentPath, { withFileTypes: true })) {
+    let entries;
+    try {
+      entries = import_node_fs.default.readdirSync(currentPath, { withFileTypes: true });
+    } catch (error) {
+      if (error.code === "EACCES" || error.code === "EPERM") {
+        const relativePath = import_node_path2.default.relative(repoPath, currentPath) || ".";
+        throw new Error(`Permission denied while scanning repository path: ${relativePath}`);
+      }
+      throw new Error(`Failed to scan repository path: ${currentPath}`);
+    }
+    for (const entry of entries) {
       const absolute = import_node_path2.default.join(currentPath, entry.name);
       const relative = import_node_path2.default.relative(repoPath, absolute).split(import_node_path2.default.sep).join("/");
       if (entry.isDirectory()) {
@@ -8938,13 +8945,30 @@ function collectRuntimeFiles(repoPath) {
       if (!entry.isFile() || !shouldIncludeFile(entry.name)) {
         continue;
       }
-      const stat = import_node_fs.default.statSync(absolute);
+      let stat;
+      try {
+        stat = import_node_fs.default.statSync(absolute);
+      } catch (error) {
+        if (error.code === "EACCES" || error.code === "EPERM") {
+          throw new Error(`Permission denied while scanning repository file: ${relative}`);
+        }
+        throw new Error(`Failed to inspect repository file: ${relative}`);
+      }
       if (stat.size > maxFileBytes) {
         continue;
       }
+      let content;
+      try {
+        content = import_node_fs.default.readFileSync(absolute, "utf8");
+      } catch (error) {
+        if (error.code === "EACCES" || error.code === "EPERM") {
+          throw new Error(`Permission denied while reading repository file: ${relative}`);
+        }
+        throw new Error(`Failed to read repository file: ${relative}`);
+      }
       found.push({
         path: relative,
-        content: import_node_fs.default.readFileSync(absolute, "utf8")
+        content
       });
     }
   }
@@ -8988,7 +9012,22 @@ function validateGitBranch(value) {
 function validateRepoPath(repoPath) {
   const absolute = import_node_path2.default.resolve(repoPath);
   const parsed = import_node_path2.default.parse(absolute);
-  if (absolute === parsed.root || absolute === import_node_os.default.homedir()) {
+  const unsafeRoots = /* @__PURE__ */ new Set([
+    parsed.root,
+    import_node_os.default.homedir(),
+    import_node_path2.default.resolve(import_node_os.default.tmpdir()),
+    import_node_path2.default.resolve("/tmp"),
+    import_node_path2.default.resolve("/private/tmp"),
+    import_node_path2.default.resolve("/var"),
+    import_node_path2.default.resolve("/Users"),
+    import_node_path2.default.resolve("/System"),
+    import_node_path2.default.resolve("/Library"),
+    import_node_path2.default.resolve("/Applications"),
+    import_node_path2.default.resolve("/etc"),
+    import_node_path2.default.resolve("/bin"),
+    import_node_path2.default.resolve("/usr")
+  ]);
+  if (unsafeRoots.has(absolute)) {
     throw new Error(`Refusing to package unsafe repository path: ${repoPath}`);
   }
   let stat;
@@ -9210,7 +9249,7 @@ function buildComputeSpec(options) {
 
 // src/index.ts
 var DEFAULT_TRAINFABRIC_API_URL2 = "https://api.trainfabric.com";
-var CLI_VERSION = "0.1.30";
+var CLI_VERSION = "0.1.32";
 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");
@@ -9557,23 +9596,43 @@ function requireProjectId(options, config) {
 }
 function buildRunInput(options, config = loadConfig()) {
   const sourceOptions = buildSourceOptions(options);
-  return {
-    projectId: requireProjectId(options, config),
+  const quoteId = options.quote === void 0 ? void 0 : normalizeId(options.quote, "quote", "Pricing quote ID");
+  if (!quoteId && !options.dataset) {
+    throw new Error("Dataset is required unless --quote is provided.");
+  }
+  if (!quoteId && !options.model) {
+    throw new Error("Model is required unless --quote is provided.");
+  }
+  const projectId = normalizeOptionalId(options.project ?? config.projectId, "project", "Project ID");
+  if (!quoteId && !projectId) {
+    throw new Error("Project is required. Pass --project <projectId> or run `trainfabric config:set-project <projectId>`.");
+  }
+  const input = {
     task: "sft",
     method: "lora",
-    baseModel: normalizeBaseModel(options.model),
-    datasetId: normalizeId(options.dataset, "dataset", "Dataset ID"),
+    pricingQuoteId: quoteId,
     evalDatasetId: normalizeOptionalId(options.eval, "dataset", "Eval dataset ID"),
-    pricingQuoteId: options.quote,
     ...sourceOptions,
-    mode: normalizeTrainingMode(options.mode),
-    compute: buildComputeSpec(options),
-    hyperparameters: {
+    mode: normalizeTrainingMode(options.mode)
+  };
+  if (projectId) {
+    input.projectId = projectId;
+  }
+  if (options.model) {
+    input.baseModel = normalizeBaseModel(options.model);
+  }
+  if (options.dataset) {
+    input.datasetId = normalizeId(options.dataset, "dataset", "Dataset ID");
+  }
+  if (!quoteId) {
+    input.compute = buildComputeSpec(options);
+    input.hyperparameters = {
       epochs: normalizeEpochs(options.epochs),
       lr: normalizeLearningRate(options.lr),
       batchSize: "auto"
-    }
-  };
+    };
+  }
+  return input;
 }
 function assertQuoteOptions(bundle, options) {
   if (bundle.quotes.length > 0) {
@@ -9714,14 +9773,15 @@ function normalizeHttpBaseUrl(value) {
   try {
     url = new URL(value);
   } catch {
-    throw new Error(`Invalid base URL "${value}". Use an absolute http(s) URL such as https://api.trainfabric.com.`);
+    throw new Error(`Invalid base URL "${value}". Use an absolute HTTPS origin such as https://api.trainfabric.com.`);
+  }
+  const isLocalDevHost = url.hostname === "localhost" || url.hostname === "127.0.0.1" || url.hostname === "::1";
+  if (url.protocol !== "https:" && !(isLocalDevHost && url.protocol === "http:")) {
+    throw new Error(`Invalid base URL "${value}". Use HTTPS, or http://localhost for local development.`);
   }
-  if (url.protocol !== "https:" && url.protocol !== "http:") {
-    throw new Error(`Invalid base URL "${value}". Use an absolute http(s) URL such as https://api.trainfabric.com.`);
+  if (url.pathname !== "/" || url.search || url.hash) {
+    throw new Error(`Invalid base URL "${value}". Use the API origin without a path, query, or hash.`);
   }
-  url.pathname = url.pathname.replace(/\/+$/, "");
-  url.search = "";
-  url.hash = "";
   return url.toString().replace(/\/+$/, "");
 }
 async function login(config) {
@@ -9872,7 +9932,7 @@ program2.command("datasets:upload").argument("<file>").option("--project <projec
   });
   printJson(dataset);
 });
-program2.command("runs:create").option("--project <projectId>").requiredOption("--dataset <datasetId>").requiredOption("--model <baseModel>").option("--eval <evalDatasetId>").option("--epochs <epochs>", "number of epochs", "3").option("--lr <lr>", "learning rate", "0.0002").option("--gpus <gpuCount>", "gpu count", "1").option("--nodes <nodeCount>", "node count", "1").option("--accelerator <acceleratorClass>", "optional hard accelerator constraint (for example: a10g, a100, h100)").option("--min-memory <gigabytes>", "minimum GPU memory in GB").option("--precision <precision>", "fp16_bf16, fp8, or fp32").option("--interconnect <interconnect>", "pcie or nvlink").option("--mode <mode>", "efficient, balanced, or power", "balanced").option("--quote <pricingQuoteId>", "launch the exact accepted quote id from runs:quote").option("--repo <path>", "local repo path for runtime autodetect").option("--git <url>", "git repo URL for runtime metadata").option("--branch <branch>", "git branch for runtime metadata").option("--yes", "confirm that you reviewed pricing with runs:quote --summary and accept fluctuating realized usage").description("Create a training run").action(async (options) => {
+program2.command("runs:create").option("--project <projectId>").option("--dataset <datasetId>").option("--model <baseModel>").option("--eval <evalDatasetId>").option("--epochs <epochs>", "number of epochs", "3").option("--lr <lr>", "learning rate", "0.0002").option("--gpus <gpuCount>", "gpu count", "1").option("--nodes <nodeCount>", "node count", "1").option("--accelerator <acceleratorClass>", "optional hard accelerator constraint (for example: a10g, a100, h100)").option("--min-memory <gigabytes>", "minimum GPU memory in GB").option("--precision <precision>", "fp16_bf16, fp8, or fp32").option("--interconnect <interconnect>", "pcie or nvlink").option("--mode <mode>", "efficient, balanced, or power").option("--quote <pricingQuoteId>", "launch the exact accepted quote id from runs:quote").option("--repo <path>", "local repo path for runtime autodetect").option("--git <url>", "git repo URL for runtime metadata").option("--branch <branch>", "git branch for runtime metadata").option("--yes", "confirm that you reviewed pricing with runs:quote --summary and accept fluctuating realized usage").description("Create a training run").action(async (options) => {
   const config = loadConfig();
   const runInput = buildRunInput(options, config);
   if (!options.yes) {

package/package.json

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