@aigne/core 1.72.0 → 1.74.0-beta

package/dist/loader/index.mjs

@@ -0,0 +1,228 @@
+import { isAgent } from "../utils/agent-utils.mjs";
+import { flat, isNil, isNonNullable, omitBy, tryOrThrow } from "../utils/type-utils.mjs";
+import { Agent } from "../agents/agent.mjs";
+import { camelizeSchema, chatModelSchema, imageModelSchema, optionalize } from "./schema.mjs";
+import { loadAgentFromYamlFile } from "./agent-yaml.mjs";
+import { loadAgentFromJsFile } from "./agent-js.mjs";
+import { builtinAgents } from "./agents.mjs";
+import { AFS } from "@aigne/afs";
+import { nodejs } from "@aigne/utils/nodejs";
+import { parse } from "yaml";
+import { z as z$1 } from "zod";
+
+//#region src/loader/index.ts
+const AIGNE_FILE_NAME = ["aigne.yaml", "aigne.yml"];
+async function load(path, options = {}) {
+	const { aigne, rootDir } = await loadAIGNEFile(path);
+	const flatCliAgents = (cliAgent) => {
+		if (typeof cliAgent === "string") return [cliAgent];
+		return flat(cliAgent.url, cliAgent.agents?.flatMap(flatCliAgents));
+	};
+	const allAgentPaths = new Set(flat(aigne.agents, aigne.skills, aigne.mcpServer?.agents, aigne.cli?.init, aigne.cli?.chat, aigne.cli?.agents?.flatMap((i) => typeof i === "string" ? i : flatCliAgents(i))).map((i) => nodejs.path.join(rootDir, i)));
+	const allAgents = {};
+	for (const path$1 of allAgentPaths) allAgents[path$1] = await loadAgent(path$1, {
+		...options,
+		aigne
+	});
+	const pickAgent = (path$1) => allAgents[nodejs.path.join(rootDir, path$1)];
+	const pickAgents = (paths) => paths.map((filename) => pickAgent(filename)).filter(isNonNullable);
+	const mapCliAgents = (cliAgent) => {
+		if (typeof cliAgent === "string") return { agent: pickAgent(cliAgent) };
+		return {
+			...cliAgent,
+			agent: cliAgent.url ? pickAgent(cliAgent.url) : void 0,
+			agents: cliAgent.agents?.map(mapCliAgents)
+		};
+	};
+	return {
+		...aigne,
+		rootDir,
+		model: typeof options.model === "function" ? await options.model(aigne.model) : options.model,
+		imageModel: typeof options.imageModel === "function" ? aigne.imageModel ? await options.imageModel(aigne.imageModel) : void 0 : options.imageModel,
+		agents: pickAgents(aigne.agents ?? []),
+		skills: pickAgents(aigne.skills ?? []),
+		mcpServer: { agents: pickAgents(aigne.mcpServer?.agents ?? []) },
+		cli: {
+			init: aigne.cli?.init ? pickAgents([aigne.cli.init])[0] : void 0,
+			chat: aigne.cli?.chat ? pickAgents([aigne.cli.chat])[0] : void 0,
+			agents: aigne.cli?.agents?.map(mapCliAgents)
+		}
+	};
+}
+async function loadAgent(path, options, agentOptions) {
+	if ([
+		".js",
+		".mjs",
+		".ts",
+		".mts"
+	].includes(nodejs.path.extname(path))) {
+		const agent = await loadAgentFromJsFile(path, options);
+		if (agent instanceof Agent) return agent;
+		return parseAgent(path, agent, options, agentOptions);
+	}
+	if ([".yml", ".yaml"].includes(nodejs.path.extname(path))) return parseAgent(path, await loadAgentFromYamlFile(path, options), options, agentOptions);
+	throw new Error(`Unsupported agent file type: ${path}`);
+}
+async function loadNestAgent(path, agent, options, agentOptions) {
+	return typeof agent === "object" && "type" in agent ? parseAgent(path, agent, options, agentOptions) : typeof agent === "string" ? loadAgent(nodejs.path.join(nodejs.path.dirname(path), agent), options, agentOptions) : loadAgent(nodejs.path.join(nodejs.path.dirname(path), agent.url), options, {
+		...agentOptions,
+		defaultInput: agent.defaultInput,
+		hooks: await parseHooks(path, agent.hooks, options)
+	});
+}
+async function parseHooks(path, hooks, options) {
+	hooks = [hooks].flat().filter(isNonNullable);
+	if (!hooks.length) return void 0;
+	return await Promise.all(hooks.map(async (hook) => ({
+		priority: hook.priority,
+		onStart: hook.onStart ? await loadNestAgent(path, hook.onStart, options) : void 0,
+		onSuccess: hook.onSuccess ? await loadNestAgent(path, hook.onSuccess, options) : void 0,
+		onError: hook.onError ? await loadNestAgent(path, hook.onError, options) : void 0,
+		onEnd: hook.onEnd ? await loadNestAgent(path, hook.onEnd, options) : void 0,
+		onSkillStart: hook.onSkillStart ? await loadNestAgent(path, hook.onSkillStart, options) : void 0,
+		onSkillEnd: hook.onSkillEnd ? await loadNestAgent(path, hook.onSkillEnd, options) : void 0,
+		onHandoff: hook.onHandoff ? await loadNestAgent(path, hook.onHandoff, options) : void 0
+	})));
+}
+async function loadSkills(path, skills, options) {
+	const loadedSkills = [];
+	for (const skill of skills) loadedSkills.push(await loadNestAgent(path, skill, options));
+	return loadedSkills;
+}
+async function parseAgent(path, agent, options, agentOptions) {
+	if (isAgent(agent)) return agent;
+	const memory = "memory" in agent && options?.memories?.length ? await loadMemory(options.memories, typeof agent.memory === "object" ? agent.memory.provider : void 0, typeof agent.memory === "object" ? agent.memory : {}) : void 0;
+	let afs;
+	if (agent.afs !== false && (!agent.afs || agent.afs === true) && options?.afs?.sharedAFS) afs = options.afs.sharedAFS;
+	else if (agent.afs === true) afs = new AFS();
+	else if (agent.afs) {
+		afs = new AFS({});
+		const loadAFSContextPresets = async (presets) => {
+			return Object.fromEntries(await Promise.all(Object.entries(presets).map(async ([key, value]) => {
+				const [select, per, dedupe] = await Promise.all([
+					value.select,
+					value.per,
+					value.dedupe
+				].map(async (item) => {
+					if (!item?.agent) return void 0;
+					const agent$1 = await loadNestAgent(path, item.agent, options, { afs });
+					return { invoke: (input, options$1) => options$1.context.invoke(agent$1, input, {
+						...options$1,
+						streaming: false
+					}) };
+				}));
+				return [key, {
+					...value,
+					select,
+					per,
+					dedupe
+				}];
+			})));
+		};
+		const context = {
+			search: { presets: await loadAFSContextPresets(agent.afs.context?.search?.presets || {}) },
+			list: { presets: await loadAFSContextPresets(agent.afs.context?.list?.presets || {}) }
+		};
+		afs.options.context = context;
+		for (const m of agent.afs.modules || []) {
+			const moduleName = typeof m === "string" ? m : m.module;
+			const mod = options?.afs?.availableModules?.find((mod$1) => mod$1.module === moduleName || mod$1.alias?.includes(moduleName));
+			if (!mod) throw new Error(`AFS module not found: ${typeof m === "string" ? m : m.module}`);
+			const module = await mod.load({
+				filepath: path,
+				parsed: typeof m === "string" ? {} : m.options
+			});
+			afs.mount(module);
+		}
+	}
+	const skills = "skills" in agent && agent.skills ? await loadSkills(path, agent.skills, {
+		...options,
+		afs: {
+			...options?.afs,
+			sharedAFS: agent.shareAFS && afs || options?.afs?.sharedAFS
+		}
+	}) : [];
+	const model = agent.model && typeof options?.model === "function" ? await options.model({
+		...options.aigne?.model,
+		...omitBy(agent.model, (v) => isNil(v))
+	}) : void 0;
+	const imageModel = agent.imageModel && typeof options?.imageModel === "function" ? await options.imageModel({
+		...options.aigne?.imageModel,
+		...omitBy(agent.imageModel, (v) => isNil(v))
+	}) : void 0;
+	const baseOptions = {
+		...agentOptions,
+		...agent,
+		model,
+		imageModel,
+		memory,
+		hooks: [...await parseHooks(path, agent.hooks, options) ?? [], ...[agentOptions?.hooks].flat().filter(isNonNullable)],
+		skills: [...agentOptions?.skills || [], ...skills],
+		afs: afs || agentOptions?.afs
+	};
+	let agentClass = 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) throw new Error(`Unsupported agent type: ${agent.type} from ${path}`);
+	return await agentClass.load({
+		filepath: path,
+		parsed: baseOptions,
+		options: {
+			...options,
+			loadNestAgent
+		}
+	});
+}
+async function loadMemory(memories, provider, options) {
+	const M = !provider ? memories[0] : memories.find((i) => i.name.toLowerCase().includes(provider.toLowerCase()));
+	if (!M) throw new Error(`Unsupported memory: ${provider}`);
+	return new M(options);
+}
+const cliAgentSchema = z$1.union([z$1.string(), z$1.object({
+	url: optionalize(z$1.string()),
+	name: optionalize(z$1.string()),
+	alias: optionalize(z$1.array(z$1.string())),
+	description: optionalize(z$1.string()),
+	agents: optionalize(z$1.array(z$1.lazy(() => cliAgentSchema)))
+})]);
+const aigneFileSchema = camelizeSchema(z$1.object({
+	name: optionalize(z$1.string()),
+	description: optionalize(z$1.string()),
+	model: optionalize(chatModelSchema),
+	imageModel: optionalize(imageModelSchema),
+	agents: optionalize(z$1.array(z$1.string())),
+	skills: optionalize(z$1.array(z$1.string())),
+	mcpServer: optionalize(z$1.object({ agents: optionalize(z$1.array(z$1.string())) })),
+	cli: optionalize(z$1.object({
+		init: optionalize(z$1.string()),
+		chat: optionalize(z$1.string()),
+		agents: optionalize(z$1.array(cliAgentSchema))
+	}))
+}));
+async function loadAIGNEFile(path) {
+	const file = await findAIGNEFile(path);
+	const raw = await tryOrThrow(() => nodejs.fs.readFile(file, "utf8"), (error) => /* @__PURE__ */ new Error(`Failed to load aigne.yaml from ${file}: ${error.message}`));
+	const json = tryOrThrow(() => parse(raw), (error) => /* @__PURE__ */ new Error(`Failed to parse aigne.yaml from ${file}: ${error.message}`));
+	return {
+		aigne: tryOrThrow(() => aigneFileSchema.parse({
+			...json,
+			model: json.model || json.chatModel || json.chat_model
+		}), (error) => /* @__PURE__ */ new Error(`Failed to validate aigne.yaml from ${file}: ${error.message}`)),
+		rootDir: nodejs.path.dirname(file)
+	};
+}
+async function findAIGNEFile(path) {
+	const possibleFiles = AIGNE_FILE_NAME.includes(nodejs.path.basename(path)) ? [path] : AIGNE_FILE_NAME.map((name) => nodejs.path.join(path, name));
+	for (const file of possibleFiles) try {
+		if ((await nodejs.fs.stat(file)).isFile()) return file;
+	} catch {}
+	throw new Error(`aigne.yaml not found in ${path}. Please ensure you are in the correct directory or provide a valid path.`);
+}
+
+//#endregion
+export { findAIGNEFile, load, loadAIGNEFile, loadAgent, loadNestAgent, parseAgent };
+//# sourceMappingURL=index.mjs.map

