@aigne/core 1.21.0 → 1.23.0

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";
@@ -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 skills(): Agent<any, any>[] | 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 skills(): Agent<any, any>[] | undefined;
+    get observer(): AIGNEObserver | undefined;
     get limits(): ContextLimits | undefined;
     usage: ContextUsage;
     userContext: Context["userContext"];

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

@@ -4,8 +4,8 @@ import { ProcessMode } from "../agents/team-agent.js";
 export declare function loadAgentFromYamlFile(path: string): Promise<{
     type: "mcp";
     url?: string | undefined;
-    args?: string[] | undefined;
     command?: string | undefined;
+    args?: string[] | undefined;
 } | {
     type: "team";
     name: string;

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

@@ -17,8 +17,8 @@ export interface LoadOptions {
 }
 export declare function load(options: LoadOptions): Promise<{
     model: ChatModel | undefined;
-    agents: Agent<import("../agents/agent.js").Message, import("../agents/agent.js").Message>[];
-    skills: Agent<import("../agents/agent.js").Message, import("../agents/agent.js").Message>[];
+    agents: Agent<any, any>[];
+    skills: Agent<any, any>[];
     name?: string | null | undefined;
     description?: string | null | undefined;
     chat_model?: {

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

@@ -1,8 +1,8 @@
 import { Agent, type AgentInvokeOptions, type AgentOptions, type Message } from "../agents/agent.js";
 import type { Context } from "../aigne/context.js";
 import type { MessagePayload } from "../aigne/message-queue.js";
-import type { MemoryRecorder, MemoryRecorderInput, MemoryRecorderOutput } from "./recorder.js";
-import type { MemoryRetriever, MemoryRetrieverInput, MemoryRetrieverOutput } from "./retriever.js";
+import { MemoryRecorder, type MemoryRecorderInput, type MemoryRecorderOptions, type MemoryRecorderOutput } from "./recorder.js";
+import { MemoryRetriever, type MemoryRetrieverInput, type MemoryRetrieverOptions, type MemoryRetrieverOutput } from "./retriever.js";
 export interface Memory {
     id: string;
     sessionId?: string | null;
@@ -10,7 +10,9 @@ export interface Memory {
     createdAt: string;
 }
 export declare const newMemoryId: () => string;
-export interface MemoryAgentOptions extends Partial<Pick<MemoryAgent, "recorder" | "retriever" | "autoUpdate">>, Pick<AgentOptions, "subscribeTopic" | "skills"> {
+export interface MemoryAgentOptions extends Partial<Pick<MemoryAgent, "autoUpdate">>, Pick<AgentOptions, "subscribeTopic" | "skills"> {
+    recorder?: MemoryRecorder | MemoryRecorderOptions["process"] | MemoryRecorderOptions;
+    retriever?: MemoryRetriever | MemoryRetrieverOptions["process"] | MemoryRetrieverOptions;
 }
 /**
  * A specialized agent responsible for managing, storing, and retrieving memories within the agent system.
@@ -21,6 +23,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

@@ -1,5 +1,6 @@
 import { type ZodType, z } from "zod";
-import { Agent, type AgentOptions, type Message } from "../agents/agent.js";
+import { Agent, type AgentInvokeOptions, type AgentOptions, type AgentProcessResult, type FunctionAgentFn, type Message } from "../agents/agent.js";
+import type { PromiseOrValue } from "../utils/type-utils.js";
 import type { Memory } from "./memory.js";
 /**
  * Input for memory recording operations.
@@ -13,7 +14,11 @@ export interface MemoryRecorderInput extends Message {
      * Array of content items to record as memories.
      * Each item in this array will typically be converted into a separate memory entry.
      */
-    content: unknown[];
+    content: {
+        role: "user" | "agent";
+        content: Message;
+        source?: string;
+    }[];
 }
 /**
  * @hidden
@@ -62,6 +67,9 @@ export declare const memoryRecorderOutputSchema: z.ZodObject<{
         createdAt: string;
     }[];
 }>;
+export interface MemoryRecorderOptions extends Omit<AgentOptions<MemoryRecorderInput, MemoryRecorderOutput>, "inputSchema" | "outputSchema"> {
+    process?: FunctionAgentFn<MemoryRecorderInput, MemoryRecorderOutput>;
+}
 /**
  * Abstract base class for agents that record and store memories.
  *
@@ -76,11 +84,14 @@ export declare const memoryRecorderOutputSchema: z.ZodObject<{
  * Custom implementations should extend this class and provide concrete
  * implementations of the process method to handle the actual storage logic.
  */
-export declare abstract class MemoryRecorder extends Agent<MemoryRecorderInput, MemoryRecorderOutput> {
+export declare class MemoryRecorder extends Agent<MemoryRecorderInput, MemoryRecorderOutput> {
+    tag: string;
     /**
      * Creates a new MemoryRecorder instance with predefined input and output schemas.
      *
      * @param options - Configuration options for the memory recorder agent
      */
-    constructor(options: Omit<AgentOptions<MemoryRecorderInput, MemoryRecorderOutput>, "inputSchema" | "outputSchema">);
+    constructor(options: MemoryRecorderOptions);
+    private _process?;
+    process(input: MemoryRecorderInput, options: AgentInvokeOptions): PromiseOrValue<AgentProcessResult<MemoryRecorderOutput>>;
 }

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

@@ -1,5 +1,6 @@
 import { z } from "zod";
-import { Agent, type AgentOptions, type Message } from "../agents/agent.js";
+import { Agent, type AgentInvokeOptions, type AgentOptions, type AgentProcessResult, type FunctionAgentFn, type Message } from "../agents/agent.js";
+import type { PromiseOrValue } from "../utils/type-utils.js";
 import type { Memory } from "./memory.js";
 /**
  * Input for memory retrieval operations.
@@ -39,11 +40,11 @@ export declare const memoryRetrieverInputSchema: z.ZodObject<{
     limit: z.ZodOptional<z.ZodNumber>;
     search: z.ZodOptional<z.ZodString>;
 }, "strip", z.ZodTypeAny, {
-    limit?: number | undefined;
     search?: string | undefined;
-}, {
     limit?: number | undefined;
+}, {
     search?: string | undefined;
+    limit?: number | undefined;
 }>;
 /**
  * @hidden
@@ -75,6 +76,9 @@ export declare const memoryRetrieverOutputSchema: z.ZodObject<{
         createdAt: string;
     }[];
 }>;
+export interface MemoryRetrieverOptions extends Omit<AgentOptions<MemoryRetrieverInput, MemoryRetrieverOutput>, "inputSchema" | "outputSchema"> {
+    process?: FunctionAgentFn<MemoryRetrieverInput, MemoryRetrieverOutput>;
+}
 /**
  * Abstract base class for agents that retrieve memories from storage.
  *
@@ -89,11 +93,14 @@ export declare const memoryRetrieverOutputSchema: z.ZodObject<{
  * Custom implementations should extend this class and provide concrete
  * implementations of the process method to handle the actual retrieval logic.
  */
-export declare abstract class MemoryRetriever extends Agent<MemoryRetrieverInput, MemoryRetrieverOutput> {
+export declare class MemoryRetriever extends Agent<MemoryRetrieverInput, MemoryRetrieverOutput> {
+    tag: string;
     /**
      * Creates a new MemoryRetriever instance with predefined input and output schemas.
      *
      * @param options - Configuration options for the memory retriever agent
      */
-    constructor(options: Omit<AgentOptions<MemoryRetrieverInput, MemoryRetrieverOutput>, "inputSchema" | "outputSchema">);
+    constructor(options: MemoryRetrieverOptions);
+    private _process?;
+    process(input: MemoryRetrieverInput, options: AgentInvokeOptions): PromiseOrValue<AgentProcessResult<MemoryRetrieverOutput>>;
 }

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

@@ -37,7 +37,7 @@ export type PublishTopic<O extends Message> = string | string[] | ((output: O) =
  * @template I The agent input message type
  * @template O The agent output message type
  */
-export interface AgentOptions<I extends Message = Message, O extends Message = Message> extends Partial<Pick<Agent, "guideRails" | "hooks">> {
+export interface AgentOptions<I extends Message = Message, O extends Message = Message> extends Partial<Pick<Agent, "guideRails">> {
     /**
      * Topics the agent should subscribe to
      *
@@ -106,6 +106,7 @@ export interface AgentOptions<I extends Message = Message, O extends Message = M
      * Maximum number of memory items to retrieve
      */
     maxRetrieveMemoryCount?: number;
+    hooks?: AgentHooks<I, O>;
 }
 export declare const agentOptionsSchema: ZodObject<{
     [key in keyof AgentOptions]: ZodType<AgentOptions[key]>;
@@ -161,7 +162,7 @@ export interface AgentInvokeOptions<U extends UserContext = UserContext> {
  * Here's an example of how to create a custom agent:
  * {@includeCode ../../test/agents/agent.test.ts#example-custom-agent}
  */
-export declare abstract class Agent<I extends Message = Message, O extends Message = Message> {
+export declare abstract class Agent<I extends Message = any, O extends Message = any> {
     /**
      * Custom object inspection behavior
      *
@@ -176,6 +177,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
      */
@@ -190,7 +192,7 @@ export declare abstract class Agent<I extends Message = Message, O extends Messa
      * Here's an example of using hooks:
      * {@includeCode ../../test/agents/agent.test.ts#example-agent-hooks}
      */
-    readonly hooks: AgentHooks;
+    readonly hooks: AgentHooks<I, O>;
     /**
      * List of GuideRail agents applied to this agent
      *
@@ -276,8 +278,8 @@ export declare abstract class Agent<I extends Message = Message, O extends Messa
      * Skills can be accessed by name or by array index, allowing
      * the agent to delegate tasks to specialized sub-agents
      */
-    readonly skills: Agent<Message, Message>[] & {
-        [key: string]: Agent<Message, Message>;
+    readonly skills: Agent<any, any>[] & {
+        [key: string]: Agent<any, any>;
     };
     /**
      * Whether to disable emitting events for agent actions
@@ -518,7 +520,9 @@ export interface AgentHooks<I extends Message = Message, O extends Message = Mes
     onStart?: (event: {
         context: Context;
         input: I;
-    }) => PromiseOrValue<void>;
+    }) => PromiseOrValue<void | {
+        input?: I;
+    }>;
     /**
      * Called when agent processing completes or fails
      *
@@ -533,7 +537,9 @@ export interface AgentHooks<I extends Message = Message, O extends Message = Mes
         input: I;
         output: O;
         error: Error;
-    }, "output", "error">) => PromiseOrValue<void>;
+    }, "output", "error">) => PromiseOrValue<void | {
+        output?: O;
+    }>;
     /**
      * Called before a skill (sub-agent) is invoked
      *
@@ -545,7 +551,7 @@ export interface AgentHooks<I extends Message = Message, O extends Message = Mes
     onSkillStart?: (event: {
         context: Context;
         skill: Agent;
-        input: I;
+        input: Message;
     }) => PromiseOrValue<void>;
     /**
      * Called after a skill (sub-agent) completes or fails
@@ -559,8 +565,8 @@ export interface AgentHooks<I extends Message = Message, O extends Message = Mes
     onSkillEnd?: (event: XOr<{
         context: Context;
         skill: Agent;
-        input: I;
-        output: O;
+        input: Message;
+        output: Message;
         error: Error;
     }, "output", "error">) => PromiseOrValue<void>;
     /**
@@ -723,6 +729,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
      */
@@ -303,8 +304,8 @@ export class Agent {
         if (!this.disableEvents)
             opts.context.emit("agentStarted", { agent: this, input });
         try {
-            await this.hooks.onStart?.({ context: opts.context, input });
-            const parsedInput = checkArguments(`Agent ${this.name} input`, this.inputSchema, input);
+            let parsedInput = (await this.hooks.onStart?.({ context: opts.context, input }))?.input ?? input;
+            parsedInput = checkArguments(`Agent ${this.name} input`, this.inputSchema, input);
             await this.preprocess(parsedInput, opts);
             this.checkContextStatus(opts);
             let response = await this.process(parsedInput, opts);
@@ -367,7 +368,9 @@ export class Agent {
         logger.debug("Invoke agent %s succeed with output: %O", this.name, finalOutput);
         if (!this.disableEvents)
             context.emit("agentSucceed", { agent: this, output: finalOutput });
-        await this.hooks.onEnd?.({ context, input, output: finalOutput });
+        const o = (await this.hooks.onEnd?.({ context, input, output: finalOutput }))?.output;
+        if (o)
+            return o;
         return finalOutput;
     }
     /**
@@ -596,6 +599,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

@@ -131,7 +131,8 @@ export declare const aiAgentOptionsSchema: ZodObject<{
  * Basic AIAgent creation:
  * {@includeCode ../../test/agents/ai-agent.test.ts#example-ai-agent-basic}
  */
-export declare class AIAgent<I extends Message = Message, O extends Message = Message> extends Agent<I, O> {
+export declare class AIAgent<I extends Message = any, O extends Message = any> extends Agent<I, O> {
+    tag: string;
     /**
      * Create an AIAgent with the specified options
      *

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

@@ -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
      *

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.
      *
@@ -182,26 +183,30 @@ declare class ClientWithReconnect extends Client {
     private reconnect;
     request<T extends ZodType<object>>(request: Request, resultSchema: T, options?: RequestOptions): Promise<z.infer<T>>;
 }
-export interface MCPBaseOptions<I extends Message = Message, O extends Message = Message> extends AgentOptions<I, O> {
+export interface MCPBaseOptions<I extends Message = any, O extends Message = any> extends AgentOptions<I, O> {
     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;

package/lib/esm/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.
@@ -88,16 +93,20 @@ export declare class AIGNE<U extends UserContext = UserContext> {
      * Collection of skill agents available to this AIGNE instance.
      * Provides indexed access by skill name.
      */
-    readonly skills: Agent<Message, Message>[] & {
-        [key: string]: Agent<Message, Message>;
+    readonly skills: Agent<any, any>[] & {
+        [key: string]: Agent<any, any>;
     };
     /**
      * Collection of primary agents managed by this AIGNE instance.
      * Provides indexed access by agent name.
      */
-    readonly agents: Agent<Message, Message>[] & {
-        [key: string]: Agent<Message, Message>;
+    readonly agents: Agent<any, any>[] & {
+        [key: string]: Agent<any, any>;
     };
+    /**
+     * 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.

package/lib/esm/aigne/aigne.js

@@ -1,3 +1,4 @@
+import { AIGNEObserver } from "@aigne/observability";
 import { z } from "zod";
 import { Agent, } from "../agents/agent.js";
 import { ChatModel } from "../agents/chat-model.js";
@@ -49,10 +50,15 @@ export class AIGNE {
         this.description = options?.description;
         this.model = options?.model;
         this.limits = options?.limits;
+        this.observer =
+            process.env.AIGNE_OBSERVABILITY_DISABLED === "true"
+                ? undefined
+                : (options?.observer ?? new AIGNEObserver());
         if (options?.skills?.length)
             this.skills.push(...options.skills);
         if (options?.agents?.length)
             this.addAgent(...options.agents);
+        this.observer?.serve();
         this.initProcessExitHandler();
     }
     /**
@@ -87,6 +93,10 @@ export class AIGNE {
      * Provides indexed access by agent name.
      */
     agents = createAccessorArray([], (arr, name) => arr.find((i) => i.name === name));
+    /**
+     * Observer for the AIGNE instance.
+     */
+    observer;
     /**
      * 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.
@@ -107,10 +117,17 @@ export class AIGNE {
      * @returns A new AIGNEContext instance bound to this AIGNE.
      */
     newContext(options) {
-        return new AIGNEContext(this, options);
+        const context = new AIGNEContext(this);
+        if (options?.userContext)
+            context.userContext = options.userContext;
+        if (options?.memories)
+            context.memories = options.memories;
+        return context;
     }
     invoke(agent, message, options) {
-        return new AIGNEContext(this).invoke(agent, message, options);
+        this.observer?.serve();
+        const context = new AIGNEContext(this);
+        return context.invoke(agent, message, { ...options, newContext: false });
     }
     /**
      * Publishes a message to the message queue for inter-agent communication.
@@ -126,6 +143,7 @@ export class AIGNE {
      * {@includeCode ../../test/aigne/aigne.test.ts#example-publish-message}
      */
     publish(topic, payload, options) {
+        this.observer?.serve();
         return new AIGNEContext(this).publish(topic, payload, options);
     }
     subscribe(topic, listener) {
@@ -188,5 +206,6 @@ const aigneOptionsSchema = z.object({
     model: z.instanceof(ChatModel).optional(),
     skills: z.array(z.instanceof(Agent)).optional(),
     agents: z.array(z.instanceof(Agent)).optional(),
+    observer: z.instanceof(AIGNEObserver).optional(),
 });
 const aigneAddAgentArgsSchema = z.array(z.instanceof(Agent));