@aigne/agent-library 1.16.0 → 1.16.1

package/CHANGELOG.md

@@ -13,6 +13,21 @@
     * @aigne/core bumped to 1.22.0
     * @aigne/openai bumped to 0.3.4
 
+## [1.16.1](https://github.com/AIGNE-io/aigne-framework/compare/agent-library-v1.16.0...agent-library-v1.16.1) (2025-06-25)
+
+
+### Bug Fixes
+
+* **core:** pass input/output to MemoryAgent directily ([#178](https://github.com/AIGNE-io/aigne-framework/issues/178)) ([3b20e33](https://github.com/AIGNE-io/aigne-framework/commit/3b20e33f1eefc81ac1e009b1afff14fca46644b1))
+
+
+### Dependencies
+
+* The following workspace dependencies were updated
+  * dependencies
+    * @aigne/core bumped to 1.23.1
+    * @aigne/openai bumped to 0.3.6
+
 ## [1.16.0](https://github.com/AIGNE-io/aigne-framework/compare/agent-library-v1.15.0...agent-library-v1.16.0) (2025-06-25)
 
 

package/lib/cjs/default-memory/default-memory-storage/index.js

@@ -39,19 +39,14 @@ class DefaultMemoryStorage extends storage_js_1.MemoryStorage {
         const db = await this.db;
         const memories = this.options?.enableFTS && query.search
             ? await db
-                .select({
-                id: memory_js_1.Memories.id,
-                sessionId: memory_js_1.Memories.sessionId,
-                content: memory_js_1.Memories.content,
-                createdAt: memory_js_1.Memories.createdAt,
-                updatedAt: memory_js_1.Memories.updatedAt,
-            })
+                .select()
                 .from(memory_js_1.Memories)
                 .innerJoin((0, drizzle_orm_1.sql) `Memories_fts`, (0, drizzle_orm_1.sql) `Memories_fts.id = ${memory_js_1.Memories.id}`)
-                .where((0, drizzle_orm_1.sql) `Memories_fts MATCH ${query.search}`)
+                .where((0, drizzle_orm_1.sql) `Memories_fts MATCH ${drizzle_orm_1.sql.param(this.segment(query.search).join(" OR "))}`)
                 .orderBy((0, drizzle_orm_1.sql) `bm25(Memories_fts)`)
                 .limit(limit)
                 .execute()
+                .then((rows) => rows.map((row) => row.Memories))
             : await db
                 .select()
                 .from(memory_js_1.Memories)
@@ -81,14 +76,18 @@ class DefaultMemoryStorage extends storage_js_1.MemoryStorage {
             this.options?.enableFTS &&
                 db.run((0, drizzle_orm_1.sql) `\
           insert into Memories_fts (id, content)
-          values (${id}, ${this.segment((0, yaml_1.stringify)(memory.content)).join(" ")})`),
+          values (${drizzle_orm_1.sql.param(id)}, ${drizzle_orm_1.sql.param(this.segment((0, yaml_1.stringify)(memory.content)).join(" "))})`),
         ]);
         if (!result)
             throw new Error("Failed to create memory");
         return { result: this.convertMemory(result) };
     }
     segment(str) {
-        return Array.from(new Intl.Segmenter(undefined, { granularity: "word" }).segment(str)).map((i) => i.segment);
+        return (Array.from(new Intl.Segmenter(undefined, { granularity: "word" }).segment(str))
+            .map((i) => i.segment)
+            // Remove non-alphanumeric characters and trim whitespace
+            .map((i) => i.replace(/[^\p{L}\p{N}\s]/gu, "").trim())
+            .filter(Boolean));
     }
 }
 exports.DefaultMemoryStorage = DefaultMemoryStorage;

package/lib/cjs/default-memory/index.d.ts

@@ -1,10 +1,11 @@
-import { type AgentOptions, MemoryAgent, type MemoryAgentOptions, type MemoryRecorderInput, type MemoryRecorderOutput, type MemoryRetrieverInput, type MemoryRetrieverOutput } from "@aigne/core";
+import { type AgentInvokeOptions, type AgentOptions, MemoryAgent, type MemoryAgentOptions, type MemoryRecorderInput, type MemoryRecorderOutput, MemoryRetriever, type MemoryRetrieverInput, type MemoryRetrieverOutput } from "@aigne/core";
 import { type DefaultMemoryStorageOptions } from "./default-memory-storage/index.js";
 import { MemoryStorage } from "./storage.js";
 export interface DefaultMemoryOptions extends Partial<MemoryAgentOptions> {
     storage?: MemoryStorage | DefaultMemoryStorageOptions;
     recorderOptions?: Omit<DefaultMemoryRecorderOptions, "storage">;
     retrieverOptions?: Omit<DefaultMemoryRetrieverOptions, "storage">;
+    messageKey?: string | string[];
 }
 export declare class DefaultMemory extends MemoryAgent {
     constructor(options?: DefaultMemoryOptions);
@@ -13,8 +14,23 @@ export declare class DefaultMemory extends MemoryAgent {
 export interface DefaultMemoryRetrieverOptions extends AgentOptions<MemoryRetrieverInput, MemoryRetrieverOutput> {
     storage: MemoryStorage;
     defaultRetrieveMemoryCount?: number;
+    retrieveFromMessageKey?: string | string[];
+    getSearchPattern?: DefaultMemoryRetriever["getSearchPattern"];
+    formatMessage?: DefaultMemoryRetriever["formatMessage"];
+    formatMemory?: DefaultMemoryRetriever["formatMemory"];
+}
+declare class DefaultMemoryRetriever extends MemoryRetriever {
+    constructor(options: DefaultMemoryRetrieverOptions);
+    private storage;
+    private defaultRetrieveMemoryCount?;
+    private retrieveFromMessageKey?;
+    private getSearchPattern;
+    private formatMessage;
+    private formatMemory;
+    process(input: MemoryRetrieverInput, options: AgentInvokeOptions): Promise<MemoryRetrieverOutput>;
 }
 export interface DefaultMemoryRecorderOptions extends AgentOptions<MemoryRecorderInput, MemoryRecorderOutput> {
     storage: MemoryStorage;
     rememberFromMessageKey?: string | string[];
 }
+export {};

package/lib/cjs/default-memory/index.js

@@ -2,6 +2,7 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.DefaultMemory = void 0;
 const core_1 = require("@aigne/core");
+const type_utils_js_1 = require("@aigne/core/utils/type-utils.js");
 const index_js_1 = require("./default-memory-storage/index.js");
 const storage_js_1 = require("./storage.js");
 const DEFAULT_RETRIEVE_MEMORY_COUNT = 10;
@@ -15,11 +16,13 @@ class DefaultMemory extends core_1.MemoryAgent {
             recorder: options.recorder ??
                 new DefaultMemoryRecorder({
                     ...options.recorderOptions,
+                    rememberFromMessageKey: options.recorderOptions?.rememberFromMessageKey ?? options.messageKey,
                     storage,
                 }),
             retriever: options.retriever ??
                 new DefaultMemoryRetriever({
                     ...options.retrieverOptions,
+                    retrieveFromMessageKey: options.retrieverOptions?.retrieveFromMessageKey ?? options.messageKey,
                     storage,
                 }),
             autoUpdate: options.autoUpdate ?? true,
@@ -34,15 +37,50 @@ class DefaultMemoryRetriever extends core_1.MemoryRetriever {
         super(options);
         this.storage = options.storage;
         this.defaultRetrieveMemoryCount = options.defaultRetrieveMemoryCount;
+        this.retrieveFromMessageKey = options.retrieveFromMessageKey;
+        if (options.getSearchPattern)
+            this.getSearchPattern = options.getSearchPattern;
+        if (options.formatMessage)
+            this.formatMessage = options.formatMessage;
+        if (options.formatMemory)
+            this.formatMemory = options.formatMemory;
     }
     storage;
     defaultRetrieveMemoryCount;
+    retrieveFromMessageKey;
+    getSearchPattern = (search) => {
+        if (!search || typeof search === "string")
+            return search;
+        const obj = search && this.retrieveFromMessageKey ? (0, type_utils_js_1.pick)(search, this.retrieveFromMessageKey) : search;
+        return Object.values(obj)
+            .map((v) => (typeof v === "string" ? v : undefined))
+            .join("\n");
+    };
+    formatMessage = (content) => {
+        if (!this.retrieveFromMessageKey || !(0, type_utils_js_1.isRecord)(content))
+            return content;
+        const obj = (0, type_utils_js_1.pick)(content, this.retrieveFromMessageKey);
+        return Object.values(obj)
+            .map((v) => (typeof v === "string" ? v : undefined))
+            .join("\n");
+    };
+    formatMemory = (content) => {
+        if (!(0, type_utils_js_1.isRecord)(content))
+            return content;
+        if ("input" in content || "output" in content) {
+            return {
+                user: this.formatMessage(content.input),
+                agent: this.formatMessage(content.output),
+            };
+        }
+    };
     async process(input, options) {
         const { result } = await this.storage.search({
             ...input,
+            search: this.getSearchPattern(input.search),
             limit: input.limit ?? this.defaultRetrieveMemoryCount ?? DEFAULT_RETRIEVE_MEMORY_COUNT,
         }, options);
-        return { memories: result };
+        return { memories: result.map((i) => ({ ...i, content: this.formatMemory(i.content) })) };
     }
 }
 class DefaultMemoryRecorder extends core_1.MemoryRecorder {
@@ -58,8 +96,13 @@ class DefaultMemoryRecorder extends core_1.MemoryRecorder {
         for (const item of input.content) {
             const { result } = await this.storage.create({
                 content: {
-                    ...item,
-                    content: this.processMemoryInput(item.content),
+                    input: item.input && this.rememberFromMessageKey
+                        ? (0, type_utils_js_1.pick)(item.input, this.rememberFromMessageKey)
+                        : item.input,
+                    output: item.output && this.rememberFromMessageKey
+                        ? (0, type_utils_js_1.pick)(item.output, this.rememberFromMessageKey)
+                        : item.output,
+                    source: item.source,
                 },
             }, options);
             newMemories.push(result);
@@ -68,17 +111,4 @@ class DefaultMemoryRecorder extends core_1.MemoryRecorder {
             memories: newMemories,
         };
     }
-    processMemoryInput(content) {
-        const keys = Array.isArray(this.rememberFromMessageKey)
-            ? this.rememberFromMessageKey
-            : this.rememberFromMessageKey
-                ? [this.rememberFromMessageKey]
-                : [];
-        if (!keys.length)
-            return content;
-        return Object.entries(content)
-            .filter(([key]) => keys.includes(key))
-            .map(([_, value]) => (typeof value === "string" ? value : JSON.stringify(value)))
-            .join("\n");
-    }
 }

package/lib/dts/default-memory/index.d.ts

@@ -1,10 +1,11 @@
-import { type AgentOptions, MemoryAgent, type MemoryAgentOptions, type MemoryRecorderInput, type MemoryRecorderOutput, type MemoryRetrieverInput, type MemoryRetrieverOutput } from "@aigne/core";
+import { type AgentInvokeOptions, type AgentOptions, MemoryAgent, type MemoryAgentOptions, type MemoryRecorderInput, type MemoryRecorderOutput, MemoryRetriever, type MemoryRetrieverInput, type MemoryRetrieverOutput } from "@aigne/core";
 import { type DefaultMemoryStorageOptions } from "./default-memory-storage/index.js";
 import { MemoryStorage } from "./storage.js";
 export interface DefaultMemoryOptions extends Partial<MemoryAgentOptions> {
     storage?: MemoryStorage | DefaultMemoryStorageOptions;
     recorderOptions?: Omit<DefaultMemoryRecorderOptions, "storage">;
     retrieverOptions?: Omit<DefaultMemoryRetrieverOptions, "storage">;
+    messageKey?: string | string[];
 }
 export declare class DefaultMemory extends MemoryAgent {
     constructor(options?: DefaultMemoryOptions);
@@ -13,8 +14,23 @@ export declare class DefaultMemory extends MemoryAgent {
 export interface DefaultMemoryRetrieverOptions extends AgentOptions<MemoryRetrieverInput, MemoryRetrieverOutput> {
     storage: MemoryStorage;
     defaultRetrieveMemoryCount?: number;
+    retrieveFromMessageKey?: string | string[];
+    getSearchPattern?: DefaultMemoryRetriever["getSearchPattern"];
+    formatMessage?: DefaultMemoryRetriever["formatMessage"];
+    formatMemory?: DefaultMemoryRetriever["formatMemory"];
+}
+declare class DefaultMemoryRetriever extends MemoryRetriever {
+    constructor(options: DefaultMemoryRetrieverOptions);
+    private storage;
+    private defaultRetrieveMemoryCount?;
+    private retrieveFromMessageKey?;
+    private getSearchPattern;
+    private formatMessage;
+    private formatMemory;
+    process(input: MemoryRetrieverInput, options: AgentInvokeOptions): Promise<MemoryRetrieverOutput>;
 }
 export interface DefaultMemoryRecorderOptions extends AgentOptions<MemoryRecorderInput, MemoryRecorderOutput> {
     storage: MemoryStorage;
     rememberFromMessageKey?: string | string[];
 }
+export {};

package/lib/esm/default-memory/default-memory-storage/index.js

@@ -36,19 +36,14 @@ export class DefaultMemoryStorage extends MemoryStorage {
         const db = await this.db;
         const memories = this.options?.enableFTS && query.search
             ? await db
-                .select({
-                id: Memories.id,
-                sessionId: Memories.sessionId,
-                content: Memories.content,
-                createdAt: Memories.createdAt,
-                updatedAt: Memories.updatedAt,
-            })
+                .select()
                 .from(Memories)
                 .innerJoin(sql `Memories_fts`, sql `Memories_fts.id = ${Memories.id}`)
-                .where(sql `Memories_fts MATCH ${query.search}`)
+                .where(sql `Memories_fts MATCH ${sql.param(this.segment(query.search).join(" OR "))}`)
                 .orderBy(sql `bm25(Memories_fts)`)
                 .limit(limit)
                 .execute()
+                .then((rows) => rows.map((row) => row.Memories))
             : await db
                 .select()
                 .from(Memories)
@@ -78,13 +73,17 @@ export class DefaultMemoryStorage extends MemoryStorage {
             this.options?.enableFTS &&
                 db.run(sql `\
           insert into Memories_fts (id, content)
-          values (${id}, ${this.segment(stringify(memory.content)).join(" ")})`),
+          values (${sql.param(id)}, ${sql.param(this.segment(stringify(memory.content)).join(" "))})`),
         ]);
         if (!result)
             throw new Error("Failed to create memory");
         return { result: this.convertMemory(result) };
     }
     segment(str) {
-        return Array.from(new Intl.Segmenter(undefined, { granularity: "word" }).segment(str)).map((i) => i.segment);
+        return (Array.from(new Intl.Segmenter(undefined, { granularity: "word" }).segment(str))
+            .map((i) => i.segment)
+            // Remove non-alphanumeric characters and trim whitespace
+            .map((i) => i.replace(/[^\p{L}\p{N}\s]/gu, "").trim())
+            .filter(Boolean));
     }
 }