package/dist/loader/index.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.mjs","names":["path","agent","options","mod","z"],"sources":["../../src/loader/index.ts"],"sourcesContent":["import { AFS, type AFSContext, type AFSContextPreset, type AFSModule } from \"@aigne/afs\";\nimport { nodejs } from \"@aigne/utils/nodejs\";\nimport { parse } from \"yaml\";\nimport { type ZodType, z } from \"zod\";\nimport { Agent, type AgentHooks, type AgentOptions } from \"../agents/agent.js\";\nimport type { ChatModel } from \"../agents/chat-model.js\";\nimport type { ImageModel } from \"../agents/image-model.js\";\nimport type { AIGNEOptions } from \"../aigne/aigne.js\";\nimport type { AIGNECLIAgent } from \"../aigne/type.js\";\nimport type { MemoryAgent, MemoryAgentOptions } from \"../memory/memory.js\";\nimport { isAgent } from \"../utils/agent-utils.js\";\nimport {\n  flat,\n  isNil,\n  isNonNullable,\n  omitBy,\n  type PromiseOrValue,\n  tryOrThrow,\n} from \"../utils/type-utils.js\";\nimport { loadAgentFromJsFile } from \"./agent-js.js\";\nimport {\n  type AFSContextPresetSchema,\n  type HooksSchema,\n  loadAgentFromYamlFile,\n  type NestAgentSchema,\n} from \"./agent-yaml.js\";\nimport { builtinAgents } from \"./agents.js\";\nimport { camelizeSchema, chatModelSchema, imageModelSchema, optionalize } from \"./schema.js\";\n\nconst AIGNE_FILE_NAME = [\"aigne.yaml\", \"aigne.yml\"];\n\nexport interface LoadOptions {\n  memories?: { new (parameters?: MemoryAgentOptions): MemoryAgent }[];\n  model?:\n    | ChatModel\n    | ((model?: z.infer<typeof aigneFileSchema>[\"model\"]) => PromiseOrValue<ChatModel | undefined>);\n  imageModel?:\n    | ImageModel\n    | ((\n        model?: z.infer<typeof aigneFileSchema>[\"imageModel\"],\n      ) => PromiseOrValue<ImageModel | undefined>);\n  afs?: {\n    sharedAFS?: AFS;\n    availableModules?: {\n      module: string;\n      alias?: string[];\n      load: (options: { filepath: string; parsed?: object }) => PromiseOrValue<AFSModule>;\n    }[];\n  };\n  aigne?: z.infer<typeof aigneFileSchema>;\n  require?: (modulePath: string, options: { parent?: string }) => Promise<any>;\n}\n\nexport interface AgentLoadOptions extends LoadOptions {\n  loadNestAgent: (\n    path: string,\n    agent: NestAgentSchema,\n    options: LoadOptions,\n    agentOptions?: AgentOptions<any, any> & Record<string, unknown>,\n  ) => Promise<Agent>;\n}\n\nexport async function load(path: string, options: LoadOptions = {}): Promise<AIGNEOptions> {\n  const { aigne, rootDir } = await loadAIGNEFile(path);\n\n  const flatCliAgents = (cliAgent: CliAgent): string[] => {\n    if (typeof cliAgent === \"string\") return [cliAgent];\n    return flat(cliAgent.url, cliAgent.agents?.flatMap(flatCliAgents));\n  };\n\n  const allAgentPaths = new Set(\n    flat(\n      aigne.agents,\n      aigne.skills,\n      aigne.mcpServer?.agents,\n      aigne.cli?.init,\n      aigne.cli?.chat,\n      aigne.cli?.agents?.flatMap((i) => (typeof i === \"string\" ? i : flatCliAgents(i))),\n    ).map((i) => nodejs.path.join(rootDir, i)),\n  );\n\n  const allAgents: { [path: string]: Agent } = {};\n\n  for (const path of allAgentPaths) {\n    allAgents[path] = await loadAgent(path, { ...options, aigne });\n  }\n\n  const pickAgent = (path: string) => allAgents[nodejs.path.join(rootDir, path)];\n\n  const pickAgents = (paths: string[]) =>\n    paths.map((filename) => pickAgent(filename)).filter(isNonNullable);\n\n  const mapCliAgents = (cliAgent: CliAgent): AIGNECLIAgent => {\n    if (typeof cliAgent === \"string\") return { agent: pickAgent(cliAgent) };\n\n    return {\n      ...cliAgent,\n      agent: cliAgent.url ? pickAgent(cliAgent.url) : undefined,\n      agents: cliAgent.agents?.map(mapCliAgents),\n    };\n  };\n\n  return {\n    ...aigne,\n    rootDir,\n    model: typeof options.model === \"function\" ? await options.model(aigne.model) : options.model,\n    imageModel:\n      typeof options.imageModel === \"function\"\n        ? aigne.imageModel\n          ? await options.imageModel(aigne.imageModel)\n          : undefined\n        : options.imageModel,\n    agents: pickAgents(aigne.agents ?? []),\n    skills: pickAgents(aigne.skills ?? []),\n    mcpServer: {\n      agents: pickAgents(aigne.mcpServer?.agents ?? []),\n    },\n    cli: {\n      init: aigne.cli?.init ? pickAgents([aigne.cli.init])[0] : undefined,\n      chat: aigne.cli?.chat ? pickAgents([aigne.cli.chat])[0] : undefined,\n      agents: aigne.cli?.agents?.map(mapCliAgents),\n    },\n  };\n}\n\nexport async function loadAgent(\n  path: string,\n  options: LoadOptions,\n  agentOptions?: AgentOptions,\n): Promise<Agent> {\n  if ([\".js\", \".mjs\", \".ts\", \".mts\"].includes(nodejs.path.extname(path))) {\n    const agent = await loadAgentFromJsFile(path, options);\n    if (agent instanceof Agent) return agent;\n    return parseAgent(path, agent, options, agentOptions);\n  }\n\n  if ([\".yml\", \".yaml\"].includes(nodejs.path.extname(path))) {\n    const agent = await loadAgentFromYamlFile(path, options);\n\n    return parseAgent(path, agent, options, agentOptions);\n  }\n\n  throw new Error(`Unsupported agent file type: ${path}`);\n}\n\nexport async function loadNestAgent(\n  path: string,\n  agent: NestAgentSchema,\n  options: LoadOptions,\n  agentOptions?: AgentOptions<any, any> & Record<string, unknown>,\n): Promise<Agent> {\n  return typeof agent === \"object\" && \"type\" in agent\n    ? parseAgent(path, agent, options, agentOptions)\n    : typeof agent === \"string\"\n      ? loadAgent(nodejs.path.join(nodejs.path.dirname(path), agent), options, agentOptions)\n      : loadAgent(nodejs.path.join(nodejs.path.dirname(path), agent.url), options, {\n          ...agentOptions,\n          defaultInput: agent.defaultInput,\n          hooks: await parseHooks(path, agent.hooks, options),\n        });\n}\n\nasync function parseHooks(\n  path: string,\n  hooks: HooksSchema | HooksSchema[] | undefined,\n  options: LoadOptions,\n): Promise<AgentHooks[] | undefined> {\n  hooks = [hooks].flat().filter(isNonNullable);\n  if (!hooks.length) return undefined;\n\n  type AllHooks = Required<AgentHooks>;\n\n  return await Promise.all(\n    hooks.map(\n      async (hook): Promise<{ [key in keyof AllHooks]: AllHooks[key] | undefined }> => ({\n        priority: hook.priority,\n        onStart: hook.onStart ? await loadNestAgent(path, hook.onStart, options) : undefined,\n        onSuccess: hook.onSuccess ? await loadNestAgent(path, hook.onSuccess, options) : undefined,\n        onError: hook.onError ? await loadNestAgent(path, hook.onError, options) : undefined,\n        onEnd: hook.onEnd ? await loadNestAgent(path, hook.onEnd, options) : undefined,\n        onSkillStart: hook.onSkillStart\n          ? await loadNestAgent(path, hook.onSkillStart, options)\n          : undefined,\n        onSkillEnd: hook.onSkillEnd\n          ? await loadNestAgent(path, hook.onSkillEnd, options)\n          : undefined,\n        onHandoff: hook.onHandoff ? await loadNestAgent(path, hook.onHandoff, options) : undefined,\n      }),\n    ),\n  );\n}\n\nasync function loadSkills(\n  path: string,\n  skills: NestAgentSchema[],\n  options: LoadOptions,\n): Promise<Agent[]> {\n  const loadedSkills: Agent[] = [];\n  for (const skill of skills) {\n    loadedSkills.push(await loadNestAgent(path, skill, options));\n  }\n  return loadedSkills;\n}\n\nexport async function parseAgent(\n  path: string,\n  agent: Awaited<ReturnType<typeof loadAgentFromYamlFile>>,\n  options: LoadOptions,\n  agentOptions?: AgentOptions,\n): Promise<Agent> {\n  if (isAgent(agent)) return agent;\n\n  const memory =\n    \"memory\" in agent && options?.memories?.length\n      ? await loadMemory(\n          options.memories,\n          typeof agent.memory === \"object\" ? agent.memory.provider : undefined,\n          typeof agent.memory === \"object\" ? agent.memory : {},\n        )\n      : undefined;\n\n  let afs: AFS | undefined;\n  if (agent.afs !== false && (!agent.afs || agent.afs === true) && options?.afs?.sharedAFS) {\n    afs = options.afs.sharedAFS;\n  } else if (agent.afs === true) {\n    afs = new AFS();\n  } else if (agent.afs) {\n    afs = new AFS({});\n\n    const loadAFSContextPresets = async (\n      presets: Record<string, AFSContextPresetSchema>,\n    ): Promise<Record<string, AFSContextPreset>> => {\n      return Object.fromEntries(\n        await Promise.all(\n          Object.entries(presets).map<Promise<[string, AFSContextPreset]>>(async ([key, value]) => {\n            const [select, per, dedupe] = await Promise.all(\n              [value.select, value.per, value.dedupe].map(async (item) => {\n                if (!item?.agent) return undefined;\n                const agent = await loadNestAgent(path, item.agent, options, { afs });\n                return {\n                  invoke: (input: any, options: any) =>\n                    options.context.invoke(agent, input, {\n                      ...options,\n                      streaming: false,\n                    }),\n                };\n              }),\n            );\n\n            return [key, { ...value, select, per, dedupe }];\n          }),\n        ),\n      );\n    };\n\n    const context: AFSContext = {\n      search: {\n        presets: await loadAFSContextPresets(agent.afs.context?.search?.presets || {}),\n      },\n      list: {\n        presets: await loadAFSContextPresets(agent.afs.context?.list?.presets || {}),\n      },\n    };\n\n    afs.options.context = context;\n\n    for (const m of agent.afs.modules || []) {\n      const moduleName = typeof m === \"string\" ? m : m.module;\n\n      const mod = options?.afs?.availableModules?.find(\n        (mod) => mod.module === moduleName || mod.alias?.includes(moduleName),\n      );\n      if (!mod) throw new Error(`AFS module not found: ${typeof m === \"string\" ? m : m.module}`);\n\n      const module = await mod.load({\n        filepath: path,\n        parsed: typeof m === \"string\" ? {} : m.options,\n      });\n\n      afs.mount(module);\n    }\n  }\n\n  const skills =\n    \"skills\" in agent && agent.skills\n      ? await loadSkills(path, agent.skills, {\n          ...options,\n          afs: { ...options?.afs, sharedAFS: (agent.shareAFS && afs) || options?.afs?.sharedAFS },\n        })\n      : [];\n\n  const model =\n    agent.model && typeof options?.model === \"function\"\n      ? await options.model({ ...options.aigne?.model, ...omitBy(agent.model, (v) => isNil(v)) })\n      : undefined;\n  const imageModel =\n    agent.imageModel && typeof options?.imageModel === \"function\"\n      ? await options.imageModel({\n          ...options.aigne?.imageModel,\n          ...omitBy(agent.imageModel, (v) => isNil(v)),\n        })\n      : undefined;\n\n  const baseOptions: AgentOptions<any, any> = {\n    ...agentOptions,\n    ...agent,\n    model,\n    imageModel,\n    memory,\n    hooks: [\n      ...((await parseHooks(path, agent.hooks, options)) ?? []),\n      ...[agentOptions?.hooks].flat().filter(isNonNullable),\n    ],\n    skills: [...(agentOptions?.skills || []), ...skills],\n    afs: afs || agentOptions?.afs,\n  };\n\n  let agentClass = builtinAgents[agent.type];\n\n  if (!agentClass) {\n    if (!options?.require)\n      throw new Error(\n        `Module loader is not provided to load agent type module ${agent.type} from ${path}`,\n      );\n    const Mod = await options.require(agent.type, { parent: path });\n    if (typeof Mod?.default?.prototype?.constructor !== \"function\") {\n      throw new Error(`The agent type module ${agent.type} does not export a default Agent class`);\n    }\n\n    agentClass = Mod.default;\n  }\n\n  if (!agentClass) {\n    throw new Error(`Unsupported agent type: ${agent.type} from ${path}`);\n  }\n\n  return await agentClass.load({\n    filepath: path,\n    parsed: baseOptions,\n    options: { ...options, loadNestAgent },\n  });\n}\n\nasync function loadMemory(\n  memories: NonNullable<LoadOptions[\"memories\"]>,\n  provider?: string,\n  options?: MemoryAgentOptions,\n) {\n  const M = !provider\n    ? memories[0]\n    : memories.find((i) => i.name.toLowerCase().includes(provider.toLowerCase()));\n  if (!M) throw new Error(`Unsupported memory: ${provider}`);\n\n  return new M(options);\n}\n\ntype CliAgent =\n  | string\n  | {\n      url?: string;\n      name?: string;\n      alias?: string[];\n      description?: string;\n      agents?: CliAgent[];\n    };\n\nconst cliAgentSchema: ZodType<CliAgent> = z.union([\n  z.string(),\n  z.object({\n    url: optionalize(z.string()),\n    name: optionalize(z.string()),\n    alias: optionalize(z.array(z.string())),\n    description: optionalize(z.string()),\n    agents: optionalize(z.array(z.lazy(() => cliAgentSchema))),\n  }),\n]);\n\nconst aigneFileSchema = camelizeSchema(\n  z.object({\n    name: optionalize(z.string()),\n    description: optionalize(z.string()),\n    model: optionalize(chatModelSchema),\n    imageModel: optionalize(imageModelSchema),\n    agents: optionalize(z.array(z.string())),\n    skills: optionalize(z.array(z.string())),\n    mcpServer: optionalize(\n      z.object({\n        agents: optionalize(z.array(z.string())),\n      }),\n    ),\n    cli: optionalize(\n      z.object({\n        init: optionalize(z.string()),\n        chat: optionalize(z.string()),\n        agents: optionalize(z.array(cliAgentSchema)),\n      }),\n    ),\n  }),\n);\n\nexport async function loadAIGNEFile(path: string): Promise<{\n  aigne: z.infer<typeof aigneFileSchema>;\n  rootDir: string;\n}> {\n  const file = await findAIGNEFile(path);\n\n  const raw = await tryOrThrow(\n    () => nodejs.fs.readFile(file, \"utf8\"),\n    (error) => new Error(`Failed to load aigne.yaml from ${file}: ${error.message}`),\n  );\n\n  const json = tryOrThrow(\n    () => parse(raw),\n    (error) => new Error(`Failed to parse aigne.yaml from ${file}: ${error.message}`),\n  );\n\n  const aigne = tryOrThrow(\n    () =>\n      aigneFileSchema.parse({ ...json, model: json.model || json.chatModel || json.chat_model }),\n    (error) => new Error(`Failed to validate aigne.yaml from ${file}: ${error.message}`),\n  );\n\n  return { aigne, rootDir: nodejs.path.dirname(file) };\n}\n\nexport async function findAIGNEFile(path: string): Promise<string> {\n  const possibleFiles = AIGNE_FILE_NAME.includes(nodejs.path.basename(path))\n    ? [path]\n    : AIGNE_FILE_NAME.map((name) => nodejs.path.join(path, name));\n\n  for (const file of possibleFiles) {\n    try {\n      const stat = await nodejs.fs.stat(file);\n\n      if (stat.isFile()) return file;\n    } catch {}\n  }\n\n  throw new Error(\n    `aigne.yaml not found in ${path}. Please ensure you are in the correct directory or provide a valid path.`,\n  );\n}\n"],"mappings":";;;;;;;;;;;;;AA6BA,MAAM,kBAAkB,CAAC,cAAc,YAAY;AAiCnD,eAAsB,KAAK,MAAc,UAAuB,EAAE,EAAyB;CACzF,MAAM,EAAE,OAAO,YAAY,MAAM,cAAc,KAAK;CAEpD,MAAM,iBAAiB,aAAiC;AACtD,MAAI,OAAO,aAAa,SAAU,QAAO,CAAC,SAAS;AACnD,SAAO,KAAK,SAAS,KAAK,SAAS,QAAQ,QAAQ,cAAc,CAAC;;CAGpE,MAAM,gBAAgB,IAAI,IACxB,KACE,MAAM,QACN,MAAM,QACN,MAAM,WAAW,QACjB,MAAM,KAAK,MACX,MAAM,KAAK,MACX,MAAM,KAAK,QAAQ,SAAS,MAAO,OAAO,MAAM,WAAW,IAAI,cAAc,EAAE,CAAE,CAClF,CAAC,KAAK,MAAM,OAAO,KAAK,KAAK,SAAS,EAAE,CAAC,CAC3C;CAED,MAAM,YAAuC,EAAE;AAE/C,MAAK,MAAMA,UAAQ,cACjB,WAAUA,UAAQ,MAAM,UAAUA,QAAM;EAAE,GAAG;EAAS;EAAO,CAAC;CAGhE,MAAM,aAAa,WAAiB,UAAU,OAAO,KAAK,KAAK,SAASA,OAAK;CAE7E,MAAM,cAAc,UAClB,MAAM,KAAK,aAAa,UAAU,SAAS,CAAC,CAAC,OAAO,cAAc;CAEpE,MAAM,gBAAgB,aAAsC;AAC1D,MAAI,OAAO,aAAa,SAAU,QAAO,EAAE,OAAO,UAAU,SAAS,EAAE;AAEvE,SAAO;GACL,GAAG;GACH,OAAO,SAAS,MAAM,UAAU,SAAS,IAAI,GAAG;GAChD,QAAQ,SAAS,QAAQ,IAAI,aAAa;GAC3C;;AAGH,QAAO;EACL,GAAG;EACH;EACA,OAAO,OAAO,QAAQ,UAAU,aAAa,MAAM,QAAQ,MAAM,MAAM,MAAM,GAAG,QAAQ;EACxF,YACE,OAAO,QAAQ,eAAe,aAC1B,MAAM,aACJ,MAAM,QAAQ,WAAW,MAAM,WAAW,GAC1C,SACF,QAAQ;EACd,QAAQ,WAAW,MAAM,UAAU,EAAE,CAAC;EACtC,QAAQ,WAAW,MAAM,UAAU,EAAE,CAAC;EACtC,WAAW,EACT,QAAQ,WAAW,MAAM,WAAW,UAAU,EAAE,CAAC,EAClD;EACD,KAAK;GACH,MAAM,MAAM,KAAK,OAAO,WAAW,CAAC,MAAM,IAAI,KAAK,CAAC,CAAC,KAAK;GAC1D,MAAM,MAAM,KAAK,OAAO,WAAW,CAAC,MAAM,IAAI,KAAK,CAAC,CAAC,KAAK;GAC1D,QAAQ,MAAM,KAAK,QAAQ,IAAI,aAAa;GAC7C;EACF;;AAGH,eAAsB,UACpB,MACA,SACA,cACgB;AAChB,KAAI;EAAC;EAAO;EAAQ;EAAO;EAAO,CAAC,SAAS,OAAO,KAAK,QAAQ,KAAK,CAAC,EAAE;EACtE,MAAM,QAAQ,MAAM,oBAAoB,MAAM,QAAQ;AACtD,MAAI,iBAAiB,MAAO,QAAO;AACnC,SAAO,WAAW,MAAM,OAAO,SAAS,aAAa;;AAGvD,KAAI,CAAC,QAAQ,QAAQ,CAAC,SAAS,OAAO,KAAK,QAAQ,KAAK,CAAC,CAGvD,QAAO,WAAW,MAFJ,MAAM,sBAAsB,MAAM,QAAQ,EAEzB,SAAS,aAAa;AAGvD,OAAM,IAAI,MAAM,gCAAgC,OAAO;;AAGzD,eAAsB,cACpB,MACA,OACA,SACA,cACgB;AAChB,QAAO,OAAO,UAAU,YAAY,UAAU,QAC1C,WAAW,MAAM,OAAO,SAAS,aAAa,GAC9C,OAAO,UAAU,WACf,UAAU,OAAO,KAAK,KAAK,OAAO,KAAK,QAAQ,KAAK,EAAE,MAAM,EAAE,SAAS,aAAa,GACpF,UAAU,OAAO,KAAK,KAAK,OAAO,KAAK,QAAQ,KAAK,EAAE,MAAM,IAAI,EAAE,SAAS;EACzE,GAAG;EACH,cAAc,MAAM;EACpB,OAAO,MAAM,WAAW,MAAM,MAAM,OAAO,QAAQ;EACpD,CAAC;;AAGV,eAAe,WACb,MACA,OACA,SACmC;AACnC,SAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,OAAO,cAAc;AAC5C,KAAI,CAAC,MAAM,OAAQ,QAAO;AAI1B,QAAO,MAAM,QAAQ,IACnB,MAAM,IACJ,OAAO,UAA2E;EAChF,UAAU,KAAK;EACf,SAAS,KAAK,UAAU,MAAM,cAAc,MAAM,KAAK,SAAS,QAAQ,GAAG;EAC3E,WAAW,KAAK,YAAY,MAAM,cAAc,MAAM,KAAK,WAAW,QAAQ,GAAG;EACjF,SAAS,KAAK,UAAU,MAAM,cAAc,MAAM,KAAK,SAAS,QAAQ,GAAG;EAC3E,OAAO,KAAK,QAAQ,MAAM,cAAc,MAAM,KAAK,OAAO,QAAQ,GAAG;EACrE,cAAc,KAAK,eACf,MAAM,cAAc,MAAM,KAAK,cAAc,QAAQ,GACrD;EACJ,YAAY,KAAK,aACb,MAAM,cAAc,MAAM,KAAK,YAAY,QAAQ,GACnD;EACJ,WAAW,KAAK,YAAY,MAAM,cAAc,MAAM,KAAK,WAAW,QAAQ,GAAG;EAClF,EACF,CACF;;AAGH,eAAe,WACb,MACA,QACA,SACkB;CAClB,MAAM,eAAwB,EAAE;AAChC,MAAK,MAAM,SAAS,OAClB,cAAa,KAAK,MAAM,cAAc,MAAM,OAAO,QAAQ,CAAC;AAE9D,QAAO;;AAGT,eAAsB,WACpB,MACA,OACA,SACA,cACgB;AAChB,KAAI,QAAQ,MAAM,CAAE,QAAO;CAE3B,MAAM,SACJ,YAAY,SAAS,SAAS,UAAU,SACpC,MAAM,WACJ,QAAQ,UACR,OAAO,MAAM,WAAW,WAAW,MAAM,OAAO,WAAW,QAC3D,OAAO,MAAM,WAAW,WAAW,MAAM,SAAS,EAAE,CACrD,GACD;CAEN,IAAI;AACJ,KAAI,MAAM,QAAQ,UAAU,CAAC,MAAM,OAAO,MAAM,QAAQ,SAAS,SAAS,KAAK,UAC7E,OAAM,QAAQ,IAAI;UACT,MAAM,QAAQ,KACvB,OAAM,IAAI,KAAK;UACN,MAAM,KAAK;AACpB,QAAM,IAAI,IAAI,EAAE,CAAC;EAEjB,MAAM,wBAAwB,OAC5B,YAC8C;AAC9C,UAAO,OAAO,YACZ,MAAM,QAAQ,IACZ,OAAO,QAAQ,QAAQ,CAAC,IAAyC,OAAO,CAAC,KAAK,WAAW;IACvF,MAAM,CAAC,QAAQ,KAAK,UAAU,MAAM,QAAQ,IAC1C;KAAC,MAAM;KAAQ,MAAM;KAAK,MAAM;KAAO,CAAC,IAAI,OAAO,SAAS;AAC1D,SAAI,CAAC,MAAM,MAAO,QAAO;KACzB,MAAMC,UAAQ,MAAM,cAAc,MAAM,KAAK,OAAO,SAAS,EAAE,KAAK,CAAC;AACrE,YAAO,EACL,SAAS,OAAY,cACnBC,UAAQ,QAAQ,OAAOD,SAAO,OAAO;MACnC,GAAGC;MACH,WAAW;MACZ,CAAC,EACL;MACD,CACH;AAED,WAAO,CAAC,KAAK;KAAE,GAAG;KAAO;KAAQ;KAAK;KAAQ,CAAC;KAC/C,CACH,CACF;;EAGH,MAAM,UAAsB;GAC1B,QAAQ,EACN,SAAS,MAAM,sBAAsB,MAAM,IAAI,SAAS,QAAQ,WAAW,EAAE,CAAC,EAC/E;GACD,MAAM,EACJ,SAAS,MAAM,sBAAsB,MAAM,IAAI,SAAS,MAAM,WAAW,EAAE,CAAC,EAC7E;GACF;AAED,MAAI,QAAQ,UAAU;AAEtB,OAAK,MAAM,KAAK,MAAM,IAAI,WAAW,EAAE,EAAE;GACvC,MAAM,aAAa,OAAO,MAAM,WAAW,IAAI,EAAE;GAEjD,MAAM,MAAM,SAAS,KAAK,kBAAkB,MACzC,UAAQC,MAAI,WAAW,cAAcA,MAAI,OAAO,SAAS,WAAW,CACtE;AACD,OAAI,CAAC,IAAK,OAAM,IAAI,MAAM,yBAAyB,OAAO,MAAM,WAAW,IAAI,EAAE,SAAS;GAE1F,MAAM,SAAS,MAAM,IAAI,KAAK;IAC5B,UAAU;IACV,QAAQ,OAAO,MAAM,WAAW,EAAE,GAAG,EAAE;IACxC,CAAC;AAEF,OAAI,MAAM,OAAO;;;CAIrB,MAAM,SACJ,YAAY,SAAS,MAAM,SACvB,MAAM,WAAW,MAAM,MAAM,QAAQ;EACnC,GAAG;EACH,KAAK;GAAE,GAAG,SAAS;GAAK,WAAY,MAAM,YAAY,OAAQ,SAAS,KAAK;GAAW;EACxF,CAAC,GACF,EAAE;CAER,MAAM,QACJ,MAAM,SAAS,OAAO,SAAS,UAAU,aACrC,MAAM,QAAQ,MAAM;EAAE,GAAG,QAAQ,OAAO;EAAO,GAAG,OAAO,MAAM,QAAQ,MAAM,MAAM,EAAE,CAAC;EAAE,CAAC,GACzF;CACN,MAAM,aACJ,MAAM,cAAc,OAAO,SAAS,eAAe,aAC/C,MAAM,QAAQ,WAAW;EACvB,GAAG,QAAQ,OAAO;EAClB,GAAG,OAAO,MAAM,aAAa,MAAM,MAAM,EAAE,CAAC;EAC7C,CAAC,GACF;CAEN,MAAM,cAAsC;EAC1C,GAAG;EACH,GAAG;EACH;EACA;EACA;EACA,OAAO,CACL,GAAK,MAAM,WAAW,MAAM,MAAM,OAAO,QAAQ,IAAK,EAAE,EACxD,GAAG,CAAC,cAAc,MAAM,CAAC,MAAM,CAAC,OAAO,cAAc,CACtD;EACD,QAAQ,CAAC,GAAI,cAAc,UAAU,EAAE,EAAG,GAAG,OAAO;EACpD,KAAK,OAAO,cAAc;EAC3B;CAED,IAAI,aAAa,cAAc,MAAM;AAErC,KAAI,CAAC,YAAY;AACf,MAAI,CAAC,SAAS,QACZ,OAAM,IAAI,MACR,2DAA2D,MAAM,KAAK,QAAQ,OAC/E;EACH,MAAM,MAAM,MAAM,QAAQ,QAAQ,MAAM,MAAM,EAAE,QAAQ,MAAM,CAAC;AAC/D,MAAI,OAAO,KAAK,SAAS,WAAW,gBAAgB,WAClD,OAAM,IAAI,MAAM,yBAAyB,MAAM,KAAK,wCAAwC;AAG9F,eAAa,IAAI;;AAGnB,KAAI,CAAC,WACH,OAAM,IAAI,MAAM,2BAA2B,MAAM,KAAK,QAAQ,OAAO;AAGvE,QAAO,MAAM,WAAW,KAAK;EAC3B,UAAU;EACV,QAAQ;EACR,SAAS;GAAE,GAAG;GAAS;GAAe;EACvC,CAAC;;AAGJ,eAAe,WACb,UACA,UACA,SACA;CACA,MAAM,IAAI,CAAC,WACP,SAAS,KACT,SAAS,MAAM,MAAM,EAAE,KAAK,aAAa,CAAC,SAAS,SAAS,aAAa,CAAC,CAAC;AAC/E,KAAI,CAAC,EAAG,OAAM,IAAI,MAAM,uBAAuB,WAAW;AAE1D,QAAO,IAAI,EAAE,QAAQ;;AAavB,MAAM,iBAAoCC,IAAE,MAAM,CAChDA,IAAE,QAAQ,EACVA,IAAE,OAAO;CACP,KAAK,YAAYA,IAAE,QAAQ,CAAC;CAC5B,MAAM,YAAYA,IAAE,QAAQ,CAAC;CAC7B,OAAO,YAAYA,IAAE,MAAMA,IAAE,QAAQ,CAAC,CAAC;CACvC,aAAa,YAAYA,IAAE,QAAQ,CAAC;CACpC,QAAQ,YAAYA,IAAE,MAAMA,IAAE,WAAW,eAAe,CAAC,CAAC;CAC3D,CAAC,CACH,CAAC;AAEF,MAAM,kBAAkB,eACtBA,IAAE,OAAO;CACP,MAAM,YAAYA,IAAE,QAAQ,CAAC;CAC7B,aAAa,YAAYA,IAAE,QAAQ,CAAC;CACpC,OAAO,YAAY,gBAAgB;CACnC,YAAY,YAAY,iBAAiB;CACzC,QAAQ,YAAYA,IAAE,MAAMA,IAAE,QAAQ,CAAC,CAAC;CACxC,QAAQ,YAAYA,IAAE,MAAMA,IAAE,QAAQ,CAAC,CAAC;CACxC,WAAW,YACTA,IAAE,OAAO,EACP,QAAQ,YAAYA,IAAE,MAAMA,IAAE,QAAQ,CAAC,CAAC,EACzC,CAAC,CACH;CACD,KAAK,YACHA,IAAE,OAAO;EACP,MAAM,YAAYA,IAAE,QAAQ,CAAC;EAC7B,MAAM,YAAYA,IAAE,QAAQ,CAAC;EAC7B,QAAQ,YAAYA,IAAE,MAAM,eAAe,CAAC;EAC7C,CAAC,CACH;CACF,CAAC,CACH;AAED,eAAsB,cAAc,MAGjC;CACD,MAAM,OAAO,MAAM,cAAc,KAAK;CAEtC,MAAM,MAAM,MAAM,iBACV,OAAO,GAAG,SAAS,MAAM,OAAO,GACrC,0BAAU,IAAI,MAAM,kCAAkC,KAAK,IAAI,MAAM,UAAU,CACjF;CAED,MAAM,OAAO,iBACL,MAAM,IAAI,GACf,0BAAU,IAAI,MAAM,mCAAmC,KAAK,IAAI,MAAM,UAAU,CAClF;AAQD,QAAO;EAAE,OANK,iBAEV,gBAAgB,MAAM;GAAE,GAAG;GAAM,OAAO,KAAK,SAAS,KAAK,aAAa,KAAK;GAAY,CAAC,GAC3F,0BAAU,IAAI,MAAM,sCAAsC,KAAK,IAAI,MAAM,UAAU,CACrF;EAEe,SAAS,OAAO,KAAK,QAAQ,KAAK;EAAE;;AAGtD,eAAsB,cAAc,MAA+B;CACjE,MAAM,gBAAgB,gBAAgB,SAAS,OAAO,KAAK,SAAS,KAAK,CAAC,GACtE,CAAC,KAAK,GACN,gBAAgB,KAAK,SAAS,OAAO,KAAK,KAAK,MAAM,KAAK,CAAC;AAE/D,MAAK,MAAM,QAAQ,cACjB,KAAI;AAGF,OAFa,MAAM,OAAO,GAAG,KAAK,KAAK,EAE9B,QAAQ,CAAE,QAAO;SACpB;AAGV,OAAM,IAAI,MACR,2BAA2B,KAAK,2EACjC"}

