@aigne/core 1.39.0 → 1.41.0

package/CHANGELOG.md

@@ -12,6 +12,20 @@
   * dependencies
     * @aigne/observability bumped to 0.1.0
 
+## [1.41.0](https://github.com/AIGNE-io/aigne-framework/compare/core-v1.40.0...core-v1.41.0) (2025-07-31)
+
+
+### Features
+
+* **cli:** add alias support for agent ([#297](https://github.com/AIGNE-io/aigne-framework/issues/297)) ([fa166ab](https://github.com/AIGNE-io/aigne-framework/commit/fa166ab66d19e89ddd32c34e1470450eb4fbdbbd))
+
+## [1.40.0](https://github.com/AIGNE-io/aigne-framework/compare/core-v1.39.0...core-v1.40.0) (2025-07-31)
+
+
+### Features
+
+* **cli:** support dynamic download and execution of doc-smith app ([#293](https://github.com/AIGNE-io/aigne-framework/issues/293)) ([4c40077](https://github.com/AIGNE-io/aigne-framework/commit/4c40077bacef076bc4b098879e948ef866218e39))
+
 ## [1.39.0](https://github.com/AIGNE-io/aigne-framework/compare/core-v1.38.1...core-v1.39.0) (2025-07-28)
 
 

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

@@ -61,6 +61,10 @@ export interface AgentOptions<I extends Message = Message, O extends Message = M
      * if not specified
      */
     name?: string;
+    /**
+     * Alias for the agent, can be used to refer to the agent by multiple names, especially in AIGNE CLI
+     */
+    alias?: string[];
     /**
      * Description of the agent
      *
@@ -232,6 +236,10 @@ export declare abstract class Agent<I extends Message = any, O extends Message =
      * Defaults to the class constructor name if not specified in options
      */
     readonly name: string;
+    /**
+     * Alias for the agent, can be used to refer to the agent by multiple names, especially in AIGNE CLI
+     */
+    readonly alias?: string[];
     /**
      * Default topic this agent subscribes to
      *

package/lib/cjs/agents/agent.js

@@ -102,6 +102,7 @@ class Agent {
     constructor(options = {}) {
         const { inputSchema, outputSchema } = options;
         this.name = options.name || this.constructor.name;
+        this.alias = options.alias;
         this.description = options.description;
         if (inputSchema)
             checkAgentInputOutputSchema(inputSchema);
@@ -170,6 +171,10 @@ class Agent {
      * Defaults to the class constructor name if not specified in options
      */
     name;
+    /**
+     * Alias for the agent, can be used to refer to the agent by multiple names, especially in AIGNE CLI
+     */
+    alias;
     /**
      * Default topic this agent subscribes to
      *

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

@@ -35,6 +35,12 @@ export interface AIGNEOptions {
      * Agents to use for the AIGNE instance.
      */
     agents?: Agent[];
+    mcpServer?: {
+        agents?: Agent[];
+    };
+    cli?: {
+        agents?: Agent[];
+    };
     /**
      * Limits for the AIGNE instance, such as timeout, max tokens, max invocations, etc.
      */
@@ -112,6 +118,16 @@ export declare class AIGNE<U extends UserContext = UserContext> {
     readonly agents: Agent<any, any>[] & {
         [key: string]: Agent<any, any>;
     };
+    readonly mcpServer: {
+        agents: Agent<any, any>[] & {
+            [key: string]: Agent<any, any>;
+        };
+    };
+    readonly cli: {
+        agents: Agent<any, any>[] & {
+            [key: string]: Agent<any, any>;
+        };
+    };
     /**
      * Observer for the AIGNE instance.
      */

package/lib/cjs/aigne/aigne.js

@@ -31,6 +31,7 @@ class AIGNE {
     static async load(path, options) {
         const { model, agents = [], skills = [], ...aigne } = await (0, index_js_1.load)({ ...options, path });
         return new AIGNE({
+            ...aigne,
             ...options,
             rootDir: aigne.rootDir,
             model: options?.model || model,
@@ -61,6 +62,10 @@ class AIGNE {
             this.skills.push(...options.skills);
         if (options?.agents?.length)
             this.addAgent(...options.agents);
+        if (options?.mcpServer?.agents?.length)
+            this.mcpServer.agents.push(...options.mcpServer.agents);
+        if (options?.cli?.agents?.length)
+            this.cli.agents.push(...options.cli.agents);
         this.observer?.serve();
         this.initProcessExitHandler();
     }
@@ -100,6 +105,12 @@ class AIGNE {
      * Provides indexed access by agent name.
      */
     agents = (0, type_utils_js_1.createAccessorArray)([], (arr, name) => arr.find((i) => i.name === name));
+    mcpServer = {
+        agents: (0, type_utils_js_1.createAccessorArray)([], (arr, name) => arr.find((i) => i.name === name)),
+    };
+    cli = {
+        agents: (0, type_utils_js_1.createAccessorArray)([], (arr, name) => arr.find((i) => i.name === name)),
+    };
     /**
      * Observer for the AIGNE instance.
      */

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

@@ -32,6 +32,7 @@ async function parseAgentFile(path, data) {
         }));
         const baseAgentSchema = zod_1.z.object({
             name: (0, schema_js_1.optionalize)(zod_1.z.string()),
+            alias: (0, schema_js_1.optionalize)(zod_1.z.array(zod_1.z.string())),
             description: (0, schema_js_1.optionalize)(zod_1.z.string()),
             inputSchema: (0, schema_js_1.optionalize)((0, schema_js_1.inputOutputSchema)({ path })).transform((v) => v ? (0, json_schema_to_zod_1.jsonSchemaToZod)(v) : undefined),
             defaultInput: (0, schema_js_1.optionalize)(schema_js_1.defaultInputSchema),

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

@@ -67,6 +67,16 @@ declare const aigneFileSchema: z.ZodObject<{
     } | undefined>;
     agents: z.ZodType<string[] | undefined, z.ZodTypeDef, string[] | undefined>;
     skills: z.ZodType<string[] | undefined, z.ZodTypeDef, string[] | undefined>;
+    mcpServer: z.ZodType<{
+        agents?: string[] | undefined;
+    } | undefined, z.ZodTypeDef, {
+        agents?: string[] | undefined;
+    } | undefined>;
+    cli: z.ZodType<{
+        agents?: string[] | undefined;
+    } | undefined, z.ZodTypeDef, {
+        agents?: string[] | undefined;
+    } | undefined>;
 }, "strip", z.ZodTypeAny, {
     name?: string | undefined;
     description?: string | undefined;
@@ -80,6 +90,12 @@ declare const aigneFileSchema: z.ZodObject<{
         provider?: string | null | undefined;
     } | undefined;
     agents?: string[] | undefined;
+    mcpServer?: {
+        agents?: string[] | undefined;
+    } | undefined;
+    cli?: {
+        agents?: string[] | undefined;
+    } | undefined;
 }, {
     name?: string | undefined;
     description?: string | undefined;
@@ -93,6 +109,12 @@ declare const aigneFileSchema: z.ZodObject<{
         provider?: string | null | undefined;
     } | undefined;
     agents?: string[] | undefined;
+    mcpServer?: {
+        agents?: string[] | undefined;
+    } | undefined;
+    cli?: {
+        agents?: string[] | undefined;
+    } | undefined;
 }>;
 export declare function loadAIGNEFile(path: string): Promise<{
     aigne: z.infer<typeof aigneFileSchema>;

package/lib/cjs/loader/index.js

@@ -1,14 +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.load = load;
 exports.loadAgent = loadAgent;
 exports.loadModel = loadModel;
 exports.loadAIGNEFile = loadAIGNEFile;
 const index_js_1 = require("@aigne/platform-helpers/nodejs/index.js");
-const camelize_ts_1 = __importDefault(require("camelize-ts"));
 const yaml_1 = require("yaml");
 const zod_1 = require("zod");
 const agent_js_1 = require("../agents/agent.js");
@@ -24,8 +20,9 @@ const schema_js_1 = require("./schema.js");
 const AIGNE_FILE_NAME = ["aigne.yaml", "aigne.yml"];
 async function load(options) {
     const { aigne, rootDir } = await loadAIGNEFile(options.path);
-    const agents = await Promise.all((aigne.agents ?? []).map((filename) => loadAgent(index_js_1.nodejs.path.join(rootDir, filename), options)));
-    const skills = await Promise.all((aigne.skills ?? []).map((filename) => loadAgent(index_js_1.nodejs.path.join(rootDir, filename), options)));
+    const allAgentPaths = new Set((0, type_utils_js_1.flat)(aigne.agents, aigne.skills, aigne.mcpServer?.agents, aigne.cli?.agents).map((i) => index_js_1.nodejs.path.join(rootDir, i)));
+    const allAgents = Object.fromEntries(await Promise.all(Array.from(allAgentPaths).map(async (path) => [path, await loadAgent(path, options)])));
+    const pickAgents = (paths) => paths.map((filename) => allAgents[index_js_1.nodejs.path.join(rootDir, filename)]).filter(type_utils_js_1.isNonNullable);
     return {
         ...aigne,
         rootDir,
@@ -35,8 +32,14 @@ async function load(options) {
                 create: (options) => new i(options),
             }
             : i), aigne.model),
-        agents,
-        skills,
+        agents: pickAgents(aigne.agents ?? []),
+        skills: pickAgents(aigne.skills ?? []),
+        mcpServer: {
+            agents: pickAgents(aigne.mcpServer?.agents ?? []),
+        },
+        cli: {
+            agents: pickAgents(aigne.cli?.agents ?? []),
+        },
     };
 }
 async function loadAgent(path, options, agentOptions) {
@@ -176,32 +179,34 @@ async function loadModel(models, model, modelOptions, accessKeyOptions) {
         modelOptions: { ...params, ...modelOptions },
     });
 }
-const aigneFileSchema = zod_1.z.object({
+const aigneFileSchema = (0, schema_js_1.camelizeSchema)(zod_1.z.object({
     name: (0, schema_js_1.optionalize)(zod_1.z.string()),
     description: (0, schema_js_1.optionalize)(zod_1.z.string()),
     model: (0, schema_js_1.optionalize)(zod_1.z.union([
         zod_1.z.string(),
-        zod_1.z.object({
+        (0, schema_js_1.camelizeSchema)(zod_1.z.object({
             provider: zod_1.z.string().nullish(),
             name: zod_1.z.string().nullish(),
             temperature: zod_1.z.number().min(0).max(2).nullish(),
             topP: zod_1.z.number().min(0).nullish(),
             frequencyPenalty: zod_1.z.number().min(-2).max(2).nullish(),
             presencePenalty: zod_1.z.number().min(-2).max(2).nullish(),
-        }),
+        })),
     ])).transform((v) => (typeof v === "string" ? { name: v } : v)),
     agents: (0, schema_js_1.optionalize)(zod_1.z.array(zod_1.z.string())),
     skills: (0, schema_js_1.optionalize)(zod_1.z.array(zod_1.z.string())),
-});
+    mcpServer: (0, schema_js_1.optionalize)(zod_1.z.object({
+        agents: (0, schema_js_1.optionalize)(zod_1.z.array(zod_1.z.string())),
+    })),
+    cli: (0, schema_js_1.optionalize)(zod_1.z.object({
+        agents: (0, schema_js_1.optionalize)(zod_1.z.array(zod_1.z.string())),
+    })),
+}));
 async function loadAIGNEFile(path) {
     const file = await findAIGNEFile(path);
     const raw = await (0, type_utils_js_1.tryOrThrow)(() => index_js_1.nodejs.fs.readFile(file, "utf8"), (error) => new Error(`Failed to load aigne.yaml from ${file}: ${error.message}`));
-    const json = (0, type_utils_js_1.tryOrThrow)(() => (0, camelize_ts_1.default)((0, yaml_1.parse)(raw)), (error) => new Error(`Failed to parse aigne.yaml from ${file}: ${error.message}`));
-    const aigne = (0, type_utils_js_1.tryOrThrow)(() => aigneFileSchema.parse({
-        ...json,
-        model: json.model ?? json.chatModel,
-        skills: json.skills ?? json.tools,
-    }), (error) => new Error(`Failed to validate aigne.yaml from ${file}: ${error.message}`));
+    const json = (0, type_utils_js_1.tryOrThrow)(() => (0, yaml_1.parse)(raw), (error) => new Error(`Failed to parse aigne.yaml from ${file}: ${error.message}`));
+    const aigne = (0, type_utils_js_1.tryOrThrow)(() => aigneFileSchema.parse({ ...json, model: json.model ?? json.chat_model ?? json.chatModel }), (error) => new Error(`Failed to validate aigne.yaml from ${file}: ${error.message}`));
     return { aigne, rootDir: index_js_1.nodejs.path.dirname(file) };
 }
 async function findAIGNEFile(path) {

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

@@ -30,6 +30,6 @@ export declare const defaultInputSchema: z.ZodRecord<z.ZodString, z.ZodUnion<[z.
     $get: string;
 }>, z.ZodUnknown]>>;
 export declare function optionalize<T>(schema: ZodType<T>): ZodType<T | undefined>;
-export declare function camelizeSchema<T>(schema: ZodType<T>, { shallow }?: {
+export declare function camelizeSchema<T extends ZodType>(schema: T, { shallow }?: {
     shallow?: boolean;
-}): ZodType<T>;
+}): T;

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

@@ -61,6 +61,10 @@ export interface AgentOptions<I extends Message = Message, O extends Message = M
      * if not specified
      */
     name?: string;
+    /**
+     * Alias for the agent, can be used to refer to the agent by multiple names, especially in AIGNE CLI
+     */
+    alias?: string[];
     /**
      * Description of the agent
      *
@@ -232,6 +236,10 @@ export declare abstract class Agent<I extends Message = any, O extends Message =
      * Defaults to the class constructor name if not specified in options
      */
     readonly name: string;
+    /**
+     * Alias for the agent, can be used to refer to the agent by multiple names, especially in AIGNE CLI
+     */
+    readonly alias?: string[];
     /**
      * Default topic this agent subscribes to
      *

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

@@ -35,6 +35,12 @@ export interface AIGNEOptions {
      * Agents to use for the AIGNE instance.
      */
     agents?: Agent[];
+    mcpServer?: {
+        agents?: Agent[];
+    };
+    cli?: {
+        agents?: Agent[];
+    };
     /**
      * Limits for the AIGNE instance, such as timeout, max tokens, max invocations, etc.
      */
@@ -112,6 +118,16 @@ export declare class AIGNE<U extends UserContext = UserContext> {
     readonly agents: Agent<any, any>[] & {
         [key: string]: Agent<any, any>;
     };
+    readonly mcpServer: {
+        agents: Agent<any, any>[] & {
+            [key: string]: Agent<any, any>;
+        };
+    };
+    readonly cli: {
+        agents: Agent<any, any>[] & {
+            [key: string]: Agent<any, any>;
+        };
+    };
     /**
      * Observer for the AIGNE instance.
      */

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

@@ -67,6 +67,16 @@ declare const aigneFileSchema: z.ZodObject<{
     } | undefined>;
     agents: z.ZodType<string[] | undefined, z.ZodTypeDef, string[] | undefined>;
     skills: z.ZodType<string[] | undefined, z.ZodTypeDef, string[] | undefined>;
+    mcpServer: z.ZodType<{
+        agents?: string[] | undefined;
+    } | undefined, z.ZodTypeDef, {
+        agents?: string[] | undefined;
+    } | undefined>;
+    cli: z.ZodType<{
+        agents?: string[] | undefined;
+    } | undefined, z.ZodTypeDef, {
+        agents?: string[] | undefined;
+    } | undefined>;
 }, "strip", z.ZodTypeAny, {
     name?: string | undefined;
     description?: string | undefined;
@@ -80,6 +90,12 @@ declare const aigneFileSchema: z.ZodObject<{
         provider?: string | null | undefined;
     } | undefined;
     agents?: string[] | undefined;
+    mcpServer?: {
+        agents?: string[] | undefined;
+    } | undefined;
+    cli?: {
+        agents?: string[] | undefined;
+    } | undefined;
 }, {
     name?: string | undefined;
     description?: string | undefined;
@@ -93,6 +109,12 @@ declare const aigneFileSchema: z.ZodObject<{
         provider?: string | null | undefined;
     } | undefined;
     agents?: string[] | undefined;
+    mcpServer?: {
+        agents?: string[] | undefined;
+    } | undefined;
+    cli?: {
+        agents?: string[] | undefined;
+    } | undefined;
 }>;
 export declare function loadAIGNEFile(path: string): Promise<{
     aigne: z.infer<typeof aigneFileSchema>;

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

@@ -30,6 +30,6 @@ export declare const defaultInputSchema: z.ZodRecord<z.ZodString, z.ZodUnion<[z.
     $get: string;
 }>, z.ZodUnknown]>>;
 export declare function optionalize<T>(schema: ZodType<T>): ZodType<T | undefined>;
-export declare function camelizeSchema<T>(schema: ZodType<T>, { shallow }?: {
+export declare function camelizeSchema<T extends ZodType>(schema: T, { shallow }?: {
     shallow?: boolean;
-}): ZodType<T>;
+}): T;

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

@@ -61,6 +61,10 @@ export interface AgentOptions<I extends Message = Message, O extends Message = M
      * if not specified
      */
     name?: string;
+    /**
+     * Alias for the agent, can be used to refer to the agent by multiple names, especially in AIGNE CLI
+     */
+    alias?: string[];
     /**
      * Description of the agent
      *
@@ -232,6 +236,10 @@ export declare abstract class Agent<I extends Message = any, O extends Message =
      * Defaults to the class constructor name if not specified in options
      */
     readonly name: string;
+    /**
+     * Alias for the agent, can be used to refer to the agent by multiple names, especially in AIGNE CLI
+     */
+    readonly alias?: string[];
     /**
      * Default topic this agent subscribes to
      *

package/lib/esm/agents/agent.js

@@ -57,6 +57,7 @@ export class Agent {
     constructor(options = {}) {
         const { inputSchema, outputSchema } = options;
         this.name = options.name || this.constructor.name;
+        this.alias = options.alias;
         this.description = options.description;
         if (inputSchema)
             checkAgentInputOutputSchema(inputSchema);
@@ -125,6 +126,10 @@ export class Agent {
      * Defaults to the class constructor name if not specified in options
      */
     name;
+    /**
+     * Alias for the agent, can be used to refer to the agent by multiple names, especially in AIGNE CLI
+     */
+    alias;
     /**
      * Default topic this agent subscribes to
      *

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

@@ -35,6 +35,12 @@ export interface AIGNEOptions {
      * Agents to use for the AIGNE instance.
      */
     agents?: Agent[];
+    mcpServer?: {
+        agents?: Agent[];
+    };
+    cli?: {
+        agents?: Agent[];
+    };
     /**
      * Limits for the AIGNE instance, such as timeout, max tokens, max invocations, etc.
      */
@@ -112,6 +118,16 @@ export declare class AIGNE<U extends UserContext = UserContext> {
     readonly agents: Agent<any, any>[] & {
         [key: string]: Agent<any, any>;
     };
+    readonly mcpServer: {
+        agents: Agent<any, any>[] & {
+            [key: string]: Agent<any, any>;
+        };
+    };
+    readonly cli: {
+        agents: Agent<any, any>[] & {
+            [key: string]: Agent<any, any>;
+        };
+    };
     /**
      * Observer for the AIGNE instance.
      */

package/lib/esm/aigne/aigne.js

@@ -28,6 +28,7 @@ export class AIGNE {
     static async load(path, options) {
         const { model, agents = [], skills = [], ...aigne } = await load({ ...options, path });
         return new AIGNE({
+            ...aigne,
             ...options,
             rootDir: aigne.rootDir,
             model: options?.model || model,
@@ -58,6 +59,10 @@ export class AIGNE {
             this.skills.push(...options.skills);
         if (options?.agents?.length)
             this.addAgent(...options.agents);
+        if (options?.mcpServer?.agents?.length)
+            this.mcpServer.agents.push(...options.mcpServer.agents);
+        if (options?.cli?.agents?.length)
+            this.cli.agents.push(...options.cli.agents);
         this.observer?.serve();
         this.initProcessExitHandler();
     }
@@ -97,6 +102,12 @@ export class AIGNE {
      * Provides indexed access by agent name.
      */
     agents = createAccessorArray([], (arr, name) => arr.find((i) => i.name === name));
+    mcpServer = {
+        agents: createAccessorArray([], (arr, name) => arr.find((i) => i.name === name)),
+    };
+    cli = {
+        agents: createAccessorArray([], (arr, name) => arr.find((i) => i.name === name)),
+    };
     /**
      * Observer for the AIGNE instance.
      */

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

@@ -28,6 +28,7 @@ export async function parseAgentFile(path, data) {
         }));
         const baseAgentSchema = z.object({
             name: optionalize(z.string()),
+            alias: optionalize(z.array(z.string())),
             description: optionalize(z.string()),
             inputSchema: optionalize(inputOutputSchema({ path })).transform((v) => v ? jsonSchemaToZod(v) : undefined),
             defaultInput: optionalize(defaultInputSchema),

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

@@ -67,6 +67,16 @@ declare const aigneFileSchema: z.ZodObject<{
     } | undefined>;
     agents: z.ZodType<string[] | undefined, z.ZodTypeDef, string[] | undefined>;
     skills: z.ZodType<string[] | undefined, z.ZodTypeDef, string[] | undefined>;
+    mcpServer: z.ZodType<{
+        agents?: string[] | undefined;
+    } | undefined, z.ZodTypeDef, {
+        agents?: string[] | undefined;
+    } | undefined>;
+    cli: z.ZodType<{
+        agents?: string[] | undefined;
+    } | undefined, z.ZodTypeDef, {
+        agents?: string[] | undefined;
+    } | undefined>;
 }, "strip", z.ZodTypeAny, {
     name?: string | undefined;
     description?: string | undefined;
@@ -80,6 +90,12 @@ declare const aigneFileSchema: z.ZodObject<{
         provider?: string | null | undefined;
     } | undefined;
     agents?: string[] | undefined;
+    mcpServer?: {
+        agents?: string[] | undefined;
+    } | undefined;
+    cli?: {
+        agents?: string[] | undefined;
+    } | undefined;
 }, {
     name?: string | undefined;
     description?: string | undefined;
@@ -93,6 +109,12 @@ declare const aigneFileSchema: z.ZodObject<{
         provider?: string | null | undefined;
     } | undefined;
     agents?: string[] | undefined;
+    mcpServer?: {
+        agents?: string[] | undefined;
+    } | undefined;
+    cli?: {
+        agents?: string[] | undefined;
+    } | undefined;
 }>;
 export declare function loadAIGNEFile(path: string): Promise<{
     aigne: z.infer<typeof aigneFileSchema>;

package/lib/esm/loader/index.js

@@ -1,5 +1,4 @@
 import { nodejs } from "@aigne/platform-helpers/nodejs/index.js";
-import camelize from "camelize-ts";
 import { parse } from "yaml";
 import { z } from "zod";
 import { Agent, FunctionAgent } from "../agents/agent.js";
@@ -8,15 +7,16 @@ import { MCPAgent } from "../agents/mcp-agent.js";
 import { TeamAgent } from "../agents/team-agent.js";
 import { TransformAgent } from "../agents/transform-agent.js";
 import { PromptBuilder } from "../prompt/prompt-builder.js";
-import { isNonNullable, tryOrThrow } from "../utils/type-utils.js";
+import { flat, isNonNullable, tryOrThrow } from "../utils/type-utils.js";
 import { loadAgentFromJsFile } from "./agent-js.js";
 import { loadAgentFromYamlFile } from "./agent-yaml.js";
-import { optionalize } from "./schema.js";
+import { camelizeSchema, optionalize } from "./schema.js";
 const AIGNE_FILE_NAME = ["aigne.yaml", "aigne.yml"];
 export async function load(options) {
     const { aigne, rootDir } = await loadAIGNEFile(options.path);
-    const agents = await Promise.all((aigne.agents ?? []).map((filename) => loadAgent(nodejs.path.join(rootDir, filename), options)));
-    const skills = await Promise.all((aigne.skills ?? []).map((filename) => loadAgent(nodejs.path.join(rootDir, filename), options)));
+    const allAgentPaths = new Set(flat(aigne.agents, aigne.skills, aigne.mcpServer?.agents, aigne.cli?.agents).map((i) => nodejs.path.join(rootDir, i)));
+    const allAgents = Object.fromEntries(await Promise.all(Array.from(allAgentPaths).map(async (path) => [path, await loadAgent(path, options)])));
+    const pickAgents = (paths) => paths.map((filename) => allAgents[nodejs.path.join(rootDir, filename)]).filter(isNonNullable);
     return {
         ...aigne,
         rootDir,
@@ -26,8 +26,14 @@ export async function load(options) {
                 create: (options) => new i(options),
             }
             : i), aigne.model),
-        agents,
-        skills,
+        agents: pickAgents(aigne.agents ?? []),
+        skills: pickAgents(aigne.skills ?? []),
+        mcpServer: {
+            agents: pickAgents(aigne.mcpServer?.agents ?? []),
+        },
+        cli: {
+            agents: pickAgents(aigne.cli?.agents ?? []),
+        },
     };
 }
 export async function loadAgent(path, options, agentOptions) {
@@ -167,32 +173,34 @@ export async function loadModel(models, model, modelOptions, accessKeyOptions) {
         modelOptions: { ...params, ...modelOptions },
     });
 }
-const aigneFileSchema = z.object({
+const aigneFileSchema = camelizeSchema(z.object({
     name: optionalize(z.string()),
     description: optionalize(z.string()),
     model: optionalize(z.union([
         z.string(),
-        z.object({
+        camelizeSchema(z.object({
             provider: z.string().nullish(),
             name: z.string().nullish(),
             temperature: z.number().min(0).max(2).nullish(),
             topP: z.number().min(0).nullish(),
             frequencyPenalty: z.number().min(-2).max(2).nullish(),
             presencePenalty: z.number().min(-2).max(2).nullish(),
-        }),
+        })),
     ])).transform((v) => (typeof v === "string" ? { name: v } : v)),
     agents: optionalize(z.array(z.string())),
     skills: optionalize(z.array(z.string())),
-});
+    mcpServer: optionalize(z.object({
+        agents: optionalize(z.array(z.string())),
+    })),
+    cli: optionalize(z.object({
+        agents: optionalize(z.array(z.string())),
+    })),
+}));
 export async function loadAIGNEFile(path) {
     const file = await findAIGNEFile(path);
     const raw = await tryOrThrow(() => nodejs.fs.readFile(file, "utf8"), (error) => new Error(`Failed to load aigne.yaml from ${file}: ${error.message}`));
-    const json = tryOrThrow(() => camelize(parse(raw)), (error) => new Error(`Failed to parse aigne.yaml from ${file}: ${error.message}`));
-    const aigne = tryOrThrow(() => aigneFileSchema.parse({
-        ...json,
-        model: json.model ?? json.chatModel,
-        skills: json.skills ?? json.tools,
-    }), (error) => new Error(`Failed to validate aigne.yaml from ${file}: ${error.message}`));
+    const json = tryOrThrow(() => parse(raw), (error) => new Error(`Failed to parse aigne.yaml from ${file}: ${error.message}`));
+    const aigne = tryOrThrow(() => aigneFileSchema.parse({ ...json, model: json.model ?? json.chat_model ?? json.chatModel }), (error) => new Error(`Failed to validate aigne.yaml from ${file}: ${error.message}`));
     return { aigne, rootDir: nodejs.path.dirname(file) };
 }
 async function findAIGNEFile(path) {

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

@@ -30,6 +30,6 @@ export declare const defaultInputSchema: z.ZodRecord<z.ZodString, z.ZodUnion<[z.
     $get: string;
 }>, z.ZodUnknown]>>;
 export declare function optionalize<T>(schema: ZodType<T>): ZodType<T | undefined>;
-export declare function camelizeSchema<T>(schema: ZodType<T>, { shallow }?: {
+export declare function camelizeSchema<T extends ZodType>(schema: T, { shallow }?: {
     shallow?: boolean;
-}): ZodType<T>;
+}): T;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@aigne/core",
-  "version": "1.39.0",
+  "version": "1.41.0",
   "description": "AIGNE core library for building AI-powered applications",
   "publishConfig": {
     "access": "public"