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

package/CHANGELOG.md

@@ -1,5 +1,41 @@
 # 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)
 
 

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<any, any> & 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/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/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/dts/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/dts/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/dts/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/dts/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<any, any> & 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/dts/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/esm/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/esm/aigne/aigne.js

@@ -96,6 +96,10 @@ export class AIGNE {
      * @hidden
      */
     messageQueue = new 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.
@@ -197,7 +201,7 @@ export 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();
         }
@@ -217,12 +221,17 @@ export 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);
     }
 }
 const aigneOptionsSchema = z.object({

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

@@ -17,7 +17,6 @@ import { newEmptyContextUsage } from "./usage.js";
  * @hidden
  */
 export class AIGNEContext {
-    static exitCount = 0;
     constructor(parent, { reset } = {}) {
         const tracer = parent?.observer?.tracer;
         if (parent instanceof AIGNEContext && !reset) {
@@ -41,7 +40,7 @@ export class AIGNEContext {
             this.rootId = this.span?.spanContext?.().traceId ?? v7();
         }
         this.id = this.span?.spanContext()?.spanId ?? v7();
-        this.initProcessExitHandler();
+        this.internal.contextIds.add(this.id);
     }
     id;
     parentId;
@@ -197,26 +196,6 @@ export 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: 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;
@@ -252,12 +231,14 @@ export class AIGNEContext {
                     await this.observer?.update(this.id, { output });
                     span.setStatus({ code: SpanStatusCode.OK });
                     span.end();
+                    this.internal.contextIds.delete(this.id);
                     break;
                 }
                 case "agentFailed": {
                     const { error } = args[0];
                     span.setStatus({ code: SpanStatusCode.ERROR, message: error.message });
                     span.end();
+                    this.internal.contextIds.delete(this.id);
                     break;
                 }
             }
@@ -285,6 +266,7 @@ class AIGNEContextShared {
         this.parent = parent;
         this.messageQueue = this.parent?.messageQueue ?? new MessageQueue();
         this.events = this.parent?.events ?? new Emitter();
+        this.contextIds = this.parent?.contextIds ?? new Set();
     }
     messageQueue;
     events;
@@ -310,6 +292,7 @@ class AIGNEContextShared {
         return this.parent?.limits;
     }
     usage = newEmptyContextUsage();
+    contextIds;
     userContext = {};
     memories = [];
     hooks = [];

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

@@ -4,7 +4,7 @@ import { tryOrThrow } from "../utils/type-utils.js";
 import { parseAgentFile } from "./agent-yaml.js";
 import { LoadJsAgentError } from "./error.js";
 const importFn = new Function("path", "return import(path)");
-export async function loadAgentFromJsFile(path) {
+export async function loadAgentFromJsFile(path, options) {
     const url = nodejs.path.isAbsolute(path) ? nodejs.url.pathToFileURL(path).toString() : path;
     const { default: agent } = await tryOrThrow(() => importFn(url), (error) => new LoadJsAgentError(`Failed to load agent definition from ${url}: ${error.message}`));
     if (isAgent(agent))
@@ -14,5 +14,5 @@ export 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/esm/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/esm/loader/agent-yaml.js

@@ -7,9 +7,9 @@ import { roleSchema } from "../agents/chat-model.js";
 import { ProcessMode } from "../agents/team-agent.js";
 import { tryOrThrow } from "../utils/type-utils.js";
 import { codeToFunctionAgentFn } from "./function-agent.js";
-import { camelizeSchema, chatModelSchema, defaultInputSchema, imageModelSchema, inputOutputSchema, optionalize, } from "./schema.js";
-export async function parseAgentFile(path, data) {
-    const agentSchema = getAgentSchema({ filepath: path });
+import { camelizeSchema, chatModelSchema, defaultInputSchema, imageModelSchema, inputOutputSchema, optionalize, preprocessSchema, } from "./schema.js";
+export async function parseAgentFile(path, data, options) {
+    const agentSchema = getAgentSchema({ filepath: path, options });
     return agentSchema.parseAsync({
         ...data,
         model: data.model || data.chatModel || data.chat_model,
@@ -18,22 +18,11 @@ export async function parseAgentFile(path, data) {
 export async function loadAgentFromYamlFile(path, options) {
     const raw = await tryOrThrow(() => nodejs.fs.readFile(path, "utf8"), (error) => new Error(`Failed to load agent definition from ${path}: ${error.message}`));
     const json = tryOrThrow(() => 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 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 = z.union([
@@ -70,7 +59,7 @@ export const getInstructionsSchema = ({ filepath }) => z
     }
     return [await parseInstructionItem({ filepath })(v)];
 });
-export const getAgentSchema = ({ filepath }) => {
+export const getAgentSchema = ({ filepath, options, }) => {
     const agentSchema = z.lazy(() => {
         const nestAgentSchema = z.lazy(() => z.union([
             agentSchema,
@@ -127,7 +116,22 @@ export const getAgentSchema = ({ filepath }) => {
             shareAFS: optionalize(z.boolean()),
         });
         const instructionsSchema = getInstructionsSchema({ filepath: filepath });
-        return camelizeSchema(z.union([
+        return camelizeSchema(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;
+        }, z.union([
             z
                 .object({
                 type: z.string(),
@@ -198,7 +202,7 @@ export const getAgentSchema = ({ filepath }) => {
                 })
                     .extend(baseAgentSchema.shape),
             ]),
-        ]));
+        ])));
     });
     return agentSchema;
 };

package/lib/esm/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<any, any> & 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/esm/loader/index.js

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

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

@@ -91,3 +91,6 @@ export function optionalize(schema) {
 export function camelizeSchema(schema, { shallow = true } = {}) {
     return z.preprocess((v) => (isRecord(v) ? camelize(v, shallow) : v), schema);
 }
+export function preprocessSchema(fn, schema) {
+    return z.preprocess(fn, schema);
+}

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

@@ -19,6 +19,7 @@ export async function getAFSSkills(afs) {
                 return {
                     status: "success",
                     tool: "afs_list",
+                    path: input.path,
                     options: input.options,
                     message,
                     result,
@@ -36,6 +37,10 @@ export async function getAFSSkills(afs) {
                 options: z
                     .object({
                     limit: z.number().optional().describe("Maximum number of entries to return"),
+                    caseSensitive: z
+                        .boolean()
+                        .optional()
+                        .describe("Whether the search is case sensitive, default is false"),
                 })
                     .optional(),
             }),
@@ -44,6 +49,7 @@ export async function getAFSSkills(afs) {
                 return {
                     status: "success",
                     tool: "afs_search",
+                    path: input.path,
                     query: input.query,
                     options: input.options,
                     ...result,
@@ -80,6 +86,7 @@ Usage:
                     status: "success",
                     tool: "afs_read",
                     path: input.path,
+                    withLineNumbers: input.withLineNumbers,
                     ...result,
                     result: {
                         ...result.result,

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

@@ -1,7 +1,8 @@
 import { createParser } from "eventsource-parser";
-import { produce } from "immer";
+import { produce, setAutoFreeze } from "immer";
 import { isAgentResponseDelta, isAgentResponseProgress, } from "../agents/agent.js";
 import { tryOrThrow } from "./type-utils.js";
+setAutoFreeze(false);
 export class EventStreamParser extends TransformStream {
     constructor() {
         let parser;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@aigne/core",
-  "version": "1.71.0-beta.2",
+  "version": "1.71.0-beta.4",
   "description": "The functional core of agentic AI",
   "publishConfig": {
     "access": "public"
@@ -92,10 +92,10 @@
     "zod": "^3.25.67",
     "zod-from-json-schema": "^0.0.5",
     "zod-to-json-schema": "^3.24.6",
-    "@aigne/afs": "^1.3.0-beta",
-    "@aigne/afs-history": "^1.1.3-beta",
-    "@aigne/observability-api": "^0.11.12",
-    "@aigne/platform-helpers": "^0.6.5"
+    "@aigne/afs": "^1.3.0-beta.2",
+    "@aigne/afs-history": "^1.1.3-beta.2",
+    "@aigne/platform-helpers": "^0.6.6-beta",
+    "@aigne/observability-api": "^0.11.13-beta.1"
   },
   "devDependencies": {
     "@types/bun": "^1.2.22",