langtrain 0.1.9 → 0.1.10

package/dist/cli.mjs

@@ -12631,9 +12631,11 @@ function saveConfig(config) {
   }
   fs.writeFileSync(CONFIG_FILE, JSON.stringify(config, null, 2));
 }
+var packageJson = __require(path5.join(__dirname, "../package.json"));
 async function main() {
   const program = new 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 = `
@@ -12645,23 +12647,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(gradient(["#00DC82", "#36E4DA", "#0047E1"])(banner));
-    intro(`${bgCyan(black(" Langtrain SDK v0.1.9 "))}`);
+    intro(`${bgCyan(black(` Langtrain SDK v${version} `))}`);
     const config = getConfig();
     if (!config.apiKey) {
       intro(yellow("Authentication required"));
-      const apiKey = await password({
-        message: "Enter your Langtrain API Key:",
-        validate(value) {
-          if (!value || value.length === 0) return "API Key is required";
-        }
-      });
-      if (isCancel(apiKey)) {
-        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 () => {
+        outro("Goodbye!");
         process.exit(0);
       }
-      saveConfig({ ...config, apiKey });
-      intro(green("Successfully logged in!"));
-    }
+    };
     while (true) {
       const operation = await select({
         message: "Select an operation:",
@@ -12679,33 +12682,26 @@ async function main() {
           { value: "exit", label: "  \u21B3 Exit" }
         ]
       });
-      if (isCancel(operation) || operation === "exit") {
+      if (isCancel(operation)) {
         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) {
-        outro(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) {
+            outro(red(`Error: ${error.message}`));
+          }
+        }
       }
     }
   });