@aigne/core 1.22.0 → 1.23.1

package/lib/esm/memory/recorder.js

@@ -1,10 +1,14 @@
 import { z } from "zod";
-import { Agent } from "../agents/agent.js";
+import { Agent, } from "../agents/agent.js";
 /**
  * @hidden
  */
 export const memoryRecorderInputSchema = z.object({
-    content: z.array(z.unknown()),
+    content: z.array(z.object({
+        input: z.record(z.string(), z.unknown()).optional(),
+        output: z.record(z.string(), z.unknown()).optional(),
+        source: z.string().optional(),
+    })),
 });
 /**
  * @hidden
@@ -43,5 +47,13 @@ export class MemoryRecorder extends Agent {
             inputSchema: memoryRecorderInputSchema,
             outputSchema: memoryRecorderOutputSchema,
         });
+        this._process = options.process;
+    }
+    _process;
+    process(input, options) {
+        if (!this._process) {
+            throw new Error("MemoryRecorder process function is not defined.");
+        }
+        return this._process(input, options);
     }
 }

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

@@ -1,5 +1,6 @@
 import { z } from "zod";
-import { Agent, type AgentOptions, type Message } from "../agents/agent.js";
+import { Agent, type AgentInvokeOptions, type AgentOptions, type AgentProcessResult, type FunctionAgentFn, type Message } from "../agents/agent.js";
+import type { PromiseOrValue } from "../utils/type-utils.js";
 import type { Memory } from "./memory.js";
 /**
  * Input for memory retrieval operations.
@@ -17,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.
@@ -37,13 +38,13 @@ 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 | Record<string, unknown> | undefined;
     limit?: number | undefined;
-    search?: string | undefined;
 }, {
+    search?: string | Record<string, unknown> | undefined;
     limit?: number | undefined;
-    search?: string | undefined;
 }>;
 /**
  * @hidden
@@ -75,6 +76,9 @@ export declare const memoryRetrieverOutputSchema: z.ZodObject<{
         createdAt: string;
     }[];
 }>;
+export interface MemoryRetrieverOptions extends Omit<AgentOptions<MemoryRetrieverInput, MemoryRetrieverOutput>, "inputSchema" | "outputSchema"> {
+    process?: FunctionAgentFn<MemoryRetrieverInput, MemoryRetrieverOutput>;
+}
 /**
  * Abstract base class for agents that retrieve memories from storage.
  *
@@ -89,12 +93,14 @@ export declare const memoryRetrieverOutputSchema: z.ZodObject<{
  * Custom implementations should extend this class and provide concrete
  * implementations of the process method to handle the actual retrieval logic.
  */
-export declare abstract class MemoryRetriever extends Agent<MemoryRetrieverInput, MemoryRetrieverOutput> {
+export declare class MemoryRetriever extends Agent<MemoryRetrieverInput, MemoryRetrieverOutput> {
     tag: string;
     /**
      * Creates a new MemoryRetriever instance with predefined input and output schemas.
      *
      * @param options - Configuration options for the memory retriever agent
      */
-    constructor(options: Omit<AgentOptions<MemoryRetrieverInput, MemoryRetrieverOutput>, "inputSchema" | "outputSchema">);
+    constructor(options: MemoryRetrieverOptions);
+    private _process?;
+    process(input: MemoryRetrieverInput, options: AgentInvokeOptions): PromiseOrValue<AgentProcessResult<MemoryRetrieverOutput>>;
 }

package/lib/esm/memory/retriever.js

@@ -1,11 +1,11 @@
 import { z } from "zod";
-import { Agent } from "../agents/agent.js";
+import { Agent, } from "../agents/agent.js";
 /**
  * @hidden
  */
 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
@@ -44,5 +44,13 @@ export class MemoryRetriever extends Agent {
             inputSchema: memoryRetrieverInputSchema,
             outputSchema: memoryRetrieverOutputSchema,
         });
+        this._process = options.process;
+    }
+    _process;
+    process(input, options) {
+        if (!this._process) {
+            throw new Error("MemoryRetriever process function is not implemented.");
+        }
+        return this._process(input, options);
     }
 }

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.22.0",
+  "version": "1.23.1",
   "description": "AIGNE core library for building AI-powered applications",
   "publishConfig": {
     "access": "public"
@@ -81,7 +81,7 @@
     "yaml": "^2.7.1",
     "zod": "^3.24.4",
     "zod-to-json-schema": "^3.24.5",
-    "@aigne/observability": "^0.1.0",
+    "@aigne/observability": "^0.1.1",
     "@aigne/platform-helpers": "^0.1.2"
   },
   "devDependencies": {