@ainetwork/adk 0.0.1 → 0.0.2

package/dist/cjs/modules/memory/memory.module.d.ts

@@ -1,35 +1,5 @@
-import type { ChatObject, SessionObject } from "../../types/memory.js";
-/**
- * Abstract base class for memory storage implementations.
- *
- * Provides an interface for storing and retrieving conversation history
- * and query-intent pairs. Implementations can use various backends
- * such as in-memory storage, databases, or file systems.
- */
-export declare abstract class BaseMemory {
-    /**
-     * Retrieves the conversation history for a specific session.
-     *
-     * @param sessionId - Unique identifier for the session
-     * @returns Promise resolving to the session's conversation history
-     */
-    abstract getSessionHistory(sessionId: string): Promise<SessionObject>;
-    /**
-     * Adds a new chat message to the session history.
-     *
-     * @param sessionId - Unique identifier for the session
-     * @param chat - The chat object to add to the history
-     */
-    abstract updateSessionHistory(sessionId: string, chat: ChatObject): Promise<void>;
-    /**
-     * Stores a query-intent pair for analysis or training purposes.
-     *
-     * @param query - The user's input query
-     * @param intent - The detected intent for the query
-     * @param sessionId - Unique identifier for the session
-     */
-    abstract storeQueryAndIntent(query: string, intent: string, sessionId: string): Promise<void>;
-}
+import { MemoryType } from "../../types/memory.js";
+import type { BaseMemory } from "./base.memory.js";
 /**
  * Module wrapper for memory storage implementations.
  *
@@ -48,7 +18,7 @@ export declare abstract class BaseMemory {
  */
 export declare class MemoryModule {
     /** The memory storage implementation */
-    private memory;
+    private memoryMap;
     /**
      * Creates a new MemoryModule with the specified storage implementation.
      *
@@ -60,6 +30,6 @@ export declare class MemoryModule {
      *
      * @returns The active memory storage instance
      */
-    getMemory(): BaseMemory;
+    getMemory(type?: MemoryType): BaseMemory | undefined;
 }
 //# sourceMappingURL=memory.module.d.ts.map

