@almyty/chat 0.1.3 → 0.1.4

package/dist/index.d.ts

@@ -1,20 +1,9 @@
 #!/usr/bin/env node
 /**
- * @almyty/chat — interactive REPL with almyty agents.
+ * @almyty/chat — interactive agent chat.
  *
  *   npx @almyty/chat                       # pick from a menu
- *   npx @almyty/chat <agent>               # start chatting with that agent
+ *   npx @almyty/chat <agent>               # start chatting
  *   npx @almyty/chat <agent> --resume <id> # resume a conversation
- *
- * Reads credentials from ~/.almyty/credentials.json (created by
- * `npx @almyty/auth login`).
- *
- * Slash commands inside the REPL:
- *
- *   /switch <agent>   switch to a different agent
- *   /agents           list agents in your organization
- *   /clear            clear the screen
- *   /help             show command list
- *   /quit, /exit      leave the REPL
  */
 export {};

package/dist/index.js

@@ -1,133 +1,171 @@
 #!/usr/bin/env node
 /**
- * @almyty/chat — interactive REPL with almyty agents.
+ * @almyty/chat — interactive agent chat.
  *
  *   npx @almyty/chat                       # pick from a menu
- *   npx @almyty/chat <agent>               # start chatting with that agent
+ *   npx @almyty/chat <agent>               # start chatting
  *   npx @almyty/chat <agent> --resume <id> # resume a conversation
- *
- * Reads credentials from ~/.almyty/credentials.json (created by
- * `npx @almyty/auth login`).
- *
- * Slash commands inside the REPL:
- *
- *   /switch <agent>   switch to a different agent
- *   /agents           list agents in your organization
- *   /clear            clear the screen
- *   /help             show command list
- *   /quit, /exit      leave the REPL
  */
 import { createInterface } from 'readline';
 import { AlmytyClient, resolveCredentialsOrExit } from '@almyty/client';
