@aigne/core 1.1.0-beta.2 → 1.1.0-beta.4

package/CHANGELOG.md

@@ -0,0 +1,17 @@
+## 1.1.0-beta.4 (2025-3-12)
+
+- feat: support run puppeteer example chat loop in terminal
+
+## 1.1.0-beta.3 (2025-3-11)
+
+- chore: set module type for core package
+
+## 1.1.0-beta.2 (2025-3-11)
+
+- feat: use PromptBuilder instead of string instructions
+- refactor: use tools instead of skills
+- chore(examples): add puppeteer-mcp-server example
+
+## 1.1.0-beta.1 (2025-3-11)
+
+- feat: add Agent FunctionAgent AIAgent MCPAgent and ExecutionEngine

package/lib/cjs/agents/agent.d.ts

@@ -19,6 +19,7 @@ export interface AgentOptions<I extends AgentInput = AgentInput, O extends Agent
     }>;
     includeInputInOutput?: boolean;
     tools?: (Agent | FunctionAgentFn)[];
+    disableLogging?: boolean;
 }
 export declare class Agent<I extends AgentInput = AgentInput, O extends AgentOutput = AgentOutput> extends EventEmitter {
     static from<I extends AgentInput = AgentInput, O extends AgentOutput = AgentOutput>(options: AgentOptions<I, O>): Agent<I, O>;
@@ -37,6 +38,7 @@ export declare class Agent<I extends AgentInput = AgentInput, O extends AgentOut
     readonly tools: Agent<AgentInput, AgentOutput>[] & {
         [key: string]: Agent<AgentInput, AgentOutput>;
     };
+    private disableLogging?;
     addTool<I extends AgentInput, O extends AgentOutput>(tool: Agent<I, O> | FunctionAgentFn<I, O>): void;
     get isCallable(): boolean;
     call(input: I | string, context?: Context): Promise<O>;

package/lib/cjs/agents/agent.js

@@ -26,6 +26,7 @@ class Agent extends node_events_1.default {
         this.publishTopic = options.publishTopic;
         if (options.tools?.length)
             this.tools.push(...options.tools.map(functionToAgent));
+        this.disableLogging = options.disableLogging;
     }
     name;
     description;
@@ -37,6 +38,7 @@ class Agent extends node_events_1.default {
     tools = new Proxy([], {
         get: (t, p, r) => Reflect.get(t, p, r) ?? t.find((t) => t.name === p),
     });
+    disableLogging;
     addTool(tool) {
         this.tools.push(typeof tool === "function" ? functionToAgent(tool) : tool);
     }
@@ -48,19 +50,11 @@ class Agent extends node_events_1.default {
             throw new Error("Agent must implement process method");
         const _input = typeof input === "string" ? (0, prompt_builder_1.userInput)(input) : input;
         const parsedInput = this.inputSchema.passthrough().parse(_input);
-        logger_1.logger.debug(`Call agent ${this.name} start`, parsedInput);
-        const output = await this.process(parsedInput, context);
-        const parsedOutput = this.outputSchema.passthrough().parse(output);
-        const finalOutput = this.includeInputInOutput
-            ? { ...parsedInput, ...parsedOutput }
-            : parsedOutput;
-        logger_1.logger.debug(`Call agent ${this.name} successfully`, {
-            ...finalOutput,
-            ...(finalOutput[types_1.transferAgentOutputKey]
-                ? { [types_1.transferAgentOutputKey]: finalOutput[types_1.transferAgentOutputKey].agent.name }
-                : {}),
+        const result = this.process(parsedInput, context).then((output) => {
+            const parsedOutput = this.outputSchema.passthrough().parse(output);
+            return this.includeInputInOutput ? { ...parsedInput, ...parsedOutput } : parsedOutput;
         });
-        return finalOutput;
+        return logger_1.logger.debug.spinner(result, `Call agent ${this.name}`, (output) => logger_1.logger.debug("%O", { input, output }), { disabled: this.disableLogging });
     }
     async shutdown() { }
 }

package/lib/cjs/agents/mcp-agent.js

@@ -10,7 +10,7 @@ const logger_1 = require("../utils/logger");
 const agent_1 = require("./agent");
 const MCP_AGENT_CLIENT_NAME = "MCPAgent";
 const MCP_AGENT_CLIENT_VERSION = "0.0.1";
-const debug = logger_1.logger.base.extend("mcp-agent:debug", "/");
+const debug = logger_1.logger.base.extend("mcp");
 function isSSEServerParameters(options) {
     return "url" in options && typeof options.url === "string";
 }
@@ -34,13 +34,9 @@ class MCPAgent extends agent_1.Agent {
             name: MCP_AGENT_CLIENT_NAME,
             version: MCP_AGENT_CLIENT_VERSION,
         });
-        debug(`Connecting to MCP server with transport ${transport.constructor.name}`);
-        await client.connect(transport);
-        debug(`Connected to MCP server with transport ${transport.constructor.name}`);
+        await debug.spinner(client.connect(transport), "Connecting to MCP server");
         const mcpServer = getMCPServerName(client);
-        debug(`Listing tools from ${mcpServer}`);
-        const { tools: mcpTools } = await client.listTools();
-        debug(`Listed tools from ${mcpServer}`, mcpTools.map((i) => i.name));
+        const { tools: mcpTools } = await debug.spinner(client.listTools(), `Listing tools from ${mcpServer}`, ({ tools }) => debug("%O", tools));
         const tools = mcpTools.map((tool) => {
             return new MCPTool({
                 client,
@@ -79,9 +75,7 @@ class MCPTool extends agent_1.Agent {
         return getMCPServerName(this.client);
     }
     async process(input) {
-        debug(`Start call tool ${this.name} from ${this.mcpServer} with input`, input);
-        const result = await this.client.callTool({ name: this.name, arguments: input });
-        debug(`End call tool ${this.name} from ${this.mcpServer} with result`, result);
+        const result = await debug.spinner(this.client.callTool({ name: this.name, arguments: input }), `Call tool ${this.name} from ${this.mcpServer}`, (output) => debug("%O", { input, output }));
         return result;
     }
 }

package/lib/cjs/execution-engine/index.d.ts

@@ -33,6 +33,7 @@ export declare class ExecutionEngine extends EventEmitter implements Context {
     private runChatLoop;
     private callAgent;
     shutdown(): Promise<void>;
+    private initProcessExitHandler;
 }
 export declare class UserAgent<I extends AgentInput = AgentInput, O extends AgentOutput = AgentOutput> extends Agent<I, O> {
     options: AgentOptions<I, O> & {

package/lib/cjs/execution-engine/index.js

@@ -18,6 +18,7 @@ class ExecutionEngine extends node_events_1.default {
         this.tools = options?.tools ?? [];
         if (options?.agents?.length)
             this.addAgent(...options.agents);
+        this.initProcessExitHandler();
     }
     messageQueue = new message_queue_1.MessageQueue();
     model;
@@ -175,12 +176,17 @@ class ExecutionEngine extends node_events_1.default {
             await agent.shutdown();
         }
     }
+    initProcessExitHandler() {
+        const shutdownAndExit = () => this.shutdown().finally(() => process.exit(0));
+        process.on("SIGINT", shutdownAndExit);
+        process.on("exit", shutdownAndExit);
+    }
 }
 exports.ExecutionEngine = ExecutionEngine;
 class UserAgent extends agent_1.Agent {
     options;
     constructor(options) {
-        super(options);
+        super({ ...options, disableLogging: true });
         this.options = options;
     }
     process(input) {

package/lib/cjs/index.d.ts

@@ -8,3 +8,4 @@ export * from "./models/chat";
 export * from "./models/chat-openai";
 export * from "./prompt/prompt-builder";
 export * from "./prompt/template";
+export * from "./utils/run-chat-loop";

package/lib/cjs/index.js

@@ -24,3 +24,4 @@ __exportStar(require("./models/chat"), exports);
 __exportStar(require("./models/chat-openai"), exports);
 __exportStar(require("./prompt/prompt-builder"), exports);
 __exportStar(require("./prompt/template"), exports);
+__exportStar(require("./utils/run-chat-loop"), exports);

package/lib/cjs/utils/logger.d.ts

@@ -1,6 +1,14 @@
-import debug from "debug";
+import { type Debugger } from "debug";
+interface DebugWithSpinner extends Debugger {
+    spinner<T>(promise: Promise<T>, message?: string, callback?: (result: T) => void, options?: {
+        disabled?: boolean;
+    }): Promise<T>;
+    extend: (namespace: string) => DebugWithSpinner;
+}
+declare function spinner<T>(promise: Promise<T>, message?: string, callback?: (result: T) => void): Promise<T>;
 export declare const logger: {
-    base: debug.Debugger;
-    debug: debug.Debugger;
-    error: debug.Debugger;
+    base: DebugWithSpinner;
+    debug: DebugWithSpinner;
+    spinner: typeof spinner;
 };
+export {};

package/lib/cjs/utils/logger.js

@@ -5,8 +5,63 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.logger = void 0;
 const debug_1 = __importDefault(require("debug"));
+const ora_1 = __importDefault(require("ora"));
+debug_1.default.log = (...args) => {
+    const { isSpinning } = globalSpinner;
+    if (isSpinning)
+        globalSpinner.stop();
+    console.log(...args);
+    if (isSpinning)
+        globalSpinner.start();
+};
+function createDebugger(namespace) {
+    const i = (0, debug_1.default)(namespace);
+    function overrideExtend(debug) {
+        const originalExtend = debug.extend;
+        debug.extend = (namespace) => {
+            const extended = originalExtend.call(debug, namespace);
+            overrideExtend(extended);
+            extended.spinner = async (promise, message, callback, options) => {
+                if (!extended.enabled || options?.disabled)
+                    return promise;
+                return spinner(promise, message, callback);
+            };
+            return extended;
+        };
+    }
+    overrideExtend(i);
+    return i;
+}
+const globalSpinner = (0, ora_1.default)();
+async function spinner(promise, message, callback) {
+    const { isSpinning, text } = globalSpinner;
+    try {
+        globalSpinner.start(message || " ");
+        const result = await promise;
+        if (message)
+            globalSpinner.succeed(message);
+        else
+            globalSpinner.stop();
+        // NOTE: This is a workaround to make sure the spinner stops spinning before the next tick
+        await new Promise((resolve) => setTimeout(resolve, 100));
+        callback?.(result);
+        // NOTE: This is a workaround to make sure the spinner stops spinning before the next tick
+        await new Promise((resolve) => setTimeout(resolve, 100));
+        return result;
+    }
+    catch (error) {
+        globalSpinner.fail(message || " ");
+        throw error;
+    }
+    finally {
+        // Recover the spinner state
+        if (isSpinning)
+            globalSpinner.start(text || " ");
+    }
+}
+const base = createDebugger("aigne");
 exports.logger = {
-    base: (0, debug_1.default)("@aigne/core"),
-    debug: (0, debug_1.default)("@aigne/core:debug"),
-    error: (0, debug_1.default)("@aigne/core:error"),
+    base,
+    debug: base.extend("core"),
+    spinner,
 };

package/lib/cjs/utils/run-chat-loop.d.ts

@@ -0,0 +1,8 @@
+import type { AgentOutput } from "../agents/agent";
+import type { UserAgent } from "../execution-engine";
+export interface ChatLoopOptions {
+    welcome?: string;
+    defaultQuestion?: string;
+    onResponse?: (response: AgentOutput) => void;
+}
+export declare function runChatLoopInTerminal(userAgent: UserAgent, options?: ChatLoopOptions): Promise<void>;

package/lib/cjs/utils/run-chat-loop.js

@@ -0,0 +1,44 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.runChatLoopInTerminal = runChatLoopInTerminal;
+const inquirer_1 = __importDefault(require("inquirer"));
+const logger_1 = require("./logger");
+async function runChatLoopInTerminal(userAgent, options) {
+    if (options?.welcome)
+        console.log(options.welcome);
+    for (let i = 0;; i++) {
+        const { question } = await inquirer_1.default.prompt([
+            {
+                type: "input",
+                name: "question",
+                message: ">",
+                default: i === 0 ? options?.defaultQuestion : undefined,
+            },
+        ]);
+        if (!question.trim())
+            continue;
+        const cmd = COMMANDS[question.trim()];
+        if (cmd) {
+            await cmd();
+            continue;
+        }
+        const response = await logger_1.logger.spinner(userAgent.call(question));
+        if (options?.onResponse)
+            options.onResponse(response);
+        else
+            console.log(response);
+    }
+}
+const COMMANDS = {
+    "/exit": () => process.exit(0),
+    "/help": () => {
+        console.log(`\
+Commands:
+  /exit - exit the chat loop
+  /help - show this help message
+`);
+    },
+};

package/lib/dts/agents/agent.d.ts

@@ -19,6 +19,7 @@ export interface AgentOptions<I extends AgentInput = AgentInput, O extends Agent
     }>;
     includeInputInOutput?: boolean;
     tools?: (Agent | FunctionAgentFn)[];
+    disableLogging?: boolean;
 }
 export declare class Agent<I extends AgentInput = AgentInput, O extends AgentOutput = AgentOutput> extends EventEmitter {
     static from<I extends AgentInput = AgentInput, O extends AgentOutput = AgentOutput>(options: AgentOptions<I, O>): Agent<I, O>;
@@ -37,6 +38,7 @@ export declare class Agent<I extends AgentInput = AgentInput, O extends AgentOut
     readonly tools: Agent<AgentInput, AgentOutput>[] & {
         [key: string]: Agent<AgentInput, AgentOutput>;
     };
+    private disableLogging?;
     addTool<I extends AgentInput, O extends AgentOutput>(tool: Agent<I, O> | FunctionAgentFn<I, O>): void;
     get isCallable(): boolean;
     call(input: I | string, context?: Context): Promise<O>;

package/lib/dts/execution-engine/index.d.ts

@@ -33,6 +33,7 @@ export declare class ExecutionEngine extends EventEmitter implements Context {
     private runChatLoop;
     private callAgent;
     shutdown(): Promise<void>;
+    private initProcessExitHandler;
 }
 export declare class UserAgent<I extends AgentInput = AgentInput, O extends AgentOutput = AgentOutput> extends Agent<I, O> {
     options: AgentOptions<I, O> & {

package/lib/dts/index.d.ts

@@ -8,3 +8,4 @@ export * from "./models/chat";
 export * from "./models/chat-openai";
 export * from "./prompt/prompt-builder";
 export * from "./prompt/template";
+export * from "./utils/run-chat-loop";

package/lib/dts/utils/logger.d.ts

@@ -1,6 +1,14 @@
-import debug from "debug";
+import { type Debugger } from "debug";
+interface DebugWithSpinner extends Debugger {
+    spinner<T>(promise: Promise<T>, message?: string, callback?: (result: T) => void, options?: {
+        disabled?: boolean;
+    }): Promise<T>;
+    extend: (namespace: string) => DebugWithSpinner;
+}
+declare function spinner<T>(promise: Promise<T>, message?: string, callback?: (result: T) => void): Promise<T>;
 export declare const logger: {
-    base: debug.Debugger;
-    debug: debug.Debugger;
-    error: debug.Debugger;
+    base: DebugWithSpinner;
+    debug: DebugWithSpinner;
+    spinner: typeof spinner;
 };
+export {};

package/lib/dts/utils/run-chat-loop.d.ts

@@ -0,0 +1,8 @@
+import type { AgentOutput } from "../agents/agent";
+import type { UserAgent } from "../execution-engine";
+export interface ChatLoopOptions {
+    welcome?: string;
+    defaultQuestion?: string;
+    onResponse?: (response: AgentOutput) => void;
+}
+export declare function runChatLoopInTerminal(userAgent: UserAgent, options?: ChatLoopOptions): Promise<void>;

package/lib/esm/agents/agent.d.ts

@@ -19,6 +19,7 @@ export interface AgentOptions<I extends AgentInput = AgentInput, O extends Agent
     }>;
     includeInputInOutput?: boolean;
     tools?: (Agent | FunctionAgentFn)[];
+    disableLogging?: boolean;
 }
 export declare class Agent<I extends AgentInput = AgentInput, O extends AgentOutput = AgentOutput> extends EventEmitter {
     static from<I extends AgentInput = AgentInput, O extends AgentOutput = AgentOutput>(options: AgentOptions<I, O>): Agent<I, O>;
@@ -37,6 +38,7 @@ export declare class Agent<I extends AgentInput = AgentInput, O extends AgentOut
     readonly tools: Agent<AgentInput, AgentOutput>[] & {
         [key: string]: Agent<AgentInput, AgentOutput>;
     };
+    private disableLogging?;
     addTool<I extends AgentInput, O extends AgentOutput>(tool: Agent<I, O> | FunctionAgentFn<I, O>): void;
     get isCallable(): boolean;
     call(input: I | string, context?: Context): Promise<O>;

package/lib/esm/agents/agent.js

@@ -2,7 +2,7 @@ import EventEmitter from "node:events";
 import { z } from "zod";
 import { userInput } from "../prompt/prompt-builder";
 import { logger } from "../utils/logger";
-import { transferAgentOutputKey, transferToAgentOutput } from "./types";
+import { transferToAgentOutput } from "./types";
 export class Agent extends EventEmitter {
     static from(options) {
         return new Agent(options);
@@ -20,6 +20,7 @@ export class Agent extends EventEmitter {
         this.publishTopic = options.publishTopic;
         if (options.tools?.length)
             this.tools.push(...options.tools.map(functionToAgent));
+        this.disableLogging = options.disableLogging;
     }
     name;
     description;
@@ -31,6 +32,7 @@ export class Agent extends EventEmitter {
     tools = new Proxy([], {
         get: (t, p, r) => Reflect.get(t, p, r) ?? t.find((t) => t.name === p),
     });
+    disableLogging;
     addTool(tool) {
         this.tools.push(typeof tool === "function" ? functionToAgent(tool) : tool);
     }
@@ -42,19 +44,11 @@ export class Agent extends EventEmitter {
             throw new Error("Agent must implement process method");
         const _input = typeof input === "string" ? userInput(input) : input;
         const parsedInput = this.inputSchema.passthrough().parse(_input);
-        logger.debug(`Call agent ${this.name} start`, parsedInput);
-        const output = await this.process(parsedInput, context);
-        const parsedOutput = this.outputSchema.passthrough().parse(output);
-        const finalOutput = this.includeInputInOutput
-            ? { ...parsedInput, ...parsedOutput }
-            : parsedOutput;
-        logger.debug(`Call agent ${this.name} successfully`, {
-            ...finalOutput,
-            ...(finalOutput[transferAgentOutputKey]
-                ? { [transferAgentOutputKey]: finalOutput[transferAgentOutputKey].agent.name }
-                : {}),
+        const result = this.process(parsedInput, context).then((output) => {
+            const parsedOutput = this.outputSchema.passthrough().parse(output);
+            return this.includeInputInOutput ? { ...parsedInput, ...parsedOutput } : parsedOutput;
         });
-        return finalOutput;
+        return logger.debug.spinner(result, `Call agent ${this.name}`, (output) => logger.debug("%O", { input, output }), { disabled: this.disableLogging });
     }
     async shutdown() { }
 }

package/lib/esm/agents/mcp-agent.js

@@ -7,7 +7,7 @@ import { logger } from "../utils/logger";
 import { Agent } from "./agent";
 const MCP_AGENT_CLIENT_NAME = "MCPAgent";
 const MCP_AGENT_CLIENT_VERSION = "0.0.1";
-const debug = logger.base.extend("mcp-agent:debug", "/");
+const debug = logger.base.extend("mcp");
 function isSSEServerParameters(options) {
     return "url" in options && typeof options.url === "string";
 }
@@ -31,13 +31,9 @@ export class MCPAgent extends Agent {
             name: MCP_AGENT_CLIENT_NAME,
             version: MCP_AGENT_CLIENT_VERSION,
         });
-        debug(`Connecting to MCP server with transport ${transport.constructor.name}`);
-        await client.connect(transport);
-        debug(`Connected to MCP server with transport ${transport.constructor.name}`);
+        await debug.spinner(client.connect(transport), "Connecting to MCP server");
         const mcpServer = getMCPServerName(client);
-        debug(`Listing tools from ${mcpServer}`);
-        const { tools: mcpTools } = await client.listTools();
-        debug(`Listed tools from ${mcpServer}`, mcpTools.map((i) => i.name));
+        const { tools: mcpTools } = await debug.spinner(client.listTools(), `Listing tools from ${mcpServer}`, ({ tools }) => debug("%O", tools));
         const tools = mcpTools.map((tool) => {
             return new MCPTool({
                 client,
@@ -75,9 +71,7 @@ export class MCPTool extends Agent {
         return getMCPServerName(this.client);
     }
     async process(input) {
-        debug(`Start call tool ${this.name} from ${this.mcpServer} with input`, input);
-        const result = await this.client.callTool({ name: this.name, arguments: input });
-        debug(`End call tool ${this.name} from ${this.mcpServer} with result`, result);
+        const result = await debug.spinner(this.client.callTool({ name: this.name, arguments: input }), `Call tool ${this.name} from ${this.mcpServer}`, (output) => debug("%O", { input, output }));
         return result;
     }
 }

package/lib/esm/execution-engine/index.d.ts

@@ -33,6 +33,7 @@ export declare class ExecutionEngine extends EventEmitter implements Context {
     private runChatLoop;
     private callAgent;
     shutdown(): Promise<void>;
+    private initProcessExitHandler;
 }
 export declare class UserAgent<I extends AgentInput = AgentInput, O extends AgentOutput = AgentOutput> extends Agent<I, O> {
     options: AgentOptions<I, O> & {

package/lib/esm/execution-engine/index.js

@@ -12,6 +12,7 @@ export class ExecutionEngine extends EventEmitter {
         this.tools = options?.tools ?? [];
         if (options?.agents?.length)
             this.addAgent(...options.agents);
+        this.initProcessExitHandler();
     }
     messageQueue = new MessageQueue();
     model;
@@ -169,11 +170,16 @@ export class ExecutionEngine extends EventEmitter {
             await agent.shutdown();
         }
     }
+    initProcessExitHandler() {
+        const shutdownAndExit = () => this.shutdown().finally(() => process.exit(0));
+        process.on("SIGINT", shutdownAndExit);
+        process.on("exit", shutdownAndExit);
+    }
 }
 export class UserAgent extends Agent {
     options;
     constructor(options) {
-        super(options);
+        super({ ...options, disableLogging: true });
         this.options = options;
     }
     process(input) {

package/lib/esm/index.d.ts

@@ -8,3 +8,4 @@ export * from "./models/chat";
 export * from "./models/chat-openai";
 export * from "./prompt/prompt-builder";
 export * from "./prompt/template";
+export * from "./utils/run-chat-loop";

package/lib/esm/index.js

@@ -8,3 +8,4 @@ export * from "./models/chat";
 export * from "./models/chat-openai";
 export * from "./prompt/prompt-builder";
 export * from "./prompt/template";
+export * from "./utils/run-chat-loop";

package/lib/esm/utils/logger.d.ts

@@ -1,6 +1,14 @@
-import debug from "debug";
+import { type Debugger } from "debug";
+interface DebugWithSpinner extends Debugger {
+    spinner<T>(promise: Promise<T>, message?: string, callback?: (result: T) => void, options?: {
+        disabled?: boolean;
+    }): Promise<T>;
+    extend: (namespace: string) => DebugWithSpinner;
+}
+declare function spinner<T>(promise: Promise<T>, message?: string, callback?: (result: T) => void): Promise<T>;
 export declare const logger: {
-    base: debug.Debugger;
-    debug: debug.Debugger;
-    error: debug.Debugger;
+    base: DebugWithSpinner;
+    debug: DebugWithSpinner;
+    spinner: typeof spinner;
 };
+export {};

package/lib/esm/utils/logger.js

@@ -1,6 +1,61 @@
 import debug from "debug";
+import ora from "ora";
+debug.log = (...args) => {
+    const { isSpinning } = globalSpinner;
+    if (isSpinning)
+        globalSpinner.stop();
+    console.log(...args);
+    if (isSpinning)
+        globalSpinner.start();
+};
+function createDebugger(namespace) {
+    const i = debug(namespace);
+    function overrideExtend(debug) {
+        const originalExtend = debug.extend;
+        debug.extend = (namespace) => {
+            const extended = originalExtend.call(debug, namespace);
+            overrideExtend(extended);
+            extended.spinner = async (promise, message, callback, options) => {
+                if (!extended.enabled || options?.disabled)
+                    return promise;
+                return spinner(promise, message, callback);
+            };
+            return extended;
+        };
+    }
+    overrideExtend(i);
+    return i;
+}
+const globalSpinner = ora();
+async function spinner(promise, message, callback) {
+    const { isSpinning, text } = globalSpinner;
+    try {
+        globalSpinner.start(message || " ");
+        const result = await promise;
+        if (message)
+            globalSpinner.succeed(message);
+        else
+            globalSpinner.stop();
+        // NOTE: This is a workaround to make sure the spinner stops spinning before the next tick
+        await new Promise((resolve) => setTimeout(resolve, 100));
+        callback?.(result);
+        // NOTE: This is a workaround to make sure the spinner stops spinning before the next tick
+        await new Promise((resolve) => setTimeout(resolve, 100));
+        return result;
+    }
+    catch (error) {
+        globalSpinner.fail(message || " ");
+        throw error;
+    }
+    finally {
+        // Recover the spinner state
+        if (isSpinning)
+            globalSpinner.start(text || " ");
+    }
+}
+const base = createDebugger("aigne");
 export const logger = {
-    base: debug("@aigne/core"),
-    debug: debug("@aigne/core:debug"),
-    error: debug("@aigne/core:error"),
+    base,
+    debug: base.extend("core"),
+    spinner,
 };

package/lib/esm/utils/run-chat-loop.d.ts

@@ -0,0 +1,8 @@
+import type { AgentOutput } from "../agents/agent";
+import type { UserAgent } from "../execution-engine";
+export interface ChatLoopOptions {
+    welcome?: string;
+    defaultQuestion?: string;
+    onResponse?: (response: AgentOutput) => void;
+}
+export declare function runChatLoopInTerminal(userAgent: UserAgent, options?: ChatLoopOptions): Promise<void>;

package/lib/esm/utils/run-chat-loop.js

@@ -0,0 +1,38 @@
+import inquirer from "inquirer";
+import { logger } from "./logger";
+export async function runChatLoopInTerminal(userAgent, options) {
+    if (options?.welcome)
+        console.log(options.welcome);
+    for (let i = 0;; i++) {
+        const { question } = await inquirer.prompt([
+            {
+                type: "input",
+                name: "question",
+                message: ">",
+                default: i === 0 ? options?.defaultQuestion : undefined,
+            },
+        ]);
+        if (!question.trim())
+            continue;
+        const cmd = COMMANDS[question.trim()];
+        if (cmd) {
+            await cmd();
+            continue;
+        }
+        const response = await logger.spinner(userAgent.call(question));
+        if (options?.onResponse)
+            options.onResponse(response);
+        else
+            console.log(response);
+    }
+}
+const COMMANDS = {
+    "/exit": () => process.exit(0),
+    "/help": () => {
+        console.log(`\
+Commands:
+  /exit - exit the chat loop
+  /help - show this help message
+`);
+    },
+};

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@aigne/core",
-  "version": "1.1.0-beta.2",
+  "version": "1.1.0-beta.4",
   "description": "AIGNE core library",
   "publishConfig": {
     "access": "public"
@@ -8,16 +8,24 @@
   "author": "Arcblock <blocklet@arcblock.io> https://github.com/blocklet",
   "homepage": "https://github.com/blocklet/ai-studio#readme",
   "license": "ISC",
+  "type": "module",
   "main": "./lib/cjs/index.js",
   "module": "./lib/esm/index.js",
   "types": "./lib/dts/index.d.ts",
+  "exports": {
+    ".": {
+      "import": "./lib/esm/index.js",
+      "require": "./lib/cjs/index.js",
+      "types": "./lib/dts/index.d.ts"
+    }
+  },
   "files": [
     "lib/cjs",
     "lib/dts",
     "lib/esm",
     "LICENSE",
-    "package.json",
-    "README.md"
+    "README.md",
+    "CHANGELOG.md"
   ],
   "repository": {
     "type": "git",
@@ -30,10 +38,12 @@
     "@types/debug": "^4.1.12",
     "@types/mustache": "^4.2.5",
     "debug": "^4.4.0",
+    "inquirer": "^12.4.3",
     "lodash": "^4.17.21",
     "mustache": "^4.2.0",
     "nanoid": "^5.1.3",
     "openai": "^4.86.2",
+    "ora": "^8.2.0",
     "zod": "^3.24.2",
     "zod-to-json-schema": "^3.24.3"
   },