trainfabric 0.1.21 → 0.1.23

package/dist/index.cjs

@@ -8817,13 +8817,22 @@ var import_node_fs = __toESM(require("node:fs"), 1);
 var import_node_os = __toESM(require("node:os"), 1);
 var import_node_path2 = __toESM(require("node:path"), 1);
 function collectRuntimeFiles(repoPath) {
-  const manifestFiles = /* @__PURE__ */ new Set(["training.yaml", "train.runtime.yaml", "pyproject.toml", "requirements.txt", "environment.yml", "Dockerfile"]);
+  const manifestFiles = /* @__PURE__ */ new Set([
+    "training.yaml",
+    "train.runtime.yaml",
+    "pyproject.toml",
+    "requirements.txt",
+    "environment.yml",
+    "Dockerfile"
+  ]);
+  const configExtensions = /* @__PURE__ */ new Set([".yaml", ".yml", ".json", ".toml"]);
   const ignoredDirectories = /* @__PURE__ */ new Set([".git", ".hg", ".svn", "node_modules", ".venv", "venv", "__pycache__", ".mypy_cache", ".pytest_cache", "dist", "build"]);
   const found = [];
   const maxFiles = 256;
   const maxFileBytes = 256 * 1024;
   function shouldIncludeFile(fileName) {
-    return manifestFiles.has(fileName) || fileName.endsWith(".py");
+    const extension = import_node_path2.default.extname(fileName).toLowerCase();
+    return manifestFiles.has(fileName) || configExtensions.has(extension) || fileName.endsWith(".py");
   }
   function walk(currentPath) {
     if (found.length >= maxFiles) {
@@ -8945,7 +8954,7 @@ function buildComputeSpec(options) {
 
 // src/index.ts
 var DEFAULT_TRAINFABRIC_API_URL2 = "https://api.trainfabric.com";
-var CLI_VERSION = "0.1.21";
+var CLI_VERSION = "0.1.23";
 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");
@@ -9409,6 +9418,21 @@ function createClient(config) {
     projectId: config.projectId
   });
 }
+function normalizeHttpBaseUrl(value) {
+  let url;
+  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.`);
+  }
+  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.`);
+  }
+  url.pathname = url.pathname.replace(/\/+$/, "");
+  url.search = "";
+  url.hash = "";
+  return url.toString().replace(/\/+$/, "");
+}
 async function login(config) {
   const apiKey = await promptApiKey();
   if (!apiKey) {
@@ -9454,20 +9478,32 @@ program2.command("config:show").description("Show the active CLI configuration")
   printJson(visibleConfig(loadConfig()));
 });
 program2.command("config:set-base-url").argument("<baseUrl>").description("Set the backend base URL").action((baseUrl) => {
+  const normalizedBaseUrl = normalizeHttpBaseUrl(String(baseUrl));
   const config = updateConfig((current) => {
-    current.baseUrl = String(baseUrl);
+    current.baseUrl = normalizedBaseUrl;
   });
   printJson(visibleConfig(config));
 });
-program2.command("config:set-org").argument("<orgId>").description("Set the default organization ID").action((orgId) => {
-  const config = updateConfig((current) => {
-    current.orgId = String(orgId);
+program2.command("config:set-org").argument("<orgId>").description("Set the default organization ID").action(async (orgId) => {
+  const currentConfig = loadConfig();
+  const organization = await createClient(currentConfig).organizations.get(String(orgId));
+  const updatedConfig = updateConfig((current) => {
+    current.orgId = organization.id;
+    if (current.projectId) {
+      current.projectId = void 0;
+    }
   });
-  printJson(visibleConfig(config));
+  printJson(visibleConfig(updatedConfig));
 });
-program2.command("config:set-project").argument("<projectId>").description("Set the default project ID").action((projectId) => {
+program2.command("config:set-project").argument("<projectId>").description("Set the default project ID").action(async (projectId) => {
+  const currentConfig = loadConfig();
+  const project = await createClient(currentConfig).projects.get(String(projectId));
+  if (currentConfig.orgId && project.organizationId !== currentConfig.orgId) {
+    throw new Error(`Project ${project.id} does not belong to configured organization ${currentConfig.orgId}.`);
+  }
   const config = updateConfig((current) => {
-    current.projectId = String(projectId);
+    current.projectId = project.id;
+    current.orgId ??= project.organizationId;
   });
   printJson(visibleConfig(config));
 });

package/package.json

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