@aigne/core 1.20.1 → 1.22.0

package/lib/dts/aigne/aigne.d.ts

@@ -1,3 +1,4 @@
+import { AIGNEObserver } from "@aigne/observability";
 import { Agent, type AgentResponse, type AgentResponseStream, type Message } from "../agents/agent.js";
 import { ChatModel } from "../agents/chat-model.js";
 import type { UserAgent } from "../agents/user-agent.js";
@@ -33,6 +34,10 @@ export interface AIGNEOptions {
      * Limits for the AIGNE instance, such as timeout, max tokens, max invocations, etc.
      */
     limits?: ContextLimits;
+    /**
+     * Observer for the AIGNE instance.
+     */
+    observer?: AIGNEObserver;
 }
 /**
  * AIGNE is a class that orchestrates multiple agents to build complex AI applications.
@@ -98,6 +103,10 @@ export declare class AIGNE<U extends UserContext = UserContext> {
     readonly agents: Agent<Message, Message>[] & {
         [key: string]: Agent<Message, Message>;
     };
+    /**
+     * Observer for the AIGNE instance.
+     */
+    readonly observer?: AIGNEObserver;
     /**
      * Adds one or more agents to this AIGNE instance.
      * Each agent is attached to this AIGNE instance, allowing it to access the AIGNE's resources.
@@ -111,7 +120,7 @@ export declare class AIGNE<U extends UserContext = UserContext> {
      *
      * @returns A new AIGNEContext instance bound to this AIGNE.
      */
