langtrain 0.1.9 → 0.1.10

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "langtrain",
-  "version": "0.1.9",
+  "version": "0.1.10",
   "description": "Unified JavaScript SDK for Langtrain Ecosystem",
   "main": "./dist/index.js",
   "module": "./dist/index.mjs",

package/src/cli.ts

@@ -28,16 +28,16 @@ function saveConfig(config: any) {
     fs.writeFileSync(CONFIG_FILE, JSON.stringify(config, null, 2));
 }
 
-// Initialize clients with config
-// Clients are initialized inside main loop to support re-login
+const packageJson = require(path.join(__dirname, '../package.json'));
 
 async function main() {
     const program = new Command();
+    const version = packageJson.version;
 
     program
         .name('langtrain')
-        .description('Langtrain CLI for AI Model Fine-tuning and Generation')
-        .version('0.1.9');
+        .description(packageJson.description || 'Langtrain CLI for AI Model Fine-tuning and Generation')
+        .version(version);
 
     program.action(async () => {
         console.clear();
@@ -52,28 +52,26 @@ async function main() {
     ╚══════╝╚═╝  ╚═╝╚═╝  ╚═══╝ ╚═════╝   ╚═╝   ╚═╝  ╚═╝╚═╝  ╚═╝╚═╝╚═╝  ╚═══╝
     `;
         console.log(gradient(['#00DC82', '#36E4DA', '#0047E1'])(banner)); // Custom Langtrain Green-Cyan-Blue gradient
-        intro(`${bgCyan(black(' Langtrain SDK v0.1.9 '))}`);
+        intro(`${bgCyan(black(` Langtrain SDK v${version} `))}`);
 
-        // Check auth
+        // Check auth (only show login if missing)
         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');
-                process.exit(0);
-            }
-
-            saveConfig({ ...config, apiKey });
-            intro(green('Successfully logged in!'));
+            await handleLogin();
         }
 
+        // Operation Handlers Map (O(1) lookup)
+        const handlers: Record<string, (clients?: any) => Promise<void>> = {
+            '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); },
+        };
+
         while (true) {
             const operation = await select({
                 message: 'Select an operation:',
@@ -95,39 +93,31 @@ async function main() {
                 ],
             });
 
-            if (isCancel(operation) || operation === 'exit') {
+            if (isCancel(operation)) {
                 outro('Goodbye!');
                 process.exit(0);
             }
 
-            if (typeof operation === 'string' && operation.startsWith('group-')) {
-                // Header selected, just loop
-                continue;
-            }
-
-            try {
-                // Re-read config & re-init clients
-                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);
+            if (typeof operation === 'string') {
+                if (operation.startsWith('group-')) continue;
+
+                // Execute handler via map lookup
+                const handler = handlers[operation];
+                if (handler) {
+                    try {
+                        // Re-read config & re-init clients freshly for each operation
+                        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: any) {
+                        outro(red(`Error: ${error.message}`));
+                    }
                 }
-            } catch (error: any) {
-                outro(red(`Error: ${error.message}`));
-                // Don't exit, just loop back to menu
             }
         }
     });