langtrain 0.1.10 → 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,
@@ -12660,6 +12671,8 @@ async function main() {
       "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);
@@ -12671,6 +12684,8 @@ async function main() {
         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" },
@@ -12719,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...");