-    newContext(options?: Partial<Context>): AIGNEContext;
+    newContext(options?: Partial<Pick<Context, "userContext" | "memories">>): 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 +143,7 @@ export declare class AIGNE<U extends UserContext = UserContext> {
      * @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, options: InvokeOptions<U> & {
+    invoke<I extends Message, O extends Message>(agent: Agent<I, O>, message: I & Message, options: InvokeOptions<U> & {
         returnActiveAgent: true;
         streaming?: false;
     }): Promise<[O, Agent]>;
@@ -148,7 +157,7 @@ export declare class AIGNE<U extends UserContext = UserContext> {
      * @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, options: InvokeOptions<U> & {
+    invoke<I extends Message, O extends Message>(agent: Agent<I, O>, message: I & Message, options: InvokeOptions<U> & {
         returnActiveAgent: true;
         streaming: true;
     }): Promise<[AgentResponseStream<O>, Promise<Agent>]>;
@@ -165,7 +174,7 @@ export declare class AIGNE<U extends UserContext = UserContext> {
      * 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, options?: InvokeOptions<U> & {
+    invoke<I extends Message, O extends Message>(agent: Agent<I, O>, message: I & Message, options?: InvokeOptions<U> & {
         returnActiveAgent?: false;
         streaming?: false;
     }): Promise<O>;
@@ -182,7 +191,7 @@ export declare class AIGNE<U extends UserContext = UserContext> {
      * 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, options: InvokeOptions<U> & {
+    invoke<I extends Message, O extends Message>(agent: Agent<I, O>, message: I & Message, options: InvokeOptions<U> & {
         returnActiveAgent?: false;
         streaming: true;
     }): Promise<AgentResponseStream<O>>;
@@ -196,7 +205,7 @@ export declare class AIGNE<U extends UserContext = UserContext> {
      * @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, options?: InvokeOptions<U>): UserAgent<I, O> | Promise<AgentResponse<O> | [AgentResponse<O>, Agent]>;
+    invoke<I extends Message, O extends Message>(agent: Agent<I, O>, message?: I & Message, 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).

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

@@ -1,3 +1,5 @@
+import type { AIGNEObserver } from "@aigne/observability";
+import type { Span } from "@opentelemetry/api";
 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";
@@ -45,6 +47,13 @@ export interface InvokeOptions<U extends UserContext = UserContext> extends Part
     returnMetadata?: boolean;
     disableTransfer?: boolean;
     sourceAgent?: Agent;
+    /**
+     * Whether to create a new context for this invocation.
+     * If false, the invocation will use the current context.
+     *
+     * @default true
+     */
+    newContext?: boolean;
 }
 /**
  * @hidden
@@ -57,8 +66,11 @@ export interface UserContext extends Record<string, unknown> {
 export interface Context<U extends UserContext = UserContext> extends TypedEventEmitter<ContextEventMap, ContextEmitEventMap> {
     id: string;
     parentId?: string;
+    rootId: string;
     model?: ChatModel;
     skills?: Agent[];
+    observer?: AIGNEObserver;
+    span?: Span;
     usage: ContextUsage;
     limits?: ContextLimits;
     status?: "normal" | "timeout";
@@ -78,11 +90,11 @@ export interface Context<U extends UserContext = UserContext> extends TypedEvent
      * @param options.streaming return a stream of the output
      * @returns the output of the agent and the final active agent
      */
-    invoke<I extends Message, O extends Message>(agent: Agent<I, O>, message: I, options: InvokeOptions & {
+    invoke<I extends Message, O extends Message>(agent: Agent<I, O>, message: I & Message, options: InvokeOptions & {
         returnActiveAgent: true;
         streaming?: false;
     }): Promise<[O, Agent]>;
-    invoke<I extends Message, O extends Message>(agent: Agent<I, O>, message: I, options: InvokeOptions & {
+    invoke<I extends Message, O extends Message>(agent: Agent<I, O>, message: I & Message, options: InvokeOptions & {
         returnActiveAgent: true;
         streaming: true;
     }): Promise<[AgentResponseStream<O>, Promise<Agent>]>;
@@ -92,13 +104,13 @@ export interface Context<U extends UserContext = UserContext> extends TypedEvent
      * @param message Message to pass to the agent
      * @returns the output of the agent
      */
-    invoke<I extends Message, O extends Message>(agent: Agent<I, O>, message: I, options?: InvokeOptions & {
+    invoke<I extends Message, O extends Message>(agent: Agent<I, O>, message: I & Message, options?: InvokeOptions & {
         streaming?: false;
     }): Promise<O>;
-    invoke<I extends Message, O extends Message>(agent: Agent<I, O>, message: I, options: InvokeOptions & {
+    invoke<I extends Message, O extends Message>(agent: Agent<I, O>, message: I & Message, options: InvokeOptions & {
         streaming: true;
     }): Promise<AgentResponseStream<O>>;
-    invoke<I extends Message, O extends Message>(agent: Agent<I, O>, message?: I, options?: InvokeOptions): UserAgent<I, O> | Promise<AgentResponse<O> | [AgentResponse<O>, Agent]>;
+    invoke<I extends Message, O extends Message>(agent: Agent<I, O>, message?: I & Message, options?: InvokeOptions): UserAgent<I, O> | Promise<AgentResponse<O> | [AgentResponse<O>, Agent]>;
     /**
      * Publish a message to a topic, the aigne will invoke the listeners of the topic
      * @param topic topic name, or an array of topic names
@@ -126,12 +138,18 @@ export interface Context<U extends UserContext = UserContext> extends TypedEvent
  * @hidden
  */
 export declare class AIGNEContext implements Context {
-    constructor(...[parent, ...args]: ConstructorParameters<typeof AIGNEContextShared>);
-    parentId?: string;
+    constructor(parent?: ConstructorParameters<typeof AIGNEContextShared>[0], { reset }?: {
+        reset?: boolean;
+    });
     id: string;
+    parentId?: string;
+    rootId: string;
+    span?: Span;
     readonly internal: AIGNEContextShared;
+    get messageQueue(): MessageQueue;
     get model(): ChatModel | undefined;
     get skills(): Agent<Message, Message>[] | undefined;
+    get observer(): AIGNEObserver | undefined;
     get limits(): ContextLimits | undefined;
     get status(): "normal" | "timeout";
     get usage(): ContextUsage;
@@ -148,19 +166,22 @@ export declare class AIGNEContext implements Context {
     subscribe: Context["subscribe"];
     unsubscribe: Context["unsubscribe"];
     emit<K extends keyof ContextEmitEventMap>(eventName: K, ...args: Args<K, ContextEmitEventMap>): boolean;
+    private trace;
     on<K extends keyof ContextEventMap>(eventName: K, listener: Listener<K, ContextEventMap>): this;
     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 AIGNEContextShared {
     private readonly parent?;
-    constructor(parent?: (Pick<Context, "model" | "skills" | "limits"> & {
+    span?: Span;
+    constructor(parent?: (Pick<Context, "model" | "skills" | "limits" | "observer"> & {
         messageQueue?: MessageQueue;
-    }) | undefined, overrides?: Partial<Context>);
+    }) | undefined);
     readonly messageQueue: MessageQueue;
     readonly events: Emitter<any>;
     get model(): ChatModel | undefined;
     get skills(): Agent<Message, Message>[] | undefined;
+    get observer(): AIGNEObserver | undefined;
     get limits(): ContextLimits | undefined;
     usage: ContextUsage;
     userContext: Context["userContext"];
@@ -169,7 +190,7 @@ declare class AIGNEContextShared {
     private timer?;
     private initTimeout;
     get status(): "normal" | "timeout";
-    invoke<I extends Message, O extends Message>(agent: Agent<I, O>, input: I, context: Context, options?: InvokeOptions): AgentProcessAsyncGenerator<O & {
+    invoke<I extends Message, O extends Message>(agent: Agent<I, O>, input: I & Message, context: Context, options?: InvokeOptions): AgentProcessAsyncGenerator<O & {
         __activeAgent__: Agent;
     }>;
     private invokeAgent;

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

@@ -1,6 +1,6 @@
 import { type ZodObject, type ZodType, z } from "zod";
-import type { Message } from "../agents/agent.js";
-export declare function loadAgentFromJsFile(path: string): Promise<{
+import { Agent, type Message } from "../agents/agent.js";
+export declare function loadAgentFromJsFile(path: string): Promise<Agent<any, any> | {
     process: (args_0: Message) => Message;
     name: string;
     description?: string | undefined;

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

@@ -1,6 +1,29 @@
 import { type ZodObject, type ZodType, z } from "zod";
 import { AIAgentToolChoice } from "../agents/ai-agent.js";
+import { ProcessMode } from "../agents/team-agent.js";
 export declare function loadAgentFromYamlFile(path: string): Promise<{
+    type: "mcp";
+    url?: string | undefined;
+    command?: string | undefined;
+    args?: string[] | undefined;
+} | {
+    type: "team";
+    name: string;
+    description?: string | undefined;
+    skills?: string[] | undefined;
+    mode?: ProcessMode | undefined;
+    inputSchema?: ZodObject<Record<string, ZodType<any, z.ZodTypeDef, any>>, z.UnknownKeysParam, z.ZodTypeAny, {
+        [x: string]: any;
+    }, {
+        [x: string]: any;
+    }> | undefined;
+    outputSchema?: ZodObject<Record<string, ZodType<any, z.ZodTypeDef, any>>, z.UnknownKeysParam, z.ZodTypeAny, {
+        [x: string]: any;
+    }, {
+        [x: string]: any;
+    }> | undefined;
+} | {
+    instructions: string | undefined;
     type: "ai";
     name: string;
     description?: string | undefined;
@@ -9,7 +32,6 @@ export declare function loadAgentFromYamlFile(path: string): Promise<{
         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;
     }, {
@@ -23,9 +45,4 @@ export declare function loadAgentFromYamlFile(path: string): Promise<{
     inputKey?: string | undefined;
     outputKey?: string | undefined;
     toolChoice?: AIAgentToolChoice | undefined;
-} | {
-    type: "mcp";
-    url?: string | undefined;
-    args?: string[] | undefined;
-    command?: string | undefined;
 }>;

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

@@ -1,6 +1,6 @@
 import type { Camelize } from "camelize-ts";
 import { z } from "zod";
-import { type Agent } from "../agents/agent.js";
+import { 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 {

package/lib/dts/memory/memory.d.ts

@@ -21,6 +21,7 @@ export interface MemoryAgentOptions extends Partial<Pick<MemoryAgent, "recorder"
  * instead provides memory management capabilities to the system.
  */
 export declare class MemoryAgent extends Agent {
+    tag: string;
     /**
      * Creates a new MemoryAgent instance.
      */

package/lib/dts/memory/recorder.d.ts

@@ -77,6 +77,7 @@ export declare const memoryRecorderOutputSchema: z.ZodObject<{
  * implementations of the process method to handle the actual storage logic.
  */
 export declare abstract class MemoryRecorder extends Agent<MemoryRecorderInput, MemoryRecorderOutput> {
+    tag: string;
     /**
      * Creates a new MemoryRecorder instance with predefined input and output schemas.
      *

package/lib/dts/memory/retriever.d.ts

@@ -90,6 +90,7 @@ export declare const memoryRetrieverOutputSchema: z.ZodObject<{
  * implementations of the process method to handle the actual retrieval logic.
  */
 export declare abstract class MemoryRetriever extends Agent<MemoryRetrieverInput, MemoryRetrieverOutput> {
+    tag: string;
     /**
      * Creates a new MemoryRetriever instance with predefined input and output schemas.
      *

package/lib/dts/prompt/prompt-builder.d.ts

@@ -7,7 +7,7 @@ export interface PromptBuilderOptions {
     instructions?: string | ChatMessagesTemplate;
 }
 export interface PromptBuildOptions extends Pick<AgentInvokeOptions, "context"> {
-    agent?: AIAgent<any, any, any>;
+    agent?: AIAgent;
     input?: Message;
     model?: ChatModel;
     outputSchema?: Agent["outputSchema"];

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

@@ -176,6 +176,7 @@ export declare abstract class Agent<I extends Message = Message, O extends Messa
      * List of memories this agent can use
      */
     readonly memories: MemoryAgent[];
+    tag?: string;
     /**
      * Maximum number of memory items to retrieve
      */
@@ -341,7 +342,7 @@ export declare abstract class Agent<I extends Message = Message, O extends Messa
      * Here's an example of invoking an agent with regular mode:
      * {@includeCode ../../test/agents/agent.test.ts#example-invoke}
      */
-    invoke(input: I, options?: Partial<AgentInvokeOptions> & {
+    invoke(input: I & Message, options?: Partial<AgentInvokeOptions> & {
         streaming?: false;
     }): Promise<O>;
     /**
@@ -359,7 +360,7 @@ export declare abstract class Agent<I extends Message = Message, O extends Messa
      * Here's an example of invoking an agent with streaming response:
      * {@includeCode ../../test/agents/agent.test.ts#example-invoke-streaming}
      */
-    invoke(input: I, options: Partial<AgentInvokeOptions> & {
+    invoke(input: I & Message, options: Partial<AgentInvokeOptions> & {
         streaming: true;
     }): Promise<AgentResponseStream<O>>;
     /**
@@ -371,8 +372,8 @@ export declare abstract class Agent<I extends Message = Message, O extends Messa
      * @param options Invocation options
      * @returns Agent response (streaming or regular)
      */
-    invoke(input: I, options?: Partial<AgentInvokeOptions>): Promise<AgentResponse<O>>;
-    protected invokeSkill<I extends Message, O extends Message>(skill: Agent<I, O>, input: I, options: AgentInvokeOptions): Promise<O>;
+    invoke(input: I & Message, options?: Partial<AgentInvokeOptions>): Promise<AgentResponse<O>>;
+    protected invokeSkill<I extends Message, O extends Message>(skill: Agent<I, O>, input: I & Message, options: AgentInvokeOptions): Promise<O>;
     /**
      * Process agent output
      *
@@ -723,6 +724,7 @@ export interface FunctionAgentOptions<I extends Message = Message, O extends Mes
  * {@includeCode ../../test/agents/agent.test.ts#example-function-agent}
  */
 export declare class FunctionAgent<I extends Message = Message, O extends Message = Message> extends Agent<I, O> {
+    tag: string;
     /**
      * Create a function agent from a function or options
      *

package/lib/esm/agents/agent.js

@@ -83,6 +83,7 @@ export class Agent {
      * List of memories this agent can use
      */
     memories = [];
+    tag;
     /**
      * Maximum number of memory items to retrieve
      */
@@ -596,6 +597,7 @@ function checkAgentInputOutputSchema(schema) {
  * {@includeCode ../../test/agents/agent.test.ts#example-function-agent}
  */
 export class FunctionAgent extends Agent {
+    tag = "FunctionAgent";
     /**
      * Create a function agent from a function or options
      *

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

@@ -3,6 +3,7 @@ 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";
 import type { GuideRailAgentOutput } from "./guide-rail-agent.js";
+export declare const DEFAULT_OUTPUT_KEY = "message";
 /**
  * Configuration options for an AI Agent
  *
@@ -12,7 +13,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<InputKey extends string = string, I extends Message & InputMessage<InputKey> = Message & InputMessage<InputKey>, O extends Message = Message> extends AgentOptions<Omit<I, InputKey> & Partial<InputMessage<InputKey>>, O> {
+export interface AIAgentOptions<I extends Message = Message, O extends Message = Message> extends AgentOptions<I, O> {
     /**
      * The language model to use for this agent
      *
@@ -29,7 +30,7 @@ export interface AIAgentOptions<InputKey extends string = string, I extends Mess
     /**
      * Pick a message from input to use as the user's message
      */
-    inputKey?: InputKey;
+    inputKey?: string;
     /**
      * Custom key to use for text output in the response
      *
@@ -110,9 +111,6 @@ export declare const aiAgentToolChoiceSchema: z.ZodUnion<[z.ZodNativeEnum<typeof
 export declare const aiAgentOptionsSchema: ZodObject<{
     [key in keyof AIAgentOptions]: ZodType<AIAgentOptions[key]>;
 }>;
-type InputMessage<K> = K extends string ? {
-    [key in K]: string;
-} : Message;
 /**
  * AI-powered agent that leverages language models
  *
@@ -133,7 +131,8 @@ type InputMessage<K> = K extends string ? {
  * Basic AIAgent creation:
  * {@includeCode ../../test/agents/ai-agent.test.ts#example-ai-agent-basic}
  */
-export declare class AIAgent<InputKey extends string = string, I extends Message & InputMessage<InputKey> = Message & InputMessage<InputKey>, O extends Message = Message> extends Agent<I, O> {
+export declare class AIAgent<I extends Message = Message, O extends Message = Message> extends Agent<I, O> {
+    tag: string;
     /**
      * Create an AIAgent with the specified options
      *
@@ -146,13 +145,13 @@ export declare class AIAgent<InputKey extends string = string, I extends Message
      * AI agent with custom instructions:
      * {@includeCode ../../test/agents/ai-agent.test.ts#example-ai-agent-instructions}
      */
-    static from<InputKey extends string, I extends Message & InputMessage<InputKey>, O extends Message>(options: AIAgentOptions<InputKey, I, O>): AIAgent<InputKey, I, O>;
+    static from<I extends Message, O extends Message>(options: AIAgentOptions<I, O>): AIAgent<I, O>;
     /**
      * Create an AIAgent instance
      *
      * @param options Configuration options for the AI agent
      */
-    constructor(options: AIAgentOptions<InputKey, I, O>);
+    constructor(options: AIAgentOptions<I, O>);
     /**
      * The language model used by this agent
      *
@@ -173,7 +172,7 @@ export declare class AIAgent<InputKey extends string = string, I extends Message
     /**
      * Pick a message from input to use as the user's message
      */
-    inputKey?: InputKey;
+    inputKey?: string;
     /**
      * Custom key to use for text output in the response
      *
@@ -235,4 +234,3 @@ export declare class AIAgent<InputKey extends string = string, I extends Message
      */
     _processRouter(input: I, model: ChatModel, modelInput: ChatModelInput, options: AgentInvokeOptions, toolsMap: Map<string, Agent>): AgentProcessAsyncGenerator<O>;
 }
-export {};

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

@@ -5,7 +5,7 @@ import { checkArguments, isEmpty } from "../utils/type-utils.js";
 import { Agent, agentOptionsSchema, isAgentResponseDelta, } from "./agent.js";
 import { ChatModel, } from "./chat-model.js";
 import { isTransferAgentOutput } from "./types.js";
-const DEFAULT_OUTPUT_KEY = "message";
+export const DEFAULT_OUTPUT_KEY = "message";
 /**
  * Tool choice options for AI agents
  *
@@ -77,6 +77,7 @@ export const aiAgentOptionsSchema = agentOptionsSchema.extend({
  * {@includeCode ../../test/agents/ai-agent.test.ts#example-ai-agent-basic}
  */
 export class AIAgent extends Agent {
+    tag = "AIAgent";
     /**
      * Create an AIAgent with the specified options
      *
@@ -98,7 +99,7 @@ export class AIAgent extends Agent {
      * @param options Configuration options for the AI agent
      */
     constructor(options) {
-        super({ ...options, inputSchema: options.inputSchema });
+        super(options);
         checkArguments("AIAgent", aiAgentOptionsSchema, options);
         this.model = options.model;
         this.instructions =

package/lib/esm/agents/chat-model.d.ts

@@ -24,6 +24,7 @@ import { Agent, type AgentInvokeOptions, type AgentProcessResult, type Message }
  * {@includeCode ../../test/agents/chat-model.test.ts#example-chat-model-tools}
  */
 export declare abstract class ChatModel extends Agent<ChatModelInput, ChatModelOutput> {
+    tag: string;
     constructor();
     /**
      * Indicates whether the model supports parallel tool calls

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

@@ -24,6 +24,7 @@ import { Agent } from "./agent.js";
  * {@includeCode ../../test/agents/chat-model.test.ts#example-chat-model-tools}
  */
 export class ChatModel extends Agent {
+    tag = "ChatModelAgent";
     constructor() {
         super({
             inputSchema: chatModelInputSchema,

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

@@ -57,6 +57,7 @@ export type SSEServerParameters = {
  * {@includeCode ../../test/agents/mcp-agent.test.ts#example-mcp-agent-from-sse}
  */
 export declare class MCPAgent extends Agent {
+    tag: string;
     /**
      * Create an MCPAgent from a connection to an SSE server.
      *
@@ -186,22 +187,26 @@ export interface MCPBaseOptions<I extends Message = Message, O extends Message =
     client: ClientWithReconnect;
 }
 export declare abstract class MCPBase<I extends Message, O extends Message> extends Agent<I, O> {
+    tag: string;
     constructor(options: MCPBaseOptions<I, O>);
     protected client: ClientWithReconnect;
 }
 export declare class MCPTool extends MCPBase<Message, CallToolResult> {
+    tag: string;
     process(input: Message): Promise<CallToolResult>;
 }
 export interface MCPPromptInput extends Record<string, unknown> {
     [key: string]: string;
 }
 export declare class MCPPrompt extends MCPBase<MCPPromptInput, GetPromptResult> {
+    tag: string;
     process(input: MCPPromptInput): Promise<GetPromptResult>;
 }
 export interface MCPResourceOptions extends MCPBaseOptions<MCPPromptInput, ReadResourceResult> {
     uri: string;
 }
 export declare class MCPResource extends MCPBase<MCPPromptInput, ReadResourceResult> {
+    tag: string;
     constructor(options: MCPResourceOptions);
     uri: string;
     process(input: MCPPromptInput): Promise<ReadResourceResult>;

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

@@ -49,6 +49,7 @@ function getMCPServerString(options) {
  * {@includeCode ../../test/agents/mcp-agent.test.ts#example-mcp-agent-from-sse}
  */
 export class MCPAgent extends Agent {
+    tag = "MCPAgent";
     static from(options) {
         checkArguments("MCPAgent.from", mcpAgentOptionsSchema, options);
         if (isSSEServerParameters(options)) {
@@ -244,6 +245,7 @@ class ClientWithReconnect extends Client {
     }
 }
 export class MCPBase extends Agent {
+    tag = "MCPBase";
     constructor(options) {
         super(options);
         this.client = options.client;
@@ -251,18 +253,21 @@ export class MCPBase extends Agent {
     client;
 }
 export class MCPTool extends MCPBase {
+    tag = "MCPTool";
     async process(input) {
         const result = await this.client.callTool({ name: this.name, arguments: input });
         return result;
     }
 }
 export class MCPPrompt extends MCPBase {
+    tag = "MCPPrompt";
     async process(input) {
         const result = await this.client.getPrompt({ name: this.name, arguments: input });
         return result;
     }
 }
 export class MCPResource extends MCPBase {
+    tag = "MCPResource";
     constructor(options) {
         super(options);
         this.uri = options.uri;

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

@@ -52,6 +52,7 @@ export interface TeamAgentOptions<I extends Message, O extends Message> extends
  * {@includeCode ../../test/agents/team-agent.test.ts#example-team-agent-sequential}
  */
 export declare class TeamAgent<I extends Message, O extends Message> extends Agent<I, O> {
+    tag: string;
     /**
      * Create a TeamAgent from the provided options.
      *

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

@@ -42,6 +42,7 @@ export var ProcessMode;
  * {@includeCode ../../test/agents/team-agent.test.ts#example-team-agent-sequential}
  */
 export class TeamAgent extends Agent {
+    tag = "TeamAgent";
     /**
      * Create a TeamAgent from the provided options.
      *

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

@@ -7,6 +7,7 @@ export interface UserAgentOptions<I extends Message = Message, O extends Message
     activeAgent?: Agent;
 }
 export declare class UserAgent<I extends Message = Message, O extends Message = Message> extends Agent<I, O> {
+    tag: string;
     static from<I extends Message, O extends Message>(options: UserAgentOptions<I, O>): UserAgent<I, O>;
     constructor(options: UserAgentOptions<I, O>);
     context: Context;

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

@@ -2,6 +2,7 @@ import { toMessagePayload } from "../aigne/message-queue.js";
 import { orArrayToArray } from "../utils/type-utils.js";
 import { Agent, } from "./agent.js";
 export class UserAgent extends Agent {
+    tag = "UserAgent";
     static from(options) {
         return new UserAgent(options);
     }
@@ -23,9 +24,8 @@ export class UserAgent extends Agent {
             super.publishToTopics(output, options);
     }
     invoke = ((input, options = {}) => {
-        if (!options.context)
-            this.context = this.context.newContext({ reset: true });
-        return super.invoke(input, { ...options, context: this.context });
+        options.context ??= this.context.newContext({ reset: true });
+        return super.invoke(input, options);
     });
     async process(input, options) {
         if (this._process) {
@@ -35,6 +35,9 @@ export class UserAgent extends Agent {
             const [output, agent] = await options.context.invoke(this.activeAgent, input, {
                 returnActiveAgent: true,
                 streaming: true,
+                // Do not create a new context for the nested agent invocation,
+                // We are resetting the context in the override invoke method
+                newContext: false,
             });
             agent.then((agent) => {
                 this.activeAgent = agent;