@aigne/core 1.72.0-beta.3 → 1.72.0-beta.5

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

@@ -94,6 +94,7 @@ export declare const videoModelInputSchema: z.ZodObject<{
     prompt: string;
     model?: string | undefined;
     modelOptions?: Record<string, unknown> | undefined;
+    outputFileType?: "local" | "url" | "file" | undefined;
     image?: {
         type: "url";
         url: string;
@@ -110,13 +111,13 @@ export declare const videoModelInputSchema: z.ZodObject<{
         filename?: string | undefined;
         mimeType?: string | undefined;
     } | undefined;
-    outputFileType?: "local" | "file" | "url" | undefined;
     size?: string | undefined;
     seconds?: string | undefined;
 }, {
     prompt: string;
     model?: string | undefined;
     modelOptions?: Record<string, unknown> | undefined;
+    outputFileType?: "local" | "url" | "file" | undefined;
     image?: {
         type: "url";
         url: string;
@@ -133,7 +134,6 @@ export declare const videoModelInputSchema: z.ZodObject<{
         filename?: string | undefined;
         mimeType?: string | undefined;
     } | undefined;
-    outputFileType?: "local" | "file" | "url" | undefined;
     size?: string | undefined;
     seconds?: string | undefined;
 }>;
@@ -240,7 +240,6 @@ export declare const videoModelOutputSchema: z.ZodObject<{
         filename?: string | undefined;
         mimeType?: string | undefined;
     })[];
-    model?: string | undefined;
     usage?: {
         inputTokens: number;
         outputTokens: number;
@@ -249,6 +248,7 @@ export declare const videoModelOutputSchema: z.ZodObject<{
         cacheReadInputTokens?: number | undefined;
         creditPrefix?: "$" | "€" | "¥" | undefined;
     } | undefined;
+    model?: string | undefined;
     seconds?: number | undefined;
 }, {
     videos: ({
@@ -267,7 +267,6 @@ export declare const videoModelOutputSchema: z.ZodObject<{
         filename?: string | undefined;
         mimeType?: string | undefined;
     })[];
-    model?: string | undefined;
     usage?: {
         inputTokens: number;
         outputTokens: number;
@@ -276,5 +275,6 @@ export declare const videoModelOutputSchema: z.ZodObject<{
         cacheReadInputTokens?: number | undefined;
         creditPrefix?: "$" | "€" | "¥" | undefined;
     } | undefined;
+    model?: string | undefined;
     seconds?: number | undefined;
 }>;

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

@@ -1,9 +1,6 @@
 import type { AFSOptions } from "@aigne/afs";
 import { type ZodType, z } from "zod";
-import type { AgentClass, AgentHooks, FunctionAgentFn, TaskRenderMode } from "../agents/agent.js";
-import { AIAgentToolChoice } from "../agents/ai-agent.js";
-import { type Role } from "../agents/chat-model.js";
-import { ProcessMode, type ReflectionMode } from "../agents/team-agent.js";
+import type { AgentHooks, TaskRenderMode } from "../agents/agent.js";
 import type { LoadOptions } from "./index.js";
 import { chatModelSchema, imageModelSchema } from "./schema.js";
 export interface HooksSchema {
@@ -45,7 +42,8 @@ export interface AFSContextSchema {
         presets?: Record<string, AFSContextPresetSchema>;
     };
 }
-export interface BaseAgentSchema {
+export interface AgentSchema {
+    type: string;
     name?: string;
     description?: string;
     model?: z.infer<typeof chatModelSchema>;
@@ -67,71 +65,11 @@ export interface BaseAgentSchema {
         context?: AFSContextSchema;
     });
     shareAFS?: boolean;
+    [key: string]: unknown;
 }
-export type Instructions = {
-    role: Exclude<Role, "tool">;
-    content: string;
-    path: string;
-    cacheControl?: {
-        type: "ephemeral";
-        ttl?: "5m" | "1h";
-    };
-}[];
-export interface AIAgentSchema extends BaseAgentSchema {
-    type: "ai";
-    instructions?: Instructions;
-    autoReorderSystemMessages?: boolean;
-    autoMergeSystemMessages?: boolean;
-    inputKey?: string;
-    inputFileKey?: string;
-    outputKey?: string;
-    outputFileKey?: string;
-    toolChoice?: AIAgentToolChoice;
-    toolCallsConcurrency?: number;
-    keepTextInToolUses?: boolean;
-}
-export interface ImageAgentSchema extends BaseAgentSchema {
-    type: "image";
-    instructions: Instructions;
-    inputFileKey?: string;
-}
-export interface MCPAgentSchema extends BaseAgentSchema {
-    type: "mcp";
-    url?: string;
-    command?: string;
-    args?: string[];
-}
-export interface TeamAgentSchema extends BaseAgentSchema {
-    type: "team";
-    mode?: ProcessMode;
-    iterateOn?: string;
-    concurrency?: number;
-    iterateWithPreviousOutput?: boolean;
-    includeAllStepsOutput?: boolean;
-    reflection?: Omit<ReflectionMode, "reviewer"> & {
-        reviewer: NestAgentSchema;
-    };
-}
-export interface TransformAgentSchema extends BaseAgentSchema {
-    type: "transform";
-    jsonata: string;
-}
-export interface FunctionAgentSchema extends BaseAgentSchema {
-    type: "function";
-    process: FunctionAgentFn;
-}
-export interface ThirdAgentSchema extends BaseAgentSchema {
-    agentClass?: AgentClass;
-    type: "";
-    [key: string]: any;
-}
-export type AgentSchema = AIAgentSchema | ImageAgentSchema | MCPAgentSchema | TeamAgentSchema | TransformAgentSchema | FunctionAgentSchema | ThirdAgentSchema;
 export declare function parseAgentFile(path: string, data: any, options: LoadOptions): Promise<AgentSchema>;
 export declare function loadAgentFromYamlFile(path: string, options: LoadOptions): Promise<AgentSchema>;
-export declare const getInstructionsSchema: ({ filepath }: {
-    filepath: string;
-}) => ZodType<Instructions>;
-export declare const getAgentSchema: ({ filepath, options, }: {
+export declare const getAgentSchema: ({ filepath }: {
     filepath: string;
     options?: LoadOptions;
 }) => ZodType<AgentSchema, z.ZodTypeDef, AgentSchema>;

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

@@ -1,17 +1,13 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.getNestAgentSchema = exports.getAgentSchema = exports.getInstructionsSchema = void 0;
+exports.getNestAgentSchema = exports.getAgentSchema = void 0;
 exports.parseAgentFile = parseAgentFile;
 exports.loadAgentFromYamlFile = loadAgentFromYamlFile;
 const json_schema_to_zod_1 = require("@aigne/json-schema-to-zod");
 const index_js_1 = require("@aigne/platform-helpers/nodejs/index.js");
 const yaml_1 = require("yaml");
 const zod_1 = require("zod");
-const ai_agent_js_1 = require("../agents/ai-agent.js");
-const chat_model_js_1 = require("../agents/chat-model.js");
-const team_agent_js_1 = require("../agents/team-agent.js");
 const type_utils_js_1 = require("../utils/type-utils.js");
-const function_agent_js_1 = require("./function-agent.js");
 const schema_js_1 = require("./schema.js");
 async function parseAgentFile(path, data, options) {
     const agentSchema = (0, exports.getAgentSchema)({ filepath: path, options });
@@ -30,52 +26,7 @@ async function loadAgentFromYamlFile(path, options) {
     }, options), (error) => new Error(`Failed to validate agent definition from ${path}: ${error.message}`));
     return agent;
 }
-const instructionItemSchema = (0, schema_js_1.camelizeSchema)(zod_1.z.union([
-    zod_1.z.object({
-        role: chat_model_js_1.roleSchema.default("system"),
-        url: zod_1.z.string(),
-        cacheControl: (0, schema_js_1.optionalize)(zod_1.z.object({
-            type: zod_1.z.literal("ephemeral"),
-            ttl: (0, schema_js_1.optionalize)(zod_1.z.union([zod_1.z.literal("5m"), zod_1.z.literal("1h")])),
-        })),
-    }),
-    zod_1.z.object({
-        role: chat_model_js_1.roleSchema.default("system"),
-        content: zod_1.z.string(),
-        cacheControl: (0, schema_js_1.optionalize)(zod_1.z.object({
-            type: zod_1.z.literal("ephemeral"),
-            ttl: (0, schema_js_1.optionalize)(zod_1.z.union([zod_1.z.literal("5m"), zod_1.z.literal("1h")])),
-        })),
-    }),
-]));
-const parseInstructionItem = ({ filepath }) => async ({ role, cacheControl, ...v }) => {
-    if (role === "tool")
-        throw new Error(`'tool' role is not allowed in instruction item in agent file ${filepath}`);
-    if ("content" in v && typeof v.content === "string") {
-        return { role, content: v.content, path: filepath, cacheControl };
-    }
-    if ("url" in v && typeof v.url === "string") {
-        const url = index_js_1.nodejs.path.isAbsolute(v.url)
-            ? v.url
-            : index_js_1.nodejs.path.join(index_js_1.nodejs.path.dirname(filepath), v.url);
-        return index_js_1.nodejs.fs
-            .readFile(url, "utf8")
-            .then((content) => ({ role, content, path: url, cacheControl }));
-    }
-    throw new Error(`Invalid instruction item in agent file ${filepath}. Expected 'content' or 'url' property`);
-};
-const getInstructionsSchema = ({ filepath }) => zod_1.z
-    .union([zod_1.z.string(), instructionItemSchema, zod_1.z.array(instructionItemSchema)])
-    .transform(async (v) => {
-    if (typeof v === "string")
-        return [{ role: "system", content: v, path: filepath }];
-    if (Array.isArray(v)) {
-        return Promise.all(v.map((item) => parseInstructionItem({ filepath })(item)));
-    }
-    return [await parseInstructionItem({ filepath })(v)];
-});
-exports.getInstructionsSchema = getInstructionsSchema;
-const getAgentSchema = ({ filepath, options, }) => {
+const getAgentSchema = ({ filepath }) => {
     const agentSchema = zod_1.z.lazy(() => {
         const nestAgentSchema = zod_1.z.lazy(() => zod_1.z.union([
             agentSchema,
@@ -111,6 +62,7 @@ const getAgentSchema = ({ filepath, options, }) => {
             }))),
         });
         const baseAgentSchema = zod_1.z.object({
+            type: zod_1.z.string(),
             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()),
@@ -149,94 +101,7 @@ const getAgentSchema = ({ filepath, options, }) => {
             ])),
             shareAFS: (0, schema_js_1.optionalize)(zod_1.z.boolean()),
         });
-        const instructionsSchema = (0, exports.getInstructionsSchema)({ filepath: filepath });
-        return (0, schema_js_1.camelizeSchema)((0, schema_js_1.preprocessSchema)(async (json) => {
-            if (typeof json === "object" &&
-                json &&
-                "type" in json &&
-                typeof json.type === "string" &&
-                !["ai", "image", "mcp", "team", "transform", "function"].includes(json.type)) {
-                if (!options?.require)
-                    throw new Error(`Module loader is not provided to load agent type module ${json.type} from ${filepath}`);
-                const Mod = await options.require(json.type, { parent: filepath });
-                if (typeof Mod?.default?.prototype?.constructor !== "function") {
-                    throw new Error(`The agent type module ${json.type} does not export a default Agent class`);
-                }
-                Object.assign(json, { agentClass: Mod.default });
-            }
-            return json;
-        }, zod_1.z.union([
-            zod_1.z
-                .object({
-                type: zod_1.z.string(),
-                agentClass: zod_1.z.custom((v) => typeof v?.prototype?.constructor === "function"),
-            })
-                .extend(baseAgentSchema.shape)
-                .passthrough(),
-            zod_1.z.discriminatedUnion("type", [
-                zod_1.z
-                    .object({
-                    type: zod_1.z.literal("ai"),
-                    instructions: (0, schema_js_1.optionalize)(instructionsSchema),
-                    autoReorderSystemMessages: (0, schema_js_1.optionalize)(zod_1.z.boolean()),
-                    autoMergeSystemMessages: (0, schema_js_1.optionalize)(zod_1.z.boolean()),
-                    inputKey: (0, schema_js_1.optionalize)(zod_1.z.string()),
-                    outputKey: (0, schema_js_1.optionalize)(zod_1.z.string()),
-                    inputFileKey: (0, schema_js_1.optionalize)(zod_1.z.string()),
-                    outputFileKey: (0, schema_js_1.optionalize)(zod_1.z.string()),
-                    toolChoice: (0, schema_js_1.optionalize)(zod_1.z.nativeEnum(ai_agent_js_1.AIAgentToolChoice)),
-                    toolCallsConcurrency: (0, schema_js_1.optionalize)(zod_1.z.number().int().min(0)),
-                    keepTextInToolUses: (0, schema_js_1.optionalize)(zod_1.z.boolean()),
-                    catchToolsError: (0, schema_js_1.optionalize)(zod_1.z.boolean()),
-                    structuredStreamMode: (0, schema_js_1.optionalize)(zod_1.z.boolean()),
-                })
-                    .extend(baseAgentSchema.shape),
-                zod_1.z
-                    .object({
-                    type: zod_1.z.literal("image"),
-                    instructions: instructionsSchema,
-                    inputFileKey: (0, schema_js_1.optionalize)(zod_1.z.string()),
-                })
-                    .extend(baseAgentSchema.shape),
-                zod_1.z
-                    .object({
-                    type: zod_1.z.literal("mcp"),
-                    url: (0, schema_js_1.optionalize)(zod_1.z.string()),
-                    command: (0, schema_js_1.optionalize)(zod_1.z.string()),
-                    args: (0, schema_js_1.optionalize)(zod_1.z.array(zod_1.z.string())),
-                })
-                    .extend(baseAgentSchema.shape),
-                zod_1.z
-                    .object({
-                    type: zod_1.z.literal("team"),
-                    mode: (0, schema_js_1.optionalize)(zod_1.z.nativeEnum(team_agent_js_1.ProcessMode)),
-                    iterateOn: (0, schema_js_1.optionalize)(zod_1.z.string()),
-                    concurrency: (0, schema_js_1.optionalize)(zod_1.z.number().int().min(1)),
-                    iterateWithPreviousOutput: (0, schema_js_1.optionalize)(zod_1.z.boolean()),
-                    includeAllStepsOutput: (0, schema_js_1.optionalize)(zod_1.z.boolean()),
-                    reflection: (0, schema_js_1.camelizeSchema)((0, schema_js_1.optionalize)(zod_1.z.object({
-                        reviewer: nestAgentSchema,
-                        isApproved: zod_1.z.string(),
-                        maxIterations: (0, schema_js_1.optionalize)(zod_1.z.number().int().min(1)),
-                        returnLastOnMaxIterations: (0, schema_js_1.optionalize)(zod_1.z.boolean()),
-                        customErrorMessage: (0, schema_js_1.optionalize)(zod_1.z.string()),
-                    }))),
-                })
-                    .extend(baseAgentSchema.shape),
-                zod_1.z
-                    .object({
-                    type: zod_1.z.literal("transform"),
-                    jsonata: zod_1.z.string(),
-                })
-                    .extend(baseAgentSchema.shape),
-                zod_1.z
-                    .object({
-                    type: zod_1.z.literal("function"),
-                    process: zod_1.z.preprocess((v) => (typeof v === "string" ? (0, function_agent_js_1.codeToFunctionAgentFn)(v) : v), zod_1.z.custom()),
-                })
-                    .extend(baseAgentSchema.shape),
-            ]),
-        ])));
+        return (0, schema_js_1.camelizeSchema)(baseAgentSchema.passthrough());
     });
     return agentSchema;
 };

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

