@aigne/core 1.63.0-beta → 1.63.0-beta.10

package/CHANGELOG.md

@@ -1,5 +1,102 @@
 # Changelog
 
+## [1.63.0-beta.10](https://github.com/AIGNE-io/aigne-framework/compare/core-v1.63.0-beta.9...core-v1.63.0-beta.10) (2025-10-16)
+
+
+### Dependencies
+
+* The following workspace dependencies were updated
+  * dependencies
+    * @aigne/observability-api bumped to 0.11.2-beta.5
+
+## [1.63.0-beta.9](https://github.com/AIGNE-io/aigne-framework/compare/core-v1.63.0-beta.8...core-v1.63.0-beta.9) (2025-10-16)
+
+
+### Bug Fixes
+
+* allow custom user messages and prevent duplicate user content ([#632](https://github.com/AIGNE-io/aigne-framework/issues/632)) ([6c883b2](https://github.com/AIGNE-io/aigne-framework/commit/6c883b2d57a65e9b46232cece91fc6aa1de03aba))
+* **models:** auto retry when got emtpy response from gemini ([#636](https://github.com/AIGNE-io/aigne-framework/issues/636)) ([9367cef](https://github.com/AIGNE-io/aigne-framework/commit/9367cef49ea4c0c87b8a36b454deb2efaee6886f))
+
+
+### Dependencies
+
+* The following workspace dependencies were updated
+  * dependencies
+    * @aigne/observability-api bumped to 0.11.2-beta.4
+
+## [1.63.0-beta.8](https://github.com/AIGNE-io/aigne-framework/compare/core-v1.63.0-beta.7...core-v1.63.0-beta.8) (2025-10-16)
+
+
+### Dependencies
+
+* The following workspace dependencies were updated
+  * dependencies
+    * @aigne/observability-api bumped to 0.11.2-beta.3
+
+## [1.63.0-beta.7](https://github.com/AIGNE-io/aigne-framework/compare/core-v1.63.0-beta.6...core-v1.63.0-beta.7) (2025-10-15)
+
+
+### Bug Fixes
+
+* add options for system message reordering and merging ([#624](https://github.com/AIGNE-io/aigne-framework/issues/624)) ([8ca466d](https://github.com/AIGNE-io/aigne-framework/commit/8ca466d49d1e4ed08bc90922f39c0d3ed60c4fd5))
+
+## [1.63.0-beta.6](https://github.com/AIGNE-io/aigne-framework/compare/core-v1.63.0-beta.5...core-v1.63.0-beta.6) (2025-10-15)
+
+
+### Dependencies
+
+* The following workspace dependencies were updated
+  * dependencies
+    * @aigne/observability-api bumped to 0.11.2-beta.2
+
+## [1.63.0-beta.5](https://github.com/AIGNE-io/aigne-framework/compare/core-v1.63.0-beta.4...core-v1.63.0-beta.5) (2025-10-13)
+
+
+### Bug Fixes
+
+* **core:** auto merge multiple system messages ([#619](https://github.com/AIGNE-io/aigne-framework/issues/619)) ([e9e62c0](https://github.com/AIGNE-io/aigne-framework/commit/e9e62c03c45f5a9b75d44a07588b2b179e262aad))
+
+## [1.63.0-beta.4](https://github.com/AIGNE-io/aigne-framework/compare/core-v1.63.0-beta.3...core-v1.63.0-beta.4) (2025-10-12)
+
+
+### Features
+
+* **afs:** add configurable history injection to AFS system ([#611](https://github.com/AIGNE-io/aigne-framework/issues/611)) ([689b5d7](https://github.com/AIGNE-io/aigne-framework/commit/689b5d76d8cc82f548e8be74e63f18e7a6216c31))
+
+## [1.63.0-beta.3](https://github.com/AIGNE-io/aigne-framework/compare/core-v1.63.0-beta.2...core-v1.63.0-beta.3) (2025-10-11)
+
+
+### Features
+
+* **afs:** add module system fs for afs ([#594](https://github.com/AIGNE-io/aigne-framework/issues/594)) ([83c7b65](https://github.com/AIGNE-io/aigne-framework/commit/83c7b6555d21c606a5005eb05f6686882fb8ffa3))
+
+
+### Dependencies
+
+* The following workspace dependencies were updated
+  * dependencies
+    * @aigne/afs bumped to 1.1.0-beta
+    * @aigne/observability-api bumped to 0.11.2-beta.1
+
+## [1.63.0-beta.2](https://github.com/AIGNE-io/aigne-framework/compare/core-v1.63.0-beta.1...core-v1.63.0-beta.2) (2025-10-09)
+
+
+### Bug Fixes
+
+* input schema of AI agent should includes input key and input file key ([#600](https://github.com/AIGNE-io/aigne-framework/issues/600)) ([b4ca076](https://github.com/AIGNE-io/aigne-framework/commit/b4ca076d6b4a1a1ecb8d4ebb008abd0d7561aadd))
+
+## [1.63.0-beta.1](https://github.com/AIGNE-io/aigne-framework/compare/core-v1.63.0-beta...core-v1.63.0-beta.1) (2025-10-09)
+
+
+### Features
+
+* **core:** add `toolCallsConcurrency` support for AI agent ([#598](https://github.com/AIGNE-io/aigne-framework/issues/598)) ([84df406](https://github.com/AIGNE-io/aigne-framework/commit/84df406bfa9e3bdf159509f4b9cf2301ec80b155))
+
+
+### Bug Fixes
+
+* **core:** add input_file_key support for agent yaml ([#597](https://github.com/AIGNE-io/aigne-framework/issues/597)) ([63414a3](https://github.com/AIGNE-io/aigne-framework/commit/63414a3d46c74c686e7f033c224ca6175bea8c3f))
+
 ## [1.63.0-beta](https://github.com/AIGNE-io/aigne-framework/compare/core-v1.62.0...core-v1.63.0-beta) (2025-10-07)
 
 

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

@@ -20,7 +20,8 @@ export declare const DEFAULT_INPUT_ACTION_GET = "$get";
  */
 export interface Message extends Record<string, unknown> {
     $meta?: {
-        usage: ContextUsage;
+        usage?: ContextUsage;
+        [key: string]: unknown;
     };
 }
 /**
@@ -122,6 +123,7 @@ export interface AgentOptions<I extends Message = Message, O extends Message = M
      */
     memory?: MemoryAgent | MemoryAgent[];
     afs?: true | AFSOptions | AFS | ((afs: AFS) => AFS);
+    afsConfig?: AFSConfig;
     asyncMemoryRecord?: boolean;
     /**
      * Maximum number of memory items to retrieve
@@ -130,6 +132,10 @@ export interface AgentOptions<I extends Message = Message, O extends Message = M
     hooks?: AgentHooks<I, O> | AgentHooks<I, O>[];
     retryOnError?: Agent<I, O>["retryOnError"] | boolean;
 }
+export interface AFSConfig {
+    injectHistory?: boolean;
+    historyWindowSize?: number;
+}
 export declare const agentOptionsSchema: ZodObject<{
     [key in keyof AgentOptions]: ZodType<AgentOptions[key]>;
 }>;
@@ -212,6 +218,7 @@ export declare abstract class Agent<I extends Message = any, O extends Message =
      */
     readonly memories: MemoryAgent[];
     afs?: AFS;
+    afsConfig?: AFSConfig;
     asyncMemoryRecord?: boolean;
     tag?: string;
     /**

package/lib/cjs/agents/agent.js

@@ -161,6 +161,7 @@ class Agent {
                     : options.afs instanceof afs_1.AFS
                         ? options.afs
                         : new afs_1.AFS(options.afs);
+        this.afsConfig = options.afsConfig;
         this.asyncMemoryRecord = options.asyncMemoryRecord;
         this.maxRetrieveMemoryCount = options.maxRetrieveMemoryCount;
         this.hooks = (0, type_utils_js_1.flat)(options.hooks);
@@ -179,6 +180,7 @@ class Agent {
      */
     memories = [];
     afs;
+    afsConfig;
     asyncMemoryRecord;
     tag;
     /**
@@ -449,7 +451,10 @@ class Agent {
                     (0, stream_utils_js_1.mergeAgentResponseChunk)(output, chunk);
                     yield chunk;
                 }
-                const result = await this.processAgentOutput(input, output, options);
+                let result = await this.processAgentOutput(input, output, options);
+                if (attempt > 0) {
+                    result = { ...result, $meta: { ...result.$meta, retries: attempt } };
+                }
                 if (result && !(0, fast_deep_equal_1.default)(result, output)) {
                     yield { delta: { json: result } };
                 }
@@ -459,7 +464,8 @@ class Agent {
             catch (error) {
                 if (this.retryOnError?.retries) {
                     const { retries, minTimeout = DEFAULT_RETRY_MIN_TIMEOUT, factor = DEFAULT_RETRY_FACTOR, randomize = false, shouldRetry, } = this.retryOnError;
-                    if (attempt++ < retries && (!shouldRetry || (await shouldRetry(error)))) {
+                    if (attempt < retries && (!shouldRetry || (await shouldRetry(error)))) {
+                        attempt++;
                         const timeout = minTimeout * factor ** (attempt - 1) * (randomize ? 1 + Math.random() : 1);
                         logger_js_1.logger.warn(`Agent ${this.name} attempt ${attempt} of ${retries} failed with error: ${error}. Retrying in ${timeout}ms...`);
                         await new Promise((resolve) => setTimeout(resolve, timeout));
@@ -467,8 +473,12 @@ class Agent {
                     }
                 }
                 const res = await this.processAgentError(input, error, options);
-                if (!res.retry)
-                    throw res.error ?? error;
+                if (!res.retry) {
+                    const e = res.error ?? error;
+                    if (attempt > 0)
+                        e.message += ` (after ${attempt} retries)`;
+                    throw e;
+                }
             }
         }
     }

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

@@ -3,7 +3,7 @@ import { PromptBuilder } from "../prompt/prompt-builder.js";
 import { Agent, type AgentInvokeOptions, type AgentOptions, type AgentProcessAsyncGenerator, type AgentProcessResult, type Message } from "./agent.js";
 import type { ChatModel, ChatModelInput } from "./chat-model.js";
 import type { GuideRailAgentOutput } from "./guide-rail-agent.js";
-import type { FileType } from "./model.js";
+import { type FileType } from "./model.js";
 export declare const DEFAULT_OUTPUT_KEY = "message";
 export declare const DEFAULT_OUTPUT_FILE_KEY = "files";
 /**
@@ -23,6 +23,8 @@ export interface AIAgentOptions<I extends Message = Message, O extends Message =
      * more complex prompt templates
      */
     instructions?: string | PromptBuilder;
+    autoReorderSystemMessages?: boolean;
+    autoMergeSystemMessages?: boolean;
     /**
      * Pick a message from input to use as the user's message
      */
@@ -42,6 +44,12 @@ export interface AIAgentOptions<I extends Message = Message, O extends Message =
      * @default AIAgentToolChoice.auto
      */
     toolChoice?: AIAgentToolChoice | Agent;
+    /**
+     * Maximum number of tool calls to execute concurrently
+     *
+     * @default 1
+     */
+    toolCallsConcurrency?: number;
     /**
      * Whether to preserve text generated during tool usage in the final output
      */
@@ -210,6 +218,8 @@ export declare class AIAgent<I extends Message = any, O extends Message = any> e
      * {@includeCode ../../test/agents/ai-agent.test.ts#example-ai-agent-prompt-builder}
      */
     instructions: PromptBuilder;
+    autoReorderSystemMessages?: boolean;
+    autoMergeSystemMessages?: boolean;
     /**
      * Pick a message from input to use as the user's message
      */
@@ -237,6 +247,12 @@ export declare class AIAgent<I extends Message = any, O extends Message = any> e
      * {@includeCode ../../test/agents/ai-agent.test.ts#example-ai-agent-router}
      */
     toolChoice?: AIAgentToolChoice | Agent;
+    /**
+     * Maximum number of tool calls to execute concurrently
+     *
+     * @default 1
+     */
+    toolCallsConcurrency?: number;
     /**
      * Whether to preserve text generated during tool usage in the final output
      */
@@ -299,6 +315,7 @@ export declare class AIAgent<I extends Message = any, O extends Message = any> e
         metadataEnd: string;
         parse: (raw: string) => object;
     };
+    get inputSchema(): ZodType<I>;
     /**
      * Process an input message and generate a response
      *

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

@@ -1,6 +1,10 @@
 "use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.AIAgent = exports.aiAgentOptionsSchema = exports.aiAgentToolChoiceSchema = exports.AIAgentToolChoice = exports.DEFAULT_OUTPUT_FILE_KEY = exports.DEFAULT_OUTPUT_KEY = void 0;
+const fastq_1 = __importDefault(require("fastq"));
 const zod_1 = require("zod");
 const prompt_builder_js_1 = require("../prompt/prompt-builder.js");
 const structured_stream_instructions_js_1 = require("../prompt/prompts/structured-stream-instructions.js");
@@ -8,6 +12,7 @@ const template_js_1 = require("../prompt/template.js");
 const structured_stream_extractor_js_1 = require("../utils/structured-stream-extractor.js");
 const type_utils_js_1 = require("../utils/type-utils.js");
 const agent_js_1 = require("./agent.js");
+const model_js_1 = require("./model.js");
 const types_js_1 = require("./types.js");
 exports.DEFAULT_OUTPUT_KEY = "message";
 exports.DEFAULT_OUTPUT_FILE_KEY = "files";
@@ -58,6 +63,7 @@ exports.aiAgentOptionsSchema = agent_js_1.agentOptionsSchema.extend({
     inputKey: zod_1.z.string().optional(),
     outputKey: zod_1.z.string().optional(),
     toolChoice: exports.aiAgentToolChoiceSchema.optional(),
+    toolCallsConcurrency: zod_1.z.number().int().min(0).optional(),
     keepTextInToolUses: zod_1.z.boolean().optional(),
     memoryAgentsAsTools: zod_1.z.boolean().optional(),
     memoryPromptTemplate: zod_1.z.string().optional(),
@@ -111,12 +117,15 @@ class AIAgent extends agent_js_1.Agent {
             typeof options.instructions === "string"
                 ? prompt_builder_js_1.PromptBuilder.from(options.instructions)
                 : (options.instructions ?? new prompt_builder_js_1.PromptBuilder());
+        this.autoReorderSystemMessages = options.autoReorderSystemMessages ?? false;
+        this.autoMergeSystemMessages = options.autoMergeSystemMessages ?? false;
         this.inputKey = options.inputKey;
         this.inputFileKey = options.inputFileKey;
         this.outputKey = options.outputKey || exports.DEFAULT_OUTPUT_KEY;
         this.outputFileKey = options.outputFileKey || exports.DEFAULT_OUTPUT_FILE_KEY;
         this.outputFileType = options.outputFileType;
         this.toolChoice = options.toolChoice;
+        this.toolCallsConcurrency = options.toolCallsConcurrency || 1;
         this.keepTextInToolUses = options.keepTextInToolUses;
         this.memoryAgentsAsTools = options.memoryAgentsAsTools;
         this.memoryPromptTemplate = options.memoryPromptTemplate;
@@ -146,6 +155,8 @@ class AIAgent extends agent_js_1.Agent {
      * {@includeCode ../../test/agents/ai-agent.test.ts#example-ai-agent-prompt-builder}
      */
     instructions;
+    autoReorderSystemMessages;
+    autoMergeSystemMessages;
     /**
      * Pick a message from input to use as the user's message
      */
@@ -173,6 +184,12 @@ class AIAgent extends agent_js_1.Agent {
      * {@includeCode ../../test/agents/ai-agent.test.ts#example-ai-agent-router}
      */
     toolChoice;
+    /**
+     * Maximum number of tool calls to execute concurrently
+     *
+     * @default 1
+     */
+    toolCallsConcurrency;
     /**
      * Whether to preserve text generated during tool usage in the final output
      */
@@ -230,6 +247,20 @@ class AIAgent extends agent_js_1.Agent {
      * which outputs structured data in YAML format within <metadata> tags.
      */
     customStructuredStreamInstructions;
+    get inputSchema() {
+        let schema = super.inputSchema;
+        if (this.inputKey) {
+            schema = schema.extend({
+                [this.inputKey]: zod_1.z.string().nullish(),
+            });
+        }
+        if (this.inputFileKey) {
+            schema = schema.extend({
+                [this.inputFileKey]: model_js_1.fileUnionContentsSchema.nullish(),
+            });
+        }
+        return schema;
+    }
     /**
      * Process an input message and generate a response
      *
@@ -287,31 +318,43 @@ class AIAgent extends agent_js_1.Agent {
                     yield { delta: { text: { [outputKey]: "\n" } } };
                 }
                 const executedToolCalls = [];
+                let error;
+                const queue = fastq_1.default.promise(async ({ tool, call }) => {
+                    try {
+                        // NOTE: should pass both arguments (model generated) and input (user provided) to the tool
+                        const output = await this.invokeSkill(tool, { ...input, ...call.function.arguments }, options).catch((error) => {
+                            if (!this.catchToolsError) {
+                                return Promise.reject(error);
+                            }
+                            return {
+                                isError: true,
+                                error: {
+                                    message: error.message,
+                                },
+                            };
+                        });
+                        executedToolCalls.push({ call, output });
+                    }
+                    catch (e) {
+                        error = e;
+                        queue.killAndDrain();
+                    }
+                }, this.toolCallsConcurrency || 1);
                 // Execute tools
                 for (const call of toolCalls) {
                     const tool = toolsMap.get(call.function.name);
                     if (!tool)
                         throw new Error(`Tool not found: ${call.function.name}`);
-                    // NOTE: should pass both arguments (model generated) and input (user provided) to the tool
-                    const output = await this.invokeSkill(tool, { ...input, ...call.function.arguments }, options).catch((error) => {
-                        if (!this.catchToolsError) {
-                            return Promise.reject(error);
-                        }
-                        return {
-                            isError: true,
-                            error: {
-                                message: error.message,
-                            },
-                        };
-                    });
-                    // NOTE: Return transfer output immediately
-                    if ((0, types_js_1.isTransferAgentOutput)(output)) {
-                        return output;
-                    }
-                    executedToolCalls.push({ call, output });
+                    queue.push({ tool, call });
                 }
+                await queue.drained();
+                if (error)
+                    throw error;
                 // Continue LLM function calling loop if any tools were executed
                 if (executedToolCalls.length) {
+                    const transferOutput = executedToolCalls.find((i) => (0, types_js_1.isTransferAgentOutput)(i.output))?.output;
+                    if (transferOutput)
+                        return transferOutput;
                     toolCallMessages.push(await template_js_1.AgentMessageTemplate.from(undefined, executedToolCalls.map(({ call }) => call)).format(), ...(await Promise.all(executedToolCalls.map(({ call, output }) => template_js_1.ToolMessageTemplate.from(output, call.id).format()))));
                     continue;
                 }

package/lib/cjs/agents/chat-model.js

@@ -42,7 +42,7 @@ const type_utils_js_1 = require("../utils/type-utils.js");
 const agent_js_1 = require("./agent.js");
 const model_js_1 = require("./model.js");
 const CHAT_MODEL_DEFAULT_RETRY_OPTIONS = {
-    retries: 3,
+    retries: 10,
     shouldRetry: async (error) => error instanceof StructuredOutputError || (await Promise.resolve().then(() => __importStar(require("is-network-error")))).default(error),
 };
 class StructuredOutputError extends Error {

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

@@ -1,5 +1,6 @@
+import type { AFSOptions } from "@aigne/afs";
 import { type ZodType, z } from "zod";
-import type { AgentHooks, FunctionAgentFn, TaskRenderMode } from "../agents/agent.js";
+import type { AFSConfig, AgentHooks, FunctionAgentFn, TaskRenderMode } from "../agents/agent.js";
 import { AIAgentToolChoice } from "../agents/ai-agent.js";
 import { type Role } from "../agents/chat-model.js";
 import { ProcessMode, type ReflectionMode } from "../agents/team-agent.js";
@@ -19,6 +20,10 @@ export type NestAgentSchema = string | {
     defaultInput?: Record<string, any>;
     hooks?: HooksSchema | HooksSchema[];
 } | AgentSchema;
+export type AFSModuleSchema = string | {
+    module: string;
+    options?: Record<string, any>;
+};
 export interface BaseAgentSchema {
     name?: string;
     description?: string;
@@ -36,6 +41,10 @@ export interface BaseAgentSchema {
         provider: string;
         subscribeTopic?: string[];
     };
+    afs?: boolean | (Omit<AFSOptions, "modules"> & {
+        modules?: AFSModuleSchema[];
+    });
+    afsConfig?: AFSConfig;
 }
 export type Instructions = {
     role: Exclude<Role, "tool">;
@@ -45,9 +54,13 @@ export type Instructions = {
 export interface AIAgentSchema extends BaseAgentSchema {
     type: "ai";
     instructions?: Instructions;
+    autoReorderSystemMessages?: boolean;
+    autoMergeSystemMessages?: boolean;
     inputKey?: string;
+    inputFileKey?: string;
     outputKey?: string;
     toolChoice?: AIAgentToolChoice;
+    toolCallsConcurrency?: number;
     keepTextInToolUses?: boolean;
 }
 export interface ImageAgentSchema extends BaseAgentSchema {

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

@@ -53,6 +53,25 @@ async function parseAgentFile(path, data) {
                     subscribeTopic: (0, schema_js_1.optionalize)(zod_1.z.array(zod_1.z.string())),
                 })),
             ])),
+            afs: (0, schema_js_1.optionalize)(zod_1.z.union([
+                zod_1.z.boolean(),
+                (0, schema_js_1.camelizeSchema)(zod_1.z.object({
+                    storage: (0, schema_js_1.optionalize)(zod_1.z.object({
+                        url: (0, schema_js_1.optionalize)(zod_1.z.string()),
+                    })),
+                    modules: (0, schema_js_1.optionalize)(zod_1.z.array(zod_1.z.union([
+                        zod_1.z.string(),
+                        zod_1.z.object({
+                            module: zod_1.z.string(),
+                            options: (0, schema_js_1.optionalize)(zod_1.z.record(zod_1.z.any())),
+                        }),
+                    ]))),
+                })),
+            ])),
+            afsConfig: (0, schema_js_1.optionalize)((0, schema_js_1.camelizeSchema)(zod_1.z.object({
+                injectHistory: (0, schema_js_1.optionalize)(zod_1.z.boolean()),
+                historyWindowSize: (0, schema_js_1.optionalize)(zod_1.z.number().int().min(1)),
+            }))),
         });
         const instructionItemSchema = zod_1.z.union([
             zod_1.z.object({
@@ -93,9 +112,14 @@ async function parseAgentFile(path, data) {
                 .object({
                 type: zod_1.z.literal("ai"),
                 instructions: (0, schema_js_1.optionalize)(instructionsSchema),
+                autoReorderSystemMessages: (0, schema_js_1.optionalize)(zod_1.z.boolean()),
+                autoMergeSystemMessages: (0, schema_js_1.optionalize)(zod_1.z.boolean()),
                 inputKey: (0, schema_js_1.optionalize)(zod_1.z.string()),
                 outputKey: (0, schema_js_1.optionalize)(zod_1.z.string()),
+                inputFileKey: (0, schema_js_1.optionalize)(zod_1.z.string()),
+                outputFileKey: (0, schema_js_1.optionalize)(zod_1.z.string()),
                 toolChoice: (0, schema_js_1.optionalize)(zod_1.z.nativeEnum(ai_agent_js_1.AIAgentToolChoice)),
+                toolCallsConcurrency: (0, schema_js_1.optionalize)(zod_1.z.number().int().min(0)),
                 keepTextInToolUses: (0, schema_js_1.optionalize)(zod_1.z.boolean()),
                 structuredStreamMode: (0, schema_js_1.optionalize)(zod_1.z.boolean()),
             })

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

@@ -1,3 +1,4 @@
+import { type AFSModule } from "@aigne/afs";
 import { type ZodType, z } from "zod";
 import { Agent, type AgentOptions } from "../agents/agent.js";
 import type { ChatModel } from "../agents/chat-model.js";
@@ -11,6 +12,12 @@ export interface LoadOptions {
     }[];
     model?: ChatModel | ((model?: z.infer<typeof aigneFileSchema>["model"]) => PromiseOrValue<ChatModel | undefined>);
     imageModel?: ImageModel | ((model?: z.infer<typeof aigneFileSchema>["imageModel"]) => PromiseOrValue<ImageModel | undefined>);
+    afs?: {
+        availableModules?: {
+            module: string;
+            create: (options?: Record<string, any>) => PromiseOrValue<AFSModule>;
+        }[];
+    };
 }
 export declare function load(path: string, options?: LoadOptions): Promise<AIGNEOptions>;
 export declare function loadAgent(path: string, options?: LoadOptions, agentOptions?: AgentOptions): Promise<Agent>;

package/lib/cjs/loader/index.js

@@ -3,6 +3,7 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.load = load;
 exports.loadAgent = loadAgent;
 exports.loadAIGNEFile = loadAIGNEFile;
+const afs_1 = require("@aigne/afs");
 const index_js_1 = require("@aigne/platform-helpers/nodejs/index.js");
 const yaml_1 = require("yaml");
 const zod_1 = require("zod");
@@ -112,6 +113,26 @@ async function parseAgent(path, agent, options, agentOptions) {
     const memory = "memory" in agent && options?.memories?.length
         ? await loadMemory(options.memories, typeof agent.memory === "object" ? agent.memory.provider : undefined, typeof agent.memory === "object" ? agent.memory : {})
         : undefined;
+    let afs;
+    if (typeof agent.afs === "boolean") {
+        if (agent.afs) {
+            afs = new afs_1.AFS();
+        }
+    }
+    else if (agent.afs) {
+        afs = new afs_1.AFS({
+            ...agent.afs,
+            modules: agent.afs.modules &&
+                (await Promise.all(agent.afs.modules.map((m) => {
+                    const mod = typeof m === "string"
+                        ? options?.afs?.availableModules?.find((mod) => mod.module === m)
+                        : options?.afs?.availableModules?.find((mod) => mod.module === m.module);
+                    if (!mod)
+                        throw new Error(`AFS module not found: ${typeof m === "string" ? m : m.module}`);
+                    return mod.create(typeof m === "string" ? {} : m.options);
+                }))),
+        });
+    }
     const model = agent.model && typeof options?.model === "function"
         ? await options.model(agent.model)
         : undefined;
@@ -129,6 +150,7 @@ async function parseAgent(path, agent, options, agentOptions) {
             ...((await parseHooks(path, agent.hooks, options)) ?? []),
             ...[agentOptions?.hooks].flat().filter(type_utils_js_1.isNonNullable),
         ],
+        afs,
     };
     let instructions;
     if ("instructions" in agent && agent.instructions) {

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

@@ -31,6 +31,7 @@ export declare class PromptBuilder {
         prompt: string;
     }>;
     private buildMessages;
+    private refineMessages;
     private convertMemoriesToMessages;
     private buildResponseFormat;
     private buildTools;