@aigne/core 1.71.0-beta.1 → 1.71.0-beta.4

package/CHANGELOG.md

@@ -1,5 +1,48 @@
 # Changelog
 
+## [1.71.0-beta.4](https://github.com/AIGNE-io/aigne-framework/compare/core-v1.71.0-beta.3...core-v1.71.0-beta.4) (2025-12-10)
+
+
+### Bug Fixes
+
+* bump version ([af04b69](https://github.com/AIGNE-io/aigne-framework/commit/af04b6931951afa35d52065430acc7fef4b10087))
+* **core:** support load third agent in skills ([#819](https://github.com/AIGNE-io/aigne-framework/issues/819)) ([bcbb140](https://github.com/AIGNE-io/aigne-framework/commit/bcbb1404d2fe9c709d99a8c28883b21dd107a844))
+
+
+### Dependencies
+
+* The following workspace dependencies were updated
+  * dependencies
+    * @aigne/afs bumped to 1.3.0-beta.2
+    * @aigne/afs-history bumped to 1.1.3-beta.2
+    * @aigne/observability-api bumped to 0.11.13-beta.1
+    * @aigne/platform-helpers bumped to 0.6.6-beta
+
+## [1.71.0-beta.3](https://github.com/AIGNE-io/aigne-framework/compare/core-v1.71.0-beta.2...core-v1.71.0-beta.3) (2025-12-10)
+
+
+### Bug Fixes
+
+* **afs:** add case-sensitive option for search with case-insensitive default ([#814](https://github.com/AIGNE-io/aigne-framework/issues/814)) ([9dc9446](https://github.com/AIGNE-io/aigne-framework/commit/9dc944635104fc311e7756b4bde0a20275cfe8ec))
+* **core:** disable Immer autofreeze to return mutable response objects ([#817](https://github.com/AIGNE-io/aigne-framework/issues/817)) ([a3d0651](https://github.com/AIGNE-io/aigne-framework/commit/a3d06512cdadb9d85f92b7e8d2fd85b4f35a804b))
+* **observability:** improve trace shutdown handling and exit status  ([#813](https://github.com/AIGNE-io/aigne-framework/issues/813)) ([6215f13](https://github.com/AIGNE-io/aigne-framework/commit/6215f13243b23103c1793a4559798f0e90722384))
+
+
+### Dependencies
+
+* The following workspace dependencies were updated
+  * dependencies
+    * @aigne/afs bumped to 1.3.0-beta.1
+    * @aigne/afs-history bumped to 1.1.3-beta.1
+    * @aigne/observability-api bumped to 0.11.13-beta
+
+## [1.71.0-beta.2](https://github.com/AIGNE-io/aigne-framework/compare/core-v1.71.0-beta.1...core-v1.71.0-beta.2) (2025-12-09)
+
+
+### Bug Fixes
+
+* **orchestrator:** enhance prompts with detailed guidance ([#811](https://github.com/AIGNE-io/aigne-framework/issues/811)) ([5656f38](https://github.com/AIGNE-io/aigne-framework/commit/5656f38c09e458e18b90e962a5e85c96755be2e4))
+
 ## [1.71.0-beta.1](https://github.com/AIGNE-io/aigne-framework/compare/core-v1.71.0-beta...core-v1.71.0-beta.1) (2025-12-08)
 
 

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

@@ -118,6 +118,7 @@ export interface AgentOptions<I extends Message = Message, O extends Message = M
      * agentSucceed, or agentFailed
      */
     disableEvents?: boolean;
+    historyConfig?: Agent["historyConfig"];
     /**
      * One or more memory agents this agent can use
      */
@@ -346,6 +347,11 @@ export declare abstract class Agent<I extends Message = any, O extends Message =
      * agentSucceed, or agentFailed
      */
     private disableEvents?;
+    historyConfig?: {
+        disabled?: boolean;
+        maxTokens?: number;
+        maxItems?: number;
+    };
     private subscriptions;
     /**
      * Attach agent to context:

package/lib/cjs/agents/agent.js

@@ -151,6 +151,7 @@ class Agent {
         if (options.skills?.length)
             this.skills.push(...options.skills.map(functionToAgent));
         this.disableEvents = options.disableEvents;
+        this.historyConfig = options.historyConfig;
         if (Array.isArray(options.memory)) {
             this.memories.push(...options.memory);
         }
@@ -324,6 +325,7 @@ class Agent {
      * agentSucceed, or agentFailed
      */
     disableEvents;
+    historyConfig;
     subscriptions = [];
     /**
      * Attach agent to context:
@@ -567,7 +569,8 @@ class Agent {
         const o = await this.callHooks(["onSuccess", "onEnd"], { input, output: finalOutput }, options);
         if (o?.output)
             finalOutput = o.output;
-        this.afs?.emit("agentSucceed", { input, output: finalOutput });
+        if (this.historyConfig?.disabled !== true)
+            this.afs?.emit("agentSucceed", { input, output: finalOutput });
         if (!this.disableEvents)
             context.emit("agentSucceed", { agent: this, output: finalOutput });
         return finalOutput;

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

@@ -113,6 +113,10 @@ export declare class AIGNE<U extends UserContext = UserContext> {
      * @hidden
      */
     readonly messageQueue: MessageQueue;
+    /**
+     * Collection of all context IDs created by this AIGNE instance.
+     */
+    readonly contextIds: Set<string>;
     /**
      * Collection of skill agents available to this AIGNE instance.
      * Provides indexed access by skill name.
@@ -324,7 +328,8 @@ export declare class AIGNE<U extends UserContext = UserContext> {
     [Symbol.asyncDispose](): Promise<void>;
     /**
      * Initializes handlers for process exit events to ensure clean shutdown.
-     * This registers handlers for SIGINT and exit events to properly terminate all agents.
+     * This registers handlers for SIGINT/SIGTERM to properly terminate all agents.
+     * Note: 'exit' event cannot run async code, so we handle cleanup in signal handlers.
      */
     private initProcessExitHandler;
 }

package/lib/cjs/aigne/aigne.js

@@ -99,6 +99,10 @@ class AIGNE {
      * @hidden
      */
     messageQueue = new message_queue_js_1.MessageQueue();
+    /**
+     * Collection of all context IDs created by this AIGNE instance.
+     */
+    contextIds = new Set();
     /**
      * Collection of skill agents available to this AIGNE instance.
      * Provides indexed access by skill name.
@@ -200,7 +204,7 @@ class AIGNE {
      */
     async shutdown() {
         // Close observer first to flush any pending traces
-        await this.observer?.close();
+        await this.observer?.close(Array.from(this.contextIds));
         for (const tool of this.skills) {
             await tool.shutdown();
         }
@@ -220,12 +224,17 @@ class AIGNE {
     }
     /**
      * Initializes handlers for process exit events to ensure clean shutdown.
-     * This registers handlers for SIGINT and exit events to properly terminate all agents.
+     * This registers handlers for SIGINT/SIGTERM to properly terminate all agents.
+     * Note: 'exit' event cannot run async code, so we handle cleanup in signal handlers.
      */
     initProcessExitHandler() {
-        const shutdownAndExit = () => this.shutdown();
+        const originalExit = process.exit;
+        // @ts-ignore
+        process.exit = (...args) => {
+            this.shutdown().finally(() => originalExit(...args));
+        };
+        const shutdownAndExit = () => this.shutdown().finally(() => originalExit(0));
         process.on("SIGINT", shutdownAndExit);
-        process.on("exit", shutdownAndExit);
     }
 }
 exports.AIGNE = AIGNE;

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

@@ -149,7 +149,6 @@ export interface Context<U extends UserContext = UserContext> extends TypedEvent
  * @hidden
  */
 export declare class AIGNEContext implements Context {
-    static exitCount: number;
     constructor(parent?: ConstructorParameters<typeof AIGNEContextShared>[0], { reset }?: {
         reset?: boolean;
     });
@@ -183,7 +182,6 @@ export declare class AIGNEContext implements Context {
     subscribe: Context["subscribe"];
     unsubscribe: Context["unsubscribe"];
     emit<K extends keyof ContextEmitEventMap>(eventName: K, ...args: Args<K, ContextEmitEventMap>): boolean;
-    initProcessExitHandler(): void;
     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;
@@ -194,6 +192,7 @@ declare class AIGNEContextShared {
     constructor(parent?: (Pick<Context, "model" | "imageModel" | "agents" | "skills" | "limits" | "observer" | "metadata"> & {
         messageQueue?: MessageQueue;
         events?: Emitter<any>;
+        contextIds?: Set<string>;
     }) | undefined);
     readonly messageQueue: MessageQueue;
     readonly events: Emitter<any>;
@@ -205,6 +204,7 @@ declare class AIGNEContextShared {
     get metadata(): AIGNEMetadata | undefined;
     get limits(): ContextLimits | undefined;
     usage: ContextUsage;
+    contextIds: Set<string>;
     userContext: Context["userContext"];
     memories: Context["memories"];
     hooks: AgentHooks[];

package/lib/cjs/aigne/context.js

@@ -23,7 +23,6 @@ const usage_js_1 = require("./usage.js");
  * @hidden
  */
 class AIGNEContext {
-    static exitCount = 0;
     constructor(parent, { reset } = {}) {
         const tracer = parent?.observer?.tracer;
         if (parent instanceof AIGNEContext && !reset) {
@@ -47,7 +46,7 @@ class AIGNEContext {
             this.rootId = this.span?.spanContext?.().traceId ?? (0, uuid_1.v7)();
         }
         this.id = this.span?.spanContext()?.spanId ?? (0, uuid_1.v7)();
-        this.initProcessExitHandler();
+        this.internal.contextIds.add(this.id);
     }
     id;
     parentId;
@@ -203,26 +202,6 @@ class AIGNEContext {
         this.trace(eventName, args, b);
         return this.internal.events.emit(eventName, ...newArgs);
     }
-    initProcessExitHandler() {
-        if (this.parentId)
-            return;
-        AIGNEContext.exitCount++;
-        process.once("SIGINT", async () => {
-            try {
-                if (process.env.AIGNE_OBSERVABILITY_DISABLED)
-                    return;
-                await this.observer?.update(this.id, {
-                    status: { code: api_1.SpanStatusCode.ERROR, message: "SIGINT" },
-                });
-            }
-            finally {
-                AIGNEContext.exitCount--;
-                if (AIGNEContext.exitCount === 0) {
-                    process.exit(0);
-                }
-            }
-        });
-    }
     async trace(eventName, args, b) {
         if (process.env.AIGNE_OBSERVABILITY_DISABLED)
             return;
@@ -258,12 +237,14 @@ class AIGNEContext {
                     await this.observer?.update(this.id, { output });
                     span.setStatus({ code: api_1.SpanStatusCode.OK });
                     span.end();
+                    this.internal.contextIds.delete(this.id);
                     break;
                 }
                 case "agentFailed": {
                     const { error } = args[0];
                     span.setStatus({ code: api_1.SpanStatusCode.ERROR, message: error.message });
                     span.end();
+                    this.internal.contextIds.delete(this.id);
                     break;
                 }
             }
@@ -292,6 +273,7 @@ class AIGNEContextShared {
         this.parent = parent;
         this.messageQueue = this.parent?.messageQueue ?? new message_queue_js_1.MessageQueue();
         this.events = this.parent?.events ?? new strict_event_emitter_1.Emitter();
+        this.contextIds = this.parent?.contextIds ?? new Set();
     }
     messageQueue;
     events;
@@ -317,6 +299,7 @@ class AIGNEContextShared {
         return this.parent?.limits;
     }
     usage = (0, usage_js_1.newEmptyContextUsage)();
+    contextIds;
     userContext = {};
     memories = [];
     hooks = [];

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

@@ -1 +1,2 @@
-export declare function loadAgentFromJsFile(path: string): Promise<import("../index.js").Agent<any, any> | import("./agent-yaml.js").AgentSchema>;
+import type { LoadOptions } from "./index.js";
+export declare function loadAgentFromJsFile(path: string, options: LoadOptions): Promise<import("../index.js").Agent<any, any> | import("./agent-yaml.js").AgentSchema>;

package/lib/cjs/loader/agent-js.js

@@ -7,7 +7,7 @@ const type_utils_js_1 = require("../utils/type-utils.js");
 const agent_yaml_js_1 = require("./agent-yaml.js");
 const error_js_1 = require("./error.js");
 const importFn = new Function("path", "return import(path)");
-async function loadAgentFromJsFile(path) {
+async function loadAgentFromJsFile(path, options) {
     const url = index_js_1.nodejs.path.isAbsolute(path) ? index_js_1.nodejs.url.pathToFileURL(path).toString() : path;
     const { default: agent } = await (0, type_utils_js_1.tryOrThrow)(() => importFn(url), (error) => new error_js_1.LoadJsAgentError(`Failed to load agent definition from ${url}: ${error.message}`));
     if ((0, agent_utils_js_1.isAgent)(agent))
@@ -17,5 +17,5 @@ async function loadAgentFromJsFile(path) {
         process: agent,
         name: agent.agent_name || agent.agentName || agent.name,
         ...agent,
-    }), (error) => new Error(`Failed to parse agent from ${path}: ${error.message}`));
+    }, options), (error) => new Error(`Failed to parse agent from ${path}: ${error.message}`));
 }

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

@@ -101,13 +101,14 @@ export interface ThirdAgentSchema extends BaseAgentSchema {
     [key: string]: any;
 }
 export type AgentSchema = AIAgentSchema | ImageAgentSchema | MCPAgentSchema | TeamAgentSchema | TransformAgentSchema | FunctionAgentSchema | ThirdAgentSchema;
-export declare function parseAgentFile(path: string, data: any): Promise<AgentSchema>;
-export declare function loadAgentFromYamlFile(path: string, options?: LoadOptions): Promise<AgentSchema>;
+export declare function parseAgentFile(path: string, data: any, options: LoadOptions): Promise<AgentSchema>;
+export declare function loadAgentFromYamlFile(path: string, options: LoadOptions): Promise<AgentSchema>;
 export declare const getInstructionsSchema: ({ filepath }: {
     filepath: string;
 }) => ZodType<Instructions>;
-export declare const getAgentSchema: ({ filepath }: {
+export declare const getAgentSchema: ({ filepath, options, }: {
     filepath: string;
+    options?: LoadOptions;
 }) => ZodType<AgentSchema, z.ZodTypeDef, AgentSchema>;
 export declare const getNestAgentSchema: ({ filepath, }: {
     filepath: string;

package/lib/cjs/loader/agent-yaml.js

@@ -13,8 +13,8 @@ const team_agent_js_1 = require("../agents/team-agent.js");
 const type_utils_js_1 = require("../utils/type-utils.js");
 const function_agent_js_1 = require("./function-agent.js");
 const schema_js_1 = require("./schema.js");
-async function parseAgentFile(path, data) {
-    const agentSchema = (0, exports.getAgentSchema)({ filepath: path });
+async function parseAgentFile(path, data, options) {
+    const agentSchema = (0, exports.getAgentSchema)({ filepath: path, options });
     return agentSchema.parseAsync({
         ...data,
         model: data.model || data.chatModel || data.chat_model,
@@ -23,22 +23,11 @@ async function parseAgentFile(path, data) {
 async function loadAgentFromYamlFile(path, options) {
     const raw = await (0, type_utils_js_1.tryOrThrow)(() => index_js_1.nodejs.fs.readFile(path, "utf8"), (error) => new Error(`Failed to load agent definition from ${path}: ${error.message}`));
     const json = (0, type_utils_js_1.tryOrThrow)(() => (0, yaml_1.parse)(raw), (error) => new Error(`Failed to parse agent definition from ${path}: ${error.message}`));
-    if (!["ai", "image", "mcp", "team", "transform", "function"].includes(json?.type)) {
-        if (typeof json?.type === "string") {
-            if (!options?.require)
-                throw new Error(`Module loader is not provided to load agent type module ${json.type} from ${path}`);
-            const Mod = await options.require(json.type, { parent: path });
-            if (typeof Mod?.default?.prototype?.constructor !== "function") {
-                throw new Error(`The agent type module ${json.type} does not export a default Agent class`);
-            }
-            json.agentClass = Mod.default;
-        }
-    }
     const agent = await (0, type_utils_js_1.tryOrThrow)(async () => await parseAgentFile(path, {
         ...json,
         type: json.type ?? "ai",
         skills: json.skills ?? json.tools,
-    }), (error) => new Error(`Failed to validate agent definition from ${path}: ${error.message}`));
+    }, options), (error) => new Error(`Failed to validate agent definition from ${path}: ${error.message}`));
     return agent;
 }
 const instructionItemSchema = zod_1.z.union([
@@ -76,7 +65,7 @@ const getInstructionsSchema = ({ filepath }) => zod_1.z
     return [await parseInstructionItem({ filepath })(v)];
 });
 exports.getInstructionsSchema = getInstructionsSchema;
-const getAgentSchema = ({ filepath }) => {
+const getAgentSchema = ({ filepath, options, }) => {
     const agentSchema = zod_1.z.lazy(() => {
         const nestAgentSchema = zod_1.z.lazy(() => zod_1.z.union([
             agentSchema,
@@ -133,7 +122,22 @@ const getAgentSchema = ({ filepath }) => {
             shareAFS: (0, schema_js_1.optionalize)(zod_1.z.boolean()),
         });
         const instructionsSchema = (0, exports.getInstructionsSchema)({ filepath: filepath });
-        return (0, schema_js_1.camelizeSchema)(zod_1.z.union([
+        return (0, schema_js_1.camelizeSchema)((0, schema_js_1.preprocessSchema)(async (json) => {
+            if (typeof json === "object" &&
+                json &&
+                "type" in json &&
+                typeof json.type === "string" &&
+                !["ai", "image", "mcp", "team", "transform", "function"].includes(json.type)) {
+                if (!options?.require)
+                    throw new Error(`Module loader is not provided to load agent type module ${json.type} from ${filepath}`);
+                const Mod = await options.require(json.type, { parent: filepath });
+                if (typeof Mod?.default?.prototype?.constructor !== "function") {
+                    throw new Error(`The agent type module ${json.type} does not export a default Agent class`);
+                }
+                Object.assign(json, { agentClass: Mod.default });
+            }
+            return json;
+        }, zod_1.z.union([
             zod_1.z
                 .object({
                 type: zod_1.z.string(),
@@ -204,7 +208,7 @@ const getAgentSchema = ({ filepath }) => {
                 })
                     .extend(baseAgentSchema.shape),
             ]),
-        ]));
+        ])));
     });
     return agentSchema;
 };

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

@@ -28,9 +28,9 @@ export interface LoadOptions {
     }) => Promise<any>;
 }
 export declare function load(path: string, options?: LoadOptions): Promise<AIGNEOptions>;
-export declare function loadAgent(path: string, options?: LoadOptions, agentOptions?: AgentOptions): Promise<Agent>;
-export declare function loadNestAgent(path: string, agent: NestAgentSchema, options?: LoadOptions, agentOptions?: AgentOptions & Record<string, unknown>): Promise<Agent>;
-export declare function parseAgent(path: string, agent: Awaited<ReturnType<typeof loadAgentFromYamlFile>>, options?: LoadOptions, agentOptions?: AgentOptions): Promise<Agent>;
+export declare function loadAgent(path: string, options: LoadOptions, agentOptions?: AgentOptions): Promise<Agent>;
+export declare function loadNestAgent(path: string, agent: NestAgentSchema, options: LoadOptions, agentOptions?: AgentOptions<any, any> & Record<string, unknown>): Promise<Agent>;
+export declare function parseAgent(path: string, agent: Awaited<ReturnType<typeof loadAgentFromYamlFile>>, options: LoadOptions, agentOptions?: AgentOptions): Promise<Agent>;
 type CliAgent = string | {
     url?: string;
     name?: string;

package/lib/cjs/loader/index.js

@@ -70,7 +70,7 @@ async function load(path, options = {}) {
 }
 async function loadAgent(path, options, agentOptions) {
     if ([".js", ".mjs", ".ts", ".mts"].includes(index_js_1.nodejs.path.extname(path))) {
-        const agent = await (0, agent_js_js_1.loadAgentFromJsFile)(path);
+        const agent = await (0, agent_js_js_1.loadAgentFromJsFile)(path, options);
         if (agent instanceof agent_js_1.Agent)
             return agent;
         return parseAgent(path, agent, options, agentOptions);

package/lib/cjs/loader/schema.d.ts

@@ -148,4 +148,5 @@ export declare function optionalize<T>(schema: ZodType<T>): ZodType<T | undefine
 export declare function camelizeSchema<T extends ZodType>(schema: T, { shallow }?: {
     shallow?: boolean;
 }): T;
+export declare function preprocessSchema<T extends ZodType>(fn: (data: unknown) => unknown, schema: T): T;
 export {};

package/lib/cjs/loader/schema.js

@@ -3,6 +3,7 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.imageModelSchema = exports.chatModelSchema = exports.defaultInputSchema = exports.inputOutputSchema = void 0;
 exports.optionalize = optionalize;
 exports.camelizeSchema = camelizeSchema;
+exports.preprocessSchema = preprocessSchema;
 const index_js_1 = require("@aigne/platform-helpers/nodejs/index.js");
 const yaml_1 = require("yaml");
 const zod_1 = require("zod");
@@ -97,3 +98,6 @@ function optionalize(schema) {
 function camelizeSchema(schema, { shallow = true } = {}) {
     return zod_1.z.preprocess((v) => ((0, type_utils_js_1.isRecord)(v) ? (0, camelize_js_1.camelize)(v, shallow) : v), schema);
 }
+function preprocessSchema(fn, schema) {
+    return zod_1.z.preprocess(fn, schema);
+}

package/lib/cjs/prompt/filters/index.js

@@ -37,13 +37,15 @@ exports.setupFilters = setupFilters;
 const tson = __importStar(require("@zenoaihq/tson"));
 const yaml_1 = require("yaml");
 function setupFilters(env) {
-    env.addFilter("yaml.stringify", (obj) => {
-        return (0, yaml_1.stringify)(obj);
-    });
-    env.addFilter("json.stringify", (obj, ...args) => {
-        return JSON.stringify(obj, ...args);
-    });
-    env.addFilter("tson.stringify", (obj) => {
-        return tson.dumps(obj);
-    });
+    env.addFilter("yaml.stringify", async (obj, cb) => {
+        cb(null, (0, yaml_1.stringify)(await obj));
+    }, true);
+    env.addFilter("json.stringify", async (obj, ...args) => {
+        const jsonArgs = args.slice(0, -1);
+        const cb = args[args.length - 1];
+        cb(null, JSON.stringify(await obj, ...jsonArgs));
+    }, true);
+    env.addFilter("tson.stringify", async (obj, cb) => {
+        cb(null, tson.dumps(await obj));
+    }, true);
 }

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

@@ -37,6 +37,10 @@ export declare class PromptBuilder {
     }>;
     private getTemplateVariables;
     private buildMessages;
+    getHistories(options: PromptBuildOptions): Promise<{
+        role: "user" | "agent";
+        content: unknown;
+    }[]>;
     private refineMessages;
     private convertMemoriesToMessages;
     private buildResponseFormat;

package/lib/cjs/prompt/prompt-builder.js

@@ -95,10 +95,36 @@ class PromptBuilder {
         };
     }
     getTemplateVariables(options) {
+        const self = this;
         return {
             userContext: options.context?.userContext,
             ...options.context?.userContext,
             ...options.input,
+            $afs: {
+                get enabled() {
+                    return !!options.agent?.afs;
+                },
+                description: afs_builtin_prompt_js_1.AFS_DESCRIPTION_PROMPT_TEMPLATE,
+                get modules() {
+                    return options.agent?.afs
+                        ?.listModules()
+                        .then((list) => list.map((i) => (0, type_utils_js_1.pick)(i, ["name", "path", "description"])));
+                },
+                get histories() {
+                    return self.getHistories(options);
+                },
+                get skills() {
+                    const afs = options.agent?.afs;
+                    if (!afs)
+                        return [];
+                    return (0, afs_js_1.getAFSSkills)(afs).then((skills) => skills.map((s) => (0, type_utils_js_1.pick)(s, ["name", "description"])));
+                },
+            },
+            $agent: {
+                get skills() {
+                    return options.agent?.skills.map((s) => (0, type_utils_js_1.pick)(s, ["name", "description"]));
+                },
+            },
         };
     }
     async buildMessages(options) {
@@ -120,7 +146,7 @@ class PromptBuilder {
             memories.push(...options.context.memories);
         }
         const afs = options.agent?.afs;
-        if (afs) {
+        if (afs && options.agent?.historyConfig?.disabled !== true) {
             const historyModule = (await afs.listModules()).find((m) => m.module instanceof afs_history_1.AFSHistory);
             messages.push(await template_js_1.SystemMessageTemplate.from(await (0, afs_builtin_prompt_js_1.getAFSSystemPrompt)(afs)).format({}));
             if (historyModule) {
@@ -198,6 +224,34 @@ class PromptBuilder {
         messages.push(...otherCustomMessages);
         return this.refineMessages(options, messages);
     }
+    async getHistories(options) {
+        const { agent } = options;
+        const afs = agent?.afs;
+        if (!afs)
+            return [];
+        const historyModule = (await afs.listModules()).find((m) => m.module instanceof afs_history_1.AFSHistory);
+        if (!historyModule)
+            return [];
+        const { list: history } = await afs.list(historyModule.path, {
+            limit: agent.historyConfig?.maxItems || 10,
+            orderBy: [["createdAt", "desc"]],
+        });
+        return history
+            .reverse()
+            .map((i) => {
+            if (!i.content)
+                return;
+            const { input, output } = i.content;
+            if (!input || !output)
+                return;
+            return [
+                { role: "user", content: input },
+                { role: "agent", content: output },
+            ];
+        })
+            .filter(type_utils_js_1.isNonNullable)
+            .flat();
+    }
     refineMessages(options, messages) {
         const { autoReorderSystemMessages, autoMergeSystemMessages } = options.agent ?? {};
         if (!autoReorderSystemMessages && !autoMergeSystemMessages)

package/lib/cjs/prompt/prompts/afs-builtin-prompt.d.ts

@@ -1,3 +1,4 @@
 import type { AFS } from "@aigne/afs";
 export declare function getAFSSystemPrompt(afs: AFS): Promise<string>;
 export declare const AFS_EXECUTABLE_TOOLS_PROMPT_TEMPLATE = "<afs_executable_tools>\nHere are the executable tools available in the AFS you can use:\n\n{{ tools | yaml.stringify }}\n</afs_executable_tools>\n";
+export declare const AFS_DESCRIPTION_PROMPT_TEMPLATE = "AFS (Agentic File System) provides tools to interact with a virtual file system,\nallowing you to list, search, read, and write files, or execute a useful tool from the available modules.\nYou can use these tools to manage and retrieve files as needed.\n";

package/lib/cjs/prompt/prompts/afs-builtin-prompt.js

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.AFS_EXECUTABLE_TOOLS_PROMPT_TEMPLATE = void 0;
+exports.AFS_DESCRIPTION_PROMPT_TEMPLATE = exports.AFS_EXECUTABLE_TOOLS_PROMPT_TEMPLATE = void 0;
 exports.getAFSSystemPrompt = getAFSSystemPrompt;
 const yaml_1 = require("yaml");
 const type_utils_js_1 = require("../../utils/type-utils.js");
@@ -8,9 +8,7 @@ async function getAFSSystemPrompt(afs) {
     return `\
 
 <afs_usage>
-AFS (Agentic File System) provides tools to interact with a virtual file system,
-allowing you to list, search, read, and write files, or execute a useful tool from the available modules.
-Use these tools to manage and retrieve files as needed.
+${exports.AFS_DESCRIPTION_PROMPT_TEMPLATE}
 
 Provided modules:
 ${(0, yaml_1.stringify)((await afs.listModules()).map((i) => (0, type_utils_js_1.pick)(i, ["name", "path", "description"])))}
@@ -31,3 +29,8 @@ Here are the executable tools available in the AFS you can use:
 {{ tools | yaml.stringify }}
 </afs_executable_tools>
 `;
+exports.AFS_DESCRIPTION_PROMPT_TEMPLATE = `\
+AFS (Agentic File System) provides tools to interact with a virtual file system,
+allowing you to list, search, read, and write files, or execute a useful tool from the available modules.
+You can use these tools to manage and retrieve files as needed.
+`;

package/lib/cjs/prompt/skills/afs.js

@@ -22,6 +22,7 @@ async function getAFSSkills(afs) {
                 return {
                     status: "success",
                     tool: "afs_list",
+                    path: input.path,
                     options: input.options,
                     message,
                     result,
@@ -39,6 +40,10 @@ async function getAFSSkills(afs) {
                 options: zod_1.z
                     .object({
                     limit: zod_1.z.number().optional().describe("Maximum number of entries to return"),
+                    caseSensitive: zod_1.z
+                        .boolean()
+                        .optional()
+                        .describe("Whether the search is case sensitive, default is false"),
                 })
                     .optional(),
             }),
@@ -47,6 +52,7 @@ async function getAFSSkills(afs) {
                 return {
                     status: "success",
                     tool: "afs_search",
+                    path: input.path,
                     query: input.query,
                     options: input.options,
                     ...result,
@@ -83,6 +89,7 @@ Usage:
                     status: "success",
                     tool: "afs_read",
                     path: input.path,
+                    withLineNumbers: input.withLineNumbers,
                     ...result,
                     result: {
                         ...result.result,

package/lib/cjs/utils/event-stream.js

@@ -5,6 +5,7 @@ const eventsource_parser_1 = require("eventsource-parser");
 const immer_1 = require("immer");
 const agent_js_1 = require("../agents/agent.js");
 const type_utils_js_1 = require("./type-utils.js");
+(0, immer_1.setAutoFreeze)(false);
 class EventStreamParser extends TransformStream {
     constructor() {
         let parser;

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

@@ -118,6 +118,7 @@ export interface AgentOptions<I extends Message = Message, O extends Message = M
      * agentSucceed, or agentFailed
      */
     disableEvents?: boolean;
+    historyConfig?: Agent["historyConfig"];
     /**
      * One or more memory agents this agent can use
      */
@@ -346,6 +347,11 @@ export declare abstract class Agent<I extends Message = any, O extends Message =
      * agentSucceed, or agentFailed
      */
     private disableEvents?;
+    historyConfig?: {
+        disabled?: boolean;
+        maxTokens?: number;
+        maxItems?: number;
+    };
     private subscriptions;
     /**
      * Attach agent to context: