langtrain 0.1.9 → 0.1.11

package/README.md

@@ -4,7 +4,7 @@
     <source media="(prefers-color-scheme: light)" srcset="https://raw.githubusercontent.com/langtrain-ai/langtrain-sdk/main/public/langtrain-black.svg">
     <img alt="Langtrain Logo" src="https://raw.githubusercontent.com/langtrain-ai/langtrain-sdk/main/public/langtrain-black.svg" width="250">
   </picture>
-  <h1>Langtrain SDK</h1>
+  <h3>Langtrain SDK</h3>
   <p>
     The unified intelligence layer for JavaScript applications. <br/>
     Combine computer vision and LLM fine-tuning in a single, high-performance SDK.

package/dist/cli.js

@@ -12595,6 +12595,17 @@ var AgentClient = class {
     const response = await this.client.get("/agents", { params });
     return response.data.agents;
   }
+  async get(agentId) {
+    const response = await this.client.get(`/agents/${agentId}`);
+    return response.data;
+  }
+  async create(agent) {
+    const response = await this.client.post("/agents/", agent);
+    return response.data;
+  }
+  async delete(agentId) {
+    await this.client.delete(`/agents/${agentId}`);
+  }
   async execute(agentId, input, messages = [], conversationId) {
     const response = await this.client.post(`/agents/${agentId}/execute`, {
       input,
@@ -12626,9 +12637,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,29 +12653,34 @@ 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),
+      "agent-create": (c) => handleAgentCreate(c.agent),
+      "agent-delete": (c) => handleAgentDelete(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:",
         options: [
           { value: "group-agents", label: "\u{1F916} Agents (Server)", hint: "Chat with custom agents" },
           { value: "agent-list", label: "  \u21B3 List & Run Agents" },
+          { value: "agent-create", label: "  \u21B3 Create New Agent" },
+          { value: "agent-delete", label: "  \u21B3 Delete Agent" },
           { 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" },
@@ -12674,33 +12692,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}`));
+          }
+        }
       }
     }
   });
@@ -12718,6 +12729,96 @@ async function handleLogin() {
   saveConfig({ ...config, apiKey });
   (0, import_prompts.intro)((0, import_colors.green)("API Key updated successfully!"));
 }
+async function handleAgentCreate(client) {
+  const name = await (0, import_prompts.text)({
+    message: "Agent Name:",
+    placeholder: "e.g. Support Bot",
+    validate(value) {
+      if (value.length === 0) return "Name is required!";
+    }
+  });
+  if ((0, import_prompts.isCancel)(name)) return;
+  const description = await (0, import_prompts.text)({
+    message: "Description:",
+    placeholder: "e.g. A helpful support assistant"
+  });
+  if ((0, import_prompts.isCancel)(description)) return;
+  const systemPrompt = await (0, import_prompts.text)({
+    message: "System Prompt:",
+    placeholder: "e.g. You are a helpful assistant.",
+    initialValue: "You are a helpful assistant."
+  });
+  if ((0, import_prompts.isCancel)(systemPrompt)) return;
+  const s = (0, import_prompts.spinner)();
+  s.start("Creating agent...");
+  try {
+    const agents = await client.list();
+    let workspaceId = "";
+    if (agents.length > 0) {
+      workspaceId = agents[0].workspace_id;
+    } else {
+      s.stop((0, import_colors.yellow)("Workspace ID needed (no existing agents found)."));
+      const wid = await (0, import_prompts.text)({
+        message: "Enter Workspace ID (UUID):",
+        validate(value) {
+          if (value.length === 0) return "Required";
+        }
+      });
+      if ((0, import_prompts.isCancel)(wid)) return;
+      workspaceId = wid;
+      s.start("Creating agent...");
+    }
+    const agent = await client.create({
+      workspace_id: workspaceId,
+      name,
+      description,
+      config: {
+        system_prompt: systemPrompt,
+        model: "gpt-4o"
+        // Default or prompt? simplified
+      }
+    });
+    s.stop((0, import_colors.green)(`Agent "${agent.name}" created successfully! ID: ${agent.id}`));
+  } catch (e) {
+    s.stop((0, import_colors.red)("Failed to create agent."));
+    throw e;
+  }
+}
+async function handleAgentDelete(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 to delete."));
+    return;
+  }
+  const agentId = await (0, import_prompts.select)({
+    message: "Select an agent to DELETE:",
+    options: agents.map((a) => ({ value: a.id, label: a.name, hint: a.description || "No description" }))
+  });
+  if ((0, import_prompts.isCancel)(agentId)) return;
+  const confirm = await (0, import_prompts.select)({
+    message: `Are you sure you want to delete this agent?`,
+    options: [
+      { value: "yes", label: "Yes, delete it", hint: "Cannot be undone" },
+      { value: "no", label: "No, keep it" }
+    ]
+  });
+  if (confirm !== "yes") {
+    (0, import_prompts.intro)((0, import_colors.gray)("Deletion cancelled."));
+    return;
+  }
+  const d = (0, import_prompts.spinner)();
+  d.start("Deleting agent...");
+  try {
+    await client.delete(agentId);
+    d.stop((0, import_colors.green)("Agent deleted successfully."));
+  } catch (e) {
+    d.stop((0, import_colors.red)("Failed to delete agent."));
+    throw e;
+  }
+}
 async function handleAgentList(client) {
   const s = (0, import_prompts.spinner)();
   s.start("Fetching agents...");