@alejandroroman/agent-kit 0.2.0 → 0.2.1

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,15 +1,37 @@
 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";
+    const prevLogLevel = process.env.LOG_LEVEL ?? "info";
+    setLogLevel("silent");
+    try {
+        return await _runSetupAgent(opts, model);
+    }
+    finally {
+        setLogLevel(prevLogLevel);
+    }
+}
+async function _runSetupAgent(opts, model) {
     return new Promise((resolve) => {
         let finished = false;
+        const spinner = p.spinner();
         const toolRegistry = new ToolRegistry();
         const tools = createSetupTools({
             configPath: opts.configPath,
@@ -27,19 +49,52 @@ export async function runSetupAgent(opts) {
         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...");
+            }
+            else if (phase === "tool_use") {
+                spinner.message(TOOL_LABELS[detail ?? ""] ?? "Working...");
+            }
+            // "complete" phase — spinner will be stopped by the caller
+        };
+        const runWithSpinner = async (loopMessages, maxIterations) => {
+            spinner.start("Thinking...");
+            try {
+                const result = await runAgentLoop(loopMessages, {
+                    model,
+                    systemPrompt,
+                    toolRegistry,
+                    maxIterations,
+                    source: "cli",
+                    onProgress,
+                });
+                spinner.stop("");
+                return result;
+            }
+            catch (err) {
+                spinner.stop("Error");
+                throw err;
+            }
+        };
         const rl = readline.createInterface({
             input: process.stdin,
             output: process.stdout,
         });
+        const cleanup = async () => {
+            rl.close();
+            process.stdin.pause();
+            await new Promise((r) => setTimeout(r, 200));
+        };
         const ask = (prompt) => {
             if (finished) {
-                rl.close();
+                cleanup();
                 return;
             }
             rl.question(prompt ?? "You: ", async (input) => {
                 const trimmed = input.trim();
                 if (trimmed === "/quit" || trimmed === "/exit") {
-                    rl.close();
+                    await cleanup();
                     resolve("Setup cancelled.");
                     return;
                 }
@@ -49,13 +104,7 @@ export async function runSetupAgent(opts) {
                 }
                 messages.push({ role: "user", content: trimmed });
                 try {
-                    const result = await runAgentLoop(messages, {
-                        model,
-                        systemPrompt,
-                        toolRegistry,
-                        maxIterations: 30,
-                        source: "cli",
-                    });
+                    const result = await runWithSpinner(messages, 30);
                     messages.push(...result.messages.slice(messages.length));
                     if (!finished) {
                         console.log(`\n${result.text}\n`);
@@ -68,7 +117,7 @@ export async function runSetupAgent(opts) {
                     ask();
                 }
                 else {
-                    rl.close();
+                    await cleanup();
                 }
             });
         };
@@ -79,13 +128,7 @@ export async function runSetupAgent(opts) {
                     ? "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",
-                });
+                const result = await runWithSpinner(messages, 10);
                 messages.push(...result.messages.slice(messages.length));
                 console.log(`\n${result.text}\n`);
             }

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.1",
   "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"
   }
-}
+}