langtrain 0.1.10 → 0.1.11

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "langtrain",
-  "version": "0.1.10",
+  "version": "0.1.11",
   "description": "Unified JavaScript SDK for Langtrain Ecosystem",
   "main": "./dist/index.js",
   "module": "./dist/index.mjs",

package/src/agent.ts

@@ -43,6 +43,20 @@ export class AgentClient {
         return response.data.agents;
     }
 
+    async get(agentId: string): Promise<Agent> {
+        const response = await this.client.get<Agent>(`/agents/${agentId}`);
+        return response.data;
+    }
+
+    async create(agent: AgentCreate): Promise<Agent> {
+        const response = await this.client.post<Agent>('/agents/', agent);
+        return response.data;
+    }
+
+    async delete(agentId: string): Promise<void> {
+        await this.client.delete(`/agents/${agentId}`);
+    }
+
     async execute(agentId: string, input: any, messages: any[] = [], conversationId?: string): Promise<AgentRun> {
         const response = await this.client.post<AgentRun>(`/agents/${agentId}/execute`, {
             input,
@@ -52,3 +66,19 @@ export class AgentClient {
         return response.data;
     }
 }
+
+export interface AgentConfig {
+    system_prompt?: string;
+    temperature?: number;
+    max_tokens?: number;
+    tools?: string[];
+    [key: string]: any;
+}
+
+export interface AgentCreate {
+    workspace_id: string; // UUID
+    name: string;
+    description?: string;
+    model_id?: string; // UUID
+    config?: AgentConfig;
+}

package/src/cli.ts

@@ -2,7 +2,7 @@
 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';
-import { Langvision, Langtune, AgentClient, Agent } from './index';
+import { Langvision, Langtune, AgentClient, Agent, AgentCreate } from './index';
 import fs from 'fs';
 import path from 'path';
 import os from 'os';
@@ -69,6 +69,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); },
         };
 
