@aigne/core 1.32.2 → 1.33.1

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.d.ts

@@ -1,5 +1,7 @@
 import { type ZodType, z } from "zod";
-export declare const inputOutputSchema: z.ZodObject<{
+export 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">>;
@@ -14,5 +16,10 @@ export declare const inputOutputSchema: z.ZodObject<{
     properties: Record<string, any>;
     required?: string[] | undefined;
     additionalProperties?: boolean | undefined;
-}>;
+}>, any, {
+    type: "object";
+    properties: Record<string, any>;
+    required?: string[] | undefined;
+    additionalProperties?: boolean | undefined;
+}>]>;
 export declare function optionalize<T>(schema: ZodType<T>): ZodType<T | undefined>;

package/lib/esm/loader/schema.js

@@ -1,10 +1,46 @@
+import { nodejs } from "@aigne/platform-helpers/nodejs/index.js";
+import { parse } from "yaml";
 import { z } from "zod";
-export const inputOutputSchema = z.object({
-    type: z.literal("object"),
-    properties: z.record(z.any()),
-    required: z.array(z.string()).optional(),
-    additionalProperties: z.boolean().optional(),
-});
+export const inputOutputSchema = ({ path }) => {
+    const jsonSchemaSchema = z
+        .object({
+        type: z.literal("object"),
+        properties: z.record(z.any()),
+        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);
+    });
+    return z.union([
+        z
+            .string()
+            .transform((v) => nodejs.fs
+            .readFile(nodejs.path.join(nodejs.path.dirname(path), v), "utf8")
+            .then((raw) => jsonSchemaSchema.parseAsync(parse(raw)))),
+        jsonSchemaSchema,
+    ]);
+};
 export function optionalize(schema) {
     return schema.nullish().transform((v) => v ?? undefined);
 }

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(),

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@aigne/core",
-  "version": "1.32.2",
+  "version": "1.33.1",
   "description": "AIGNE core library for building AI-powered applications",
   "publishConfig": {
     "access": "public"
@@ -63,7 +63,7 @@
   },
   "dependencies": {
     "@aigne/json-schema-to-zod": "^1.3.3",
-    "@modelcontextprotocol/sdk": "^1.13.3",
+    "@modelcontextprotocol/sdk": "^1.15.0",
     "@types/debug": "^4.1.12",
     "camelize-ts": "^3.0.0",
     "content-type": "^1.0.5",
@@ -83,16 +83,16 @@
     "yaml": "^2.8.0",
     "zod": "^3.25.67",
     "zod-to-json-schema": "^3.24.6",
-    "@aigne/observability-api": "^0.7.1",
-    "@aigne/platform-helpers": "^0.3.0"
+    "@aigne/observability-api": "^0.7.2",
+    "@aigne/platform-helpers": "^0.3.1"
   },
   "devDependencies": {
-    "@types/bun": "^1.2.17",
+    "@types/bun": "^1.2.18",
     "@types/compression": "^1.8.1",
     "@types/content-type": "^1.1.9",
     "@types/express": "^5.0.3",
     "@types/mustache": "^4.2.6",
-    "@types/node": "^24.0.10",
+    "@types/node": "^24.0.12",
     "@types/nunjucks": "^3.2.6",
     "compression": "^1.8.0",
     "detect-port": "^2.1.0",

package/README.zh.md

@@ -1,79 +0,0 @@
-# @aigne/core
-
-[![GitHub star chart](https://img.shields.io/github/stars/AIGNE-io/aigne-framework?style=flat-square)](https://star-history.com/#AIGNE-io/aigne-framework)
-[![Open Issues](https://img.shields.io/github/issues-raw/AIGNE-io/aigne-framework?style=flat-square)](https://github.com/AIGNE-io/aigne-framework/issues)
-[![codecov](https://codecov.io/gh/AIGNE-io/aigne-framework/graph/badge.svg?token=DO07834RQL)](https://codecov.io/gh/AIGNE-io/aigne-framework)
-[![NPM Version](https://img.shields.io/npm/v/@aigne/core)](https://www.npmjs.com/package/@aigne/core)
-[![Elastic-2.0 licensed](https://img.shields.io/npm/l/@aigne/core)](https://github.com/AIGNE-io/aigne-framework/blob/main/LICENSE)
-
-[English](./README.md) | **中文**
-
-[AIGNE Framework](https://github.com/AIGNE-io/aigne-framework) 的核心库，用于构建 AI 驱动的应用程序。
-
-## 简介
-
-`@aigne/core` 是 [AIGNE Framework](https://github.com/AIGNE-io/aigne-framework) 的基础组件，提供构建 AI 驱动应用程序所需的核心模块和工具。该包实现了框架的核心功能，包括代理系统、aigne 环境、模型集成和工作流模式支持。
-
-## 特性
-
-* **多 AI 模型支持**：内置支持 OpenAI、Gemini、Claude、Nova 等主流 AI 模型，可轻松扩展支持其他模型
-* **代理系统**：强大的代理抽象，支持 AI 代理、功能代理、MCP 代理等
-* **AIGNE 环境**：灵活处理代理之间的通信和工作流执行
-* **工作流模式**：支持顺序、并发、路由、交接等多种工作流模式
-* **MCP 协议集成**：通过模型上下文协议与外部系统无缝集成
-* **TypeScript 支持**：全面的类型定义，提供出色的开发体验
-
-## 安装
-
-### 使用 npm
-
-```bash
-npm install @aigne/core
-```
-
-### 使用 yarn
-
-```bash
-yarn add @aigne/core
-```
-
-### 使用 pnpm
-
-```bash
-pnpm add @aigne/core
-```
-
-## 基本用法
-
-```typescript
-import { AIAgent, AIGNE } from "@aigne/core";
-import { OpenAIChatModel } from "@aigne/core/models/openai-chat-model.js";
-
-// 创建 AI 模型实例
-const model = new OpenAIChatModel({
-  apiKey: process.env.OPENAI_API_KEY,
-  model: process.env.DEFAULT_CHAT_MODEL || "gpt-4-turbo",
-});
-
-// 创建 AI 代理
-const agent = AIAgent.from({
-  name: "Assistant",
-  instructions: "You are a helpful assistant.",
-});
-
-// AIGNE: AIGNE Framework 的主要执行引擎
-const aigne = new AIGNE({ model });
-
-// 使用 AIGNE 调用代理
-const userAgent = await aigne.invoke(agent);
-
-// 向代理发送消息
-const response = await userAgent.invoke(
-  "Hello, can you help me write a short article?",
-);
-console.log(response);
-```
-
-## 许可证
-
-Elastic-2.0