@aigne/core 1.22.0 → 1.23.0

package/CHANGELOG.md

@@ -1,23 +1,35 @@
-## [1.17.0](https://github.com/AIGNE-io/aigne-framework/compare/core-v1.16.0...core-v1.17.0) (2025-05-25)
+## [1.22.0](https://github.com/AIGNE-io/aigne-framework/compare/core-v1.21.0...core-v1.22.0) (2025-06-24)
 
 
 ### Features
 
-* add user context support ([#131](https://github.com/AIGNE-io/aigne-framework/issues/131)) ([4dd9d20](https://github.com/AIGNE-io/aigne-framework/commit/4dd9d20953f6ac33933723db56efd9b44bafeb02))
+* support observability for cli and blocklet ([#155](https://github.com/AIGNE-io/aigne-framework/issues/155)) ([5baa705](https://github.com/AIGNE-io/aigne-framework/commit/5baa705a33cfdba1efc5ccbe18674c27513ca97d))
 
-## [1.22.0](https://github.com/AIGNE-io/aigne-framework/compare/core-v1.21.0...core-v1.22.0) (2025-06-24)
+
+### Dependencies
+
+* The following workspace dependencies were updated
+  * dependencies
+    * @aigne/observability bumped to 0.1.0
+
+## [1.23.0](https://github.com/AIGNE-io/aigne-framework/compare/core-v1.22.0...core-v1.23.0) (2025-06-25)
 
 
 ### Features
 
-* support observability for cli and blocklet ([#155](https://github.com/AIGNE-io/aigne-framework/issues/155)) ([5baa705](https://github.com/AIGNE-io/aigne-framework/commit/5baa705a33cfdba1efc5ccbe18674c27513ca97d))
+* support remember custom fields from message ([#174](https://github.com/AIGNE-io/aigne-framework/issues/174)) ([664069d](https://github.com/AIGNE-io/aigne-framework/commit/664069d343137f69d0c103b2b5eff545ab0051fb))
+
+
+### Bug Fixes
+
+* **blocklet:** ensure only admins can access traces ([#173](https://github.com/AIGNE-io/aigne-framework/issues/173)) ([9c5cc06](https://github.com/AIGNE-io/aigne-framework/commit/9c5cc06c5841b9684d16c5c60af764d8c98c9c3e))
 
 
 ### Dependencies
 
 * The following workspace dependencies were updated
   * dependencies
-    * @aigne/observability bumped to 0.1.0
+    * @aigne/observability bumped to 0.1.1
 
 ## [1.21.0](https://github.com/AIGNE-io/aigne-framework/compare/core-v1.20.1...core-v1.21.0) (2025-06-20)
 
@@ -118,6 +130,15 @@
   * dependencies
     * @aigne/platform-helpers bumped to 0.1.0
 
+
+## [1.17.0](https://github.com/AIGNE-io/aigne-framework/compare/core-v1.16.0...core-v1.17.0) (2025-05-25)
+
+
+### Features
+
+* add user context support ([#131](https://github.com/AIGNE-io/aigne-framework/issues/131)) ([4dd9d20](https://github.com/AIGNE-io/aigne-framework/commit/4dd9d20953f6ac33933723db56efd9b44bafeb02))
+
+
 ## [1.16.0](https://github.com/AIGNE-io/aigne-framework/compare/core-v1.15.0...core-v1.16.0) (2025-05-23)
 
 
@@ -320,85 +341,3 @@
 * **MCP:** catch list resource error treat as empty list ([1885fab](https://github.com/AIGNE-io/aigne-framework/commit/1885fab3585e0dd1467b127e5b47cd0b98282bab))
 * rename @aigne/core-next to @aigne/core ([3a81009](https://github.com/AIGNE-io/aigne-framework/commit/3a8100962c81813217b687ae28e8de604419c622))
 * use text resource from MCP correctly ([8b9eba8](https://github.com/AIGNE-io/aigne-framework/commit/8b9eba83352ec096a2a5d4f410d4c4bde7420bce))
-
-## [1.2.0](https://github.com/AIGNE-io/aigne-framework/compare/core-next-v1.1.0...core-next-v1.2.0) (2025-03-18)
-
-- chore: release v1.2.0
-
-## 1.1.0-beta.17 (2025-3-18)
-
-- chore: add support for esm module
-
-
-## 1.1.0-beta.16 (2025-3-18)
-
-- chore: add puppeteer in linux need docker_container
-
-## 1.1.0-beta.15 (2025-3-18)
-
-- chore: make coverage report as text to terminal
-- chore: update contributing docs
-
-## 1.1.0-beta.14 (2025-3-18)
-
-- chore(example): add code-execution example
-
-## 1.1.0-beta.13 (2025-3-18)
-
-- feat: add OrchestratorAgent in agent library
-
-## 1.1.0-beta.12 (2025-3-14)
-
-- chore(example): add concurrency reflection handoff workflow examples
-
-## 1.1.0-beta.11 (2025-3-14)
-
-- feat(core): add sequential and parallel helper function
-- chore(examples): add workflow-sequential example
-
-## 1.1.0-beta.10 (2025-3-13)
-
-- chore: ensure required environment variables have values
-
-## 1.1.0-beta.9 (2025-3-13)
-
-- fix(MCP): catch list resource error treat as empty list
-
-## 1.1.0-beta.8 (2025-3-13)
-
-- fix(AIAgent): should pass both arguments (model generated) and input (user input) to tool
-- chore(examples): add workflow-router example
-- chore(examples): rename examples puppeteer-mcp-server and sqlite-mcp-server to mcp-server-puppeteer and mcp-server-sqlite
-
-## 1.1.0-beta.7 (2025-3-13)
-
-- chore: rename @aigne/core to @aigne/core
-
-## 1.1.0-beta.6 (2025-3-13)
-
-- chore(examples): default enable mcp debug message for examples
-
-## 1.1.0-beta.5 (2025-3-13)
-
-- feat: support chat history in PromptBuilder
-- feat: add `prompts` for MCPAgent to consume prompts from MCP server
-- chore: add sqlite-mcp-server example
-- test: add more unit test cases
-
-## 1.1.0-beta.4 (2025-3-12)
-
-- feat: support run puppeteer example chat loop in terminal
-
-## 1.1.0-beta.3 (2025-3-11)
-
-- chore: set module type for core package
-
-## 1.1.0-beta.2 (2025-3-11)
-
-- feat: use PromptBuilder instead of string instructions
-- refactor: use tools instead of skills
-- chore(examples): add puppeteer-mcp-server example
-
-## 1.1.0-beta.1 (2025-3-11)
-
-- feat: add Agent FunctionAgent AIAgent MCPAgent and ExecutionEngine

package/lib/cjs/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
      *
@@ -191,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
      *
@@ -277,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
@@ -519,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
      *
@@ -534,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
      *
@@ -546,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
@@ -560,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>;
     /**

package/lib/cjs/agents/agent.js

@@ -348,8 +348,8 @@ class Agent {
         if (!this.disableEvents)
             opts.context.emit("agentStarted", { agent: this, input });
         try {
-            await this.hooks.onStart?.({ context: opts.context, input });
-            const parsedInput = (0, type_utils_js_1.checkArguments)(`Agent ${this.name} input`, this.inputSchema, input);
+            let parsedInput = (await this.hooks.onStart?.({ context: opts.context, input }))?.input ?? input;
+            parsedInput = (0, type_utils_js_1.checkArguments)(`Agent ${this.name} input`, this.inputSchema, input);
             await this.preprocess(parsedInput, opts);
             this.checkContextStatus(opts);
             let response = await this.process(parsedInput, opts);
@@ -412,7 +412,9 @@ class Agent {
         logger_js_1.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;
     }
     /**

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

@@ -131,7 +131,7 @@ 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/cjs/agents/mcp-agent.d.ts

@@ -183,7 +183,7 @@ 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> {

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

@@ -93,15 +93,15 @@ 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.

package/lib/cjs/aigne/aigne.js

@@ -128,6 +128,7 @@ class AIGNE {
         return context;
     }
     invoke(agent, message, options) {
+        this.observer?.serve();
         const context = new context_js_1.AIGNEContext(this);
         return context.invoke(agent, message, { ...options, newContext: false });
     }
@@ -145,6 +146,7 @@ class AIGNE {
      * {@includeCode ../../test/aigne/aigne.test.ts#example-publish-message}
      */
     publish(topic, payload, options) {
+        this.observer?.serve();
         return new context_js_1.AIGNEContext(this).publish(topic, payload, options);
     }
     subscribe(topic, listener) {

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

@@ -148,7 +148,7 @@ export declare class AIGNEContext implements Context {
     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";
@@ -180,7 +180,7 @@ declare class AIGNEContextShared {
     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;

package/lib/cjs/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/cjs/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.

package/lib/cjs/memory/memory.js

@@ -5,6 +5,8 @@ const uuid_1 = require("uuid");
 const zod_1 = require("zod");
 const agent_js_1 = require("../agents/agent.js");
 const type_utils_js_1 = require("../utils/type-utils.js");
+const recorder_js_1 = require("./recorder.js");
+const retriever_js_1 = require("./retriever.js");
 const newMemoryId = () => (0, uuid_1.v7)();
 exports.newMemoryId = newMemoryId;
 /**
@@ -26,8 +28,20 @@ class MemoryAgent extends agent_js_1.Agent {
             subscribeTopic: options.subscribeTopic,
             skills: options.skills,
         });
-        this.recorder = options.recorder;
-        this.retriever = options.retriever;
+        this.recorder =
+            options.recorder instanceof recorder_js_1.MemoryRecorder
+                ? options.recorder
+                : options.recorder &&
+                    new recorder_js_1.MemoryRecorder(typeof options.recorder === "function"
+                        ? { process: options.recorder }
+                        : options.recorder);
+        this.retriever =
+            options.retriever instanceof retriever_js_1.MemoryRetriever
+                ? options.retriever
+                : options.retriever &&
+                    new retriever_js_1.MemoryRetriever(typeof options.retriever === "function"
+                        ? { process: options.retriever }
+                        : options.retriever);
         this.autoUpdate = options.autoUpdate;
     }
     _retriever;

package/lib/cjs/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,12 +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/cjs/memory/recorder.js

@@ -7,7 +7,11 @@ const agent_js_1 = require("../agents/agent.js");
  * @hidden
  */
 exports.memoryRecorderInputSchema = zod_1.z.object({
-    content: zod_1.z.array(zod_1.z.unknown()),
+    content: zod_1.z.array(zod_1.z.object({
+        role: zod_1.z.union([zod_1.z.literal("user"), zod_1.z.literal("agent")]),
+        content: zod_1.z.record(zod_1.z.string(), zod_1.z.unknown()),
+        source: zod_1.z.string().optional(),
+    })),
 });
 /**
  * @hidden
@@ -46,6 +50,14 @@ class MemoryRecorder extends agent_js_1.Agent {
             inputSchema: exports.memoryRecorderInputSchema,
             outputSchema: exports.memoryRecorderOutputSchema,
         });
+        this._process = options.process;
+    }
+    _process;
+    process(input, options) {
+        if (!this._process) {
+            throw new Error("MemoryRecorder process function is not defined.");
+        }
+        return this._process(input, options);
     }
 }
 exports.MemoryRecorder = MemoryRecorder;

package/lib/cjs/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,12 +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/cjs/memory/retriever.js

@@ -47,6 +47,14 @@ class MemoryRetriever extends agent_js_1.Agent {
             inputSchema: exports.memoryRetrieverInputSchema,
             outputSchema: exports.memoryRetrieverOutputSchema,
         });
+        this._process = options.process;
+    }
+    _process;
+    process(input, options) {
+        if (!this._process) {
+            throw new Error("MemoryRetriever process function is not implemented.");
+        }
+        return this._process(input, options);
     }
 }
 exports.MemoryRetriever = MemoryRetriever;

package/lib/dts/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
      *
@@ -191,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
      *
@@ -277,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
@@ -519,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
      *
@@ -534,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
      *
@@ -546,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
@@ -560,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>;
     /**

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

@@ -131,7 +131,7 @@ 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/dts/agents/mcp-agent.d.ts

@@ -183,7 +183,7 @@ 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> {

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

@@ -93,15 +93,15 @@ 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.

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

@@ -148,7 +148,7 @@ export declare class AIGNEContext implements Context {
     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";
@@ -180,7 +180,7 @@ declare class AIGNEContextShared {
     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;

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.

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,12 +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,12 +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
      *
@@ -191,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
      *
@@ -277,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
@@ -519,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
      *
@@ -534,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
      *
@@ -546,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
@@ -560,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>;
     /**

package/lib/esm/agents/agent.js

@@ -304,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);
@@ -368,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;
     }
     /**

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

@@ -131,7 +131,7 @@ 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/mcp-agent.d.ts

@@ -183,7 +183,7 @@ 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> {

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

@@ -93,15 +93,15 @@ 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.

package/lib/esm/aigne/aigne.js

@@ -125,6 +125,7 @@ export class AIGNE {
         return context;
     }
     invoke(agent, message, options) {
+        this.observer?.serve();
         const context = new AIGNEContext(this);
         return context.invoke(agent, message, { ...options, newContext: false });
     }
@@ -142,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) {

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

@@ -148,7 +148,7 @@ export declare class AIGNEContext implements Context {
     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";
@@ -180,7 +180,7 @@ declare class AIGNEContextShared {
     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;

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

package/lib/esm/memory/memory.js

@@ -2,6 +2,8 @@ import { v7 } from "uuid";
 import { z } from "zod";
 import { Agent, } from "../agents/agent.js";
 import { checkArguments, remove } from "../utils/type-utils.js";
+import { MemoryRecorder, } from "./recorder.js";
+import { MemoryRetriever, } from "./retriever.js";
 export const newMemoryId = () => v7();
 /**
  * A specialized agent responsible for managing, storing, and retrieving memories within the agent system.
@@ -22,8 +24,20 @@ export class MemoryAgent extends Agent {
             subscribeTopic: options.subscribeTopic,
             skills: options.skills,
         });
-        this.recorder = options.recorder;
-        this.retriever = options.retriever;
+        this.recorder =
+            options.recorder instanceof MemoryRecorder
+                ? options.recorder
+                : options.recorder &&
+                    new MemoryRecorder(typeof options.recorder === "function"
+                        ? { process: options.recorder }
+                        : options.recorder);
+        this.retriever =
+            options.retriever instanceof MemoryRetriever
+                ? options.retriever
+                : options.retriever &&
+                    new MemoryRetriever(typeof options.retriever === "function"
+                        ? { process: options.retriever }
+                        : options.retriever);
         this.autoUpdate = options.autoUpdate;
     }
     _retriever;

package/lib/esm/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,12 +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/esm/memory/recorder.js

@@ -1,10 +1,14 @@
 import { z } from "zod";
-import { Agent } from "../agents/agent.js";
+import { Agent, } from "../agents/agent.js";
 /**
  * @hidden
  */
 export const memoryRecorderInputSchema = z.object({
-    content: z.array(z.unknown()),
+    content: z.array(z.object({
+        role: z.union([z.literal("user"), z.literal("agent")]),
+        content: z.record(z.string(), z.unknown()),
+        source: z.string().optional(),
+    })),
 });
 /**
  * @hidden
@@ -43,5 +47,13 @@ export class MemoryRecorder extends Agent {
             inputSchema: memoryRecorderInputSchema,
             outputSchema: memoryRecorderOutputSchema,
         });
+        this._process = options.process;
+    }
+    _process;
+    process(input, options) {
+        if (!this._process) {
+            throw new Error("MemoryRecorder process function is not defined.");
+        }
+        return this._process(input, options);
     }
 }

package/lib/esm/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,12 +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/memory/retriever.js

@@ -1,5 +1,5 @@
 import { z } from "zod";
-import { Agent } from "../agents/agent.js";
+import { Agent, } from "../agents/agent.js";
 /**
  * @hidden
  */
@@ -44,5 +44,13 @@ export class MemoryRetriever extends Agent {
             inputSchema: memoryRetrieverInputSchema,
             outputSchema: memoryRetrieverOutputSchema,
         });
+        this._process = options.process;
+    }
+    _process;
+    process(input, options) {
+        if (!this._process) {
+            throw new Error("MemoryRetriever process function is not implemented.");
+        }
+        return this._process(input, options);
     }
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@aigne/core",
-  "version": "1.22.0",
+  "version": "1.23.0",
   "description": "AIGNE core library for building AI-powered applications",
   "publishConfig": {
     "access": "public"
@@ -81,8 +81,8 @@
     "yaml": "^2.7.1",
     "zod": "^3.24.4",
     "zod-to-json-schema": "^3.24.5",
-    "@aigne/observability": "^0.1.0",
-    "@aigne/platform-helpers": "^0.1.2"
+    "@aigne/platform-helpers": "^0.1.2",
+    "@aigne/observability": "^0.1.1"
   },
   "devDependencies": {
     "@types/bun": "^1.2.12",