@aigne/core 1.48.0 → 1.49.1

package/CHANGELOG.md

@@ -1,5 +1,31 @@
 # Changelog
 
+## [1.49.1](https://github.com/AIGNE-io/aigne-framework/compare/core-v1.49.0...core-v1.49.1) (2025-08-12)
+
+
+### Bug Fixes
+
+* **core:** add optional memory context toggle for AI agent ([#350](https://github.com/AIGNE-io/aigne-framework/issues/350)) ([92322cc](https://github.com/AIGNE-io/aigne-framework/commit/92322ccaf6f2b6e4440d47a7631589061c351d64))
+
+## [1.49.0](https://github.com/AIGNE-io/aigne-framework/compare/core-v1.48.0...core-v1.49.0) (2025-08-12)
+
+
+### Features
+
+* **cli:** add retry functionality and improve error handling for AIGNE Hub ([#348](https://github.com/AIGNE-io/aigne-framework/issues/348)) ([672c93a](https://github.com/AIGNE-io/aigne-framework/commit/672c93abbba8b4b234f6d810536ff4b603a97e1e))
+
+
+### Bug Fixes
+
+* **core:** examples cases that failed when using aigne-hub ([#337](https://github.com/AIGNE-io/aigne-framework/issues/337)) ([0d4a31c](https://github.com/AIGNE-io/aigne-framework/commit/0d4a31c24d9e7d26f00d1accb80719d9ad79a4c6))
+
+
+### Dependencies
+
+* The following workspace dependencies were updated
+  * dependencies
+    * @aigne/platform-helpers bumped to 0.6.1
+
 ## [1.48.0](https://github.com/AIGNE-io/aigne-framework/compare/core-v1.47.0...core-v1.48.0) (2025-08-12)
 
 

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

@@ -405,6 +405,7 @@ export declare abstract class Agent<I extends Message = any, O extends Message =
      * @returns Agent response (streaming or regular)
      */
     invoke(input: I & Message, options?: Partial<AgentInvokeOptions>): Promise<AgentResponse<O>>;
+    private processStreamingAndRetry;
     private callHooks;
     private mergeDefaultInput;
     protected invokeSkill<I extends Message, O extends Message>(skill: Agent<I, O>, input: I & Message, options: AgentInvokeOptions): Promise<O>;
@@ -582,12 +583,14 @@ export interface AgentHooks<I extends Message = Message, O extends Message = Mes
         error: Error;
     }, "output", "error">) => PromiseOrValue<void | {
         output?: O;
+        retry?: boolean;
     }>) | Agent<XOr<{
         input: I;
         output: O;
         error: Error;
     }, "output", "error">, {
         output?: O;
+        retry?: boolean;
     }>;
     onSuccess?: ((event: {
         context: Context;
@@ -607,9 +610,13 @@ export interface AgentHooks<I extends Message = Message, O extends Message = Mes
         agent: Agent;
         input: I;
         error: Error;
-    }) => void) | Agent<{
+    }) => PromiseOrValue<void | {
+        retry?: boolean;
+    }>) | Agent<{
         input: I;
         error: Error;
+    }, {
+        retry?: boolean;
     }>;
     /**
      * Called before a skill (sub-agent) is invoked

package/lib/cjs/agents/agent.js

@@ -47,6 +47,7 @@ exports.textDelta = textDelta;
 exports.jsonDelta = jsonDelta;
 exports.agentProcessResultToObject = agentProcessResultToObject;
 const index_js_1 = require("@aigne/platform-helpers/nodejs/index.js");
+const fast_deep_equal_1 = __importDefault(require("fast-deep-equal"));
 const nunjucks_1 = __importDefault(require("nunjucks"));
 const zod_1 = require("zod");
 const logger_js_1 = require("../utils/logger.js");
@@ -363,7 +364,6 @@ class Agent {
         if (!this.disableEvents)
             opts.context.emit("agentStarted", { agent: this, input });
         try {
-            let response;
             const s = await this.callHooks("onStart", { input }, opts);
             if (s?.input)
                 input = s.input;
@@ -372,31 +372,47 @@ class Agent {
             input = (0, type_utils_js_1.checkArguments)(`Agent ${this.name} input`, this.inputSchema, input);
             await this.preprocess(input, opts);
             this.checkContextStatus(opts);
-            if (!response) {
-                response = await this.process(input, opts);
-                if (response instanceof Agent) {
-                    response = (0, types_js_1.transferToAgentOutput)(response);
+            const response = (0, stream_utils_js_1.asyncGeneratorToReadableStream)(this.processStreamingAndRetry(input, opts));
+            return await this.checkResponseByGuideRails(input, options.streaming ? response : await agentProcessResultToObject(response), opts);
+        }
+        catch (error) {
+            throw (await this.processAgentError(input, error, opts)).error ?? error;
+        }
+    }
+    async *processStreamingAndRetry(input, options) {
+        let output = {};
+        for (;;) {
+            // Reset output to avoid accumulating old data
+            const resetOutput = Object.fromEntries(Object.entries(output).map(([key]) => [key, null]));
+            if (!(0, type_utils_js_1.isEmpty)(resetOutput)) {
+                yield { delta: { json: resetOutput } };
+                output = {};
+            }
+            let response = await this.process(input, options);
+            if (response instanceof Agent)
+                response = (0, types_js_1.transferToAgentOutput)(response);
+            const stream = response instanceof ReadableStream
+                ? response
+                : (0, stream_utils_js_1.isAsyncGenerator)(response)
+                    ? (0, stream_utils_js_1.asyncGeneratorToReadableStream)(response)
+                    : (0, stream_utils_js_1.objectToAgentResponseStream)(response);
+            try {
+                for await (const chunk of stream) {
+                    (0, stream_utils_js_1.mergeAgentResponseChunk)(output, chunk);
+                    yield chunk;
                 }
+                const result = await this.processAgentOutput(input, output, options);
+                if (result && !(0, fast_deep_equal_1.default)(result, output)) {
+                    yield { delta: { json: result } };
+                }
+                // Close the stream after processing
+                break;
             }
-            if (opts.streaming) {
-                const stream = response instanceof ReadableStream
-                    ? response
-                    : (0, stream_utils_js_1.isAsyncGenerator)(response)
-                        ? (0, stream_utils_js_1.asyncGeneratorToReadableStream)(response)
-                        : (0, stream_utils_js_1.objectToAgentResponseStream)(response);
-                return this.checkResponseByGuideRails(input, (0, stream_utils_js_1.onAgentResponseStreamEnd)(stream, {
-                    onResult: async (result) => {
-                        return await this.processAgentOutput(input, result, opts);
-                    },
-                    onError: async (error) => {
-                        return await this.processAgentError(input, error, opts);
-                    },
-                }), opts);
+            catch (error) {
+                const res = await this.processAgentError(input, error, options);
+                if (!res.retry)
+                    throw res.error ?? error;
             }
-            return await this.checkResponseByGuideRails(input, this.processAgentOutput(input, await agentProcessResultToObject(response), opts), opts);
-        }
-        catch (error) {
-            throw await this.processAgentError(input, error, opts);
         }
     }
     async callHooks(hook, input, options) {
@@ -483,12 +499,15 @@ class Agent {
      * @param options Invocation options
      */
     async processAgentError(input, error, options) {
+        if ("$error_has_been_processed" in error && error.$error_has_been_processed)
+            return {};
+        Object.defineProperty(error, "$error_has_been_processed", { value: true, enumerable: false });
         logger_js_1.logger.error("Invoke agent %s failed with error: %O", this.name, error);
         if (!this.disableEvents)
             options.context.emit("agentFailed", { agent: this, error });
-        await this.callHooks("onError", { input, error }, options);
-        await this.callHooks("onEnd", { input, error }, options);
-        return error;
+        const res = (await this.callHooks("onError", { input, error }, options)) ?? {};
+        Object.assign(res, await this.callHooks("onEnd", { input, error }, options));
+        return { ...res };
     }
     /**
      * Check agent invocation usage to prevent exceeding limits

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

@@ -112,6 +112,7 @@ export interface AIAgentOptions<I extends Message = Message, O extends Message =
      * The template receives a {{memories}} variable containing serialized memory content.
      */
     memoryPromptTemplate?: string;
+    useMemoriesFromContext?: boolean;
 }
 /**
  * Tool choice options for AI agents
@@ -253,6 +254,7 @@ export declare class AIAgent<I extends Message = any, O extends Message = any> e
      * The template receives a {{memories}} variable containing serialized memory content.
      */
     memoryPromptTemplate?: string;
