langtrain 0.1.6 → 0.1.9

package/dist/cli.mjs

@@ -51,8 +51,8 @@ var require_windows = __commonJS({
     "use strict";
     module.exports = isexe;
     isexe.sync = sync;
-    var fs = __require("fs");
-    function checkPathExt(path5, options) {
+    var fs2 = __require("fs");
+    function checkPathExt(path6, options) {
       var pathext = options.pathExt !== void 0 ? options.pathExt : process.env.PATHEXT;
       if (!pathext) {
         return true;
@@ -63,25 +63,25 @@ var require_windows = __commonJS({
       }
       for (var i = 0; i < pathext.length; i++) {
         var p = pathext[i].toLowerCase();
-        if (p && path5.substr(-p.length).toLowerCase() === p) {
+        if (p && path6.substr(-p.length).toLowerCase() === p) {
           return true;
         }
       }
       return false;
     }
-    function checkStat(stat, path5, options) {
+    function checkStat(stat, path6, options) {
       if (!stat.isSymbolicLink() && !stat.isFile()) {
         return false;
       }
-      return checkPathExt(path5, options);
+      return checkPathExt(path6, options);
     }
-    function isexe(path5, options, cb) {
-      fs.stat(path5, function(er, stat) {
-        cb(er, er ? false : checkStat(stat, path5, options));
+    function isexe(path6, options, cb) {
+      fs2.stat(path6, function(er, stat) {
+        cb(er, er ? false : checkStat(stat, path6, options));
       });
     }
-    function sync(path5, options) {
-      return checkStat(fs.statSync(path5), path5, options);
+    function sync(path6, options) {
+      return checkStat(fs2.statSync(path6), path6, options);
     }
   }
 });
@@ -92,14 +92,14 @@ var require_mode = __commonJS({
     "use strict";
     module.exports = isexe;
     isexe.sync = sync;
-    var fs = __require("fs");
-    function isexe(path5, options, cb) {
-      fs.stat(path5, function(er, stat) {
+    var fs2 = __require("fs");
+    function isexe(path6, options, cb) {
+      fs2.stat(path6, function(er, stat) {
         cb(er, er ? false : checkStat(stat, options));
       });
     }
-    function sync(path5, options) {
-      return checkStat(fs.statSync(path5), options);
+    function sync(path6, options) {
+      return checkStat(fs2.statSync(path6), options);
     }
     function checkStat(stat, options) {
       return stat.isFile() && checkMode(stat, options);
@@ -124,7 +124,7 @@ var require_mode = __commonJS({
 var require_isexe = __commonJS({
   "../langvision/js/node_modules/isexe/index.js"(exports, module) {
     "use strict";
-    var fs = __require("fs");
+    var fs2 = __require("fs");
     var core;
     if (process.platform === "win32" || global.TESTING_WINDOWS) {
       core = require_windows();
@@ -133,7 +133,7 @@ var require_isexe = __commonJS({
     }
     module.exports = isexe;
     isexe.sync = sync;
-    function isexe(path5, options, cb) {
+    function isexe(path6, options, cb) {
       if (typeof options === "function") {
         cb = options;
         options = {};
@@ -143,7 +143,7 @@ var require_isexe = __commonJS({
           throw new TypeError("callback not provided");
         }
         return new Promise(function(resolve, reject) {
-          isexe(path5, options || {}, function(er, is) {
+          isexe(path6, options || {}, function(er, is) {
             if (er) {
               reject(er);
             } else {
@@ -152,7 +152,7 @@ var require_isexe = __commonJS({
           });
         });
       }
-      core(path5, options || {}, function(er, is) {
+      core(path6, options || {}, function(er, is) {
         if (er) {
           if (er.code === "EACCES" || options && options.ignoreErrors) {
             er = null;
@@ -162,9 +162,9 @@ var require_isexe = __commonJS({
         cb(er, is);
       });
     }
-    function sync(path5, options) {
+    function sync(path6, options) {
       try {
-        return core.sync(path5, options || {});
+        return core.sync(path6, options || {});
       } catch (er) {
         if (options && options.ignoreErrors || er.code === "EACCES") {
           return false;
@@ -181,7 +181,7 @@ var require_which = __commonJS({
   "../langvision/js/node_modules/which/which.js"(exports, module) {
     "use strict";
     var isWindows = process.platform === "win32" || process.env.OSTYPE === "cygwin" || process.env.OSTYPE === "msys";
-    var path5 = __require("path");
+    var path6 = __require("path");
     var COLON = isWindows ? ";" : ":";
     var isexe = require_isexe();
     var getNotFoundError = (cmd) => Object.assign(new Error(`not found: ${cmd}`), { code: "ENOENT" });
@@ -219,7 +219,7 @@ var require_which = __commonJS({
           return opt.all && found.length ? resolve(found) : reject(getNotFoundError(cmd));
         const ppRaw = pathEnv[i];
         const pathPart = /^".*"$/.test(ppRaw) ? ppRaw.slice(1, -1) : ppRaw;
-        const pCmd = path5.join(pathPart, cmd);
+        const pCmd = path6.join(pathPart, cmd);
         const p = !pathPart && /^\.[\\\/]/.test(cmd) ? cmd.slice(0, 2) + pCmd : pCmd;
         resolve(subStep(p, i, 0));
       });
@@ -246,7 +246,7 @@ var require_which = __commonJS({
       for (let i = 0; i < pathEnv.length; i++) {
         const ppRaw = pathEnv[i];
         const pathPart = /^".*"$/.test(ppRaw) ? ppRaw.slice(1, -1) : ppRaw;
-        const pCmd = path5.join(pathPart, cmd);
+        const pCmd = path6.join(pathPart, cmd);
         const p = !pathPart && /^\.[\\\/]/.test(cmd) ? cmd.slice(0, 2) + pCmd : pCmd;
         for (let j = 0; j < pathExt.length; j++) {
           const cur = p + pathExt[j];
@@ -294,7 +294,7 @@ var require_path_key = __commonJS({
 var require_resolveCommand = __commonJS({
   "../langvision/js/node_modules/cross-spawn/lib/util/resolveCommand.js"(exports, module) {
     "use strict";
-    var path5 = __require("path");
+    var path6 = __require("path");
     var which = require_which();
     var getPathKey = require_path_key();
     function resolveCommandAttempt(parsed, withoutPathExt) {
@@ -312,7 +312,7 @@ var require_resolveCommand = __commonJS({
       try {
         resolved = which.sync(parsed.command, {
           path: env[getPathKey({ env })],
-          pathExt: withoutPathExt ? path5.delimiter : void 0
+          pathExt: withoutPathExt ? path6.delimiter : void 0
         });
       } catch (e) {
       } finally {
@@ -321,7 +321,7 @@ var require_resolveCommand = __commonJS({
         }
       }
       if (resolved) {
-        resolved = path5.resolve(hasCustomCwd ? parsed.options.cwd : "", resolved);
+        resolved = path6.resolve(hasCustomCwd ? parsed.options.cwd : "", resolved);
       }
       return resolved;
     }
@@ -375,8 +375,8 @@ var require_shebang_command = __commonJS({
       if (!match) {
         return null;
       }
-      const [path5, argument] = match[0].replace(/#! ?/, "").split(" ");
-      const binary = path5.split("/").pop();
+      const [path6, argument] = match[0].replace(/#! ?/, "").split(" ");
+      const binary = path6.split("/").pop();
       if (binary === "env") {
         return argument;
       }
@@ -389,16 +389,16 @@ var require_shebang_command = __commonJS({
 var require_readShebang = __commonJS({
   "../langvision/js/node_modules/cross-spawn/lib/util/readShebang.js"(exports, module) {
     "use strict";
-    var fs = __require("fs");
+    var fs2 = __require("fs");
     var shebangCommand = require_shebang_command();
     function readShebang(command) {
       const size = 150;
       const buffer = Buffer.alloc(size);
       let fd;
       try {
-        fd = fs.openSync(command, "r");
-        fs.readSync(fd, buffer, 0, size, 0);
-        fs.closeSync(fd);
+        fd = fs2.openSync(command, "r");
+        fs2.readSync(fd, buffer, 0, size, 0);
+        fs2.closeSync(fd);
       } catch (e) {
       }
       return shebangCommand(buffer.toString());
@@ -411,7 +411,7 @@ var require_readShebang = __commonJS({
 var require_parse = __commonJS({
   "../langvision/js/node_modules/cross-spawn/lib/parse.js"(exports, module) {
     "use strict";
-    var path5 = __require("path");
+    var path6 = __require("path");
     var resolveCommand = require_resolveCommand();
     var escape = require_escape();
     var readShebang = require_readShebang();
@@ -436,7 +436,7 @@ var require_parse = __commonJS({
       const needsShell = !isExecutableRegExp.test(commandFile);
       if (parsed.options.forceShell || needsShell) {
         const needsDoubleEscapeMetaChars = isCmdShimRegExp.test(commandFile);
-        parsed.command = path5.normalize(parsed.command);
+        parsed.command = path6.normalize(parsed.command);
         parsed.command = escape.command(parsed.command);
         parsed.args = parsed.args.map((arg) => escape.argument(arg, needsDoubleEscapeMetaChars));
         const shellCommand = [parsed.command].concat(parsed.args).join(" ");
@@ -592,8 +592,8 @@ var require_windows2 = __commonJS({
     "use strict";
     module.exports = isexe;
     isexe.sync = sync;
-    var fs = __require("fs");
-    function checkPathExt(path5, options) {
+    var fs2 = __require("fs");
+    function checkPathExt(path6, options) {
       var pathext = options.pathExt !== void 0 ? options.pathExt : process.env.PATHEXT;
       if (!pathext) {
         return true;
@@ -604,25 +604,25 @@ var require_windows2 = __commonJS({
       }
       for (var i = 0; i < pathext.length; i++) {
         var p = pathext[i].toLowerCase();
-        if (p && path5.substr(-p.length).toLowerCase() === p) {
+        if (p && path6.substr(-p.length).toLowerCase() === p) {
           return true;
         }
       }
       return false;
     }
-    function checkStat(stat, path5, options) {
+    function checkStat(stat, path6, options) {
       if (!stat.isSymbolicLink() && !stat.isFile()) {
         return false;
       }
-      return checkPathExt(path5, options);
+      return checkPathExt(path6, options);
     }
-    function isexe(path5, options, cb) {
-      fs.stat(path5, function(er, stat) {
-        cb(er, er ? false : checkStat(stat, path5, options));
+    function isexe(path6, options, cb) {
+      fs2.stat(path6, function(er, stat) {
+        cb(er, er ? false : checkStat(stat, path6, options));
       });
     }
-    function sync(path5, options) {
-      return checkStat(fs.statSync(path5), path5, options);
+    function sync(path6, options) {
+      return checkStat(fs2.statSync(path6), path6, options);
     }
   }
 });
@@ -633,14 +633,14 @@ var require_mode2 = __commonJS({
     "use strict";
     module.exports = isexe;
     isexe.sync = sync;
-    var fs = __require("fs");
-    function isexe(path5, options, cb) {
-      fs.stat(path5, function(er, stat) {
+    var fs2 = __require("fs");
+    function isexe(path6, options, cb) {
+      fs2.stat(path6, function(er, stat) {
         cb(er, er ? false : checkStat(stat, options));
       });
     }
-    function sync(path5, options) {
-      return checkStat(fs.statSync(path5), options);
+    function sync(path6, options) {
+      return checkStat(fs2.statSync(path6), options);
     }
     function checkStat(stat, options) {
       return stat.isFile() && checkMode(stat, options);
@@ -665,7 +665,7 @@ var require_mode2 = __commonJS({
 var require_isexe2 = __commonJS({
   "../langtune/js/node_modules/isexe/index.js"(exports, module) {
     "use strict";
-    var fs = __require("fs");
+    var fs2 = __require("fs");
     var core;
     if (process.platform === "win32" || global.TESTING_WINDOWS) {
       core = require_windows2();
@@ -674,7 +674,7 @@ var require_isexe2 = __commonJS({
     }
     module.exports = isexe;
     isexe.sync = sync;
-    function isexe(path5, options, cb) {
+    function isexe(path6, options, cb) {
       if (typeof options === "function") {
         cb = options;
         options = {};
@@ -684,7 +684,7 @@ var require_isexe2 = __commonJS({
           throw new TypeError("callback not provided");
         }
         return new Promise(function(resolve, reject) {
-          isexe(path5, options || {}, function(er, is) {
+          isexe(path6, options || {}, function(er, is) {
             if (er) {
               reject(er);
             } else {
@@ -693,7 +693,7 @@ var require_isexe2 = __commonJS({
           });
         });
       }
-      core(path5, options || {}, function(er, is) {
+      core(path6, options || {}, function(er, is) {
         if (er) {
           if (er.code === "EACCES" || options && options.ignoreErrors) {
             er = null;
@@ -703,9 +703,9 @@ var require_isexe2 = __commonJS({
         cb(er, is);
       });
     }
-    function sync(path5, options) {
+    function sync(path6, options) {
       try {
-        return core.sync(path5, options || {});
+        return core.sync(path6, options || {});
       } catch (er) {
         if (options && options.ignoreErrors || er.code === "EACCES") {
           return false;
@@ -722,7 +722,7 @@ var require_which2 = __commonJS({
   "../langtune/js/node_modules/which/which.js"(exports, module) {
     "use strict";
     var isWindows = process.platform === "win32" || process.env.OSTYPE === "cygwin" || process.env.OSTYPE === "msys";
-    var path5 = __require("path");
+    var path6 = __require("path");
     var COLON = isWindows ? ";" : ":";
     var isexe = require_isexe2();
     var getNotFoundError = (cmd) => Object.assign(new Error(`not found: ${cmd}`), { code: "ENOENT" });
@@ -760,7 +760,7 @@ var require_which2 = __commonJS({
           return opt.all && found.length ? resolve(found) : reject(getNotFoundError(cmd));
         const ppRaw = pathEnv[i];
         const pathPart = /^".*"$/.test(ppRaw) ? ppRaw.slice(1, -1) : ppRaw;
-        const pCmd = path5.join(pathPart, cmd);
+        const pCmd = path6.join(pathPart, cmd);
         const p = !pathPart && /^\.[\\\/]/.test(cmd) ? cmd.slice(0, 2) + pCmd : pCmd;
         resolve(subStep(p, i, 0));
       });
@@ -787,7 +787,7 @@ var require_which2 = __commonJS({
       for (let i = 0; i < pathEnv.length; i++) {
         const ppRaw = pathEnv[i];
         const pathPart = /^".*"$/.test(ppRaw) ? ppRaw.slice(1, -1) : ppRaw;
-        const pCmd = path5.join(pathPart, cmd);
+        const pCmd = path6.join(pathPart, cmd);
         const p = !pathPart && /^\.[\\\/]/.test(cmd) ? cmd.slice(0, 2) + pCmd : pCmd;
         for (let j = 0; j < pathExt.length; j++) {
           const cur = p + pathExt[j];
@@ -835,7 +835,7 @@ var require_path_key2 = __commonJS({
 var require_resolveCommand2 = __commonJS({
   "../langtune/js/node_modules/cross-spawn/lib/util/resolveCommand.js"(exports, module) {
     "use strict";
-    var path5 = __require("path");
+    var path6 = __require("path");
     var which = require_which2();
     var getPathKey = require_path_key2();
     function resolveCommandAttempt(parsed, withoutPathExt) {
@@ -853,7 +853,7 @@ var require_resolveCommand2 = __commonJS({
       try {
         resolved = which.sync(parsed.command, {
           path: env[getPathKey({ env })],
-          pathExt: withoutPathExt ? path5.delimiter : void 0
+          pathExt: withoutPathExt ? path6.delimiter : void 0
         });
       } catch (e) {
       } finally {
@@ -862,7 +862,7 @@ var require_resolveCommand2 = __commonJS({
         }
       }
       if (resolved) {
-        resolved = path5.resolve(hasCustomCwd ? parsed.options.cwd : "", resolved);
+        resolved = path6.resolve(hasCustomCwd ? parsed.options.cwd : "", resolved);
       }
       return resolved;
     }
@@ -916,8 +916,8 @@ var require_shebang_command2 = __commonJS({
       if (!match) {
         return null;
       }
-      const [path5, argument] = match[0].replace(/#! ?/, "").split(" ");
-      const binary = path5.split("/").pop();
+      const [path6, argument] = match[0].replace(/#! ?/, "").split(" ");
+      const binary = path6.split("/").pop();
       if (binary === "env") {
         return argument;
       }
@@ -930,16 +930,16 @@ var require_shebang_command2 = __commonJS({
 var require_readShebang2 = __commonJS({
   "../langtune/js/node_modules/cross-spawn/lib/util/readShebang.js"(exports, module) {
     "use strict";
-    var fs = __require("fs");
+    var fs2 = __require("fs");
     var shebangCommand = require_shebang_command2();
     function readShebang(command) {
       const size = 150;
       const buffer = Buffer.alloc(size);
       let fd;
       try {
-        fd = fs.openSync(command, "r");
-        fs.readSync(fd, buffer, 0, size, 0);
-        fs.closeSync(fd);
+        fd = fs2.openSync(command, "r");
+        fs2.readSync(fd, buffer, 0, size, 0);
+        fs2.closeSync(fd);
       } catch (e) {
       }
       return shebangCommand(buffer.toString());
@@ -952,7 +952,7 @@ var require_readShebang2 = __commonJS({
 var require_parse2 = __commonJS({
   "../langtune/js/node_modules/cross-spawn/lib/parse.js"(exports, module) {
     "use strict";
-    var path5 = __require("path");
+    var path6 = __require("path");
     var resolveCommand = require_resolveCommand2();
     var escape = require_escape2();
     var readShebang = require_readShebang2();
@@ -977,7 +977,7 @@ var require_parse2 = __commonJS({
       const needsShell = !isExecutableRegExp.test(commandFile);
       if (parsed.options.forceShell || needsShell) {
         const needsDoubleEscapeMetaChars = isCmdShimRegExp.test(commandFile);
-        parsed.command = path5.normalize(parsed.command);
+        parsed.command = path6.normalize(parsed.command);
         parsed.command = escape.command(parsed.command);
         parsed.args = parsed.args.map((arg) => escape.argument(arg, needsDoubleEscapeMetaChars));
         const shellCommand = [parsed.command].concat(parsed.args).join(" ");
@@ -1128,8 +1128,8 @@ var require_merge_stream2 = __commonJS({
 });
 
 // src/cli.ts
-import { intro, outro, select, text, spinner, isCancel, cancel } from "@clack/prompts";
-import { bgCyan, black, red, green } from "kleur/colors";
+import { intro, outro, select, text, spinner, isCancel, cancel, password } from "@clack/prompts";
+import { bgCyan, black, red, green, yellow, gray } from "kleur/colors";
 import { Command } from "commander";
 
 // ../langvision/js/node_modules/execa/index.js
@@ -3208,8 +3208,8 @@ function getErrorMap() {
 
 // ../langvision/js/node_modules/zod/v3/helpers/parseUtil.js
 var makeIssue = (params) => {
-  const { data, path: path5, errorMaps, issueData } = params;
-  const fullPath = [...path5, ...issueData.path || []];
+  const { data, path: path6, errorMaps, issueData } = params;
+  const fullPath = [...path6, ...issueData.path || []];
   const fullIssue = {
     ...issueData,
     path: fullPath
@@ -3325,11 +3325,11 @@ var errorUtil;
 
 // ../langvision/js/node_modules/zod/v3/types.js
 var ParseInputLazyPath = class {
-  constructor(parent, value, path5, key) {
+  constructor(parent, value, path6, key) {
     this._cachedPath = [];
     this.parent = parent;
     this.data = value;
-    this._path = path5;
+    this._path = path6;
     this._key = key;
   }
   get path() {
@@ -6784,6 +6784,13 @@ var FinetuneConfigSchema = external_exports.object({
 var Langvision = class {
   constructor(options = {}) {
     this.cliPath = options.cliPath || "langvision";
+    this.apiKey = options.apiKey;
+  }
+  getEnv() {
+    return {
+      ...process.env,
+      LANGVISION_API_KEY: this.apiKey || process.env.LANGVISION_API_KEY
+    };
   }
   /**
    * Run a fine-tuning job using the local CLI
@@ -6808,7 +6815,10 @@ var Langvision = class {
       validated.outputDir
     ];
     try {
-      await execa(this.cliPath, args, { stdio: "inherit" });
+      await execa(this.cliPath, args, {
+        stdio: "inherit",
+        env: this.getEnv()
+      });
     } catch (error) {
       throw new Error(`Langvision fine-tuning failed: ${error}`);
     }
@@ -6831,7 +6841,9 @@ var Langvision = class {
       args.push("--temperature", options.temperature.toString());
     }
     try {
-      const { stdout } = await execa(this.cliPath, args);
+      const { stdout } = await execa(this.cliPath, args, {
+        env: this.getEnv()
+      });
       return stdout;
     } catch (error) {
       throw new Error(`Langvision generation failed: ${error}`);
@@ -8915,8 +8927,8 @@ function getErrorMap2() {
 
 // ../langtune/js/node_modules/zod/v3/helpers/parseUtil.js
 var makeIssue2 = (params) => {
-  const { data, path: path5, errorMaps, issueData } = params;
-  const fullPath = [...path5, ...issueData.path || []];
+  const { data, path: path6, errorMaps, issueData } = params;
+  const fullPath = [...path6, ...issueData.path || []];
   const fullIssue = {
     ...issueData,
     path: fullPath
@@ -9032,11 +9044,11 @@ var errorUtil2;
 
 // ../langtune/js/node_modules/zod/v3/types.js
 var ParseInputLazyPath2 = class {
-  constructor(parent, value, path5, key) {
+  constructor(parent, value, path6, key) {
     this._cachedPath = [];
     this.parent = parent;
     this.data = value;
-    this._path = path5;
+    this._path = path6;
     this._key = key;
   }
   get path() {
@@ -12494,6 +12506,13 @@ var FinetuneConfigSchema2 = external_exports2.object({
 var Langtune = class {
   constructor(options = {}) {
     this.cliPath = options.cliPath || "langtune";
+    this.apiKey = options.apiKey;
+  }
+  getEnv() {
+    return {
+      ...process.env,
+      LANGTUNE_API_KEY: this.apiKey || process.env.LANGTUNE_API_KEY
+    };
   }
   /**
    * Run a fine-tuning job using the local CLI
@@ -12526,7 +12545,10 @@ var Langtune = class {
       args.push("--use-lisa");
     }
     try {
-      await execa2(this.cliPath, args, { stdio: "inherit" });
+      await execa2(this.cliPath, args, {
+        stdio: "inherit",
+        env: this.getEnv()
+      });
     } catch (error) {
       throw new Error(`Langtune fine-tuning failed: ${error}`);
     }
@@ -12549,7 +12571,9 @@ var Langtune = class {
       args.push("--temperature", options.temperature.toString());
     }
     try {
-      const { stdout } = await execa2(this.cliPath, args);
+      const { stdout } = await execa2(this.cliPath, args, {
+        env: this.getEnv()
+      });
       return stdout;
     } catch (error) {
       throw new Error(`Langtune generation failed: ${error}`);
@@ -12557,48 +12581,194 @@ var Langtune = class {
   }
 };
 
+// src/agent.ts
+import axios from "axios";
+var AgentClient = class {
+  constructor(config) {
+    this.config = config;
+    this.client = axios.create({
+      baseURL: config.baseUrl || "https://api.langtrain.ai/api/v1",
+      headers: {
+        "X-API-Key": config.apiKey,
+        "Content-Type": "application/json"
+      }
+    });
+  }
+  async list(workspaceId) {
+    const params = {};
+    if (workspaceId) params.workspace_id = workspaceId;
+    const response = await this.client.get("/agents", { params });
+    return response.data.agents;
+  }
+  async execute(agentId, input, messages = [], conversationId) {
+    const response = await this.client.post(`/agents/${agentId}/execute`, {
+      input,
+      messages,
+      conversation_id: conversationId
+    });
+    return response.data;
+  }
+};
+
 // src/cli.ts
-var vision = new Langvision();
-var tune = new Langtune();
+import fs from "fs";
+import path5 from "path";
+import os3 from "os";
+import gradient from "gradient-string";
+var CONFIG_DIR = path5.join(os3.homedir(), ".langtrain");
+var CONFIG_FILE = path5.join(CONFIG_DIR, "config.json");
+function getConfig() {
+  if (!fs.existsSync(CONFIG_FILE)) return {};
+  try {
+    return JSON.parse(fs.readFileSync(CONFIG_FILE, "utf-8"));
+  } catch {
+    return {};
+  }
+}
+function saveConfig(config) {
+  if (!fs.existsSync(CONFIG_DIR)) {
+    fs.mkdirSync(CONFIG_DIR, { recursive: true });
+  }
+  fs.writeFileSync(CONFIG_FILE, JSON.stringify(config, null, 2));
+}
 async function main() {
   const program = new Command();
-  program.name("langtrain").description("Langtrain CLI for AI Model Fine-tuning and Generation").version("0.1.5");
+  program.name("langtrain").description("Langtrain CLI for AI Model Fine-tuning and Generation").version("0.1.9");
   program.action(async () => {
     console.clear();
-    intro(`${bgCyan(black(" langtrain "))}`);
-    const operation = await select({
-      message: "Select an operation:",
-      options: [
-        { value: "tune-finetune", label: "\u{1F9E0} Fine-tune Text Model (Langtune)" },
-        { value: "tune-generate", label: "\u{1F4DD} Generate Text (Langtune)" },
-        { value: "vision-finetune", label: "\u{1F441}\uFE0F Fine-tune Vision Model (Langvision)" },
-        { value: "vision-generate", label: "\u{1F5BC}\uFE0F Generate Vision Response (Langvision)" },
-        { value: "exit", label: "\u{1F6AA} Exit" }
-      ]
-    });
-    if (isCancel(operation) || operation === "exit") {
-      outro("Goodbye!");
-      process.exit(0);
-    }
-    try {
-      if (operation === "tune-finetune") {
-        await handleTuneFinetune();
-      } else if (operation === "tune-generate") {
-        await handleTuneGenerate();
-      } else if (operation === "vision-finetune") {
-        await handleVisionFinetune();
-      } else if (operation === "vision-generate") {
-        await handleVisionGenerate();
+    const banner = `
+    \u2588\u2588\u2557      \u2588\u2588\u2588\u2588\u2588\u2557 \u2588\u2588\u2588\u2557   \u2588\u2588\u2557 \u2588\u2588\u2588\u2588\u2588\u2588\u2557\u2588\u2588\u2588\u2588\u2588\u2588\u2588\u2588\u2557\u2588\u2588\u2588\u2588\u2588\u2588\u2557  \u2588\u2588\u2588\u2588\u2588\u2557 \u2588\u2588\u2557\u2588\u2588\u2588\u2557   \u2588\u2588\u2557
+    \u2588\u2588\u2551     \u2588\u2588\u2554\u2550\u2550\u2588\u2588\u2557\u2588\u2588\u2588\u2588\u2557  \u2588\u2588\u2551\u2588\u2588\u2554\u2550\u2550\u2550\u2550\u255D\u255A\u2550\u2550\u2588\u2588\u2554\u2550\u2550\u255D\u2588\u2588\u2554\u2550\u2550\u2588\u2588\u2557\u2588\u2588\u2554\u2550\u2550\u2588\u2588\u2557\u2588\u2588\u2551\u2588\u2588\u2588\u2588\u2557  \u2588\u2588\u2551
+    \u2588\u2588\u2551     \u2588\u2588\u2588\u2588\u2588\u2588\u2588\u2551\u2588\u2588\u2554\u2588\u2588\u2557 \u2588\u2588\u2551\u2588\u2588\u2551  \u2588\u2588\u2588\u2557  \u2588\u2588\u2551   \u2588\u2588\u2588\u2588\u2588\u2588\u2554\u255D\u2588\u2588\u2588\u2588\u2588\u2588\u2588\u2551\u2588\u2588\u2551\u2588\u2588\u2554\u2588\u2588\u2557 \u2588\u2588\u2551
+    \u2588\u2588\u2551     \u2588\u2588\u2554\u2550\u2550\u2588\u2588\u2551\u2588\u2588\u2551\u255A\u2588\u2588\u2557\u2588\u2588\u2551\u2588\u2588\u2551   \u2588\u2588\u2551  \u2588\u2588\u2551   \u2588\u2588\u2554\u2550\u2550\u2588\u2588\u2557\u2588\u2588\u2554\u2550\u2550\u2588\u2588\u2551\u2588\u2588\u2551\u2588\u2588\u2551\u255A\u2588\u2588\u2557\u2588\u2588\u2551
+    \u2588\u2588\u2588\u2588\u2588\u2588\u2588\u2557\u2588\u2588\u2551  \u2588\u2588\u2551\u2588\u2588\u2551 \u255A\u2588\u2588\u2588\u2588\u2551\u255A\u2588\u2588\u2588\u2588\u2588\u2588\u2554\u255D  \u2588\u2588\u2551   \u2588\u2588\u2551  \u2588\u2588\u2551\u2588\u2588\u2551  \u2588\u2588\u2551\u2588\u2588\u2551\u2588\u2588\u2551 \u255A\u2588\u2588\u2588\u2588\u2551
+    \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 "))}`);
+    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!"));
+    }
+    while (true) {
+      const operation = await select({
+        message: "Select an operation:",
+        options: [
+          { value: "group-agents", label: "\u{1F916} Agents (Server)", hint: "Chat with custom agents" },
+          { value: "agent-list", label: "  \u21B3 List & Run Agents" },
+          { value: "group-tune", label: "\u{1F9E0} Langtune (LLM)", hint: "Fine-tuning & Text Generation" },
+          { value: "tune-finetune", label: "  \u21B3 Fine-tune Text Model" },
+          { value: "tune-generate", label: "  \u21B3 Generate Text" },
+          { value: "group-vision", label: "\u{1F441}\uFE0F Langvision (Vision)", hint: "Vision Analysis & Tuning" },
+          { value: "vision-finetune", label: "  \u21B3 Fine-tune Vision Model" },
+          { value: "vision-generate", label: "  \u21B3 Generate Vision Response" },
+          { value: "group-settings", label: "\u2699\uFE0F Settings" },
+          { value: "login", label: "  \u21B3 Update API Key" },
+          { value: "exit", label: "  \u21B3 Exit" }
+        ]
+      });
+      if (isCancel(operation) || operation === "exit") {
+        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}`));
       }
-    } catch (error) {
-      outro(red(`Error: ${error.message}`));
-      process.exit(1);
     }
-    outro(green("Operation completed successfully!"));
   });
   program.parse(process.argv);
 }
-async function handleTuneFinetune() {
+async function handleLogin() {
+  const apiKey = await password({
+    message: "Enter your new Langtrain API Key:",
+    validate(value) {
+      if (!value || value.length === 0) return "API Key is required";
+    }
+  });
+  if (isCancel(apiKey)) cancel("Operation cancelled");
+  const config = getConfig();
+  saveConfig({ ...config, apiKey });
+  intro(green("API Key updated successfully!"));
+}
+async function handleAgentList(client) {
+  const s = spinner();
+  s.start("Fetching agents...");
+  const agents = await client.list();
+  s.stop(`Found ${agents.length} agents`);
+  if (agents.length === 0) {
+    intro(yellow("No agents found in your workspace."));
+    return;
+  }
+  const agentId = await select({
+    message: "Select an agent to run:",
+    options: agents.map((a) => ({ value: a.id, label: a.name, hint: a.description || "No description" }))
+  });
+  if (isCancel(agentId)) return;
+  await handleAgentRun(client, agentId, agents.find((a) => a.id === agentId)?.name || "Agent");
+}
+async function handleAgentRun(client, agentId, agentName) {
+  intro(bgCyan(black(` Chatting with ${agentName} `)));
+  console.log(gray('Type "exit" to quit conversation.'));
+  let conversationId = void 0;
+  while (true) {
+    const input = await text({
+      message: "You:",
+      placeholder: "Type a message..."
+    });
+    if (isCancel(input) || input === "exit") {
+      break;
+    }
+    const s = spinner();
+    s.start("Agent is thinking...");
+    try {
+      const result = await client.execute(agentId, { prompt: input }, [], conversationId);
+      s.stop();
+      if (result.output && result.output.response) {
+        console.log(gradient.pastel(`Agent: ${result.output.response}`));
+      } else {
+        console.log(gradient.pastel(`Agent: ${JSON.stringify(result.output)}`));
+      }
+      conversationId = result.conversation_id;
+    } catch (e) {
+      s.stop(red("Error running agent."));
+      console.error(e);
+    }
+  }
+}
+async function handleTuneFinetune(tune) {
   const model = await text({
     message: "Enter base model (e.g., gpt-3.5-turbo):",
     placeholder: "gpt-3.5-turbo",
@@ -12622,7 +12792,9 @@ async function handleTuneFinetune() {
   });
   if (isCancel(epochs)) cancel("Operation cancelled.");
   const s = spinner();
-  s.start("Starting fine-tuning job...");
+  s.start("Connecting to Langtrain Cloud...");
+  await new Promise((r) => setTimeout(r, 800));
+  s.message("Starting fine-tuning job...");
   try {
     const config = {
       model,
@@ -12636,13 +12808,13 @@ async function handleTuneFinetune() {
       outputDir: "./output"
     };
     await tune.finetune(config);
-    s.stop(green("Fine-tuning job started!"));
+    s.stop(green("Fine-tuning job started successfully! \u{1F680}"));
   } catch (e) {
     s.stop(red("Failed to start job."));
     throw e;
   }
 }
-async function handleTuneGenerate() {
+async function handleTuneGenerate(tune) {
   const model = await text({
     message: "Enter model path:",
     placeholder: "./output/model",
@@ -12655,18 +12827,18 @@ async function handleTuneGenerate() {
   });
   if (isCancel(prompt)) cancel("Operation cancelled");
   const s = spinner();
-  s.start("Generating response...");
+  s.start("Connecting to Langtrain Inference API...");
   try {
     const response = await tune.generate(model, { prompt });
     s.stop("Generation complete");
     intro("Response:");
-    console.log(response);
+    console.log(gradient.pastel(response));
   } catch (e) {
     s.stop(red("Generation failed."));
     throw e;
   }
 }
-async function handleVisionFinetune() {
+async function handleVisionFinetune(vision) {
   const model = await text({
     message: "Enter base vision model:",
     placeholder: "llava-v1.5-7b",
@@ -12684,7 +12856,9 @@ async function handleVisionFinetune() {
     initialValue: "3"
   });
   const s = spinner();
-  s.start("Starting vision fine-tuning...");
+  s.start("Analyzing dataset structure...");
+  await new Promise((r) => setTimeout(r, 800));
+  s.message("Starting vision fine-tuning on Langtrain Cloud...");
   try {
     const config = {
       model,
@@ -12696,13 +12870,13 @@ async function handleVisionFinetune() {
       outputDir: "./vision-output"
     };
     await vision.finetune(config);
-    s.stop(green("Vision fine-tuning started!"));
+    s.stop(green("Vision fine-tuning started successfully! \u{1F441}\uFE0F"));
   } catch (e) {
     s.stop(red("Failed to start vision job."));
     throw e;
   }
 }
-async function handleVisionGenerate() {
+async function handleVisionGenerate(vision) {
   const model = await text({
     message: "Enter model path:",
     placeholder: "./vision-output/model",
@@ -12716,12 +12890,14 @@ async function handleVisionGenerate() {
   });
   if (isCancel(prompt)) cancel("Operation cancelled");
   const s = spinner();
-  s.start("Generating vision response...");
+  s.start("Uploading image and context...");
+  await new Promise((r) => setTimeout(r, 600));
+  s.message("Generating vision response...");
   try {
     const response = await vision.generate(model, { prompt });
     s.stop("Generation complete");
     intro("Response:");
-    console.log(response);
+    console.log(gradient.pastel(response));
   } catch (e) {
     s.stop(red("Generation failed."));
     throw e;