langtrain 0.1.9 → 0.1.10

package/dist/cli.js

@@ -12626,9 +12626,11 @@ function saveConfig(config) {
   }
   import_fs.default.writeFileSync(CONFIG_FILE, JSON.stringify(config, null, 2));
 }
+var packageJson = require(import_path.default.join(__dirname, "../package.json"));
 async function main() {
   const program = new import_commander.Command();
-  program.name("langtrain").description("Langtrain CLI for AI Model Fine-tuning and Generation").version("0.1.9");
+  const version = packageJson.version;
+  program.name("langtrain").description(packageJson.description || "Langtrain CLI for AI Model Fine-tuning and Generation").version(version);
   program.action(async () => {
     console.clear();
     const banner = `
@@ -12640,23 +12642,24 @@ async function main() {
     \u255A\u2550\u2550\u2550\u2550\u2550\u2550\u255D\u255A\u2550\u255D  \u255A\u2550\u255D\u255A\u2550\u255D  \u255A\u2550\u2550\u2550\u255D \u255A\u2550\u2550\u2550\u2550\u2550\u255D   \u255A\u2550\u255D   \u255A\u2550\u255D  \u255A\u2550\u255D\u255A\u2550\u255D  \u255A\u2550\u255D\u255A\u2550\u255D\u255A\u2550\u255D  \u255A\u2550\u2550\u2550\u255D
     `;
     console.log((0, import_gradient_string.default)(["#00DC82", "#36E4DA", "#0047E1"])(banner));
-    (0, import_prompts.intro)(`${(0, import_colors.bgCyan)((0, import_colors.black)(" Langtrain SDK v0.1.9 "))}`);
+    (0, import_prompts.intro)(`${(0, import_colors.bgCyan)((0, import_colors.black)(` Langtrain SDK v${version} `))}`);
     const config = getConfig();
     if (!config.apiKey) {
       (0, import_prompts.intro)((0, import_colors.yellow)("Authentication required"));
-      const apiKey = await (0, import_prompts.password)({
-        message: "Enter your Langtrain API Key:",
-        validate(value) {
-          if (!value || value.length === 0) return "API Key is required";
-        }
-      });
-      if ((0, import_prompts.isCancel)(apiKey)) {
-        (0, import_prompts.cancel)("Operation cancelled");
+      await handleLogin();
+    }
+    const handlers = {
+      "login": handleLogin,
+      "tune-finetune": (c) => handleTuneFinetune(c.tune),
+      "tune-generate": (c) => handleTuneGenerate(c.tune),
+      "vision-finetune": (c) => handleVisionFinetune(c.vision),
+      "vision-generate": (c) => handleVisionGenerate(c.vision),
+      "agent-list": (c) => handleAgentList(c.agent),
+      "exit": async () => {
+        (0, import_prompts.outro)("Goodbye!");
         process.exit(0);
       }
-      saveConfig({ ...config, apiKey });
-      (0, import_prompts.intro)((0, import_colors.green)("Successfully logged in!"));
-    }
+    };
     while (true) {
       const operation = await (0, import_prompts.select)({
         message: "Select an operation:",
@@ -12674,33 +12677,26 @@ async function main() {
           { value: "exit", label: "  \u21B3 Exit" }
         ]
       });
-      if ((0, import_prompts.isCancel)(operation) || operation === "exit") {
+      if ((0, import_prompts.isCancel)(operation)) {
         (0, import_prompts.outro)("Goodbye!");
         process.exit(0);
       }
-      if (typeof operation === "string" && operation.startsWith("group-")) {
-        continue;
-      }
-      try {
-        const currentConfig = getConfig();
-        const currentVision = new Langvision({ apiKey: currentConfig.apiKey });
-        const currentTune = new Langtune({ apiKey: currentConfig.apiKey });
-        const currentAgent = new AgentClient({ apiKey: currentConfig.apiKey, baseUrl: currentConfig.baseUrl });
-        if (operation === "login") {
-          await handleLogin();
-        } else if (operation === "tune-finetune") {
-          await handleTuneFinetune(currentTune);
-        } else if (operation === "tune-generate") {
-          await handleTuneGenerate(currentTune);
-        } else if (operation === "vision-finetune") {
-          await handleVisionFinetune(currentVision);
-        } else if (operation === "vision-generate") {
-          await handleVisionGenerate(currentVision);
-        } else if (operation === "agent-list") {
-          await handleAgentList(currentAgent);
-        }
-      } catch (error) {
-        (0, import_prompts.outro)((0, import_colors.red)(`Error: ${error.message}`));
+      if (typeof operation === "string") {
+        if (operation.startsWith("group-")) continue;
+        const handler = handlers[operation];
+        if (handler) {
+          try {
+            const currentConfig = getConfig();
+            const clients = {
+              vision: new Langvision({ apiKey: currentConfig.apiKey }),
+              tune: new Langtune({ apiKey: currentConfig.apiKey }),
+              agent: new AgentClient({ apiKey: currentConfig.apiKey, baseUrl: currentConfig.baseUrl })
+            };
+            await handler(clients);
+          } catch (error) {
+            (0, import_prompts.outro)((0, import_colors.red)(`Error: ${error.message}`));
+          }
+        }
       }
     }
   });