+    useMemoriesFromContext?: boolean;
     /**
      * Whether to catch error from tool execution and continue processing.
      * If set to false, the agent will throw an error if a tool fails

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

@@ -115,6 +115,7 @@ class AIAgent extends agent_js_1.Agent {
         this.toolChoice = options.toolChoice;
         this.memoryAgentsAsTools = options.memoryAgentsAsTools;
         this.memoryPromptTemplate = options.memoryPromptTemplate;
+        this.useMemoriesFromContext = options.useMemoriesFromContext;
         if (typeof options.catchToolsError === "boolean")
             this.catchToolsError = options.catchToolsError;
         this.structuredStreamMode = options.structuredStreamMode;
@@ -187,6 +188,7 @@ class AIAgent extends agent_js_1.Agent {
      * The template receives a {{memories}} variable containing serialized memory content.
      */
     memoryPromptTemplate;
+    useMemoriesFromContext;
     /**
      * Whether to catch error from tool execution and continue processing.
      * If set to false, the agent will throw an error if a tool fails

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

@@ -11,6 +11,7 @@ export interface LoadOptions {
         new (parameters?: MemoryAgentOptions): MemoryAgent;
     }[];
     path: string;
+    model?: ChatModel;
 }
 export declare function load(options: LoadOptions): Promise<AIGNEOptions>;
 export declare function loadAgent(path: string, options?: LoadOptions, agentOptions?: AgentOptions): Promise<Agent>;

package/lib/cjs/loader/index.js

@@ -25,7 +25,7 @@ async function load(options) {
     return {
         ...aigne,
         rootDir,
-        model: await options.loadModel(aigne.model),
+        model: options?.model || (await options.loadModel(aigne.model)),
         agents: pickAgents(aigne.agents ?? []),
         skills: pickAgents(aigne.skills ?? []),
         mcpServer: {

package/lib/cjs/prompt/prompt-builder.js

@@ -79,7 +79,7 @@ class PromptBuilder {
         if (options.agent && options.context) {
             memories.push(...(await options.agent.retrieveMemories({ search: message }, { context: options.context })));
         }
-        if (options.context?.memories?.length) {
+        if (options.agent?.useMemoriesFromContext && options.context?.memories?.length) {
             memories.push(...options.context.memories);
         }
         if (memories.length)

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

@@ -405,6 +405,7 @@ export declare abstract class Agent<I extends Message = any, O extends Message =
      * @returns Agent response (streaming or regular)
      */
     invoke(input: I & Message, options?: Partial<AgentInvokeOptions>): Promise<AgentResponse<O>>;
