@aigne/core 1.16.0 → 1.17.0

package/lib/esm/agents/chat-model.js

@@ -75,12 +75,12 @@ export class ChatModel extends Agent {
      * Primarily checks if token usage exceeds limits, throwing an exception if limits are exceeded
      *
      * @param input Input message
-     * @param context Execution context
+     * @param options Options for invoking the agent
      * @throws Error if token usage exceeds maximum limit
      */
-    async preprocess(input, context) {
-        super.preprocess(input, context);
-        const { limits, usage } = context;
+    async preprocess(input, options) {
+        super.preprocess(input, options);
+        const { limits, usage } = options.context;
         const usedTokens = usage.outputTokens + usage.inputTokens;
         if (limits?.maxTokens && usedTokens >= limits.maxTokens) {
             throw new Error(`Exceeded max tokens ${usedTokens}/${limits.maxTokens}`);
@@ -109,9 +109,9 @@ export class ChatModel extends Agent {
      *
      * @param input Input message
      * @param output Output message
-     * @param context Execution context
+     * @param options Options for invoking the agent
      */
-    postprocess(input, output, context) {
+    async postprocess(input, output, options) {
         // Restore original tool names in the output
         if (output.toolCalls?.length) {
             const toolsMap = input._toolsMap;
@@ -125,11 +125,11 @@ export class ChatModel extends Agent {
                 }
             }
         }
-        super.postprocess(input, output, context);
+        super.postprocess(input, output, options);
         const { usage } = output;
         if (usage) {
-            context.usage.outputTokens += usage.outputTokens;
-            context.usage.inputTokens += usage.inputTokens;
+            options.context.usage.outputTokens += usage.outputTokens;
+            options.context.usage.inputTokens += usage.inputTokens;
         }
     }
 }

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

@@ -6,9 +6,8 @@ import type { RequestOptions } from "@modelcontextprotocol/sdk/shared/protocol.j
 import type { Transport } from "@modelcontextprotocol/sdk/shared/transport.js";
 import type { CallToolResult, GetPromptResult, Implementation, ReadResourceResult, Request } from "@modelcontextprotocol/sdk/types.js";
 import { type ZodType, z } from "zod";
-import type { Context } from "../aigne/context.js";
 import { type PromiseOrValue } from "../utils/type-utils.js";
-import { Agent, type AgentOptions, type Message } from "./agent.js";
+import { Agent, type AgentInvokeOptions, type AgentOptions, type Message } from "./agent.js";
 export interface MCPAgentOptions extends AgentOptions {
     client: Client;
     prompts?: MCPPrompt[];
@@ -150,10 +149,10 @@ export declare class MCPAgent extends Agent {
      * throws an error if called.
      *
      * @param _input Input message (unused)
-     * @param _context Execution context (unused)
+     * @param _options AgentInvokeOptions (unused)
      * @throws Error This method always throws an error since MCPAgent is not directly invokable
      */
-    process(_input: Message, _context?: Context): Promise<Message>;
+    process(_input: Message, _options: AgentInvokeOptions): Promise<Message>;
     /**
      * Shut down the agent and close the MCP connection.
      *

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

@@ -177,10 +177,10 @@ export class MCPAgent extends Agent {
      * throws an error if called.
      *
      * @param _input Input message (unused)
-     * @param _context Execution context (unused)
+     * @param _options AgentInvokeOptions (unused)
      * @throws Error This method always throws an error since MCPAgent is not directly invokable
      */
-    async process(_input, _context) {
+    async process(_input, _options) {
         throw new Error("Method not implemented.");
     }
     /**

package/lib/esm/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/esm/agents/team-agent.js

@@ -83,15 +83,15 @@ export class TeamAgent extends 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);
         }
     }
     /**
@@ -104,18 +104,18 @@ export class TeamAgent extends 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;
                 mergeAgentResponseChunk(output, chunk);
@@ -134,13 +134,13 @@ export class TeamAgent extends 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/esm/agents/user-agent.d.ts

@@ -1,7 +1,7 @@
 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 +14,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/esm/agents/user-agent.js

@@ -19,21 +19,21 @@ export class UserAgent extends 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,
             });
@@ -44,7 +44,7 @@ export class UserAgent extends 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);
             }
@@ -77,7 +77,7 @@ export class UserAgent extends Agent {
             },
         });
     }
-    checkAgentInvokesUsage(_context) {
+    checkAgentInvokesUsage(_options) {
         // ignore calls usage check for UserAgent
     }
 }

package/lib/esm/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/esm/aigne/aigne.js

@@ -106,8 +106,8 @@ export class AIGNE {
      *
      * @returns A new AIGNEContext instance bound to this AIGNE.
      */
-    newContext() {
-        return new AIGNEContext(this);
+    newContext(options) {
+        return new AIGNEContext(this, options);
     }
     invoke(agent, message, options) {
         return new AIGNEContext(this).invoke(agent, message, options);
@@ -119,13 +119,14 @@ export 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 AIGNEContext(this).publish(topic, payload);
+    publish(topic, payload, options) {
+        return new AIGNEContext(this).publish(topic, payload, options);
     }
     subscribe(topic, listener) {
         return this.messageQueue.subscribe(topic, listener);

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

@@ -38,20 +38,27 @@ 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;
+    userContext?: U;
 }
 /**
  * @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 +99,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 +121,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 +144,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>;
     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/esm/aigne/context.js

@@ -2,7 +2,7 @@ import EventEmitter from "node:events";
 import { v7 } from "uuid";
 import { z } from "zod";
 import { Agent, } from "../agents/agent.js";
-import { isTransferAgentOutput, transferAgentOutputKey } from "../agents/types.js";
+import { isTransferAgentOutput, transferAgentOutputKey, } from "../agents/types.js";
 import { UserAgent } from "../agents/user-agent.js";
 import { createMessage } from "../prompt/prompt-builder.js";
 import { agentResponseStreamToObject, asyncGeneratorToReadableStream, onAgentResponseStreamEnd, } from "../utils/stream-utils.js";
@@ -13,13 +13,13 @@ import { newEmptyContextUsage } from "./usage.js";
  * @hidden
  */
 export 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;
@@ -40,9 +40,15 @@ export 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) => {
@@ -51,6 +57,8 @@ export class AIGNEContext {
             message,
             options,
         });
+        if (options?.userContext)
+            Object.assign(this.userContext, options.userContext);
         if (isNil(message)) {
             return UserAgent.from({
                 context: this,
@@ -90,7 +98,9 @@ export 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, {
             ...toMessagePayload(payload),
             context: this,
@@ -125,11 +135,12 @@ export class AIGNEContext {
         return this;
     }
 }
-class AIGNEContextInternal {
+class AIGNEContextShared {
     parent;
-    constructor(parent) {
+    constructor(parent, overrides) {
         this.parent = parent;
         this.messageQueue = this.parent?.messageQueue ?? new MessageQueue();
+        this.userContext = overrides?.userContext ?? {};
     }
     messageQueue;
     events = new EventEmitter();
@@ -143,6 +154,7 @@ class AIGNEContextInternal {
         return this.parent?.limits;
     }
     usage = newEmptyContextUsage();
+    userContext;
     abortController = new AbortController();
     timer;
     initTimeout() {
@@ -169,12 +181,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/esm/index.d.ts

@@ -1,10 +1,11 @@
 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 "./prompt/prompt-builder.js";
 export * from "./prompt/template.js";

package/lib/esm/index.js

@@ -1,10 +1,11 @@
 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 "./prompt/prompt-builder.js";
 export * from "./prompt/template.js";

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

@@ -24,6 +24,6 @@ export declare function loadAgentFromYamlFile(path: string): Promise<{
 } | {
     type: "mcp";
     url?: string | undefined;
-    command?: string | undefined;
     args?: string[] | undefined;
+    command?: string | undefined;
 }>;

package/lib/esm/memory/default-memory/default-memory-storage/index.d.ts

@@ -0,0 +1,30 @@
+import type { AgentInvokeOptions } from "../../../agents/agent.js";
+import "sqlite3";
+import { Sequelize } from "sequelize";
+import type { Context } from "../../../aigne/context.js";
+import type { PromiseOrValue } from "../../../utils/type-utils.js";
+import type { Memory } from "../../memory.js";
+import { MemoryStorage } from "../storage.js";
+export interface DefaultMemoryStorageOptions {
+    path?: string;
+    getSessionId?: (context: Context) => PromiseOrValue<string>;
+}
+export declare class DefaultMemoryStorage extends MemoryStorage {
+    options?: DefaultMemoryStorageOptions | undefined;
+    memories: {
+        [groupId: string]: Memory[];
+    };
+    constructor(options?: DefaultMemoryStorageOptions | undefined);
+    private _models?;
+    private get models();
+    private convertMemory;
+    search(query: {
+        limit?: number;
+    }, { context }: AgentInvokeOptions): Promise<{
+        result: Memory[];
+    }>;
+    create(memory: Pick<Memory, "content">, { context }: AgentInvokeOptions): Promise<{
+        result: Memory;
+    }>;
+}
+export declare function initSequelize(path?: string): Sequelize;