@@ -78,6 +80,8 @@ async function main() {
                 options: [
                     { value: 'group-agents', label: '🤖 Agents (Server)', hint: 'Chat with custom agents' },
                     { value: 'agent-list', label: '  ↳ List & Run Agents' },
+                    { value: 'agent-create', label: '  ↳ Create New Agent' },
+                    { value: 'agent-delete', label: '  ↳ Delete Agent' },
 
                     { value: 'group-tune', label: '🧠 Langtune (LLM)', hint: 'Fine-tuning & Text Generation' },
                     { value: 'tune-finetune', label: '  ↳ Fine-tune Text Model' },
@@ -140,6 +144,130 @@ async function handleLogin() {
     intro(green('API Key updated successfully!'));
 }
 
+async function handleAgentCreate(client: AgentClient) {
+    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 {
+        // We need a workspace ID. server usually infers it from API key context if not provided?
+        // But the schema says workspace_id is required in AgentCreate.
+        // The server implementation of create_agent takes AgentCreate which has workspace_id.
+        // However, standard users might not know their exact workspace UUID. 
+        // We might need to fetch it or rely on server to fill it if we made it optional in schema (which we didn't).
+        // EDIT: Let's fetch one agent to get the workspace_id or assume one?
+        // Better: List agents, get workspace_id from first one. Hacky but works for single-workspace users.
+        // Use 'default' or similar if server supports it?
+        // Checking agents.py: verify_api_key returns workspace_id.
+        // But create_agent payload requires it.
+        // I'll try to fetch list first to get workspace ID. If list empty, we are stuck?
+        // Wait, list_agents returns `AgentListResponse` which doesn't explicitly return workspace_id at top level, but agents have it.
+        // If no agents, we can't guess it.
+        // Maybe I should fetch user profile? No endpoint for that in CLI yet.
+        // I'll try to pass a placeholder and hope server ignores it if it uses context?
+        // Server code: `workspace_id=agent_in.workspace_id`. It uses payload.
+        // I might need to ask user for workspace ID or update server to be smarter.
+        // For now, I'll attempt to LIST agents to get a workspace ID.
+
+        const agents = await client.list();
+        let workspaceId = "";
+        if (agents.length > 0) {
+            workspaceId = agents[0].workspace_id;
+        } else {
+            // Fallback: Ask user or fail?
+            // Or maybe decoding JWT/API key client side? No.
+            // I'll prompt for it if not found, with a hint.
+            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 as string;
+            s.start('Creating agent...');
+        }
+
+        const agent = await client.create({
+            workspace_id: workspaceId,
+            name: name as string,
+            description: description as string,
+            config: {
+                system_prompt: systemPrompt as string,
+                model: 'gpt-4o' // Default or prompt? simplified
+            }
+        });
+        s.stop(green(`Agent "${agent.name}" created successfully! ID: ${agent.id}`));
+    } catch (e: any) {
+        s.stop(red('Failed to create agent.'));
+        throw e;
+    }
+}
+
+async function handleAgentDelete(client: AgentClient) {
+    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 as string);
+        d.stop(green('Agent deleted successfully.'));
+    } catch (e: any) {
+        d.stop(red('Failed to delete agent.'));
+        throw e;
+    }
+}
+
 async function handleAgentList(client: AgentClient) {
     const s = spinner();
     s.start('Fetching agents...');

package/src/index.ts

@@ -4,7 +4,7 @@ export { Langvision } from 'langvision';
 export { Langtune } from 'langtune';
 
 // Export Agent Client
-export { AgentClient, Agent, AgentRun } from './agent';
+export { AgentClient, Agent, AgentRun, AgentCreate } from './agent';
 
 // Export Types with Namespaces to avoid collisions
 import * as Vision from 'langvision';

package/dist/cli.d.mts

@@ -1 +0,0 @@
-#!/usr/bin/env node

package/dist/cli.d.ts

@@ -1 +0,0 @@
-#!/usr/bin/env node

package/dist/index.d.mts

@@ -1,47 +0,0 @@
-import * as langvision from 'langvision';
-export { langvision as Vision };
-export { Langvision } from 'langvision';
-import * as langtune from 'langtune';
-export { langtune as Text };
-export { Langtune } from 'langtune';
-
-interface Agent {
-    id: string;
-    workspace_id: string;
-    name: string;
-    description?: string;
-    model_id?: string;
-    config: any;
-    is_active: boolean;
-    created_at: string;
-    updated_at: string;
-}
-interface AgentRun {
-    id: string;
-    conversation_id: string;
-    success: boolean;
-    output?: any;
-    error?: string;
-    latency_ms: number;
-    tokens_used: number;
-}
-declare class AgentClient {
-    private config;
-    private client;
-    constructor(config: {
-        apiKey: string;
-        baseUrl?: string;
-    });
-    list(workspaceId?: string): Promise<Agent[]>;
-    execute(agentId: string, input: any, messages?: any[], conversationId?: string): Promise<AgentRun>;
-}
-
-type agent_Agent = Agent;
-type agent_AgentClient = AgentClient;
-declare const agent_AgentClient: typeof AgentClient;
-type agent_AgentRun = AgentRun;
-declare namespace agent {
-  export { type agent_Agent as Agent, agent_AgentClient as AgentClient, type agent_AgentRun as AgentRun };
-}
-
-export { type Agent, AgentClient, type AgentRun, agent as AgentTypes };

package/dist/index.d.ts

@@ -1,47 +0,0 @@
-import * as langvision from 'langvision';
-export { langvision as Vision };
-export { Langvision } from 'langvision';
-import * as langtune from 'langtune';
-export { langtune as Text };
-export { Langtune } from 'langtune';
-
-interface Agent {
-    id: string;
-    workspace_id: string;
-    name: string;
-    description?: string;
-    model_id?: string;
-    config: any;
-    is_active: boolean;
-    created_at: string;
-    updated_at: string;
-}
-interface AgentRun {
-    id: string;
-    conversation_id: string;
-    success: boolean;
-    output?: any;
-    error?: string;
-    latency_ms: number;
-    tokens_used: number;
-}
-declare class AgentClient {
-    private config;
-    private client;
-    constructor(config: {
-        apiKey: string;
-        baseUrl?: string;
-    });
-    list(workspaceId?: string): Promise<Agent[]>;
-    execute(agentId: string, input: any, messages?: any[], conversationId?: string): Promise<AgentRun>;
-}
-
-type agent_Agent = Agent;
-type agent_AgentClient = AgentClient;
-declare const agent_AgentClient: typeof AgentClient;
-type agent_AgentRun = AgentRun;
-declare namespace agent {
-  export { type agent_Agent as Agent, agent_AgentClient as AgentClient, type agent_AgentRun as AgentRun };
-}
-
-export { type Agent, AgentClient, type AgentRun, agent as AgentTypes };