@aigne/core 1.10.0 → 1.11.0

package/CHANGELOG.md

@@ -22,6 +22,19 @@
 * rename @aigne/core-next to @aigne/core ([3a81009](https://github.com/AIGNE-io/aigne-framework/commit/3a8100962c81813217b687ae28e8de604419c622))
 * use text resource from MCP correctly ([8b9eba8](https://github.com/AIGNE-io/aigne-framework/commit/8b9eba83352ec096a2a5d4f410d4c4bde7420bce))
 
+## [1.11.0](https://github.com/AIGNE-io/aigne-framework/compare/core-v1.10.0...core-v1.11.0) (2025-04-23)
+
+
+### Features
+
+* **core:** enhance ClaudeChatModel to support streaming responses ([#85](https://github.com/AIGNE-io/aigne-framework/issues/85)) ([5433240](https://github.com/AIGNE-io/aigne-framework/commit/5433240e7b663ec9e9f4a79dffa05038088d54fc))
+* support set memory in agent yaml ([#90](https://github.com/AIGNE-io/aigne-framework/issues/90)) ([215118f](https://github.com/AIGNE-io/aigne-framework/commit/215118f1dc55f02322d59a3f18395a459198e031))
+
+
+### Bug Fixes
+
+* **core:** router model should support streaming response ([#88](https://github.com/AIGNE-io/aigne-framework/issues/88)) ([4fb4d92](https://github.com/AIGNE-io/aigne-framework/commit/4fb4d92f8b36011437efba3265591b2477f2d680))
+
 ## [1.10.0](https://github.com/AIGNE-io/aigne-framework/compare/core-v1.9.0...core-v1.10.0) (2025-04-22)
 
 

package/LICENSE

@@ -0,0 +1,93 @@
+Elastic License 2.0
+
+URL: https://www.elastic.co/licensing/elastic-license
+
+## Acceptance
+
+By using the software, you agree to all of the terms and conditions below.
+
+## Copyright License
+
+The licensor grants you a non-exclusive, royalty-free, worldwide,
+non-sublicensable, non-transferable license to use, copy, distribute, make
+available, and prepare derivative works of the software, in each case subject to
+the limitations and conditions below.
+
+## Limitations
+
+You may not provide the software to third parties as a hosted or managed
+service, where the service provides users with access to any substantial set of
+the features or functionality of the software.
+
+You may not move, change, disable, or circumvent the license key functionality
+in the software, and you may not remove or obscure any functionality in the
+software that is protected by the license key.
+
+You may not alter, remove, or obscure any licensing, copyright, or other notices
+of the licensor in the software. Any use of the licensor’s trademarks is subject
+to applicable law.
+
+## Patents
+
+The licensor grants you a license, under any patent claims the licensor can
+license, or becomes able to license, to make, have made, use, sell, offer for
+sale, import and have imported the software, in each case subject to the
+limitations and conditions in this license. This license does not cover any
+patent claims that you cause to be infringed by modifications or additions to
+the software. If you or your company make any written claim that the software
+infringes or contributes to infringement of any patent, your patent license for
+the software granted under these terms ends immediately. If your company makes
+such a claim, your patent license ends immediately for work on behalf of your
+company.
+
+## Notices
+
+You must ensure that anyone who gets a copy of any part of the software from you
+also gets a copy of these terms.
+
+If you modify the software, you must include in any modified copies of the
+software prominent notices stating that you have modified the software.
+
+## No Other Rights
+
+These terms do not imply any licenses other than those expressly granted in
+these terms.
+
+## Termination
+
+If you use the software in violation of these terms, such use is not licensed,
+and your licenses will automatically terminate. If the licensor provides you
+with a notice of your violation, and you cease all violation of this license no
+later than 30 days after you receive that notice, your licenses will be
+reinstated retroactively. However, if you violate these terms after such
+reinstatement, any additional violation of these terms will cause your licenses
+to terminate automatically and permanently.
+
+## No Liability
+
+*As far as the law allows, the software comes as is, without any warranty or
+condition, and the licensor will not be liable to you for any damages arising
+out of these terms or the use or nature of the software, under any kind of
+legal claim.*
+
+## Definitions
+
+The **licensor** is the entity offering these terms, and the **software** is the
+software the licensor makes available under these terms, including any portion
+of it.
+
+**you** refers to the individual or entity agreeing to these terms.
+
+**your company** is any legal entity, sole proprietorship, or other kind of
+organization that you work for, plus all organizations that have control over,
+are under the control of, or are under common control with that
+organization. **control** means ownership of substantially all the assets of an
+entity, or the power to direct its management and policies by vote, contract, or
+otherwise. Control can be direct or indirect.
+
+**your licenses** are all the licenses granted to you for the software under
+these terms.
+
+**use** means anything you do with the software requiring one of your licenses.
+
+**trademark** means trademarks, service marks, and similar rights.

package/README.md

@@ -0,0 +1,89 @@
+# @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.zh.md)
+
+Core library of [AIGNE Framework](https://github.com/AIGNE-io/aigne-framework) for building AI-powered applications.
+
+## Introduction
+
+`@aigne/core` is the foundation component of [AIGNE Framework](https://github.com/AIGNE-io/aigne-framework), providing the essential modules and tools needed to build AI-driven applications. This package implements the core functionalities of the framework, including agent systems, execution engines, model integrations, and workflow pattern support.
+
+## Features
+
+- **Multiple AI Model Support**: Built-in support for OpenAI, Gemini, Claude, and other mainstream AI models, easily extensible to support additional models
+- **Agent System**: Powerful agent abstractions supporting AI agents, function agents, MCP agents, and more
+- **Execution Engine**: Flexible execution engine handling communication between agents and workflow execution
+- **Workflow Patterns**: Support for sequential, concurrent, routing, handoff, and other workflow patterns
+- **MCP Protocol Integration**: Seamless integration with external systems through the Model Context Protocol
+- **TypeScript Support**: Comprehensive type definitions providing an excellent development experience
+
+## Installation
+
+```bash
+# Using npm
+npm install @aigne/core
+
+# Using yarn
+yarn add @aigne/core
+
+# Using pnpm
+pnpm add @aigne/core
+```
+
+## Basic Usage
+
+```typescript
+import { AIAgent, ExecutionEngine } from "@aigne/core";
+import { OpenAIChatModel } from "@aigne/core/models/openai-chat-model.js";
+
+// Create AI model instance
+const model = new OpenAIChatModel({
+  apiKey: process.env.OPENAI_API_KEY,
+  model: process.env.DEFAULT_CHAT_MODEL || "gpt-4-turbo",
+});
+
+// Create AI agent
+const agent = AIAgent.from({
+  name: "Assistant",
+  instructions: "You are a helpful assistant.",
+});
+
+// Create execution engine
+const engine = new ExecutionEngine({ model });
+
+// Use the execution engine to call the agent
+const userAgent = await engine.call(agent);
+
+// Send a message to the agent
+const response = await userAgent.call("Hello, can you help me write a short article?");
+console.log(response);
+```
+
+## Module Structure
+
+- `agents/`: Agent implementations, including AI agents, function agents, MCP agents, etc.
+- `execution-engine/`: Execution engine implementation
+- `loader/`: Loader-related functionality
+- `models/`: Integration with various AI models
+- `prompt/`: Prompt handling functionality
+- `utils/`: Utility functions and helper methods
+
+## Documentation
+
+For more detailed API documentation, please refer to:
+
+- [Agent API](../../docs/apis/agent-api.md)
+- [AI Agent API](../../docs/apis/ai-agent-api.md)
+- [Function Agent API](../../docs/apis/function-agent-api.md)
+- [MCP Agent API](../../docs/apis/mcp-agent-api.md)
+- [Execution Engine API](../../docs/apis/execution-engine-api.md)
+
+## License
+
+Elastic-2.0

package/README.zh.md

@@ -0,0 +1,89 @@
+# @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 框架](https://github.com/AIGNE-io/aigne-framework)的核心库，用于构建 AI 驱动的应用程序。
+
+## 简介
+
+`@aigne/core` 是 [AIGNE 框架](https://github.com/AIGNE-io/aigne-framework)的基础核心组件，提供了构建 AI 驱动应用程序所需的基本模块和工具。该包实现了框架的核心功能，包括代理系统、执行引擎、模型集成以及工作流模式支持等。
+
+## 特性
+
+- **多种 AI 模型支持**：内置支持 OpenAI、Gemini、Claude 等主流 AI 模型，可轻松扩展支持其他模型
+- **代理系统**：强大的代理抽象，支持 AI 代理、函数代理、MCP 代理等多种类型
+- **执行引擎**：灵活的执行引擎，处理代理之间的通信和工作流执行
+- **工作流模式**：支持顺序、并发、路由、交接等多种工作流模式
+- **MCP 协议集成**：通过模型上下文协议（Model Context Protocol）实现与外部系统的无缝集成
+- **TypeScript 支持**：完善的类型定义，提供良好的开发体验
+
+## 安装
+
+```bash
+# 使用 npm
+npm install @aigne/core
+
+# 使用 yarn
+yarn add @aigne/core
+
+# 使用 pnpm
+pnpm add @aigne/core
+```
+
+## 基本用法
+
+```typescript
+import { AIAgent, ExecutionEngine } 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: "助手",
+  instructions: "你是一个有帮助的助手。",
+});
+
+// 创建执行引擎
+const engine = new ExecutionEngine({ model });
+
+// 使用执行引擎调用代理
+const userAgent = await engine.call(agent);
+
+// 向代理发送消息
+const response = await userAgent.call("你好，能帮我写一篇短文吗？");
+console.log(response);
+```
+
+## 模块结构
+
+- `agents/`: 代理相关的实现，包括 AI 代理、函数代理、MCP 代理等
+- `execution-engine/`: 执行引擎的实现
+- `loader/`: 加载器相关功能
+- `models/`: 各种 AI 模型的集成
+- `prompt/`: 提示词处理相关功能
+- `utils/`: 工具函数和辅助方法
+
+## 文档
+
+更多详细的 API 文档，请参考：
+
+- [Agent API](../../docs/apis/agent-api.zh.md)
+- [AI Agent API](../../docs/apis/ai-agent-api.zh.md)
+- [Function Agent API](../../docs/apis/function-agent-api.zh.md)
+- [MCP Agent API](../../docs/apis/mcp-agent-api.zh.md)
+- [Execution Engine API](../../docs/apis/execution-engine-api.zh.md)
+
+## 协议
+
+Elastic-2.0

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

@@ -72,6 +72,7 @@ export declare abstract class Agent<I extends Message = Message, O extends Messa
 export type AgentResponse<T> = T | AgentResponseStream<T>;
 export type AgentResponseStream<T> = ReadableStream<AgentResponseChunk<T>>;
 export type AgentResponseChunk<T> = AgentResponseDelta<T>;
+export declare function isEmptyChunk<T>(chunk: AgentResponseChunk<T>): boolean;
 export interface AgentResponseDelta<T> {
     delta: {
         text?: Partial<{

package/lib/cjs/agents/agent.js

@@ -34,6 +34,7 @@ var __importStar = (this && this.__importStar) || (function () {
 })();
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.FunctionAgent = exports.Agent = void 0;
+exports.isEmptyChunk = isEmptyChunk;
 const node_util_1 = require("node:util");
 const zod_1 = require("zod");
 const prompt_builder_js_1 = require("../prompt/prompt-builder.js");
@@ -214,6 +215,9 @@ class Agent {
     }
 }
 exports.Agent = Agent;
+function isEmptyChunk(chunk) {
+    return (0, type_utils_js_1.isEmpty)(chunk.delta.json) && (0, type_utils_js_1.isEmpty)(chunk.delta.text);
+}
 function checkAgentInputOutputSchema(schema) {
     if (!(schema instanceof zod_1.ZodObject) && typeof schema !== "function") {
         throw new Error("schema must be a zod object or function return a zod object ");

package/lib/cjs/agents/ai-agent.d.ts

@@ -1,6 +1,7 @@
 import { z } from "zod";
 import type { Context } from "../execution-engine/context.js";
 import { ChatModel } from "../models/chat-model.js";
+import type { ChatModelInput } from "../models/chat-model.js";
 import { PromptBuilder } from "../prompt/prompt-builder.js";
 import { Agent, type AgentOptions, type AgentProcessAsyncGenerator, type Message } from "./agent.js";
 import { type TransferAgentOutput } from "./types.js";
@@ -28,35 +29,35 @@ export declare const aiAgentOptionsSchema: z.ZodObject<{
     disableLogging: z.ZodOptional<z.ZodBoolean>;
     memory: z.ZodOptional<z.ZodUnion<[z.ZodBoolean, z.ZodAny, z.ZodAny]>>;
 }, "strip", z.ZodTypeAny, {
-    description?: string | undefined;
-    memory?: any;
-    includeInputInOutput?: boolean | undefined;
-    subscribeTopic?: string | string[] | undefined;
-    publishTopic?: string | string[] | ((...args: unknown[]) => unknown) | undefined;
     tools?: (Agent<Message, Message> | ((...args: unknown[]) => unknown))[] | undefined;
     toolChoice?: Agent<Message, Message> | "auto" | "none" | "required" | "router" | undefined;
     name?: string | undefined;
+    description?: string | undefined;
     model?: ChatModel | undefined;
     instructions?: string | PromptBuilder | undefined;
     outputKey?: string | undefined;
     enableHistory?: boolean | undefined;
     maxHistoryMessages?: number | undefined;
-    disableLogging?: boolean | undefined;
-}, {
-    description?: string | undefined;
-    memory?: any;
     includeInputInOutput?: boolean | undefined;
     subscribeTopic?: string | string[] | undefined;
     publishTopic?: string | string[] | ((...args: unknown[]) => unknown) | undefined;
+    disableLogging?: boolean | undefined;
+    memory?: any;
+}, {
     tools?: (Agent<Message, Message> | ((...args: unknown[]) => unknown))[] | undefined;
     toolChoice?: Agent<Message, Message> | "auto" | "none" | "required" | "router" | undefined;
     name?: string | undefined;
+    description?: string | undefined;
     model?: ChatModel | undefined;
     instructions?: string | PromptBuilder | undefined;
     outputKey?: string | undefined;
     enableHistory?: boolean | undefined;
     maxHistoryMessages?: number | undefined;
+    includeInputInOutput?: boolean | undefined;
+    subscribeTopic?: string | string[] | undefined;
+    publishTopic?: string | string[] | ((...args: unknown[]) => unknown) | undefined;
     disableLogging?: boolean | undefined;
+    memory?: any;
 }>;
 export declare class AIAgent<I extends Message = Message, O extends Message = Message> extends Agent<I, O> {
     static from<I extends Message, O extends Message>(options: AIAgentOptions<I, O>): AIAgent<I, O>;
@@ -66,4 +67,5 @@ export declare class AIAgent<I extends Message = Message, O extends Message = Me
     outputKey?: string;
     toolChoice?: AIAgentToolChoice;
     process(input: I, context: Context): AgentProcessAsyncGenerator<O | TransferAgentOutput>;
+    processRouter(input: I, model: ChatModel, modelInput: ChatModelInput, context: Context, toolsMap: Map<string, Agent>): AgentProcessAsyncGenerator<O | TransferAgentOutput>;
 }

package/lib/cjs/agents/ai-agent.js

@@ -55,18 +55,22 @@ class AIAgent extends agent_js_1.Agent {
         const model = context.model ?? this.model;
         if (!model)
             throw new Error("model is required to run AIAgent");
-        const { toolAgents, messages, ...modelInput } = await this.instructions.build({
+        const { toolAgents, ...modelInput } = await this.instructions.build({
             agent: this,
             input,
             model,
             context,
         });
         const toolsMap = new Map(toolAgents?.map((i) => [i.name, i]));
+        if (this.toolChoice === "router") {
+            yield* this.processRouter(input, model, modelInput, context, toolsMap);
+            return;
+        }
         const toolCallMessages = [];
         const outputKey = this.outputKey || prompt_builder_js_1.MESSAGE_KEY;
         for (;;) {
             const modelOutput = {};
-            const stream = await context.call(model, { ...modelInput, messages: messages.concat(toolCallMessages) }, { streaming: true });
+            const stream = await context.call(model, { ...modelInput, messages: modelInput.messages.concat(toolCallMessages) }, { streaming: true });
             for await (const value of (0, stream_utils_js_1.readableStreamToAsyncIterator)(stream)) {
                 if (value.delta.text?.text) {
                     yield { delta: { text: { [outputKey]: value.delta.text.text } } };
@@ -94,15 +98,6 @@ class AIAgent extends agent_js_1.Agent {
                 // Continue LLM function calling loop if any tools were executed
                 if (executedToolCalls.length) {
                     toolCallMessages.push(template_js_1.AgentMessageTemplate.from(undefined, executedToolCalls.map(({ call }) => call)).format(), ...executedToolCalls.map(({ call, output }) => template_js_1.ToolMessageTemplate.from(output, call.id).format()));
-                    // Return the output of the first tool if the toolChoice is "router"
-                    if (this.toolChoice === "router") {
-                        const output = executedToolCalls[0]?.output;
-                        const { supportsParallelToolCalls } = model.getModelCapabilities();
-                        if (!output || (supportsParallelToolCalls && executedToolCalls.length !== 1)) {
-                            throw new Error("Router toolChoice requires exactly one tool to be executed");
-                        }
-                        return output;
-                    }
                     continue;
                 }
             }
@@ -119,5 +114,16 @@ class AIAgent extends agent_js_1.Agent {
             return;
         }
     }
+    async *processRouter(input, model, modelInput, context, toolsMap) {
+        const { toolCalls: [call] = [], } = await context.call(model, modelInput);
+        if (!call) {
+            throw new Error("Router toolChoice requires exactly one tool to be executed");
+        }
+        const tool = toolsMap.get(call.function.name);
+        if (!tool)
+            throw new Error(`Tool not found: ${call.function.name}`);
+        const stream = await context.call(tool, { ...call.function.arguments, ...input }, { streaming: true });
+        yield* (0, stream_utils_js_1.readableStreamToAsyncIterator)(stream);
+    }
 }
 exports.AIAgent = AIAgent;

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

@@ -4,12 +4,12 @@ export declare function loadAgentFromJsFile(path: string): Promise<{
     name: string;
     fn: (args_0: Message) => Message;
     description?: string | undefined;
-    input_schema?: ZodObject<Record<string, ZodType<any, z.ZodTypeDef, any>>, z.UnknownKeysParam, z.ZodTypeAny, {
+    inputSchema?: ZodObject<Record<string, ZodType<any, z.ZodTypeDef, any>>, z.UnknownKeysParam, z.ZodTypeAny, {
         [x: string]: any;
     }, {
         [x: string]: any;
     }> | undefined;
-    output_schema?: ZodObject<Record<string, ZodType<any, z.ZodTypeDef, any>>, z.UnknownKeysParam, z.ZodTypeAny, {
+    outputSchema?: ZodObject<Record<string, ZodType<any, z.ZodTypeDef, any>>, z.UnknownKeysParam, z.ZodTypeAny, {
         [x: string]: any;
     }, {
         [x: string]: any;

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

@@ -36,6 +36,7 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.loadAgentFromJsFile = loadAgentFromJsFile;
 const json_schema_to_zod_1 = require("@aigne/json-schema-to-zod");
 const zod_1 = require("zod");
+const camelize_js_1 = require("../utils/camelize.js");
 const type_utils_js_1 = require("../utils/type-utils.js");
 const schema_js_1 = require("./schema.js");
 const agentJsFileSchema = zod_1.z.object({
@@ -57,11 +58,9 @@ async function loadAgentFromJsFile(path) {
     if (typeof agent !== "function") {
         throw new Error(`Agent file ${path} must export a default function, but got ${typeof agent}`);
     }
-    return (0, type_utils_js_1.tryOrThrow)(() => agentJsFileSchema.parse({
+    return (0, type_utils_js_1.tryOrThrow)(() => (0, camelize_js_1.customCamelize)(agentJsFileSchema.parse({
+        ...agent,
         name: agent.agent_name || agent.name,
-        description: agent.description,
-        input_schema: agent.input_schema,
-        output_schema: agent.output_schema,
         fn: agent,
-    }), (error) => new Error(`Failed to parse agent from ${path}: ${error.message}`));
+    }), { shallowKeys: ["input_schema", "output_schema"] }), (error) => new Error(`Failed to parse agent from ${path}: ${error.message}`));
 }

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

@@ -2,21 +2,24 @@ import { type ZodObject, type ZodType, z } from "zod";
 export declare function loadAgentFromYamlFile(path: string): Promise<{
     type: "ai";
     name: string;
-    description?: string | undefined;
     tools?: string[] | undefined;
+    description?: string | undefined;
     instructions?: string | undefined;
-    input_schema?: ZodObject<Record<string, ZodType<any, z.ZodTypeDef, any>>, z.UnknownKeysParam, z.ZodTypeAny, {
+    memory?: true | {
+        subscribeTopic: string[];
+    } | undefined;
+    inputSchema?: ZodObject<Record<string, ZodType<any, z.ZodTypeDef, any>>, z.UnknownKeysParam, z.ZodTypeAny, {
         [x: string]: any;
     }, {
         [x: string]: any;
     }> | undefined;
-    tool_choice?: "auto" | "none" | "required" | "router" | undefined;
-    output_schema?: ZodObject<Record<string, ZodType<any, z.ZodTypeDef, any>>, z.UnknownKeysParam, z.ZodTypeAny, {
+    toolChoice?: "auto" | "none" | "required" | "router" | undefined;
+    outputSchema?: ZodObject<Record<string, ZodType<any, z.ZodTypeDef, any>>, z.UnknownKeysParam, z.ZodTypeAny, {
         [x: string]: any;
     }, {
         [x: string]: any;
     }> | undefined;
-    output_key?: string | undefined;
+    outputKey?: string | undefined;
 } | {
     type: "mcp";
     url?: string | undefined;

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

@@ -5,6 +5,7 @@ const promises_1 = require("node:fs/promises");
 const json_schema_to_zod_1 = require("@aigne/json-schema-to-zod");
 const yaml_1 = require("yaml");
 const zod_1 = require("zod");
+const camelize_js_1 = require("../utils/camelize.js");
 const type_utils_js_1 = require("../utils/type-utils.js");
 const schema_js_1 = require("./schema.js");
 const agentFileSchema = zod_1.z.discriminatedUnion("type", [
@@ -37,6 +38,15 @@ const agentFileSchema = zod_1.z.discriminatedUnion("type", [
             .union([zod_1.z.literal("auto"), zod_1.z.literal("none"), zod_1.z.literal("required"), zod_1.z.literal("router")])
             .nullish()
             .transform((v) => v ?? undefined),
+        memory: zod_1.z
+            .union([
+            zod_1.z.boolean(),
+            zod_1.z.object({
+                subscribe_topic: zod_1.z.array(zod_1.z.string()),
+            }),
+        ])
+            .nullish()
+            .transform((v) => v || undefined),
     }),
     zod_1.z.object({
         type: zod_1.z.literal("mcp"),
@@ -57,6 +67,8 @@ const agentFileSchema = zod_1.z.discriminatedUnion("type", [
 async function loadAgentFromYamlFile(path) {
     const raw = await (0, type_utils_js_1.tryOrThrow)(() => (0, promises_1.readFile)(path, "utf8"), (error) => new Error(`Failed to load agent definition from ${path}: ${error.message}`));
     const json = await (0, type_utils_js_1.tryOrThrow)(() => (0, yaml_1.parse)(raw), (error) => new Error(`Failed to parse agent definition from ${path}: ${error.message}`));
-    const agent = (0, type_utils_js_1.tryOrThrow)(() => agentFileSchema.parse({ ...json, type: json.type ?? "ai" }), (error) => new Error(`Failed to validate agent definition from ${path}: ${error.message}`));
+    const agent = (0, type_utils_js_1.tryOrThrow)(() => (0, camelize_js_1.customCamelize)(agentFileSchema.parse({ ...json, type: json.type ?? "ai" }), {
+        shallowKeys: ["input_schema", "output_schema"],
+    }), (error) => new Error(`Failed to validate agent definition from ${path}: ${error.message}`));
     return agent;
 }

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

@@ -8,8 +8,8 @@ export declare function load(options: LoadOptions): Promise<{
     model: ChatModel | undefined;
     agents: Agent<import("../agents/agent.js").Message, import("../agents/agent.js").Message>[];
     tools: Agent<import("../agents/agent.js").Message, import("../agents/agent.js").Message>[];
-    description?: string | null | undefined;
     name?: string | null | undefined;
+    description?: string | null | undefined;
     chat_model?: {
         name?: string | null | undefined;
         temperature?: number | null | undefined;
@@ -63,9 +63,9 @@ declare const aigneFileSchema: z.ZodObject<{
     agents: z.ZodOptional<z.ZodNullable<z.ZodArray<z.ZodString, "many">>>;
     tools: z.ZodOptional<z.ZodNullable<z.ZodArray<z.ZodString, "many">>>;
 }, "strip", z.ZodTypeAny, {
-    description?: string | null | undefined;
     tools?: string[] | null | undefined;
     name?: string | null | undefined;
+    description?: string | null | undefined;
     chat_model?: {
         name?: string | null | undefined;
         temperature?: number | null | undefined;
@@ -76,9 +76,9 @@ declare const aigneFileSchema: z.ZodObject<{
     } | null | undefined;
     agents?: string[] | null | undefined;
 }, {
-    description?: string | null | undefined;
     tools?: string[] | null | undefined;
     name?: string | null | undefined;
+    description?: string | null | undefined;
     chat_model?: string | {
         name?: string | null | undefined;
         temperature?: number | null | undefined;
@@ -90,9 +90,9 @@ declare const aigneFileSchema: z.ZodObject<{
     agents?: string[] | null | undefined;
 }>;
 export declare function loadAIGNEFile(path: string): Promise<{
-    description?: string | null | undefined;
     tools?: string[] | null | undefined;
     name?: string | null | undefined;
+    description?: string | null | undefined;
     chat_model?: {
         name?: string | null | undefined;
         temperature?: number | null | undefined;

package/lib/cjs/loader/index.js

@@ -39,26 +39,15 @@ async function load(options) {
 async function loadAgent(path) {
     if ((0, node_path_1.extname)(path) === ".js") {
         const agent = await (0, agent_js_js_1.loadAgentFromJsFile)(path);
-        return agent_js_1.FunctionAgent.from({
-            name: agent.name,
-            description: agent.description,
-            inputSchema: agent.input_schema,
-            outputSchema: agent.output_schema,
-            fn: agent.fn,
-        });
+        return agent_js_1.FunctionAgent.from(agent);
     }
     if ((0, node_path_1.extname)(path) === ".yaml" || (0, node_path_1.extname)(path) === ".yml") {
         const agent = await (0, agent_yaml_js_1.loadAgentFromYamlFile)(path);
         if (agent.type === "ai") {
             return ai_agent_js_1.AIAgent.from({
-                name: agent.name,
-                description: agent.description,
-                instructions: agent.instructions,
-                inputSchema: agent.input_schema,
-                outputSchema: agent.output_schema,
-                outputKey: agent.output_key,
-                tools: await Promise.all((agent.tools ?? []).map((filename) => loadAgent((0, node_path_1.join)((0, node_path_1.dirname)(path), filename)))),
-                toolChoice: agent.tool_choice,
+                ...agent,
+                tools: agent.tools &&
+                    (await Promise.all(agent.tools.map((filename) => loadAgent((0, node_path_1.join)((0, node_path_1.dirname)(path), filename))))),
             });
         }
         if (agent.type === "mcp") {

package/lib/cjs/models/claude-chat-model.d.ts

@@ -1,5 +1,7 @@
 import Anthropic from "@anthropic-ai/sdk";
 import { z } from "zod";
+import type { AgentCallOptions, AgentResponse } from "../agents/agent.js";
+import type { Context } from "../execution-engine/context.js";
 import { ChatModel, type ChatModelInput, type ChatModelOptions, type ChatModelOutput } from "./chat-model.js";
 export interface ClaudeChatModelOptions {
     apiKey?: string;
@@ -60,7 +62,7 @@ export declare class ClaudeChatModel extends ChatModel {
     protected _client?: Anthropic;
     get client(): Anthropic;
     get modelOptions(): ChatModelOptions | undefined;
-    process(input: ChatModelInput): Promise<ChatModelOutput>;
+    process(input: ChatModelInput, _context: Context, options?: AgentCallOptions): Promise<AgentResponse<ChatModelOutput>>;
     private extractResultFromClaudeStream;
     private requestStructuredOutput;
 }