+    private processStreamingAndRetry;
     private callHooks;
     private mergeDefaultInput;
     protected invokeSkill<I extends Message, O extends Message>(skill: Agent<I, O>, input: I & Message, options: AgentInvokeOptions): Promise<O>;
@@ -582,12 +583,14 @@ export interface AgentHooks<I extends Message = Message, O extends Message = Mes
         error: Error;
     }, "output", "error">) => PromiseOrValue<void | {
         output?: O;
+        retry?: boolean;
     }>) | Agent<XOr<{
         input: I;
         output: O;
         error: Error;
     }, "output", "error">, {
         output?: O;
+        retry?: boolean;
     }>;
     onSuccess?: ((event: {
         context: Context;
@@ -607,9 +610,13 @@ export interface AgentHooks<I extends Message = Message, O extends Message = Mes
         agent: Agent;
         input: I;
         error: Error;
-    }) => void) | Agent<{
+    }) => PromiseOrValue<void | {
+        retry?: boolean;
+    }>) | Agent<{
         input: I;
         error: Error;
+    }, {
+        retry?: boolean;
     }>;
     /**
      * Called before a skill (sub-agent) is invoked

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

@@ -112,6 +112,7 @@ export interface AIAgentOptions<I extends Message = Message, O extends Message =
      * The template receives a {{memories}} variable containing serialized memory content.
      */
     memoryPromptTemplate?: string;
