@aigne/core 1.23.0 → 1.23.1

package/CHANGELOG.md

@@ -12,6 +12,13 @@
   * dependencies
     * @aigne/observability bumped to 0.1.0
 
+## [1.23.1](https://github.com/AIGNE-io/aigne-framework/compare/core-v1.23.0...core-v1.23.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))
+
 ## [1.23.0](https://github.com/AIGNE-io/aigne-framework/compare/core-v1.22.0...core-v1.23.0) (2025-06-25)
 
 

package/lib/cjs/agents/agent.js

@@ -317,7 +317,6 @@ class Agent {
         for (const memory of this.memories) {
             const ms = (await memory.retrieve({
                 ...input,
-                search: typeof input.search === "string" ? input.search : JSON.stringify(input.search),
                 limit: input.limit ?? this.maxRetrieveMemoryCount,
             }, options.context)).memories;
             memories.push(...ms);
@@ -521,12 +520,7 @@ class Agent {
     async postprocess(input, output, options) {
         this.checkContextStatus(options);
         this.publishToTopics(output, options);
-        await this.recordMemories({
-            content: [
-                { role: "user", content: input },
-                { role: "agent", content: (0, types_js_1.replaceTransferAgentToName)(output), source: this.name },
-            ],
-        }, options);
+        await this.recordMemories({ content: [{ input, output: (0, types_js_1.replaceTransferAgentToName)(output), source: this.name }] }, options);
     }
     async publishToTopics(output, options) {
         const publishTopics = typeof this.publishTopic === "function" ? await this.publishTopic(output) : this.publishTopic;

package/lib/cjs/memory/memory.js

@@ -134,7 +134,7 @@ class MemoryAgent extends agent_js_1.Agent {
         return context.invoke(this.recorder, input);
     }
     async onMessage({ role, source, message, context }) {
-        this.record({ content: [{ role, source, content: message }] }, context);
+        this.record({ content: [role === "user" ? { input: message } : { output: message, source }] }, context);
     }
 }
 exports.MemoryAgent = MemoryAgent;

package/lib/cjs/memory/recorder.d.ts

@@ -10,13 +10,9 @@ import type { Memory } from "./memory.js";
  * should be stored as memories.
  */
 export interface MemoryRecorderInput extends Message {
-    /**
-     * Array of content items to record as memories.
-     * Each item in this array will typically be converted into a separate memory entry.
-     */
     content: {
-        role: "user" | "agent";
-        content: Message;
+        input?: Message;
+        output?: Message;
         source?: string;
     }[];
 }

package/lib/cjs/memory/recorder.js

@@ -8,8 +8,8 @@ const agent_js_1 = require("../agents/agent.js");
  */
 exports.memoryRecorderInputSchema = zod_1.z.object({
     content: zod_1.z.array(zod_1.z.object({
-        role: zod_1.z.union([zod_1.z.literal("user"), zod_1.z.literal("agent")]),
-        content: zod_1.z.record(zod_1.z.string(), zod_1.z.unknown()),
+        input: zod_1.z.record(zod_1.z.string(), zod_1.z.unknown()).optional(),
+        output: zod_1.z.record(zod_1.z.string(), zod_1.z.unknown()).optional(),
         source: zod_1.z.string().optional(),
     })),
 });

package/lib/cjs/memory/retriever.d.ts

@@ -18,7 +18,7 @@ export interface MemoryRetrieverInput extends Message {
      * Search term to filter memories by.
      * How the search is implemented depends on the specific retriever implementation.
      */
-    search?: string;
+    search?: string | Message;
 }
 /**
  * Output from memory retrieval operations.
@@ -38,12 +38,12 @@ export interface MemoryRetrieverOutput extends Message {
  */
 export declare const memoryRetrieverInputSchema: z.ZodObject<{
     limit: z.ZodOptional<z.ZodNumber>;
-    search: z.ZodOptional<z.ZodString>;
+    search: z.ZodOptional<z.ZodUnion<[z.ZodString, z.ZodRecord<z.ZodString, z.ZodUnknown>]>>;
 }, "strip", z.ZodTypeAny, {
-    search?: string | undefined;
+    search?: string | Record<string, unknown> | undefined;
     limit?: number | undefined;
 }, {
-    search?: string | undefined;
+    search?: string | Record<string, unknown> | undefined;
     limit?: number | undefined;
 }>;
 /**

package/lib/cjs/memory/retriever.js

@@ -8,7 +8,7 @@ const agent_js_1 = require("../agents/agent.js");
  */
 exports.memoryRetrieverInputSchema = zod_1.z.object({
     limit: zod_1.z.number().optional(),
-    search: zod_1.z.string().optional(),
+    search: zod_1.z.union([zod_1.z.string(), zod_1.z.record(zod_1.z.string(), zod_1.z.unknown())]).optional(),
 });
 /**
  * @hidden

package/lib/cjs/utils/type-utils.d.ts

@@ -15,6 +15,7 @@ export declare function isNotEmpty<T>(arr: T[]): arr is [T, ...T[]];
 export declare function duplicates<T>(arr: T[], key?: (item: T) => unknown): T[];
 export declare function remove<T>(arr: T[], remove: T[] | ((item: T) => boolean)): T[];
 export declare function unique<T>(arr: T[], key?: (item: T) => unknown): T[];
+export declare function pick<T extends Record<string, unknown>, K extends keyof T | string>(obj: T, ...keys: (K | K[])[]): Pick<T, Extract<K, keyof T>> & Partial<Record<Exclude<K, keyof T>, unknown>>;
 export declare function omit<T extends Record<string, unknown>, K extends keyof T>(obj: T, ...keys: (K | K[])[]): Omit<T, K>;
 export declare function omitDeep<T, K>(obj: T, ...keys: (K | K[])[]): unknown;
 export declare function omitBy<T extends Record<string, unknown>, K extends keyof T>(obj: T, predicate: (value: T[K], key: K) => boolean): Partial<T>;

package/lib/cjs/utils/type-utils.js

@@ -8,6 +8,7 @@ exports.isNotEmpty = isNotEmpty;
 exports.duplicates = duplicates;
 exports.remove = remove;
 exports.unique = unique;
+exports.pick = pick;
 exports.omit = omit;
 exports.omitDeep = omitDeep;
 exports.omitBy = omitBy;
@@ -75,6 +76,10 @@ function unique(arr, key = (item) => item) {
         return true;
     });
 }
+function pick(obj, ...keys) {
+    const flattenedKeys = new Set(keys.flat());
+    return Object.fromEntries(Object.entries(obj).filter(([key]) => flattenedKeys.has(key)));
+}
 function omit(obj, ...keys) {
     const flattenedKeys = new Set(keys.flat());
     return Object.fromEntries(Object.entries(obj).filter(([key]) => !flattenedKeys.has(key)));

package/lib/dts/memory/recorder.d.ts

@@ -10,13 +10,9 @@ import type { Memory } from "./memory.js";
  * should be stored as memories.
  */
 export interface MemoryRecorderInput extends Message {
-    /**
-     * Array of content items to record as memories.
-     * Each item in this array will typically be converted into a separate memory entry.
-     */
     content: {
-        role: "user" | "agent";
-        content: Message;
+        input?: Message;
+        output?: Message;
         source?: string;
     }[];
 }

package/lib/dts/memory/retriever.d.ts

@@ -18,7 +18,7 @@ export interface MemoryRetrieverInput extends Message {
      * Search term to filter memories by.
      * How the search is implemented depends on the specific retriever implementation.
      */
-    search?: string;
+    search?: string | Message;
 }
 /**
  * Output from memory retrieval operations.
@@ -38,12 +38,12 @@ export interface MemoryRetrieverOutput extends Message {
  */
 export declare const memoryRetrieverInputSchema: z.ZodObject<{
     limit: z.ZodOptional<z.ZodNumber>;
-    search: z.ZodOptional<z.ZodString>;
+    search: z.ZodOptional<z.ZodUnion<[z.ZodString, z.ZodRecord<z.ZodString, z.ZodUnknown>]>>;
 }, "strip", z.ZodTypeAny, {
-    search?: string | undefined;
+    search?: string | Record<string, unknown> | undefined;
     limit?: number | undefined;
 }, {
-    search?: string | undefined;
+    search?: string | Record<string, unknown> | undefined;
     limit?: number | undefined;
 }>;
 /**

package/lib/dts/utils/type-utils.d.ts

@@ -15,6 +15,7 @@ export declare function isNotEmpty<T>(arr: T[]): arr is [T, ...T[]];
 export declare function duplicates<T>(arr: T[], key?: (item: T) => unknown): T[];
 export declare function remove<T>(arr: T[], remove: T[] | ((item: T) => boolean)): T[];
 export declare function unique<T>(arr: T[], key?: (item: T) => unknown): T[];
+export declare function pick<T extends Record<string, unknown>, K extends keyof T | string>(obj: T, ...keys: (K | K[])[]): Pick<T, Extract<K, keyof T>> & Partial<Record<Exclude<K, keyof T>, unknown>>;
 export declare function omit<T extends Record<string, unknown>, K extends keyof T>(obj: T, ...keys: (K | K[])[]): Omit<T, K>;
 export declare function omitDeep<T, K>(obj: T, ...keys: (K | K[])[]): unknown;
 export declare function omitBy<T extends Record<string, unknown>, K extends keyof T>(obj: T, predicate: (value: T[K], key: K) => boolean): Partial<T>;

package/lib/esm/agents/agent.js

@@ -273,7 +273,6 @@ export class Agent {
         for (const memory of this.memories) {
             const ms = (await memory.retrieve({
                 ...input,
-                search: typeof input.search === "string" ? input.search : JSON.stringify(input.search),
                 limit: input.limit ?? this.maxRetrieveMemoryCount,
             }, options.context)).memories;
             memories.push(...ms);
@@ -477,12 +476,7 @@ export class Agent {
     async postprocess(input, output, options) {
         this.checkContextStatus(options);
         this.publishToTopics(output, options);
-        await this.recordMemories({
-            content: [
-                { role: "user", content: input },
-                { role: "agent", content: replaceTransferAgentToName(output), source: this.name },
-            ],
-        }, options);
+        await this.recordMemories({ content: [{ input, output: replaceTransferAgentToName(output), source: this.name }] }, options);
     }
     async publishToTopics(output, options) {
         const publishTopics = typeof this.publishTopic === "function" ? await this.publishTopic(output) : this.publishTopic;

package/lib/esm/memory/memory.js

@@ -130,7 +130,7 @@ export class MemoryAgent extends Agent {
         return context.invoke(this.recorder, input);
     }
     async onMessage({ role, source, message, context }) {
-        this.record({ content: [{ role, source, content: message }] }, context);
+        this.record({ content: [role === "user" ? { input: message } : { output: message, source }] }, context);
     }
 }
 const memoryAgentOptionsSchema = z.object({

package/lib/esm/memory/recorder.d.ts

@@ -10,13 +10,9 @@ import type { Memory } from "./memory.js";
  * should be stored as memories.
  */
 export interface MemoryRecorderInput extends Message {
-    /**
-     * Array of content items to record as memories.
-     * Each item in this array will typically be converted into a separate memory entry.
-     */
     content: {
-        role: "user" | "agent";
-        content: Message;
+        input?: Message;
+        output?: Message;
         source?: string;
     }[];
 }

package/lib/esm/memory/recorder.js

@@ -5,8 +5,8 @@ import { Agent, } from "../agents/agent.js";
  */
 export const memoryRecorderInputSchema = z.object({
     content: z.array(z.object({
-        role: z.union([z.literal("user"), z.literal("agent")]),
-        content: z.record(z.string(), z.unknown()),
+        input: z.record(z.string(), z.unknown()).optional(),
+        output: z.record(z.string(), z.unknown()).optional(),
         source: z.string().optional(),
     })),
 });

package/lib/esm/memory/retriever.d.ts

@@ -18,7 +18,7 @@ export interface MemoryRetrieverInput extends Message {
      * Search term to filter memories by.
      * How the search is implemented depends on the specific retriever implementation.
      */
-    search?: string;
+    search?: string | Message;
 }
 /**
  * Output from memory retrieval operations.
@@ -38,12 +38,12 @@ export interface MemoryRetrieverOutput extends Message {
  */
 export declare const memoryRetrieverInputSchema: z.ZodObject<{
     limit: z.ZodOptional<z.ZodNumber>;
-    search: z.ZodOptional<z.ZodString>;
+    search: z.ZodOptional<z.ZodUnion<[z.ZodString, z.ZodRecord<z.ZodString, z.ZodUnknown>]>>;
 }, "strip", z.ZodTypeAny, {
-    search?: string | undefined;
+    search?: string | Record<string, unknown> | undefined;
     limit?: number | undefined;
 }, {
-    search?: string | undefined;
+    search?: string | Record<string, unknown> | undefined;
     limit?: number | undefined;
 }>;
 /**

package/lib/esm/memory/retriever.js

@@ -5,7 +5,7 @@ import { Agent, } from "../agents/agent.js";
  */
 export const memoryRetrieverInputSchema = z.object({
     limit: z.number().optional(),
-    search: z.string().optional(),
+    search: z.union([z.string(), z.record(z.string(), z.unknown())]).optional(),
 });
 /**
  * @hidden

package/lib/esm/utils/type-utils.d.ts

@@ -15,6 +15,7 @@ export declare function isNotEmpty<T>(arr: T[]): arr is [T, ...T[]];
 export declare function duplicates<T>(arr: T[], key?: (item: T) => unknown): T[];
 export declare function remove<T>(arr: T[], remove: T[] | ((item: T) => boolean)): T[];
 export declare function unique<T>(arr: T[], key?: (item: T) => unknown): T[];
+export declare function pick<T extends Record<string, unknown>, K extends keyof T | string>(obj: T, ...keys: (K | K[])[]): Pick<T, Extract<K, keyof T>> & Partial<Record<Exclude<K, keyof T>, unknown>>;
 export declare function omit<T extends Record<string, unknown>, K extends keyof T>(obj: T, ...keys: (K | K[])[]): Omit<T, K>;
 export declare function omitDeep<T, K>(obj: T, ...keys: (K | K[])[]): unknown;
 export declare function omitBy<T extends Record<string, unknown>, K extends keyof T>(obj: T, predicate: (value: T[K], key: K) => boolean): Partial<T>;

package/lib/esm/utils/type-utils.js

@@ -58,6 +58,10 @@ export function unique(arr, key = (item) => item) {
         return true;
     });
 }
+export function pick(obj, ...keys) {
+    const flattenedKeys = new Set(keys.flat());
+    return Object.fromEntries(Object.entries(obj).filter(([key]) => flattenedKeys.has(key)));
+}
 export function omit(obj, ...keys) {
     const flattenedKeys = new Set(keys.flat());
     return Object.fromEntries(Object.entries(obj).filter(([key]) => !flattenedKeys.has(key)));

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@aigne/core",
-  "version": "1.23.0",
+  "version": "1.23.1",
   "description": "AIGNE core library for building AI-powered applications",
   "publishConfig": {
     "access": "public"
@@ -81,8 +81,8 @@
     "yaml": "^2.7.1",
     "zod": "^3.24.4",
     "zod-to-json-schema": "^3.24.5",
-    "@aigne/platform-helpers": "^0.1.2",
-    "@aigne/observability": "^0.1.1"
+    "@aigne/observability": "^0.1.1",
+    "@aigne/platform-helpers": "^0.1.2"
   },
   "devDependencies": {
     "@types/bun": "^1.2.12",