package/dist/loader/schema.cjs

@@ -0,0 +1,135 @@
+const require_rolldown_runtime = require('../_virtual/rolldown_runtime.cjs');
+const require_utils_type_utils = require('../utils/type-utils.cjs');
+const require_types = require('../agents/types.cjs');
+require('../agents/agent.cjs');
+const require_chat_model = require('../agents/chat-model.cjs');
+const require_template = require('../prompt/template.cjs');
+const require_prompt_builder = require('../prompt/prompt-builder.cjs');
+const require_utils_camelize = require('../utils/camelize.cjs');
+let _aigne_utils_nodejs = require("@aigne/utils/nodejs");
+let yaml = require("yaml");
+let zod = require("zod");
+
+//#region src/loader/schema.ts
+const inputOutputSchema = ({ path }) => {
+	const includeExternalSchema = async (schema) => {
+		if (schema?.type === "object" && schema.properties) return {
+			...schema,
+			properties: Object.fromEntries(await Promise.all(Object.entries(schema.properties).map(async ([key, value]) => [key, await includeExternalSchema(value)])))
+		};
+		if (schema?.type === "array" && schema.items) return {
+			...schema,
+			items: await includeExternalSchema(schema.items)
+		};
+		if (typeof schema === "string") {
+			const raw = await _aigne_utils_nodejs.nodejs.fs.readFile(_aigne_utils_nodejs.nodejs.path.join(_aigne_utils_nodejs.nodejs.path.dirname(path), schema), "utf8");
+			return nestedJsonSchema.parseAsync((0, yaml.parse)(raw));
+		}
+		return schema;
+	};
+	const nestedJsonSchema = zod.z.object({ type: zod.z.string() }).passthrough().transform((v) => includeExternalSchema(v));
+	const jsonSchemaSchema = zod.z.object({
+		type: zod.z.literal("object"),
+		properties: zod.z.record(zod.z.any()),
+		required: zod.z.array(zod.z.string()).optional(),
+		additionalProperties: zod.z.boolean().optional()
+	}).transform((v) => includeExternalSchema(v));
+	return zod.z.union([zod.z.string().transform((v) => _aigne_utils_nodejs.nodejs.fs.readFile(_aigne_utils_nodejs.nodejs.path.join(_aigne_utils_nodejs.nodejs.path.dirname(path), v), "utf8").then((raw) => jsonSchemaSchema.parseAsync((0, yaml.parse)(raw)))), jsonSchemaSchema]);
+};
+const defaultInputSchema = zod.z.record(zod.z.string(), require_types.getterSchema(zod.z.unknown()));
+const chatModelObjectSchema = camelizeSchema(zod.z.object({
+	model: optionalize(require_types.getterSchema(zod.z.string())),
+	temperature: optionalize(require_types.getterSchema(zod.z.number().min(0).max(2))),
+	topP: optionalize(require_types.getterSchema(zod.z.number().min(0))),
+	frequencyPenalty: optionalize(require_types.getterSchema(zod.z.number().min(-2).max(2))),
+	presencePenalty: optionalize(require_types.getterSchema(zod.z.number().min(-2).max(2))),
+	thinkingEffort: optionalize(require_types.getterSchema(zod.z.union([
+		zod.z.number().int(),
+		zod.z.literal("high"),
+		zod.z.literal("medium"),
+		zod.z.literal("low"),
+		zod.z.literal("minimal")
+	])))
+}).passthrough());
+const chatModelSchema = zod.z.preprocess((v) => {
+	if (!require_utils_type_utils.isRecord(v)) return v;
+	return {
+		...v,
+		model: v.model || (v.provider && v.name ? `${v.provider || ""}:${v.name || ""}` : void 0) || void 0
+	};
+}, zod.z.union([zod.z.string(), chatModelObjectSchema])).transform((v) => typeof v === "string" ? { model: v } : v);
+const imageModelObjectSchema = camelizeSchema(zod.z.object({ model: optionalize(require_types.getterSchema(zod.z.string())) }).passthrough());
+const imageModelSchema = zod.z.union([zod.z.string(), imageModelObjectSchema]).transform((v) => typeof v === "string" ? { model: v } : v);
+function optionalize(schema) {
+	return schema.nullish().transform((v) => v ?? void 0);
+}
+function camelizeSchema(schema, { shallow = true } = {}) {
+	return zod.z.preprocess((v) => require_utils_type_utils.isRecord(v) ? require_utils_camelize.camelize(v, shallow) : v, schema);
+}
+function preprocessSchema(fn, schema) {
+	return zod.z.preprocess(fn, schema);
+}
+const instructionItemSchema = camelizeSchema(zod.z.union([zod.z.object({
+	role: require_chat_model.roleSchema.default("system"),
+	url: zod.z.string(),
+	cacheControl: optionalize(zod.z.object({
+		type: zod.z.literal("ephemeral"),
+		ttl: optionalize(zod.z.union([zod.z.literal("5m"), zod.z.literal("1h")]))
+	}))
+}), zod.z.object({
+	role: require_chat_model.roleSchema.default("system"),
+	content: zod.z.string(),
+	cacheControl: optionalize(zod.z.object({
+		type: zod.z.literal("ephemeral"),
+		ttl: optionalize(zod.z.union([zod.z.literal("5m"), zod.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 = _aigne_utils_nodejs.nodejs.path.isAbsolute(v.url) ? v.url : _aigne_utils_nodejs.nodejs.path.join(_aigne_utils_nodejs.nodejs.path.dirname(filepath), v.url);
+		return _aigne_utils_nodejs.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.z.union([
+	zod.z.string(),
+	instructionItemSchema,
+	zod.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)];
+});
+function instructionsToPromptBuilder(instructions) {
+	return new require_prompt_builder.PromptBuilder({ instructions: typeof instructions === "string" ? instructions : require_template.ChatMessagesTemplate.from(require_template.parseChatMessages(instructions.map((i) => ({
+		...i,
+		options: { workingDir: _aigne_utils_nodejs.nodejs.path.dirname(i.path) }
+	})))) });
+}
+
+//#endregion
+exports.camelizeSchema = camelizeSchema;
+exports.chatModelSchema = chatModelSchema;
+exports.defaultInputSchema = defaultInputSchema;
+exports.getInstructionsSchema = getInstructionsSchema;
+exports.imageModelSchema = imageModelSchema;
+exports.inputOutputSchema = inputOutputSchema;
+exports.instructionsToPromptBuilder = instructionsToPromptBuilder;
+exports.optionalize = optionalize;
+exports.preprocessSchema = preprocessSchema;

package/dist/loader/schema.d.cts

@@ -0,0 +1,177 @@
+import { Role } from "../agents/chat-model.cjs";
+import { PromptBuilder } from "../prompt/prompt-builder.cjs";
+import { ZodType, z } from "zod";
+
+//#region src/loader/schema.d.ts
+declare const inputOutputSchema: ({
+  path
+}: {
+  path: string;
+}) => z.ZodUnion<[ZodType<any, z.ZodTypeDef, any>, z.ZodEffects<z.ZodObject<{
+  type: z.ZodLiteral<"object">;
+  properties: z.ZodRecord<z.ZodString, z.ZodAny>;
+  required: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
+  additionalProperties: z.ZodOptional<z.ZodBoolean>;
+}, "strip", z.ZodTypeAny, {
+  type: "object";
+  properties: Record<string, any>;
+  required?: string[] | undefined;
+  additionalProperties?: boolean | undefined;
+}, {
+  type: "object";
+  properties: Record<string, any>;
+  required?: string[] | undefined;
+  additionalProperties?: boolean | undefined;
+}>, any, {
+  type: "object";
+  properties: Record<string, any>;
+  required?: string[] | undefined;
+  additionalProperties?: boolean | undefined;
+}>]>;
+declare const defaultInputSchema: z.ZodRecord<z.ZodString, z.ZodUnion<[z.ZodUnknown, z.ZodObject<{
+  $get: z.ZodString;
+}, "strip", z.ZodTypeAny, {
+  $get: string;
+}, {
+  $get: string;
+}>]>>;
+declare const chatModelObjectSchema: z.ZodObject<{
+  model: ZodType<string | {
+    $get: string;
+  } | undefined, z.ZodTypeDef, string | {
+    $get: string;
+  } | undefined>;
+  temperature: ZodType<number | {
+    $get: string;
+  } | undefined, z.ZodTypeDef, number | {
+    $get: string;
+  } | undefined>;
+  topP: ZodType<number | {
+    $get: string;
+  } | undefined, z.ZodTypeDef, number | {
+    $get: string;
+  } | undefined>;
+  frequencyPenalty: ZodType<number | {
+    $get: string;
+  } | undefined, z.ZodTypeDef, number | {
+    $get: string;
+  } | undefined>;
+  presencePenalty: ZodType<number | {
+    $get: string;
+  } | undefined, z.ZodTypeDef, number | {
+    $get: string;
+  } | undefined>;
+  thinkingEffort: ZodType<number | "high" | "medium" | "low" | "minimal" | {
+    $get: string;
+  } | undefined, z.ZodTypeDef, number | "high" | "medium" | "low" | "minimal" | {
+    $get: string;
+  } | undefined>;
+}, "passthrough", z.ZodTypeAny, z.objectOutputType<{
+  model: ZodType<string | {
+    $get: string;
+  } | undefined, z.ZodTypeDef, string | {
+    $get: string;
+  } | undefined>;
+  temperature: ZodType<number | {
+    $get: string;
+  } | undefined, z.ZodTypeDef, number | {
+    $get: string;
+  } | undefined>;
+  topP: ZodType<number | {
+    $get: string;
+  } | undefined, z.ZodTypeDef, number | {
+    $get: string;
+  } | undefined>;
+  frequencyPenalty: ZodType<number | {
+    $get: string;
+  } | undefined, z.ZodTypeDef, number | {
+    $get: string;
+  } | undefined>;
+  presencePenalty: ZodType<number | {
+    $get: string;
+  } | undefined, z.ZodTypeDef, number | {
+    $get: string;
+  } | undefined>;
+  thinkingEffort: ZodType<number | "high" | "medium" | "low" | "minimal" | {
+    $get: string;
+  } | undefined, z.ZodTypeDef, number | "high" | "medium" | "low" | "minimal" | {
+    $get: string;
+  } | undefined>;
+}, z.ZodTypeAny, "passthrough">, z.objectInputType<{
+  model: ZodType<string | {
+    $get: string;
+  } | undefined, z.ZodTypeDef, string | {
+    $get: string;
+  } | undefined>;
+  temperature: ZodType<number | {
+    $get: string;
+  } | undefined, z.ZodTypeDef, number | {
+    $get: string;
+  } | undefined>;
+  topP: ZodType<number | {
+    $get: string;
+  } | undefined, z.ZodTypeDef, number | {
+    $get: string;
+  } | undefined>;
+  frequencyPenalty: ZodType<number | {
+    $get: string;
+  } | undefined, z.ZodTypeDef, number | {
+    $get: string;
+  } | undefined>;
+  presencePenalty: ZodType<number | {
+    $get: string;
+  } | undefined, z.ZodTypeDef, number | {
+    $get: string;
+  } | undefined>;
+  thinkingEffort: ZodType<number | "high" | "medium" | "low" | "minimal" | {
+    $get: string;
+  } | undefined, z.ZodTypeDef, number | "high" | "medium" | "low" | "minimal" | {
+    $get: string;
+  } | undefined>;
+}, z.ZodTypeAny, "passthrough">>;
+declare const chatModelSchema: typeof chatModelObjectSchema;
+declare const imageModelObjectSchema: z.ZodObject<{
+  model: ZodType<string | {
+    $get: string;
+  } | undefined, z.ZodTypeDef, string | {
+    $get: string;
+  } | undefined>;
+}, "passthrough", z.ZodTypeAny, z.objectOutputType<{
+  model: ZodType<string | {
+    $get: string;
+  } | undefined, z.ZodTypeDef, string | {
+    $get: string;
+  } | undefined>;
+}, z.ZodTypeAny, "passthrough">, z.objectInputType<{
+  model: ZodType<string | {
+    $get: string;
+  } | undefined, z.ZodTypeDef, string | {
+    $get: string;
+  } | undefined>;
+}, z.ZodTypeAny, "passthrough">>;
+declare const imageModelSchema: typeof imageModelObjectSchema;
+declare function optionalize<T>(schema: ZodType<T>): ZodType<T | undefined>;
+declare function camelizeSchema<T extends ZodType>(schema: T, {
+  shallow
+}?: {
+  shallow?: boolean;
+}): T;
+declare function preprocessSchema<T extends ZodType>(fn: (data: unknown) => unknown, schema: T): T;
+type Instructions = {
+  role: Exclude<Role, "tool">;
+  content: string;
+  path: string;
+  cacheControl?: {
+    type: "ephemeral";
+    ttl?: "5m" | "1h";
+  };
+}[];
+declare const getInstructionsSchema: ({
+  filepath
+}: {
+  filepath: string;
+}) => ZodType<Instructions>;
+declare function instructionsToPromptBuilder(instructions: Instructions | string): PromptBuilder;
+//#endregion
+export { Instructions, camelizeSchema, chatModelSchema, defaultInputSchema, getInstructionsSchema, imageModelSchema, inputOutputSchema, instructionsToPromptBuilder, optionalize, preprocessSchema };
+//# sourceMappingURL=schema.d.cts.map

package/dist/loader/schema.d.cts.map

@@ -0,0 +1 @@
+{"version":3,"file":"schema.d.cts","names":[],"sources":["../../src/loader/schema.ts"],"mappings":";;;;;cAUa,iBAAA;EAAA;AAAA;EAAA,IAAA;AAAA,MAA+C,CAAA,CAAA,QAAA,EAAA,OAAA,MAAA,CAAA,CAAA,UAAA,QAAA,CAAA,CAAA,UAAA,CAAA,CAAA,CAAA,SAAA;EAAA,IAAA;;;;;;;;;;;;;;;;;;;;cA2D/C,kBAAA,EAAkB,CAAA,CAAA,SAAA,CAAA,CAAA,CAAA,SAAA,EAAA,CAAA,CAAA,QAAA,EAAA,CAAA,CAAA,UAAA,EAAA,CAAA,CAAA,SAAA;EAAA,IAAA;;;;;;cAEzB,qBAAA,EAAqB,CAAA,CAAA,SAAA;EAAA,KAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;cAuBd,eAAA,SAgBY,qBAAA;AAAA,cAEnB,sBAAA,EAAsB,CAAA,CAAA,SAAA;EAAA,KAAA;;;;;;;;;;;;;;;;;;cAQf,gBAAA,SAIY,sBAAA;AAAA,iBAET,WAAA,GAAA,CAAA,MAAA,EAAuB,OAAA,CAAQ,CAAA,IAAK,OAAA,CAAQ,CAAA;AAAA,iBAI5C,cAAA,WAAyB,OAAA,CAAA,CAAA,MAAA,EAC/B,CAAA;EAAA;AAAA;EAAA,OAAA;AAAA,IAEP,CAAA;AAAA,iBAIa,gBAAA,WAA2B,OAAA,CAAA,CAAA,EAAA,GAAA,IAAA,uBAAA,MAAA,EAAiD,CAAA,GAAI,CAAA;AAAA,KA6BpF,YAAA;EAAA,IAAA,EACJ,OAAA,CAAQ,IAAA;EAAA,OAAA;EAAA,IAAA;EAAA,YAAA;IAAA,IAAA;IAAA,GAAA;EAAA;AAAA;AAAA,cAmCH,qBAAA;EAAA;AAAA;EAAA,QAAA;AAAA,MAWQ,OAAA,CAAQ,YAAA;AAAA,iBAEb,2BAAA,CAAA,YAAA,EAA0C,YAAA,YAAqB,aAAA"}