+    useMemoriesFromContext?: boolean;
 }
 /**
  * Tool choice options for AI agents
@@ -253,6 +254,7 @@ export declare class AIAgent<I extends Message = any, O extends Message = any> e
      * The template receives a {{memories}} variable containing serialized memory content.
      */
     memoryPromptTemplate?: string;
+    useMemoriesFromContext?: boolean;
     /**
      * Whether to catch error from tool execution and continue processing.
      * If set to false, the agent will throw an error if a tool fails

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

@@ -11,6 +11,7 @@ export interface LoadOptions {
         new (parameters?: MemoryAgentOptions): MemoryAgent;
     }[];
     path: string;
+    model?: ChatModel;
 }
 export declare function load(options: LoadOptions): Promise<AIGNEOptions>;
 export declare function loadAgent(path: string, options?: LoadOptions, agentOptions?: AgentOptions): Promise<Agent>;

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

@@ -405,6 +405,7 @@ export declare abstract class Agent<I extends Message = any, O extends Message =
      * @returns Agent response (streaming or regular)
      */
     invoke(input: I & Message, options?: Partial<AgentInvokeOptions>): Promise<AgentResponse<O>>;
+    private processStreamingAndRetry;
     private callHooks;
     private mergeDefaultInput;
     protected invokeSkill<I extends Message, O extends Message>(skill: Agent<I, O>, input: I & Message, options: AgentInvokeOptions): Promise<O>;
