@aigne/core 1.17.0 → 1.18.0

package/CHANGELOG.md

@@ -1,33 +1,23 @@
-## [1.3.0](https://github.com/AIGNE-io/aigne-framework/compare/core-v1.2.0...core-v1.3.0) (2025-03-24)
+## [1.17.0](https://github.com/AIGNE-io/aigne-framework/compare/core-v1.16.0...core-v1.17.0) (2025-05-25)
 
 
 ### Features
 
-* add Agent FunctionAgent AIAgent MCPAgent and ExecutionEngine ([4d2a5a1](https://github.com/AIGNE-io/aigne-framework/commit/4d2a5a1b3366b8f935f50a0937c2da6e49073348))
-* add OrchestratorAgent in agent library ([25a5e9e](https://github.com/AIGNE-io/aigne-framework/commit/25a5e9e6c60d747c8bf484ac884b31dc02c14757))
-* add sequential and parallel helper function ([a295697](https://github.com/AIGNE-io/aigne-framework/commit/a295697b5694754e02954fc5c7f382a3b219a3ab))
-* add support for MCP resources ([1ded2fb](https://github.com/AIGNE-io/aigne-framework/commit/1ded2fbf222fa8984e75df0852ff384524f73b04))
-* **core:** add ChatModelClaude to use models of anthropic ([#30](https://github.com/AIGNE-io/aigne-framework/issues/30)) ([0a62a64](https://github.com/AIGNE-io/aigne-framework/commit/0a62a6499e3da723a4646e67952051708ce7de6a))
-* **core:** add support for subscribing topics for agent memory ([#28](https://github.com/AIGNE-io/aigne-framework/issues/28)) ([eeecc67](https://github.com/AIGNE-io/aigne-framework/commit/eeecc67049a60ebcc4cdba0fbcd987b3d81f4af6))
-* **prompt-builder:** support chat history in PromptBuilder ([6ca05f2](https://github.com/AIGNE-io/aigne-framework/commit/6ca05f28eddb683a4f1e228865f8bbf8a8e190f1))
-* support run puppeteer example chat loop in terminal ([85ce7f8](https://github.com/AIGNE-io/aigne-framework/commit/85ce7f8de8b443c86e50815dd7bcab99f869c4ce))
-* use PromptBuilder instead of string instructions ([e4cb2cb](https://github.com/AIGNE-io/aigne-framework/commit/e4cb2cb6baf4f9bcef390567a4a174e9246c29a3))
+* add user context support ([#131](https://github.com/AIGNE-io/aigne-framework/issues/131)) ([4dd9d20](https://github.com/AIGNE-io/aigne-framework/commit/4dd9d20953f6ac33933723db56efd9b44bafeb02))
 
+## [1.18.0](https://github.com/AIGNE-io/aigne-framework/compare/core-v1.17.0...core-v1.18.0) (2025-05-29)
 
-### Bug Fixes
 
-* **AIAgent:** should pass both arguments (model generated) and input (user input) to tool ([c49d64e](https://github.com/AIGNE-io/aigne-framework/commit/c49d64ee35f7efd83b0f82f43205bb1c40f999e8))
-* **core:** enforce stricter input/output type checks ([#26](https://github.com/AIGNE-io/aigne-framework/issues/26)) ([ef8cf53](https://github.com/AIGNE-io/aigne-framework/commit/ef8cf53586aff08a809909c56ab2a20f215fa129))
-* **MCP:** catch list resource error treat as empty list ([1885fab](https://github.com/AIGNE-io/aigne-framework/commit/1885fab3585e0dd1467b127e5b47cd0b98282bab))
-* rename @aigne/core-next to @aigne/core ([3a81009](https://github.com/AIGNE-io/aigne-framework/commit/3a8100962c81813217b687ae28e8de604419c622))
-* use text resource from MCP correctly ([8b9eba8](https://github.com/AIGNE-io/aigne-framework/commit/8b9eba83352ec096a2a5d4f410d4c4bde7420bce))
+### Features
 
-## [1.17.0](https://github.com/AIGNE-io/aigne-framework/compare/core-v1.16.0...core-v1.17.0) (2025-05-25)
+* add memory agents support for client agent ([#139](https://github.com/AIGNE-io/aigne-framework/issues/139)) ([57044fa](https://github.com/AIGNE-io/aigne-framework/commit/57044fa87b8abcba395cd05f941d6d312ab65764))
 
 
-### Features
+### Dependencies
 
-* add user context support ([#131](https://github.com/AIGNE-io/aigne-framework/issues/131)) ([4dd9d20](https://github.com/AIGNE-io/aigne-framework/commit/4dd9d20953f6ac33933723db56efd9b44bafeb02))
+* The following workspace dependencies were updated
+  * dependencies
+    * @aigne/platform-helpers bumped to 0.1.0
 
 ## [1.16.0](https://github.com/AIGNE-io/aigne-framework/compare/core-v1.15.0...core-v1.16.0) (2025-05-23)
 
@@ -208,6 +198,30 @@
 
 * **core:** use system message as user message for claude model if needed ([#32](https://github.com/AIGNE-io/aigne-framework/issues/32)) ([316a6d5](https://github.com/AIGNE-io/aigne-framework/commit/316a6d51f885cceee4020c24695f6588f6b7425f))
 
+## [1.3.0](https://github.com/AIGNE-io/aigne-framework/compare/core-v1.2.0...core-v1.3.0) (2025-03-24)
+
+
+### Features
+
+* add Agent FunctionAgent AIAgent MCPAgent and ExecutionEngine ([4d2a5a1](https://github.com/AIGNE-io/aigne-framework/commit/4d2a5a1b3366b8f935f50a0937c2da6e49073348))
+* add OrchestratorAgent in agent library ([25a5e9e](https://github.com/AIGNE-io/aigne-framework/commit/25a5e9e6c60d747c8bf484ac884b31dc02c14757))
+* add sequential and parallel helper function ([a295697](https://github.com/AIGNE-io/aigne-framework/commit/a295697b5694754e02954fc5c7f382a3b219a3ab))
+* add support for MCP resources ([1ded2fb](https://github.com/AIGNE-io/aigne-framework/commit/1ded2fbf222fa8984e75df0852ff384524f73b04))
+* **core:** add ChatModelClaude to use models of anthropic ([#30](https://github.com/AIGNE-io/aigne-framework/issues/30)) ([0a62a64](https://github.com/AIGNE-io/aigne-framework/commit/0a62a6499e3da723a4646e67952051708ce7de6a))
+* **core:** add support for subscribing topics for agent memory ([#28](https://github.com/AIGNE-io/aigne-framework/issues/28)) ([eeecc67](https://github.com/AIGNE-io/aigne-framework/commit/eeecc67049a60ebcc4cdba0fbcd987b3d81f4af6))
+* **prompt-builder:** support chat history in PromptBuilder ([6ca05f2](https://github.com/AIGNE-io/aigne-framework/commit/6ca05f28eddb683a4f1e228865f8bbf8a8e190f1))
+* support run puppeteer example chat loop in terminal ([85ce7f8](https://github.com/AIGNE-io/aigne-framework/commit/85ce7f8de8b443c86e50815dd7bcab99f869c4ce))
+* use PromptBuilder instead of string instructions ([e4cb2cb](https://github.com/AIGNE-io/aigne-framework/commit/e4cb2cb6baf4f9bcef390567a4a174e9246c29a3))
+
+
+### Bug Fixes
+
+* **AIAgent:** should pass both arguments (model generated) and input (user input) to tool ([c49d64e](https://github.com/AIGNE-io/aigne-framework/commit/c49d64ee35f7efd83b0f82f43205bb1c40f999e8))
+* **core:** enforce stricter input/output type checks ([#26](https://github.com/AIGNE-io/aigne-framework/issues/26)) ([ef8cf53](https://github.com/AIGNE-io/aigne-framework/commit/ef8cf53586aff08a809909c56ab2a20f215fa129))
+* **MCP:** catch list resource error treat as empty list ([1885fab](https://github.com/AIGNE-io/aigne-framework/commit/1885fab3585e0dd1467b127e5b47cd0b98282bab))
+* rename @aigne/core-next to @aigne/core ([3a81009](https://github.com/AIGNE-io/aigne-framework/commit/3a8100962c81813217b687ae28e8de604419c622))
+* use text resource from MCP correctly ([8b9eba8](https://github.com/AIGNE-io/aigne-framework/commit/8b9eba83352ec096a2a5d4f410d4c4bde7420bce))
+
 ## [1.2.0](https://github.com/AIGNE-io/aigne-framework/compare/core-next-v1.1.0...core-next-v1.2.0) (2025-03-18)
 
 - chore: release v1.2.0

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

@@ -1,8 +1,10 @@
-import { inspect } from "node:util";
 import { ZodObject, type ZodType } from "zod";
 import type { Context, UserContext } from "../aigne/context.js";
 import type { MessagePayload } from "../aigne/message-queue.js";
-import type { MemoryAgent } from "../memory/memory.js";
+import type { Memory, MemoryAgent } from "../memory/memory.js";
+import type { MemoryRecorderInput } from "../memory/recorder.js";
+import type { MemoryRetrieverInput } from "../memory/retriever.js";
+import { nodejs } from "../utils/nodejs.js";
 import { type Nullish, type PromiseOrValue, type XOr } from "../utils/type-utils.js";
 import type { GuideRailAgent, GuideRailAgentOutput } from "./guide-rail-agent.js";
 import { type TransferAgentOutput } from "./types.js";
@@ -95,6 +97,10 @@ export interface AgentOptions<I extends Message = Message, O extends Message = M
      * One or more memory agents this agent can use
      */
     memory?: MemoryAgent | MemoryAgent[];
+    /**
+     * Maximum number of memory items to retrieve
+     */
+    maxRetrieveMemoryCount?: number;
 }
 export declare const agentOptionsSchema: ZodObject<{
     [key in keyof AgentOptions]: ZodType<AgentOptions[key]>;
@@ -125,6 +131,8 @@ export interface AgentInvokeOptions<U extends UserContext = UserContext> {
      * and returns the final JSON result
      */
     streaming?: boolean;
+    userContext?: U;
+    memories?: Pick<Memory, "content">[];
 }
 /**
  * Agent is the base class for all agents.
@@ -149,11 +157,24 @@ export interface AgentInvokeOptions<U extends UserContext = UserContext> {
  * {@includeCode ../../test/agents/agent.test.ts#example-custom-agent}
  */
 export declare abstract class Agent<I extends Message = Message, O extends Message = Message> {
+    /**
+     * Custom object inspection behavior
+     *
+     * When using Node.js's util.inspect function to inspect an agent,
+     * only the agent's name will be shown, making output more concise
+     *
+     * @returns Agent name
+     */
+    [nodejs.customInspect]: () => string;
     constructor(options?: AgentOptions<I, O>);
     /**
      * List of memories this agent can use
      */
     readonly memories: MemoryAgent[];
+    /**
+     * Maximum number of memory items to retrieve
+     */
+    maxRetrieveMemoryCount?: number;
     /**
      * Lifecycle hooks for agent processing.
      *
@@ -297,6 +318,10 @@ export declare abstract class Agent<I extends Message = Message, O extends Messa
      */
     private checkContextStatus;
     private newDefaultContext;
+    retrieveMemories(input: Pick<MemoryRetrieverInput, "limit"> & {
+        search?: Message | string;
+    }, options: Pick<AgentInvokeOptions, "context">): Promise<Pick<Memory, "content">[]>;
+    recordMemories(input: MemoryRecorderInput, options: Pick<AgentInvokeOptions, "context">): Promise<void>;
     /**
      * Invoke the agent with regular (non-streaming) response
      *
@@ -460,15 +485,6 @@ export declare abstract class Agent<I extends Message = Message, O extends Messa
      * {@includeCode ../../test/agents/agent.test.ts#example-agent-shutdown-by-using}
      */
     shutdown(): Promise<void>;
-    /**
-     * Custom object inspection behavior
-     *
-     * When using Node.js's util.inspect function to inspect an agent,
-     * only the agent's name will be shown, making output more concise
-     *
-     * @returns Agent name
-     */
-    [inspect.custom](): string;
     /**
      * Async dispose method for shutdown the agent
      *

package/lib/cjs/agents/agent.js

@@ -40,10 +40,10 @@ exports.FunctionAgent = exports.Agent = exports.agentOptionsSchema = void 0;
 exports.isEmptyChunk = isEmptyChunk;
 exports.textDelta = textDelta;
 exports.jsonDelta = jsonDelta;
-const node_util_1 = require("node:util");
 const zod_1 = require("zod");
 const prompt_builder_js_1 = require("../prompt/prompt-builder.js");
 const logger_js_1 = require("../utils/logger.js");
+const nodejs_js_1 = require("../utils/nodejs.js");
 const stream_utils_js_1 = require("../utils/stream-utils.js");
 const type_utils_js_1 = require("../utils/type-utils.js");
 const types_js_1 = require("./types.js");
@@ -61,6 +61,7 @@ exports.agentOptionsSchema = zod_1.z.object({
     skills: zod_1.z.array(zod_1.z.union([zod_1.z.custom(), zod_1.z.custom()])).optional(),
     disableEvents: zod_1.z.boolean().optional(),
     memory: zod_1.z.union([zod_1.z.custom(), zod_1.z.array(zod_1.z.custom())]).optional(),
+    maxRetrieveMemoryCount: zod_1.z.number().optional(),
     hooks: zod_1.z
         .object({
         onStart: zod_1.z.custom().optional(),
@@ -117,6 +118,7 @@ class Agent {
         else if (options.memory) {
             this.memories.push(options.memory);
         }
+        this.maxRetrieveMemoryCount = options.maxRetrieveMemoryCount;
         this.hooks = options.hooks ?? {};
         this.guideRails = options.guideRails;
     }
@@ -124,6 +126,10 @@ class Agent {
      * List of memories this agent can use
      */
     memories = [];
+    /**
+     * Maximum number of memory items to retrieve
+     */
+    maxRetrieveMemoryCount;
     /**
      * Lifecycle hooks for agent processing.
      *
@@ -304,11 +310,33 @@ class Agent {
     async newDefaultContext() {
         return Promise.resolve().then(() => __importStar(require("../aigne/context.js"))).then((m) => new m.AIGNEContext());
     }
+    async retrieveMemories(input, options) {
+        const memories = [];
+        for (const memory of this.memories) {
+            const ms = (await memory.retrieve({
+                ...input,
+                search: typeof input === "string" ? input : input && (0, prompt_builder_js_1.getMessage)(input),
+                limit: input.limit ?? this.maxRetrieveMemoryCount,
+            }, options.context)).memories;
+            memories.push(...ms);
+        }
+        return memories;
+    }
+    async recordMemories(input, options) {
+        for (const memory of this.memories) {
+            if (memory.autoUpdate) {
+                await memory.record(input, options.context);
+            }
+        }
+    }
     async invoke(input, options = {}) {
         const opts = {
             ...options,
             context: options.context ?? (await this.newDefaultContext()),
         };
+        if (options.userContext) {
+            Object.assign(opts.context.userContext, options.userContext);
+        }
         const message = typeof input === "string" ? (0, prompt_builder_js_1.createMessage)(input) : input;
         logger_js_1.logger.debug("Invoke agent %s started with input: %O", this.name, input);
         if (!this.disableEvents)
@@ -482,16 +510,12 @@ class Agent {
     async postprocess(input, output, options) {
         this.checkContextStatus(options);
         this.publishToTopics(output, options);
-        for (const memory of this.memories) {
-            if (memory.autoUpdate) {
-                await memory.record({
-                    content: [
-                        { role: "user", content: input },
-                        { role: "agent", content: (0, types_js_1.replaceTransferAgentToName)(output), source: this.name },
-                    ],
-                }, options.context);
-            }
-        }
+        await this.recordMemories({
+            content: [
+                { role: "user", content: input },
+                { role: "agent", content: (0, types_js_1.replaceTransferAgentToName)(output), source: this.name },
+            ],
+        }, options);
     }
     async publishToTopics(output, options) {
         const publishTopics = typeof this.publishTopic === "function" ? await this.publishTopic(output) : this.publishTopic;
@@ -533,7 +557,7 @@ class Agent {
      *
      * @returns Agent name
      */
-    [node_util_1.inspect.custom]() {
+    [nodejs_js_1.nodejs.customInspect]() {
         return this.name;
     }
     /**

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

@@ -1,5 +1,4 @@
 import { type ZodObject, type ZodType, z } from "zod";
-import { type DefaultMemoryOptions } from "../memory/default-memory/index.js";
 import { PromptBuilder } from "../prompt/prompt-builder.js";
 import { Agent, type AgentInvokeOptions, type AgentOptions, type AgentProcessAsyncGenerator, type Message } from "./agent.js";
 import { ChatModel, type ChatModelInput } from "./chat-model.js";
@@ -13,7 +12,7 @@ import type { GuideRailAgentOutput } from "./guide-rail-agent.js";
  * @template I The input message type the agent accepts
  * @template O The output message type the agent returns
  */
-export interface AIAgentOptions<I extends Message = Message, O extends Message = Message> extends Omit<AgentOptions<I, O>, "memory"> {
+export interface AIAgentOptions<I extends Message = Message, O extends Message = Message> extends AgentOptions<I, O> {
     /**
      * The language model to use for this agent
      *
@@ -65,7 +64,6 @@ export interface AIAgentOptions<I extends Message = Message, O extends Message =
      * The template receives a {{memories}} variable containing serialized memory content.
      */
     memoryPromptTemplate?: string;
-    memory?: AgentOptions<I, O>["memory"] | DefaultMemoryOptions | true;
 }
 /**
  * Tool choice options for AI agents

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

@@ -2,8 +2,6 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.AIAgent = exports.aiAgentOptionsSchema = exports.aiAgentToolChoiceSchema = exports.AIAgentToolChoice = void 0;
 const zod_1 = require("zod");
-const index_js_1 = require("../memory/default-memory/index.js");
-const memory_js_1 = require("../memory/memory.js");
 const prompt_builder_js_1 = require("../prompt/prompt-builder.js");
 const template_js_1 = require("../prompt/template.js");
 const type_utils_js_1 = require("../utils/type-utils.js");
@@ -101,14 +99,7 @@ class AIAgent extends agent_js_1.Agent {
      * @param options Configuration options for the AI agent
      */
     constructor(options) {
-        super({
-            ...options,
-            memory: !options.memory
-                ? undefined
-                : Array.isArray(options.memory) || options.memory instanceof memory_js_1.MemoryAgent
-                    ? options.memory
-                    : new index_js_1.DefaultMemory(options.memory === true ? {} : options.memory),
-        });
+        super(options);
         (0, type_utils_js_1.checkArguments)("AIAgent", exports.aiAgentOptionsSchema, options);
         this.model = options.model;
         this.instructions =
@@ -193,10 +184,10 @@ class AIAgent extends agent_js_1.Agent {
         if (!model)
             throw new Error("model is required to run AIAgent");
         const { toolAgents, ...modelInput } = await this.instructions.build({
+            ...options,
             agent: this,
             input,
             model,
-            context: options.context,
         });
         const toolsMap = new Map(toolAgents?.map((i) => [i.name, i]));
         if (this.toolChoice === "router") {

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

@@ -1,6 +1,6 @@
 import { Client, type ClientOptions } from "@modelcontextprotocol/sdk/client/index.js";
 import { type SSEClientTransportOptions } from "@modelcontextprotocol/sdk/client/sse.js";
-import { type StdioServerParameters } from "@modelcontextprotocol/sdk/client/stdio.js";
+import type { StdioServerParameters } from "@modelcontextprotocol/sdk/client/stdio.js";
 import { type StreamableHTTPClientTransportOptions } from "@modelcontextprotocol/sdk/client/streamableHttp.js";
 import type { RequestOptions } from "@modelcontextprotocol/sdk/shared/protocol.js";
 import type { Transport } from "@modelcontextprotocol/sdk/shared/transport.js";

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

@@ -4,15 +4,16 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.MCPResource = exports.MCPPrompt = exports.MCPTool = exports.MCPBase = exports.MCPAgent = void 0;
+const stdio_client_transport_js_1 = require("@aigne/platform-helpers/mcp/stdio-client-transport.js");
 const index_js_1 = require("@modelcontextprotocol/sdk/client/index.js");
 const sse_js_1 = require("@modelcontextprotocol/sdk/client/sse.js");
-const stdio_js_1 = require("@modelcontextprotocol/sdk/client/stdio.js");
 const streamableHttp_js_1 = require("@modelcontextprotocol/sdk/client/streamableHttp.js");
 const uriTemplate_js_1 = require("@modelcontextprotocol/sdk/shared/uriTemplate.js");
 const p_retry_1 = __importDefault(require("p-retry"));
 const zod_1 = require("zod");
 const logger_js_1 = require("../utils/logger.js");
 const mcp_utils_js_1 = require("../utils/mcp-utils.js");
+const nodejs_js_1 = require("../utils/nodejs.js");
 const type_utils_js_1 = require("../utils/type-utils.js");
 const agent_js_1 = require("./agent.js");
 const MCP_AGENT_CLIENT_NAME = "AIGNE/MCPAgent";
@@ -22,7 +23,7 @@ const DEFAULT_TIMEOUT = () => zod_1.z.coerce
     .number()
     .int()
     .min(0)
-    .safeParse(process.env.MCP_TIMEOUT || process.env.TIMEOUT).data || 60e3;
+    .safeParse(nodejs_js_1.nodejs.env.MCP_TIMEOUT || nodejs_js_1.nodejs.env.TIMEOUT).data || 60e3;
 function isSSEServerParameters(options) {
     return "url" in options && typeof options.url === "string";
 }
@@ -66,14 +67,7 @@ class MCPAgent extends agent_js_1.Agent {
             return MCPAgent.fromTransport(transport, options);
         }
         if (isStdioServerParameters(options)) {
-            const transport = () => new stdio_js_1.StdioClientTransport({
-                ...options,
-                env: {
-                    ...(0, stdio_js_1.getDefaultEnvironment)(),
-                    ...options.env,
-                },
-                stderr: "pipe",
-            });
+            const transport = async () => (0, stdio_client_transport_js_1.createStdioClientTransport)(options);
             return MCPAgent.fromTransport(transport, options);
         }
         return new MCPAgent(options);
@@ -83,7 +77,7 @@ class MCPAgent extends agent_js_1.Agent {
             name: MCP_AGENT_CLIENT_NAME,
             version: MCP_AGENT_CLIENT_VERSION,
         }, undefined, isSSEServerParameters(options) ? { transportCreator, ...options } : undefined);
-        const transport = transportCreator();
+        const transport = await transportCreator();
         logger_js_1.logger.debug(`Connecting to MCP server: ${getMCPServerString(options)}`);
         await client.connect(transport);
         const mcpServer = getMCPServerName(client);

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

@@ -1,4 +1,3 @@
-import { ReadableStream } from "node:stream/web";
 import type { Context } from "../aigne/context.js";
 import { type MessagePayload } from "../aigne/message-queue.js";
 import { Agent, type AgentInvokeOptions, type AgentOptions, type AgentProcessResult, type FunctionAgentFn, type Message } from "./agent.js";

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

@@ -1,7 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.UserAgent = void 0;
-const web_1 = require("node:stream/web");
 const message_queue_js_1 = require("../aigne/message-queue.js");
 const type_utils_js_1 = require("../utils/type-utils.js");
 const agent_js_1 = require("./agent.js");
@@ -66,7 +65,7 @@ class UserAgent extends agent_js_1.Agent {
     });
     get stream() {
         let subscriptions = [];
-        return new web_1.ReadableStream({
+        return new ReadableStream({
             start: (controller) => {
                 const subscribeTopic = (0, type_utils_js_1.orArrayToArray)(this.subscribeTopic);
                 subscriptions = subscribeTopic.map((topic) => this.subscribe(topic, (message) => {

package/lib/cjs/aigne/context.d.ts

@@ -1,4 +1,4 @@
-import EventEmitter from "node:events";
+import { Emitter } from "strict-event-emitter";
 import { Agent, type AgentInvokeOptions, type AgentProcessAsyncGenerator, type AgentResponse, type AgentResponseStream, type Message } from "../agents/agent.js";
 import type { ChatModel } from "../agents/chat-model.js";
 import { UserAgent } from "../agents/user-agent.js";
@@ -42,7 +42,6 @@ export interface InvokeOptions<U extends UserContext = UserContext> extends Part
     returnActiveAgent?: boolean;
     disableTransfer?: boolean;
     sourceAgent?: Agent;
-    userContext?: U;
 }
 /**
  * @hidden
@@ -150,7 +149,7 @@ declare class AIGNEContextShared {
         messageQueue?: MessageQueue;
     }) | undefined, overrides?: Partial<Context>);
     readonly messageQueue: MessageQueue;
-    readonly events: EventEmitter<ContextEventMap>;
+    readonly events: Emitter<any>;
     get model(): ChatModel | undefined;
     get skills(): Agent<Message, Message>[] | undefined;
     get limits(): ContextLimits | undefined;

package/lib/cjs/aigne/context.js

@@ -1,10 +1,7 @@
 "use strict";
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.AIGNEContext = void 0;
-const node_events_1 = __importDefault(require("node:events"));
+const strict_event_emitter_1 = require("strict-event-emitter");
 const uuid_1 = require("uuid");
 const zod_1 = require("zod");
 const agent_js_1 = require("../agents/agent.js");
@@ -150,7 +147,8 @@ class AIGNEContextShared {
         this.userContext = overrides?.userContext ?? {};
     }
     messageQueue;
-    events = new node_events_1.default();
+    // biome-ignore lint/suspicious/noExplicitAny: <explanation>
+    events = new strict_event_emitter_1.Emitter();
     get model() {
         return this.parent?.model;
     }

package/lib/cjs/aigne/message-queue.d.ts

@@ -1,4 +1,4 @@
-import { EventEmitter } from "node:events";
+import { Emitter, type EventMap } from "strict-event-emitter";
 import type { Message } from "../agents/agent.js";
 import type { Context } from "./context.js";
 /**
@@ -30,11 +30,14 @@ export type MessageQueueListener = (message: MessagePayload) => void;
  * @hidden
  */
 export type Unsubscribe = () => void;
+interface MessageQueueEventMap extends EventMap {
+    [key: string]: any[];
+}
 /**
  * @hidden
  */
 export declare class MessageQueue {
-    events: EventEmitter<[never]>;
+    events: Emitter<MessageQueueEventMap>;
     publish(topic: string | string[], payload: MessagePayload): void;
     error(error: Error): void;
     subscribe(topic: string | string[], listener?: undefined): Promise<MessagePayload>;
@@ -42,3 +45,4 @@ export declare class MessageQueue {
     subscribe(topic: string | string[], listener?: MessageQueueListener): Unsubscribe | Promise<MessagePayload>;
     unsubscribe(topic: string | string[], listener: MessageQueueListener): void;
 }
+export {};

package/lib/cjs/aigne/message-queue.js

@@ -2,7 +2,7 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.MessageQueue = exports.UserOutputTopic = exports.UserInputTopic = void 0;
 exports.toMessagePayload = toMessagePayload;
-const node_events_1 = require("node:events");
+const strict_event_emitter_1 = require("strict-event-emitter");
 const zod_1 = require("zod");
 const prompt_builder_js_1 = require("../prompt/prompt-builder.js");
 const type_utils_js_1 = require("../utils/type-utils.js");
@@ -40,7 +40,7 @@ function toMessagePayload(payload, options) {
  * @hidden
  */
 class MessageQueue {
-    events = new node_events_1.EventEmitter();
+    events = new strict_event_emitter_1.Emitter();
     publish(topic, payload) {
         (0, type_utils_js_1.checkArguments)("MessageQueue.publish", publishArgsSchema, {
             topic,

package/lib/cjs/index.d.ts

@@ -7,5 +7,6 @@ export * from "./agents/team-agent.js";
 export * from "./agents/types.js";
 export * from "./agents/user-agent.js";
 export * from "./aigne/index.js";
+export * from "./memory/index.js";
 export * from "./prompt/prompt-builder.js";
 export * from "./prompt/template.js";

package/lib/cjs/index.js

@@ -23,5 +23,6 @@ __exportStar(require("./agents/team-agent.js"), exports);
 __exportStar(require("./agents/types.js"), exports);
 __exportStar(require("./agents/user-agent.js"), exports);
 __exportStar(require("./aigne/index.js"), exports);
+__exportStar(require("./memory/index.js"), exports);
 __exportStar(require("./prompt/prompt-builder.js"), exports);
 __exportStar(require("./prompt/template.js"), exports);

package/lib/cjs/loader/agent-js.d.ts

@@ -1,8 +1,8 @@
 import { type ZodObject, type ZodType, z } from "zod";
 import type { Message } from "../agents/agent.js";
 export declare function loadAgentFromJsFile(path: string): Promise<{
-    name: string;
     process: (args_0: Message) => Message;
+    name: string;
     description?: string | undefined;
     inputSchema?: ZodObject<Record<string, ZodType<any, z.ZodTypeDef, any>>, z.UnknownKeysParam, z.ZodTypeAny, {
         [x: string]: any;

package/lib/cjs/loader/agent-yaml.d.ts

@@ -5,10 +5,11 @@ export declare function loadAgentFromYamlFile(path: string): Promise<{
     name: string;
     description?: string | undefined;
     skills?: string[] | undefined;
+    instructions?: string | undefined;
     memory?: true | {
+        provider: string;
         subscribeTopic?: string[] | undefined;
     } | undefined;
-    instructions?: string | undefined;
     inputSchema?: ZodObject<Record<string, ZodType<any, z.ZodTypeDef, any>>, z.UnknownKeysParam, z.ZodTypeAny, {
         [x: string]: any;
     }, {

package/lib/cjs/loader/agent-yaml.js

@@ -1,12 +1,12 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.loadAgentFromYamlFile = loadAgentFromYamlFile;
-const promises_1 = require("node:fs/promises");
 const json_schema_to_zod_1 = require("@aigne/json-schema-to-zod");
 const yaml_1 = require("yaml");
 const zod_1 = require("zod");
 const ai_agent_js_1 = require("../agents/ai-agent.js");
 const camelize_js_1 = require("../utils/camelize.js");
+const nodejs_js_1 = require("../utils/nodejs.js");
 const type_utils_js_1 = require("../utils/type-utils.js");
 const schema_js_1 = require("./schema.js");
 const agentFileSchema = zod_1.z.discriminatedUnion("type", [
@@ -43,6 +43,7 @@ const agentFileSchema = zod_1.z.discriminatedUnion("type", [
             .union([
             zod_1.z.boolean(),
             zod_1.z.object({
+                provider: zod_1.z.string(),
                 subscribe_topic: zod_1.z
                     .array(zod_1.z.string())
                     .nullish()
@@ -69,7 +70,7 @@ const agentFileSchema = zod_1.z.discriminatedUnion("type", [
     }),
 ]);
 async function loadAgentFromYamlFile(path) {
-    const raw = await (0, type_utils_js_1.tryOrThrow)(() => (0, promises_1.readFile)(path, "utf8"), (error) => new Error(`Failed to load agent definition from ${path}: ${error.message}`));
+    const raw = await (0, type_utils_js_1.tryOrThrow)(() => nodejs_js_1.nodejs.fs.readFile(path, "utf8"), (error) => new Error(`Failed to load agent definition from ${path}: ${error.message}`));
     const json = await (0, type_utils_js_1.tryOrThrow)(() => (0, yaml_1.parse)(raw), (error) => new Error(`Failed to parse agent definition from ${path}: ${error.message}`));
     const agent = (0, type_utils_js_1.tryOrThrow)(() => (0, camelize_js_1.customCamelize)(agentFileSchema.parse({
         ...json,

package/lib/cjs/loader/index.d.ts

@@ -2,6 +2,7 @@ import type { Camelize } from "camelize-ts";
 import { z } from "zod";
 import { type Agent } from "../agents/agent.js";
 import type { ChatModel, ChatModelOptions } from "../agents/chat-model.js";
+import type { MemoryAgent, MemoryAgentOptions } from "../memory/memory.js";
 export interface LoadOptions {
     models: {
         new (parameters: {
@@ -9,6 +10,9 @@ export interface LoadOptions {
             modelOptions?: ChatModelOptions;
         }): ChatModel;
     }[];
+    memories?: {
+        new (parameters?: MemoryAgentOptions): MemoryAgent;
+    }[];
     path: string;
 }
 export declare function load(options: LoadOptions): Promise<{
@@ -26,7 +30,7 @@ export declare function load(options: LoadOptions): Promise<{
         presence_penalty?: number | null | undefined;
     } | null | undefined;
 }>;
-export declare function loadAgent(path: string): Promise<Agent>;
+export declare function loadAgent(path: string, options?: LoadOptions): Promise<Agent>;
 export declare function loadModel(models: LoadOptions["models"], model?: Camelize<z.infer<typeof aigneFileSchema>["chat_model"]>, modelOptions?: ChatModelOptions): Promise<ChatModel | undefined>;
 declare const aigneFileSchema: z.ZodObject<{
     name: z.ZodOptional<z.ZodNullable<z.ZodString>>;