langtrain 0.1.9 → 0.1.11

package/dist/cli.mjs

@@ -12600,6 +12600,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,
@@ -12631,9 +12642,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,29 +12658,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(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),
+      "agent-create": (c) => handleAgentCreate(c.agent),
+      "agent-delete": (c) => handleAgentDelete(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:",
         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" },
@@ -12679,33 +12697,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}`));
+          }
+        }
       }
     }
   });
@@ -12723,6 +12734,96 @@ async function handleLogin() {
   saveConfig({ ...config, apiKey });
   intro(green("API Key updated successfully!"));
 }
+async function handleAgentCreate(client) {
+  const name = await text({
+    message: "Agent Name:",
+    placeholder: "e.g. Support Bot",
+    validate(value) {
+      if (value.length === 0) return "Name is required!";
+    }
+  });
+  if (isCancel(name)) return;
+  const description = await text({
+    message: "Description:",
+    placeholder: "e.g. A helpful support assistant"
+  });
+  if (isCancel(description)) return;
+  const systemPrompt = await text({
+    message: "System Prompt:",
+    placeholder: "e.g. You are a helpful assistant.",
+    initialValue: "You are a helpful assistant."
+  });
+  if (isCancel(systemPrompt)) return;
+  const s = 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(yellow("Workspace ID needed (no existing agents found)."));
+      const wid = await text({
+        message: "Enter Workspace ID (UUID):",
+        validate(value) {
+          if (value.length === 0) return "Required";
+        }
+      });
+      if (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(green(`Agent "${agent.name}" created successfully! ID: ${agent.id}`));
+  } catch (e) {
+    s.stop(red("Failed to create agent."));
+    throw e;
+  }
+}
+async function handleAgentDelete(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 to delete."));
+    return;
+  }
+  const agentId = await select({
+    message: "Select an agent to DELETE:",
+    options: agents.map((a) => ({ value: a.id, label: a.name, hint: a.description || "No description" }))
+  });
+  if (isCancel(agentId)) return;
+  const confirm = await 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") {
+    intro(gray("Deletion cancelled."));
+    return;
+  }
+  const d = spinner();
+  d.start("Deleting agent...");
+  try {
+    await client.delete(agentId);
+    d.stop(green("Agent deleted successfully."));
+  } catch (e) {
+    d.stop(red("Failed to delete agent."));
+    throw e;
+  }
+}
 async function handleAgentList(client) {
   const s = spinner();
   s.start("Fetching agents...");