@@ -0,0 +1,4 @@
+import type { AgentClass } from "../agents/types.ts";
+export declare const builtinAgents: {
+    [type: string]: AgentClass;
+};

package/lib/cjs/loader/agents.js

@@ -0,0 +1,17 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.builtinAgents = void 0;
+const agent_js_1 = require("../agents/agent.js");
+const ai_agent_js_1 = require("../agents/ai-agent.js");
+const image_agent_js_1 = require("../agents/image-agent.js");
+const mcp_agent_js_1 = require("../agents/mcp-agent.js");
+const team_agent_js_1 = require("../agents/team-agent.js");
+const transform_agent_js_1 = require("../agents/transform-agent.js");
+exports.builtinAgents = {
+    ai: ai_agent_js_1.AIAgent,
+    function: agent_js_1.FunctionAgent,
+    image: image_agent_js_1.ImageAgent,
+    mcp: mcp_agent_js_1.MCPAgent,
+    team: team_agent_js_1.TeamAgent,
+    transform: transform_agent_js_1.TransformAgent,
+};

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

@@ -5,9 +5,8 @@ import type { ChatModel } from "../agents/chat-model.js";
 import type { ImageModel } from "../agents/image-model.js";
 import type { AIGNEOptions } from "../aigne/aigne.js";
 import type { MemoryAgent, MemoryAgentOptions } from "../memory/memory.js";
