@aigne/core 1.33.0 → 1.33.2

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

@@ -2,6 +2,7 @@ import type { Camelize } from "camelize-ts";
 import { z } from "zod";
 import { Agent } from "../agents/agent.js";
 import type { ChatModel, ChatModelOptions } from "../agents/chat-model.js";
+import type { AIGNEOptions } from "../aigne/aigne.js";
 import type { MemoryAgent, MemoryAgentOptions } from "../memory/memory.js";
 interface LoadableModelClass {
     new (parameters: {
@@ -23,103 +24,72 @@ export interface LoadOptions {
     }[];
     path: string;
 }
-export declare function load(options: LoadOptions): Promise<{
-    model: ChatModel | undefined;
-    agents: Agent<any, any>[];
-    skills: Agent<any, any>[];
-    name?: string | null | undefined;
-    description?: string | null | undefined;
-    chat_model?: {
-        name?: string | null | undefined;
-        temperature?: number | null | undefined;
-        provider?: string | null | undefined;
-        top_p?: number | null | undefined;
-        frequency_penalty?: number | null | undefined;
-        presence_penalty?: number | null | undefined;
-    } | null | undefined;
-}>;
+export declare function load(options: LoadOptions): Promise<AIGNEOptions>;
 export declare function loadAgent(path: string, options?: LoadOptions): Promise<Agent>;
-export declare function loadModel(models: LoadableModel[], model?: Camelize<z.infer<typeof aigneFileSchema>["chat_model"]>, modelOptions?: ChatModelOptions): Promise<ChatModel | undefined>;
+export declare function loadModel(models: LoadableModel[], model?: Camelize<z.infer<typeof aigneFileSchema>["model"]>, modelOptions?: ChatModelOptions): Promise<ChatModel | undefined>;
 declare const aigneFileSchema: z.ZodObject<{
-    name: z.ZodOptional<z.ZodNullable<z.ZodString>>;
-    description: z.ZodOptional<z.ZodNullable<z.ZodString>>;
-    chat_model: z.ZodEffects<z.ZodOptional<z.ZodNullable<z.ZodUnion<[z.ZodString, z.ZodObject<{
-        provider: z.ZodOptional<z.ZodNullable<z.ZodString>>;
-        name: z.ZodOptional<z.ZodNullable<z.ZodString>>;
-        temperature: z.ZodOptional<z.ZodNullable<z.ZodNumber>>;
-        top_p: z.ZodOptional<z.ZodNullable<z.ZodNumber>>;
-        frequency_penalty: z.ZodOptional<z.ZodNullable<z.ZodNumber>>;
-        presence_penalty: z.ZodOptional<z.ZodNullable<z.ZodNumber>>;
-    }, "strip", z.ZodTypeAny, {
+    name: z.ZodType<string | undefined, z.ZodTypeDef, string | undefined>;
+    description: z.ZodType<string | undefined, z.ZodTypeDef, string | undefined>;
+    model: z.ZodEffects<z.ZodType<string | {
         name?: string | null | undefined;
         temperature?: number | null | undefined;
+        topP?: number | null | undefined;
+        frequencyPenalty?: number | null | undefined;
+        presencePenalty?: number | null | undefined;
         provider?: string | null | undefined;
-        top_p?: number | null | undefined;
-        frequency_penalty?: number | null | undefined;
-        presence_penalty?: number | null | undefined;
-    }, {
+    } | undefined, z.ZodTypeDef, string | {
         name?: string | null | undefined;
         temperature?: number | null | undefined;
+        topP?: number | null | undefined;
+        frequencyPenalty?: number | null | undefined;
+        presencePenalty?: number | null | undefined;
         provider?: string | null | undefined;
-        top_p?: number | null | undefined;
-        frequency_penalty?: number | null | undefined;
-        presence_penalty?: number | null | undefined;
-    }>]>>>, {
+    } | undefined>, {
         name?: string | null | undefined;
         temperature?: number | null | undefined;
+        topP?: number | null | undefined;
+        frequencyPenalty?: number | null | undefined;
+        presencePenalty?: number | null | undefined;
         provider?: string | null | undefined;
-        top_p?: number | null | undefined;
-        frequency_penalty?: number | null | undefined;
-        presence_penalty?: number | null | undefined;
-    } | null | undefined, string | {
+    } | undefined, string | {
         name?: string | null | undefined;
         temperature?: number | null | undefined;
+        topP?: number | null | undefined;
+        frequencyPenalty?: number | null | undefined;
+        presencePenalty?: number | null | undefined;
         provider?: string | null | undefined;
-        top_p?: number | null | undefined;
-        frequency_penalty?: number | null | undefined;
-        presence_penalty?: number | null | undefined;
-    } | null | undefined>;
-    agents: z.ZodOptional<z.ZodNullable<z.ZodArray<z.ZodString, "many">>>;
-    skills: z.ZodOptional<z.ZodNullable<z.ZodArray<z.ZodString, "many">>>;
+    } | undefined>;
+    agents: z.ZodType<string[] | undefined, z.ZodTypeDef, string[] | undefined>;
+    skills: z.ZodType<string[] | undefined, z.ZodTypeDef, string[] | undefined>;
 }, "strip", z.ZodTypeAny, {
-    name?: string | null | undefined;
-    description?: string | null | undefined;
-    skills?: string[] | null | undefined;
-    chat_model?: {
+    name?: string | undefined;
+    description?: string | undefined;
+    skills?: string[] | undefined;
+    model?: {
         name?: string | null | undefined;
         temperature?: number | null | undefined;
+        topP?: number | null | undefined;
+        frequencyPenalty?: number | null | undefined;
+        presencePenalty?: number | null | undefined;
         provider?: string | null | undefined;
-        top_p?: number | null | undefined;
-        frequency_penalty?: number | null | undefined;
-        presence_penalty?: number | null | undefined;
-    } | null | undefined;
-    agents?: string[] | null | undefined;
+    } | undefined;
+    agents?: string[] | undefined;
 }, {
-    name?: string | null | undefined;
-    description?: string | null | undefined;
-    skills?: string[] | null | undefined;
-    chat_model?: string | {
+    name?: string | undefined;
+    description?: string | undefined;
+    skills?: string[] | undefined;
+    model?: string | {
         name?: string | null | undefined;
         temperature?: number | null | undefined;
+        topP?: number | null | undefined;
+        frequencyPenalty?: number | null | undefined;
+        presencePenalty?: number | null | undefined;
         provider?: string | null | undefined;
-        top_p?: number | null | undefined;
-        frequency_penalty?: number | null | undefined;
-        presence_penalty?: number | null | undefined;
-    } | null | undefined;
-    agents?: string[] | null | undefined;
+    } | undefined;
+    agents?: string[] | undefined;
 }>;
 export declare function loadAIGNEFile(path: string): Promise<{
-    name?: string | null | undefined;
-    description?: string | null | undefined;
-    skills?: string[] | null | undefined;
-    chat_model?: {
-        name?: string | null | undefined;
-        temperature?: number | null | undefined;
-        provider?: string | null | undefined;
-        top_p?: number | null | undefined;
-        frequency_penalty?: number | null | undefined;
-        presence_penalty?: number | null | undefined;
-    } | null | undefined;
-    agents?: string[] | null | undefined;
+    aigne: z.infer<typeof aigneFileSchema>;
+    rootDir: string;
 }>;
 export {};

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

@@ -151,7 +151,7 @@ const chatModelInputMessageSchema = z.object({
         type: z.literal("function"),
         function: z.object({
             name: z.string(),
-            arguments: z.record(z.unknown()),
+            arguments: z.record(z.string(), z.unknown()),
         }),
     }))
         .optional(),
@@ -165,7 +165,7 @@ const chatModelInputResponseFormatSchema = z.discriminatedUnion("type", [
         jsonSchema: z.object({
             name: z.string(),
             description: z.string().optional(),
-            schema: z.record(z.unknown()),
+            schema: z.record(z.string(), z.unknown()),
             strict: z.boolean().optional(),
         }),
     }),
@@ -175,7 +175,7 @@ const chatModelInputToolSchema = z.object({
     function: z.object({
         name: z.string(),
         description: z.string().optional(),
-        parameters: z.record(z.unknown()),
+        parameters: z.record(z.string(), z.unknown()),
     }),
 });
 const chatModelInputToolChoiceSchema = z.union([
@@ -204,7 +204,7 @@ const chatModelOutputToolCallSchema = z.object({
     type: z.literal("function"),
     function: z.object({
         name: z.string(),
-        arguments: z.record(z.unknown()),
+        arguments: z.record(z.string(), z.unknown()),
     }),
 });
 const chatModelOutputUsageSchema = z.object({
@@ -213,7 +213,7 @@ const chatModelOutputUsageSchema = z.object({
 });
 const chatModelOutputSchema = z.object({
     text: z.string().optional(),
-    json: z.record(z.unknown()).optional(),
+    json: z.record(z.string(), z.unknown()).optional(),
     toolCalls: z.array(chatModelOutputToolCallSchema).optional(),
     usage: chatModelOutputUsageSchema.optional(),
     model: z.string().optional(),

package/lib/esm/agents/mcp-agent.js

@@ -302,6 +302,6 @@ const mcpAgentOptionsSchema = z.union([
     z.object({
         command: z.string(),
         args: z.array(z.string()).optional(),
-        env: z.record(z.string()).optional(),
+        env: z.record(z.string(), z.string()).optional(),
     }),
 ]);

package/lib/esm/agents/team-agent.js

@@ -111,6 +111,8 @@ export class TeamAgent extends Agent {
      * @returns A stream of message chunks that collectively form the response
      */
     process(input, options) {
+        if (!this.skills.length)
+            throw new Error("TeamAgent must have at least one skill defined.");
         if (this.iterateOn) {
             return this._processIterator(this.iterateOn, input, options);
         }

package/lib/esm/aigne/aigne.d.ts

@@ -10,6 +10,11 @@ import type { ContextLimits } from "./usage.js";
  * Options for the AIGNE class.
  */
 export interface AIGNEOptions {
+    /**
+     * Optional root directory for this AIGNE instance.
+     * This is used to resolve relative paths for agents and skills.
+     */
+    rootDir?: string;
     /**
      * The name of the AIGNE instance.
      */
@@ -67,6 +72,10 @@ export declare class AIGNE<U extends UserContext = UserContext> {
      * @param options - Configuration options for the AIGNE instance including name, description, model, and agents.
      */
     constructor(options?: AIGNEOptions);
+    /**
+     * Optional root directory for this AIGNE instance.
+     */
+    rootDir?: string;
     /**
      * Optional name identifier for this AIGNE instance.
      */

package/lib/esm/aigne/aigne.js

@@ -26,9 +26,10 @@ export class AIGNE {
      * @returns A fully initialized AIGNE instance with configured agents and skills.
      */
     static async load(path, options) {
-        const { model, agents, skills, ...aigne } = await load({ ...options, path });
+        const { model, agents = [], skills = [], ...aigne } = await load({ ...options, path });
         return new AIGNE({
             ...options,
+            rootDir: aigne.rootDir,
             model: options?.model || model,
             name: options?.name || aigne.name || undefined,
             description: options?.description || aigne.description || undefined,
@@ -44,6 +45,7 @@ export class AIGNE {
     constructor(options) {
         if (options)
             checkArguments("AIGNE", aigneOptionsSchema, options);
+        this.rootDir = options?.rootDir;
         this.name = options?.name;
         this.description = options?.description;
         this.model = options?.model;
@@ -59,6 +61,10 @@ export class AIGNE {
         this.observer?.serve();
         this.initProcessExitHandler();
     }
+    /**
+     * Optional root directory for this AIGNE instance.
+     */
+    rootDir;
     /**
      * Optional name identifier for this AIGNE instance.
      */

package/lib/esm/aigne/context.js

@@ -381,6 +381,6 @@ async function* withAbortSignal(signal, error, fn) {
 }
 const aigneContextInvokeArgsSchema = z.object({
     agent: z.union([z.custom(), z.custom()]),
-    message: z.union([z.record(z.unknown()), z.string()]).optional(),
+    message: z.union([z.record(z.string(), z.unknown()), z.string()]).optional(),
     options: z.object({ returnActiveAgent: z.boolean().optional() }).optional(),
 });

package/lib/esm/aigne/message-queue.js

@@ -87,18 +87,18 @@ function once(events, event, listener) {
 }
 const subscribeArgsSchema = z.object({
     topic: z.union([z.string(), z.array(z.string())]),
-    listener: z.function(z.tuple([z.any()]), z.any()).optional(),
+    listener: z.custom().optional(),
 });
 const unsubscribeArgsSchema = z.object({
     topic: z.union([z.string(), z.array(z.string())]),
-    listener: z.function(z.tuple([z.any()]), z.any()),
+    listener: z.custom(),
 });
 const publishArgsSchema = z.object({
     topic: z.union([z.string(), z.array(z.string())]),
     payload: z.object({
         role: z.union([z.literal("user"), z.literal("agent")]),
         source: z.string().optional(),
-        message: z.union([z.string(), z.record(z.unknown())]),
+        message: z.union([z.string(), z.record(z.string(), z.unknown())]),
         context: z.any(),
     }),
 });

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

@@ -62,11 +62,11 @@ export async function loadAgentFromYamlFile(path) {
         ]);
     });
     const raw = await tryOrThrow(() => nodejs.fs.readFile(path, "utf8"), (error) => new Error(`Failed to load agent definition from ${path}: ${error.message}`));
-    const json = await tryOrThrow(() => parse(raw), (error) => new Error(`Failed to parse agent definition from ${path}: ${error.message}`));
-    const agent = await tryOrThrow(async () => await agentSchema.parseAsync(camelize({
+    const json = tryOrThrow(() => camelize(parse(raw)), (error) => new Error(`Failed to parse agent definition from ${path}: ${error.message}`));
+    const agent = await tryOrThrow(async () => await agentSchema.parseAsync({
         ...json,
         type: json.type ?? "ai",
         skills: json.skills ?? json.tools,
-    })), (error) => new Error(`Failed to validate agent definition from ${path}: ${error.message}`));
+    }), (error) => new Error(`Failed to validate agent definition from ${path}: ${error.message}`));
     return agent;
 }

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

@@ -2,6 +2,7 @@ import type { Camelize } from "camelize-ts";
 import { z } from "zod";
 import { Agent } from "../agents/agent.js";
 import type { ChatModel, ChatModelOptions } from "../agents/chat-model.js";
+import type { AIGNEOptions } from "../aigne/aigne.js";
 import type { MemoryAgent, MemoryAgentOptions } from "../memory/memory.js";
 interface LoadableModelClass {
     new (parameters: {
@@ -23,103 +24,72 @@ export interface LoadOptions {
     }[];
     path: string;
 }
-export declare function load(options: LoadOptions): Promise<{
-    model: ChatModel | undefined;
-    agents: Agent<any, any>[];
-    skills: Agent<any, any>[];
-    name?: string | null | undefined;
-    description?: string | null | undefined;
-    chat_model?: {
-        name?: string | null | undefined;
-        temperature?: number | null | undefined;
-        provider?: string | null | undefined;
-        top_p?: number | null | undefined;
-        frequency_penalty?: number | null | undefined;
-        presence_penalty?: number | null | undefined;
-    } | null | undefined;
-}>;
+export declare function load(options: LoadOptions): Promise<AIGNEOptions>;
 export declare function loadAgent(path: string, options?: LoadOptions): Promise<Agent>;
-export declare function loadModel(models: LoadableModel[], model?: Camelize<z.infer<typeof aigneFileSchema>["chat_model"]>, modelOptions?: ChatModelOptions): Promise<ChatModel | undefined>;
+export declare function loadModel(models: LoadableModel[], model?: Camelize<z.infer<typeof aigneFileSchema>["model"]>, modelOptions?: ChatModelOptions): Promise<ChatModel | undefined>;
 declare const aigneFileSchema: z.ZodObject<{
-    name: z.ZodOptional<z.ZodNullable<z.ZodString>>;
-    description: z.ZodOptional<z.ZodNullable<z.ZodString>>;
-    chat_model: z.ZodEffects<z.ZodOptional<z.ZodNullable<z.ZodUnion<[z.ZodString, z.ZodObject<{
-        provider: z.ZodOptional<z.ZodNullable<z.ZodString>>;
-        name: z.ZodOptional<z.ZodNullable<z.ZodString>>;
-        temperature: z.ZodOptional<z.ZodNullable<z.ZodNumber>>;
-        top_p: z.ZodOptional<z.ZodNullable<z.ZodNumber>>;
-        frequency_penalty: z.ZodOptional<z.ZodNullable<z.ZodNumber>>;
-        presence_penalty: z.ZodOptional<z.ZodNullable<z.ZodNumber>>;
-    }, "strip", z.ZodTypeAny, {
+    name: z.ZodType<string | undefined, z.ZodTypeDef, string | undefined>;
+    description: z.ZodType<string | undefined, z.ZodTypeDef, string | undefined>;
+    model: z.ZodEffects<z.ZodType<string | {
         name?: string | null | undefined;
         temperature?: number | null | undefined;
+        topP?: number | null | undefined;
+        frequencyPenalty?: number | null | undefined;
+        presencePenalty?: number | null | undefined;
         provider?: string | null | undefined;
-        top_p?: number | null | undefined;
-        frequency_penalty?: number | null | undefined;
-        presence_penalty?: number | null | undefined;
-    }, {
+    } | undefined, z.ZodTypeDef, string | {
         name?: string | null | undefined;
         temperature?: number | null | undefined;
+        topP?: number | null | undefined;
+        frequencyPenalty?: number | null | undefined;
+        presencePenalty?: number | null | undefined;
         provider?: string | null | undefined;
-        top_p?: number | null | undefined;
-        frequency_penalty?: number | null | undefined;
-        presence_penalty?: number | null | undefined;
-    }>]>>>, {
+    } | undefined>, {
         name?: string | null | undefined;
         temperature?: number | null | undefined;
+        topP?: number | null | undefined;
+        frequencyPenalty?: number | null | undefined;
+        presencePenalty?: number | null | undefined;
         provider?: string | null | undefined;
-        top_p?: number | null | undefined;
-        frequency_penalty?: number | null | undefined;
-        presence_penalty?: number | null | undefined;
-    } | null | undefined, string | {
+    } | undefined, string | {
         name?: string | null | undefined;
         temperature?: number | null | undefined;
+        topP?: number | null | undefined;
+        frequencyPenalty?: number | null | undefined;
+        presencePenalty?: number | null | undefined;
         provider?: string | null | undefined;
-        top_p?: number | null | undefined;
-        frequency_penalty?: number | null | undefined;
-        presence_penalty?: number | null | undefined;
-    } | null | undefined>;
-    agents: z.ZodOptional<z.ZodNullable<z.ZodArray<z.ZodString, "many">>>;
-    skills: z.ZodOptional<z.ZodNullable<z.ZodArray<z.ZodString, "many">>>;
+    } | undefined>;
+    agents: z.ZodType<string[] | undefined, z.ZodTypeDef, string[] | undefined>;
+    skills: z.ZodType<string[] | undefined, z.ZodTypeDef, string[] | undefined>;
 }, "strip", z.ZodTypeAny, {
-    name?: string | null | undefined;
-    description?: string | null | undefined;
-    skills?: string[] | null | undefined;
-    chat_model?: {
+    name?: string | undefined;
+    description?: string | undefined;
+    skills?: string[] | undefined;
+    model?: {
         name?: string | null | undefined;
         temperature?: number | null | undefined;
+        topP?: number | null | undefined;
+        frequencyPenalty?: number | null | undefined;
+        presencePenalty?: number | null | undefined;
         provider?: string | null | undefined;
-        top_p?: number | null | undefined;
-        frequency_penalty?: number | null | undefined;
-        presence_penalty?: number | null | undefined;
-    } | null | undefined;
-    agents?: string[] | null | undefined;
+    } | undefined;
+    agents?: string[] | undefined;
 }, {
-    name?: string | null | undefined;
-    description?: string | null | undefined;
-    skills?: string[] | null | undefined;
-    chat_model?: string | {
+    name?: string | undefined;
+    description?: string | undefined;
+    skills?: string[] | undefined;
+    model?: string | {
         name?: string | null | undefined;
         temperature?: number | null | undefined;
+        topP?: number | null | undefined;
+        frequencyPenalty?: number | null | undefined;
+        presencePenalty?: number | null | undefined;
         provider?: string | null | undefined;
-        top_p?: number | null | undefined;
-        frequency_penalty?: number | null | undefined;
-        presence_penalty?: number | null | undefined;
-    } | null | undefined;
-    agents?: string[] | null | undefined;
+    } | undefined;
+    agents?: string[] | undefined;
 }>;
 export declare function loadAIGNEFile(path: string): Promise<{
-    name?: string | null | undefined;
-    description?: string | null | undefined;
-    skills?: string[] | null | undefined;
-    chat_model?: {
-        name?: string | null | undefined;
-        temperature?: number | null | undefined;
-        provider?: string | null | undefined;
-        top_p?: number | null | undefined;
-        frequency_penalty?: number | null | undefined;
-        presence_penalty?: number | null | undefined;
-    } | null | undefined;
-    agents?: string[] | null | undefined;
+    aigne: z.infer<typeof aigneFileSchema>;
+    rootDir: string;
 }>;
 export {};

package/lib/esm/loader/index.js

@@ -1,4 +1,5 @@
 import { nodejs } from "@aigne/platform-helpers/nodejs/index.js";
+import camelize from "camelize-ts";
 import { parse } from "yaml";
 import { z } from "zod";
 import { Agent, FunctionAgent } from "../agents/agent.js";
@@ -10,21 +11,21 @@ import { PromptBuilder } from "../prompt/prompt-builder.js";
 import { tryOrThrow } from "../utils/type-utils.js";
 import { loadAgentFromJsFile } from "./agent-js.js";
 import { loadAgentFromYamlFile } from "./agent-yaml.js";
+import { optionalize } from "./schema.js";
 const AIGNE_FILE_NAME = ["aigne.yaml", "aigne.yml"];
 export async function load(options) {
-    const aigneFilePath = await getAIGNEFilePath(options.path);
-    const rootDir = nodejs.path.dirname(aigneFilePath);
-    const aigne = await loadAIGNEFile(aigneFilePath);
+    const { aigne, rootDir } = await loadAIGNEFile(options.path);
     const agents = await Promise.all((aigne.agents ?? []).map((filename) => loadAgent(nodejs.path.join(rootDir, filename), options)));
     const skills = await Promise.all((aigne.skills ?? []).map((filename) => loadAgent(nodejs.path.join(rootDir, filename), options)));
     return {
         ...aigne,
+        rootDir,
         model: await loadModel(options.models.map((i) => typeof i === "function"
             ? {
                 name: i.name,
                 create: (options) => new i(options),
             }
-            : i), aigne.chat_model),
+            : i), aigne.model),
         agents,
         skills,
     };
@@ -121,39 +122,44 @@ export async function loadModel(models, model, modelOptions) {
     });
 }
 const aigneFileSchema = z.object({
-    name: z.string().nullish(),
-    description: z.string().nullish(),
-    chat_model: z
-        .union([
+    name: optionalize(z.string()),
+    description: optionalize(z.string()),
+    model: optionalize(z.union([
         z.string(),
         z.object({
             provider: z.string().nullish(),
             name: z.string().nullish(),
             temperature: z.number().min(0).max(2).nullish(),
-            top_p: z.number().min(0).nullish(),
-            frequency_penalty: z.number().min(-2).max(2).nullish(),
-            presence_penalty: z.number().min(-2).max(2).nullish(),
+            topP: z.number().min(0).nullish(),
+            frequencyPenalty: z.number().min(-2).max(2).nullish(),
+            presencePenalty: z.number().min(-2).max(2).nullish(),
         }),
-    ])
-        .nullish()
-        .transform((v) => (typeof v === "string" ? { name: v } : v)),
-    agents: z.array(z.string()).nullish(),
-    skills: z.array(z.string()).nullish(),
+    ])).transform((v) => (typeof v === "string" ? { name: v } : v)),
+    agents: optionalize(z.array(z.string())),
+    skills: optionalize(z.array(z.string())),
 });
 export async function loadAIGNEFile(path) {
-    const raw = await tryOrThrow(() => nodejs.fs.readFile(path, "utf8"), (error) => new Error(`Failed to load aigne.yaml from ${path}: ${error.message}`));
-    const json = await tryOrThrow(() => parse(raw), (error) => new Error(`Failed to parse aigne.yaml from ${path}: ${error.message}`));
-    const agent = tryOrThrow(() => aigneFileSchema.parse({ ...json, skills: json.skills ?? json.tools }), (error) => new Error(`Failed to validate aigne.yaml from ${path}: ${error.message}`));
-    return agent;
+    const file = await findAIGNEFile(path);
+    const raw = await tryOrThrow(() => nodejs.fs.readFile(file, "utf8"), (error) => new Error(`Failed to load aigne.yaml from ${file}: ${error.message}`));
+    const json = tryOrThrow(() => camelize(parse(raw)), (error) => new Error(`Failed to parse aigne.yaml from ${file}: ${error.message}`));
+    const aigne = tryOrThrow(() => aigneFileSchema.parse({
+        ...json,
+        model: json.model ?? json.chatModel,
+        skills: json.skills ?? json.tools,
+    }), (error) => new Error(`Failed to validate aigne.yaml from ${file}: ${error.message}`));
+    return { aigne, rootDir: nodejs.path.dirname(file) };
 }
-async function getAIGNEFilePath(path) {
-    const s = await nodejs.fs.stat(path);
-    if (s.isDirectory()) {
-        for (const file of AIGNE_FILE_NAME) {
-            const filePath = nodejs.path.join(path, file);
-            if ((await nodejs.fs.stat(filePath)).isFile())
-                return filePath;
+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 {
+            const stat = await nodejs.fs.stat(file);
+            if (stat.isFile())
+                return file;
         }
+        catch { }
     }
-    return path;
+    throw new Error(`aigne.yaml not found in ${path}. Please ensure you are in the correct directory or provide a valid path.`);
 }

package/lib/esm/loader/schema.js

@@ -2,6 +2,32 @@ import { nodejs } from "@aigne/platform-helpers/nodejs/index.js";
 import { parse } from "yaml";
 import { z } from "zod";
 export 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) };
+        }
+        // Load nested schemas from file
+        if (typeof schema === "string") {
+            const raw = await nodejs.fs.readFile(nodejs.path.join(nodejs.path.dirname(path), schema), "utf8");
+            return nestedJsonSchema.parseAsync(parse(raw));
+        }
+        return schema;
+    };
+    const nestedJsonSchema = z
+        .object({
+        type: z.string(),
+    })
+        .passthrough()
+        .transform((v) => includeExternalSchema(v));
     const jsonSchemaSchema = z
         .object({
         type: z.literal("object"),
@@ -9,29 +35,7 @@ export const inputOutputSchema = ({ path }) => {
         required: z.array(z.string()).optional(),
         additionalProperties: z.boolean().optional(),
     })
-        .transform((v) => {
-        const t = 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 t(value),
-                    ]))),
-                };
-            }
-            if (schema?.type === "array" && schema.items) {
-                return { ...schema, items: await t(schema.items) };
-            }
-            // Load nested schemas from file
-            if (typeof schema === "string") {
-                const raw = await nodejs.fs.readFile(nodejs.path.join(nodejs.path.dirname(path), schema), "utf8");
-                return jsonSchemaSchema.parseAsync(parse(raw));
-            }
-            return schema;
-        };
-        return t(v);
-    });
+        .transform((v) => includeExternalSchema(v));
     return z.union([
         z
             .string()

package/lib/esm/prompt/template.js

@@ -152,7 +152,7 @@ const chatModelOutputToolCallSchema = z.object({
     type: z.literal("function"),
     function: z.object({
         name: z.string(),
-        arguments: z.record(z.unknown()),
+        arguments: z.record(z.string(), z.unknown()),
     }),
 });
 const agentChatMessageSchema = z.object({
@@ -164,7 +164,7 @@ const agentChatMessageSchema = z.object({
 const toolChatMessageSchema = z.object({
     role: z.literal("tool"),
     content: z
-        .union([z.string(), z.record(z.unknown())])
+        .union([z.string(), z.record(z.string(), z.unknown())])
         .transform((val) => (typeof val !== "string" ? JSON.stringify(val) : val)),
     toolCallId: z.string(),
     name: z.string().optional(),