@@ -582,12 +583,14 @@ export interface AgentHooks<I extends Message = Message, O extends Message = Mes
         error: Error;
     }, "output", "error">) => PromiseOrValue<void | {
         output?: O;
+        retry?: boolean;
     }>) | Agent<XOr<{
         input: I;
         output: O;
         error: Error;
     }, "output", "error">, {
         output?: O;
+        retry?: boolean;
     }>;
     onSuccess?: ((event: {
         context: Context;
@@ -607,9 +610,13 @@ export interface AgentHooks<I extends Message = Message, O extends Message = Mes
         agent: Agent;
         input: I;
         error: Error;
-    }) => void) | Agent<{
+    }) => PromiseOrValue<void | {
+        retry?: boolean;
+    }>) | Agent<{
         input: I;
         error: Error;
+    }, {
+        retry?: boolean;
     }>;
     /**
      * Called before a skill (sub-agent) is invoked

package/lib/esm/agents/agent.js

@@ -1,8 +1,9 @@
 import { nodejs } from "@aigne/platform-helpers/nodejs/index.js";
+import equal from "fast-deep-equal";
 import nunjucks from "nunjucks";
 import { ZodObject, z } from "zod";
 import { logger } from "../utils/logger.js";
-import { agentResponseStreamToObject, asyncGeneratorToReadableStream, isAsyncGenerator, objectToAgentResponseStream, onAgentResponseStreamEnd, } from "../utils/stream-utils.js";
+import { agentResponseStreamToObject, asyncGeneratorToReadableStream, isAsyncGenerator, mergeAgentResponseChunk, objectToAgentResponseStream, onAgentResponseStreamEnd, } from "../utils/stream-utils.js";
 import { checkArguments, createAccessorArray, flat, isEmpty, isNil, isRecord, } from "../utils/type-utils.js";
 import { replaceTransferAgentToName, transferToAgentOutput, } from "./types.js";
 export * from "./types.js";
@@ -315,7 +316,6 @@ export class Agent {
         if (!this.disableEvents)
             opts.context.emit("agentStarted", { agent: this, input });
         try {
-            let response;
             const s = await this.callHooks("onStart", { input }, opts);
             if (s?.input)
                 input = s.input;
@@ -324,31 +324,47 @@ export class Agent {
             input = checkArguments(`Agent ${this.name} input`, this.inputSchema, input);
             await this.preprocess(input, opts);
             this.checkContextStatus(opts);
-            if (!response) {
-                response = await this.process(input, opts);
-                if (response instanceof Agent) {
-                    response = transferToAgentOutput(response);
+            const response = asyncGeneratorToReadableStream(this.processStreamingAndRetry(input, opts));
+            return await this.checkResponseByGuideRails(input, options.streaming ? response : await agentProcessResultToObject(response), opts);
+        }
+        catch (error) {
+            throw (await this.processAgentError(input, error, opts)).error ?? error;
+        }
+    }
+    async *processStreamingAndRetry(input, options) {
+        let output = {};
+        for (;;) {
+            // Reset output to avoid accumulating old data
+            const resetOutput = Object.fromEntries(Object.entries(output).map(([key]) => [key, null]));
+            if (!isEmpty(resetOutput)) {
+                yield { delta: { json: resetOutput } };
+                output = {};
+            }
+            let response = await this.process(input, options);
+            if (response instanceof Agent)
+                response = transferToAgentOutput(response);
+            const stream = response instanceof ReadableStream
+                ? response
+                : isAsyncGenerator(response)
+                    ? asyncGeneratorToReadableStream(response)
+                    : objectToAgentResponseStream(response);
+            try {
+                for await (const chunk of stream) {
+                    mergeAgentResponseChunk(output, chunk);
+                    yield chunk;
                 }
+                const result = await this.processAgentOutput(input, output, options);
+                if (result && !equal(result, output)) {
+                    yield { delta: { json: result } };
+                }
+                // Close the stream after processing
+                break;
             }
-            if (opts.streaming) {
-                const stream = response instanceof ReadableStream
-                    ? response
-                    : isAsyncGenerator(response)
-                        ? asyncGeneratorToReadableStream(response)
-                        : objectToAgentResponseStream(response);
-                return this.checkResponseByGuideRails(input, onAgentResponseStreamEnd(stream, {
-                    onResult: async (result) => {
-                        return await this.processAgentOutput(input, result, opts);
-                    },
-                    onError: async (error) => {
-                        return await this.processAgentError(input, error, opts);
-                    },
-                }), opts);
+            catch (error) {
+                const res = await this.processAgentError(input, error, options);
+                if (!res.retry)
+                    throw res.error ?? error;
             }
-            return await this.checkResponseByGuideRails(input, this.processAgentOutput(input, await agentProcessResultToObject(response), opts), opts);
-        }
-        catch (error) {
-            throw await this.processAgentError(input, error, opts);
         }
     }
     async callHooks(hook, input, options) {
@@ -435,12 +451,15 @@ export class Agent {
      * @param options Invocation options
      */
     async processAgentError(input, error, options) {
+        if ("$error_has_been_processed" in error && error.$error_has_been_processed)
+            return {};
+        Object.defineProperty(error, "$error_has_been_processed", { value: true, enumerable: false });
         logger.error("Invoke agent %s failed with error: %O", this.name, error);
         if (!this.disableEvents)
             options.context.emit("agentFailed", { agent: this, error });
-        await this.callHooks("onError", { input, error }, options);
-        await this.callHooks("onEnd", { input, error }, options);
-        return error;
+        const res = (await this.callHooks("onError", { input, error }, options)) ?? {};
+        Object.assign(res, await this.callHooks("onEnd", { input, error }, options));
+        return { ...res };
     }
     /**
      * Check agent invocation usage to prevent exceeding limits

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

@@ -112,6 +112,7 @@ export interface AIAgentOptions<I extends Message = Message, O extends Message =
      * The template receives a {{memories}} variable containing serialized memory content.
      */
     memoryPromptTemplate?: string;
