langtrain 0.1.6 → 0.1.10

package/dist/cli.js

@@ -46,8 +46,8 @@ var require_windows = __commonJS({
     "use strict";
     module2.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;
@@ -58,25 +58,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);
     }
   }
 });
@@ -87,14 +87,14 @@ var require_mode = __commonJS({
     "use strict";
     module2.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);
@@ -119,7 +119,7 @@ var require_mode = __commonJS({
 var require_isexe = __commonJS({
   "../langvision/js/node_modules/isexe/index.js"(exports2, module2) {
     "use strict";
-    var fs = require("fs");
+    var fs2 = require("fs");
     var core;
     if (process.platform === "win32" || global.TESTING_WINDOWS) {
       core = require_windows();
@@ -128,7 +128,7 @@ var require_isexe = __commonJS({
     }
     module2.exports = isexe;
     isexe.sync = sync;
-    function isexe(path5, options, cb) {
+    function isexe(path6, options, cb) {
       if (typeof options === "function") {
         cb = options;
         options = {};
@@ -138,7 +138,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 {
@@ -147,7 +147,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;
@@ -157,9 +157,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;
@@ -176,7 +176,7 @@ var require_which = __commonJS({
   "../langvision/js/node_modules/which/which.js"(exports2, module2) {
     "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" });
@@ -214,7 +214,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));
       });
@@ -241,7 +241,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];
@@ -289,7 +289,7 @@ var require_path_key = __commonJS({
 var require_resolveCommand = __commonJS({
   "../langvision/js/node_modules/cross-spawn/lib/util/resolveCommand.js"(exports2, module2) {
     "use strict";
-    var path5 = require("path");
+    var path6 = require("path");
     var which = require_which();
     var getPathKey = require_path_key();
     function resolveCommandAttempt(parsed, withoutPathExt) {
@@ -307,7 +307,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 {
@@ -316,7 +316,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;
     }
@@ -370,8 +370,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;
       }
@@ -384,16 +384,16 @@ var require_shebang_command = __commonJS({
 var require_readShebang = __commonJS({
   "../langvision/js/node_modules/cross-spawn/lib/util/readShebang.js"(exports2, module2) {
     "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());
@@ -406,7 +406,7 @@ var require_readShebang = __commonJS({
 var require_parse = __commonJS({
   "../langvision/js/node_modules/cross-spawn/lib/parse.js"(exports2, module2) {
     "use strict";
-    var path5 = require("path");
+    var path6 = require("path");
     var resolveCommand = require_resolveCommand();
     var escape = require_escape();
     var readShebang = require_readShebang();
@@ -431,7 +431,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(" ");
@@ -587,8 +587,8 @@ var require_windows2 = __commonJS({
     "use strict";
     module2.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;
@@ -599,25 +599,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);
     }
   }
 });
@@ -628,14 +628,14 @@ var require_mode2 = __commonJS({
     "use strict";
     module2.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);
@@ -660,7 +660,7 @@ var require_mode2 = __commonJS({
 var require_isexe2 = __commonJS({
   "../langtune/js/node_modules/isexe/index.js"(exports2, module2) {
     "use strict";
-    var fs = require("fs");
+    var fs2 = require("fs");
     var core;
     if (process.platform === "win32" || global.TESTING_WINDOWS) {
       core = require_windows2();
@@ -669,7 +669,7 @@ var require_isexe2 = __commonJS({
     }
     module2.exports = isexe;
     isexe.sync = sync;
-    function isexe(path5, options, cb) {
+    function isexe(path6, options, cb) {
       if (typeof options === "function") {
         cb = options;
         options = {};
@@ -679,7 +679,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 {
@@ -688,7 +688,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;
@@ -698,9 +698,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;
@@ -717,7 +717,7 @@ var require_which2 = __commonJS({
   "../langtune/js/node_modules/which/which.js"(exports2, module2) {
     "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" });
@@ -755,7 +755,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));
       });
@@ -782,7 +782,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];
@@ -830,7 +830,7 @@ var require_path_key2 = __commonJS({
 var require_resolveCommand2 = __commonJS({
   "../langtune/js/node_modules/cross-spawn/lib/util/resolveCommand.js"(exports2, module2) {
     "use strict";
-    var path5 = require("path");
+    var path6 = require("path");
     var which = require_which2();
     var getPathKey = require_path_key2();
     function resolveCommandAttempt(parsed, withoutPathExt) {
@@ -848,7 +848,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 {
@@ -857,7 +857,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;
     }
@@ -911,8 +911,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;
       }
@@ -925,16 +925,16 @@ var require_shebang_command2 = __commonJS({
 var require_readShebang2 = __commonJS({
   "../langtune/js/node_modules/cross-spawn/lib/util/readShebang.js"(exports2, module2) {
     "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());
@@ -947,7 +947,7 @@ var require_readShebang2 = __commonJS({
 var require_parse2 = __commonJS({
   "../langtune/js/node_modules/cross-spawn/lib/parse.js"(exports2, module2) {
     "use strict";
-    var path5 = require("path");
+    var path6 = require("path");
     var resolveCommand = require_resolveCommand2();
     var escape = require_escape2();
     var readShebang = require_readShebang2();
@@ -972,7 +972,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(" ");
@@ -3203,8 +3203,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
@@ -3320,11 +3320,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() {
@@ -6779,6 +6779,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
@@ -6803,7 +6810,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}`);
     }
@@ -6826,7 +6836,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}`);
@@ -8910,8 +8922,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
@@ -9027,11 +9039,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() {
@@ -12489,6 +12501,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
@@ -12521,7 +12540,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}`);
     }
@@ -12544,7 +12566,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}`);
@@ -12552,48 +12576,190 @@ var Langtune = class {
   }
 };
 
+// src/agent.ts
+var import_axios = __toESM(require("axios"));
+var AgentClient = class {
+  constructor(config) {
+    this.config = config;
+    this.client = import_axios.default.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();
+var import_fs = __toESM(require("fs"));
+var import_path = __toESM(require("path"));
+var import_os = __toESM(require("os"));
+var import_gradient_string = __toESM(require("gradient-string"));
+var CONFIG_DIR = import_path.default.join(import_os.default.homedir(), ".langtrain");
+var CONFIG_FILE = import_path.default.join(CONFIG_DIR, "config.json");
+function getConfig() {
+  if (!import_fs.default.existsSync(CONFIG_FILE)) return {};
+  try {
+    return JSON.parse(import_fs.default.readFileSync(CONFIG_FILE, "utf-8"));
+  } catch {
+    return {};
+  }
+}
+function saveConfig(config) {
+  if (!import_fs.default.existsSync(CONFIG_DIR)) {
+    import_fs.default.mkdirSync(CONFIG_DIR, { recursive: true });
+  }
+  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.5");
+  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();
-    (0, import_prompts.intro)(`${(0, import_colors.bgCyan)((0, import_colors.black)(" langtrain "))}`);
-    const operation = await (0, import_prompts.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 ((0, import_prompts.isCancel)(operation) || operation === "exit") {
-      (0, import_prompts.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((0, import_gradient_string.default)(["#00DC82", "#36E4DA", "#0047E1"])(banner));
+    (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"));
+      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);
+      }
+    };
+    while (true) {
+      const operation = await (0, import_prompts.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 ((0, import_prompts.isCancel)(operation)) {
+        (0, import_prompts.outro)("Goodbye!");
+        process.exit(0);
+      }
+      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}`));
+          }
+        }
       }
-    } catch (error) {
-      (0, import_prompts.outro)((0, import_colors.red)(`Error: ${error.message}`));
-      process.exit(1);
     }
-    (0, import_prompts.outro)((0, import_colors.green)("Operation completed successfully!"));
   });
   program.parse(process.argv);
 }
-async function handleTuneFinetune() {
+async function handleLogin() {
+  const apiKey = await (0, import_prompts.password)({
+    message: "Enter your new 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");
+  const config = getConfig();
+  saveConfig({ ...config, apiKey });
+  (0, import_prompts.intro)((0, import_colors.green)("API Key updated successfully!"));
+}
+async function handleAgentList(client) {
+  const s = (0, import_prompts.spinner)();
+  s.start("Fetching agents...");
+  const agents = await client.list();
+  s.stop(`Found ${agents.length} agents`);
+  if (agents.length === 0) {
+    (0, import_prompts.intro)((0, import_colors.yellow)("No agents found in your workspace."));
+    return;
+  }
+  const agentId = await (0, import_prompts.select)({
+    message: "Select an agent to run:",
+    options: agents.map((a) => ({ value: a.id, label: a.name, hint: a.description || "No description" }))
+  });
+  if ((0, import_prompts.isCancel)(agentId)) return;
+  await handleAgentRun(client, agentId, agents.find((a) => a.id === agentId)?.name || "Agent");
+}
+async function handleAgentRun(client, agentId, agentName) {
+  (0, import_prompts.intro)((0, import_colors.bgCyan)((0, import_colors.black)(` Chatting with ${agentName} `)));
+  console.log((0, import_colors.gray)('Type "exit" to quit conversation.'));
+  let conversationId = void 0;
+  while (true) {
+    const input = await (0, import_prompts.text)({
+      message: "You:",
+      placeholder: "Type a message..."
+    });
+    if ((0, import_prompts.isCancel)(input) || input === "exit") {
+      break;
+    }
+    const s = (0, import_prompts.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(import_gradient_string.default.pastel(`Agent: ${result.output.response}`));
+      } else {
+        console.log(import_gradient_string.default.pastel(`Agent: ${JSON.stringify(result.output)}`));
+      }
+      conversationId = result.conversation_id;
+    } catch (e) {
+      s.stop((0, import_colors.red)("Error running agent."));
+      console.error(e);
+    }
+  }
+}
+async function handleTuneFinetune(tune) {
   const model = await (0, import_prompts.text)({
     message: "Enter base model (e.g., gpt-3.5-turbo):",
     placeholder: "gpt-3.5-turbo",
@@ -12617,7 +12783,9 @@ async function handleTuneFinetune() {
   });
   if ((0, import_prompts.isCancel)(epochs)) (0, import_prompts.cancel)("Operation cancelled.");
   const s = (0, import_prompts.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,
@@ -12631,13 +12799,13 @@ async function handleTuneFinetune() {
       outputDir: "./output"
     };
     await tune.finetune(config);
-    s.stop((0, import_colors.green)("Fine-tuning job started!"));
+    s.stop((0, import_colors.green)("Fine-tuning job started successfully! \u{1F680}"));
   } catch (e) {
     s.stop((0, import_colors.red)("Failed to start job."));
     throw e;
   }
 }
-async function handleTuneGenerate() {
+async function handleTuneGenerate(tune) {
   const model = await (0, import_prompts.text)({
     message: "Enter model path:",
     placeholder: "./output/model",
@@ -12650,18 +12818,18 @@ async function handleTuneGenerate() {
   });
   if ((0, import_prompts.isCancel)(prompt)) (0, import_prompts.cancel)("Operation cancelled");
   const s = (0, import_prompts.spinner)();
-  s.start("Generating response...");
+  s.start("Connecting to Langtrain Inference API...");
   try {
     const response = await tune.generate(model, { prompt });
     s.stop("Generation complete");
     (0, import_prompts.intro)("Response:");
-    console.log(response);
+    console.log(import_gradient_string.default.pastel(response));
   } catch (e) {
     s.stop((0, import_colors.red)("Generation failed."));
     throw e;
   }
 }
-async function handleVisionFinetune() {
+async function handleVisionFinetune(vision) {
   const model = await (0, import_prompts.text)({
     message: "Enter base vision model:",
     placeholder: "llava-v1.5-7b",
@@ -12679,7 +12847,9 @@ async function handleVisionFinetune() {
     initialValue: "3"
   });
   const s = (0, import_prompts.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,
@@ -12691,13 +12861,13 @@ async function handleVisionFinetune() {
       outputDir: "./vision-output"
     };
     await vision.finetune(config);
-    s.stop((0, import_colors.green)("Vision fine-tuning started!"));
+    s.stop((0, import_colors.green)("Vision fine-tuning started successfully! \u{1F441}\uFE0F"));
   } catch (e) {
     s.stop((0, import_colors.red)("Failed to start vision job."));
     throw e;
   }
 }
-async function handleVisionGenerate() {
+async function handleVisionGenerate(vision) {
   const model = await (0, import_prompts.text)({
     message: "Enter model path:",
     placeholder: "./vision-output/model",
@@ -12711,12 +12881,14 @@ async function handleVisionGenerate() {
   });
   if ((0, import_prompts.isCancel)(prompt)) (0, import_prompts.cancel)("Operation cancelled");
   const s = (0, import_prompts.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");
     (0, import_prompts.intro)("Response:");
-    console.log(response);
+    console.log(import_gradient_string.default.pastel(response));
   } catch (e) {
     s.stop((0, import_colors.red)("Generation failed."));
     throw e;