trainfabric 0.1.9 → 0.1.11

package/dist/index.cjs

@@ -8204,6 +8204,9 @@ var DatasetsClient = class extends ResourceClient {
 
 // ../sdk/dist/client/deployments.js
 var DeploymentsClient = class extends ResourceClient {
+  list() {
+    return this.requestGet("/v1/deployments");
+  }
   create(input) {
     return this.requestPost("/v1/deployments", input);
   }
@@ -8789,16 +8792,36 @@ var import_node_os = __toESM(require("node:os"), 1);
 var import_node_path2 = __toESM(require("node:path"), 1);
 function collectRuntimeFiles(repoPath) {
   const candidates = ["training.yaml", "train.runtime.yaml", "pyproject.toml", "requirements.txt", "environment.yml", "Dockerfile", "train.py"];
-  return candidates.map((fileName) => {
-    const filePath = import_node_path2.default.join(repoPath, fileName);
-    if (!import_node_fs.default.existsSync(filePath)) {
-      return void 0;
+  const ignoredDirectories = /* @__PURE__ */ new Set([".git", ".hg", ".svn", "node_modules", ".venv", "venv", "__pycache__", ".mypy_cache", ".pytest_cache", "dist", "build"]);
+  const found = [];
+  function walk(currentPath) {
+    if (found.length >= 64) {
+      return;
     }
-    return {
-      path: fileName,
-      content: import_node_fs.default.readFileSync(filePath, "utf8")
-    };
-  }).filter(Boolean);
+    for (const entry of import_node_fs.default.readdirSync(currentPath, { withFileTypes: true })) {
+      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()) {
+        if (!ignoredDirectories.has(entry.name)) {
+          walk(absolute);
+        }
+        continue;
+      }
+      if (!entry.isFile() || !candidates.includes(entry.name)) {
+        continue;
+      }
+      const stat = import_node_fs.default.statSync(absolute);
+      if (stat.size > 256 * 1024) {
+        continue;
+      }
+      found.push({
+        path: relative,
+        content: import_node_fs.default.readFileSync(absolute, "utf8")
+      });
+    }
+  }
+  walk(repoPath);
+  return found.sort((left, right) => left.path.localeCompare(right.path));
 }
 function buildSourceOptions(options) {
   if (options.repo) {
@@ -8891,7 +8914,7 @@ function buildComputeSpec(options) {
 
 // src/index.ts
 var DEFAULT_TRAINFABRIC_API_URL2 = "https://api.trainfabric.com";
-var CLI_VERSION = "0.1.9";
+var CLI_VERSION = "0.1.11";
 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");
@@ -9061,7 +9084,7 @@ function ensureConfigDir() {
   import_node_fs2.default.chmodSync(CONFIG_DIR, 448);
 }
 function canUseMacKeychain() {
-  return process.platform === "darwin" && process.env.TRAINFABRIC_DISABLE_KEYCHAIN !== "1" && process.env.CI !== "1" && Boolean(process.stdin.isTTY && process.stdout.isTTY) && (0, import_node_child_process.spawnSync)("security", ["-h"]).status === 0;
+  return process.platform === "darwin" && process.env.TRAINFABRIC_DISABLE_KEYCHAIN !== "1" && process.env.CI !== "1" && (0, import_node_child_process.spawnSync)("security", ["-h"]).status === 0;
 }
 function encodeSecretPayload(payload) {
   return JSON.stringify(payload);
@@ -9527,6 +9550,9 @@ program2.command("models:list").description("List exported models").action(async
 program2.command("models:export").argument("<modelId>").description("Fetch the export package path for a model").action(async (modelId) => {
   printJson(await createClient(loadConfig()).models.export(String(modelId)));
 });
+program2.command("deployments:list").description("List deployments").action(async () => {
+  printJson(await createClient(loadConfig()).deployments.list());
+});
 program2.command("deployments:create").requiredOption("--model <modelId>").description("Create a deployment for a model").action(async (options) => {
   printJson(await createClient(loadConfig()).deployments.create({ modelId: options.model }));
 });

package/package.json

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