@aigne/core 1.44.0 → 1.46.0

package/CHANGELOG.md

@@ -12,6 +12,45 @@
   * dependencies
     * @aigne/observability bumped to 0.1.0
 
+## [1.46.0](https://github.com/AIGNE-io/aigne-framework/compare/core-v1.45.0...core-v1.46.0) (2025-08-06)
+
+
+### Features
+
+* **cli:** support custom task title of agent in cli ([#328](https://github.com/AIGNE-io/aigne-framework/issues/328)) ([128d75f](https://github.com/AIGNE-io/aigne-framework/commit/128d75fb42ca470b47a2793d79c92d7bb64cfedb))
+
+
+### Bug Fixes
+
+* **core:** improve hook handling in agent and context ([#325](https://github.com/AIGNE-io/aigne-framework/issues/325)) ([c858fec](https://github.com/AIGNE-io/aigne-framework/commit/c858fecb08453c2daca9708f4b8a9c135fac40b0))
+
+
+### Dependencies
+
+* The following workspace dependencies were updated
+  * dependencies
+    * @aigne/platform-helpers bumped to 0.6.0
+
+## [1.45.0](https://github.com/AIGNE-io/aigne-framework/compare/core-v1.44.0...core-v1.45.0) (2025-08-06)
+
+
+### Features
+
+* **core:** add concurrency support for team agent ([#323](https://github.com/AIGNE-io/aigne-framework/issues/323)) ([5743260](https://github.com/AIGNE-io/aigne-framework/commit/57432603a45208ad3503b9fc4c64f07c8151f9ee))
+
+
+### Bug Fixes
+
+* **core:** remove lodash dependency ensure core support both esm and cjs ([#324](https://github.com/AIGNE-io/aigne-framework/issues/324)) ([d6c2452](https://github.com/AIGNE-io/aigne-framework/commit/d6c2452b660a163c73f2c628ffdc2a12949360b0))
+* **models:** aigne-hub adapter not working in node.js v21 ([#320](https://github.com/AIGNE-io/aigne-framework/issues/320)) ([2884d00](https://github.com/AIGNE-io/aigne-framework/commit/2884d00b83e153ae7465ef1369fcd22d7c6d43e0))
+
+
+### Dependencies
+
+* The following workspace dependencies were updated
+  * dependencies
+    * @aigne/platform-helpers bumped to 0.5.1
+
 ## [1.44.0](https://github.com/AIGNE-io/aigne-framework/compare/core-v1.43.1...core-v1.44.0) (2025-08-05)
 
 

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

@@ -72,6 +72,7 @@ export interface AgentOptions<I extends Message = Message, O extends Message = M
      * for documentation and debugging
      */
     description?: string;
+    taskTitle?: string;
     /**
      * Zod schema defining the input message structure
      *
@@ -256,6 +257,7 @@ export declare abstract class Agent<I extends Message = any, O extends Message =
      * each other's roles in a multi-agent system
      */
     readonly description?: string;
+    taskTitle?: string;
     private readonly _inputSchema?;
     defaultInput?: Partial<{
         [key in keyof I]: {

package/lib/cjs/agents/agent.js

@@ -104,6 +104,7 @@ class Agent {
         this.name = options.name || this.constructor.name;
         this.alias = options.alias;
         this.description = options.description;
+        this.taskTitle = options.taskTitle;
         if (inputSchema)
             checkAgentInputOutputSchema(inputSchema);
         if (outputSchema)
@@ -193,6 +194,7 @@ class Agent {
      * each other's roles in a multi-agent system
      */
     description;
+    taskTitle;
     _inputSchema;
     defaultInput;
     _outputSchema;
@@ -388,7 +390,7 @@ class Agent {
     async callHooks(hook, input, options) {
         const { context } = options;
         const result = {};
-        for (const hooks of (0, type_utils_js_1.flat)(options.hooks, this.hooks)) {
+        for (const hooks of (0, type_utils_js_1.flat)(options.hooks, options.context.hooks, this.hooks)) {
             const h = hooks[hook];
             if (!h)
                 continue;

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

@@ -1,7 +1,37 @@
 "use strict";
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.MCPResource = exports.MCPPrompt = exports.MCPTool = exports.MCPBase = exports.MCPAgent = void 0;
 const stdio_client_transport_js_1 = require("@aigne/platform-helpers/mcp/stdio-client-transport.js");
@@ -10,7 +40,6 @@ const index_js_2 = require("@modelcontextprotocol/sdk/client/index.js");
 const sse_js_1 = require("@modelcontextprotocol/sdk/client/sse.js");
 const streamableHttp_js_1 = require("@modelcontextprotocol/sdk/client/streamableHttp.js");
 const uriTemplate_js_1 = require("@modelcontextprotocol/sdk/shared/uriTemplate.js");
-const p_retry_1 = __importDefault(require("p-retry"));
 const zod_1 = require("zod");
 const logger_js_1 = require("../utils/logger.js");
 const mcp_utils_js_1 = require("../utils/mcp-utils.js");
@@ -222,7 +251,8 @@ class ClientWithReconnect extends index_js_2.Client {
         const transportCreator = this.reconnectOptions?.transportCreator;
         if (!transportCreator)
             throw new Error("reconnect requires a transportCreator");
-        await (0, p_retry_1.default)(async () => {
+        const retry = await Promise.resolve().then(() => __importStar(require("p-retry")));
+        await retry.default(async () => {
             await this.close();
             await this.connect(await transportCreator(), {
                 timeout: this.reconnectOptions?.timeout ?? DEFAULT_TIMEOUT(),

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

@@ -137,6 +137,22 @@ export interface TeamAgentOptions<I extends Message, O extends Message> extends
      * - The processing results are streamed incrementally as each iteration completes
      */
     iterateOn?: keyof I;
+    /**
+     * The maximum number of concurrent operations when processing array items with `iterateOn`.
+     *
+     * This property controls how many array elements are processed simultaneously when
+     * iterating over an array field. A higher concurrency value allows for faster
+     * parallel processing but may consume more resources.
+     *
+     * @remarks
+     * - Only applies when `iterateOn` is specified
+     * - Cannot be used together with `iterateWithPreviousOutput` (concurrency > 1)
+     * - Uses a queue-based approach to limit concurrent operations
+     * - Each concurrent operation processes one array element through the entire agent workflow
+     *
+     * @default 1
+     */
+    concurrency?: number;
     /**
      * Controls whether to merge the output from each iteration back into the array items
      * for subsequent iterations when using `iterateOn`.
@@ -228,6 +244,16 @@ export declare class TeamAgent<I extends Message, O extends Message> extends Age
      * @see TeamAgentOptions.iterateOn for detailed documentation
      */
     iterateOn?: keyof I;
+    /**
+     * The maximum number of concurrent operations when processing array items.
+     *
+     * This property controls the concurrency level for iterative processing when `iterateOn`
+     * is used. It determines how many array elements are processed simultaneously.
+     *
+     * @see TeamAgentOptions.concurrency for detailed documentation
+     * @default 1
+     */
+    concurrency: number;
     /**
      * Controls whether to merge the output from each iteration back into the array items
      * for subsequent iterations when using `iterateOn`.

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

@@ -5,6 +5,7 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.TeamAgent = exports.DEFAULT_REFLECTION_MAX_ITERATIONS = exports.ProcessMode = void 0;
 const node_assert_1 = __importDefault(require("node:assert"));
+const fastq_1 = __importDefault(require("fastq"));
 const immer_1 = require("immer");
 const stream_utils_js_1 = require("../utils/stream-utils.js");
 const type_utils_js_1 = require("../utils/type-utils.js");
@@ -82,8 +83,12 @@ class TeamAgent extends agent_js_1.Agent {
             maxIterations: options.reflection.maxIterations ?? exports.DEFAULT_REFLECTION_MAX_ITERATIONS,
         };
         this.iterateOn = options.iterateOn;
+        this.concurrency = options.concurrency ?? 1;
         this.iterateWithPreviousOutput = options.iterateWithPreviousOutput;
         this.includeAllStepsOutput = options.includeAllStepsOutput;
+        if (this.concurrency !== 1 && this.iterateWithPreviousOutput) {
+            throw new Error(`iterateWithPreviousOutput cannot be used with concurrency > 1, concurrency: ${this.concurrency}`);
+        }
     }
     /**
      * The processing mode that determines how agents in the team are executed.
@@ -109,6 +114,16 @@ class TeamAgent extends agent_js_1.Agent {
      * @see TeamAgentOptions.iterateOn for detailed documentation
      */
     iterateOn;
+    /**
+     * The maximum number of concurrent operations when processing array items.
+     *
+     * This property controls the concurrency level for iterative processing when `iterateOn`
+     * is used. It determines how many array elements are processed simultaneously.
+     *
+     * @see TeamAgentOptions.concurrency for detailed documentation
+     * @default 1
+     */
+    concurrency;
     /**
      * Controls whether to merge the output from each iteration back into the array items
      * for subsequent iterations when using `iterateOn`.
@@ -180,23 +195,27 @@ class TeamAgent extends agent_js_1.Agent {
         (0, node_assert_1.default)(this.iterateOn, "iterateInputKey must be defined for iterator processing");
         let arr = input[this.iterateOn];
         arr = Array.isArray(arr) ? [...arr] : (0, type_utils_js_1.isNil)(arr) ? [arr] : [];
-        const result = [];
-        for (let i = 0; i < arr.length; i++) {
-            const item = arr[i];
+        const results = new Array(arr.length);
+        const queue = fastq_1.default.promise(async ({ item, index }) => {
             if (!(0, type_utils_js_1.isRecord)(item))
                 throw new TypeError(`Expected ${String(key)} to be an object, got ${typeof item}`);
-            const res = await (0, agent_js_1.agentProcessResultToObject)(await this._processNonIterator({ ...input, [key]: arr, ...item }, { ...options, streaming: false }));
+            const o = await (0, agent_js_1.agentProcessResultToObject)(await this._processNonIterator({ ...input, [key]: arr, ...item }, { ...options, streaming: false }));
+            const res = (0, type_utils_js_1.omit)(o, key);
             // Merge the item result with the original item used for next iteration
             if (this.iterateWithPreviousOutput) {
                 arr = (0, immer_1.produce)(arr, (draft) => {
-                    const item = draft[i];
+                    const item = draft[index];
                     (0, node_assert_1.default)(item);
                     Object.assign(item, res);
                 });
             }
-            result.push((0, type_utils_js_1.omit)(res, key));
-            yield { delta: { json: { [key]: result } } };
+            results[index] = res;
+        }, this.concurrency);
+        for (let index = 0; index < arr.length; index++) {
+            queue.push({ index, item: arr[index] });
         }
+        await queue.drained();
+        yield { delta: { json: { [key]: results } } };
     }
     _processNonIterator(input, options) {
         switch (this.mode) {

package/lib/cjs/aigne/context.d.ts

@@ -1,7 +1,7 @@
 import type { AIGNEObserver } from "@aigne/observability-api";
 import type { Span } from "@opentelemetry/api";
 import { Emitter } from "strict-event-emitter";
-import { type Agent, type AgentInvokeOptions, type AgentProcessAsyncGenerator, type AgentResponse, type AgentResponseStream, type Message } from "../agents/agent.js";
+import { type Agent, type AgentHooks, type AgentInvokeOptions, type AgentProcessAsyncGenerator, type AgentResponse, type AgentResponseStream, type Message } from "../agents/agent.js";
 import type { ChatModel } from "../agents/chat-model.js";
 import { UserAgent } from "../agents/user-agent.js";
 import type { Memory } from "../memory/memory.js";
@@ -80,6 +80,7 @@ export interface Context<U extends UserContext = UserContext> extends TypedEvent
     limits?: ContextLimits;
     status?: "normal" | "timeout";
     userContext: U;
+    hooks?: AgentHooks[];
     memories: Pick<Memory, "content">[];
     /**
      * Create a user agent to consistently invoke an agent
@@ -164,11 +165,14 @@ export declare class AIGNEContext implements Context {
     set userContext(userContext: Context["userContext"]);
     get memories(): Context["memories"];
     set memories(memories: Context["memories"]);
+    get hooks(): AgentHooks[];
+    set hooks(hooks: AgentHooks[]);
     newContext({ reset }?: {
         reset?: boolean;
     }): AIGNEContext;
     invoke: Context["invoke"];
     private onInvocationResult;
+    private processOptions;
     publish: Context["publish"];
     subscribe: Context["subscribe"];
     unsubscribe: Context["unsubscribe"];
@@ -197,6 +201,7 @@ declare class AIGNEContextShared {
     usage: ContextUsage;
     userContext: Context["userContext"];
     memories: Context["memories"];
+    hooks: AgentHooks[];
     private abortController;
     private timer?;
     private initTimeout;

package/lib/cjs/aigne/context.js

@@ -94,6 +94,12 @@ class AIGNEContext {
     set memories(memories) {
         this.internal.memories = memories;
     }
+    get hooks() {
+        return this.internal.hooks;
+    }
+    set hooks(hooks) {
+        this.internal.hooks = hooks;
+    }
     newContext({ reset } = {}) {
         return new AIGNEContext(this, { reset });
     }
@@ -103,14 +109,7 @@ class AIGNEContext {
             message,
             options,
         });
-        if (options?.userContext) {
-            Object.assign(this.userContext, options.userContext);
-            options.userContext = undefined;
-        }
-        if (options?.memories?.length) {
-            this.memories.push(...options.memories);
-            options.memories = undefined;
-        }
+        this.processOptions(options);
         if ((0, type_utils_js_1.isNil)(message)) {
             return user_agent_js_1.UserAgent.from({
                 context: this,
@@ -176,7 +175,7 @@ class AIGNEContext {
             },
         };
     }
-    publish = ((topic, payload, options) => {
+    processOptions(options) {
         if (options?.userContext) {
             Object.assign(this.userContext, options.userContext);
             options.userContext = undefined;
@@ -185,6 +184,13 @@ class AIGNEContext {
             this.memories.push(...options.memories);
             options.memories = undefined;
         }
+        if (options?.hooks) {
+            this.hooks.push(...(0, type_utils_js_1.flat)(options.hooks));
+            options.hooks = undefined;
+        }
+    }
+    publish = ((topic, payload, options) => {
+        this.processOptions(options);
         const newContext = options?.newContext === false ? this : this.newContext();
         return this.internal.messageQueue.publish(topic, {
             ...(0, message_queue_js_1.toMessagePayload)(payload),
@@ -318,6 +324,7 @@ class AIGNEContextShared {
     usage = (0, usage_js_1.newEmptyContextUsage)();
     userContext = {};
     memories = [];
+    hooks = [];
     abortController = new AbortController();
     timer;
     initTimeout() {

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

@@ -19,6 +19,7 @@ export type NestAgentSchema = string | {
 export interface BaseAgentSchema {
     name?: string;
     description?: string;
+    taskTitle?: string;
     inputSchema?: ZodType<Record<string, any>>;
     defaultInput?: Record<string, any>;
     outputSchema?: ZodType<Record<string, any>>;
@@ -46,6 +47,9 @@ export interface TeamAgentSchema extends BaseAgentSchema {
     type: "team";
     mode?: ProcessMode;
     iterateOn?: string;
+    concurrency?: number;
+    iterateWithPreviousOutput?: boolean;
+    includeAllStepsOutput?: boolean;
     reflection?: Omit<ReflectionMode, "reviewer"> & {
         reviewer: NestAgentSchema;
     };

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

@@ -34,6 +34,7 @@ async function parseAgentFile(path, data) {
             name: (0, schema_js_1.optionalize)(zod_1.z.string()),
             alias: (0, schema_js_1.optionalize)(zod_1.z.array(zod_1.z.string())),
             description: (0, schema_js_1.optionalize)(zod_1.z.string()),
+            taskTitle: (0, schema_js_1.optionalize)(zod_1.z.string()),
             inputSchema: (0, schema_js_1.optionalize)((0, schema_js_1.inputOutputSchema)({ path })).transform((v) => v ? (0, json_schema_to_zod_1.jsonSchemaToZod)(v) : undefined),
             defaultInput: (0, schema_js_1.optionalize)(schema_js_1.defaultInputSchema),
             outputSchema: (0, schema_js_1.optionalize)((0, schema_js_1.inputOutputSchema)({ path })).transform((v) => v ? (0, json_schema_to_zod_1.jsonSchemaToZod)(v) : undefined),
@@ -78,6 +79,9 @@ async function parseAgentFile(path, data) {
                 type: zod_1.z.literal("team"),
                 mode: (0, schema_js_1.optionalize)(zod_1.z.nativeEnum(team_agent_js_1.ProcessMode)),
                 iterateOn: (0, schema_js_1.optionalize)(zod_1.z.string()),
+                concurrency: (0, schema_js_1.optionalize)(zod_1.z.number().int().min(1)),
+                iterateWithPreviousOutput: (0, schema_js_1.optionalize)(zod_1.z.boolean()),
+                includeAllStepsOutput: (0, schema_js_1.optionalize)(zod_1.z.boolean()),
                 reflection: (0, schema_js_1.camelizeSchema)((0, schema_js_1.optionalize)(zod_1.z.object({
                     reviewer: nestAgentSchema,
                     isApproved: zod_1.z.string(),

package/lib/cjs/loader/schema.js

@@ -1,16 +1,13 @@
 "use strict";
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.defaultInputSchema = exports.inputOutputSchema = void 0;
 exports.optionalize = optionalize;
 exports.camelizeSchema = camelizeSchema;
 const index_js_1 = require("@aigne/platform-helpers/nodejs/index.js");
-const camelize_ts_1 = __importDefault(require("camelize-ts"));
 const yaml_1 = require("yaml");
 const zod_1 = require("zod");
 const agent_js_1 = require("../agents/agent.js");
+const camelize_js_1 = require("../utils/camelize.js");
 const type_utils_js_1 = require("../utils/type-utils.js");
 const inputOutputSchema = ({ path }) => {
     const includeExternalSchema = async (schema) => {
@@ -67,5 +64,5 @@ function optionalize(schema) {
     return schema.nullish().transform((v) => v ?? undefined);
 }
 function camelizeSchema(schema, { shallow = true } = {}) {
-    return zod_1.z.preprocess((v) => ((0, type_utils_js_1.isRecord)(v) ? (0, camelize_ts_1.default)(v, shallow) : v), schema);
+    return zod_1.z.preprocess((v) => ((0, type_utils_js_1.isRecord)(v) ? (0, camelize_js_1.camelize)(v, shallow) : v), schema);
 }

package/lib/cjs/utils/camelize.d.ts

@@ -0,0 +1 @@
+export declare function camelize<T>(obj: T, shallow?: boolean): any;

package/lib/cjs/utils/camelize.js

@@ -0,0 +1,21 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.camelize = camelize;
+const type_utils_js_1 = require("./type-utils.js");
+function camelize(obj, shallow = false) {
+    if (Array.isArray(obj)) {
+        return shallow ? obj : obj.map((item) => camelize(item, false));
+    }
+    if ((0, type_utils_js_1.isRecord)(obj)) {
+        return Object.fromEntries(Object.entries(obj).map(([key, value]) => [
+            camelCase(key),
+            shallow ? value : camelize(value, false),
+        ]));
+    }
+    return obj;
+}
+function camelCase(key) {
+    key = key.replace(/[-_#@$\s]+(.)?/g, (_, char) => char.toUpperCase());
+    key = key.charAt(0).toLowerCase() + key.slice(1);
+    return key;
+}

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

@@ -72,6 +72,7 @@ export interface AgentOptions<I extends Message = Message, O extends Message = M
      * for documentation and debugging
      */
     description?: string;
+    taskTitle?: string;
     /**
      * Zod schema defining the input message structure
      *
@@ -256,6 +257,7 @@ export declare abstract class Agent<I extends Message = any, O extends Message =
      * each other's roles in a multi-agent system
      */
     readonly description?: string;
+    taskTitle?: string;
     private readonly _inputSchema?;
     defaultInput?: Partial<{
         [key in keyof I]: {

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

@@ -137,6 +137,22 @@ export interface TeamAgentOptions<I extends Message, O extends Message> extends
      * - The processing results are streamed incrementally as each iteration completes
      */
     iterateOn?: keyof I;
+    /**
+     * The maximum number of concurrent operations when processing array items with `iterateOn`.
+     *
+     * This property controls how many array elements are processed simultaneously when
+     * iterating over an array field. A higher concurrency value allows for faster
+     * parallel processing but may consume more resources.
+     *
+     * @remarks
+     * - Only applies when `iterateOn` is specified
+     * - Cannot be used together with `iterateWithPreviousOutput` (concurrency > 1)
+     * - Uses a queue-based approach to limit concurrent operations
+     * - Each concurrent operation processes one array element through the entire agent workflow
+     *
+     * @default 1
+     */
+    concurrency?: number;
     /**
      * Controls whether to merge the output from each iteration back into the array items
      * for subsequent iterations when using `iterateOn`.
@@ -228,6 +244,16 @@ export declare class TeamAgent<I extends Message, O extends Message> extends Age
      * @see TeamAgentOptions.iterateOn for detailed documentation
      */
     iterateOn?: keyof I;
+    /**
+     * The maximum number of concurrent operations when processing array items.
+     *
+     * This property controls the concurrency level for iterative processing when `iterateOn`
+     * is used. It determines how many array elements are processed simultaneously.
+     *
+     * @see TeamAgentOptions.concurrency for detailed documentation
+     * @default 1
+     */
+    concurrency: number;
     /**
      * Controls whether to merge the output from each iteration back into the array items
      * for subsequent iterations when using `iterateOn`.

package/lib/dts/aigne/context.d.ts

@@ -1,7 +1,7 @@
 import type { AIGNEObserver } from "@aigne/observability-api";
 import type { Span } from "@opentelemetry/api";
 import { Emitter } from "strict-event-emitter";
-import { type Agent, type AgentInvokeOptions, type AgentProcessAsyncGenerator, type AgentResponse, type AgentResponseStream, type Message } from "../agents/agent.js";
+import { type Agent, type AgentHooks, type AgentInvokeOptions, type AgentProcessAsyncGenerator, type AgentResponse, type AgentResponseStream, type Message } from "../agents/agent.js";
 import type { ChatModel } from "../agents/chat-model.js";
 import { UserAgent } from "../agents/user-agent.js";
 import type { Memory } from "../memory/memory.js";
@@ -80,6 +80,7 @@ export interface Context<U extends UserContext = UserContext> extends TypedEvent
     limits?: ContextLimits;
     status?: "normal" | "timeout";
     userContext: U;
+    hooks?: AgentHooks[];
     memories: Pick<Memory, "content">[];
     /**
      * Create a user agent to consistently invoke an agent
@@ -164,11 +165,14 @@ export declare class AIGNEContext implements Context {
     set userContext(userContext: Context["userContext"]);
     get memories(): Context["memories"];
     set memories(memories: Context["memories"]);
+    get hooks(): AgentHooks[];
+    set hooks(hooks: AgentHooks[]);
     newContext({ reset }?: {
         reset?: boolean;
     }): AIGNEContext;
     invoke: Context["invoke"];
     private onInvocationResult;
+    private processOptions;
     publish: Context["publish"];
     subscribe: Context["subscribe"];
     unsubscribe: Context["unsubscribe"];
@@ -197,6 +201,7 @@ declare class AIGNEContextShared {
     usage: ContextUsage;
     userContext: Context["userContext"];
     memories: Context["memories"];
+    hooks: AgentHooks[];
     private abortController;
     private timer?;
     private initTimeout;

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

@@ -19,6 +19,7 @@ export type NestAgentSchema = string | {
 export interface BaseAgentSchema {
     name?: string;
     description?: string;
+    taskTitle?: string;
     inputSchema?: ZodType<Record<string, any>>;
     defaultInput?: Record<string, any>;
     outputSchema?: ZodType<Record<string, any>>;
@@ -46,6 +47,9 @@ export interface TeamAgentSchema extends BaseAgentSchema {
     type: "team";
     mode?: ProcessMode;
     iterateOn?: string;
+    concurrency?: number;
+    iterateWithPreviousOutput?: boolean;
+    includeAllStepsOutput?: boolean;
     reflection?: Omit<ReflectionMode, "reviewer"> & {
         reviewer: NestAgentSchema;
     };

package/lib/dts/utils/camelize.d.ts

@@ -0,0 +1 @@
+export declare function camelize<T>(obj: T, shallow?: boolean): any;

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

@@ -72,6 +72,7 @@ export interface AgentOptions<I extends Message = Message, O extends Message = M
      * for documentation and debugging
      */
     description?: string;
+    taskTitle?: string;
     /**
      * Zod schema defining the input message structure
      *
@@ -256,6 +257,7 @@ export declare abstract class Agent<I extends Message = any, O extends Message =
      * each other's roles in a multi-agent system
      */
     readonly description?: string;
+    taskTitle?: string;
     private readonly _inputSchema?;
     defaultInput?: Partial<{
         [key in keyof I]: {

package/lib/esm/agents/agent.js

@@ -59,6 +59,7 @@ export class Agent {
         this.name = options.name || this.constructor.name;
         this.alias = options.alias;
         this.description = options.description;
+        this.taskTitle = options.taskTitle;
         if (inputSchema)
             checkAgentInputOutputSchema(inputSchema);
         if (outputSchema)
@@ -148,6 +149,7 @@ export class Agent {
      * each other's roles in a multi-agent system
      */
     description;
+    taskTitle;
     _inputSchema;
     defaultInput;
     _outputSchema;
@@ -343,7 +345,7 @@ export class Agent {
     async callHooks(hook, input, options) {
         const { context } = options;
         const result = {};
-        for (const hooks of flat(options.hooks, this.hooks)) {
+        for (const hooks of flat(options.hooks, options.context.hooks, this.hooks)) {
             const h = hooks[hook];
             if (!h)
                 continue;

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

@@ -4,7 +4,6 @@ import { Client } from "@modelcontextprotocol/sdk/client/index.js";
 import { SSEClientTransport, } from "@modelcontextprotocol/sdk/client/sse.js";
 import { StreamableHTTPClientTransport, } from "@modelcontextprotocol/sdk/client/streamableHttp.js";
 import { UriTemplate } from "@modelcontextprotocol/sdk/shared/uriTemplate.js";
-import pRetry from "p-retry";
 import { z } from "zod";
 import { logger } from "../utils/logger.js";
 import { promptFromMCPPrompt, resourceFromMCPResource, toolFromMCPTool, } from "../utils/mcp-utils.js";
@@ -215,7 +214,8 @@ class ClientWithReconnect extends Client {
         const transportCreator = this.reconnectOptions?.transportCreator;
         if (!transportCreator)
             throw new Error("reconnect requires a transportCreator");
-        await pRetry(async () => {
+        const retry = await import("p-retry");
+        await retry.default(async () => {
             await this.close();
             await this.connect(await transportCreator(), {
                 timeout: this.reconnectOptions?.timeout ?? DEFAULT_TIMEOUT(),

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

@@ -137,6 +137,22 @@ export interface TeamAgentOptions<I extends Message, O extends Message> extends
      * - The processing results are streamed incrementally as each iteration completes
      */
     iterateOn?: keyof I;
+    /**
+     * The maximum number of concurrent operations when processing array items with `iterateOn`.
+     *
+     * This property controls how many array elements are processed simultaneously when
+     * iterating over an array field. A higher concurrency value allows for faster
+     * parallel processing but may consume more resources.
+     *
+     * @remarks
+     * - Only applies when `iterateOn` is specified
+     * - Cannot be used together with `iterateWithPreviousOutput` (concurrency > 1)
+     * - Uses a queue-based approach to limit concurrent operations
+     * - Each concurrent operation processes one array element through the entire agent workflow
+     *
+     * @default 1
+     */
+    concurrency?: number;
     /**
      * Controls whether to merge the output from each iteration back into the array items
      * for subsequent iterations when using `iterateOn`.
@@ -228,6 +244,16 @@ export declare class TeamAgent<I extends Message, O extends Message> extends Age
      * @see TeamAgentOptions.iterateOn for detailed documentation
      */
     iterateOn?: keyof I;
+    /**
+     * The maximum number of concurrent operations when processing array items.
+     *
+     * This property controls the concurrency level for iterative processing when `iterateOn`
+     * is used. It determines how many array elements are processed simultaneously.
+     *
+     * @see TeamAgentOptions.concurrency for detailed documentation
+     * @default 1
+     */
+    concurrency: number;
     /**
      * Controls whether to merge the output from each iteration back into the array items
      * for subsequent iterations when using `iterateOn`.

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

@@ -1,4 +1,5 @@
 import assert from "node:assert";
+import fastq from "fastq";
 import { produce } from "immer";
 import { mergeAgentResponseChunk } from "../utils/stream-utils.js";
 import { isEmpty, isNil, isRecord, omit } from "../utils/type-utils.js";
@@ -76,8 +77,12 @@ export class TeamAgent extends Agent {
             maxIterations: options.reflection.maxIterations ?? DEFAULT_REFLECTION_MAX_ITERATIONS,
         };
         this.iterateOn = options.iterateOn;
+        this.concurrency = options.concurrency ?? 1;
         this.iterateWithPreviousOutput = options.iterateWithPreviousOutput;
         this.includeAllStepsOutput = options.includeAllStepsOutput;
+        if (this.concurrency !== 1 && this.iterateWithPreviousOutput) {
+            throw new Error(`iterateWithPreviousOutput cannot be used with concurrency > 1, concurrency: ${this.concurrency}`);
+        }
     }
     /**
      * The processing mode that determines how agents in the team are executed.
@@ -103,6 +108,16 @@ export class TeamAgent extends Agent {
      * @see TeamAgentOptions.iterateOn for detailed documentation
      */
     iterateOn;
+    /**
+     * The maximum number of concurrent operations when processing array items.
+     *
+     * This property controls the concurrency level for iterative processing when `iterateOn`
+     * is used. It determines how many array elements are processed simultaneously.
+     *
+     * @see TeamAgentOptions.concurrency for detailed documentation
+     * @default 1
+     */
+    concurrency;
     /**
      * Controls whether to merge the output from each iteration back into the array items
      * for subsequent iterations when using `iterateOn`.
@@ -174,23 +189,27 @@ export class TeamAgent extends Agent {
         assert(this.iterateOn, "iterateInputKey must be defined for iterator processing");
         let arr = input[this.iterateOn];
         arr = Array.isArray(arr) ? [...arr] : isNil(arr) ? [arr] : [];
-        const result = [];
-        for (let i = 0; i < arr.length; i++) {
-            const item = arr[i];
+        const results = new Array(arr.length);
+        const queue = fastq.promise(async ({ item, index }) => {
             if (!isRecord(item))
                 throw new TypeError(`Expected ${String(key)} to be an object, got ${typeof item}`);
-            const res = await agentProcessResultToObject(await this._processNonIterator({ ...input, [key]: arr, ...item }, { ...options, streaming: false }));
+            const o = await agentProcessResultToObject(await this._processNonIterator({ ...input, [key]: arr, ...item }, { ...options, streaming: false }));
+            const res = omit(o, key);
             // Merge the item result with the original item used for next iteration
             if (this.iterateWithPreviousOutput) {
                 arr = produce(arr, (draft) => {
-                    const item = draft[i];
+                    const item = draft[index];
                     assert(item);
                     Object.assign(item, res);
                 });
             }
-            result.push(omit(res, key));
-            yield { delta: { json: { [key]: result } } };
+            results[index] = res;
+        }, this.concurrency);
+        for (let index = 0; index < arr.length; index++) {
+            queue.push({ index, item: arr[index] });
         }
+        await queue.drained();
+        yield { delta: { json: { [key]: results } } };
     }
     _processNonIterator(input, options) {
         switch (this.mode) {

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

@@ -1,7 +1,7 @@
 import type { AIGNEObserver } from "@aigne/observability-api";
 import type { Span } from "@opentelemetry/api";
 import { Emitter } from "strict-event-emitter";
-import { type Agent, type AgentInvokeOptions, type AgentProcessAsyncGenerator, type AgentResponse, type AgentResponseStream, type Message } from "../agents/agent.js";
+import { type Agent, type AgentHooks, type AgentInvokeOptions, type AgentProcessAsyncGenerator, type AgentResponse, type AgentResponseStream, type Message } from "../agents/agent.js";
 import type { ChatModel } from "../agents/chat-model.js";
 import { UserAgent } from "../agents/user-agent.js";
 import type { Memory } from "../memory/memory.js";
@@ -80,6 +80,7 @@ export interface Context<U extends UserContext = UserContext> extends TypedEvent
     limits?: ContextLimits;
     status?: "normal" | "timeout";
     userContext: U;
+    hooks?: AgentHooks[];
     memories: Pick<Memory, "content">[];
     /**
      * Create a user agent to consistently invoke an agent
@@ -164,11 +165,14 @@ export declare class AIGNEContext implements Context {
     set userContext(userContext: Context["userContext"]);
     get memories(): Context["memories"];
     set memories(memories: Context["memories"]);
+    get hooks(): AgentHooks[];
+    set hooks(hooks: AgentHooks[]);
     newContext({ reset }?: {
         reset?: boolean;
     }): AIGNEContext;
     invoke: Context["invoke"];
     private onInvocationResult;
+    private processOptions;
     publish: Context["publish"];
     subscribe: Context["subscribe"];
     unsubscribe: Context["unsubscribe"];
@@ -197,6 +201,7 @@ declare class AIGNEContextShared {
     usage: ContextUsage;
     userContext: Context["userContext"];
     memories: Context["memories"];
+    hooks: AgentHooks[];
     private abortController;
     private timer?;
     private initTimeout;

package/lib/esm/aigne/context.js

@@ -88,6 +88,12 @@ export class AIGNEContext {
     set memories(memories) {
         this.internal.memories = memories;
     }
+    get hooks() {
+        return this.internal.hooks;
+    }
+    set hooks(hooks) {
+        this.internal.hooks = hooks;
+    }
     newContext({ reset } = {}) {
         return new AIGNEContext(this, { reset });
     }
@@ -97,14 +103,7 @@ export class AIGNEContext {
             message,
             options,
         });
-        if (options?.userContext) {
-            Object.assign(this.userContext, options.userContext);
-            options.userContext = undefined;
-        }
-        if (options?.memories?.length) {
-            this.memories.push(...options.memories);
-            options.memories = undefined;
-        }
+        this.processOptions(options);
         if (isNil(message)) {
             return UserAgent.from({
                 context: this,
@@ -170,7 +169,7 @@ export class AIGNEContext {
             },
         };
     }
-    publish = ((topic, payload, options) => {
+    processOptions(options) {
         if (options?.userContext) {
             Object.assign(this.userContext, options.userContext);
             options.userContext = undefined;
@@ -179,6 +178,13 @@ export class AIGNEContext {
             this.memories.push(...options.memories);
             options.memories = undefined;
         }
+        if (options?.hooks) {
+            this.hooks.push(...flat(options.hooks));
+            options.hooks = undefined;
+        }
+    }
+    publish = ((topic, payload, options) => {
+        this.processOptions(options);
         const newContext = options?.newContext === false ? this : this.newContext();
         return this.internal.messageQueue.publish(topic, {
             ...toMessagePayload(payload),
@@ -311,6 +317,7 @@ class AIGNEContextShared {
     usage = newEmptyContextUsage();
     userContext = {};
     memories = [];
+    hooks = [];
     abortController = new AbortController();
     timer;
     initTimeout() {

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

@@ -19,6 +19,7 @@ export type NestAgentSchema = string | {
 export interface BaseAgentSchema {
     name?: string;
     description?: string;
+    taskTitle?: string;
     inputSchema?: ZodType<Record<string, any>>;
     defaultInput?: Record<string, any>;
     outputSchema?: ZodType<Record<string, any>>;
@@ -46,6 +47,9 @@ export interface TeamAgentSchema extends BaseAgentSchema {
     type: "team";
     mode?: ProcessMode;
     iterateOn?: string;
+    concurrency?: number;
+    iterateWithPreviousOutput?: boolean;
+    includeAllStepsOutput?: boolean;
     reflection?: Omit<ReflectionMode, "reviewer"> & {
         reviewer: NestAgentSchema;
     };

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

@@ -30,6 +30,7 @@ export async function parseAgentFile(path, data) {
             name: optionalize(z.string()),
             alias: optionalize(z.array(z.string())),
             description: optionalize(z.string()),
+            taskTitle: optionalize(z.string()),
             inputSchema: optionalize(inputOutputSchema({ path })).transform((v) => v ? jsonSchemaToZod(v) : undefined),
             defaultInput: optionalize(defaultInputSchema),
             outputSchema: optionalize(inputOutputSchema({ path })).transform((v) => v ? jsonSchemaToZod(v) : undefined),
@@ -74,6 +75,9 @@ export async function parseAgentFile(path, data) {
                 type: z.literal("team"),
                 mode: optionalize(z.nativeEnum(ProcessMode)),
                 iterateOn: optionalize(z.string()),
+                concurrency: optionalize(z.number().int().min(1)),
+                iterateWithPreviousOutput: optionalize(z.boolean()),
+                includeAllStepsOutput: optionalize(z.boolean()),
                 reflection: camelizeSchema(optionalize(z.object({
                     reviewer: nestAgentSchema,
                     isApproved: z.string(),

package/lib/esm/loader/schema.js

@@ -1,8 +1,8 @@
 import { nodejs } from "@aigne/platform-helpers/nodejs/index.js";
-import camelize from "camelize-ts";
 import { parse } from "yaml";
 import { z } from "zod";
 import { DEFAULT_INPUT_ACTION_GET } from "../agents/agent.js";
+import { camelize } from "../utils/camelize.js";
 import { isRecord } from "../utils/type-utils.js";
 export const inputOutputSchema = ({ path }) => {
     const includeExternalSchema = async (schema) => {

package/lib/esm/utils/camelize.d.ts

@@ -0,0 +1 @@
+export declare function camelize<T>(obj: T, shallow?: boolean): any;

package/lib/esm/utils/camelize.js

@@ -0,0 +1,18 @@
+import { isRecord } from "./type-utils.js";
+export function camelize(obj, shallow = false) {
+    if (Array.isArray(obj)) {
+        return shallow ? obj : obj.map((item) => camelize(item, false));
+    }
+    if (isRecord(obj)) {
+        return Object.fromEntries(Object.entries(obj).map(([key, value]) => [
+            camelCase(key),
+            shallow ? value : camelize(value, false),
+        ]));
+    }
+    return obj;
+}
+function camelCase(key) {
+    key = key.replace(/[-_#@$\s]+(.)?/g, (_, char) => char.toUpperCase());
+    key = key.charAt(0).toLowerCase() + key.slice(1);
+    return key;
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@aigne/core",
-  "version": "1.44.0",
+  "version": "1.46.0",
   "description": "AIGNE core library for building AI-powered applications",
   "publishConfig": {
     "access": "public"
@@ -76,11 +76,10 @@
     "debug": "^4.4.1",
     "eventsource-parser": "^3.0.3",
     "fast-deep-equal": "^3.1.3",
+    "fastq": "^1.19.1",
     "immer": "^10.1.1",
     "jaison": "^2.0.2",
     "jsonata": "^2.0.6",
-    "mustache": "^4.2.0",
-    "nanoid": "^5.1.5",
     "nunjucks": "^3.2.4",
     "p-retry": "^6.2.1",
     "raw-body": "^3.0.0",
@@ -91,14 +90,13 @@
     "zod": "^3.25.67",
     "zod-to-json-schema": "^3.24.6",
     "@aigne/observability-api": "^0.9.0",
-    "@aigne/platform-helpers": "^0.5.0"
+    "@aigne/platform-helpers": "^0.6.0"
   },
   "devDependencies": {
     "@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.12",
     "@types/nunjucks": "^3.2.6",
     "compression": "^1.8.0",