package/lib/esm/default-memory/index.d.ts

@@ -1,10 +1,11 @@
-import { type AgentOptions, MemoryAgent, type MemoryAgentOptions, type MemoryRecorderInput, type MemoryRecorderOutput, type MemoryRetrieverInput, type MemoryRetrieverOutput } from "@aigne/core";
+import { type AgentInvokeOptions, type AgentOptions, MemoryAgent, type MemoryAgentOptions, type MemoryRecorderInput, type MemoryRecorderOutput, MemoryRetriever, type MemoryRetrieverInput, type MemoryRetrieverOutput } from "@aigne/core";
 import { type DefaultMemoryStorageOptions } from "./default-memory-storage/index.js";
 import { MemoryStorage } from "./storage.js";
 export interface DefaultMemoryOptions extends Partial<MemoryAgentOptions> {
     storage?: MemoryStorage | DefaultMemoryStorageOptions;
     recorderOptions?: Omit<DefaultMemoryRecorderOptions, "storage">;
     retrieverOptions?: Omit<DefaultMemoryRetrieverOptions, "storage">;
+    messageKey?: string | string[];
 }
 export declare class DefaultMemory extends MemoryAgent {
     constructor(options?: DefaultMemoryOptions);
@@ -13,8 +14,23 @@ export declare class DefaultMemory extends MemoryAgent {
 export interface DefaultMemoryRetrieverOptions extends AgentOptions<MemoryRetrieverInput, MemoryRetrieverOutput> {
     storage: MemoryStorage;
     defaultRetrieveMemoryCount?: number;
+    retrieveFromMessageKey?: string | string[];
+    getSearchPattern?: DefaultMemoryRetriever["getSearchPattern"];
+    formatMessage?: DefaultMemoryRetriever["formatMessage"];
+    formatMemory?: DefaultMemoryRetriever["formatMemory"];
+}
+declare class DefaultMemoryRetriever extends MemoryRetriever {
+    constructor(options: DefaultMemoryRetrieverOptions);
+    private storage;
+    private defaultRetrieveMemoryCount?;
+    private retrieveFromMessageKey?;
+    private getSearchPattern;
+    private formatMessage;
+    private formatMemory;
+    process(input: MemoryRetrieverInput, options: AgentInvokeOptions): Promise<MemoryRetrieverOutput>;
 }
 export interface DefaultMemoryRecorderOptions extends AgentOptions<MemoryRecorderInput, MemoryRecorderOutput> {
     storage: MemoryStorage;
     rememberFromMessageKey?: string | string[];
 }
+export {};

package/lib/esm/default-memory/index.js

@@ -1,4 +1,5 @@
 import { MemoryAgent, MemoryRecorder, MemoryRetriever, } from "@aigne/core";
+import { isRecord, pick } from "@aigne/core/utils/type-utils.js";
 import { DefaultMemoryStorage, } from "./default-memory-storage/index.js";
 import { MemoryStorage } from "./storage.js";
 const DEFAULT_RETRIEVE_MEMORY_COUNT = 10;
@@ -12,11 +13,13 @@ export class DefaultMemory extends MemoryAgent {
             recorder: options.recorder ??
                 new DefaultMemoryRecorder({
                     ...options.recorderOptions,
+                    rememberFromMessageKey: options.recorderOptions?.rememberFromMessageKey ?? options.messageKey,
                     storage,
                 }),
             retriever: options.retriever ??
                 new DefaultMemoryRetriever({
                     ...options.retrieverOptions,
+                    retrieveFromMessageKey: options.retrieverOptions?.retrieveFromMessageKey ?? options.messageKey,
                     storage,
                 }),
             autoUpdate: options.autoUpdate ?? true,
@@ -30,15 +33,50 @@ class DefaultMemoryRetriever extends MemoryRetriever {
         super(options);
         this.storage = options.storage;
         this.defaultRetrieveMemoryCount = options.defaultRetrieveMemoryCount;
+        this.retrieveFromMessageKey = options.retrieveFromMessageKey;
+        if (options.getSearchPattern)
+            this.getSearchPattern = options.getSearchPattern;
+        if (options.formatMessage)
+            this.formatMessage = options.formatMessage;
+        if (options.formatMemory)
+            this.formatMemory = options.formatMemory;
     }
     storage;
     defaultRetrieveMemoryCount;
+    retrieveFromMessageKey;
+    getSearchPattern = (search) => {
+        if (!search || typeof search === "string")
+            return search;
+        const obj = search && this.retrieveFromMessageKey ? pick(search, this.retrieveFromMessageKey) : search;
+        return Object.values(obj)
+            .map((v) => (typeof v === "string" ? v : undefined))
+            .join("\n");
+    };
+    formatMessage = (content) => {
+        if (!this.retrieveFromMessageKey || !isRecord(content))
+            return content;
+        const obj = pick(content, this.retrieveFromMessageKey);
+        return Object.values(obj)
+            .map((v) => (typeof v === "string" ? v : undefined))
+            .join("\n");
+    };
+    formatMemory = (content) => {
+        if (!isRecord(content))
+            return content;
+        if ("input" in content || "output" in content) {
+            return {
+                user: this.formatMessage(content.input),
+                agent: this.formatMessage(content.output),
+            };
+        }
+    };
     async process(input, options) {
         const { result } = await this.storage.search({
             ...input,
+            search: this.getSearchPattern(input.search),
             limit: input.limit ?? this.defaultRetrieveMemoryCount ?? DEFAULT_RETRIEVE_MEMORY_COUNT,
         }, options);
-        return { memories: result };
+        return { memories: result.map((i) => ({ ...i, content: this.formatMemory(i.content) })) };
     }
 }
 class DefaultMemoryRecorder extends MemoryRecorder {
@@ -54,8 +92,13 @@ class DefaultMemoryRecorder extends MemoryRecorder {
         for (const item of input.content) {
             const { result } = await this.storage.create({
                 content: {
-                    ...item,
-                    content: this.processMemoryInput(item.content),
+                    input: item.input && this.rememberFromMessageKey
+                        ? pick(item.input, this.rememberFromMessageKey)
+                        : item.input,
+                    output: item.output && this.rememberFromMessageKey
+                        ? pick(item.output, this.rememberFromMessageKey)
+                        : item.output,
+                    source: item.source,
                 },
             }, options);
             newMemories.push(result);
@@ -64,17 +107,4 @@ class DefaultMemoryRecorder extends MemoryRecorder {
             memories: newMemories,
         };
     }
-    processMemoryInput(content) {
-        const keys = Array.isArray(this.rememberFromMessageKey)
-            ? this.rememberFromMessageKey
-            : this.rememberFromMessageKey
-                ? [this.rememberFromMessageKey]
-                : [];
-        if (!keys.length)
-            return content;
-        return Object.entries(content)
-            .filter(([key]) => keys.includes(key))
-            .map(([_, value]) => (typeof value === "string" ? value : JSON.stringify(value)))
-            .join("\n");
-    }
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@aigne/agent-library",
-  "version": "1.16.0",
+  "version": "1.16.1",
   "description": "Collection of agent libraries for AIGNE framework",
   "publishConfig": {
     "access": "public"
@@ -47,9 +47,9 @@
     "uuid": "^11.1.0",
     "yaml": "^2.7.1",
     "zod": "^3.24.4",
-    "@aigne/core": "^1.23.0",
-    "@aigne/openai": "^0.3.5",
-    "@aigne/sqlite": "^0.1.1"
+    "@aigne/core": "^1.23.1",
+    "@aigne/sqlite": "^0.1.1",
+    "@aigne/openai": "^0.3.6"
   },
   "devDependencies": {
     "@types/bun": "^1.2.12",