@aigne/core 1.16.0 → 1.18.0

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

@@ -1,6 +1,5 @@
-import type { Context } from "../aigne/context.js";
 import { type PromiseOrValue } from "../utils/type-utils.js";
-import { Agent, type AgentOptions, type AgentProcessResult, type Message } from "./agent.js";
+import { Agent, type AgentInvokeOptions, type AgentOptions, type AgentProcessResult, type Message } from "./agent.js";
 /**
  * Defines the processing modes available for a TeamAgent.
  *
@@ -89,10 +88,10 @@ export declare class TeamAgent<I extends Message, O extends Message> extends Age
      * - In parallel mode: Process input through all agents simultaneously and combine their outputs
      *
      * @param input The message to process
-     * @param context The execution context
+     * @param options The invocation options
      * @returns A stream of message chunks that collectively form the response
      */
-    process(input: I, context: Context): PromiseOrValue<AgentProcessResult<O>>;
+    process(input: I, options: AgentInvokeOptions): PromiseOrValue<AgentProcessResult<O>>;
     /**
      * Process input sequentially through each agent in the team.
      *
@@ -103,12 +102,12 @@ export declare class TeamAgent<I extends Message, O extends Message> extends Age
      * 4. Updates the team's agent list with any changes that occurred during processing
      *
      * @param input The message to process
-     * @param context The execution context
+     * @param options The invocation options
      * @returns A stream of message chunks from all agents
      *
      * @private
      */
-    _processSequential(input: I, context: Context): PromiseOrValue<AgentProcessResult<O>>;
+    _processSequential(input: I, options: AgentInvokeOptions): PromiseOrValue<AgentProcessResult<O>>;
     /**
      * Process input in parallel through all agents in the team.
      *
@@ -118,10 +117,10 @@ export declare class TeamAgent<I extends Message, O extends Message> extends Age
      * 3. Updates the team's agent list with any changes that occurred during processing
      *
      * @param input The message to process
-     * @param context The execution context
+     * @param options The invocation options
      * @returns A stream of combined message chunks from all agents
      *
      * @private
      */
-    _processParallel(input: I, context: Context): PromiseOrValue<AgentProcessResult<O>>;
+    _processParallel(input: I, options: AgentInvokeOptions): PromiseOrValue<AgentProcessResult<O>>;
 }

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