-const VERSION = '0.1.0';
-function printHelp() {
-    console.log(`
-@almyty/chat v${VERSION}
-
-Usage:
-  npx @almyty/chat                       Pick an agent from a menu
-  npx @almyty/chat <name|id>             Start chatting with that agent
-  npx @almyty/chat <name|id> --resume <conversation-id>
-                                         Resume an existing conversation
-
-Inside the REPL:
-  /switch <agent>   switch to a different agent
-  /agents           list agents in your organization
-  /clear            clear the screen
-  /help             show command list
-  /quit, /exit      leave the REPL
-
-Login:
-  npx @almyty/auth login                 Browser-based login
-`);
+const VERSION = '0.1.3';
+// ── ANSI helpers ────────────────────────────────────────────────
+const DIM = '\x1b[2m';
+const BOLD = '\x1b[1m';
+const RESET = '\x1b[0m';
+const VIOLET = '\x1b[38;5;141m';
+const CYAN = '\x1b[36m';
+const RED = '\x1b[31m';
+const GRAY = '\x1b[90m';
+const CLEAR_LINE = '\x1b[2K\r';
+const isTTY = process.stdout.isTTY ?? false;
+function dim(s) { return isTTY ? `${DIM}${s}${RESET}` : s; }
+function bold(s) { return isTTY ? `${BOLD}${s}${RESET}` : s; }
+function violet(s) { return isTTY ? `${VIOLET}${s}${RESET}` : s; }
+function cyan(s) { return isTTY ? `${CYAN}${s}${RESET}` : s; }
+function red(s) { return isTTY ? `${RED}${s}${RESET}` : s; }
+function gray(s) { return isTTY ? `${GRAY}${s}${RESET}` : s; }
+// ── Spinner ─────────────────────────────────────────────────────
+const SPINNER_FRAMES = ['⠋', '⠙', '⠹', '⠸', '⠼', '⠴', '⠦', '⠧', '⠇', '⠏'];
+class Spinner {
+    interval = null;
+    frame = 0;
+    start(label = 'Thinking') {
+        if (!isTTY)
+            return;
+        this.frame = 0;
+        this.interval = setInterval(() => {
+            const f = SPINNER_FRAMES[this.frame % SPINNER_FRAMES.length];
+            process.stdout.write(`${CLEAR_LINE}${GRAY}${f} ${label}...${RESET}`);
+            this.frame++;
+        }, 80);
+    }
+    stop() {
+        if (this.interval) {
+            clearInterval(this.interval);
+            this.interval = null;
+            process.stdout.write(CLEAR_LINE);
+        }
+    }
 }
-function printAgentBanner(agent) {
+// ── Word wrapping ───────────────────────────────────────────────
+function wordWrap(text, width) {
+    if (width <= 0)
+        return text;
+    const lines = [];
+    for (const paragraph of text.split('\n')) {
+        if (paragraph.length <= width) {
+            lines.push(paragraph);
+            continue;
+        }
+        let line = '';
+        for (const word of paragraph.split(/\s+/)) {
+            if (line.length + word.length + 1 > width && line.length > 0) {
+                lines.push(line);
+                line = word;
+            }
+            else {
+                line = line ? `${line} ${word}` : word;
+            }
+        }
+        if (line)
+            lines.push(line);
+    }
+    return lines.join('\n');
+}
+// ── Display ─────────────────────────────────────────────────────
+function printBanner(agent) {
+    const cols = process.stdout.columns || 80;
+    const line = gray('─'.repeat(Math.min(cols, 60)));
     console.log('');
-    console.log(`  💬 ${agent.name}${agent.mode ? ` [${agent.mode}]` : ''}`);
+    console.log(line);
+    console.log(`  ${violet(bold(agent.name))}  ${dim(agent.mode ?? '')}`);
     if (agent.description)
-        console.log(`     ${agent.description}`);
-    console.log('');
-    console.log('  Type your message, or /help for commands. /quit to exit.');
+        console.log(`  ${dim(agent.description)}`);
+    console.log(line);
+    console.log(dim('  Type a message to chat. /help for commands, /quit to exit.'));
     console.log('');
 }
-function formatStep(step) {
-    if (!step)
-        return null;
-    switch (step.type) {
-        case 'llm_call':
-            if (step.output?.content)
-                return `  ${step.output.content}`;
-            return `  · llm_call (${step.tokens?.input ?? 0}↓ / ${step.tokens?.output ?? 0}↑)`;
-        case 'tool_call':
-            return `  · tool_call → ${step.input?.tool ?? '?'}`;
-        case 'sub_agent_call':
-            return `  · sub_agent_call → ${step.input?.agentId ?? '?'}`;
-        case 'error':
-            return `  ! error: ${step.error}`;
-        default:
-            return null;
+function printResponse(text) {
+    const cols = process.stdout.columns || 80;
+    const maxWidth = Math.min(cols - 2, 78);
+    const wrapped = wordWrap(text.trim(), maxWidth);
+    console.log('');
+    for (const line of wrapped.split('\n')) {
+        console.log(`  ${line}`);
     }
+    console.log('');
+}
+function printToolCall(name) {
+    process.stdout.write(`${CLEAR_LINE}${gray(`  > ${name}`)}\n`);
+}
+function printError(msg) {
+    console.log(`\n  ${red('error')} ${msg}\n`);
 }
+// ── Agent picker ────────────────────────────────────────────────
 async function pickAgent(client) {
     const agents = await client.listAgents();
     if (agents.length === 0) {
-        console.error('No agents in your organization. Create one at https://app.almyty.com/agents');
+        printError('No agents found. Create one at https://app.almyty.com/agents');
         return null;
     }
-    if (agents.length === 1) {
+    if (agents.length === 1)
         return agents[0];
-    }
     console.log('');
-    console.log('  Pick an agent:');
+    console.log(bold('  Select an agent:'));
+    console.log('');
     agents.forEach((a, i) => {
-        const mode = a.mode ? ` [${a.mode}]` : '';
-        console.log(`    ${i + 1}. ${a.name}${mode}`);
+        const mode = a.mode ? dim(` [${a.mode}]`) : '';
+        console.log(`  ${cyan(`${i + 1}.`)} ${a.name}${mode}`);
     });
     console.log('');
     const rl = createInterface({ input: process.stdin, output: process.stdout });
     const answer = await new Promise((resolve) => {
-        rl.question('  Enter number: ', (input) => {
+        rl.question(`  ${cyan('>')} `, (input) => {
             rl.close();
             resolve(input.trim());
         });
     });
     const idx = parseInt(answer, 10) - 1;
     if (Number.isNaN(idx) || idx < 0 || idx >= agents.length) {
-        console.error('Invalid selection.');
+        printError('Invalid selection.');
         return null;
     }
     return agents[idx];
 }
+// ── Slash commands ──────────────────────────────────────────────
 async function handleSlashCommand(command, state) {
     const [cmd, ...rest] = command.slice(1).trim().split(/\s+/);
     switch (cmd) {
         case 'quit':
         case 'exit':
+        case 'q':
             return { exit: true };
         case 'help':
+        case 'h':
             console.log('');
-            console.log('  /switch <agent>   switch to a different agent');
-            console.log('  /agents           list agents in your organization');
-            console.log('  /clear            clear the screen');
-            console.log('  /quit, /exit      leave the REPL');
-            console.log('  /help             this help');
+            console.log(`  ${cyan('/switch')} ${dim('<agent>')}   switch to a different agent`);
+            console.log(`  ${cyan('/agents')}            list available agents`);
+            console.log(`  ${cyan('/clear')}             clear the screen`);
+            console.log(`  ${cyan('/quit')}              exit`);
             console.log('');
             return {};
         case 'clear':
+        case 'c':
             process.stdout.write('\x1b[2J\x1b[H');
-            printAgentBanner(state.agent);
+            printBanner(state.agent);
             return {};
         case 'agents': {
             const agents = await state.client.listAgents();
             if (agents.length === 0) {
-                console.log('  (no agents in this organization)');
+                console.log(dim('  (no agents)'));
                 return {};
             }
             console.log('');
             for (const a of agents) {
-                const marker = a.id === state.agent.id ? '→' : ' ';
-                const mode = a.mode ? ` [${a.mode}]` : '';
-                console.log(`  ${marker} ${a.name}${mode}${a.description ? ' — ' + a.description : ''}`);
+                const marker = a.id === state.agent.id ? violet('>') : ' ';
+                const mode = a.mode ? dim(` [${a.mode}]`) : '';
+                console.log(`  ${marker} ${a.name}${mode}`);
             }
             console.log('');
             return {};
@@ -135,83 +173,106 @@ async function handleSlashCommand(command, state) {
         case 'switch': {
             const target = rest.join(' ').trim();
             if (!target) {
-                console.log('  Usage: /switch <agent name or id>');
+                console.log(dim('  Usage: /switch <agent>'));
                 return {};
             }
             const next = await state.client.findAgentByNameOrId(target);
             if (!next) {
-                console.log(`  Agent "${target}" not found.`);
+                printError(`Agent "${target}" not found.`);
                 return {};
             }
             state.agent = next;
             state.pendingRunId = null;
             state.conversationId = null;
-            printAgentBanner(state.agent);
+            printBanner(state.agent);
             return {};
         }
         default:
-            console.log(`  Unknown command: /${cmd}. Type /help for available commands.`);
+            console.log(dim(`  Unknown command: /${cmd}`));
             return {};
     }
 }
+// ── Turn handling ───────────────────────────────────────────────
 async function handleAutonomousTurn(state, message) {
+    const spinner = new Spinner();
     let runId;
-    if (state.pendingRunId) {
-        await state.client.sendRunInput(state.agent.id, state.pendingRunId, message);
-        runId = state.pendingRunId;
-        state.pendingRunId = null;
-    }
-    else {
-        const run = await state.client.startRun(state.agent.id, message, {
-            conversationId: state.conversationId ?? undefined,
-        });
-        runId = run.id;
-        if (run.conversationId) {
-            state.conversationId = run.conversationId;
+    try {
+        spinner.start();
+        if (state.pendingRunId) {
+            await state.client.sendRunInput(state.agent.id, state.pendingRunId, message);
+            runId = state.pendingRunId;
+            state.pendingRunId = null;
         }
-    }
-    let printed = 0;
-    const final = await state.client.pollRun(state.agent.id, runId, {
-        onStep: (run) => {
-            const newSteps = (run.steps ?? []).slice(printed);
-            for (const step of newSteps) {
-                const line = formatStep(step);
-                if (line)
-                    console.log(line);
+        else {
+            const run = await state.client.startRun(state.agent.id, message, {
+                conversationId: state.conversationId ?? undefined,
+            });
+            runId = run.id;
+            if (run.conversationId) {
+                state.conversationId = run.conversationId;
             }
-            printed = run.steps?.length ?? printed;
-        },
-    });
-    if (final.status === 'waiting_input') {
-        state.pendingRunId = runId;
-        console.log('');
-        console.log('  (agent is waiting for your input — answer below)');
-        console.log('');
-        return;
-    }
-    if (final.status === 'completed') {
-        if (final.output != null) {
-            console.log('');
-            console.log(typeof final.output === 'string' ? `  ${final.output}` : JSON.stringify(final.output, null, 2));
         }
-        return;
+        // Poll for completion — show tool calls as they happen
+        const seenSteps = new Set();
+        const final = await state.client.pollRun(state.agent.id, runId, {
+            intervalMs: 800,
+            onStep: (run) => {
+                const steps = run.steps ?? [];
+                for (let i = 0; i < steps.length; i++) {
+                    if (seenSteps.has(i))
+                        continue;
+                    seenSteps.add(i);
+                    const step = steps[i];
+                    if (step?.type === 'tool_call' && step?.input?.tool) {
+                        spinner.stop();
+                        printToolCall(step.input.tool);
+                        spinner.start();
+                    }
+                }
+            },
+        });
+        spinner.stop();
+        if (final.status === 'waiting_input') {
+            state.pendingRunId = runId;
+            console.log(dim('\n  (waiting for your input)\n'));
+            return;
+        }
+        if (final.status === 'completed' && final.output != null) {
+            const text = typeof final.output === 'string' ? final.output : JSON.stringify(final.output, null, 2);
+            printResponse(text);
+            return;
+        }
+        if (final.status === 'failed') {
+            printError(final.error || 'Run failed');
+            return;
+        }
+        if (final.status === 'cancelled') {
+            console.log(dim('\n  (cancelled)\n'));
+            return;
+        }
+    }
+    catch (err) {
+        spinner.stop();
+        throw err;
     }
-    console.log('');
-    console.log(`  Run ${final.status}${final.error ? `: ${final.error}` : ''}`);
 }
 async function handleWorkflowTurn(state, message) {
-    const result = await state.client.invokeAgent(state.agent.id, { message });
-    const output = result?.output ?? result?.data?.output ?? result;
-    if (output == null) {
-        console.log('  (no output)');
-        return;
-    }
-    console.log('');
-    if (typeof output === 'string') {
-        console.log(`  ${output}`);
+    const spinner = new Spinner();
+    try {
+        spinner.start();
+        const result = await state.client.invokeAgent(state.agent.id, { message });
+        spinner.stop();
+        const output = result?.output ?? result?.data?.output ?? result;
+        if (output == null) {
+            console.log(dim('\n  (no output)\n'));
+            return;
+        }
+        const text = typeof output === 'string' ? output : JSON.stringify(output, null, 2);
+        printResponse(text);
     }
-    else {
-        console.log(JSON.stringify(output, null, 2).split('\n').map((l) => '  ' + l).join('\n'));
+    catch (err) {
+        spinner.stop();
+        throw err;
     }
 }
 async function handleUserTurn(state, message) {
@@ -220,16 +281,18 @@ async function handleUserTurn(state, message) {
     }
     return handleWorkflowTurn(state, message);
 }
+// ── Main loop ───────────────────────────────────────────────────
 async function startChat(client, agent, resumeConversationId) {
     const state = { client, agent, pendingRunId: null, conversationId: resumeConversationId ?? null };
     if (resumeConversationId) {
-        console.log(`  Resuming conversation ${resumeConversationId}`);
+        console.log(dim(`  Resuming conversation ${resumeConversationId}`));
     }
-    printAgentBanner(state.agent);
+    printBanner(state.agent);
+    const prompt = `${VIOLET}>${RESET} `;
     const rl = createInterface({
         input: process.stdin,
         output: process.stdout,
-        prompt: '> ',
+        prompt: isTTY ? prompt : '> ',
     });
     rl.prompt();
     rl.on('line', async (raw) => {
@@ -247,7 +310,7 @@ async function startChat(client, agent, resumeConversationId) {
                 }
             }
             catch (err) {
-                console.error(`  Error: ${err.message}`);
+                printError(err.message);
             }
             rl.prompt();
             return;
@@ -256,17 +319,16 @@ async function startChat(client, agent, resumeConversationId) {
             await handleUserTurn(state, line);
         }
         catch (err) {
-            console.error(`  Error: ${err.message}`);
+            printError(err.message);
         }
-        console.log('');
         rl.prompt();
     });
     rl.on('close', () => {
         console.log('');
-        console.log('Chat ended.');
         process.exit(0);
     });
 }
+// ── Entry point ─────────────────────────────────────────────────
 async function main() {
     const argv = process.argv.slice(2);
     if (argv.includes('--version') || argv.includes('-v')) {
@@ -274,18 +336,33 @@ async function main() {
         return;
     }
     if (argv.includes('--help') || argv.includes('-h')) {
-        printHelp();
+        console.log(`
+${bold('almyty chat')} ${dim(`v${VERSION}`)}
+
+${bold('Usage:')}
+  npx @almyty/chat                                 Pick an agent
+  npx @almyty/chat ${dim('<agent>')}                          Start chatting
+  npx @almyty/chat ${dim('<agent>')} --resume ${dim('<conv-id>')}    Resume conversation
+
+${bold('Commands:')}
+  /switch ${dim('<agent>')}   switch agent
+  /agents          list agents
+  /clear           clear screen
+  /quit            exit
+
+${bold('Auth:')}
+  npx @almyty/auth login
+`);
         return;
     }
     const creds = resolveCredentialsOrExit();
     const client = new AlmytyClient(creds.url, creds.token);
-    // Parse --resume <conversation-id>
     let resumeConversationId;
     const resumeIdx = argv.indexOf('--resume');
     if (resumeIdx !== -1) {
         const nextArg = argv[resumeIdx + 1];
         if (!nextArg || nextArg.startsWith('-')) {
-            console.error('Usage: --resume <conversation-id>');
+            printError('Usage: --resume <conversation-id>');
             process.exit(1);
         }
         resumeConversationId = nextArg;
@@ -293,9 +370,12 @@ async function main() {
     let agent = null;
     const ref = argv.find((a) => !a.startsWith('-') && a !== resumeConversationId);
     if (ref) {
+        const spinner = new Spinner();
+        spinner.start('Connecting');
         agent = await client.findAgentByNameOrId(ref);
+        spinner.stop();
         if (!agent) {
-            console.error(`Agent not found: ${ref}`);
+            printError(`Agent not found: ${ref}`);
             process.exit(1);
         }
     }
@@ -307,6 +387,6 @@ async function main() {
     await startChat(client, agent, resumeConversationId);
 }
 main().catch((err) => {
-    console.error(`Error: ${err.message}`);
+    printError(err.message);
     process.exit(1);
 });

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@almyty/chat",
-  "version": "0.1.3",
+  "version": "0.1.4",
   "description": "Interactive chat REPL with almyty agents",
   "type": "module",
   "main": "dist/index.js",