@aigne/core 1.23.0 → 1.24.0

package/CHANGELOG.md

@@ -12,6 +12,27 @@
   * dependencies
     * @aigne/observability bumped to 0.1.0
 
+## [1.24.0](https://github.com/AIGNE-io/aigne-framework/compare/core-v1.23.1...core-v1.24.0) (2025-06-26)
+
+
+### Features
+
+* **transport:** support invoke server side chat model ([#182](https://github.com/AIGNE-io/aigne-framework/issues/182)) ([f81a1bf](https://github.com/AIGNE-io/aigne-framework/commit/f81a1bf883abda1845ccee09b270e5f583e287ab))
+
+
+### Dependencies
+
+* The following workspace dependencies were updated
+  * dependencies
+    * @aigne/observability bumped to 0.1.2
+
+## [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/agents/ai-agent.d.ts

@@ -1,7 +1,7 @@
 import { type ZodObject, type ZodType, z } from "zod";
 import { PromptBuilder } from "../prompt/prompt-builder.js";
 import { Agent, type AgentInvokeOptions, type AgentOptions, type AgentProcessAsyncGenerator, type Message } from "./agent.js";
-import { ChatModel, type ChatModelInput } from "./chat-model.js";
+import type { ChatModel, ChatModelInput } from "./chat-model.js";
 import type { GuideRailAgentOutput } from "./guide-rail-agent.js";
 export declare const DEFAULT_OUTPUT_KEY = "message";
 /**
@@ -100,7 +100,7 @@ export declare enum AIAgentToolChoice {
  *
  * @hidden
  */
-export declare const aiAgentToolChoiceSchema: z.ZodUnion<[z.ZodNativeEnum<typeof AIAgentToolChoice>, ZodType<Agent<Message, Message>, z.ZodTypeDef, Agent<Message, Message>>]>;
+export declare const aiAgentToolChoiceSchema: z.ZodUnion<[z.ZodNativeEnum<typeof AIAgentToolChoice>, ZodType<Agent<any, any>, z.ZodTypeDef, Agent<any, any>>]>;
 /**
  * Zod schema for validating AIAgentOptions
  *

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

@@ -6,7 +6,6 @@ const prompt_builder_js_1 = require("../prompt/prompt-builder.js");
 const template_js_1 = require("../prompt/template.js");
 const type_utils_js_1 = require("../utils/type-utils.js");
 const agent_js_1 = require("./agent.js");
-const chat_model_js_1 = require("./chat-model.js");
 const types_js_1 = require("./types.js");
 exports.DEFAULT_OUTPUT_KEY = "message";
 /**
@@ -40,7 +39,7 @@ var AIAgentToolChoice;
  *
  * @hidden
  */
-exports.aiAgentToolChoiceSchema = zod_1.z.union([zod_1.z.nativeEnum(AIAgentToolChoice), zod_1.z.instanceof(agent_js_1.Agent)], {
+exports.aiAgentToolChoiceSchema = zod_1.z.union([zod_1.z.nativeEnum(AIAgentToolChoice), zod_1.z.custom()], {
     message: `aiAgentToolChoice must be ${Object.values(AIAgentToolChoice).join(", ")}, or an Agent`,
 });
 /**
@@ -51,8 +50,8 @@ exports.aiAgentToolChoiceSchema = zod_1.z.union([zod_1.z.nativeEnum(AIAgentToolC
  * @hidden
  */
 exports.aiAgentOptionsSchema = agent_js_1.agentOptionsSchema.extend({
-    model: zod_1.z.instanceof(chat_model_js_1.ChatModel).optional(),
-    instructions: zod_1.z.union([zod_1.z.string(), zod_1.z.instanceof(prompt_builder_js_1.PromptBuilder)]).optional(),
+    model: zod_1.z.custom().optional(),
+    instructions: zod_1.z.union([zod_1.z.string(), zod_1.z.custom()]).optional(),
     inputKey: zod_1.z.string().optional(),
     outputKey: zod_1.z.string().optional(),
     toolChoice: exports.aiAgentToolChoiceSchema.optional(),

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

@@ -308,7 +308,7 @@ const mcpAgentOptionsSchema = zod_1.z.union([
         opts: zod_1.z.object({}).optional(),
         timeout: zod_1.z.number().optional(),
         maxReconnects: zod_1.z.number().optional(),
-        shouldReconnect: zod_1.z.function().args(zod_1.z.instanceof(Error)).returns(zod_1.z.boolean()).optional(),
+        shouldReconnect: zod_1.z.custom().optional(),
     }),
     zod_1.z.object({
         command: zod_1.z.string(),

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

@@ -1,6 +1,6 @@
 import { AIGNEObserver } from "@aigne/observability";
-import { Agent, type AgentResponse, type AgentResponseStream, type Message } from "../agents/agent.js";
-import { ChatModel } from "../agents/chat-model.js";
+import type { Agent, AgentResponse, AgentResponseStream, Message } from "../agents/agent.js";
+import type { ChatModel } from "../agents/chat-model.js";
 import type { UserAgent } from "../agents/user-agent.js";
 import { type LoadOptions } from "../loader/index.js";
 import { AIGNEContext, type Context, type InvokeOptions, type UserContext } from "./context.js";

package/lib/cjs/aigne/aigne.js

@@ -3,8 +3,6 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.AIGNE = void 0;
 const observability_1 = require("@aigne/observability");
 const zod_1 = require("zod");
-const agent_js_1 = require("../agents/agent.js");
-const chat_model_js_1 = require("../agents/chat-model.js");
 const index_js_1 = require("../loader/index.js");
 const type_utils_js_1 = require("../utils/type-utils.js");
 const context_js_1 = require("./context.js");
@@ -207,9 +205,9 @@ class AIGNE {
 }
 exports.AIGNE = AIGNE;
 const aigneOptionsSchema = zod_1.z.object({
-    model: zod_1.z.instanceof(chat_model_js_1.ChatModel).optional(),
-    skills: zod_1.z.array(zod_1.z.instanceof(agent_js_1.Agent)).optional(),
-    agents: zod_1.z.array(zod_1.z.instanceof(agent_js_1.Agent)).optional(),
-    observer: zod_1.z.instanceof(observability_1.AIGNEObserver).optional(),
+    model: zod_1.z.custom().optional(),
+    skills: zod_1.z.array(zod_1.z.custom()).optional(),
+    agents: zod_1.z.array(zod_1.z.custom()).optional(),
+    observer: zod_1.z.custom().optional(),
 });
-const aigneAddAgentArgsSchema = zod_1.z.array(zod_1.z.instanceof(agent_js_1.Agent));
+const aigneAddAgentArgsSchema = zod_1.z.array(zod_1.z.custom());

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

@@ -1,7 +1,7 @@
 import type { AIGNEObserver } from "@aigne/observability";
 import type { Span } from "@opentelemetry/api";
 import { Emitter } from "strict-event-emitter";
-import { Agent, type AgentInvokeOptions, type AgentProcessAsyncGenerator, type AgentResponse, type AgentResponseStream, type Message } from "../agents/agent.js";
+import { type Agent, 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";

package/lib/cjs/aigne/context.js

@@ -34,7 +34,7 @@ class AIGNEContext {
                 this.span = tracer?.startSpan("childAIGNEContext", undefined, parentContext);
             }
             else {
-                if (!process.env.AIGNE_OBSERVABILITY_DISABLED) {
+                if (parent.observer && !process.env.AIGNE_OBSERVABILITY_DISABLED) {
                     throw new Error("parent span is not set");
                 }
             }
@@ -369,7 +369,7 @@ async function* withAbortSignal(signal, error, fn) {
     }
 }
 const aigneContextInvokeArgsSchema = zod_1.z.object({
-    agent: zod_1.z.union([zod_1.z.function(), zod_1.z.instanceof(agent_js_1.Agent)]),
+    agent: zod_1.z.union([zod_1.z.custom(), zod_1.z.custom()]),
     message: zod_1.z.union([zod_1.z.record(zod_1.z.unknown()), zod_1.z.string()]).optional(),
     options: zod_1.z.object({ returnActiveAgent: zod_1.z.boolean().optional() }).optional(),
 });

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,7 +15,8 @@ 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 omit<T extends Record<string, unknown>, K extends keyof T>(obj: T, ...keys: (K | K[])[]): Omit<T, K>;
+export declare function pick<T extends object, K extends keyof T>(obj: T, ...keys: (K | K[])[]): Pick<T, K>;
+export declare function omit<T extends object, 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>;
 export declare function orArrayToArray<T>(value?: T | T[]): 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/agents/ai-agent.d.ts

@@ -1,7 +1,7 @@
 import { type ZodObject, type ZodType, z } from "zod";
 import { PromptBuilder } from "../prompt/prompt-builder.js";
 import { Agent, type AgentInvokeOptions, type AgentOptions, type AgentProcessAsyncGenerator, type Message } from "./agent.js";
-import { ChatModel, type ChatModelInput } from "./chat-model.js";
+import type { ChatModel, ChatModelInput } from "./chat-model.js";
 import type { GuideRailAgentOutput } from "./guide-rail-agent.js";
 export declare const DEFAULT_OUTPUT_KEY = "message";
 /**
@@ -100,7 +100,7 @@ export declare enum AIAgentToolChoice {
  *
  * @hidden
  */
-export declare const aiAgentToolChoiceSchema: z.ZodUnion<[z.ZodNativeEnum<typeof AIAgentToolChoice>, ZodType<Agent<Message, Message>, z.ZodTypeDef, Agent<Message, Message>>]>;
+export declare const aiAgentToolChoiceSchema: z.ZodUnion<[z.ZodNativeEnum<typeof AIAgentToolChoice>, ZodType<Agent<any, any>, z.ZodTypeDef, Agent<any, any>>]>;
 /**
  * Zod schema for validating AIAgentOptions
  *

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

@@ -1,6 +1,6 @@
 import { AIGNEObserver } from "@aigne/observability";
-import { Agent, type AgentResponse, type AgentResponseStream, type Message } from "../agents/agent.js";
-import { ChatModel } from "../agents/chat-model.js";
+import type { Agent, AgentResponse, AgentResponseStream, Message } from "../agents/agent.js";
+import type { ChatModel } from "../agents/chat-model.js";
 import type { UserAgent } from "../agents/user-agent.js";
 import { type LoadOptions } from "../loader/index.js";
 import { AIGNEContext, type Context, type InvokeOptions, type UserContext } from "./context.js";

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

@@ -1,7 +1,7 @@
 import type { AIGNEObserver } from "@aigne/observability";
 import type { Span } from "@opentelemetry/api";
 import { Emitter } from "strict-event-emitter";
-import { Agent, type AgentInvokeOptions, type AgentProcessAsyncGenerator, type AgentResponse, type AgentResponseStream, type Message } from "../agents/agent.js";
+import { type Agent, 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";

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,7 +15,8 @@ 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 omit<T extends Record<string, unknown>, K extends keyof T>(obj: T, ...keys: (K | K[])[]): Omit<T, K>;
+export declare function pick<T extends object, K extends keyof T>(obj: T, ...keys: (K | K[])[]): Pick<T, K>;
+export declare function omit<T extends object, 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>;
 export declare function orArrayToArray<T>(value?: T | T[]): 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/agents/ai-agent.d.ts

@@ -1,7 +1,7 @@
 import { type ZodObject, type ZodType, z } from "zod";
 import { PromptBuilder } from "../prompt/prompt-builder.js";
 import { Agent, type AgentInvokeOptions, type AgentOptions, type AgentProcessAsyncGenerator, type Message } from "./agent.js";
-import { ChatModel, type ChatModelInput } from "./chat-model.js";
+import type { ChatModel, ChatModelInput } from "./chat-model.js";
 import type { GuideRailAgentOutput } from "./guide-rail-agent.js";
 export declare const DEFAULT_OUTPUT_KEY = "message";
 /**
@@ -100,7 +100,7 @@ export declare enum AIAgentToolChoice {
  *
  * @hidden
  */
-export declare const aiAgentToolChoiceSchema: z.ZodUnion<[z.ZodNativeEnum<typeof AIAgentToolChoice>, ZodType<Agent<Message, Message>, z.ZodTypeDef, Agent<Message, Message>>]>;
+export declare const aiAgentToolChoiceSchema: z.ZodUnion<[z.ZodNativeEnum<typeof AIAgentToolChoice>, ZodType<Agent<any, any>, z.ZodTypeDef, Agent<any, any>>]>;
 /**
  * Zod schema for validating AIAgentOptions
  *

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

@@ -3,7 +3,6 @@ import { PromptBuilder } from "../prompt/prompt-builder.js";
 import { AgentMessageTemplate, ToolMessageTemplate } from "../prompt/template.js";
 import { checkArguments, isEmpty } from "../utils/type-utils.js";
 import { Agent, agentOptionsSchema, isAgentResponseDelta, } from "./agent.js";
-import { ChatModel, } from "./chat-model.js";
 import { isTransferAgentOutput } from "./types.js";
 export const DEFAULT_OUTPUT_KEY = "message";
 /**
@@ -37,7 +36,7 @@ export var AIAgentToolChoice;
  *
  * @hidden
  */
-export const aiAgentToolChoiceSchema = z.union([z.nativeEnum(AIAgentToolChoice), z.instanceof(Agent)], {
+export const aiAgentToolChoiceSchema = z.union([z.nativeEnum(AIAgentToolChoice), z.custom()], {
     message: `aiAgentToolChoice must be ${Object.values(AIAgentToolChoice).join(", ")}, or an Agent`,
 });
 /**
@@ -48,8 +47,8 @@ export const aiAgentToolChoiceSchema = z.union([z.nativeEnum(AIAgentToolChoice),
  * @hidden
  */
 export const aiAgentOptionsSchema = agentOptionsSchema.extend({
-    model: z.instanceof(ChatModel).optional(),
-    instructions: z.union([z.string(), z.instanceof(PromptBuilder)]).optional(),
+    model: z.custom().optional(),
+    instructions: z.union([z.string(), z.custom()]).optional(),
     inputKey: z.string().optional(),
     outputKey: z.string().optional(),
     toolChoice: aiAgentToolChoiceSchema.optional(),

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

@@ -297,7 +297,7 @@ const mcpAgentOptionsSchema = z.union([
         opts: z.object({}).optional(),
         timeout: z.number().optional(),
         maxReconnects: z.number().optional(),
-        shouldReconnect: z.function().args(z.instanceof(Error)).returns(z.boolean()).optional(),
+        shouldReconnect: z.custom().optional(),
     }),
     z.object({
         command: z.string(),

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

@@ -1,6 +1,6 @@
 import { AIGNEObserver } from "@aigne/observability";
-import { Agent, type AgentResponse, type AgentResponseStream, type Message } from "../agents/agent.js";
-import { ChatModel } from "../agents/chat-model.js";
+import type { Agent, AgentResponse, AgentResponseStream, Message } from "../agents/agent.js";
+import type { ChatModel } from "../agents/chat-model.js";
 import type { UserAgent } from "../agents/user-agent.js";
 import { type LoadOptions } from "../loader/index.js";
 import { AIGNEContext, type Context, type InvokeOptions, type UserContext } from "./context.js";

package/lib/esm/aigne/aigne.js

@@ -1,7 +1,5 @@
 import { AIGNEObserver } from "@aigne/observability";
 import { z } from "zod";
-import { Agent, } from "../agents/agent.js";
-import { ChatModel } from "../agents/chat-model.js";
 import { load } from "../loader/index.js";
 import { checkArguments, createAccessorArray } from "../utils/type-utils.js";
 import { AIGNEContext } from "./context.js";
@@ -203,9 +201,9 @@ export class AIGNE {
     }
 }
 const aigneOptionsSchema = z.object({
-    model: z.instanceof(ChatModel).optional(),
-    skills: z.array(z.instanceof(Agent)).optional(),
-    agents: z.array(z.instanceof(Agent)).optional(),
-    observer: z.instanceof(AIGNEObserver).optional(),
+    model: z.custom().optional(),
+    skills: z.array(z.custom()).optional(),
+    agents: z.array(z.custom()).optional(),
+    observer: z.custom().optional(),
 });
-const aigneAddAgentArgsSchema = z.array(z.instanceof(Agent));
+const aigneAddAgentArgsSchema = z.array(z.custom());

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

@@ -1,7 +1,7 @@
 import type { AIGNEObserver } from "@aigne/observability";
 import type { Span } from "@opentelemetry/api";
 import { Emitter } from "strict-event-emitter";
-import { Agent, type AgentInvokeOptions, type AgentProcessAsyncGenerator, type AgentResponse, type AgentResponseStream, type Message } from "../agents/agent.js";
+import { type Agent, 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";

package/lib/esm/aigne/context.js

@@ -3,7 +3,7 @@ import equal from "fast-deep-equal";
 import { Emitter } from "strict-event-emitter";
 import { v7 } from "uuid";
 import { z } from "zod";
-import { Agent, isAgentResponseDelta, isEmptyChunk, } from "../agents/agent.js";
+import { isAgentResponseDelta, isEmptyChunk, } from "../agents/agent.js";
 import { isTransferAgentOutput, transferAgentOutputKey, } from "../agents/types.js";
 import { UserAgent } from "../agents/user-agent.js";
 import { AgentResponseProgressStream } from "../utils/event-stream.js";
@@ -28,7 +28,7 @@ export class AIGNEContext {
                 this.span = tracer?.startSpan("childAIGNEContext", undefined, parentContext);
             }
             else {
-                if (!process.env.AIGNE_OBSERVABILITY_DISABLED) {
+                if (parent.observer && !process.env.AIGNE_OBSERVABILITY_DISABLED) {
                     throw new Error("parent span is not set");
                 }
             }
@@ -362,7 +362,7 @@ async function* withAbortSignal(signal, error, fn) {
     }
 }
 const aigneContextInvokeArgsSchema = z.object({
-    agent: z.union([z.function(), z.instanceof(Agent)]),
+    agent: z.union([z.custom(), z.custom()]),
     message: z.union([z.record(z.unknown()), z.string()]).optional(),
     options: z.object({ returnActiveAgent: z.boolean().optional() }).optional(),
 });

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,7 +15,8 @@ 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 omit<T extends Record<string, unknown>, K extends keyof T>(obj: T, ...keys: (K | K[])[]): Omit<T, K>;
+export declare function pick<T extends object, K extends keyof T>(obj: T, ...keys: (K | K[])[]): Pick<T, K>;
+export declare function omit<T extends object, 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>;
 export declare function orArrayToArray<T>(value?: T | T[]): 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.24.0",
   "description": "AIGNE core library for building AI-powered applications",
   "publishConfig": {
     "access": "public"
@@ -82,7 +82,7 @@
     "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.2"
   },
   "devDependencies": {
     "@types/bun": "^1.2.12",