@aigne/core 1.59.0 → 1.60.0

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

@@ -80,6 +80,8 @@ export 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)
@@ -180,6 +182,8 @@ export class Agent {
      * each other's roles in a multi-agent system
      */
     description;
+    model;
+    imageModel;
     taskTitle;
     async renderTaskTitle(input) {
         if (!this.taskTitle)
@@ -445,10 +449,9 @@ export 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;
         }
@@ -457,6 +460,13 @@ export 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
      *
@@ -551,7 +561,7 @@ export 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/esm/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/esm/agents/ai-agent.js

@@ -103,7 +103,6 @@ export class AIAgent extends Agent {
     constructor(options) {
         super(options);
         checkArguments("AIAgent", aiAgentOptionsSchema, options);
-        this.model = options.model;
         this.instructions =
             typeof options.instructions === "string"
                 ? PromptBuilder.from(options.instructions)
@@ -131,12 +130,6 @@ export class AIAgent extends 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
      *
@@ -234,7 +227,7 @@ export class AIAgent extends 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({
@@ -251,7 +244,7 @@ export class AIAgent extends 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;
                 stream = stream.pipeThrough(new ExtractMetadataTransform({ start: metadataStart, end: metadataEnd, parse }));
@@ -336,7 +329,7 @@ export class AIAgent extends 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,
         });
@@ -346,7 +339,7 @@ export class AIAgent extends 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/esm/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/esm/agents/chat-model.js

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

package/lib/esm/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/esm/agents/image-agent.js

@@ -4,7 +4,6 @@ import { checkArguments } from "../utils/type-utils.js";
 import { Agent, agentOptionsSchema, } from "./agent.js";
 import { imageModelOutputSchema } from "./image-model.js";
 export const imageAgentOptionsSchema = agentOptionsSchema.extend({
-    model: z.custom().optional(),
     instructions: z.union([z.string(), z.custom()]),
     modelOptions: z.record(z.any()).optional(),
 });
@@ -16,21 +15,19 @@ export class ImageAgent extends Agent {
     constructor(options) {
         super({ ...options, outputSchema: imageModelOutputSchema });
         checkArguments("ImageAgent", imageAgentOptionsSchema, options);
-        this.model = options.model;
         this.instructions =
             typeof options.instructions === "string"
                 ? 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 }));
     }
 }

package/lib/esm/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/esm/agents/team-agent.js

@@ -171,7 +171,10 @@ export class TeamAgent extends Agent {
         for (;;) {
             const output = await 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);
@@ -246,7 +249,7 @@ export class TeamAgent extends 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
@@ -272,7 +275,7 @@ export class TeamAgent extends 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/esm/agents/user-agent.js

@@ -32,7 +32,8 @@ export class UserAgent extends 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/esm/aigne/context.js

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

package/lib/esm/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/esm/loader/agent-yaml.js

@@ -5,7 +5,7 @@ import { z } from "zod";
 import { AIAgentToolChoice } from "../agents/ai-agent.js";
 import { ProcessMode } from "../agents/team-agent.js";
 import { tryOrThrow } from "../utils/type-utils.js";
-import { camelizeSchema, defaultInputSchema, inputOutputSchema, optionalize } from "./schema.js";
+import { camelizeSchema, chatModelSchema, defaultInputSchema, imageModelSchema, inputOutputSchema, optionalize, } from "./schema.js";
 export async function parseAgentFile(path, data) {
     const agentSchema = z.lazy(() => {
         const nestAgentSchema = z.lazy(() => z.union([
@@ -31,6 +31,8 @@ export async function parseAgentFile(path, data) {
             name: optionalize(z.string()),
             alias: optionalize(z.array(z.string())),
             description: optionalize(z.string()),
+            model: optionalize(chatModelSchema),
+            imageModel: optionalize(imageModelSchema),
             taskTitle: optionalize(z.string()),
             taskRenderMode: optionalize(z.union([z.literal("hide"), z.literal("collapse")])),
             inputSchema: optionalize(inputOutputSchema({ path })).transform((v) => v ? jsonSchemaToZod(v) : undefined),
@@ -114,7 +116,10 @@ export async function parseAgentFile(path, data) {
                 .extend(baseAgentSchema.shape),
         ]));
     });
-    return agentSchema.parseAsync(data);
+    return agentSchema.parseAsync({
+        ...data,
+        model: data.model || data.chatModel || data.chat_model,
+    });
 }
 export async function loadAgentFromYamlFile(path) {
     const raw = await tryOrThrow(() => nodejs.fs.readFile(path, "utf8"), (error) => new Error(`Failed to load agent definition from ${path}: ${error.message}`));