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

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 & 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/dts/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/dts/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/esm/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/esm/agents/agent.js

@@ -103,6 +103,7 @@ export 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);
         }
@@ -276,6 +277,7 @@ export class Agent {
      * agentSucceed, or agentFailed
      */
     disableEvents;
+    historyConfig;
     subscriptions = [];
     /**
      * Attach agent to context:
@@ -519,7 +521,8 @@ export 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/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 & 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/filters/index.js

@@ -1,13 +1,15 @@
 import * as tson from "@zenoaihq/tson";
 import { stringify } from "yaml";
 export function setupFilters(env) {
-    env.addFilter("yaml.stringify", (obj) => {
-        return 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, 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/esm/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/esm/prompt/prompt-builder.js

@@ -8,8 +8,8 @@ import { DEFAULT_OUTPUT_FILE_KEY, DEFAULT_OUTPUT_KEY } from "../agents/ai-agent.
 import { fileUnionContentsSchema } from "../agents/model.js";
 import { optionalize } from "../loader/schema.js";
 import { outputSchemaToResponseFormatSchema } from "../utils/json-schema.js";
-import { checkArguments, flat, isNonNullable, isRecord, partition, unique, } from "../utils/type-utils.js";
-import { AFS_EXECUTABLE_TOOLS_PROMPT_TEMPLATE, getAFSSystemPrompt, } from "./prompts/afs-builtin-prompt.js";
+import { checkArguments, flat, isNonNullable, isRecord, partition, pick, unique, } from "../utils/type-utils.js";
+import { AFS_DESCRIPTION_PROMPT_TEMPLATE, AFS_EXECUTABLE_TOOLS_PROMPT_TEMPLATE, getAFSSystemPrompt, } from "./prompts/afs-builtin-prompt.js";
 import { MEMORY_MESSAGE_TEMPLATE } from "./prompts/memory-message-template.js";
 import { STRUCTURED_STREAM_INSTRUCTIONS } from "./prompts/structured-stream-instructions.js";
 import { getAFSSkills } from "./skills/afs.js";
@@ -92,10 +92,36 @@ export 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_DESCRIPTION_PROMPT_TEMPLATE,
+                get modules() {
+                    return options.agent?.afs
+                        ?.listModules()
+                        .then((list) => list.map((i) => pick(i, ["name", "path", "description"])));
+                },
+                get histories() {
+                    return self.getHistories(options);
+                },
+                get skills() {
+                    const afs = options.agent?.afs;
+                    if (!afs)
+                        return [];
+                    return getAFSSkills(afs).then((skills) => skills.map((s) => pick(s, ["name", "description"])));
+                },
+            },
+            $agent: {
+                get skills() {
+                    return options.agent?.skills.map((s) => pick(s, ["name", "description"]));
+                },
+            },
         };
     }
     async buildMessages(options) {
@@ -117,7 +143,7 @@ export 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 AFSHistory);
             messages.push(await SystemMessageTemplate.from(await getAFSSystemPrompt(afs)).format({}));
             if (historyModule) {
@@ -195,6 +221,34 @@ export 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 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(isNonNullable)
+            .flat();
+    }
     refineMessages(options, messages) {
         const { autoReorderSystemMessages, autoMergeSystemMessages } = options.agent ?? {};
         if (!autoReorderSystemMessages && !autoMergeSystemMessages)

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