@@ -86,15 +86,15 @@ class TeamAgent extends agent_js_1.Agent {
      * - In parallel mode: Process input through all agents simultaneously and combine their outputs
      *
      * @param input The message to process
-     * @param context The execution context
+     * @param options The invocation options
      * @returns A stream of message chunks that collectively form the response
      */
-    process(input, context) {
+    process(input, options) {
         switch (this.mode) {
             case ProcessMode.sequential:
-                return this._processSequential(input, context);
+                return this._processSequential(input, options);
             case ProcessMode.parallel:
-                return this._processParallel(input, context);
+                return this._processParallel(input, options);
         }
     }
     /**
@@ -107,18 +107,18 @@ class TeamAgent extends agent_js_1.Agent {
      * 4. Updates the team's agent list with any changes that occurred during processing
      *
      * @param input The message to process
-     * @param context The execution context
+     * @param options The invocation options
      * @returns A stream of message chunks from all agents
      *
      * @private
      */
-    async *_processSequential(input, context) {
+    async *_processSequential(input, options) {
         const output = {};
         // Clone the agents to run, so that we can update the agents list during the loop
         const agents = [...this.skills];
         const newAgents = [];
         for (const agent of agents) {
-            const [o, transferToAgent] = await context.invoke(agent, { ...input, ...output }, { returnActiveAgent: true, streaming: true });
+            const [o, transferToAgent] = await options.context.invoke(agent, { ...input, ...output }, { returnActiveAgent: true, streaming: true });
             for await (const chunk of o) {
                 yield chunk;
                 (0, stream_utils_js_1.mergeAgentResponseChunk)(output, chunk);
@@ -137,13 +137,13 @@ class TeamAgent extends agent_js_1.Agent {
      * 3. Updates the team's agent list with any changes that occurred during processing
      *
      * @param input The message to process
-     * @param context The execution context
+     * @param options The invocation options
      * @returns A stream of combined message chunks from all agents
      *
      * @private
      */
-    async *_processParallel(input, context) {
-        const result = await Promise.all(this.skills.map((agent) => context.invoke(agent, input, { returnActiveAgent: true, streaming: true })));
+    async *_processParallel(input, options) {
+        const result = await Promise.all(this.skills.map((agent) => options.context.invoke(agent, input, { returnActiveAgent: true, streaming: true })));
         const streams = result.map((i) => i[0]);
         const read = async (index, reader) => {
             const promise = reader.read();

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

@@ -1,7 +1,6 @@
-import { ReadableStream } from "node:stream/web";
 import type { Context } from "../aigne/context.js";
 import { type MessagePayload } from "../aigne/message-queue.js";
-import { Agent, type AgentOptions, type AgentProcessResult, type FunctionAgentFn, type Message } from "./agent.js";
+import { Agent, type AgentInvokeOptions, type AgentOptions, type AgentProcessResult, type FunctionAgentFn, type Message } from "./agent.js";
 export interface UserAgentOptions<I extends Message = Message, O extends Message = Message> extends AgentOptions<I, O> {
     context: Context;
     process?: FunctionAgentFn<I, O>;
@@ -14,14 +13,14 @@ export declare class UserAgent<I extends Message = Message, O extends Message =
     private _process?;
     private activeAgent?;
     protected subscribeToTopics(context: Pick<Context, "subscribe">): void;
-    protected publishToTopics(output: O, context: Context): Promise<void>;
+    protected publishToTopics(output: O, options: AgentInvokeOptions): Promise<void>;
     invoke: Agent<I, O>["invoke"];
-    process(input: I, context: Context): Promise<AgentProcessResult<O>>;
+    process(input: I, options: AgentInvokeOptions): Promise<AgentProcessResult<O>>;
     publish: Context["publish"];
     subscribe: Context["subscribe"];
     unsubscribe: Context["unsubscribe"];
     get stream(): ReadableStream<MessagePayload & {
         topic: string;
     }>;
-    protected checkAgentInvokesUsage(_context: Context): void;
+    protected checkAgentInvokesUsage(_options: AgentInvokeOptions): void;
 }

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");
@@ -22,21 +21,21 @@ class UserAgent extends agent_js_1.Agent {
         if (this._process)
             super.subscribeToTopics(context);
     }
-    async publishToTopics(output, context) {
+    async publishToTopics(output, options) {
         if (this._process)
-            super.publishToTopics(output, context);
+            super.publishToTopics(output, options);
     }
-    invoke = ((input, context, options) => {
-        if (!context)
+    invoke = ((input, options = {}) => {
+        if (!options.context)
             this.context = this.context.newContext({ reset: true });
-        return super.invoke(input, context ?? this.context, options);
+        return super.invoke(input, { ...options, context: this.context });
     });
-    async process(input, context) {
+    async process(input, options) {
         if (this._process) {
-            return this._process(input, context);
+            return this._process(input, options);
         }
         if (this.activeAgent) {
-            const [output, agent] = await context.invoke(this.activeAgent, input, {
+            const [output, agent] = await options.context.invoke(this.activeAgent, input, {
                 returnActiveAgent: true,
                 streaming: true,
             });
@@ -47,7 +46,7 @@ class UserAgent extends agent_js_1.Agent {
         }
         const publicTopic = typeof this.publishTopic === "function" ? await this.publishTopic(input) : this.publishTopic;
         if (publicTopic?.length) {
-            context.publish(publicTopic, input);
+            options.context.publish(publicTopic, input);
             if (this.subscribeTopic) {
                 return this.subscribe(this.subscribeTopic).then((res) => res.message);
             }
@@ -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) => {
@@ -80,7 +79,7 @@ class UserAgent extends agent_js_1.Agent {
             },
         });
     }
-    checkAgentInvokesUsage(_context) {
+    checkAgentInvokesUsage(_options) {
         // ignore calls usage check for UserAgent
     }
 }

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

@@ -2,7 +2,7 @@ import { Agent, type AgentResponse, type AgentResponseStream, type Message } fro
 import { ChatModel } from "../agents/chat-model.js";
 import type { UserAgent } from "../agents/user-agent.js";
 import { type LoadOptions } from "../loader/index.js";
-import { AIGNEContext, type InvokeOptions } from "./context.js";
+import { AIGNEContext, type Context, type InvokeOptions, type UserContext } from "./context.js";
 import { type MessagePayload, MessageQueue, type MessageQueueListener, type Unsubscribe } from "./message-queue.js";
 import type { ContextLimits } from "./usage.js";
 /**
@@ -46,7 +46,7 @@ export interface AIGNEOptions {
  * Here's an example of how to use AIGNE with streaming response:
  * {@includeCode ../../test/aigne/aigne.test.ts#example-streaming}
  */
-export declare class AIGNE {
+export declare class AIGNE<U extends UserContext = UserContext> {
     /**
      * Loads an AIGNE instance from a directory containing an aigne.yaml file and agent definitions.
      * This static method provides a convenient way to initialize an AIGNE system from configuration files.
@@ -111,7 +111,7 @@ export declare class AIGNE {
      *
      * @returns A new AIGNEContext instance bound to this AIGNE.
      */
-    newContext(): AIGNEContext;
+    newContext(options?: Partial<Context>): AIGNEContext;
     /**
      * Creates a user agent for consistent interactions with a specified agent.
      * This method allows you to create a wrapper around an agent for repeated invocations.
@@ -134,7 +134,7 @@ export declare class AIGNE {
      * @param options.streaming - Must be false to return a response stream
      * @returns A promise resolving to a tuple containing the agent's response and the final active agent
      */
-    invoke<I extends Message, O extends Message>(agent: Agent<I, O>, message: I | string, options: InvokeOptions & {
+    invoke<I extends Message, O extends Message>(agent: Agent<I, O>, message: I | string, options: InvokeOptions<U> & {
         returnActiveAgent: true;
         streaming?: false;
     }): Promise<[O, Agent]>;
@@ -148,7 +148,7 @@ export declare class AIGNE {
      * @param options.streaming - Must be true to return a response stream
      * @returns A promise resolving to a tuple containing the agent's response stream and a promise for the final agent
      */
-    invoke<I extends Message, O extends Message>(agent: Agent<I, O>, message: I | string, options: InvokeOptions & {
+    invoke<I extends Message, O extends Message>(agent: Agent<I, O>, message: I | string, options: InvokeOptions<U> & {
         returnActiveAgent: true;
         streaming: true;
     }): Promise<[AgentResponseStream<O>, Promise<Agent>]>;
@@ -165,7 +165,7 @@ export declare class AIGNE {
      * Here's a simple example of how to invoke an agent:
      * {@includeCode ../../test/aigne/aigne.test.ts#example-simple}
      */
-    invoke<I extends Message, O extends Message>(agent: Agent<I, O>, message: I | string, options?: InvokeOptions & {
+    invoke<I extends Message, O extends Message>(agent: Agent<I, O>, message: I | string, options?: InvokeOptions<U> & {
         returnActiveAgent?: false;
         streaming?: false;
     }): Promise<O>;
@@ -182,7 +182,7 @@ export declare class AIGNE {
      * Here's an example of how to invoke an agent with streaming response:
      * {@includeCode ../../test/aigne/aigne.test.ts#example-streaming}
      */
-    invoke<I extends Message, O extends Message>(agent: Agent<I, O>, message: I | string, options: InvokeOptions & {
+    invoke<I extends Message, O extends Message>(agent: Agent<I, O>, message: I | string, options: InvokeOptions<U> & {
         returnActiveAgent?: false;
         streaming: true;
     }): Promise<AgentResponseStream<O>>;
@@ -196,7 +196,7 @@ export declare class AIGNE {
      * @returns Either a UserAgent (when no message provided) or a promise resolving to the agent's response
      * with optional active agent information based on the provided options
      */
-    invoke<I extends Message, O extends Message>(agent: Agent<I, O>, message?: I | string, options?: InvokeOptions): UserAgent<I, O> | Promise<AgentResponse<O> | [AgentResponse<O>, Agent]>;
+    invoke<I extends Message, O extends Message>(agent: Agent<I, O>, message?: I | string, options?: InvokeOptions<U>): UserAgent<I, O> | Promise<AgentResponse<O> | [AgentResponse<O>, Agent]>;
     /**
      * Publishes a message to the message queue for inter-agent communication.
      * This method broadcasts a message to all subscribers of the specified topic(s).
@@ -204,12 +204,13 @@ export declare class AIGNE {
      *
      * @param topic - The topic or array of topics to publish the message to
      * @param payload - The message payload to be delivered to subscribers
+     * @param options - Optional configuration parameters for the publish operation
      *
      * @example
      * Here's an example of how to publish a message:
      * {@includeCode ../../test/aigne/aigne.test.ts#example-publish-message}
      */
-    publish(topic: string | string[], payload: Omit<MessagePayload, "context"> | Message | string): void;
+    publish(topic: string | string[], payload: Omit<MessagePayload, "context"> | Message | string, options?: InvokeOptions<U>): void;
     /**
      * Subscribes to receive the next message on a specific topic.
      * This overload returns a Promise that resolves with the next message published to the topic.

package/lib/cjs/aigne/aigne.js

@@ -109,8 +109,8 @@ class AIGNE {
      *
      * @returns A new AIGNEContext instance bound to this AIGNE.
      */
-    newContext() {
-        return new context_js_1.AIGNEContext(this);
+    newContext(options) {
+        return new context_js_1.AIGNEContext(this, options);
     }
     invoke(agent, message, options) {
         return new context_js_1.AIGNEContext(this).invoke(agent, message, options);
@@ -122,13 +122,14 @@ class AIGNE {
      *
      * @param topic - The topic or array of topics to publish the message to
      * @param payload - The message payload to be delivered to subscribers
+     * @param options - Optional configuration parameters for the publish operation
      *
      * @example
      * Here's an example of how to publish a message:
      * {@includeCode ../../test/aigne/aigne.test.ts#example-publish-message}
      */
-    publish(topic, payload) {
-        return new context_js_1.AIGNEContext(this).publish(topic, payload);
+    publish(topic, payload, options) {
+        return new context_js_1.AIGNEContext(this).publish(topic, payload, options);
     }
     subscribe(topic, listener) {
         return this.messageQueue.subscribe(topic, listener);

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";
@@ -38,7 +38,7 @@ export type ContextEmitEventMap = {
 /**
  * @hidden
  */
-export interface InvokeOptions extends AgentInvokeOptions {
+export interface InvokeOptions<U extends UserContext = UserContext> extends Partial<Omit<AgentInvokeOptions<U>, "context">> {
     returnActiveAgent?: boolean;
     disableTransfer?: boolean;
     sourceAgent?: Agent;
@@ -46,12 +46,18 @@ export interface InvokeOptions extends AgentInvokeOptions {
 /**
  * @hidden
  */
-export interface Context extends TypedEventEmitter<ContextEventMap, ContextEmitEventMap> {
+export interface UserContext extends Record<string, unknown> {
+}
+/**
+ * @hidden
+ */
+export interface Context<U extends UserContext = UserContext> extends TypedEventEmitter<ContextEventMap, ContextEmitEventMap> {
     model?: ChatModel;
     skills?: Agent[];
     usage: ContextUsage;
     limits?: ContextLimits;
     status?: "normal" | "timeout";
+    userContext: U;
     /**
      * Create a user agent to consistently invoke an agent
      * @param agent Agent to invoke
@@ -92,7 +98,7 @@ export interface Context extends TypedEventEmitter<ContextEventMap, ContextEmitE
      * @param topic topic name, or an array of topic names
      * @param payload message to publish
      */
-    publish(topic: string | string[], payload: Omit<MessagePayload, "context"> | Message | string): void;
+    publish(topic: string | string[], payload: Omit<MessagePayload, "context"> | Message | string, options?: InvokeOptions): void;
     subscribe(topic: string | string[], listener?: undefined): Promise<MessagePayload>;
     subscribe(topic: string | string[], listener: MessageQueueListener): Unsubscribe;
     subscribe(topic: string | string[], listener?: MessageQueueListener): Unsubscribe | Promise<MessagePayload>;
@@ -114,15 +120,17 @@ export interface Context extends TypedEventEmitter<ContextEventMap, ContextEmitE
  * @hidden
  */
 export declare class AIGNEContext implements Context {
-    constructor(parent?: ConstructorParameters<typeof AIGNEContextInternal>[0]);
+    constructor(...[parent, ...args]: ConstructorParameters<typeof AIGNEContextShared>);
     parentId?: string;
     id: string;
-    readonly internal: AIGNEContextInternal;
+    readonly internal: AIGNEContextShared;
     get model(): ChatModel | undefined;
     get skills(): Agent<Message, Message>[] | undefined;
     get limits(): ContextLimits | undefined;
     get status(): "normal" | "timeout";
     get usage(): ContextUsage;
+    get userContext(): Context["userContext"];
+    set userContext(userContext: Context["userContext"]);
     newContext({ reset }?: {
         reset?: boolean;
     }): AIGNEContext;
@@ -135,17 +143,18 @@ export declare class AIGNEContext implements Context {
     once<K extends keyof ContextEventMap>(eventName: K, listener: Listener<K, ContextEventMap>): this;
     off<K extends keyof ContextEventMap>(eventName: K, listener: Listener<K, ContextEventMap>): this;
 }
-declare class AIGNEContextInternal {
+declare class AIGNEContextShared {
     private readonly parent?;
     constructor(parent?: (Pick<Context, "model" | "skills" | "limits"> & {
         messageQueue?: MessageQueue;
-    }) | undefined);
+    }) | 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;
     usage: ContextUsage;
+    userContext: Context["userContext"];
     private abortController;
     private timer?;
     private initTimeout;

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");
@@ -19,13 +16,13 @@ const usage_js_1 = require("./usage.js");
  * @hidden
  */
 class AIGNEContext {
-    constructor(parent) {
+    constructor(...[parent, ...args]) {
         if (parent instanceof AIGNEContext) {
             this.parentId = parent.id;
             this.internal = parent.internal;
         }
         else {
-            this.internal = new AIGNEContextInternal(parent);
+            this.internal = new AIGNEContextShared(parent, ...args);
         }
     }
     parentId;
@@ -46,9 +43,15 @@ class AIGNEContext {
     get usage() {
         return this.internal.usage;
     }
+    get userContext() {
+        return this.internal.userContext;
+    }
+    set userContext(userContext) {
+        this.internal.userContext = userContext;
+    }
     newContext({ reset } = {}) {
         if (reset)
-            return new AIGNEContext(this.internal);
+            return new AIGNEContext(this, { userContext: {} });
         return new AIGNEContext(this);
     }
     invoke = ((agent, message, options) => {
@@ -57,6 +60,8 @@ class AIGNEContext {
             message,
             options,
         });
+        if (options?.userContext)
+            Object.assign(this.userContext, options.userContext);
         if ((0, type_utils_js_1.isNil)(message)) {
             return user_agent_js_1.UserAgent.from({
                 context: this,
@@ -96,7 +101,9 @@ class AIGNEContext {
             return stream;
         });
     });
-    publish = ((topic, payload) => {
+    publish = ((topic, payload, options) => {
+        if (options?.userContext)
+            Object.assign(this.userContext, options.userContext);
         return this.internal.messageQueue.publish(topic, {
             ...(0, message_queue_js_1.toMessagePayload)(payload),
             context: this,
@@ -132,14 +139,16 @@ class AIGNEContext {
     }
 }
 exports.AIGNEContext = AIGNEContext;
-class AIGNEContextInternal {
+class AIGNEContextShared {
     parent;
-    constructor(parent) {
+    constructor(parent, overrides) {
         this.parent = parent;
         this.messageQueue = this.parent?.messageQueue ?? new message_queue_js_1.MessageQueue();
+        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;
     }
@@ -150,6 +159,7 @@ class AIGNEContextInternal {
         return this.parent?.limits;
     }
     usage = (0, usage_js_1.newEmptyContextUsage)();
+    userContext;
     abortController = new AbortController();
     timer;
     initTimeout() {
@@ -176,12 +186,13 @@ class AIGNEContextInternal {
             const result = {};
             if (options?.sourceAgent && activeAgent !== options.sourceAgent) {
                 options.sourceAgent.hooks.onHandoff?.({
+                    context,
                     source: options.sourceAgent,
                     target: activeAgent,
                     input,
                 });
             }
-            const stream = await activeAgent.invoke(input, context, { streaming: true });
+            const stream = await activeAgent.invoke(input, { ...options, context, streaming: true });
             for await (const value of stream) {
                 if (value.delta.text) {
                     yield { delta: { text: value.delta.text } };

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

@@ -1,10 +1,12 @@
 export * from "./agents/agent.js";
 export * from "./agents/ai-agent.js";
+export * from "./agents/chat-model.js";
+export * from "./agents/guide-rail-agent.js";
 export * from "./agents/mcp-agent.js";
 export * from "./agents/team-agent.js";
 export * from "./agents/types.js";
 export * from "./agents/user-agent.js";
 export * from "./aigne/index.js";
-export * from "./agents/chat-model.js";
+export * from "./memory/index.js";
 export * from "./prompt/prompt-builder.js";
 export * from "./prompt/template.js";

package/lib/cjs/index.js

@@ -16,11 +16,13 @@ var __exportStar = (this && this.__exportStar) || function(m, exports) {
 Object.defineProperty(exports, "__esModule", { value: true });
 __exportStar(require("./agents/agent.js"), exports);
 __exportStar(require("./agents/ai-agent.js"), exports);
+__exportStar(require("./agents/chat-model.js"), exports);
+__exportStar(require("./agents/guide-rail-agent.js"), exports);
 __exportStar(require("./agents/mcp-agent.js"), exports);
 __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("./agents/chat-model.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;
     }, {
@@ -24,6 +25,6 @@ export declare function loadAgentFromYamlFile(path: string): Promise<{
 } | {
     type: "mcp";
     url?: string | undefined;
-    command?: string | undefined;
     args?: string[] | undefined;
+    command?: string | undefined;
 }>;

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,