@aigne/core 1.63.0-beta → 1.63.0-beta.2

package/CHANGELOG.md

@@ -1,5 +1,24 @@
 # Changelog
 
+## [1.63.0-beta.2](https://github.com/AIGNE-io/aigne-framework/compare/core-v1.63.0-beta.1...core-v1.63.0-beta.2) (2025-10-09)
+
+
+### Bug Fixes
+
+* input schema of AI agent should includes input key and input file key ([#600](https://github.com/AIGNE-io/aigne-framework/issues/600)) ([b4ca076](https://github.com/AIGNE-io/aigne-framework/commit/b4ca076d6b4a1a1ecb8d4ebb008abd0d7561aadd))
+
+## [1.63.0-beta.1](https://github.com/AIGNE-io/aigne-framework/compare/core-v1.63.0-beta...core-v1.63.0-beta.1) (2025-10-09)
+
+
+### Features
+
+* **core:** add `toolCallsConcurrency` support for AI agent ([#598](https://github.com/AIGNE-io/aigne-framework/issues/598)) ([84df406](https://github.com/AIGNE-io/aigne-framework/commit/84df406bfa9e3bdf159509f4b9cf2301ec80b155))
+
+
+### Bug Fixes
+
+* **core:** add input_file_key support for agent yaml ([#597](https://github.com/AIGNE-io/aigne-framework/issues/597)) ([63414a3](https://github.com/AIGNE-io/aigne-framework/commit/63414a3d46c74c686e7f033c224ca6175bea8c3f))
+
 ## [1.63.0-beta](https://github.com/AIGNE-io/aigne-framework/compare/core-v1.62.0...core-v1.63.0-beta) (2025-10-07)
 
 

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

@@ -3,7 +3,7 @@ import { PromptBuilder } from "../prompt/prompt-builder.js";
 import { Agent, type AgentInvokeOptions, type AgentOptions, type AgentProcessAsyncGenerator, type AgentProcessResult, type Message } from "./agent.js";
 import type { ChatModel, ChatModelInput } from "./chat-model.js";
 import type { GuideRailAgentOutput } from "./guide-rail-agent.js";
-import type { FileType } from "./model.js";
+import { type FileType } from "./model.js";
 export declare const DEFAULT_OUTPUT_KEY = "message";
 export declare const DEFAULT_OUTPUT_FILE_KEY = "files";
 /**
@@ -42,6 +42,12 @@ export interface AIAgentOptions<I extends Message = Message, O extends Message =
      * @default AIAgentToolChoice.auto
      */
     toolChoice?: AIAgentToolChoice | Agent;
+    /**
+     * Maximum number of tool calls to execute concurrently
+     *
+     * @default 1
+     */
+    toolCallsConcurrency?: number;
     /**
      * Whether to preserve text generated during tool usage in the final output
      */
@@ -237,6 +243,12 @@ export declare class AIAgent<I extends Message = any, O extends Message = any> e
      * {@includeCode ../../test/agents/ai-agent.test.ts#example-ai-agent-router}
      */
     toolChoice?: AIAgentToolChoice | Agent;
+    /**
+     * Maximum number of tool calls to execute concurrently
+     *
+     * @default 1
+     */
+    toolCallsConcurrency?: number;
     /**
      * Whether to preserve text generated during tool usage in the final output
      */
@@ -299,6 +311,7 @@ export declare class AIAgent<I extends Message = any, O extends Message = any> e
         metadataEnd: string;
         parse: (raw: string) => object;
     };
+    get inputSchema(): ZodType<I>;
     /**
      * Process an input message and generate a response
      *

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

@@ -1,6 +1,10 @@
 "use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.AIAgent = exports.aiAgentOptionsSchema = exports.aiAgentToolChoiceSchema = exports.AIAgentToolChoice = exports.DEFAULT_OUTPUT_FILE_KEY = exports.DEFAULT_OUTPUT_KEY = void 0;
+const fastq_1 = __importDefault(require("fastq"));
 const zod_1 = require("zod");
 const prompt_builder_js_1 = require("../prompt/prompt-builder.js");
 const structured_stream_instructions_js_1 = require("../prompt/prompts/structured-stream-instructions.js");
@@ -8,6 +12,7 @@ const template_js_1 = require("../prompt/template.js");
 const structured_stream_extractor_js_1 = require("../utils/structured-stream-extractor.js");
 const type_utils_js_1 = require("../utils/type-utils.js");
 const agent_js_1 = require("./agent.js");
+const model_js_1 = require("./model.js");
 const types_js_1 = require("./types.js");
 exports.DEFAULT_OUTPUT_KEY = "message";
 exports.DEFAULT_OUTPUT_FILE_KEY = "files";
@@ -58,6 +63,7 @@ exports.aiAgentOptionsSchema = agent_js_1.agentOptionsSchema.extend({
     inputKey: zod_1.z.string().optional(),
     outputKey: zod_1.z.string().optional(),
     toolChoice: exports.aiAgentToolChoiceSchema.optional(),
+    toolCallsConcurrency: zod_1.z.number().int().min(0).optional(),
     keepTextInToolUses: zod_1.z.boolean().optional(),
     memoryAgentsAsTools: zod_1.z.boolean().optional(),
     memoryPromptTemplate: zod_1.z.string().optional(),
@@ -117,6 +123,7 @@ class AIAgent extends agent_js_1.Agent {
         this.outputFileKey = options.outputFileKey || exports.DEFAULT_OUTPUT_FILE_KEY;
         this.outputFileType = options.outputFileType;
         this.toolChoice = options.toolChoice;
+        this.toolCallsConcurrency = options.toolCallsConcurrency || 1;
         this.keepTextInToolUses = options.keepTextInToolUses;
         this.memoryAgentsAsTools = options.memoryAgentsAsTools;
         this.memoryPromptTemplate = options.memoryPromptTemplate;
@@ -173,6 +180,12 @@ class AIAgent extends agent_js_1.Agent {
      * {@includeCode ../../test/agents/ai-agent.test.ts#example-ai-agent-router}
      */
     toolChoice;
+    /**
+     * Maximum number of tool calls to execute concurrently
+     *
+     * @default 1
+     */
+    toolCallsConcurrency;
     /**
      * Whether to preserve text generated during tool usage in the final output
      */
@@ -230,6 +243,20 @@ class AIAgent extends agent_js_1.Agent {
      * which outputs structured data in YAML format within <metadata> tags.
      */
     customStructuredStreamInstructions;
+    get inputSchema() {
+        let schema = super.inputSchema;
+        if (this.inputKey) {
+            schema = schema.extend({
+                [this.inputKey]: zod_1.z.string().nullish(),
+            });
+        }
+        if (this.inputFileKey) {
+            schema = schema.extend({
+                [this.inputFileKey]: model_js_1.fileUnionContentsSchema.nullish(),
+            });
+        }
+        return schema;
+    }
     /**
      * Process an input message and generate a response
      *
@@ -287,31 +314,43 @@ class AIAgent extends agent_js_1.Agent {
                     yield { delta: { text: { [outputKey]: "\n" } } };
                 }
                 const executedToolCalls = [];
+                let error;
+                const queue = fastq_1.default.promise(async ({ tool, call }) => {
+                    try {
+                        // NOTE: should pass both arguments (model generated) and input (user provided) to the tool
+                        const output = await this.invokeSkill(tool, { ...input, ...call.function.arguments }, options).catch((error) => {
+                            if (!this.catchToolsError) {
+                                return Promise.reject(error);
+                            }
+                            return {
+                                isError: true,
+                                error: {
+                                    message: error.message,
+                                },
+                            };
+                        });
+                        executedToolCalls.push({ call, output });
+                    }
+                    catch (e) {
+                        error = e;
+                        queue.killAndDrain();
+                    }
+                }, this.toolCallsConcurrency || 1);
                 // Execute tools
                 for (const call of toolCalls) {
                     const tool = toolsMap.get(call.function.name);
                     if (!tool)
                         throw new Error(`Tool not found: ${call.function.name}`);
-                    // NOTE: should pass both arguments (model generated) and input (user provided) to the tool
-                    const output = await this.invokeSkill(tool, { ...input, ...call.function.arguments }, options).catch((error) => {
-                        if (!this.catchToolsError) {
-                            return Promise.reject(error);
-                        }
-                        return {
-                            isError: true,
-                            error: {
-                                message: error.message,
-                            },
-                        };
-                    });
-                    // NOTE: Return transfer output immediately
-                    if ((0, types_js_1.isTransferAgentOutput)(output)) {
-                        return output;
-                    }
-                    executedToolCalls.push({ call, output });
+                    queue.push({ tool, call });
                 }
+                await queue.drained();
+                if (error)
+                    throw error;
                 // Continue LLM function calling loop if any tools were executed
                 if (executedToolCalls.length) {
+                    const transferOutput = executedToolCalls.find((i) => (0, types_js_1.isTransferAgentOutput)(i.output))?.output;
+                    if (transferOutput)
+                        return transferOutput;
                     toolCallMessages.push(await template_js_1.AgentMessageTemplate.from(undefined, executedToolCalls.map(({ call }) => call)).format(), ...(await Promise.all(executedToolCalls.map(({ call, output }) => template_js_1.ToolMessageTemplate.from(output, call.id).format()))));
                     continue;
                 }

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

@@ -46,8 +46,10 @@ export interface AIAgentSchema extends BaseAgentSchema {
     type: "ai";
     instructions?: Instructions;
     inputKey?: string;
+    inputFileKey?: string;
     outputKey?: string;
     toolChoice?: AIAgentToolChoice;
+    toolCallsConcurrency?: number;
     keepTextInToolUses?: boolean;
 }
 export interface ImageAgentSchema extends BaseAgentSchema {

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

@@ -95,7 +95,10 @@ async function parseAgentFile(path, data) {
                 instructions: (0, schema_js_1.optionalize)(instructionsSchema),
                 inputKey: (0, schema_js_1.optionalize)(zod_1.z.string()),
                 outputKey: (0, schema_js_1.optionalize)(zod_1.z.string()),
+                inputFileKey: (0, schema_js_1.optionalize)(zod_1.z.string()),
+                outputFileKey: (0, schema_js_1.optionalize)(zod_1.z.string()),
                 toolChoice: (0, schema_js_1.optionalize)(zod_1.z.nativeEnum(ai_agent_js_1.AIAgentToolChoice)),
+                toolCallsConcurrency: (0, schema_js_1.optionalize)(zod_1.z.number().int().min(0)),
                 keepTextInToolUses: (0, schema_js_1.optionalize)(zod_1.z.boolean()),
                 structuredStreamMode: (0, schema_js_1.optionalize)(zod_1.z.boolean()),
             })

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

@@ -3,7 +3,7 @@ import { PromptBuilder } from "../prompt/prompt-builder.js";
 import { Agent, type AgentInvokeOptions, type AgentOptions, type AgentProcessAsyncGenerator, type AgentProcessResult, type Message } from "./agent.js";
 import type { ChatModel, ChatModelInput } from "./chat-model.js";
 import type { GuideRailAgentOutput } from "./guide-rail-agent.js";
-import type { FileType } from "./model.js";
+import { type FileType } from "./model.js";
 export declare const DEFAULT_OUTPUT_KEY = "message";
 export declare const DEFAULT_OUTPUT_FILE_KEY = "files";
 /**
@@ -42,6 +42,12 @@ export interface AIAgentOptions<I extends Message = Message, O extends Message =
      * @default AIAgentToolChoice.auto
      */
     toolChoice?: AIAgentToolChoice | Agent;
+    /**
+     * Maximum number of tool calls to execute concurrently
+     *
+     * @default 1
+     */
+    toolCallsConcurrency?: number;
     /**
      * Whether to preserve text generated during tool usage in the final output
      */
@@ -237,6 +243,12 @@ export declare class AIAgent<I extends Message = any, O extends Message = any> e
      * {@includeCode ../../test/agents/ai-agent.test.ts#example-ai-agent-router}
      */
     toolChoice?: AIAgentToolChoice | Agent;
+    /**
+     * Maximum number of tool calls to execute concurrently
+     *
+     * @default 1
+     */
+    toolCallsConcurrency?: number;
     /**
      * Whether to preserve text generated during tool usage in the final output
      */
@@ -299,6 +311,7 @@ export declare class AIAgent<I extends Message = any, O extends Message = any> e
         metadataEnd: string;
         parse: (raw: string) => object;
     };
+    get inputSchema(): ZodType<I>;
     /**
      * Process an input message and generate a response
      *

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

@@ -46,8 +46,10 @@ export interface AIAgentSchema extends BaseAgentSchema {
     type: "ai";
     instructions?: Instructions;
     inputKey?: string;
+    inputFileKey?: string;
     outputKey?: string;
     toolChoice?: AIAgentToolChoice;
+    toolCallsConcurrency?: number;
     keepTextInToolUses?: boolean;
 }
 export interface ImageAgentSchema extends BaseAgentSchema {

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

@@ -3,7 +3,7 @@ import { PromptBuilder } from "../prompt/prompt-builder.js";
 import { Agent, type AgentInvokeOptions, type AgentOptions, type AgentProcessAsyncGenerator, type AgentProcessResult, type Message } from "./agent.js";
 import type { ChatModel, ChatModelInput } from "./chat-model.js";
 import type { GuideRailAgentOutput } from "./guide-rail-agent.js";
-import type { FileType } from "./model.js";
+import { type FileType } from "./model.js";
 export declare const DEFAULT_OUTPUT_KEY = "message";
 export declare const DEFAULT_OUTPUT_FILE_KEY = "files";
 /**
@@ -42,6 +42,12 @@ export interface AIAgentOptions<I extends Message = Message, O extends Message =
      * @default AIAgentToolChoice.auto
      */
     toolChoice?: AIAgentToolChoice | Agent;
+    /**
+     * Maximum number of tool calls to execute concurrently
+     *
+     * @default 1
+     */
+    toolCallsConcurrency?: number;
     /**
      * Whether to preserve text generated during tool usage in the final output
      */
@@ -237,6 +243,12 @@ export declare class AIAgent<I extends Message = any, O extends Message = any> e
      * {@includeCode ../../test/agents/ai-agent.test.ts#example-ai-agent-router}
      */
     toolChoice?: AIAgentToolChoice | Agent;
+    /**
+     * Maximum number of tool calls to execute concurrently
+     *
+     * @default 1
+     */
+    toolCallsConcurrency?: number;
     /**
      * Whether to preserve text generated during tool usage in the final output
      */
@@ -299,6 +311,7 @@ export declare class AIAgent<I extends Message = any, O extends Message = any> e
         metadataEnd: string;
         parse: (raw: string) => object;
     };
+    get inputSchema(): ZodType<I>;
     /**
      * Process an input message and generate a response
      *

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

@@ -1,3 +1,4 @@
+import fastq from "fastq";
 import { z } from "zod";
 import { PromptBuilder } from "../prompt/prompt-builder.js";
 import { STRUCTURED_STREAM_INSTRUCTIONS } from "../prompt/prompts/structured-stream-instructions.js";
@@ -5,6 +6,7 @@ import { AgentMessageTemplate, ToolMessageTemplate } from "../prompt/template.js
 import { ExtractMetadataTransform } from "../utils/structured-stream-extractor.js";
 import { checkArguments, isEmpty } from "../utils/type-utils.js";
 import { Agent, agentOptionsSchema, isAgentResponseDelta, } from "./agent.js";
+import { fileUnionContentsSchema } from "./model.js";
 import { isTransferAgentOutput } from "./types.js";
 export const DEFAULT_OUTPUT_KEY = "message";
 export const DEFAULT_OUTPUT_FILE_KEY = "files";
@@ -55,6 +57,7 @@ export const aiAgentOptionsSchema = agentOptionsSchema.extend({
     inputKey: z.string().optional(),
     outputKey: z.string().optional(),
     toolChoice: aiAgentToolChoiceSchema.optional(),
+    toolCallsConcurrency: z.number().int().min(0).optional(),
     keepTextInToolUses: z.boolean().optional(),
     memoryAgentsAsTools: z.boolean().optional(),
     memoryPromptTemplate: z.string().optional(),
@@ -114,6 +117,7 @@ export class AIAgent extends Agent {
         this.outputFileKey = options.outputFileKey || DEFAULT_OUTPUT_FILE_KEY;
         this.outputFileType = options.outputFileType;
         this.toolChoice = options.toolChoice;
+        this.toolCallsConcurrency = options.toolCallsConcurrency || 1;
         this.keepTextInToolUses = options.keepTextInToolUses;
         this.memoryAgentsAsTools = options.memoryAgentsAsTools;
         this.memoryPromptTemplate = options.memoryPromptTemplate;
@@ -170,6 +174,12 @@ export class AIAgent extends Agent {
      * {@includeCode ../../test/agents/ai-agent.test.ts#example-ai-agent-router}
      */
     toolChoice;
+    /**
+     * Maximum number of tool calls to execute concurrently
+     *
+     * @default 1
+     */
+    toolCallsConcurrency;
     /**
      * Whether to preserve text generated during tool usage in the final output
      */
@@ -227,6 +237,20 @@ export class AIAgent extends Agent {
      * which outputs structured data in YAML format within <metadata> tags.
      */
     customStructuredStreamInstructions;
+    get inputSchema() {
+        let schema = super.inputSchema;
+        if (this.inputKey) {
+            schema = schema.extend({
+                [this.inputKey]: z.string().nullish(),
+            });
+        }
+        if (this.inputFileKey) {
+            schema = schema.extend({
+                [this.inputFileKey]: fileUnionContentsSchema.nullish(),
+            });
+        }
+        return schema;
+    }
     /**
      * Process an input message and generate a response
      *
@@ -284,31 +308,43 @@ export class AIAgent extends Agent {
                     yield { delta: { text: { [outputKey]: "\n" } } };
                 }
                 const executedToolCalls = [];
+                let error;
+                const queue = fastq.promise(async ({ tool, call }) => {
+                    try {
+                        // NOTE: should pass both arguments (model generated) and input (user provided) to the tool
+                        const output = await this.invokeSkill(tool, { ...input, ...call.function.arguments }, options).catch((error) => {
+                            if (!this.catchToolsError) {
+                                return Promise.reject(error);
+                            }
+                            return {
+                                isError: true,
+                                error: {
+                                    message: error.message,
+                                },
+                            };
+                        });
+                        executedToolCalls.push({ call, output });
+                    }
+                    catch (e) {
+                        error = e;
+                        queue.killAndDrain();
+                    }
+                }, this.toolCallsConcurrency || 1);
                 // Execute tools
                 for (const call of toolCalls) {
                     const tool = toolsMap.get(call.function.name);
                     if (!tool)
                         throw new Error(`Tool not found: ${call.function.name}`);
-                    // NOTE: should pass both arguments (model generated) and input (user provided) to the tool
-                    const output = await this.invokeSkill(tool, { ...input, ...call.function.arguments }, options).catch((error) => {
-                        if (!this.catchToolsError) {
-                            return Promise.reject(error);
-                        }
-                        return {
-                            isError: true,
-                            error: {
-                                message: error.message,
-                            },
-                        };
-                    });
-                    // NOTE: Return transfer output immediately
-                    if (isTransferAgentOutput(output)) {
-                        return output;
-                    }
-                    executedToolCalls.push({ call, output });
+                    queue.push({ tool, call });
                 }
+                await queue.drained();
+                if (error)
+                    throw error;
                 // Continue LLM function calling loop if any tools were executed
                 if (executedToolCalls.length) {
+                    const transferOutput = executedToolCalls.find((i) => isTransferAgentOutput(i.output))?.output;
+                    if (transferOutput)
+                        return transferOutput;
                     toolCallMessages.push(await AgentMessageTemplate.from(undefined, executedToolCalls.map(({ call }) => call)).format(), ...(await Promise.all(executedToolCalls.map(({ call, output }) => ToolMessageTemplate.from(output, call.id).format()))));
                     continue;
                 }

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

@@ -46,8 +46,10 @@ export interface AIAgentSchema extends BaseAgentSchema {
     type: "ai";
     instructions?: Instructions;
     inputKey?: string;
+    inputFileKey?: string;
     outputKey?: string;
     toolChoice?: AIAgentToolChoice;
+    toolCallsConcurrency?: number;
     keepTextInToolUses?: boolean;
 }
 export interface ImageAgentSchema extends BaseAgentSchema {

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

@@ -91,7 +91,10 @@ export async function parseAgentFile(path, data) {
                 instructions: optionalize(instructionsSchema),
                 inputKey: optionalize(z.string()),
                 outputKey: optionalize(z.string()),
+                inputFileKey: optionalize(z.string()),
+                outputFileKey: optionalize(z.string()),
                 toolChoice: optionalize(z.nativeEnum(AIAgentToolChoice)),
+                toolCallsConcurrency: optionalize(z.number().int().min(0)),
                 keepTextInToolUses: optionalize(z.boolean()),
                 structuredStreamMode: optionalize(z.boolean()),
             })

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@aigne/core",
-  "version": "1.63.0-beta",
+  "version": "1.63.0-beta.2",
   "description": "The functional core of agentic AI",
   "publishConfig": {
     "access": "public"
@@ -92,9 +92,9 @@
     "zod": "^3.25.67",
     "zod-from-json-schema": "^0.0.5",
     "zod-to-json-schema": "^3.24.6",
-    "@aigne/observability-api": "^0.11.2-beta",
+    "@aigne/afs": "^1.0.0",
     "@aigne/platform-helpers": "^0.6.3",
-    "@aigne/afs": "^1.0.0"
+    "@aigne/observability-api": "^0.11.2-beta"
   },
   "devDependencies": {
     "@types/bun": "^1.2.22",