@aigne/core 1.0.11 → 1.0.12-beta.10

package/lib/cjs/agent.d.ts

@@ -32,6 +32,7 @@ export declare abstract class Agent<I extends RunnableInput = RunnableInput, O e
     run(input: I, options?: RunOptions & {
         stream?: false;
     }): Promise<O>;
+    private addHistory;
     /**
      * Hook that is called before the agent result is returned.
      * @param _result The agent result.

package/lib/cjs/agent.js

@@ -118,6 +118,7 @@ class Agent extends runnable_1.Runnable {
                 logger_1.logger.debug(`AIGNE core: run agent ${this.name || this.id} success`, {
                     result,
                 });
+                await this.addHistory({ input, output: result });
                 await this.onResult(result);
             });
         }
@@ -130,9 +131,17 @@ class Agent extends runnable_1.Runnable {
             result,
         });
         // TODO: validate result against outputs schema
+        await this.addHistory({ input, output: result });
         await this.onResult(result);
         return result;
     }
+    async addHistory({ input, output }) {
+        await this.context?.historyManager?.create({ input, output }, {
+            userId: this.context?.state.userId,
+            sessionId: this.context?.state.sessionId,
+            agentId: this.id,
+        });
+    }
     /**
      * Hook that is called before the agent result is returned.
      * @param _result The agent result.

package/lib/cjs/context.d.ts

@@ -1,4 +1,5 @@
 import type { LLMModelConfiguration } from "./llm-model";
+import type { Memorable } from "./memorable";
 import type { Runnable, RunnableDefinition } from "./runnable";
 export interface ContextState {
     userId?: string;
@@ -9,8 +10,13 @@ export interface ContextConfig {
     llmModel?: LLMModelConfiguration;
 }
 export interface Context<State extends ContextState = ContextState, Config extends ContextConfig = ContextConfig> {
+    id: string;
     state: State;
     config: Config;
+    historyManager?: Memorable<{
+        input: object;
+        output: object;
+    }>;
     resolve<T extends Runnable>(id: string | RunnableDefinition | T): Promise<T>;
     register<R extends Array<RunnableDefinition | Runnable> = []>(...definition: R): void;
     resolveDependency<T>(token: string | symbol): T;

package/lib/cjs/memorable.d.ts

@@ -1,3 +1,4 @@
+import { Agent } from "./agent";
 import type { Context } from "./context";
 import type { LLMModelInputMessage } from "./llm-model";
 import { Runnable } from "./runnable";
@@ -25,6 +26,7 @@ export interface MemoryItem<T> {
     id: string;
     userId?: string;
     sessionId?: string;
+    agentId?: string;
     createdAt: string;
     updatedAt: string;
     memory: T;
@@ -41,6 +43,7 @@ export type MemoryActions<T> = {
         options?: {
             userId?: string;
             sessionId?: string;
+            agentId?: string;
             metadata?: MemoryMetadata;
         };
     };
@@ -55,6 +58,7 @@ export type MemoryActions<T> = {
             k?: number;
             userId?: string;
             sessionId?: string;
+            agentId?: string;
             filter?: MemoryMetadata;
             sort?: MemorySortOptions;
         };
@@ -69,6 +73,7 @@ export type MemoryActions<T> = {
             k?: number;
             userId?: string;
             sessionId?: string;
+            agentId?: string;
             filter?: MemoryMetadata;
             sort?: MemorySortOptions;
         };
@@ -91,6 +96,7 @@ export type MemoryActions<T> = {
         options?: {
             userId?: string;
             sessionId?: string;
+            agentId?: string;
             metadata?: MemoryMetadata;
         };
     };
@@ -128,7 +134,7 @@ export interface SortItem {
     direction: "asc" | "desc";
 }
 export type MemorySortOptions = SortItem | SortItem[];
-export declare abstract class Memorable<T, C = undefined> extends Runnable<MemoryActions<T>, MemoryActions<T>["outputs"]> {
+export declare abstract class Memorable<T, C extends Record<string, any> = Record<string, any>> extends Runnable<MemoryActions<T>, MemoryActions<T>["outputs"]> {
     constructor(context?: Context);
     abstract runner?: MemoryRunner<T, C>;
     abstract add(messages: Extract<MemoryActions<T>, {
@@ -174,10 +180,11 @@ export declare abstract class Memorable<T, C = undefined> extends Runnable<Memor
     }>["outputs"]>;
     abstract reset(): Promise<void>;
 }
-export type MemoryRunnerInput<C = undefined> = {
+export type MemoryRunnerInput<C extends Record<string, any> = Record<string, any>> = {
     messages: MemoryMessage[];
     userId?: string;
     sessionId?: string;
+    agentId?: string;
     metadata?: MemoryMetadata;
     filter?: MemoryMetadata;
     customData: C;
@@ -185,7 +192,7 @@ export type MemoryRunnerInput<C = undefined> = {
 export type MemoryRunnerOutput<T> = {
     actions: MemoryActionItem<T>[];
 };
-export declare abstract class MemoryRunner<T, C = undefined> extends Runnable<MemoryRunnerInput<C>, MemoryRunnerOutput<T>> {
+export declare abstract class MemoryRunner<T, C extends Record<string, any> = Record<string, any>> extends Agent<MemoryRunnerInput<C>, MemoryRunnerOutput<T>> {
     constructor(name: string);
 }
 export type MemorableSearchOutput<T extends Memorable<unknown>> = Awaited<ReturnType<T["search"]>>["results"];

package/lib/cjs/memorable.js

@@ -2,6 +2,7 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.MemoryRunner = exports.Memorable = void 0;
 const lodash_1 = require("lodash");
+const agent_1 = require("./agent");
 const runnable_1 = require("./runnable");
 const utils_1 = require("./utils");
 class Memorable extends runnable_1.Runnable {
@@ -16,7 +17,7 @@ class Memorable extends runnable_1.Runnable {
     }
 }
 exports.Memorable = Memorable;
-class MemoryRunner extends runnable_1.Runnable {
+class MemoryRunner extends agent_1.Agent {
     constructor(name) {
         const id = `${(0, lodash_1.camelCase)(name)}_runner`;
         super({

package/lib/cjs/runtime.d.ts

@@ -2,6 +2,7 @@ import { type DependencyContainer } from "tsyringe";
 import type { constructor as Constructor } from "tsyringe/dist/typings/types";
 import type { Context, ContextState } from "./context";
 import type { LLMModel, LLMModelConfiguration } from "./llm-model";
+import type { Memorable } from "./memorable";
 import { Runnable, type RunnableDefinition } from "./runnable";
 import type { SandboxFunctionRunner } from "./sandbox-function-runner";
 import { OrderedRecord } from "./utils/ordered-map";
@@ -17,6 +18,10 @@ export interface RuntimeOptions<Agents extends {
     config?: RuntimeConfiguration;
     state?: State;
     agents?: Agents;
+    historyManager?: Memorable<{
+        input: object;
+        output: object;
+    }>;
     llmModel?: LLMModel | Constructor<LLMModel>;
     sandboxFunctionRunner?: SandboxFunctionRunner | Constructor<SandboxFunctionRunner>;
 }
@@ -28,6 +33,10 @@ export declare class Runtime<Agents extends {
     get options(): RuntimeOptions<Agents, State>;
     get id(): string;
     get name(): string | undefined;
+    get historyManager(): Memorable<{
+        input: object;
+        output: object;
+    }, Record<string, any>> | undefined;
     config: RuntimeConfiguration;
     state: State;
     agents: Agents;

package/lib/cjs/runtime.js

@@ -47,6 +47,9 @@ let Runtime = Runtime_1 = class Runtime {
     get name() {
         return this.inner.name;
     }
+    get historyManager() {
+        return this.inner.options.historyManager;
+    }
     config;
     state;
     agents = new Proxy({}, { get: (_, prop) => this.resolveSync(prop.toString()) });

package/lib/dts/agent.d.ts

@@ -32,6 +32,7 @@ export declare abstract class Agent<I extends RunnableInput = RunnableInput, O e
     run(input: I, options?: RunOptions & {
         stream?: false;
     }): Promise<O>;
+    private addHistory;
     /**
      * Hook that is called before the agent result is returned.
      * @param _result The agent result.

package/lib/dts/context.d.ts

@@ -1,4 +1,5 @@
 import type { LLMModelConfiguration } from "./llm-model";
+import type { Memorable } from "./memorable";
 import type { Runnable, RunnableDefinition } from "./runnable";
 export interface ContextState {
     userId?: string;
@@ -9,8 +10,13 @@ export interface ContextConfig {
     llmModel?: LLMModelConfiguration;
 }
 export interface Context<State extends ContextState = ContextState, Config extends ContextConfig = ContextConfig> {
+    id: string;
     state: State;
     config: Config;
+    historyManager?: Memorable<{
+        input: object;
+        output: object;
+    }>;
     resolve<T extends Runnable>(id: string | RunnableDefinition | T): Promise<T>;
     register<R extends Array<RunnableDefinition | Runnable> = []>(...definition: R): void;
     resolveDependency<T>(token: string | symbol): T;

package/lib/dts/memorable.d.ts

@@ -1,3 +1,4 @@
+import { Agent } from "./agent";
 import type { Context } from "./context";
 import type { LLMModelInputMessage } from "./llm-model";
 import { Runnable } from "./runnable";
@@ -25,6 +26,7 @@ export interface MemoryItem<T> {
     id: string;
     userId?: string;
     sessionId?: string;
+    agentId?: string;
     createdAt: string;
     updatedAt: string;
     memory: T;
@@ -41,6 +43,7 @@ export type MemoryActions<T> = {
         options?: {
             userId?: string;
             sessionId?: string;
+            agentId?: string;
             metadata?: MemoryMetadata;
         };
     };
@@ -55,6 +58,7 @@ export type MemoryActions<T> = {
             k?: number;
             userId?: string;
             sessionId?: string;
+            agentId?: string;
             filter?: MemoryMetadata;
             sort?: MemorySortOptions;
         };
@@ -69,6 +73,7 @@ export type MemoryActions<T> = {
             k?: number;
             userId?: string;
             sessionId?: string;
+            agentId?: string;
             filter?: MemoryMetadata;
             sort?: MemorySortOptions;
         };
@@ -91,6 +96,7 @@ export type MemoryActions<T> = {
         options?: {
             userId?: string;
             sessionId?: string;
+            agentId?: string;
             metadata?: MemoryMetadata;
         };
     };
@@ -128,7 +134,7 @@ export interface SortItem {
     direction: "asc" | "desc";
 }
 export type MemorySortOptions = SortItem | SortItem[];
-export declare abstract class Memorable<T, C = undefined> extends Runnable<MemoryActions<T>, MemoryActions<T>["outputs"]> {
+export declare abstract class Memorable<T, C extends Record<string, any> = Record<string, any>> extends Runnable<MemoryActions<T>, MemoryActions<T>["outputs"]> {
     constructor(context?: Context);
     abstract runner?: MemoryRunner<T, C>;
     abstract add(messages: Extract<MemoryActions<T>, {
@@ -174,10 +180,11 @@ export declare abstract class Memorable<T, C = undefined> extends Runnable<Memor
     }>["outputs"]>;
     abstract reset(): Promise<void>;
 }
-export type MemoryRunnerInput<C = undefined> = {
+export type MemoryRunnerInput<C extends Record<string, any> = Record<string, any>> = {
     messages: MemoryMessage[];
     userId?: string;
     sessionId?: string;
+    agentId?: string;
     metadata?: MemoryMetadata;
     filter?: MemoryMetadata;
     customData: C;
@@ -185,7 +192,7 @@ export type MemoryRunnerInput<C = undefined> = {
 export type MemoryRunnerOutput<T> = {
     actions: MemoryActionItem<T>[];
 };
-export declare abstract class MemoryRunner<T, C = undefined> extends Runnable<MemoryRunnerInput<C>, MemoryRunnerOutput<T>> {
+export declare abstract class MemoryRunner<T, C extends Record<string, any> = Record<string, any>> extends Agent<MemoryRunnerInput<C>, MemoryRunnerOutput<T>> {
     constructor(name: string);
 }
 export type MemorableSearchOutput<T extends Memorable<unknown>> = Awaited<ReturnType<T["search"]>>["results"];

package/lib/dts/runtime.d.ts

@@ -2,6 +2,7 @@ import { type DependencyContainer } from "tsyringe";
 import type { constructor as Constructor } from "tsyringe/dist/typings/types";
 import type { Context, ContextState } from "./context";
 import type { LLMModel, LLMModelConfiguration } from "./llm-model";
+import type { Memorable } from "./memorable";
 import { Runnable, type RunnableDefinition } from "./runnable";
 import type { SandboxFunctionRunner } from "./sandbox-function-runner";
 import { OrderedRecord } from "./utils/ordered-map";
@@ -17,6 +18,10 @@ export interface RuntimeOptions<Agents extends {
     config?: RuntimeConfiguration;
     state?: State;
     agents?: Agents;
+    historyManager?: Memorable<{
+        input: object;
+        output: object;
+    }>;
     llmModel?: LLMModel | Constructor<LLMModel>;
     sandboxFunctionRunner?: SandboxFunctionRunner | Constructor<SandboxFunctionRunner>;
 }
@@ -28,6 +33,10 @@ export declare class Runtime<Agents extends {
     get options(): RuntimeOptions<Agents, State>;
     get id(): string;
     get name(): string | undefined;
+    get historyManager(): Memorable<{
+        input: object;
+        output: object;
+    }, Record<string, any>> | undefined;
     config: RuntimeConfiguration;
     state: State;
     agents: Agents;

package/lib/esm/agent.d.ts

@@ -32,6 +32,7 @@ export declare abstract class Agent<I extends RunnableInput = RunnableInput, O e
     run(input: I, options?: RunOptions & {
         stream?: false;
     }): Promise<O>;
+    private addHistory;
     /**
      * Hook that is called before the agent result is returned.
      * @param _result The agent result.

package/lib/esm/agent.js

@@ -115,6 +115,7 @@ export class Agent extends Runnable {
                 logger.debug(`AIGNE core: run agent ${this.name || this.id} success`, {
                     result,
                 });
+                await this.addHistory({ input, output: result });
                 await this.onResult(result);
             });
         }
@@ -127,9 +128,17 @@ export class Agent extends Runnable {
             result,
         });
         // TODO: validate result against outputs schema
+        await this.addHistory({ input, output: result });
         await this.onResult(result);
         return result;
     }
+    async addHistory({ input, output }) {
+        await this.context?.historyManager?.create({ input, output }, {
+            userId: this.context?.state.userId,
+            sessionId: this.context?.state.sessionId,
+            agentId: this.id,
+        });
+    }
     /**
      * Hook that is called before the agent result is returned.
      * @param _result The agent result.

package/lib/esm/context.d.ts

@@ -1,4 +1,5 @@
 import type { LLMModelConfiguration } from "./llm-model";
+import type { Memorable } from "./memorable";
 import type { Runnable, RunnableDefinition } from "./runnable";
 export interface ContextState {
     userId?: string;
@@ -9,8 +10,13 @@ export interface ContextConfig {
     llmModel?: LLMModelConfiguration;
 }
 export interface Context<State extends ContextState = ContextState, Config extends ContextConfig = ContextConfig> {
+    id: string;
     state: State;
     config: Config;
+    historyManager?: Memorable<{
+        input: object;
+        output: object;
+    }>;
     resolve<T extends Runnable>(id: string | RunnableDefinition | T): Promise<T>;
     register<R extends Array<RunnableDefinition | Runnable> = []>(...definition: R): void;
     resolveDependency<T>(token: string | symbol): T;

package/lib/esm/memorable.d.ts

@@ -1,3 +1,4 @@
+import { Agent } from "./agent";
 import type { Context } from "./context";
 import type { LLMModelInputMessage } from "./llm-model";
 import { Runnable } from "./runnable";
@@ -25,6 +26,7 @@ export interface MemoryItem<T> {
     id: string;
     userId?: string;
     sessionId?: string;
+    agentId?: string;
     createdAt: string;
     updatedAt: string;
     memory: T;
@@ -41,6 +43,7 @@ export type MemoryActions<T> = {
         options?: {
             userId?: string;
             sessionId?: string;
+            agentId?: string;
             metadata?: MemoryMetadata;
         };
     };
@@ -55,6 +58,7 @@ export type MemoryActions<T> = {
             k?: number;
             userId?: string;
             sessionId?: string;
+            agentId?: string;
             filter?: MemoryMetadata;
             sort?: MemorySortOptions;
         };
@@ -69,6 +73,7 @@ export type MemoryActions<T> = {
             k?: number;
             userId?: string;
             sessionId?: string;
+            agentId?: string;
             filter?: MemoryMetadata;
             sort?: MemorySortOptions;
         };
@@ -91,6 +96,7 @@ export type MemoryActions<T> = {
         options?: {
             userId?: string;
             sessionId?: string;
+            agentId?: string;
             metadata?: MemoryMetadata;
         };
     };
@@ -128,7 +134,7 @@ export interface SortItem {
     direction: "asc" | "desc";
 }
 export type MemorySortOptions = SortItem | SortItem[];
-export declare abstract class Memorable<T, C = undefined> extends Runnable<MemoryActions<T>, MemoryActions<T>["outputs"]> {
+export declare abstract class Memorable<T, C extends Record<string, any> = Record<string, any>> extends Runnable<MemoryActions<T>, MemoryActions<T>["outputs"]> {
     constructor(context?: Context);
     abstract runner?: MemoryRunner<T, C>;
     abstract add(messages: Extract<MemoryActions<T>, {
@@ -174,10 +180,11 @@ export declare abstract class Memorable<T, C = undefined> extends Runnable<Memor
     }>["outputs"]>;
     abstract reset(): Promise<void>;
 }
-export type MemoryRunnerInput<C = undefined> = {
+export type MemoryRunnerInput<C extends Record<string, any> = Record<string, any>> = {
     messages: MemoryMessage[];
     userId?: string;
     sessionId?: string;
+    agentId?: string;
     metadata?: MemoryMetadata;
     filter?: MemoryMetadata;
     customData: C;
@@ -185,7 +192,7 @@ export type MemoryRunnerInput<C = undefined> = {
 export type MemoryRunnerOutput<T> = {
     actions: MemoryActionItem<T>[];
 };
-export declare abstract class MemoryRunner<T, C = undefined> extends Runnable<MemoryRunnerInput<C>, MemoryRunnerOutput<T>> {
+export declare abstract class MemoryRunner<T, C extends Record<string, any> = Record<string, any>> extends Agent<MemoryRunnerInput<C>, MemoryRunnerOutput<T>> {
     constructor(name: string);
 }
 export type MemorableSearchOutput<T extends Memorable<unknown>> = Awaited<ReturnType<T["search"]>>["results"];

package/lib/esm/memorable.js

@@ -1,4 +1,5 @@
 import { camelCase, startCase } from "lodash";
+import { Agent } from "./agent";
 import { Runnable } from "./runnable";
 import { OrderedRecord } from "./utils";
 export class Memorable extends Runnable {
@@ -12,7 +13,7 @@ export class Memorable extends Runnable {
         }, context);
     }
 }
-export class MemoryRunner extends Runnable {
+export class MemoryRunner extends Agent {
     constructor(name) {
         const id = `${camelCase(name)}_runner`;
         super({

package/lib/esm/runtime.d.ts

@@ -2,6 +2,7 @@ import { type DependencyContainer } from "tsyringe";
 import type { constructor as Constructor } from "tsyringe/dist/typings/types";
 import type { Context, ContextState } from "./context";
 import type { LLMModel, LLMModelConfiguration } from "./llm-model";
+import type { Memorable } from "./memorable";
 import { Runnable, type RunnableDefinition } from "./runnable";
 import type { SandboxFunctionRunner } from "./sandbox-function-runner";
 import { OrderedRecord } from "./utils/ordered-map";
@@ -17,6 +18,10 @@ export interface RuntimeOptions<Agents extends {
     config?: RuntimeConfiguration;
     state?: State;
     agents?: Agents;
+    historyManager?: Memorable<{
+        input: object;
+        output: object;
+    }>;
     llmModel?: LLMModel | Constructor<LLMModel>;
     sandboxFunctionRunner?: SandboxFunctionRunner | Constructor<SandboxFunctionRunner>;
 }
@@ -28,6 +33,10 @@ export declare class Runtime<Agents extends {
     get options(): RuntimeOptions<Agents, State>;
     get id(): string;
     get name(): string | undefined;
+    get historyManager(): Memorable<{
+        input: object;
+        output: object;
+    }, Record<string, any>> | undefined;
     config: RuntimeConfiguration;
     state: State;
     agents: Agents;

package/lib/esm/runtime.js

@@ -44,6 +44,9 @@ let Runtime = Runtime_1 = class Runtime {
     get name() {
         return this.inner.name;
     }
+    get historyManager() {
+        return this.inner.options.historyManager;
+    }
     config;
     state;
     agents = new Proxy({}, { get: (_, prop) => this.resolveSync(prop.toString()) });

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@aigne/core",
-  "version": "1.0.11",
+  "version": "1.0.12-beta.10",
   "description": "AIGNE core library",
   "publishConfig": {
     "access": "public"