@aigne/core 1.58.3 → 1.60.0

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);
 }

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,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/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;
@@ -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/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;
     }
 }