@aigne/core 1.15.0 → 1.17.0

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

@@ -1,7 +1,7 @@
 import EventEmitter from "node:events";
 import { 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 { ChatModel } from "../models/chat-model.js";
 import { type OmitPropertiesFromArrayFirstElement } from "../utils/type-utils.js";
 import type { Args, Listener, TypedEventEmitter } from "../utils/typed-event-emtter.js";
 import { type MessagePayload, MessageQueue, type MessageQueueListener, type Unsubscribe } from "./message-queue.js";
@@ -38,19 +38,27 @@ export type ContextEmitEventMap = {
 /**
  * @hidden
  */
-export interface InvokeOptions extends AgentInvokeOptions {
+export interface InvokeOptions<U extends UserContext = UserContext> extends Partial<Omit<AgentInvokeOptions<U>, "context">> {
     returnActiveAgent?: boolean;
     disableTransfer?: boolean;
+    sourceAgent?: Agent;
+    userContext?: U;
 }
 /**
  * @hidden
  */
-export interface Context extends TypedEventEmitter<ContextEventMap, ContextEmitEventMap> {
+export interface UserContext extends Record<string, unknown> {
+}
+/**
+ * @hidden
+ */
+export interface Context<U extends UserContext = UserContext> extends TypedEventEmitter<ContextEventMap, ContextEmitEventMap> {
     model?: ChatModel;
     skills?: Agent[];
     usage: ContextUsage;
     limits?: ContextLimits;
     status?: "normal" | "timeout";
+    userContext: U;
     /**
      * Create a user agent to consistently invoke an agent
      * @param agent Agent to invoke
@@ -91,7 +99,7 @@ export interface Context extends TypedEventEmitter<ContextEventMap, ContextEmitE
      * @param topic topic name, or an array of topic names
      * @param payload message to publish
      */
-    publish(topic: string | string[], payload: Omit<MessagePayload, "context"> | Message | string): void;
+    publish(topic: string | string[], payload: Omit<MessagePayload, "context"> | Message | string, options?: InvokeOptions): void;
     subscribe(topic: string | string[], listener?: undefined): Promise<MessagePayload>;
     subscribe(topic: string | string[], listener: MessageQueueListener): Unsubscribe;
     subscribe(topic: string | string[], listener?: MessageQueueListener): Unsubscribe | Promise<MessagePayload>;
@@ -113,15 +121,17 @@ export interface Context extends TypedEventEmitter<ContextEventMap, ContextEmitE
  * @hidden
  */
 export declare class AIGNEContext implements Context {
-    constructor(parent?: ConstructorParameters<typeof AIGNEContextInternal>[0]);
+    constructor(...[parent, ...args]: ConstructorParameters<typeof AIGNEContextShared>);
     parentId?: string;
     id: string;
-    readonly internal: AIGNEContextInternal;
+    readonly internal: AIGNEContextShared;
     get model(): ChatModel | undefined;
     get skills(): Agent<Message, Message>[] | undefined;
     get limits(): ContextLimits | undefined;
     get status(): "normal" | "timeout";
     get usage(): ContextUsage;
+    get userContext(): Context["userContext"];
+    set userContext(userContext: Context["userContext"]);
     newContext({ reset }?: {
         reset?: boolean;
     }): AIGNEContext;
@@ -134,17 +144,18 @@ export declare class AIGNEContext implements Context {
     once<K extends keyof ContextEventMap>(eventName: K, listener: Listener<K, ContextEventMap>): this;
     off<K extends keyof ContextEventMap>(eventName: K, listener: Listener<K, ContextEventMap>): this;
 }
-declare class AIGNEContextInternal {
+declare class AIGNEContextShared {
     private readonly parent?;
     constructor(parent?: (Pick<Context, "model" | "skills" | "limits"> & {
         messageQueue?: MessageQueue;
-    }) | undefined);
+    }) | undefined, overrides?: Partial<Context>);
     readonly messageQueue: MessageQueue;
     readonly events: EventEmitter<ContextEventMap>;
     get model(): ChatModel | undefined;
     get skills(): Agent<Message, Message>[] | undefined;
     get limits(): ContextLimits | undefined;
     usage: ContextUsage;
+    userContext: Context["userContext"];
     private abortController;
     private timer?;
     private initTimeout;

package/lib/cjs/aigne/context.js

@@ -19,13 +19,13 @@ const usage_js_1 = require("./usage.js");
  * @hidden
  */
 class AIGNEContext {
-    constructor(parent) {
+    constructor(...[parent, ...args]) {
         if (parent instanceof AIGNEContext) {
             this.parentId = parent.id;
             this.internal = parent.internal;
         }
         else {
-            this.internal = new AIGNEContextInternal(parent);
+            this.internal = new AIGNEContextShared(parent, ...args);
         }
     }
     parentId;
@@ -46,9 +46,15 @@ class AIGNEContext {
     get usage() {
         return this.internal.usage;
     }
+    get userContext() {
+        return this.internal.userContext;
+    }
+    set userContext(userContext) {
+        this.internal.userContext = userContext;
+    }
     newContext({ reset } = {}) {
         if (reset)
-            return new AIGNEContext(this.internal);
+            return new AIGNEContext(this, { userContext: {} });
         return new AIGNEContext(this);
     }
     invoke = ((agent, message, options) => {
@@ -57,6 +63,8 @@ class AIGNEContext {
             message,
             options,
         });
+        if (options?.userContext)
+            Object.assign(this.userContext, options.userContext);
         if ((0, type_utils_js_1.isNil)(message)) {
             return user_agent_js_1.UserAgent.from({
                 context: this,
@@ -96,7 +104,9 @@ class AIGNEContext {
             return stream;
         });
     });
-    publish = ((topic, payload) => {
+    publish = ((topic, payload, options) => {
+        if (options?.userContext)
+            Object.assign(this.userContext, options.userContext);
         return this.internal.messageQueue.publish(topic, {
             ...(0, message_queue_js_1.toMessagePayload)(payload),
             context: this,
@@ -132,11 +142,12 @@ class AIGNEContext {
     }
 }
 exports.AIGNEContext = AIGNEContext;
-class AIGNEContextInternal {
+class AIGNEContextShared {
     parent;
-    constructor(parent) {
+    constructor(parent, overrides) {
         this.parent = parent;
         this.messageQueue = this.parent?.messageQueue ?? new message_queue_js_1.MessageQueue();
+        this.userContext = overrides?.userContext ?? {};
     }
     messageQueue;
     events = new node_events_1.default();
@@ -150,6 +161,7 @@ class AIGNEContextInternal {
         return this.parent?.limits;
     }
     usage = (0, usage_js_1.newEmptyContextUsage)();
+    userContext;
     abortController = new AbortController();
     timer;
     initTimeout() {
@@ -174,7 +186,15 @@ class AIGNEContextInternal {
         let output;
         for (;;) {
             const result = {};
-            const stream = await activeAgent.invoke(input, context, { streaming: true });
+            if (options?.sourceAgent && activeAgent !== options.sourceAgent) {
+                options.sourceAgent.hooks.onHandoff?.({
+                    context,
+                    source: options.sourceAgent,
+                    target: activeAgent,
+                    input,
+                });
+            }
+            const stream = await activeAgent.invoke(input, { ...options, context, streaming: true });
             for await (const value of stream) {
                 if (value.delta.text) {
                     yield { delta: { text: value.delta.text } };

package/lib/cjs/index.d.ts

@@ -1,10 +1,11 @@
 export * from "./agents/agent.js";
 export * from "./agents/ai-agent.js";
+export * from "./agents/chat-model.js";
+export * from "./agents/guide-rail-agent.js";
 export * from "./agents/mcp-agent.js";
 export * from "./agents/team-agent.js";
 export * from "./agents/types.js";
 export * from "./agents/user-agent.js";
 export * from "./aigne/index.js";
-export * from "./models/chat-model.js";
 export * from "./prompt/prompt-builder.js";
 export * from "./prompt/template.js";

package/lib/cjs/index.js

@@ -16,11 +16,12 @@ var __exportStar = (this && this.__exportStar) || function(m, exports) {
 Object.defineProperty(exports, "__esModule", { value: true });
 __exportStar(require("./agents/agent.js"), exports);
 __exportStar(require("./agents/ai-agent.js"), exports);
+__exportStar(require("./agents/chat-model.js"), exports);
+__exportStar(require("./agents/guide-rail-agent.js"), exports);
 __exportStar(require("./agents/mcp-agent.js"), exports);
 __exportStar(require("./agents/team-agent.js"), exports);
 __exportStar(require("./agents/types.js"), exports);
 __exportStar(require("./agents/user-agent.js"), exports);
 __exportStar(require("./aigne/index.js"), exports);
-__exportStar(require("./models/chat-model.js"), exports);
 __exportStar(require("./prompt/prompt-builder.js"), exports);
 __exportStar(require("./prompt/template.js"), exports);

package/lib/cjs/loader/agent-yaml.d.ts

@@ -14,16 +14,16 @@ export declare function loadAgentFromYamlFile(path: string): Promise<{
     }, {
         [x: string]: any;
     }> | undefined;
-    toolChoice?: AIAgentToolChoice | undefined;
     outputSchema?: ZodObject<Record<string, ZodType<any, z.ZodTypeDef, any>>, z.UnknownKeysParam, z.ZodTypeAny, {
         [x: string]: any;
     }, {
         [x: string]: any;
     }> | undefined;
     outputKey?: string | undefined;
+    toolChoice?: AIAgentToolChoice | undefined;
 } | {
     type: "mcp";
     url?: string | undefined;
-    command?: string | undefined;
     args?: string[] | undefined;
+    command?: string | undefined;
 }>;

package/lib/cjs/loader/index.d.ts

@@ -1,7 +1,14 @@
+import type { Camelize } from "camelize-ts";
 import { z } from "zod";
 import { type Agent } from "../agents/agent.js";
-import type { ChatModel, ChatModelOptions } from "../models/chat-model.js";
+import type { ChatModel, ChatModelOptions } from "../agents/chat-model.js";
 export interface LoadOptions {
+    models: {
+        new (parameters: {
+            model?: string;
+            modelOptions?: ChatModelOptions;
+        }): ChatModel;
+    }[];
     path: string;
 }
 export declare function load(options: LoadOptions): Promise<{
@@ -15,12 +22,12 @@ export declare function load(options: LoadOptions): Promise<{
         temperature?: number | null | undefined;
         provider?: string | null | undefined;
         top_p?: number | null | undefined;
-        frequent_penalty?: number | null | undefined;
+        frequency_penalty?: number | null | undefined;
         presence_penalty?: number | null | undefined;
     } | null | undefined;
 }>;
 export declare function loadAgent(path: string): Promise<Agent>;
-export declare function loadModel(model?: z.infer<typeof aigneFileSchema>["chat_model"], modelOptions?: ChatModelOptions): Promise<ChatModel | undefined>;
+export declare function loadModel(models: LoadOptions["models"], model?: Camelize<z.infer<typeof aigneFileSchema>["chat_model"]>, modelOptions?: ChatModelOptions): Promise<ChatModel | undefined>;
 declare const aigneFileSchema: z.ZodObject<{
     name: z.ZodOptional<z.ZodNullable<z.ZodString>>;
     description: z.ZodOptional<z.ZodNullable<z.ZodString>>;
@@ -29,35 +36,35 @@ declare const aigneFileSchema: z.ZodObject<{
         name: z.ZodOptional<z.ZodNullable<z.ZodString>>;
         temperature: z.ZodOptional<z.ZodNullable<z.ZodNumber>>;
         top_p: z.ZodOptional<z.ZodNullable<z.ZodNumber>>;
-        frequent_penalty: z.ZodOptional<z.ZodNullable<z.ZodNumber>>;
+        frequency_penalty: z.ZodOptional<z.ZodNullable<z.ZodNumber>>;
         presence_penalty: z.ZodOptional<z.ZodNullable<z.ZodNumber>>;
     }, "strip", z.ZodTypeAny, {
         name?: string | null | undefined;
         temperature?: number | null | undefined;
         provider?: string | null | undefined;
         top_p?: number | null | undefined;
-        frequent_penalty?: number | null | undefined;
+        frequency_penalty?: number | null | undefined;
         presence_penalty?: number | null | undefined;
     }, {
         name?: string | null | undefined;
         temperature?: number | null | undefined;
         provider?: string | null | undefined;
         top_p?: number | null | undefined;
-        frequent_penalty?: number | null | undefined;
+        frequency_penalty?: number | null | undefined;
         presence_penalty?: number | null | undefined;
     }>]>>>, {
         name?: string | null | undefined;
         temperature?: number | null | undefined;
         provider?: string | null | undefined;
         top_p?: number | null | undefined;
-        frequent_penalty?: number | null | undefined;
+        frequency_penalty?: number | null | undefined;
         presence_penalty?: number | null | undefined;
     } | null | undefined, string | {
         name?: string | null | undefined;
         temperature?: number | null | undefined;
         provider?: string | null | undefined;
         top_p?: number | null | undefined;
-        frequent_penalty?: number | null | undefined;
+        frequency_penalty?: number | null | undefined;
         presence_penalty?: number | null | undefined;
     } | null | undefined>;
     agents: z.ZodOptional<z.ZodNullable<z.ZodArray<z.ZodString, "many">>>;
@@ -71,7 +78,7 @@ declare const aigneFileSchema: z.ZodObject<{
         temperature?: number | null | undefined;
         provider?: string | null | undefined;
         top_p?: number | null | undefined;
-        frequent_penalty?: number | null | undefined;
+        frequency_penalty?: number | null | undefined;
         presence_penalty?: number | null | undefined;
     } | null | undefined;
     agents?: string[] | null | undefined;
@@ -84,7 +91,7 @@ declare const aigneFileSchema: z.ZodObject<{
         temperature?: number | null | undefined;
         provider?: string | null | undefined;
         top_p?: number | null | undefined;
-        frequent_penalty?: number | null | undefined;
+        frequency_penalty?: number | null | undefined;
         presence_penalty?: number | null | undefined;
     } | null | undefined;
     agents?: string[] | null | undefined;
@@ -98,7 +105,7 @@ export declare function loadAIGNEFile(path: string): Promise<{
         temperature?: number | null | undefined;
         provider?: string | null | undefined;
         top_p?: number | null | undefined;
-        frequent_penalty?: number | null | undefined;
+        frequency_penalty?: number | null | undefined;
         presence_penalty?: number | null | undefined;
     } | null | undefined;
     agents?: string[] | null | undefined;

package/lib/cjs/loader/index.js

@@ -11,14 +11,6 @@ const zod_1 = require("zod");
 const agent_js_1 = require("../agents/agent.js");
 const ai_agent_js_1 = require("../agents/ai-agent.js");
 const mcp_agent_js_1 = require("../agents/mcp-agent.js");
-const bedrock_chat_model_js_1 = require("../models/bedrock-chat-model.js");
-const claude_chat_model_js_1 = require("../models/claude-chat-model.js");
-const deepseek_chat_model_js_1 = require("../models/deepseek-chat-model.js");
-const gemini_chat_model_js_1 = require("../models/gemini-chat-model.js");
-const ollama_chat_model_js_1 = require("../models/ollama-chat-model.js");
-const open_router_chat_model_js_1 = require("../models/open-router-chat-model.js");
-const openai_chat_model_js_1 = require("../models/openai-chat-model.js");
-const xai_chat_model_js_1 = require("../models/xai-chat-model.js");
 const type_utils_js_1 = require("../utils/type-utils.js");
 const agent_js_js_1 = require("./agent-js.js");
 const agent_yaml_js_1 = require("./agent-yaml.js");
@@ -32,7 +24,7 @@ async function load(options) {
     const skills = await Promise.all((aigne.skills ?? []).map((filename) => loadAgent((0, node_path_1.join)(rootDir, filename))));
     return {
         ...aigne,
-        model: await loadModel(aigne.chat_model),
+        model: await loadModel(options.models, aigne.chat_model),
         agents,
         skills,
     };
@@ -70,26 +62,15 @@ async function loadAgent(path) {
 }
 const { MODEL_PROVIDER, MODEL_NAME } = process.env;
 const DEFAULT_MODEL_PROVIDER = "openai";
-const DEFAULT_MODEL_NAME = "gpt-4o-mini";
-async function loadModel(model, modelOptions) {
+async function loadModel(models, model, modelOptions) {
     const params = {
-        model: MODEL_NAME ?? model?.name ?? DEFAULT_MODEL_NAME,
+        model: MODEL_NAME ?? model?.name ?? undefined,
         temperature: model?.temperature ?? undefined,
-        topP: model?.top_p ?? undefined,
-        frequencyPenalty: model?.frequent_penalty ?? undefined,
-        presencePenalty: model?.presence_penalty ?? undefined,
+        topP: model?.topP ?? undefined,
+        frequencyPenalty: model?.frequencyPenalty ?? undefined,
+        presencePenalty: model?.presencePenalty ?? undefined,
     };
-    const availableModels = [
-        openai_chat_model_js_1.OpenAIChatModel,
-        claude_chat_model_js_1.ClaudeChatModel,
-        xai_chat_model_js_1.XAIChatModel,
-        gemini_chat_model_js_1.GeminiChatModel,
-        deepseek_chat_model_js_1.DeepSeekChatModel,
-        open_router_chat_model_js_1.OpenRouterChatModel,
-        ollama_chat_model_js_1.OllamaChatModel,
-        bedrock_chat_model_js_1.BedrockChatModel,
-    ];
-    const M = availableModels.find((m) => m.name
+    const M = models.find((m) => m.name
         .toLowerCase()
         .includes((MODEL_PROVIDER ?? model?.provider ?? DEFAULT_MODEL_PROVIDER).toLowerCase()));
     if (!M)
@@ -107,7 +88,7 @@ const aigneFileSchema = zod_1.z.object({
             name: zod_1.z.string().nullish(),
             temperature: zod_1.z.number().min(0).max(2).nullish(),
             top_p: zod_1.z.number().min(0).nullish(),
-            frequent_penalty: zod_1.z.number().min(-2).max(2).nullish(),
+            frequency_penalty: zod_1.z.number().min(-2).max(2).nullish(),
             presence_penalty: zod_1.z.number().min(-2).max(2).nullish(),
         }),
     ])

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

@@ -0,0 +1,30 @@
+import type { AgentInvokeOptions } from "../../../agents/agent.js";
+import "sqlite3";
+import { Sequelize } from "sequelize";
+import type { Context } from "../../../aigne/context.js";
+import type { PromiseOrValue } from "../../../utils/type-utils.js";
+import type { Memory } from "../../memory.js";
+import { MemoryStorage } from "../storage.js";
+export interface DefaultMemoryStorageOptions {
+    path?: string;
+    getSessionId?: (context: Context) => PromiseOrValue<string>;
+}
+export declare class DefaultMemoryStorage extends MemoryStorage {
+    options?: DefaultMemoryStorageOptions | undefined;
+    memories: {
+        [groupId: string]: Memory[];
+    };
+    constructor(options?: DefaultMemoryStorageOptions | undefined);
+    private _models?;
+    private get models();
+    private convertMemory;
+    search(query: {
+        limit?: number;
+    }, { context }: AgentInvokeOptions): Promise<{
+        result: Memory[];
+    }>;
+    create(memory: Pick<Memory, "content">, { context }: AgentInvokeOptions): Promise<{
+        result: Memory;
+    }>;
+}
+export declare function initSequelize(path?: string): Sequelize;

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

@@ -0,0 +1,69 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.DefaultMemoryStorage = void 0;
+exports.initSequelize = initSequelize;
+require("sqlite3");
+const sequelize_1 = require("sequelize");
+const logger_js_1 = require("../../../utils/logger.js");
+const storage_js_1 = require("../storage.js");
+const migrate_js_1 = require("./migrate.js");
+const memory_js_1 = require("./models/memory.js");
+const DEFAULT_MAX_MEMORY_COUNT = 10;
+class DefaultMemoryStorage extends storage_js_1.MemoryStorage {
+    options;
+    memories = {};
+    constructor(options) {
+        super();
+        this.options = options;
+    }
+    _models;
+    get models() {
+        this._models ??= (async () => {
+            const sequelize = initSequelize(this.options?.path);
+            await (0, migrate_js_1.migrate)(sequelize);
+            return {
+                Memory: (0, memory_js_1.initMemoryModel)(sequelize),
+            };
+        })();
+        return this._models;
+    }
+    convertMemory(m) {
+        return {
+            id: m.id,
+            sessionId: m.sessionId,
+            content: m.content,
+            createdAt: m.createdAt.toISOString(),
+        };
+    }
+    async search(query, { context }) {
+        const { limit = DEFAULT_MAX_MEMORY_COUNT } = query;
+        const sessionId = (await this.options?.getSessionId?.(context)) ?? null;
+        const { Memory } = await this.models;
+        const memories = await Memory.findAll({
+            where: { sessionId },
+            order: [["id", "DESC"]],
+            limit,
+        });
+        return {
+            result: memories.reverse().map(this.convertMemory),
+        };
+    }
+    async create(memory, { context }) {
+        const sessionId = (await this.options?.getSessionId?.(context)) ?? null;
+        const { Memory } = await this.models;
+        const m = await Memory.create({
+            ...memory,
+            sessionId,
+        });
+        return { result: this.convertMemory(m) };
+    }
+}
+exports.DefaultMemoryStorage = DefaultMemoryStorage;
+function initSequelize(path) {
+    const sequelize = new sequelize_1.Sequelize({
+        logging: (sql) => logger_js_1.logger.debug(sql),
+        storage: path,
+        dialect: "sqlite",
+    });
+    return sequelize;
+}

package/lib/cjs/memory/default-memory/default-memory-storage/migrate.d.ts

@@ -0,0 +1,7 @@
+import type { Sequelize } from "sequelize";
+import { type MigrateDownOptions, type MigrateUpOptions } from "umzug";
+export declare const migrate: (sequelize: Sequelize, { type, upOptions, downOptions, }?: {
+    type?: "up" | "down";
+    upOptions?: MigrateUpOptions;
+    downOptions?: MigrateDownOptions;
+}) => Promise<void>;

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

@@ -0,0 +1,53 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.migrate = void 0;
+const umzug_1 = require("umzug");
+const init = __importStar(require("./migrations/20250523165801-init.js"));
+const migrate = async (sequelize, { type = "up", upOptions, downOptions, } = {}) => {
+    const umzug = new umzug_1.Umzug({
+        migrations: [{ ...init, name: "20241224202701-init" }],
+        context: sequelize.getQueryInterface(),
+        storage: new umzug_1.SequelizeStorage({ sequelize }),
+        logger: console,
+    });
+    if (type === "down") {
+        await umzug.down(downOptions);
+    }
+    else {
+        await umzug.up(upOptions);
+    }
+};
+exports.migrate = migrate;

package/lib/cjs/memory/default-memory/default-memory-storage/migrations/20250523165801-init.d.ts

@@ -0,0 +1,7 @@
+import { type QueryInterface } from "sequelize";
+export declare const up: ({ context: queryInterface }: {
+    context: QueryInterface;
+}) => Promise<void>;
+export declare const down: ({ context: queryInterface }: {
+    context: QueryInterface;
+}) => Promise<void>;

package/lib/cjs/memory/default-memory/default-memory-storage/migrations/20250523165801-init.js

@@ -0,0 +1,33 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.down = exports.up = void 0;
+const sequelize_1 = require("sequelize");
+const up = async ({ context: queryInterface }) => {
+    await queryInterface.createTable("Memories", {
+        id: {
+            type: sequelize_1.DataTypes.STRING,
+            primaryKey: true,
+            allowNull: false,
+        },
+        createdAt: {
+            type: sequelize_1.DataTypes.DATE,
+            allowNull: false,
+        },
+        updatedAt: {
+            type: sequelize_1.DataTypes.DATE,
+            allowNull: false,
+        },
+        sessionId: {
+            type: sequelize_1.DataTypes.STRING,
+        },
+        content: {
+            type: sequelize_1.DataTypes.JSON,
+            allowNull: false,
+        },
+    });
+};
+exports.up = up;
+const down = async ({ context: queryInterface }) => {
+    await queryInterface.dropTable("Memories");
+};
+exports.down = down;

package/lib/cjs/memory/default-memory/default-memory-storage/models/memory.d.ts

@@ -0,0 +1,10 @@
+import { type CreationOptional, type InferAttributes, type InferCreationAttributes, Model } from "sequelize";
+import type { ModelStatic, Sequelize } from "sequelize";
+export interface Memory extends Model<InferAttributes<Memory>, InferCreationAttributes<Memory>> {
+    id: CreationOptional<string>;
+    createdAt: CreationOptional<Date>;
+    updatedAt: CreationOptional<Date>;
+    sessionId?: string | null;
+    content: unknown;
+}
+export declare function initMemoryModel(sequelize: Sequelize): ModelStatic<Memory>;

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

@@ -0,0 +1,32 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.initMemoryModel = initMemoryModel;
+const sequelize_1 = require("sequelize");
+const uuid_1 = require("uuid");
+const nextId = () => (0, uuid_1.v7)();
+function initMemoryModel(sequelize) {
+    return class Memory extends sequelize_1.Model {
+    }.init({
+        id: {
+            type: sequelize_1.DataTypes.STRING,
+            primaryKey: true,
+            allowNull: false,
+            defaultValue: nextId,
+        },
+        createdAt: {
+            type: sequelize_1.DataTypes.DATE,
+            allowNull: false,
+        },
+        updatedAt: {
+            type: sequelize_1.DataTypes.DATE,
+            allowNull: false,
+        },
+        sessionId: {
+            type: sequelize_1.DataTypes.STRING,
+        },
+        content: {
+            type: sequelize_1.DataTypes.JSON,
+            allowNull: false,
+        },
+    }, { sequelize });
+}

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

@@ -0,0 +1,10 @@
+import { MemoryAgent, type MemoryAgentOptions } from "../memory.js";
+import { type DefaultMemoryStorageOptions } from "./default-memory-storage/index.js";
+import { MemoryStorage } from "./storage.js";
+export interface DefaultMemoryOptions extends Partial<MemoryAgentOptions> {
+    storage?: MemoryStorage | DefaultMemoryStorageOptions;
+}
+export declare class DefaultMemory extends MemoryAgent {
+    constructor(options?: DefaultMemoryOptions);
+    storage: MemoryStorage;
+}