-import { PromptBuilder } from "../prompt/prompt-builder.js";
 import { type PromiseOrValue } from "../utils/type-utils.js";
-import { type Instructions, loadAgentFromYamlFile, type NestAgentSchema } from "./agent-yaml.js";
+import { loadAgentFromYamlFile, type NestAgentSchema } from "./agent-yaml.js";
 export interface LoadOptions {
     memories?: {
         new (parameters?: MemoryAgentOptions): MemoryAgent;
@@ -19,7 +18,10 @@ export interface LoadOptions {
         availableModules?: {
             module: string;
             alias?: string[];
-            create: (options?: Record<string, any>) => PromiseOrValue<AFSModule>;
+            load: (options: {
+                filepath: string;
+                parsed?: object;
+            }) => PromiseOrValue<AFSModule>;
         }[];
     };
     aigne?: z.infer<typeof aigneFileSchema>;
@@ -27,6 +29,9 @@ export interface LoadOptions {
         parent?: string;
     }) => Promise<any>;
 }
+export interface AgentLoadOptions extends LoadOptions {
+    loadNestAgent: (path: string, agent: NestAgentSchema, options: LoadOptions, agentOptions?: AgentOptions<any, any> & Record<string, unknown>) => Promise<Agent>;
+}
 export declare function load(path: string, options?: LoadOptions): Promise<AIGNEOptions>;
 export declare function loadAgent(path: string, options: LoadOptions, agentOptions?: AgentOptions): Promise<Agent>;
 export declare function loadNestAgent(path: string, agent: NestAgentSchema, options: LoadOptions, agentOptions?: AgentOptions<any, any> & Record<string, unknown>): Promise<Agent>;
