@aigne/core 1.58.3 → 1.60.0

package/CHANGELOG.md

@@ -1,5 +1,26 @@
 # 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)
+
+
+### Features
+
+* support custom prefer input file type ([#469](https://github.com/AIGNE-io/aigne-framework/issues/469)) ([db0161b](https://github.com/AIGNE-io/aigne-framework/commit/db0161bbac52542c771ee2f40f361636b0668075))
+
 ## [1.58.3](https://github.com/AIGNE-io/aigne-framework/compare/core-v1.58.2...core-v1.58.3) (2025-09-08)
 
 

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,6 +3,10 @@ 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>, "model" | "inputSchema" | "outputSchema"> {
+    model?: string;
+    modelOptions?: Omit<ModelOptions, "model">;
+}
 /**
  * ChatModel is an abstract base class for interacting with Large Language Models (LLMs).
  *
@@ -27,8 +31,9 @@ export declare class StructuredOutputError extends Error {
  * {@includeCode ../../test/agents/chat-model.test.ts#example-chat-model-tools}
  */
 export declare abstract class ChatModel extends Agent<ChatModelInput, ChatModelOutput> {
+    options?: ChatModelOptions | undefined;
     tag: string;
-    constructor(options?: Omit<AgentOptions<ChatModelInput, ChatModelOutput>, "inputSchema" | "outputSchema">);
+    constructor(options?: ChatModelOptions | undefined);
     get credential(): PromiseOrValue<{
         url?: string;
         apiKey?: string;
@@ -145,7 +150,7 @@ export interface ChatModelInput extends Message {
     /**
      * Model-specific configuration options
      */
-    modelOptions?: ChatModelOptions;
+    modelOptions?: ModelOptions;
 }
 /**
  * Message role types
@@ -578,7 +583,7 @@ export type Modality = "text" | "image" | "audio";
  *
  * Contains various parameters for controlling model behavior, such as model name, temperature, etc.
  */
-export interface ChatModelOptions {
+export interface ModelOptions {
     /**
      * Model name or version
      */
@@ -604,6 +609,7 @@ export interface ChatModelOptions {
      */
     parallelToolCalls?: boolean;
     modalities?: Modality[];
+    preferFileInputType?: "file" | "url";
 }
 /**
  * Output message format for ChatModel

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

@@ -76,10 +76,11 @@ exports.StructuredOutputError = StructuredOutputError;
  * {@includeCode ../../test/agents/chat-model.test.ts#example-chat-model-tools}
  */
 class ChatModel extends agent_js_1.Agent {
+    options;
     tag = "ChatModelAgent";
     constructor(options) {
         if (options)
-            (0, type_utils_js_1.checkArguments)("ChatModel", agent_js_1.agentOptionsSchema, options);
+            (0, type_utils_js_1.checkArguments)("ChatModel", chatModelOptionsSchema, options);
         const retryOnError = options?.retryOnError === false
             ? false
             : options?.retryOnError === true
@@ -93,7 +94,9 @@ class ChatModel extends agent_js_1.Agent {
             inputSchema: chatModelInputSchema,
             outputSchema: chatModelOutputSchema,
             retryOnError,
+            model: undefined,
         });
+        this.options = options;
     }
     get credential() {
         return {};
@@ -188,6 +191,18 @@ class ChatModel extends agent_js_1.Agent {
                             mimeType: item.mimeType || ChatModel.getMimeType(item.filename || item.path),
                         };
                     }
+                    if ((input.modelOptions?.preferFileInputType ||
+                        this.options?.modelOptions?.preferFileInputType) !== "url") {
+                        if (item.type === "url") {
+                            return {
+                                ...item,
+                                type: "file",
+                                data: Buffer.from(await (await this.downloadFile(item.url)).arrayBuffer()).toString("base64"),
+                                url: undefined,
+                                mimeType: item.mimeType || ChatModel.getMimeType(item.filename || item.url),
+                            };
+                        }
+                    }
                     return item;
                 })),
             };
@@ -379,7 +394,7 @@ const chatModelInputToolChoiceSchema = zod_1.z.union([
     zod_1.z.literal("required"),
     chatModelInputToolSchema,
 ]);
-const chatModelOptionsSchema = zod_1.z.object({
+const modelOptionsSchema = zod_1.z.object({
     model: zod_1.z.string().optional(),
     temperature: zod_1.z.number().optional(),
     topP: zod_1.z.number().optional(),
@@ -388,12 +403,16 @@ const chatModelOptionsSchema = zod_1.z.object({
     parallelToolCalls: zod_1.z.boolean().optional().default(true),
     modalities: zod_1.z.array(zod_1.z.enum(["text", "image", "audio"])).optional(),
 });
+const chatModelOptionsSchema = agent_js_1.agentOptionsSchema.extend({
+    model: zod_1.z.string().optional(),
+    modelOptions: modelOptionsSchema.optional(),
+});
 const chatModelInputSchema = zod_1.z.object({
     messages: zod_1.z.array(chatModelInputMessageSchema),
     responseFormat: chatModelInputResponseFormatSchema.optional(),
     tools: zod_1.z.array(chatModelInputToolSchema).optional(),
     toolChoice: chatModelInputToolChoiceSchema.optional(),
-    modelOptions: chatModelOptionsSchema.optional(),
+    modelOptions: modelOptionsSchema.optional(),
 });
 var FileOutputType;
 (function (FileOutputType) {

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;
@@ -32,14 +32,14 @@ export declare const imageModelInputSchema: z.ZodObject<{
 }, "strip", z.ZodTypeAny, {
     prompt: string;
     model?: string | undefined;
-    responseFormat?: "base64" | "url" | undefined;
     image?: string | string[] | undefined;
+    responseFormat?: "base64" | "url" | undefined;
     n?: number | undefined;
 }, {
     prompt: string;
     model?: string | undefined;
-    responseFormat?: "base64" | "url" | undefined;
     image?: string | string[] | undefined;
+    responseFormat?: "base64" | "url" | undefined;
     n?: number | undefined;
 }>;
 export interface ImageModelOutput extends Message {

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;