@alejandroroman/agent-kit 0.2.0 → 0.2.2

package/dist/agent/loop.js

@@ -109,6 +109,7 @@ async function _runAgentLoop(initialMessages, config, runSpan, tracer, runCtx) {
             // Prune images from old messages to save context tokens
             const prunedMessages = pruneImages(messages, 4);
             // LLM call span
+            config.onProgress?.("thinking");
             const llmSpan = tracer.startSpan("llm.call", undefined, iterCtx);
             let response;
             try {
@@ -194,6 +195,7 @@ async function _runAgentLoop(initialMessages, config, runSpan, tracer, runCtx) {
                         log.error({ err, runId }, "failed to record agent run");
                     }
                 }
+                config.onProgress?.("complete");
                 log.info({ label, stopReason: response.stopReason, iterations: iteration, inputTokens: totalUsage.inputTokens, outputTokens: totalUsage.outputTokens, elapsed }, "loop end");
                 runSpan.setAttribute(ATTR.ITERATION, iteration);
                 return {
@@ -219,6 +221,7 @@ async function _runAgentLoop(initialMessages, config, runSpan, tracer, runCtx) {
                     else {
                         const toolStart = Date.now();
                         try {
+                            config.onProgress?.("tool_use", call.name);
                             result = await tool.execute(call.arguments);
                             const toolElapsed = Date.now() - toolStart;
                             log.debug({ label, iteration, tool: call.name, resultLength: result.length, snippet: truncate(result, 150), elapsed: toolElapsed }, "tool result");

package/dist/agent/types.d.ts

@@ -2,6 +2,7 @@ import type { Tool } from "../tools/types.js";
 import type { ToolRegistry } from "../tools/registry.js";
 import type { TokenUsage } from "../llm/types.js";
 import type { UsageStore } from "../usage/store.js";
+export type ProgressPhase = "thinking" | "tool_use" | "complete";
 export interface AgentConfig {
     model: string;
     fallbacks?: string[];
@@ -15,6 +16,7 @@ export interface AgentConfig {
     source?: string;
     maxToolResultSize?: number;
     compactionThreshold?: number;
+    onProgress?: (phase: ProgressPhase, detail?: string) => void;
 }
 export interface AgentResult {
     text: string;

package/dist/cli/setup-agent/index.js

@@ -1,100 +1,120 @@
-import * as readline from "readline";
+import * as p from "@clack/prompts";
 import { runAgentLoop } from "../../agent/loop.js";
 import { ToolRegistry } from "../../tools/registry.js";
 import { createSetupTools } from "./tools.js";
 import { SETUP_SOUL, CREATE_SOUL } from "./soul.js";
 import { dateContext } from "../../text.js";
+import { setLogLevel } from "../../logger.js";
+const TOOL_LABELS = {
+    check_environment: "Checking environment...",
+    list_builtin_tools: "Looking at available tools...",
+    list_available_skills: "Scanning skills...",
+    save_agent: "Saving agent...",
+    create_skill: "Creating skill...",
+    add_cron_job: "Adding cron job...",
+    read_file: "Reading file...",
+    finish_setup: "Finishing up...",
+};
 export async function runSetupAgent(opts) {
-    // Ensure the API key is available for the Anthropic provider
     process.env.ANTHROPIC_API_KEY = opts.apiKey;
     const model = "anthropic:claude-sonnet-4-6";
-    return new Promise((resolve) => {
-        let finished = false;
-        const toolRegistry = new ToolRegistry();
-        const tools = createSetupTools({
-            configPath: opts.configPath,
-            dataDir: opts.dataDir,
-            skillsDir: opts.skillsDir,
-            basePath: opts.basePath,
-            onFinish: (summary) => {
-                finished = true;
-                resolve(summary);
-            },
-        });
-        for (const tool of tools) {
-            toolRegistry.register(tool);
+    const prevLogLevel = process.env.LOG_LEVEL ?? "info";
+    setLogLevel("silent");
+    try {
+        return await _runSetupAgent(opts, model);
+    }
+    finally {
+        setLogLevel(prevLogLevel);
+    }
+}
+async function _runSetupAgent(opts, model) {
+    let finished = false;
+    let summary = "Setup cancelled.";
+    const spinner = p.spinner();
+    const toolRegistry = new ToolRegistry();
+    const tools = createSetupTools({
+        configPath: opts.configPath,
+        dataDir: opts.dataDir,
+        skillsDir: opts.skillsDir,
+        basePath: opts.basePath,
+        onFinish: (s) => {
+            finished = true;
+            summary = s;
+        },
+    });
+    for (const tool of tools) {
+        toolRegistry.register(tool);
+    }
+    const soul = opts.mode === "init" ? SETUP_SOUL : CREATE_SOUL;
+    const systemPrompt = [soul, dateContext()].join("\n\n");
+    const messages = [];
+    const onProgress = (phase, detail) => {
+        if (phase === "thinking") {
+            spinner.message("Thinking...");
         }
-        const soul = opts.mode === "init" ? SETUP_SOUL : CREATE_SOUL;
-        const systemPrompt = [soul, dateContext()].join("\n\n");
-        const messages = [];
-        const rl = readline.createInterface({
-            input: process.stdin,
-            output: process.stdout,
-        });
-        const ask = (prompt) => {
-            if (finished) {
-                rl.close();
-                return;
-            }
-            rl.question(prompt ?? "You: ", async (input) => {
-                const trimmed = input.trim();
-                if (trimmed === "/quit" || trimmed === "/exit") {
-                    rl.close();
-                    resolve("Setup cancelled.");
-                    return;
-                }
-                if (!trimmed) {
-                    ask();
-                    return;
-                }
-                messages.push({ role: "user", content: trimmed });
-                try {
-                    const result = await runAgentLoop(messages, {
-                        model,
-                        systemPrompt,
-                        toolRegistry,
-                        maxIterations: 30,
-                        source: "cli",
-                    });
-                    messages.push(...result.messages.slice(messages.length));
-                    if (!finished) {
-                        console.log(`\n${result.text}\n`);
-                    }
-                }
-                catch (err) {
-                    console.error(`\n  Error: ${err instanceof Error ? err.message : err}\n`);
-                }
-                if (!finished) {
-                    ask();
-                }
-                else {
-                    rl.close();
-                }
+        else if (phase === "tool_use") {
+            spinner.message(TOOL_LABELS[detail ?? ""] ?? "Working...");
+        }
+    };
+    const runWithSpinner = async (loopMessages, maxIterations) => {
+        spinner.start("Thinking...");
+        try {
+            const result = await runAgentLoop(loopMessages, {
+                model,
+                systemPrompt,
+                toolRegistry,
+                maxIterations,
+                source: "cli",
+                onProgress,
             });
-        };
-        // Kick off with an initial greeting from the agent
-        (async () => {
-            try {
-                const greeting = opts.mode === "init"
-                    ? "The user just ran `agent-kit init`. Greet them, then call check_environment to see what's available. After that, ask what kind of agent they'd like to create first."
-                    : "The user wants to add a new agent to their existing setup. Greet them, call check_environment and list_available_skills to see the current state, then ask what kind of agent they'd like to create.";
-                messages.push({ role: "user", content: greeting });
-                const result = await runAgentLoop(messages, {
-                    model,
-                    systemPrompt,
-                    toolRegistry,
-                    maxIterations: 10,
-                    source: "cli",
-                });
-                messages.push(...result.messages.slice(messages.length));
-                console.log(`\n${result.text}\n`);
-            }
-            catch (err) {
-                console.error(`\n  Error starting setup agent: ${err instanceof Error ? err.message : err}\n`);
-            }
+            spinner.stop("");
+            return result;
+        }
+        catch (err) {
+            spinner.stop("Error");
+            throw err;
+        }
+    };
+    // Greeting
+    try {
+        const greeting = opts.mode === "init"
+            ? "The user just ran `agent-kit init`. Greet them, then call check_environment to see what's available. After that, ask what kind of agent they'd like to create first."
+            : "The user wants to add a new agent to their existing setup. Greet them, call check_environment and list_available_skills to see the current state, then ask what kind of agent they'd like to create.";
+        messages.push({ role: "user", content: greeting });
+        const result = await runWithSpinner(messages, 10);
+        messages.push(...result.messages.slice(messages.length));
+        console.log(`\n${result.text}\n`);
+    }
+    catch (err) {
+        console.error(`\n  Error starting setup agent: ${err instanceof Error ? err.message : err}\n`);
+        return summary;
+    }
+    // Conversation loop using clack prompts (no readline — avoids stdin conflict with spinner)
+    while (!finished) {
+        const input = await p.text({
+            message: "You",
+            placeholder: "Type your response... (/quit to exit)",
+        });
+        if (p.isCancel(input)) {
+            return "Setup cancelled.";
+        }
+        const trimmed = input.trim();
+        if (trimmed === "/quit" || trimmed === "/exit") {
+            return "Setup cancelled.";
+        }
+        if (!trimmed)
+            continue;
+        messages.push({ role: "user", content: trimmed });
+        try {
+            const result = await runWithSpinner(messages, 30);
+            messages.push(...result.messages.slice(messages.length));
             if (!finished) {
-                ask();
+                console.log(`\n${result.text}\n`);
             }
-        })();
-    });
+        }
+        catch (err) {
+            console.error(`\n  Error: ${err instanceof Error ? err.message : err}\n`);
+        }
+    }
+    return summary;
 }

package/dist/logger.d.ts

@@ -1,2 +1,3 @@
 import pino from "pino";
 export declare function createLogger(name: string): pino.Logger<never, boolean>;
+export declare function setLogLevel(level: pino.Level | "silent"): void;

package/dist/logger.js

@@ -21,3 +21,6 @@ const root = pino({
 export function createLogger(name) {
     return root.child({ component: name });
 }
+export function setLogLevel(level) {
+    root.level = level;
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@alejandroroman/agent-kit",
-  "version": "0.2.0",
+  "version": "0.2.2",
   "type": "module",
   "main": "./dist/index.js",
   "exports": {
@@ -12,6 +12,19 @@
   "files": [
     "dist/"
   ],
+  "scripts": {
+    "build": "pnpm --filter @agent-kit/memory run build && tsc && cp -r packages/memory/dist dist/_memory && node -e \"const f='dist/tools/builtin/memory.js',s=require('fs');s.writeFileSync(f,s.readFileSync(f,'utf8').replaceAll('from \\\"@agent-kit/memory\\\"','from \\\"../../_memory/index.js\\\"'))\"",
+    "prepublishOnly": "pnpm run build",
+    "cli": "tsx src/cli.ts",
+    "dev": "doppler run -- tsx src/index.ts",
+    "dev:local": "tsx src/index.ts",
+    "memory": "doppler run -- pnpm --filter @agent-kit/memory start",
+    "ledger": "doppler run -- pnpm --filter @agent-kit/ledger ledger",
+    "dev:dashboard": "pnpm --filter @agent-kit/dashboard dev",
+    "agent:validate": "tsx src/scripts/validate-agent-cli.ts",
+    "test": "vitest run",
+    "test:watch": "vitest"
+  },
   "dependencies": {
     "@anthropic-ai/sdk": "^0.78.0",
     "@clack/prompts": "^1.0.1",
@@ -35,25 +48,13 @@
     "zod": "^4.3.6"
   },
   "devDependencies": {
+    "@agent-kit/memory": "workspace:*",
     "@types/better-sqlite3": "^7.6.13",
     "@types/node": "^25.3.0",
     "@types/node-cron": "^3.0.11",
     "@types/sharp": "^0.32.0",
     "tsx": "^4.21.0",
     "typescript": "^5.9.3",
-    "vitest": "^4.0.18",
-    "@agent-kit/memory": "0.1.0"
-  },
-  "scripts": {
-    "build": "pnpm --filter @agent-kit/memory run build && tsc && cp -r packages/memory/dist dist/_memory && node -e \"const f='dist/tools/builtin/memory.js',s=require('fs');s.writeFileSync(f,s.readFileSync(f,'utf8').replaceAll('from \\\"@agent-kit/memory\\\"','from \\\"../../_memory/index.js\\\"'))\"",
-    "cli": "tsx src/cli.ts",
-    "dev": "doppler run -- tsx src/index.ts",
-    "dev:local": "tsx src/index.ts",
-    "memory": "doppler run -- pnpm --filter @agent-kit/memory start",
-    "ledger": "doppler run -- pnpm --filter @agent-kit/ledger ledger",
-    "dev:dashboard": "pnpm --filter @agent-kit/dashboard dev",
-    "agent:validate": "tsx src/scripts/validate-agent-cli.ts",
-    "test": "vitest run",
-    "test:watch": "vitest"
+    "vitest": "^4.0.18"
   }
-}
+}