@@ -67,9 +72,9 @@ declare const aigneFileSchema: z.ZodObject<{
         } | undefined, z.ZodTypeDef, number | {
             $get: string;
         } | undefined>;
-        thinkingEffort: ZodType<number | "high" | "medium" | "low" | "minimal" | {
+        thinkingEffort: ZodType<number | "high" | "low" | "medium" | "minimal" | {
             $get: string;
-        } | undefined, z.ZodTypeDef, number | "high" | "medium" | "low" | "minimal" | {
+        } | undefined, z.ZodTypeDef, number | "high" | "low" | "medium" | "minimal" | {
             $get: string;
         } | undefined>;
     }, z.ZodTypeAny, "passthrough"> | undefined, z.ZodTypeDef, z.objectInputType<{
@@ -98,9 +103,9 @@ declare const aigneFileSchema: z.ZodObject<{
         } | undefined, z.ZodTypeDef, number | {
             $get: string;
         } | undefined>;
-        thinkingEffort: ZodType<number | "high" | "medium" | "low" | "minimal" | {
+        thinkingEffort: ZodType<number | "high" | "low" | "medium" | "minimal" | {
             $get: string;
-        } | undefined, z.ZodTypeDef, number | "high" | "medium" | "low" | "minimal" | {
+        } | undefined, z.ZodTypeDef, number | "high" | "low" | "medium" | "minimal" | {
             $get: string;
         } | undefined>;
     }, z.ZodTypeAny, "passthrough"> | undefined>;
@@ -132,6 +137,7 @@ declare const aigneFileSchema: z.ZodObject<{
         chat?: string | undefined;
     } | undefined>;
 }, "strip", z.ZodTypeAny, {
+    name?: string | undefined;
     model?: z.objectInputType<{
         model: ZodType<string | {
             $get: string;
@@ -158,13 +164,12 @@ declare const aigneFileSchema: z.ZodObject<{
         } | undefined, z.ZodTypeDef, number | {
             $get: string;
         } | undefined>;
-        thinkingEffort: ZodType<number | "high" | "medium" | "low" | "minimal" | {
+        thinkingEffort: ZodType<number | "high" | "low" | "medium" | "minimal" | {
             $get: string;
-        } | undefined, z.ZodTypeDef, number | "high" | "medium" | "low" | "minimal" | {
+        } | undefined, z.ZodTypeDef, number | "high" | "low" | "medium" | "minimal" | {
             $get: string;
         } | undefined>;
     }, z.ZodTypeAny, "passthrough"> | undefined;
-    name?: string | undefined;
     description?: string | undefined;
     imageModel?: z.objectInputType<{
         model: ZodType<string | {
@@ -183,6 +188,7 @@ declare const aigneFileSchema: z.ZodObject<{
         chat?: string | undefined;
     } | undefined;
 }, {
+    name?: string | undefined;
     model?: z.objectInputType<{
         model: ZodType<string | {
             $get: string;
@@ -209,13 +215,12 @@ declare const aigneFileSchema: z.ZodObject<{
         } | undefined, z.ZodTypeDef, number | {
             $get: string;
         } | undefined>;
-        thinkingEffort: ZodType<number | "high" | "medium" | "low" | "minimal" | {
+        thinkingEffort: ZodType<number | "high" | "low" | "medium" | "minimal" | {
             $get: string;
-        } | undefined, z.ZodTypeDef, number | "high" | "medium" | "low" | "minimal" | {
+        } | undefined, z.ZodTypeDef, number | "high" | "low" | "medium" | "minimal" | {
             $get: string;
         } | undefined>;
     }, z.ZodTypeAny, "passthrough"> | undefined;
-    name?: string | undefined;
     description?: string | undefined;
     imageModel?: z.objectInputType<{
         model: ZodType<string | {
@@ -239,5 +244,4 @@ export declare function loadAIGNEFile(path: string): Promise<{
     rootDir: string;
 }>;
 export declare function findAIGNEFile(path: string): Promise<string>;
-export declare function instructionsToPromptBuilder(instructions: Instructions): PromptBuilder;
 export {};

package/lib/cjs/loader/index.js

@@ -6,23 +6,16 @@ exports.loadNestAgent = loadNestAgent;
 exports.parseAgent = parseAgent;
 exports.loadAIGNEFile = loadAIGNEFile;
 exports.findAIGNEFile = findAIGNEFile;
-exports.instructionsToPromptBuilder = instructionsToPromptBuilder;
 const afs_1 = require("@aigne/afs");
 const index_js_1 = require("@aigne/platform-helpers/nodejs/index.js");
 const yaml_1 = require("yaml");
 const zod_1 = require("zod");
 const agent_js_1 = require("../agents/agent.js");
-const ai_agent_js_1 = require("../agents/ai-agent.js");
-const image_agent_js_1 = require("../agents/image-agent.js");
-const mcp_agent_js_1 = require("../agents/mcp-agent.js");
-const team_agent_js_1 = require("../agents/team-agent.js");
-const transform_agent_js_1 = require("../agents/transform-agent.js");
-const prompt_builder_js_1 = require("../prompt/prompt-builder.js");
-const template_js_1 = require("../prompt/template.js");
 const agent_utils_js_1 = require("../utils/agent-utils.js");
 const type_utils_js_1 = require("../utils/type-utils.js");
 const agent_js_js_1 = require("./agent-js.js");
 const agent_yaml_js_1 = require("./agent-yaml.js");
+const agents_js_1 = require("./agents.js");
 const schema_js_1 = require("./schema.js");
 const AIGNE_FILE_NAME = ["aigne.yaml", "aigne.yml"];
 async function load(path, options = {}) {
@@ -163,7 +156,10 @@ async function parseAgent(path, agent, options, agentOptions) {
             const mod = options?.afs?.availableModules?.find((mod) => mod.module === moduleName || mod.alias?.includes(moduleName));
             if (!mod)
                 throw new Error(`AFS module not found: ${typeof m === "string" ? m : m.module}`);
-            const module = await mod.create(typeof m === "string" ? {} : m.options);
+            const module = await mod.load({
+                filepath: path,
+                parsed: typeof m === "string" ? {} : m.options,
+            });
             afs.mount(module);
         }
     }
@@ -195,73 +191,24 @@ async function parseAgent(path, agent, options, agentOptions) {
         skills: [...(agentOptions?.skills || []), ...skills],
         afs: afs || agentOptions?.afs,
     };
-    let instructions;
-    if ("instructions" in agent && agent.instructions && ["ai", "image"].includes(agent.type)) {
-        instructions = instructionsToPromptBuilder(agent.instructions);
-    }
-    switch (agent.type) {
-        case "ai": {
-            return ai_agent_js_1.AIAgent.from({
-                ...baseOptions,
-                instructions,
-            });
-        }
-        case "image": {
-            if (!instructions)
-                throw new Error(`Missing required instructions for image agent at path: ${path}`);
-            return image_agent_js_1.ImageAgent.from({
-                ...baseOptions,
-                instructions,
-            });
-        }
-        case "mcp": {
-            if (agent.url) {
-                return mcp_agent_js_1.MCPAgent.from({
-                    ...baseOptions,
-                    url: agent.url,
-                });
-            }
-            if (agent.command) {
-                return mcp_agent_js_1.MCPAgent.from({
-                    ...baseOptions,
-                    command: agent.command,
-                    args: agent.args,
-                });
-            }
-            throw new Error(`Missing url or command in mcp agent: ${path}`);
-        }
-        case "team": {
-            return team_agent_js_1.TeamAgent.from({
-                ...baseOptions,
-                mode: agent.mode,
-                iterateOn: agent.iterateOn,
-                reflection: agent.reflection && {
-                    ...agent.reflection,
-                    reviewer: await loadNestAgent(path, agent.reflection.reviewer, options),
-                },
-            });
-        }
-        case "transform": {
-            return transform_agent_js_1.TransformAgent.from({
-                ...baseOptions,
-                jsonata: agent.jsonata,
-            });
-        }
-        case "function": {
-            return agent_js_1.FunctionAgent.from({
-                ...baseOptions,
-                process: agent.process,
-            });
+    let agentClass = agents_js_1.builtinAgents[agent.type];
+    if (!agentClass) {
+        if (!options?.require)
+            throw new Error(`Module loader is not provided to load agent type module ${agent.type} from ${path}`);
+        const Mod = await options.require(agent.type, { parent: path });
+        if (typeof Mod?.default?.prototype?.constructor !== "function") {
+            throw new Error(`The agent type module ${agent.type} does not export a default Agent class`);
         }
+        agentClass = Mod.default;
     }
-    if ("agentClass" in agent && agent.agentClass) {
-        return await agent.agentClass.load({
-            filepath: path,
-            parsed: baseOptions,
-            options,
-        });
+    if (!agentClass) {
+        throw new Error(`Unsupported agent type: ${agent.type} from ${path}`);
     }
-    throw new Error(`Unsupported agent type: ${"type" in agent ? agent.type : "unknown"} at path: ${path}`);
+    return await agentClass.load({
+        filepath: path,
+        parsed: baseOptions,
+        options: { ...options, loadNestAgent },
+    });
 }
 async function loadMemory(memories, provider, options) {
     const M = !provider
@@ -317,11 +264,3 @@ async function findAIGNEFile(path) {
     }
     throw new Error(`aigne.yaml not found in ${path}. Please ensure you are in the correct directory or provide a valid path.`);
 }
-function instructionsToPromptBuilder(instructions) {
-    return new prompt_builder_js_1.PromptBuilder({
-        instructions: template_js_1.ChatMessagesTemplate.from((0, template_js_1.parseChatMessages)(instructions.map((i) => ({
-            ...i,
-            options: { workingDir: index_js_1.nodejs.path.dirname(i.path) },
-        })))),
-    });
-}

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

@@ -1,4 +1,6 @@
 import { type ZodType, z } from "zod";
+import { type Role } from "../agents/chat-model.js";
+import { PromptBuilder } from "../prompt/prompt-builder.js";
 export declare const inputOutputSchema: ({ path }: {
     path: string;
 }) => z.ZodUnion<[ZodType<any, z.ZodTypeDef, any>, z.ZodEffects<z.ZodObject<{
@@ -55,9 +57,9 @@ declare const chatModelObjectSchema: z.ZodObject<{
     } | undefined, z.ZodTypeDef, number | {
         $get: string;
     } | undefined>;
-    thinkingEffort: ZodType<number | "high" | "medium" | "low" | "minimal" | {
+    thinkingEffort: ZodType<number | "high" | "low" | "medium" | "minimal" | {
         $get: string;
-    } | undefined, z.ZodTypeDef, number | "high" | "medium" | "low" | "minimal" | {
+    } | undefined, z.ZodTypeDef, number | "high" | "low" | "medium" | "minimal" | {
         $get: string;
     } | undefined>;
 }, "passthrough", z.ZodTypeAny, z.objectOutputType<{
@@ -86,9 +88,9 @@ declare const chatModelObjectSchema: z.ZodObject<{
     } | undefined, z.ZodTypeDef, number | {
         $get: string;
     } | undefined>;
-    thinkingEffort: ZodType<number | "high" | "medium" | "low" | "minimal" | {
+    thinkingEffort: ZodType<number | "high" | "low" | "medium" | "minimal" | {
         $get: string;
-    } | undefined, z.ZodTypeDef, number | "high" | "medium" | "low" | "minimal" | {
+    } | undefined, z.ZodTypeDef, number | "high" | "low" | "medium" | "minimal" | {
         $get: string;
     } | undefined>;
 }, z.ZodTypeAny, "passthrough">, z.objectInputType<{
@@ -117,9 +119,9 @@ declare const chatModelObjectSchema: z.ZodObject<{
     } | undefined, z.ZodTypeDef, number | {
         $get: string;
     } | undefined>;
-    thinkingEffort: ZodType<number | "high" | "medium" | "low" | "minimal" | {
+    thinkingEffort: ZodType<number | "high" | "low" | "medium" | "minimal" | {
         $get: string;
-    } | undefined, z.ZodTypeDef, number | "high" | "medium" | "low" | "minimal" | {
+    } | undefined, z.ZodTypeDef, number | "high" | "low" | "medium" | "minimal" | {
         $get: string;
     } | undefined>;
 }, z.ZodTypeAny, "passthrough">>;
@@ -149,4 +151,17 @@ export declare function camelizeSchema<T extends ZodType>(schema: T, { shallow }
     shallow?: boolean;
 }): T;
 export declare function preprocessSchema<T extends ZodType>(fn: (data: unknown) => unknown, schema: T): T;
+export type Instructions = {
+    role: Exclude<Role, "tool">;
+    content: string;
+    path: string;
+    cacheControl?: {
+        type: "ephemeral";
+        ttl?: "5m" | "1h";
+    };
+}[];
+export declare const getInstructionsSchema: ({ filepath }: {
+    filepath: string;
+}) => ZodType<Instructions>;
+export declare function instructionsToPromptBuilder(instructions: Instructions | string): PromptBuilder;
 export {};