+    useMemoriesFromContext?: boolean;
 }
 /**
  * Tool choice options for AI agents
@@ -253,6 +254,7 @@ export declare class AIAgent<I extends Message = any, O extends Message = any> e
      * The template receives a {{memories}} variable containing serialized memory content.
      */
     memoryPromptTemplate?: string;
+    useMemoriesFromContext?: boolean;
     /**
      * Whether to catch error from tool execution and continue processing.
      * If set to false, the agent will throw an error if a tool fails

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

@@ -112,6 +112,7 @@ export class AIAgent extends Agent {
         this.toolChoice = options.toolChoice;
         this.memoryAgentsAsTools = options.memoryAgentsAsTools;
         this.memoryPromptTemplate = options.memoryPromptTemplate;
+        this.useMemoriesFromContext = options.useMemoriesFromContext;
         if (typeof options.catchToolsError === "boolean")
             this.catchToolsError = options.catchToolsError;
         this.structuredStreamMode = options.structuredStreamMode;
@@ -184,6 +185,7 @@ export class AIAgent extends Agent {
      * The template receives a {{memories}} variable containing serialized memory content.
      */
     memoryPromptTemplate;
+    useMemoriesFromContext;
     /**
      * Whether to catch error from tool execution and continue processing.
      * If set to false, the agent will throw an error if a tool fails

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

@@ -11,6 +11,7 @@ export interface LoadOptions {
         new (parameters?: MemoryAgentOptions): MemoryAgent;
     }[];
     path: string;
+    model?: ChatModel;
 }
 export declare function load(options: LoadOptions): Promise<AIGNEOptions>;
 export declare function loadAgent(path: string, options?: LoadOptions, agentOptions?: AgentOptions): Promise<Agent>;

package/lib/esm/loader/index.js

@@ -20,7 +20,7 @@ export async function load(options) {
     return {
         ...aigne,
         rootDir,
-        model: await options.loadModel(aigne.model),
+        model: options?.model || (await options.loadModel(aigne.model)),
         agents: pickAgents(aigne.agents ?? []),
         skills: pickAgents(aigne.skills ?? []),
         mcpServer: {

package/lib/esm/prompt/prompt-builder.js

@@ -76,7 +76,7 @@ export class PromptBuilder {
         if (options.agent && options.context) {
             memories.push(...(await options.agent.retrieveMemories({ search: message }, { context: options.context })));
         }
-        if (options.context?.memories?.length) {
+        if (options.agent?.useMemoriesFromContext && options.context?.memories?.length) {
             memories.push(...options.context.memories);
         }
         if (memories.length)

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@aigne/core",
-  "version": "1.48.0",
+  "version": "1.49.1",
   "description": "The functional core of agentic AI",
   "publishConfig": {
     "access": "public"
@@ -89,8 +89,8 @@
     "yaml": "^2.8.0",
     "zod": "^3.25.67",
     "zod-to-json-schema": "^3.24.6",
-    "@aigne/platform-helpers": "^0.6.0",
-    "@aigne/observability-api": "^0.9.0"
+    "@aigne/observability-api": "^0.9.0",
+    "@aigne/platform-helpers": "^0.6.1"
   },
   "devDependencies": {
     "@types/bun": "^1.2.18",