package/dist/cjs/modules/memory/memory.module.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"memory.module.d.ts","sourceRoot":"","sources":["../../../../src/modules/memory/memory.module.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,UAAU,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AAEnE;;;;;;GAMG;AACH,8BAAsB,UAAU;IAC/B;;;;;OAKG;IACH,QAAQ,CAAC,iBAAiB,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,aAAa,CAAC;IACrE;;;;;OAKG;IACH,QAAQ,CAAC,oBAAoB,CAC5B,SAAS,EAAE,MAAM,EACjB,IAAI,EAAE,UAAU,GACd,OAAO,CAAC,IAAI,CAAC;IAChB;;;;;;OAMG;IACH,QAAQ,CAAC,mBAAmB,CAC3B,KAAK,EAAE,MAAM,EACb,MAAM,EAAE,MAAM,EACd,SAAS,EAAE,MAAM,GACf,OAAO,CAAC,IAAI,CAAC;CAChB;AAED;;;;;;;;;;;;;;;GAeG;AACH,qBAAa,YAAY;IACxB,wCAAwC;IACxC,OAAO,CAAC,MAAM,CAAa;IAE3B;;;;OAIG;gBACS,MAAM,EAAE,UAAU;IAI9B;;;;OAIG;IACI,SAAS;CAIhB"}
+{"version":3,"file":"memory.module.d.ts","sourceRoot":"","sources":["../../../../src/modules/memory/memory.module.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAC;AAC/C,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AACnD;;;;;;;;;;;;;;;GAeG;AACH,qBAAa,YAAY;IACxB,wCAAwC;IACxC,OAAO,CAAC,SAAS,CAA8B;IAE/C;;;;OAIG;gBACS,MAAM,EAAE,UAAU;IAK9B;;;;OAIG;IACI,SAAS,CAAC,IAAI,CAAC,EAAE,UAAU,GAAG,UAAU,GAAG,SAAS;CAW3D"}

package/dist/cjs/modules/memory/memory.module.js

@@ -1,16 +1,7 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.MemoryModule = exports.BaseMemory = void 0;
-/**
- * Abstract base class for memory storage implementations.
- *
- * Provides an interface for storing and retrieving conversation history
- * and query-intent pairs. Implementations can use various backends
- * such as in-memory storage, databases, or file systems.
- */
-class BaseMemory {
-}
-exports.BaseMemory = BaseMemory;
+exports.MemoryModule = void 0;
+const memory_js_1 = require("../../types/memory.js");
 /**
  * Module wrapper for memory storage implementations.
  *
@@ -34,16 +25,23 @@ class MemoryModule {
      * @param memory - The memory storage implementation to use
      */
     constructor(memory) {
-        this.memory = memory;
+        this.memoryMap = new Map();
+        this.memoryMap.set(memory_js_1.MemoryType._DEFAULT, memory);
     }
     /**
      * Returns the current memory storage implementation.
      *
      * @returns The active memory storage instance
      */
-    getMemory() {
-        // TODO: Support multi-memory for each type of memory?
-        return this.memory;
+    getMemory(type) {
+        if (!type) {
+            return this.memoryMap.get(memory_js_1.MemoryType._DEFAULT);
+        }
+        const memory = this.memoryMap.get(type);
+        if (!memory) {
+            return this.memoryMap.get(memory_js_1.MemoryType._DEFAULT);
+        }
+        return memory;
     }
 }
 exports.MemoryModule = MemoryModule;

package/dist/cjs/modules/memory/memory.module.js.map

@@ -1 +1 @@
-{"version":3,"file":"memory.module.js","sourceRoot":"","sources":["../../../../src/modules/memory/memory.module.ts"],"names":[],"mappings":";;;AAEA;;;;;;GAMG;AACH,MAAsB,UAAU;CA8B/B;AA9BD,gCA8BC;AAED;;;;;;;;;;;;;;;GAeG;AACH,MAAa,YAAY;IAIxB;;;;OAIG;IACH,YAAY,MAAkB;QAC7B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACtB,CAAC;IAED;;;;OAIG;IACI,SAAS;QACf,sDAAsD;QACtD,OAAO,IAAI,CAAC,MAAM,CAAC;IACpB,CAAC;CACD;AAtBD,oCAsBC"}
+{"version":3,"file":"memory.module.js","sourceRoot":"","sources":["../../../../src/modules/memory/memory.module.ts"],"names":[],"mappings":";;;AAAA,iDAA+C;AAE/C;;;;;;;;;;;;;;;GAeG;AACH,MAAa,YAAY;IAIxB;;;;OAIG;IACH,YAAY,MAAkB;QAC7B,IAAI,CAAC,SAAS,GAAG,IAAI,GAAG,EAAE,CAAC;QAC3B,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,sBAAU,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;IACjD,CAAC;IAED;;;;OAIG;IACI,SAAS,CAAC,IAAiB;QACjC,IAAI,CAAC,IAAI,EAAE,CAAC;YACX,OAAO,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,sBAAU,CAAC,QAAQ,CAAC,CAAC;QAChD,CAAC;QAED,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QACxC,IAAI,CAAC,MAAM,EAAE,CAAC;YACb,OAAO,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,sBAAU,CAAC,QAAQ,CAAC,CAAC;QAChD,CAAC;QACD,OAAO,MAAM,CAAC;IACf,CAAC;CACD;AA9BD,oCA8BC"}

package/dist/cjs/modules/memory/{inmemory.d.ts → providers/inmemory.d.ts}

@@ -2,8 +2,8 @@
  * WARNING: FOR TESTING PURPOSES ONLY      *
  * DO NOT USE IN PRODUCTION                *
  *******************************************/
-import type { ChatObject, SessionObject } from "../../types/memory.js";
-import { BaseMemory } from "./memory.module.js";
+import type { ChatObject, SessionObject } from "../../../types/memory.js";
+import { BaseMemory } from "../base/index.js";
 export declare class InMemoryMemory extends BaseMemory {
     sessionHistory: Map<string, SessionObject>;
     constructor();

package/dist/cjs/modules/memory/providers/inmemory.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"inmemory.d.ts","sourceRoot":"","sources":["../../../../../src/modules/memory/providers/inmemory.ts"],"names":[],"mappings":"AAAA;;;6CAG6C;AAG7C,OAAO,KAAK,EAAE,UAAU,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AACnE,OAAO,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAE9C,qBAAa,cAAe,SAAQ,UAAU;IACtC,cAAc,EAAE,GAAG,CAAC,MAAM,EAAE,aAAa,CAAC,CAAC;;IAOrC,iBAAiB,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,aAAa,CAAC;IAI5D,oBAAoB,CAChC,SAAS,EAAE,MAAM,EACjB,IAAI,EAAE,UAAU,GACd,OAAO,CAAC,IAAI,CAAC;IAOH,mBAAmB,CAC/B,KAAK,EAAE,MAAM,EACb,MAAM,EAAE,MAAM,EACd,SAAS,EAAE,MAAM;CAElB"}

package/dist/cjs/modules/memory/{inmemory.js → providers/inmemory.js}

@@ -6,8 +6,8 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.InMemoryMemory = void 0;
 const node_crypto_1 = require("node:crypto");
-const memory_module_js_1 = require("./memory.module.js");
-class InMemoryMemory extends memory_module_js_1.BaseMemory {
+const index_js_1 = require("../base/index.js");
+class InMemoryMemory extends index_js_1.BaseMemory {
     constructor() {
         super();
         this.sessionHistory = new Map();

package/dist/cjs/modules/memory/providers/inmemory.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"inmemory.js","sourceRoot":"","sources":["../../../../../src/modules/memory/providers/inmemory.ts"],"names":[],"mappings":";AAAA;;;6CAG6C;;;AAE7C,6CAAyC;AAEzC,+CAA8C;AAE9C,MAAa,cAAe,SAAQ,qBAAU;IAG7C;QACC,KAAK,EAAE,CAAC;QACR,IAAI,CAAC,cAAc,GAAG,IAAI,GAAG,EAAyB,CAAC;IACxD,CAAC;IAEM,KAAK,CAAC,iBAAiB,CAAC,SAAiB;QAC/C,OAAO,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;IAC5D,CAAC;IAEM,KAAK,CAAC,oBAAoB,CAChC,SAAiB,EACjB,IAAgB;QAEhB,MAAM,SAAS,GAAG,IAAA,wBAAU,GAAE,CAAC;QAC/B,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;QACxD,OAAO,CAAC,KAAK,CAAC,SAAS,CAAC,GAAG,IAAI,CAAC;QAChC,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;IAC7C,CAAC;IAEM,KAAK,CAAC,mBAAmB,CAC/B,KAAa,EACb,MAAc,EACd,SAAiB,IACf,CAAC;CACJ;AA3BD,wCA2BC"}

package/dist/cjs/modules/memory/providers/mongodb/index.d.ts

@@ -0,0 +1,12 @@
+import type { ChatObject, SessionObject } from "../../../../types/memory.js";
+import { BaseMemory } from "../../base/index.js";
+export declare class MongoDBMemory extends BaseMemory {
+    private isConnected;
+    constructor(uri: string);
+    connect(uri: string): Promise<void>;
+    disconnect(): Promise<void>;
+    getSessionHistory(sessionId: string): Promise<SessionObject>;
+    updateSessionHistory(sessionId: string, chat: ChatObject): Promise<void>;
+    storeQueryAndIntent(query: string, intent: string, sessionId: string): Promise<void>;
+}
+//# sourceMappingURL=index.d.ts.map

package/dist/cjs/modules/memory/providers/mongodb/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../../../src/modules/memory/providers/mongodb/index.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,UAAU,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AAEnE,OAAO,EAAE,UAAU,EAAE,MAAM,qBAAqB,CAAC;AAOjD,qBAAa,aAAc,SAAQ,UAAU;IAC5C,OAAO,CAAC,WAAW,CAAS;gBAEhB,GAAG,EAAE,MAAM;IAKV,OAAO,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAenC,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IAe3B,iBAAiB,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,aAAa,CAAC;IAqB5D,oBAAoB,CAChC,SAAS,EAAE,MAAM,EACjB,IAAI,EAAE,UAAU,GACd,OAAO,CAAC,IAAI,CAAC;IAaH,mBAAmB,CAC/B,KAAK,EAAE,MAAM,EACb,MAAM,EAAE,MAAM,EACd,SAAS,EAAE,MAAM,GACf,OAAO,CAAC,IAAI,CAAC;CAiBhB"}

package/dist/cjs/modules/memory/providers/mongodb/index.js

@@ -0,0 +1,91 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.MongoDBMemory = void 0;
+const mongoose_1 = __importDefault(require("mongoose"));
+const logger_js_1 = require("../../../../utils/logger.js");
+const index_js_1 = require("../../base/index.js");
+const chats_model_js_1 = require("./models/chats.model.js");
+class MongoDBMemory extends index_js_1.BaseMemory {
+    constructor(uri) {
+        super();
+        this.isConnected = false;
+        this.connect(uri);
+    }
+    async connect(uri) {
+        if (this.isConnected) {
+            return;
+        }
+        try {
+            await mongoose_1.default.connect(uri);
+            this.isConnected = true;
+            logger_js_1.loggers.agent.info("MongoDB connected successfully");
+        }
+        catch (error) {
+            logger_js_1.loggers.agent.error("Failed to connect to MongoDB:", error);
+            throw error;
+        }
+    }
+    async disconnect() {
+        if (!this.isConnected) {
+            return;
+        }
+        try {
+            await mongoose_1.default.disconnect();
+            this.isConnected = false;
+            logger_js_1.loggers.agent.info("MongoDB disconnected successfully");
+        }
+        catch (error) {
+            logger_js_1.loggers.agent.error("Failed to disconnect from MongoDB:", error);
+            throw error;
+        }
+    }
+    async getSessionHistory(sessionId) {
+        const chats = await chats_model_js_1.ChatModel.find({ sessionId }).sort({
+            timestamp: 1,
+        });
+        logger_js_1.loggers.agent.info(`Found ${chats.length} chats for session ${sessionId}`);
+        const sessionObject = { chats: {} };
+        chats.forEach((chat) => {
+            const chatId = chat._id?.toString() || chat.id;
+            sessionObject.chats[chatId] = {
+                role: chat.role,
+                content: chat.content,
+                timestamp: chat.timestamp,
+                metadata: chat.metadata,
+            };
+        });
+        return sessionObject;
+    }
+    async updateSessionHistory(sessionId, chat) {
+        logger_js_1.loggers.agent.info(`Updating session history for session ${sessionId}`);
+        logger_js_1.loggers.agent.info(`Chat: ${JSON.stringify(chat)}`);
+        await chats_model_js_1.ChatModel.create({
+            sessionId,
+            role: chat.role,
+            content: chat.content,
+            timestamp: chat.timestamp,
+            metadata: chat.metadata,
+        });
+    }
+    async storeQueryAndIntent(query, intent, sessionId) {
+        // Intent 정보를 metadata에 저장
+        const chat = {
+            role: chats_model_js_1.ChatRole.USER,
+            content: {
+                type: "text",
+                parts: [query],
+            },
+            timestamp: Date.now(),
+            metadata: {
+                intent,
+                query,
+            },
+        };
+        await this.updateSessionHistory(sessionId, chat);
+    }
+}
+exports.MongoDBMemory = MongoDBMemory;
+//# sourceMappingURL=index.js.map

package/dist/cjs/modules/memory/providers/mongodb/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../../../src/modules/memory/providers/mongodb/index.ts"],"names":[],"mappings":";;;;;;AAAA,wDAAgC;AAEhC,iDAA4C;AAC5C,kDAAiD;AACjD,4DAIiC;AAEjC,MAAa,aAAc,SAAQ,qBAAU;IAG5C,YAAY,GAAW;QACtB,KAAK,EAAE,CAAC;QAHD,gBAAW,GAAG,KAAK,CAAC;QAI3B,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;IACnB,CAAC;IAEM,KAAK,CAAC,OAAO,CAAC,GAAW;QAC/B,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YACtB,OAAO;QACR,CAAC;QAED,IAAI,CAAC;YACJ,MAAM,kBAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;YAC5B,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;YACxB,mBAAO,CAAC,KAAK,CAAC,IAAI,CAAC,gCAAgC,CAAC,CAAC;QACtD,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,mBAAO,CAAC,KAAK,CAAC,KAAK,CAAC,+BAA+B,EAAE,KAAK,CAAC,CAAC;YAC5D,MAAM,KAAK,CAAC;QACb,CAAC;IACF,CAAC;IAEM,KAAK,CAAC,UAAU;QACtB,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YACvB,OAAO;QACR,CAAC;QAED,IAAI,CAAC;YACJ,MAAM,kBAAQ,CAAC,UAAU,EAAE,CAAC;YAC5B,IAAI,CAAC,WAAW,GAAG,KAAK,CAAC;YACzB,mBAAO,CAAC,KAAK,CAAC,IAAI,CAAC,mCAAmC,CAAC,CAAC;QACzD,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,mBAAO,CAAC,KAAK,CAAC,KAAK,CAAC,oCAAoC,EAAE,KAAK,CAAC,CAAC;YACjE,MAAM,KAAK,CAAC;QACb,CAAC;IACF,CAAC;IAEM,KAAK,CAAC,iBAAiB,CAAC,SAAiB;QAC/C,MAAM,KAAK,GAAG,MAAM,0BAAS,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC,IAAI,CAAC;YACtD,SAAS,EAAE,CAAC;SACZ,CAAC,CAAC;QAEH,mBAAO,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,KAAK,CAAC,MAAM,sBAAsB,SAAS,EAAE,CAAC,CAAC;QAE3E,MAAM,aAAa,GAAkB,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;QACnD,KAAK,CAAC,OAAO,CAAC,CAAC,IAAkB,EAAE,EAAE;YACpC,MAAM,MAAM,GAAG,IAAI,CAAC,GAAG,EAAE,QAAQ,EAAE,IAAI,IAAI,CAAC,EAAE,CAAC;YAC/C,aAAa,CAAC,KAAK,CAAC,MAAM,CAAC,GAAG;gBAC7B,IAAI,EAAE,IAAI,CAAC,IAAgB;gBAC3B,OAAO,EAAE,IAAI,CAAC,OAAO;gBACrB,SAAS,EAAE,IAAI,CAAC,SAAS;gBACzB,QAAQ,EAAE,IAAI,CAAC,QAAQ;aACvB,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,OAAO,aAAa,CAAC;IACtB,CAAC;IAEM,KAAK,CAAC,oBAAoB,CAChC,SAAiB,EACjB,IAAgB;QAEhB,mBAAO,CAAC,KAAK,CAAC,IAAI,CAAC,wCAAwC,SAAS,EAAE,CAAC,CAAC;QACxE,mBAAO,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAEpD,MAAM,0BAAS,CAAC,MAAM,CAAC;YACtB,SAAS;YACT,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,QAAQ,EAAE,IAAI,CAAC,QAAQ;SACvB,CAAC,CAAC;IACJ,CAAC;IAEM,KAAK,CAAC,mBAAmB,CAC/B,KAAa,EACb,MAAc,EACd,SAAiB;QAEjB,0BAA0B;QAC1B,MAAM,IAAI,GAAe;YACxB,IAAI,EAAE,yBAAQ,CAAC,IAAI;YACnB,OAAO,EAAE;gBACR,IAAI,EAAE,MAAM;gBACZ,KAAK,EAAE,CAAC,KAAK,CAAC;aACd;YACD,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;YACrB,QAAQ,EAAE;gBACT,MAAM;gBACN,KAAK;aACL;SACD,CAAC;QAEF,MAAM,IAAI,CAAC,oBAAoB,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;IAClD,CAAC;CACD;AAhGD,sCAgGC"}

package/dist/cjs/modules/memory/providers/mongodb/models/chats.model.d.ts

@@ -0,0 +1,26 @@
+import mongoose, { type Document } from "mongoose";
+export declare enum ChatRole {
+    USER = "USER",
+    SYSTEM = "SYSTEM",
+    MODEL = "MODEL"
+}
+export interface ChatDocument extends Document {
+    sessionId: string;
+    role: ChatRole;
+    content: {
+        type: string;
+        parts: any[];
+    };
+    timestamp: number;
+    metadata?: {
+        [key: string]: unknown;
+    };
+    createdAt: Date;
+    updatedAt: Date;
+}
+export declare const ChatModel: mongoose.Model<ChatDocument, {}, {}, {}, mongoose.Document<unknown, {}, ChatDocument, {}> & ChatDocument & Required<{
+    _id: unknown;
+}> & {
+    __v: number;
+}, any>;
+//# sourceMappingURL=chats.model.d.ts.map

package/dist/cjs/modules/memory/providers/mongodb/models/chats.model.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"chats.model.d.ts","sourceRoot":"","sources":["../../../../../../../src/modules/memory/providers/mongodb/models/chats.model.ts"],"names":[],"mappings":"AAAA,OAAO,QAAQ,EAAE,EAAE,KAAK,QAAQ,EAAU,MAAM,UAAU,CAAC;AAG3D,oBAAY,QAAQ;IACnB,IAAI,SAAS;IACb,MAAM,WAAW;IACjB,KAAK,UAAU;CACf;AA2CD,MAAM,WAAW,YAAa,SAAQ,QAAQ;IAC7C,SAAS,EAAE,MAAM,CAAC;IAClB,IAAI,EAAE,QAAQ,CAAC;IACf,OAAO,EAAE;QACR,IAAI,EAAE,MAAM,CAAC;QACb,KAAK,EAAE,GAAG,EAAE,CAAC;KACb,CAAC;IACF,SAAS,EAAE,MAAM,CAAC;IAClB,QAAQ,CAAC,EAAE;QAAE,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAA;KAAE,CAAC;IACtC,SAAS,EAAE,IAAI,CAAC;IAChB,SAAS,EAAE,IAAI,CAAC;CAChB;AAGD,eAAO,MAAM,SAAS;;;;OAAyD,CAAC"}

package/dist/cjs/modules/memory/providers/mongodb/models/chats.model.js

@@ -0,0 +1,79 @@
+"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.ChatModel = exports.ChatRole = void 0;
+const mongoose_1 = __importStar(require("mongoose"));
+// ChatRole enum
+var ChatRole;
+(function (ChatRole) {
+    ChatRole["USER"] = "USER";
+    ChatRole["SYSTEM"] = "SYSTEM";
+    ChatRole["MODEL"] = "MODEL";
+})(ChatRole || (exports.ChatRole = ChatRole = {}));
+// ChatContentObject schema
+const ChatContentObjectSchema = new mongoose_1.Schema({
+    type: { type: String, required: true },
+    parts: { type: [mongoose_1.Schema.Types.Mixed], required: true },
+}, { _id: false });
+// ChatObject schema - 개별 문서로 저장
+const ChatObjectSchema = new mongoose_1.Schema({
+    sessionId: {
+        type: String,
+        required: true,
+        index: true,
+    },
+    role: {
+        type: String,
+        enum: Object.values(ChatRole),
+        required: true,
+    },
+    content: {
+        type: ChatContentObjectSchema,
+        required: true,
+    },
+    timestamp: {
+        type: Number,
+        required: true,
+    },
+    metadata: {
+        type: mongoose_1.Schema.Types.Mixed,
+        default: {},
+    },
+}, {
+    timestamps: true,
+});
+// Export the model
+exports.ChatModel = mongoose_1.default.model("Chat", ChatObjectSchema);
+//# sourceMappingURL=chats.model.js.map

package/dist/cjs/modules/memory/providers/mongodb/models/chats.model.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"chats.model.js","sourceRoot":"","sources":["../../../../../../../src/modules/memory/providers/mongodb/models/chats.model.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,qDAA2D;AAE3D,gBAAgB;AAChB,IAAY,QAIX;AAJD,WAAY,QAAQ;IACnB,yBAAa,CAAA;IACb,6BAAiB,CAAA;IACjB,2BAAe,CAAA;AAChB,CAAC,EAJW,QAAQ,wBAAR,QAAQ,QAInB;AAED,2BAA2B;AAC3B,MAAM,uBAAuB,GAAG,IAAI,iBAAM,CACzC;IACC,IAAI,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,QAAQ,EAAE,IAAI,EAAE;IACtC,KAAK,EAAE,EAAE,IAAI,EAAE,CAAC,iBAAM,CAAC,KAAK,CAAC,KAAK,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE;CACrD,EACD,EAAE,GAAG,EAAE,KAAK,EAAE,CACd,CAAC;AAEF,gCAAgC;AAChC,MAAM,gBAAgB,GAAG,IAAI,iBAAM,CAClC;IACC,SAAS,EAAE;QACV,IAAI,EAAE,MAAM;QACZ,QAAQ,EAAE,IAAI;QACd,KAAK,EAAE,IAAI;KACX;IACD,IAAI,EAAE;QACL,IAAI,EAAE,MAAM;QACZ,IAAI,EAAE,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC;QAC7B,QAAQ,EAAE,IAAI;KACd;IACD,OAAO,EAAE;QACR,IAAI,EAAE,uBAAuB;QAC7B,QAAQ,EAAE,IAAI;KACd;IACD,SAAS,EAAE;QACV,IAAI,EAAE,MAAM;QACZ,QAAQ,EAAE,IAAI;KACd;IACD,QAAQ,EAAE;QACT,IAAI,EAAE,iBAAM,CAAC,KAAK,CAAC,KAAK;QACxB,OAAO,EAAE,EAAE;KACX;CACD,EACD;IACC,UAAU,EAAE,IAAI;CAChB,CACD,CAAC;AAgBF,mBAAmB;AACN,QAAA,SAAS,GAAG,kBAAQ,CAAC,KAAK,CAAe,MAAM,EAAE,gBAAgB,CAAC,CAAC"}

package/dist/cjs/modules/memory/providers/mongodb/models/intentTriggeringInfos.model.d.ts

@@ -0,0 +1,27 @@
+import mongoose, { type Document } from "mongoose";
+export interface IntentTriggeringInfoDocument extends Document {
+    context: {
+        messages: Array<{
+            role: string;
+            content: string | Array<{
+                type: "text";
+                text: string;
+            } | {
+                type: "image_url";
+                image_url: {
+                    url: string;
+                };
+            }>;
+        }>;
+    };
+    intent: {
+        name: string;
+        description: string;
+    };
+}
+export declare const IntentTriggeringInfoModel: mongoose.Model<IntentTriggeringInfoDocument, {}, {}, {}, mongoose.Document<unknown, {}, IntentTriggeringInfoDocument, {}> & IntentTriggeringInfoDocument & Required<{
+    _id: unknown;
+}> & {
+    __v: number;
+}, any>;
+//# sourceMappingURL=intentTriggeringInfos.model.d.ts.map

package/dist/cjs/modules/memory/providers/mongodb/models/intentTriggeringInfos.model.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"intentTriggeringInfos.model.d.ts","sourceRoot":"","sources":["../../../../../../../src/modules/memory/providers/mongodb/models/intentTriggeringInfos.model.ts"],"names":[],"mappings":"AAAA,OAAO,QAAQ,EAAE,EAAE,KAAK,QAAQ,EAAU,MAAM,UAAU,CAAC;AAiB3D,MAAM,WAAW,4BAA6B,SAAQ,QAAQ;IAC7D,OAAO,EAAE;QACR,QAAQ,EAAE,KAAK,CAAC;YACf,IAAI,EAAE,MAAM,CAAC;YACb,OAAO,EACJ,MAAM,GACN,KAAK,CACH;gBAAE,IAAI,EAAE,MAAM,CAAC;gBAAC,IAAI,EAAE,MAAM,CAAA;aAAE,GAC9B;gBAAE,IAAI,EAAE,WAAW,CAAC;gBAAC,SAAS,EAAE;oBAAE,GAAG,EAAE,MAAM,CAAA;iBAAE,CAAA;aAAE,CAClD,CAAC;SACL,CAAC,CAAC;KACH,CAAC;IACF,MAAM,EAAE;QACP,IAAI,EAAE,MAAM,CAAC;QACb,WAAW,EAAE,MAAM,CAAC;KACpB,CAAC;CACF;AAYD,eAAO,MAAM,yBAAyB;;;;OAIpC,CAAC"}

package/dist/cjs/modules/memory/providers/mongodb/models/intentTriggeringInfos.model.js

@@ -0,0 +1,59 @@
+"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.IntentTriggeringInfoModel = void 0;
+const mongoose_1 = __importStar(require("mongoose"));
+const MessageSchema = new mongoose_1.Schema({
+    role: {
+        type: String,
+        enum: ["system", "user", "assistant", "tool", "function"],
+        required: true,
+    },
+    content: {
+        type: mongoose_1.Schema.Types.Mixed, // string 또는 object array
+        required: true,
+    },
+}, { _id: false });
+const IntentTriggeringInfoSchema = new mongoose_1.Schema({
+    context: {
+        messages: [MessageSchema],
+    },
+    intent: {
+        name: { type: String, required: true },
+        description: { type: String, required: true },
+    },
+});
+exports.IntentTriggeringInfoModel = mongoose_1.default.model("IntentTriggeringInfo", IntentTriggeringInfoSchema);
+//# sourceMappingURL=intentTriggeringInfos.model.js.map

package/dist/cjs/modules/memory/providers/mongodb/models/intentTriggeringInfos.model.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"intentTriggeringInfos.model.js","sourceRoot":"","sources":["../../../../../../../src/modules/memory/providers/mongodb/models/intentTriggeringInfos.model.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,qDAA2D;AAE3D,MAAM,aAAa,GAAG,IAAI,iBAAM,CAC/B;IACC,IAAI,EAAE;QACL,IAAI,EAAE,MAAM;QACZ,IAAI,EAAE,CAAC,QAAQ,EAAE,MAAM,EAAE,WAAW,EAAE,MAAM,EAAE,UAAU,CAAC;QACzD,QAAQ,EAAE,IAAI;KACd;IACD,OAAO,EAAE;QACR,IAAI,EAAE,iBAAM,CAAC,KAAK,CAAC,KAAK,EAAE,yBAAyB;QACnD,QAAQ,EAAE,IAAI;KACd;CACD,EACD,EAAE,GAAG,EAAE,KAAK,EAAE,CACd,CAAC;AAoBF,MAAM,0BAA0B,GAAG,IAAI,iBAAM,CAA+B;IAC3E,OAAO,EAAE;QACR,QAAQ,EAAE,CAAC,aAAa,CAAC;KACzB;IACD,MAAM,EAAE;QACP,IAAI,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,QAAQ,EAAE,IAAI,EAAE;QACtC,WAAW,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,QAAQ,EAAE,IAAI,EAAE;KAC7C;CACD,CAAC,CAAC;AAEU,QAAA,yBAAyB,GACrC,kBAAQ,CAAC,KAAK,CACb,sBAAsB,EACtB,0BAA0B,CAC1B,CAAC"}

package/dist/cjs/modules/models/base/index.d.ts

@@ -0,0 +1,57 @@
+import type { SessionObject } from "../../../types/memory.js";
+import type { FetchResponse, IAgentTool } from "../../../types/tool.js";
+/**
+ * Abstract base class for AI model implementations.
+ *
+ * Provides a common interface for different AI model providers (OpenAI, Gemini, etc.)
+ * to integrate with the AIN-ADK framework. Each model implementation must handle
+ * message formatting, tool conversion, and API communication.
+ *
+ * @typeParam MessageType - The message format used by the specific model API
+ * @typeParam FunctionType - The function/tool format used by the specific model API
+ */
+export declare abstract class BaseModel<MessageType, FunctionType> {
+    /**
+     * Generates an array of messages from session history and current query.
+     *
+     * @param sessionHistory - Previous conversation history
+     * @param query - Current user query
+     * @param systemPrompt - Optional system prompt to set context
+     * @returns Array of messages formatted for the specific model API
+     */
+    abstract generateMessages(params: {
+        query: string;
+        sessionHistory?: SessionObject;
+        systemPrompt?: string;
+    }): MessageType[];
+    /**
+     * Appends a new message to the existing message array.
+     *
+     * @param messages - Existing message array to expand
+     * @param message - New message content to append
+     */
+    abstract appendMessages(messages: MessageType[], message: string): void;
+    /**
+     * Converts protocol-agnostic tools to model-specific function format.
+     *
+     * @param tools - Array of agent tools from MCP or A2A sources
+     * @returns Array of functions in the format required by the model API
+     */
+    abstract convertToolsToFunctions(tools: IAgentTool[]): FunctionType[];
+    /**
+     * Fetches a response from the model API without tool support.
+     *
+     * @param messages - Array of messages to send to the model
+     * @returns Promise resolving to the model's response
+     */
+    abstract fetch(messages: MessageType[]): Promise<FetchResponse>;
+    /**
+     * Fetches a response from the model API with tool/function support.
+     *
+     * @param messages - Array of messages to send to the model
+     * @param functions - Array of available functions/tools the model can call
+     * @returns Promise resolving to the model's response, possibly including tool calls
+     */
+    abstract fetchWithContextMessage(messages: MessageType[], functions: FunctionType[]): Promise<FetchResponse>;
+}
+//# sourceMappingURL=index.d.ts.map

package/dist/cjs/modules/models/base/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../../src/modules/models/base/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AACvD,OAAO,KAAK,EAAE,aAAa,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAEjE;;;;;;;;;GASG;AACH,8BAAsB,SAAS,CAAC,WAAW,EAAE,YAAY;IACxD;;;;;;;OAOG;IACH,QAAQ,CAAC,gBAAgB,CAAC,MAAM,EAAE;QACjC,KAAK,EAAE,MAAM,CAAC;QACd,cAAc,CAAC,EAAE,aAAa,CAAC;QAC/B,YAAY,CAAC,EAAE,MAAM,CAAC;KACtB,GAAG,WAAW,EAAE;IAEjB;;;;;OAKG;IACH,QAAQ,CAAC,cAAc,CAAC,QAAQ,EAAE,WAAW,EAAE,EAAE,OAAO,EAAE,MAAM,GAAG,IAAI;IAEvE;;;;;OAKG;IACH,QAAQ,CAAC,uBAAuB,CAAC,KAAK,EAAE,UAAU,EAAE,GAAG,YAAY,EAAE;IAErE;;;;;OAKG;IACH,QAAQ,CAAC,KAAK,CAAC,QAAQ,EAAE,WAAW,EAAE,GAAG,OAAO,CAAC,aAAa,CAAC;IAE/D;;;;;;OAMG;IACH,QAAQ,CAAC,uBAAuB,CAC/B,QAAQ,EAAE,WAAW,EAAE,EACvB,SAAS,EAAE,YAAY,EAAE,GACvB,OAAO,CAAC,aAAa,CAAC;CACzB"}

package/dist/cjs/modules/models/base/index.js

@@ -0,0 +1,17 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.BaseModel = void 0;
+/**
+ * Abstract base class for AI model implementations.
+ *
+ * Provides a common interface for different AI model providers (OpenAI, Gemini, etc.)
+ * to integrate with the AIN-ADK framework. Each model implementation must handle
+ * message formatting, tool conversion, and API communication.
+ *
+ * @typeParam MessageType - The message format used by the specific model API
+ * @typeParam FunctionType - The function/tool format used by the specific model API
+ */
+class BaseModel {
+}
+exports.BaseModel = BaseModel;
+//# sourceMappingURL=index.js.map

package/dist/cjs/modules/models/base/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../../src/modules/models/base/index.ts"],"names":[],"mappings":";;;AAGA;;;;;;;;;GASG;AACH,MAAsB,SAAS;CAkD9B;AAlDD,8BAkDC"}

package/dist/cjs/modules/models/base.model.d.ts

@@ -0,0 +1,57 @@
+import type { SessionObject } from "../../types/memory.js";
+import type { FetchResponse, IAgentTool } from "../../types/tool.js";
+/**
+ * Abstract base class for AI model implementations.
+ *
+ * Provides a common interface for different AI model providers (OpenAI, Gemini, etc.)
+ * to integrate with the AIN-ADK framework. Each model implementation must handle
+ * message formatting, tool conversion, and API communication.
+ *
+ * @typeParam MessageType - The message format used by the specific model API
+ * @typeParam FunctionType - The function/tool format used by the specific model API
+ */
+export declare abstract class BaseModel<MessageType, FunctionType> {
+    /**
+     * Generates an array of messages from session history and current query.
+     *
+     * @param sessionHistory - Previous conversation history
+     * @param query - Current user query
+     * @param systemPrompt - Optional system prompt to set context
+     * @returns Array of messages formatted for the specific model API
+     */
+    abstract generateMessages(params: {
+        query: string;
+        sessionHistory?: SessionObject;
+        systemPrompt?: string;
+    }): MessageType[];
+    /**
+     * Appends a new message to the existing message array.
+     *
+     * @param messages - Existing message array to expand
+     * @param message - New message content to append
+     */
+    abstract appendMessages(messages: MessageType[], message: string): void;
+    /**
+     * Converts protocol-agnostic tools to model-specific function format.
+     *
+     * @param tools - Array of agent tools from MCP or A2A sources
+     * @returns Array of functions in the format required by the model API
+     */
+    abstract convertToolsToFunctions(tools: IAgentTool[]): FunctionType[];
+    /**
+     * Fetches a response from the model API without tool support.
+     *
+     * @param messages - Array of messages to send to the model
+     * @returns Promise resolving to the model's response
+     */
+    abstract fetch(messages: MessageType[]): Promise<FetchResponse>;
+    /**
+     * Fetches a response from the model API with tool/function support.
+     *
+     * @param messages - Array of messages to send to the model
+     * @param functions - Array of available functions/tools the model can call
+     * @returns Promise resolving to the model's response, possibly including tool calls
+     */
+    abstract fetchWithContextMessage(messages: MessageType[], functions: FunctionType[]): Promise<FetchResponse>;
+}
+//# sourceMappingURL=base.model.d.ts.map

package/dist/cjs/modules/models/base.model.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"base.model.d.ts","sourceRoot":"","sources":["../../../../src/modules/models/base.model.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AACvD,OAAO,KAAK,EAAE,aAAa,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAEjE;;;;;;;;;GASG;AACH,8BAAsB,SAAS,CAAC,WAAW,EAAE,YAAY;IACxD;;;;;;;OAOG;IACH,QAAQ,CAAC,gBAAgB,CAAC,MAAM,EAAE;QACjC,KAAK,EAAE,MAAM,CAAC;QACd,cAAc,CAAC,EAAE,aAAa,CAAC;QAC/B,YAAY,CAAC,EAAE,MAAM,CAAC;KACtB,GAAG,WAAW,EAAE;IAEjB;;;;;OAKG;IACH,QAAQ,CAAC,cAAc,CAAC,QAAQ,EAAE,WAAW,EAAE,EAAE,OAAO,EAAE,MAAM,GAAG,IAAI;IAEvE;;;;;OAKG;IACH,QAAQ,CAAC,uBAAuB,CAAC,KAAK,EAAE,UAAU,EAAE,GAAG,YAAY,EAAE;IAErE;;;;;OAKG;IACH,QAAQ,CAAC,KAAK,CAAC,QAAQ,EAAE,WAAW,EAAE,GAAG,OAAO,CAAC,aAAa,CAAC;IAE/D;;;;;;OAMG;IACH,QAAQ,CAAC,uBAAuB,CAC/B,QAAQ,EAAE,WAAW,EAAE,EACvB,SAAS,EAAE,YAAY,EAAE,GACvB,OAAO,CAAC,aAAa,CAAC;CACzB"}

package/dist/cjs/modules/models/base.model.js

@@ -0,0 +1,17 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.BaseModel = void 0;
+/**
+ * Abstract base class for AI model implementations.
+ *
+ * Provides a common interface for different AI model providers (OpenAI, Gemini, etc.)
+ * to integrate with the AIN-ADK framework. Each model implementation must handle
+ * message formatting, tool conversion, and API communication.
+ *
+ * @typeParam MessageType - The message format used by the specific model API
+ * @typeParam FunctionType - The function/tool format used by the specific model API
+ */
+class BaseModel {
+}
+exports.BaseModel = BaseModel;
+//# sourceMappingURL=base.model.js.map

package/dist/cjs/modules/models/base.model.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"base.model.js","sourceRoot":"","sources":["../../../../src/modules/models/base.model.ts"],"names":[],"mappings":";;;AAGA;;;;;;;;;GASG;AACH,MAAsB,SAAS;CAkD9B;AAlDD,8BAkDC"}