@aigne/core 1.59.0 → 1.60.0

package/CHANGELOG.md

@@ -1,5 +1,19 @@
 # Changelog
 
+## [1.60.0](https://github.com/AIGNE-io/aigne-framework/compare/core-v1.59.0...core-v1.60.0) (2025-09-10)
+
+
+### Features
+
+* support custom model for every agents ([#472](https://github.com/AIGNE-io/aigne-framework/issues/472)) ([0bda78a](https://github.com/AIGNE-io/aigne-framework/commit/0bda78a2ebf537e953d855882d68cb37d94d1d10))
+
+
+### Dependencies
+
+* The following workspace dependencies were updated
+  * dependencies
+    * @aigne/observability-api bumped to 0.10.3
+
 ## [1.59.0](https://github.com/AIGNE-io/aigne-framework/compare/core-v1.58.3...core-v1.59.0) (2025-09-09)
 
 

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

@@ -8,7 +8,9 @@ import type { Memory, MemoryAgent } from "../memory/memory.js";
 import type { MemoryRecorderInput } from "../memory/recorder.js";
 import type { MemoryRetrieverInput } from "../memory/retriever.js";
 import { type Nullish, type PromiseOrValue, type XOr } from "../utils/type-utils.js";
+import type { ChatModel } from "./chat-model.js";
 import type { GuideRailAgent, GuideRailAgentOutput } from "./guide-rail-agent.js";
+import type { ImageModel } from "./image-model.js";
 import { type TransferAgentOutput } from "./types.js";
 export * from "./types.js";
 export declare const DEFAULT_INPUT_ACTION_GET = "$get";
@@ -73,6 +75,8 @@ export interface AgentOptions<I extends Message = Message, O extends Message = M
      * for documentation and debugging
      */
     description?: string;
+    model?: ChatModel;
+    imageModel?: ImageModel;
     taskTitle?: string | ((input: I) => PromiseOrValue<string | undefined>);
     taskRenderMode?: TaskRenderMode;
     /**
@@ -142,6 +146,8 @@ export interface AgentInvokeOptions<U extends UserContext = UserContext> {
      * agent system and maintain proper isolation and resource control.
      */
     context: Context<U>;
+    model?: ChatModel;
+    imageModel?: ImageModel;
     /**
      * Whether to enable streaming response
      *
@@ -269,6 +275,8 @@ export declare abstract class Agent<I extends Message = any, O extends Message =
      * each other's roles in a multi-agent system
      */
     readonly description?: string;
+    model?: ChatModel;
+    imageModel?: ImageModel;
     taskTitle?: string | ((input: Message) => PromiseOrValue<string | undefined>);
     renderTaskTitle(input: I): Promise<string | undefined>;
     taskRenderMode?: TaskRenderMode;
@@ -422,6 +430,7 @@ export declare abstract class Agent<I extends Message = any, O extends Message =
     private callHooks;
     private mergeDefaultInput;
     protected invokeSkill<I extends Message, O extends Message>(skill: Agent<I, O>, input: I & Message, options: AgentInvokeOptions): Promise<O>;
+    protected invokeChildAgent: Context["invoke"];
     /**
      * Process agent output
      *

package/lib/cjs/agents/agent.js

@@ -128,6 +128,8 @@ class Agent {
         this.name = options.name || this.constructor.name;
         this.alias = options.alias;
         this.description = options.description;
+        this.model = options.model;
+        this.imageModel = options.imageModel;
         this.taskTitle = options.taskTitle;
         this.taskRenderMode = options.taskRenderMode;
         if (inputSchema)
@@ -228,6 +230,8 @@ class Agent {
      * each other's roles in a multi-agent system
      */
     description;
+    model;
+    imageModel;
     taskTitle;
     async renderTaskTitle(input) {
         if (!this.taskTitle)
@@ -493,10 +497,9 @@ class Agent {
         return input;
     }
     async invokeSkill(skill, input, options) {
-        const { context } = options;
         await this.callHooks("onSkillStart", { skill, input }, options);
         try {
-            const output = await context.invoke(skill, input);
+            const output = await this.invokeChildAgent(skill, input, { ...options, streaming: false });
             await this.callHooks("onSkillEnd", { skill, input, output }, options);
             return output;
         }
@@ -505,6 +508,13 @@ class Agent {
             throw error;
         }
     }
+    invokeChildAgent = ((agent, input, options) => {
+        return options.context.invoke(agent, input, {
+            ...options,
+            model: this.model || options.model,
+            imageModel: this.imageModel || options.imageModel,
+        });
+    });
     /**
      * Process agent output
      *
@@ -599,7 +609,7 @@ class Agent {
         return { ...(await this.onGuideRailError(error)), $status: "GuideRailError" };
     }
     async runGuideRails(input, output, options) {
-        const result = await Promise.all((this.guideRails ?? []).map((i) => options.context.invoke(i, { input, output })));
+        const result = await Promise.all((this.guideRails ?? []).map((i) => this.invokeChildAgent(i, { input, output }, { ...options, streaming: false })));
         return result.find((i) => !!i.abort);
     }
     /**

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

@@ -15,12 +15,6 @@ export declare const DEFAULT_FILE_OUTPUT_KEY = "files";
  * @template O The output message type the agent returns
  */
 export interface AIAgentOptions<I extends Message = Message, O extends Message = Message> extends AgentOptions<I, O> {
-    /**
-     * The language model to use for this agent
-     *
-     * If not provided, the agent will use the model from the context
-     */
-    model?: ChatModel;
     /**
      * Instructions to guide the AI model's behavior
      *
@@ -200,12 +194,6 @@ export declare class AIAgent<I extends Message = any, O extends Message = any> e
      * @param options Configuration options for the AI agent
      */
     constructor(options: AIAgentOptions<I, O>);
-    /**
-     * The language model used by this agent
-     *
-     * If not set on the agent, the model from the context will be used
-     */
-    model?: ChatModel;
     /**
      * Instructions for the language model
      *

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

@@ -106,7 +106,6 @@ class AIAgent extends agent_js_1.Agent {
     constructor(options) {
         super(options);
         (0, type_utils_js_1.checkArguments)("AIAgent", exports.aiAgentOptionsSchema, options);
-        this.model = options.model;
         this.instructions =
             typeof options.instructions === "string"
                 ? prompt_builder_js_1.PromptBuilder.from(options.instructions)
@@ -134,12 +133,6 @@ class AIAgent extends agent_js_1.Agent {
             throw new Error("AIAgent requires either inputKey or instructions to be set");
         }
     }
-    /**
-     * The language model used by this agent
-     *
-     * If not set on the agent, the model from the context will be used
-     */
-    model;
     /**
      * Instructions for the language model
      *
@@ -237,7 +230,7 @@ class AIAgent extends agent_js_1.Agent {
      * @protected
      */
     async *process(input, options) {
-        const model = this.model ?? options.context.model;
+        const model = this.model || options.model || options.context.model;
         if (!model)
             throw new Error("model is required to run AIAgent");
         const { toolAgents, ...modelInput } = await this.instructions.build({
@@ -254,7 +247,7 @@ class AIAgent extends agent_js_1.Agent {
         const outputKey = this.outputKey;
         for (;;) {
             const modelOutput = {};
-            let stream = await options.context.invoke(model, { ...modelInput, messages: modelInput.messages.concat(toolCallMessages) }, { ...options, streaming: true });
+            let stream = await this.invokeChildAgent(model, { ...modelInput, messages: modelInput.messages.concat(toolCallMessages) }, { ...options, streaming: true });
             if (this.structuredStreamMode) {
                 const { metadataStart, metadataEnd, parse } = this.customStructuredStreamInstructions || structured_stream_instructions_js_1.STRUCTURED_STREAM_INSTRUCTIONS;
                 stream = stream.pipeThrough(new structured_stream_extractor_js_1.ExtractMetadataTransform({ start: metadataStart, end: metadataEnd, parse }));
@@ -339,7 +332,7 @@ class AIAgent extends agent_js_1.Agent {
      * @protected
      */
     async *_processRouter(input, model, modelInput, options, toolsMap) {
-        const { toolCalls: [call] = [] } = await options.context.invoke(model, modelInput, {
+        const { toolCalls: [call] = [] } = await this.invokeChildAgent(model, modelInput, {
             ...options,
             streaming: false,
         });
@@ -349,7 +342,7 @@ class AIAgent extends agent_js_1.Agent {
         const tool = toolsMap.get(call.function.name);
         if (!tool)
             throw new Error(`Tool not found: ${call.function.name}`);
-        const stream = await options.context.invoke(tool, { ...call.function.arguments, ...input }, { ...options, streaming: true, sourceAgent: this });
+        const stream = await this.invokeChildAgent(tool, { ...call.function.arguments, ...input }, { ...options, streaming: true, sourceAgent: this });
         return yield* stream;
     }
 }

package/lib/cjs/agents/chat-model.d.ts

@@ -3,7 +3,7 @@ import { type PromiseOrValue } from "../utils/type-utils.js";
 import { Agent, type AgentInvokeOptions, type AgentOptions, type AgentProcessResult, type AgentResponse, type AgentResponseStream, type Message } from "./agent.js";
 export declare class StructuredOutputError extends Error {
 }
-export interface ChatModelOptions extends Omit<AgentOptions<ChatModelInput, ChatModelOutput>, "inputSchema" | "outputSchema"> {
+export interface ChatModelOptions extends Omit<AgentOptions<ChatModelInput, ChatModelOutput>, "model" | "inputSchema" | "outputSchema"> {
     model?: string;
     modelOptions?: Omit<ModelOptions, "model">;
 }

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

@@ -94,6 +94,7 @@ class ChatModel extends agent_js_1.Agent {
             inputSchema: chatModelInputSchema,
             outputSchema: chatModelOutputSchema,
             retryOnError,
+            model: undefined,
         });
         this.options = options;
     }

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

@@ -1,9 +1,8 @@
 import { type ZodObject, type ZodType } from "zod";
 import { PromptBuilder } from "../prompt/prompt-builder.js";
 import { Agent, type AgentInvokeOptions, type AgentOptions, type Message } from "./agent.js";
-import { type ImageModel, type ImageModelOutput } from "./image-model.js";
+import { type ImageModelOutput } from "./image-model.js";
 export interface ImageAgentOptions<I extends Message = any, O extends ImageModelOutput = any> extends Omit<AgentOptions<I, O>, "outputSchema"> {
-    model?: ImageModel;
     instructions: string | PromptBuilder;
     modelOptions?: Record<string, any>;
 }
@@ -14,7 +13,6 @@ export declare class ImageAgent<I extends Message = any, O extends ImageModelOut
     tag: string;
     static from<I extends Message = any, O extends ImageModelOutput = any>(options: ImageAgentOptions<I, O>): ImageAgent<I, O>;
     constructor(options: ImageAgentOptions<I, O>);
-    model?: ImageModel;
     instructions: PromptBuilder;
     modelOptions?: Record<string, any>;
     process(input: I, options: AgentInvokeOptions): Promise<O>;

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

@@ -10,7 +10,6 @@ const type_utils_js_1 = require("../utils/type-utils.js");
 const agent_js_1 = require("./agent.js");
 const image_model_js_1 = require("./image-model.js");
 exports.imageAgentOptionsSchema = agent_js_1.agentOptionsSchema.extend({
-    model: zod_1.default.custom().optional(),
     instructions: zod_1.default.union([zod_1.default.string(), zod_1.default.custom()]),
     modelOptions: zod_1.default.record(zod_1.default.any()).optional(),
 });
@@ -22,22 +21,20 @@ class ImageAgent extends agent_js_1.Agent {
     constructor(options) {
         super({ ...options, outputSchema: image_model_js_1.imageModelOutputSchema });
         (0, type_utils_js_1.checkArguments)("ImageAgent", exports.imageAgentOptionsSchema, options);
-        this.model = options.model;
         this.instructions =
             typeof options.instructions === "string"
                 ? prompt_builder_js_1.PromptBuilder.from(options.instructions)
                 : options.instructions;
         this.modelOptions = options.modelOptions;
     }
-    model;
     instructions;
     modelOptions;
     async process(input, options) {
-        const model = this.model ?? options.context.imageModel;
-        if (!model)
+        const imageModel = this.imageModel || options.imageModel || options.context.imageModel;
+        if (!imageModel)
             throw new Error("image model is required to run ImageAgent");
         const { prompt } = await this.instructions.buildImagePrompt({ input });
-        return (await options.context.invoke(model, { ...input, ...this.modelOptions, prompt }, { ...options, streaming: false }));
+        return (await this.invokeChildAgent(imageModel, { ...input, ...this.modelOptions, prompt }, { ...options, streaming: false }));
     }
 }
 exports.ImageAgent = ImageAgent;

package/lib/cjs/agents/image-model.d.ts

@@ -2,7 +2,7 @@ import { z } from "zod";
 import type { PromiseOrValue } from "../utils/type-utils.js";
 import { Agent, type AgentInvokeOptions, type AgentOptions, type AgentProcessResult, type Message } from "./agent.js";
 import { type ChatModelOutputUsage } from "./chat-model.js";
-export interface ImageModelOptions<I extends ImageModelInput = ImageModelInput, O extends ImageModelOutput = ImageModelOutput> extends AgentOptions<I, O> {
+export interface ImageModelOptions<I extends ImageModelInput = ImageModelInput, O extends ImageModelOutput = ImageModelOutput> extends Omit<AgentOptions<I, O>, "model"> {
 }
 export declare abstract class ImageModel<I extends ImageModelInput = ImageModelInput, O extends ImageModelOutput = ImageModelOutput> extends Agent<I, O> {
     tag: string;

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

@@ -177,7 +177,10 @@ class TeamAgent extends agent_js_1.Agent {
         for (;;) {
             const output = await (0, agent_js_1.agentProcessResultToObject)(await this._processNonReflection(previousOutput, options));
             Object.assign(previousOutput, output);
-            const reviewOutput = await options.context.invoke(this.reflection.reviewer, previousOutput);
+            const reviewOutput = await this.invokeChildAgent(this.reflection.reviewer, previousOutput, {
+                ...options,
+                streaming: false,
+            });
             Object.assign(previousOutput, reviewOutput);
             const { isApproved } = this.reflection;
             const approved = typeof isApproved === "string" ? reviewOutput[isApproved] : await isApproved(reviewOutput);
@@ -252,7 +255,7 @@ class TeamAgent extends agent_js_1.Agent {
     async *_processSequential(input, options) {
         const output = {};
         for (const agent of this.skills) {
-            const o = await options.context.invoke(agent, { ...input, ...output }, { ...options, streaming: true });
+            const o = await this.invokeChildAgent(agent, { ...input, ...output }, { ...options, streaming: true });
             const isLast = agent === this.skills[this.skills.length - 1];
             for await (const chunk of o) {
                 // Only yield the chunk if it is the last agent in the sequence
@@ -278,7 +281,7 @@ class TeamAgent extends agent_js_1.Agent {
      * @private
      */
     async *_processParallel(input, options) {
-        const streams = await Promise.all(this.skills.map((agent) => options.context.invoke(agent, input, { ...options, streaming: true })));
+        const streams = await Promise.all(this.skills.map((agent) => this.invokeChildAgent(agent, input, { ...options, streaming: true })));
         const read = async (index, reader) => {
             const promise = reader.read();
             return promise.then((result) => ({ ...result, reader, index }));

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

@@ -35,7 +35,8 @@ class UserAgent extends agent_js_1.Agent {
             return this._process(input, options);
         }
         if (this.activeAgent) {
-            const [output, agent] = await options.context.invoke(this.activeAgent, input, {
+            const [output, agent] = await this.invokeChildAgent(this.activeAgent, input, {
+                ...options,
                 returnActiveAgent: true,
                 streaming: true,
                 // Do not create a new context for the nested agent invocation,

package/lib/cjs/aigne/context.js

@@ -354,6 +354,8 @@ class AIGNEContextShared {
                     hooks: options.hooks,
                     context,
                     streaming: true,
+                    model: options.model,
+                    imageModel: options.imageModel,
                 });
                 for await (const value of stream) {
                     if ((0, agent_js_1.isAgentResponseDelta)(value)) {

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

@@ -1,7 +1,8 @@
-import { type ZodType } from "zod";
+import { type ZodType, z } from "zod";
 import type { AgentHooks, FunctionAgentFn, TaskRenderMode } from "../agents/agent.js";
 import { AIAgentToolChoice } from "../agents/ai-agent.js";
 import { ProcessMode, type ReflectionMode } from "../agents/team-agent.js";
+import { chatModelSchema, imageModelSchema } from "./schema.js";
 export interface HooksSchema {
     priority?: AgentHooks["priority"];
     onStart?: NestAgentSchema;
@@ -20,6 +21,8 @@ export type NestAgentSchema = string | {
 export interface BaseAgentSchema {
     name?: string;
     description?: string;
+    model?: z.infer<typeof chatModelSchema>;
+    imageModel?: z.infer<typeof imageModelSchema>;
     taskTitle?: string;
     taskRenderMode?: TaskRenderMode;
     inputSchema?: ZodType<Record<string, any>>;
@@ -74,5 +77,5 @@ export interface FunctionAgentSchema extends BaseAgentSchema {
     process: FunctionAgentFn;
 }
 export type AgentSchema = AIAgentSchema | ImageAgentSchema | MCPAgentSchema | TeamAgentSchema | TransformAgentSchema | FunctionAgentSchema;
-export declare function parseAgentFile(path: string, data: object): Promise<AgentSchema>;
+export declare function parseAgentFile(path: string, data: any): Promise<AgentSchema>;
 export declare function loadAgentFromYamlFile(path: string): Promise<AgentSchema>;

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

@@ -35,6 +35,8 @@ async function parseAgentFile(path, data) {
             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()),
+            model: (0, schema_js_1.optionalize)(schema_js_1.chatModelSchema),
+            imageModel: (0, schema_js_1.optionalize)(schema_js_1.imageModelSchema),
             taskTitle: (0, schema_js_1.optionalize)(zod_1.z.string()),
             taskRenderMode: (0, schema_js_1.optionalize)(zod_1.z.union([zod_1.z.literal("hide"), zod_1.z.literal("collapse")])),
             inputSchema: (0, schema_js_1.optionalize)((0, schema_js_1.inputOutputSchema)({ path })).transform((v) => v ? (0, json_schema_to_zod_1.jsonSchemaToZod)(v) : undefined),
@@ -118,7 +120,10 @@ async function parseAgentFile(path, data) {
                 .extend(baseAgentSchema.shape),
         ]));
     });
-    return agentSchema.parseAsync(data);
+    return agentSchema.parseAsync({
+        ...data,
+        model: data.model || data.chatModel || data.chat_model,
+    });
 }
 async function loadAgentFromYamlFile(path) {
     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}`));

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

@@ -9,7 +9,7 @@ export interface LoadOptions {
     memories?: {
         new (parameters?: MemoryAgentOptions): MemoryAgent;
     }[];
-    model?: ChatModel | ((model?: z.infer<typeof aigneFileSchema>["chatModel"]) => PromiseOrValue<ChatModel | undefined>);
+    model?: ChatModel | ((model?: z.infer<typeof aigneFileSchema>["model"]) => PromiseOrValue<ChatModel | undefined>);
     imageModel?: ImageModel | ((model?: z.infer<typeof aigneFileSchema>["imageModel"]) => PromiseOrValue<ImageModel | undefined>);
     key?: string | number;
 }
@@ -18,38 +18,22 @@ export declare function loadAgent(path: string, options?: LoadOptions, agentOpti
 declare const aigneFileSchema: z.ZodObject<{
     name: z.ZodType<string | undefined, z.ZodTypeDef, string | undefined>;
     description: z.ZodType<string | undefined, z.ZodTypeDef, string | undefined>;
-    chatModel: z.ZodEffects<z.ZodEffects<z.ZodType<string | {
+    model: z.ZodType<{
         model?: string | undefined;
         temperature?: number | undefined;
         topP?: number | undefined;
         frequencyPenalty?: number | undefined;
         presencePenalty?: number | undefined;
-    } | undefined, z.ZodTypeDef, string | {
-        model?: string | undefined;
-        temperature?: number | undefined;
-        topP?: number | undefined;
-        frequencyPenalty?: number | undefined;
-        presencePenalty?: number | undefined;
-    } | undefined>, string | {
-        model?: string | undefined;
-        temperature?: number | undefined;
-        topP?: number | undefined;
-        frequencyPenalty?: number | undefined;
-        presencePenalty?: number | undefined;
-    } | undefined, unknown>, {
+    } | undefined, z.ZodTypeDef, {
         model?: string | undefined;
         temperature?: number | undefined;
         topP?: number | undefined;
         frequencyPenalty?: number | undefined;
         presencePenalty?: number | undefined;
-    } | undefined, unknown>;
-    imageModel: z.ZodEffects<z.ZodType<string | {
-        model?: string | undefined;
-    } | undefined, z.ZodTypeDef, string | {
-        model?: string | undefined;
-    } | undefined>, {
+    } | undefined>;
+    imageModel: z.ZodType<{
         model?: string | undefined;
-    } | undefined, string | {
+    } | undefined, z.ZodTypeDef, {
         model?: string | undefined;
     } | undefined>;
     agents: z.ZodType<string[] | undefined, z.ZodTypeDef, string[] | undefined>;
@@ -67,20 +51,20 @@ declare const aigneFileSchema: z.ZodObject<{
         chat?: string | undefined;
     } | undefined>;
 }, "strip", z.ZodTypeAny, {
-    name?: string | undefined;
-    description?: string | undefined;
-    skills?: string[] | undefined;
-    imageModel?: {
-        model?: string | undefined;
-    } | undefined;
-    agents?: string[] | undefined;
-    chatModel?: {
+    model?: {
         model?: string | undefined;
         temperature?: number | undefined;
         topP?: number | undefined;
         frequencyPenalty?: number | undefined;
         presencePenalty?: number | undefined;
     } | undefined;
+    name?: string | undefined;
+    description?: string | undefined;
+    imageModel?: {
+        model?: string | undefined;
+    } | undefined;
+    skills?: string[] | undefined;
+    agents?: string[] | undefined;
     mcpServer?: {
         agents?: string[] | undefined;
     } | undefined;
@@ -89,14 +73,20 @@ declare const aigneFileSchema: z.ZodObject<{
         chat?: string | undefined;
     } | undefined;
 }, {
+    model?: {
+        model?: string | undefined;
+        temperature?: number | undefined;
+        topP?: number | undefined;
+        frequencyPenalty?: number | undefined;
+        presencePenalty?: number | undefined;
+    } | undefined;
     name?: string | undefined;
     description?: string | undefined;
-    skills?: string[] | undefined;
-    imageModel?: string | {
+    imageModel?: {
         model?: string | undefined;
     } | undefined;
+    skills?: string[] | undefined;
     agents?: string[] | undefined;
-    chatModel?: unknown;
     mcpServer?: {
         agents?: string[] | undefined;
     } | undefined;

package/lib/cjs/loader/index.js

@@ -27,7 +27,7 @@ async function load(path, options = {}) {
     return {
         ...aigne,
         rootDir,
-        model: typeof options.model === "function" ? await options.model(aigne.chatModel) : options.model,
+        model: typeof options.model === "function" ? await options.model(aigne.model) : options.model,
         imageModel: typeof options.imageModel === "function"
             ? await options.imageModel(aigne.imageModel)
             : options.imageModel,
@@ -92,9 +92,17 @@ 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;
+    const model = agent.model && typeof options?.model === "function"
+        ? await options.model(agent.model)
+        : undefined;
+    const imageModel = agent.imageModel && typeof options?.imageModel === "function"
+        ? await options.imageModel(agent.imageModel)
+        : undefined;
     const baseOptions = {
         ...agentOptions,
         ...agent,
+        model,
+        imageModel,
         skills,
         memory,
         hooks: [
@@ -175,28 +183,8 @@ async function loadMemory(memories, provider, options) {
 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()),
-    chatModel: zod_1.z
-        .preprocess((v) => {
-        if (!(0, type_utils_js_1.isRecord)(v))
-            return v;
-        return { ...v, model: v.model || `${v.provider || ""}:${v.name || ""}` };
-    }, (0, schema_js_1.optionalize)(zod_1.z.union([
-        zod_1.z.string(),
-        (0, schema_js_1.camelizeSchema)(zod_1.z.object({
-            model: (0, schema_js_1.optionalize)(zod_1.z.string()),
-            temperature: (0, schema_js_1.optionalize)(zod_1.z.number().min(0).max(2)),
-            topP: (0, schema_js_1.optionalize)(zod_1.z.number().min(0)),
-            frequencyPenalty: (0, schema_js_1.optionalize)(zod_1.z.number().min(-2).max(2)),
-            presencePenalty: (0, schema_js_1.optionalize)(zod_1.z.number().min(-2).max(2)),
-        })),
-    ])))
-        .transform((v) => (typeof v === "string" ? { model: v } : v)),
-    imageModel: (0, schema_js_1.optionalize)(zod_1.z.union([
-        zod_1.z.string(),
-        (0, schema_js_1.camelizeSchema)(zod_1.z.object({
-            model: (0, schema_js_1.optionalize)(zod_1.z.string()),
-        })),
-    ])).transform((v) => (typeof v === "string" ? { model: v } : v)),
+    model: (0, schema_js_1.optionalize)(schema_js_1.chatModelSchema),
+    imageModel: (0, schema_js_1.optionalize)(schema_js_1.imageModelSchema),
     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({
@@ -211,7 +199,7 @@ 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, 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), (error) => new Error(`Failed to validate aigne.yaml from ${file}: ${error.message}`));
+    const aigne = (0, type_utils_js_1.tryOrThrow)(() => aigneFileSchema.parse({ ...json, model: json.model || json.chatModel || json.chat_model }), (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

@@ -29,7 +29,36 @@ export declare const defaultInputSchema: z.ZodRecord<z.ZodString, z.ZodUnion<[z.
 }, {
     $get: string;
 }>, z.ZodUnknown]>>;
+declare const chatModelObjectSchema: z.ZodObject<{
+    model: ZodType<string | undefined, z.ZodTypeDef, string | undefined>;
+    temperature: ZodType<number | undefined, z.ZodTypeDef, number | undefined>;
+    topP: ZodType<number | undefined, z.ZodTypeDef, number | undefined>;
+    frequencyPenalty: ZodType<number | undefined, z.ZodTypeDef, number | undefined>;
+    presencePenalty: ZodType<number | undefined, z.ZodTypeDef, number | undefined>;
+}, "strip", z.ZodTypeAny, {
+    model?: string | undefined;
+    temperature?: number | undefined;
+    topP?: number | undefined;
+    frequencyPenalty?: number | undefined;
+    presencePenalty?: number | undefined;
+}, {
+    model?: string | undefined;
+    temperature?: number | undefined;
+    topP?: number | undefined;
+    frequencyPenalty?: number | undefined;
+    presencePenalty?: number | undefined;
+}>;
+export declare const chatModelSchema: typeof chatModelObjectSchema;
+declare const imageModelObjectSchema: z.ZodObject<{
+    model: ZodType<string | undefined, z.ZodTypeDef, string | undefined>;
+}, "strip", z.ZodTypeAny, {
+    model?: string | undefined;
+}, {
+    model?: string | undefined;
+}>;
+export declare const imageModelSchema: typeof imageModelObjectSchema;
 export declare function optionalize<T>(schema: ZodType<T>): ZodType<T | undefined>;
 export declare function camelizeSchema<T extends ZodType>(schema: T, { shallow }?: {
     shallow?: boolean;
 }): T;
+export {};

package/lib/cjs/loader/schema.js

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.defaultInputSchema = exports.inputOutputSchema = void 0;
+exports.imageModelSchema = exports.chatModelSchema = exports.defaultInputSchema = exports.inputOutputSchema = void 0;
 exports.optionalize = optionalize;
 exports.camelizeSchema = camelizeSchema;
 const index_js_1 = require("@aigne/platform-helpers/nodejs/index.js");
@@ -60,6 +60,26 @@ exports.defaultInputSchema = zod_1.z.record(zod_1.z.string(), zod_1.z.union([
     }),
     zod_1.z.unknown(),
 ]));
+const chatModelObjectSchema = camelizeSchema(zod_1.z.object({
+    model: optionalize(zod_1.z.string()),
+    temperature: optionalize(zod_1.z.number().min(0).max(2)),
+    topP: optionalize(zod_1.z.number().min(0)),
+    frequencyPenalty: optionalize(zod_1.z.number().min(-2).max(2)),
+    presencePenalty: optionalize(zod_1.z.number().min(-2).max(2)),
+}));
+exports.chatModelSchema = zod_1.z
+    .preprocess((v) => {
+    if (!(0, type_utils_js_1.isRecord)(v))
+        return v;
+    return { ...v, model: v.model || `${v.provider || ""}:${v.name || ""}` };
+}, zod_1.z.union([zod_1.z.string(), chatModelObjectSchema]))
+    .transform((v) => typeof v === "string" ? { model: v } : v);
+const imageModelObjectSchema = camelizeSchema(zod_1.z.object({
+    model: optionalize(zod_1.z.string()),
+}));
+exports.imageModelSchema = zod_1.z
+    .union([zod_1.z.string(), imageModelObjectSchema])
+    .transform((v) => typeof v === "string" ? { model: v } : v);
 function optionalize(schema) {
     return schema.nullish().transform((v) => v ?? undefined);
 }