trainfabric 0.1.26 → 0.1.28

package/dist/index.cjs

@@ -8905,6 +8905,14 @@ function validateGitUrl(value) {
   }
   return value;
 }
+function validateGitBranch(value) {
+  const branch = String(value ?? "").trim();
+  const invalidCharacterPattern = /[\x00-\x20~^:?*[\]\\]/;
+  if (!branch || branch === "@" || branch.startsWith("-") || branch.startsWith("/") || branch.endsWith("/") || branch.endsWith(".") || branch.includes("//") || branch.includes("..") || branch.includes("@{") || invalidCharacterPattern.test(branch) || branch.split("/").some((part) => !part || part.startsWith(".") || part.endsWith(".lock"))) {
+    throw new Error("Git branch is invalid. Use a valid git branch name, for example main or feature/train-run.");
+  }
+  return branch;
+}
 function validateRepoPath(repoPath) {
   const absolute = import_node_path2.default.resolve(repoPath);
   const parsed = import_node_path2.default.parse(absolute);
@@ -8921,6 +8929,9 @@ function buildSourceOptions(options) {
   if (options.repo && options.git) {
     throw new Error("Use either --repo or --git, not both.");
   }
+  if (options.branch && !options.git) {
+    throw new Error("--branch only applies when --git is used.");
+  }
   if (options.repo) {
     const repoPath = validateRepoPath(options.repo);
     const files = collectRuntimeFiles(repoPath);
@@ -8937,7 +8948,7 @@ function buildSourceOptions(options) {
       source: {
         kind: "git",
         repoUrl: validateGitUrl(options.git),
-        branch: options.branch
+        branch: options.branch === void 0 ? void 0 : validateGitBranch(options.branch)
       }
     };
   }
@@ -8983,6 +8994,9 @@ function normalizeHumanName(value, label) {
   if (!/[A-Za-z0-9]/.test(name)) {
     throw new Error(`${label} is required.`);
   }
+  if (!/^[A-Za-z0-9][A-Za-z0-9 ._-]{0,79}$/.test(name) || name.includes("..")) {
+    throw new Error(`${label} contains invalid characters. Use letters, numbers, spaces, dots, underscores, or hyphens.`);
+  }
   return name;
 }
 function normalizeId(value, kind, label) {
@@ -9024,6 +9038,8 @@ var BASE_MODEL_ALIASES = {
 var ACCELERATOR_CLASSES = /* @__PURE__ */ new Set(["a10g", "l4", "a40", "l40s", "a100", "h100", "h200", "b200"]);
 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);
 function normalizePositiveInteger(value, label, fallback) {
   const parsed = Number(value ?? String(fallback));
   if (!Number.isInteger(parsed) || parsed <= 0) {
@@ -9052,7 +9068,35 @@ function normalizeOptionalChoice(value, allowed, label, examples) {
   return normalized;
 }
 function normalizeBaseModel(model) {
-  return BASE_MODEL_ALIASES[model] ?? model;
+  const normalized = BASE_MODEL_ALIASES[String(model ?? "").trim().toLowerCase()];
+  if (normalized) {
+    return normalized;
+  }
+  const rawModel = String(model ?? "").trim();
+  if (!BASE_MODELS.has(rawModel)) {
+    throw new Error("Model is invalid. Use llama-3-8b, mistral-7b, or qwen-2.5-7b.");
+  }
+  return rawModel;
+}
+function normalizeTrainingMode(mode) {
+  if (mode === void 0) {
+    return void 0;
+  }
+  const normalized = String(mode).trim().toLowerCase();
+  if (!TRAINING_MODES.has(normalized)) {
+    throw new Error("Mode is invalid. Use efficient, balanced, or power.");
+  }
+  return normalized;
+}
+function normalizeEpochs(epochs) {
+  return normalizePositiveInteger(epochs, "Epochs", 3);
+}
+function normalizeLearningRate(lr) {
+  const parsed = Number(lr ?? "0.0002");
+  if (!Number.isFinite(parsed) || parsed < 1e-5 || parsed > 0.01) {
+    throw new Error("Learning rate must be a number between 0.00001 and 0.01.");
+  }
+  return parsed;
 }
 function buildComputeSpec(options) {
   const gpuCount = normalizePositiveInteger(options.gpus, "GPU count", 1);
@@ -9089,7 +9133,7 @@ function buildComputeSpec(options) {
 
 // src/index.ts
 var DEFAULT_TRAINFABRIC_API_URL2 = "https://api.trainfabric.com";
-var CLI_VERSION = "0.1.26";
+var CLI_VERSION = "0.1.28";
 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");
@@ -9441,15 +9485,15 @@ function buildRunInput(options, config = loadConfig()) {
     task: "sft",
     method: "lora",
     baseModel: normalizeBaseModel(options.model),
-    datasetId: options.dataset,
-    evalDatasetId: options.eval,
+    datasetId: normalizeId(options.dataset, "dataset", "Dataset ID"),
+    evalDatasetId: normalizeOptionalId(options.eval, "dataset", "Eval dataset ID"),
     pricingQuoteId: options.quote,
     ...sourceOptions,
-    mode: options.mode,
+    mode: normalizeTrainingMode(options.mode),
     compute: buildComputeSpec(options),
     hyperparameters: {
-      epochs: Number(options.epochs ?? "3"),
-      lr: Number(options.lr ?? "0.0002"),
+      epochs: normalizeEpochs(options.epochs),
+      lr: normalizeLearningRate(options.lr),
       batchSize: "auto"
     }
   };
@@ -9868,7 +9912,12 @@ program2.command("deployments:create").requiredOption("--model <modelId>").optio
     await client.projects.get(projectId);
   }
   const modelId = normalizeId(options.model, "model", "Model ID");
-  printJson(await client.deployments.create({ modelId }));
+  try {
+    printJson(await client.deployments.create({ modelId }));
+  } catch (error) {
+    printJson({ error: error instanceof Error ? error.message : String(error) });
+    process.exitCode = 1;
+  }
 });
 program2.command("deployments:get").argument("<deploymentId>").description("Fetch a deployment").action(async (deploymentId) => {
   printJson(await createClient(loadConfig()).deployments.get(normalizeId(deploymentId, "deployment", "Deployment ID")));

package/package.json

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