@ainetwork/adk-provider-memory-mongodb 0.1.2 → 0.1.3

package/dist/chunk-JCE5NAY5.js

@@ -0,0 +1,29 @@
+// models/intent.model.ts
+import mongoose, { Schema } from "mongoose";
+var IntentObjectSchema = new Schema(
+  {
+    name: {
+      type: String,
+      required: true,
+      index: true
+    },
+    description: {
+      type: String,
+      required: true
+    },
+    prompt: {
+      type: String,
+      required: false
+    },
+    llm: {
+      type: String,
+      required: false
+    }
+  }
+);
+var IntentModel = mongoose.model("Intent", IntentObjectSchema);
+
+export {
+  IntentModel
+};
+//# sourceMappingURL=chunk-JCE5NAY5.js.map

package/dist/chunk-JCE5NAY5.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../models/intent.model.ts"],"sourcesContent":["import mongoose, { type Document, Schema } from \"mongoose\";\n\nconst IntentObjectSchema = new Schema(\n\t{\n\t\tname: {\n\t\t\ttype: String,\n\t\t\trequired: true,\n\t\t\tindex: true,\n\t\t},\n\t\tdescription: {\n\t\t\ttype: String,\n\t\t\trequired: true,\n\t\t},\n\t\tprompt: {\n\t\t\ttype: String,\n\t\t\trequired: false,\n\t\t},\n\t\tllm: {\n\t\t\ttype: String,\n\t\t\trequired: false,\n\t\t},\n\t}\n);\n\nexport interface IntentDocument extends Document {\n\tname: string;\n\tdescription: string;\n\tprompt?: string;\n\tllm?: string;\n}\n\nexport const IntentModel = mongoose.model<IntentDocument>(\"Intent\", IntentObjectSchema);"],"mappings":";AAAA,OAAO,YAA2B,cAAc;AAEhD,IAAM,qBAAqB,IAAI;AAAA,EAC9B;AAAA,IACC,MAAM;AAAA,MACL,MAAM;AAAA,MACN,UAAU;AAAA,MACV,OAAO;AAAA,IACR;AAAA,IACA,aAAa;AAAA,MACZ,MAAM;AAAA,MACN,UAAU;AAAA,IACX;AAAA,IACA,QAAQ;AAAA,MACP,MAAM;AAAA,MACN,UAAU;AAAA,IACX;AAAA,IACA,KAAK;AAAA,MACJ,MAAM;AAAA,MACN,UAAU;AAAA,IACX;AAAA,EACD;AACD;AASO,IAAM,cAAc,SAAS,MAAsB,UAAU,kBAAkB;","names":[]}

package/dist/{chunk-2SB2M62A.js → chunk-SS7OXU6N.js}

@@ -1,11 +1,37 @@
 // models/chats.model.ts
-import mongoose, { Schema } from "mongoose";
+import { Schema } from "mongoose";
 var ChatRole = /* @__PURE__ */ ((ChatRole2) => {
   ChatRole2["USER"] = "USER";
   ChatRole2["SYSTEM"] = "SYSTEM";
   ChatRole2["MODEL"] = "MODEL";
   return ChatRole2;
 })(ChatRole || {});
+var SessionObjectSchema = new Schema(
+  {
+    sessionId: {
+      type: String,
+      required: true,
+      index: true
+    },
+    userId: {
+      type: String,
+      required: true,
+      index: true
+    },
+    title: {
+      type: String,
+      required: false
+    },
+    created_at: {
+      type: Number,
+      required: true
+    },
+    updated_at: {
+      type: Number,
+      required: true
+    }
+  }
+);
 var ChatContentObjectSchema = new Schema(
   {
     type: { type: String, required: true },
@@ -37,15 +63,13 @@ var ChatObjectSchema = new Schema(
       type: Schema.Types.Mixed,
       default: {}
     }
-  },
-  {
-    timestamps: true
   }
 );
-var ChatModel = mongoose.model("Chat", ChatObjectSchema);
 
 export {
   ChatRole,
-  ChatModel
+  SessionObjectSchema,
+  ChatContentObjectSchema,
+  ChatObjectSchema
 };
-//# sourceMappingURL=chunk-2SB2M62A.js.map
+//# sourceMappingURL=chunk-SS7OXU6N.js.map

package/dist/chunk-SS7OXU6N.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../models/chats.model.ts"],"sourcesContent":["import { type Document, Schema } from \"mongoose\";\n\n// ChatRole enum\nexport enum ChatRole {\n\tUSER = \"USER\",\n\tSYSTEM = \"SYSTEM\",\n\tMODEL = \"MODEL\",\n}\n\nexport const SessionObjectSchema = new Schema(\n\t{\n\t\tsessionId: {\n\t\t\ttype: String,\n\t\t\trequired: true,\n\t\t\tindex: true,\n\t\t},\n\t\tuserId: {\n\t\t\ttype: String,\n\t\t\trequired: true,\n\t\t\tindex: true,\n\t\t},\n\t\ttitle: {\n\t\t\ttype: String,\n\t\t\trequired: false,\n\t\t},\n\t\tcreated_at: {\n\t\t\ttype: Number,\n\t\t\trequired: true,\n\t\t},\n\t\tupdated_at: {\n\t\t\ttype: Number,\n\t\t\trequired: true,\n\t\t}\n\t},\n);\n\nexport interface SessionDocument extends Document {\n\tsessionId: string;\n\tuserId: string;\n\ttitle?: string;\n\tcreated_at: number;\n\tupdated_at: number;\n}\n\n// ChatContentObject schema\nexport const ChatContentObjectSchema = new Schema(\n\t{\n\t\ttype: { type: String, required: true },\n\t\tparts: { type: [Schema.Types.Mixed], required: true },\n\t},\n\t{ _id: false },\n);\n\n// ChatObject schema - 개별 문서로 저장\nexport const ChatObjectSchema = new Schema(\n\t{\n\t\tsessionId: {\n\t\t\ttype: String,\n\t\t\trequired: true,\n\t\t\tindex: true,\n\t\t},\n\t\trole: {\n\t\t\ttype: String,\n\t\t\tenum: Object.values(ChatRole),\n\t\t\trequired: true,\n\t\t},\n\t\tcontent: {\n\t\t\ttype: ChatContentObjectSchema,\n\t\t\trequired: true,\n\t\t},\n\t\ttimestamp: {\n\t\t\ttype: Number,\n\t\t\trequired: true,\n\t\t},\n\t\tmetadata: {\n\t\t\ttype: Schema.Types.Mixed,\n\t\t\tdefault: {},\n\t\t},\n\t},\n);\n\n// Chat Document interface\nexport interface ChatDocument extends Document {\n\tsessionId: string;\n\trole: ChatRole;\n\tcontent: {\n\t\ttype: string;\n\t\tparts: any[];\n\t};\n\ttimestamp: number;\n\tmetadata?: { [key: string]: unknown };\n\tcreatedAt: Date;\n\tupdatedAt: Date;\n}\n"],"mappings":";AAAA,SAAwB,cAAc;AAG/B,IAAK,WAAL,kBAAKA,cAAL;AACN,EAAAA,UAAA,UAAO;AACP,EAAAA,UAAA,YAAS;AACT,EAAAA,UAAA,WAAQ;AAHG,SAAAA;AAAA,GAAA;AAML,IAAM,sBAAsB,IAAI;AAAA,EACtC;AAAA,IACC,WAAW;AAAA,MACV,MAAM;AAAA,MACN,UAAU;AAAA,MACV,OAAO;AAAA,IACR;AAAA,IACA,QAAQ;AAAA,MACP,MAAM;AAAA,MACN,UAAU;AAAA,MACV,OAAO;AAAA,IACR;AAAA,IACA,OAAO;AAAA,MACN,MAAM;AAAA,MACN,UAAU;AAAA,IACX;AAAA,IACA,YAAY;AAAA,MACX,MAAM;AAAA,MACN,UAAU;AAAA,IACX;AAAA,IACA,YAAY;AAAA,MACX,MAAM;AAAA,MACN,UAAU;AAAA,IACX;AAAA,EACD;AACD;AAWO,IAAM,0BAA0B,IAAI;AAAA,EAC1C;AAAA,IACC,MAAM,EAAE,MAAM,QAAQ,UAAU,KAAK;AAAA,IACrC,OAAO,EAAE,MAAM,CAAC,OAAO,MAAM,KAAK,GAAG,UAAU,KAAK;AAAA,EACrD;AAAA,EACA,EAAE,KAAK,MAAM;AACd;AAGO,IAAM,mBAAmB,IAAI;AAAA,EACnC;AAAA,IACC,WAAW;AAAA,MACV,MAAM;AAAA,MACN,UAAU;AAAA,MACV,OAAO;AAAA,IACR;AAAA,IACA,MAAM;AAAA,MACL,MAAM;AAAA,MACN,MAAM,OAAO,OAAO,QAAQ;AAAA,MAC5B,UAAU;AAAA,IACX;AAAA,IACA,SAAS;AAAA,MACR,MAAM;AAAA,MACN,UAAU;AAAA,IACX;AAAA,IACA,WAAW;AAAA,MACV,MAAM;AAAA,MACN,UAAU;AAAA,IACX;AAAA,IACA,UAAU;AAAA,MACT,MAAM,OAAO,MAAM;AAAA,MACnB,SAAS,CAAC;AAAA,IACX;AAAA,EACD;AACD;","names":["ChatRole"]}

package/dist/index.cjs

@@ -30,29 +30,101 @@ var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: tru
 // index.ts
 var index_exports = {};
 __export(index_exports, {
-  MongoDBMemory: () => MongoDBMemory
+  MongoDBIntent: () => MongoDBIntent,
+  MongoDBSession: () => MongoDBSession
 });
 module.exports = __toCommonJS(index_exports);
-var import_modules = require("@ainetwork/adk/modules");
+
+// implements/session.memory.ts
+var import_node_crypto = require("crypto");
+
+// implements/base.memory.ts
+var import_mongoose = require("mongoose");
 var import_logger = require("@ainetwork/adk/utils/logger");
-var import_mongoose2 = __toESM(require("mongoose"), 1);
+var MongoDBMemory = class {
+  _isConnected = false;
+  _uri;
+  _mongoose;
+  constructor(uri) {
+    this._uri = uri;
+    this._mongoose = new import_mongoose.Mongoose();
+  }
+  getInstance() {
+    return this._mongoose;
+  }
+  async connect() {
+    if (this._isConnected) {
+      return;
+    }
+    try {
+      await this._mongoose.connect(this._uri);
+      this._isConnected = true;
+      import_logger.loggers.agent.info("MongoDB connected successfully");
+    } catch (error) {
+      import_logger.loggers.agent.error("Failed to connect to MongoDB:", error);
+      throw error;
+    }
+  }
+  async disconnect() {
+    if (!this.isConnected) {
+      return;
+    }
+    try {
+      await this._mongoose?.disconnect();
+      this._isConnected = false;
+      import_logger.loggers.agent.info("MongoDB disconnected successfully");
+    } catch (error) {
+      import_logger.loggers.agent.error("Failed to disconnect from MongoDB:", error);
+      throw error;
+    }
+  }
+  isConnected() {
+    return this._isConnected;
+  }
+};
 
 // models/chats.model.ts
-var import_mongoose = __toESM(require("mongoose"), 1);
-var ChatRole = /* @__PURE__ */ ((ChatRole2) => {
-  ChatRole2["USER"] = "USER";
-  ChatRole2["SYSTEM"] = "SYSTEM";
-  ChatRole2["MODEL"] = "MODEL";
-  return ChatRole2;
+var import_mongoose2 = require("mongoose");
+var ChatRole = /* @__PURE__ */ ((ChatRole3) => {
+  ChatRole3["USER"] = "USER";
+  ChatRole3["SYSTEM"] = "SYSTEM";
+  ChatRole3["MODEL"] = "MODEL";
+  return ChatRole3;
 })(ChatRole || {});
-var ChatContentObjectSchema = new import_mongoose.Schema(
+var SessionObjectSchema = new import_mongoose2.Schema(
+  {
+    sessionId: {
+      type: String,
+      required: true,
+      index: true
+    },
+    userId: {
+      type: String,
+      required: true,
+      index: true
+    },
+    title: {
+      type: String,
+      required: false
+    },
+    created_at: {
+      type: Number,
+      required: true
+    },
+    updated_at: {
+      type: Number,
+      required: true
+    }
+  }
+);
+var ChatContentObjectSchema = new import_mongoose2.Schema(
   {
     type: { type: String, required: true },
-    parts: { type: [import_mongoose.Schema.Types.Mixed], required: true }
+    parts: { type: [import_mongoose2.Schema.Types.Mixed], required: true }
   },
   { _id: false }
 );
-var ChatObjectSchema = new import_mongoose.Schema(
+var ChatObjectSchema = new import_mongoose2.Schema(
   {
     sessionId: {
       type: String,
@@ -73,54 +145,28 @@ var ChatObjectSchema = new import_mongoose.Schema(
       required: true
     },
     metadata: {
-      type: import_mongoose.Schema.Types.Mixed,
+      type: import_mongoose2.Schema.Types.Mixed,
       default: {}
     }
-  },
-  {
-    timestamps: true
   }
 );
-var ChatModel = import_mongoose.default.model("Chat", ChatObjectSchema);
 
-// index.ts
-var MongoDBMemory = class extends import_modules.BaseMemory {
-  isConnected = false;
+// implements/session.memory.ts
+var import_logger2 = require("@ainetwork/adk/utils/logger");
+var MongoDBSession = class extends MongoDBMemory {
+  chatModel;
+  sessionModel;
   constructor(uri) {
-    super();
-    this.connect(uri);
-  }
-  async connect(uri) {
-    if (this.isConnected) {
-      return;
-    }
-    try {
-      await import_mongoose2.default.connect(uri);
-      this.isConnected = true;
-      import_logger.loggers.agent.info("MongoDB connected successfully");
-    } catch (error) {
-      import_logger.loggers.agent.error("Failed to connect to MongoDB:", error);
-      throw error;
-    }
+    super(uri);
+    const _mongoose = super.getInstance();
+    this.chatModel = _mongoose.model("Chat", ChatObjectSchema);
+    this.sessionModel = _mongoose.model("Session", SessionObjectSchema);
   }
-  async disconnect() {
-    if (!this.isConnected) {
-      return;
-    }
-    try {
-      await import_mongoose2.default.disconnect();
-      this.isConnected = false;
-      import_logger.loggers.agent.info("MongoDB disconnected successfully");
-    } catch (error) {
-      import_logger.loggers.agent.error("Failed to disconnect from MongoDB:", error);
-      throw error;
-    }
-  }
-  async getSessionHistory(sessionId) {
-    const chats = await ChatModel.find({ sessionId }).sort({
+  async getSession(sessionId, userId) {
+    const chats = await this.chatModel.find({ sessionId }).sort({
       timestamp: 1
     });
-    import_logger.loggers.agent.info(`Found ${chats.length} chats for session ${sessionId}`);
+    import_logger2.loggers.agent.debug(`Found ${chats.length} chats for session ${sessionId}`);
     const sessionObject = { chats: {} };
     chats.forEach((chat) => {
       const chatId = chat._id?.toString() || chat.id;
@@ -133,35 +179,137 @@ var MongoDBMemory = class extends import_modules.BaseMemory {
     });
     return sessionObject;
   }
-  async updateSessionHistory(sessionId, chat) {
-    import_logger.loggers.agent.info(`Updating session history for session ${sessionId}`);
-    import_logger.loggers.agent.info(`Chat: ${JSON.stringify(chat)}`);
-    await ChatModel.create({
+  async createSession(userId, sessionId) {
+    await this.sessionModel.create({
       sessionId,
+      userId,
+      updated_at: Date.now(),
+      created_at: Date.now()
+    });
+  }
+  async addChatToSession(userId, sessionId, chat) {
+    const newId = (0, import_node_crypto.randomUUID)();
+    const session = await this.sessionModel.findOne({ sessionId, userId });
+    if (!session) {
+      await this.createSession(userId, sessionId);
+    } else {
+      await this.sessionModel.updateOne({ sessionId, userId }, {
+        updated_at: Date.now()
+      });
+    }
+    await this.chatModel.create({
+      sessionId,
+      chatId: newId,
+      userId,
       role: chat.role,
       content: chat.content,
       timestamp: chat.timestamp,
       metadata: chat.metadata
     });
   }
-  async storeQueryAndIntent(query, intent, sessionId) {
-    const chat = {
-      role: "USER" /* USER */,
-      content: {
-        type: "text",
-        parts: [query]
-      },
-      timestamp: Date.now(),
-      metadata: {
-        intent,
-        query
-      }
-    };
-    await this.updateSessionHistory(sessionId, chat);
+  async deleteSession(userId, sessionId) {
+    const chats = await this.chatModel.find({ userId, sessionId }).sort({
+      timestamp: 1
+    });
+    chats?.forEach((chat) => {
+      chat.deleteOne();
+    });
+    const session = await this.sessionModel.findOne({ sessionId, userId });
+    session?.deleteOne();
+  }
+  async listSessions(userId) {
+    const sessions = await this.sessionModel.find({ userId }).sort({
+      updated_at: -1
+    });
+    const data = sessions.map((session) => {
+      return {
+        sessionId: session.sessionId,
+        title: session.title,
+        updatedAt: session.updated_at
+      };
+    });
+    return data;
+  }
+};
+
+// implements/intent.memory.ts
+var import_node_crypto2 = require("crypto");
+
+// models/intent.model.ts
+var import_mongoose3 = __toESM(require("mongoose"), 1);
+var IntentObjectSchema = new import_mongoose3.Schema(
+  {
+    name: {
+      type: String,
+      required: true,
+      index: true
+    },
+    description: {
+      type: String,
+      required: true
+    },
+    prompt: {
+      type: String,
+      required: false
+    },
+    llm: {
+      type: String,
+      required: false
+    }
+  }
+);
+var IntentModel = import_mongoose3.default.model("Intent", IntentObjectSchema);
+
+// implements/intent.memory.ts
+var MongoDBIntent = class extends MongoDBMemory {
+  async getIntent(intentId) {
+    const intent = await IntentModel.findById(intentId);
+    if (intent) {
+      return {
+        name: intent.name,
+        description: intent.description,
+        prompt: intent.prompt,
+        llm: intent.llm
+      };
+    }
+    return void 0;
+  }
+  async saveIntent(intent) {
+    const newId = (0, import_node_crypto2.randomUUID)();
+    await IntentModel.create({
+      _id: newId,
+      name: intent.name,
+      description: intent.description,
+      prompt: intent.prompt,
+      llm: intent.llm
+    });
+  }
+  async updateIntent(intentId, intent) {
+    await IntentModel.updateOne({
+      _id: intentId
+    }, {
+      name: intent.name,
+      description: intent.description,
+      prompt: intent.prompt,
+      llm: intent.llm
+    });
+  }
+  async deleteIntent(intentId) {
+    await IntentModel.deleteOne({ _id: intentId });
+  }
+  async listIntents() {
+    const intents = await IntentModel.find();
+    return intents.map((intent) => ({
+      name: intent.name,
+      description: intent.description,
+      prompt: intent.prompt,
+      llm: intent.llm
+    }));
   }
 };
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
-  MongoDBMemory
+  MongoDBIntent,
+  MongoDBSession
 });
 //# sourceMappingURL=index.cjs.map

package/dist/index.cjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../index.ts","../models/chats.model.ts"],"sourcesContent":["import { BaseMemory } from \"@ainetwork/adk/modules\";\nimport type { ChatObject, SessionObject } from \"@ainetwork/adk/types/memory\";\nimport { loggers } from \"@ainetwork/adk/utils/logger\";\nimport mongoose from \"mongoose\";\nimport {\n\ttype ChatDocument,\n\tChatModel,\n\tChatRole,\n} from \"./models/chats.model\";\n\nexport class MongoDBMemory extends BaseMemory {\n\tprivate isConnected = false;\n\n\tconstructor(uri: string) {\n\t\tsuper();\n\t\tthis.connect(uri);\n\t}\n\n\tpublic async connect(uri: string): Promise<void> {\n\t\tif (this.isConnected) {\n\t\t\treturn;\n\t\t}\n\n\t\ttry {\n\t\t\tawait mongoose.connect(uri);\n\t\t\tthis.isConnected = true;\n\t\t\tloggers.agent.info(\"MongoDB connected successfully\");\n\t\t} catch (error) {\n\t\t\tloggers.agent.error(\"Failed to connect to MongoDB:\", error);\n\t\t\tthrow error;\n\t\t}\n\t}\n\n\tpublic async disconnect(): Promise<void> {\n\t\tif (!this.isConnected) {\n\t\t\treturn;\n\t\t}\n\n\t\ttry {\n\t\t\tawait mongoose.disconnect();\n\t\t\tthis.isConnected = false;\n\t\t\tloggers.agent.info(\"MongoDB disconnected successfully\");\n\t\t} catch (error) {\n\t\t\tloggers.agent.error(\"Failed to disconnect from MongoDB:\", error);\n\t\t\tthrow error;\n\t\t}\n\t}\n\n\tpublic async getSessionHistory(sessionId: string): Promise<SessionObject> {\n\t\tconst chats = await ChatModel.find({ sessionId }).sort({\n\t\t\ttimestamp: 1,\n\t\t});\n\n\t\tloggers.agent.info(`Found ${chats.length} chats for session ${sessionId}`);\n\n\t\tconst sessionObject: SessionObject = { chats: {} };\n\t\tchats.forEach((chat: ChatDocument) => {\n\t\t\tconst chatId = chat._id?.toString() || chat.id;\n\t\t\tsessionObject.chats[chatId] = {\n\t\t\t\trole: chat.role as ChatRole,\n\t\t\t\tcontent: chat.content,\n\t\t\t\ttimestamp: chat.timestamp,\n\t\t\t\tmetadata: chat.metadata,\n\t\t\t};\n\t\t});\n\n\t\treturn sessionObject;\n\t}\n\n\tpublic async updateSessionHistory(\n\t\tsessionId: string,\n\t\tchat: ChatObject,\n\t): Promise<void> {\n\t\tloggers.agent.info(`Updating session history for session ${sessionId}`);\n\t\tloggers.agent.info(`Chat: ${JSON.stringify(chat)}`);\n\n\t\tawait ChatModel.create({\n\t\t\tsessionId,\n\t\t\trole: chat.role,\n\t\t\tcontent: chat.content,\n\t\t\ttimestamp: chat.timestamp,\n\t\t\tmetadata: chat.metadata,\n\t\t});\n\t}\n\n\tpublic async storeQueryAndIntent(\n\t\tquery: string,\n\t\tintent: string,\n\t\tsessionId: string,\n\t): Promise<void> {\n\t\t// Intent 정보를 metadata에 저장\n\t\tconst chat: ChatObject = {\n\t\t\trole: ChatRole.USER,\n\t\t\tcontent: {\n\t\t\t\ttype: \"text\",\n\t\t\t\tparts: [query],\n\t\t\t},\n\t\t\ttimestamp: Date.now(),\n\t\t\tmetadata: {\n\t\t\t\tintent,\n\t\t\t\tquery,\n\t\t\t},\n\t\t};\n\n\t\tawait this.updateSessionHistory(sessionId, chat);\n\t}\n}\n","import mongoose, { type Document, Schema } from \"mongoose\";\n\n// ChatRole enum\nexport enum ChatRole {\n\tUSER = \"USER\",\n\tSYSTEM = \"SYSTEM\",\n\tMODEL = \"MODEL\",\n}\n\n// ChatContentObject schema\nconst ChatContentObjectSchema = new Schema(\n\t{\n\t\ttype: { type: String, required: true },\n\t\tparts: { type: [Schema.Types.Mixed], required: true },\n\t},\n\t{ _id: false },\n);\n\n// ChatObject schema - 개별 문서로 저장\nconst ChatObjectSchema = new Schema(\n\t{\n\t\tsessionId: {\n\t\t\ttype: String,\n\t\t\trequired: true,\n\t\t\tindex: true,\n\t\t},\n\t\trole: {\n\t\t\ttype: String,\n\t\t\tenum: Object.values(ChatRole),\n\t\t\trequired: true,\n\t\t},\n\t\tcontent: {\n\t\t\ttype: ChatContentObjectSchema,\n\t\t\trequired: true,\n\t\t},\n\t\ttimestamp: {\n\t\t\ttype: Number,\n\t\t\trequired: true,\n\t\t},\n\t\tmetadata: {\n\t\t\ttype: Schema.Types.Mixed,\n\t\t\tdefault: {},\n\t\t},\n\t},\n\t{\n\t\ttimestamps: true,\n\t},\n);\n\n// Chat Document interface\nexport interface ChatDocument extends Document {\n\tsessionId: string;\n\trole: ChatRole;\n\tcontent: {\n\t\ttype: string;\n\t\tparts: any[];\n\t};\n\ttimestamp: number;\n\tmetadata?: { [key: string]: unknown };\n\tcreatedAt: Date;\n\tupdatedAt: Date;\n}\n\n// Export the model\nexport const ChatModel = mongoose.model<ChatDocument>(\"Chat\", ChatObjectSchema);\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,qBAA2B;AAE3B,oBAAwB;AACxB,IAAAA,mBAAqB;;;ACHrB,sBAAgD;AAGzC,IAAK,WAAL,kBAAKC,cAAL;AACN,EAAAA,UAAA,UAAO;AACP,EAAAA,UAAA,YAAS;AACT,EAAAA,UAAA,WAAQ;AAHG,SAAAA;AAAA,GAAA;AAOZ,IAAM,0BAA0B,IAAI;AAAA,EACnC;AAAA,IACC,MAAM,EAAE,MAAM,QAAQ,UAAU,KAAK;AAAA,IACrC,OAAO,EAAE,MAAM,CAAC,uBAAO,MAAM,KAAK,GAAG,UAAU,KAAK;AAAA,EACrD;AAAA,EACA,EAAE,KAAK,MAAM;AACd;AAGA,IAAM,mBAAmB,IAAI;AAAA,EAC5B;AAAA,IACC,WAAW;AAAA,MACV,MAAM;AAAA,MACN,UAAU;AAAA,MACV,OAAO;AAAA,IACR;AAAA,IACA,MAAM;AAAA,MACL,MAAM;AAAA,MACN,MAAM,OAAO,OAAO,QAAQ;AAAA,MAC5B,UAAU;AAAA,IACX;AAAA,IACA,SAAS;AAAA,MACR,MAAM;AAAA,MACN,UAAU;AAAA,IACX;AAAA,IACA,WAAW;AAAA,MACV,MAAM;AAAA,MACN,UAAU;AAAA,IACX;AAAA,IACA,UAAU;AAAA,MACT,MAAM,uBAAO,MAAM;AAAA,MACnB,SAAS,CAAC;AAAA,IACX;AAAA,EACD;AAAA,EACA;AAAA,IACC,YAAY;AAAA,EACb;AACD;AAiBO,IAAM,YAAY,gBAAAC,QAAS,MAAoB,QAAQ,gBAAgB;;;ADtDvE,IAAM,gBAAN,cAA4B,0BAAW;AAAA,EACrC,cAAc;AAAA,EAEtB,YAAY,KAAa;AACxB,UAAM;AACN,SAAK,QAAQ,GAAG;AAAA,EACjB;AAAA,EAEA,MAAa,QAAQ,KAA4B;AAChD,QAAI,KAAK,aAAa;AACrB;AAAA,IACD;AAEA,QAAI;AACH,YAAM,iBAAAC,QAAS,QAAQ,GAAG;AAC1B,WAAK,cAAc;AACnB,4BAAQ,MAAM,KAAK,gCAAgC;AAAA,IACpD,SAAS,OAAO;AACf,4BAAQ,MAAM,MAAM,iCAAiC,KAAK;AAC1D,YAAM;AAAA,IACP;AAAA,EACD;AAAA,EAEA,MAAa,aAA4B;AACxC,QAAI,CAAC,KAAK,aAAa;AACtB;AAAA,IACD;AAEA,QAAI;AACH,YAAM,iBAAAA,QAAS,WAAW;AAC1B,WAAK,cAAc;AACnB,4BAAQ,MAAM,KAAK,mCAAmC;AAAA,IACvD,SAAS,OAAO;AACf,4BAAQ,MAAM,MAAM,sCAAsC,KAAK;AAC/D,YAAM;AAAA,IACP;AAAA,EACD;AAAA,EAEA,MAAa,kBAAkB,WAA2C;AACzE,UAAM,QAAQ,MAAM,UAAU,KAAK,EAAE,UAAU,CAAC,EAAE,KAAK;AAAA,MACtD,WAAW;AAAA,IACZ,CAAC;AAED,0BAAQ,MAAM,KAAK,SAAS,MAAM,MAAM,sBAAsB,SAAS,EAAE;AAEzE,UAAM,gBAA+B,EAAE,OAAO,CAAC,EAAE;AACjD,UAAM,QAAQ,CAAC,SAAuB;AACrC,YAAM,SAAS,KAAK,KAAK,SAAS,KAAK,KAAK;AAC5C,oBAAc,MAAM,MAAM,IAAI;AAAA,QAC7B,MAAM,KAAK;AAAA,QACX,SAAS,KAAK;AAAA,QACd,WAAW,KAAK;AAAA,QAChB,UAAU,KAAK;AAAA,MAChB;AAAA,IACD,CAAC;AAED,WAAO;AAAA,EACR;AAAA,EAEA,MAAa,qBACZ,WACA,MACgB;AAChB,0BAAQ,MAAM,KAAK,wCAAwC,SAAS,EAAE;AACtE,0BAAQ,MAAM,KAAK,SAAS,KAAK,UAAU,IAAI,CAAC,EAAE;AAElD,UAAM,UAAU,OAAO;AAAA,MACtB;AAAA,MACA,MAAM,KAAK;AAAA,MACX,SAAS,KAAK;AAAA,MACd,WAAW,KAAK;AAAA,MAChB,UAAU,KAAK;AAAA,IAChB,CAAC;AAAA,EACF;AAAA,EAEA,MAAa,oBACZ,OACA,QACA,WACgB;AAEhB,UAAM,OAAmB;AAAA,MACxB;AAAA,MACA,SAAS;AAAA,QACR,MAAM;AAAA,QACN,OAAO,CAAC,KAAK;AAAA,MACd;AAAA,MACA,WAAW,KAAK,IAAI;AAAA,MACpB,UAAU;AAAA,QACT;AAAA,QACA;AAAA,MACD;AAAA,IACD;AAEA,UAAM,KAAK,qBAAqB,WAAW,IAAI;AAAA,EAChD;AACD;","names":["import_mongoose","ChatRole","mongoose","mongoose"]}
+{"version":3,"sources":["../index.ts","../implements/session.memory.ts","../implements/base.memory.ts","../models/chats.model.ts","../implements/intent.memory.ts","../models/intent.model.ts"],"sourcesContent":["export { MongoDBSession } from \"./implements/session.memory\";\nexport { MongoDBIntent } from \"./implements/intent.memory\";","import { randomUUID } from \"node:crypto\";\nimport type { ChatObject, SessionMetadata, SessionObject } from \"@ainetwork/adk/types/memory\";\nimport { ISessionMemory } from \"@ainetwork/adk/modules\";\nimport { MongoDBMemory } from \"./base.memory\";\nimport {\n\tChatDocument,\n\tChatRole,\n  ChatObjectSchema,\n  SessionObjectSchema,\n  SessionDocument\n} from \"../models/chats.model\";\nimport { loggers } from \"@ainetwork/adk/utils/logger\";\nimport { Model } from \"mongoose\";\n\nexport class MongoDBSession extends MongoDBMemory implements ISessionMemory {\n  private chatModel: Model<ChatDocument>;\n  private sessionModel: Model<SessionDocument>;\n\n  constructor(uri: string) {\n    super(uri);\n    const _mongoose = super.getInstance();\n    this.chatModel = _mongoose.model<ChatDocument>(\"Chat\", ChatObjectSchema);\n    this.sessionModel = _mongoose.model<SessionDocument>(\"Session\", SessionObjectSchema);\n  }\n\n  public async getSession(sessionId: string, userId?: string): Promise<SessionObject | undefined> {\n\t\tconst chats = await this.chatModel.find({ sessionId }).sort({\n\t\t\ttimestamp: 1,\n\t\t});\n\n\t\tloggers.agent.debug(`Found ${chats.length} chats for session ${sessionId}`);\n\n\t\tconst sessionObject: SessionObject = { chats: {} };\n\t\tchats.forEach((chat: ChatDocument) => {\n\t\t\tconst chatId = chat._id?.toString() || chat.id;\n\t\t\tsessionObject.chats[chatId] = {\n\t\t\t\trole: chat.role as ChatRole,\n\t\t\t\tcontent: chat.content,\n\t\t\t\ttimestamp: chat.timestamp,\n\t\t\t\tmetadata: chat.metadata,\n\t\t\t};\n\t\t});\n\n\t\treturn sessionObject;\n  };\n\n\tpublic async createSession(userId: string, sessionId: string): Promise<void> {\n    await this.sessionModel.create({\n      sessionId,\n      userId,\n      updated_at: Date.now(),\n      created_at: Date.now(),\n    });\n  };\n\n\tpublic async addChatToSession(userId: string, sessionId: string, chat: ChatObject): Promise<void> {\n    const newId = randomUUID();\n    const session = await this.sessionModel.findOne({ sessionId, userId });\n    if (!session) {\n      await this.createSession(userId, sessionId);\n    } else {\n      await this.sessionModel.updateOne({ sessionId, userId }, {\n        updated_at: Date.now(),\n      });\n    }\n\t\tawait this.chatModel.create({\n\t\t\tsessionId,\n      chatId: newId,\n      userId,\n\t\t\trole: chat.role,\n\t\t\tcontent: chat.content,\n\t\t\ttimestamp: chat.timestamp,\n\t\t\tmetadata: chat.metadata,\n\t\t});\n  };\n\n\tpublic async deleteSession(userId: string, sessionId: string): Promise<void> {\n\t\tconst chats = await this.chatModel.find({ userId, sessionId }).sort({\n\t\t\ttimestamp: 1,\n\t\t});\n\n\t\tchats?.forEach((chat: ChatDocument) => {\n      chat.deleteOne();\n\t\t});\n    \n    const session = await this.sessionModel.findOne({ sessionId, userId });\n    session?.deleteOne();\n  };\n\n\tpublic async listSessions(userId: string): Promise<SessionMetadata[]> {\n    const sessions = await this.sessionModel.find({ userId }).sort({\n      updated_at: -1,\n    });\n    const data: SessionMetadata[] = sessions.map((session: SessionDocument) => {\n      return {\n        sessionId: session.sessionId,\n        title: session.title,\n        updatedAt: session.updated_at\n      } as SessionMetadata;\n    })\n    return data;\n  };\n}","import { IMemory } from \"node_modules/@ainetwork/adk/dist/esm/modules/memory/base.memory\";\nimport mongoose, { Mongoose } from \"mongoose\";\nimport { loggers } from \"@ainetwork/adk/utils/logger\";\n\nexport class MongoDBMemory implements IMemory {\n  private _isConnected: boolean = false;\n  private _uri: string;\n  private _mongoose: Mongoose;\n\n  constructor(uri: string) {\n    this._uri = uri;\n    this._mongoose = new Mongoose();\n  }\n\n  public getInstance(): Mongoose {\n    return this._mongoose;\n  }\n\n  public async connect(): Promise<void> {\n\t\tif (this._isConnected) {\n\t\t\treturn;\n\t\t}\n\n\t\ttry {\n      await this._mongoose.connect(this._uri);\n\t\t\tthis._isConnected = true;\n\t\t\tloggers.agent.info(\"MongoDB connected successfully\");\n\t\t} catch (error) {\n\t\t\tloggers.agent.error(\"Failed to connect to MongoDB:\", error);\n\t\t\tthrow error;\n\t\t}\n  }\n\n  public async disconnect(): Promise<void> {\n\t\tif (!this.isConnected) {\n\t\t\treturn;\n\t\t}\n\n\t\ttry {\n\t\t\tawait this._mongoose?.disconnect();\n\t\t\tthis._isConnected = false;\n\t\t\tloggers.agent.info(\"MongoDB disconnected successfully\");\n\t\t} catch (error) {\n\t\t\tloggers.agent.error(\"Failed to disconnect from MongoDB:\", error);\n\t\t\tthrow error;\n\t\t}\n  }\n\n  public isConnected(): boolean {\n    return this._isConnected;\n  }\n}","import { type Document, Schema } from \"mongoose\";\n\n// ChatRole enum\nexport enum ChatRole {\n\tUSER = \"USER\",\n\tSYSTEM = \"SYSTEM\",\n\tMODEL = \"MODEL\",\n}\n\nexport const SessionObjectSchema = new Schema(\n\t{\n\t\tsessionId: {\n\t\t\ttype: String,\n\t\t\trequired: true,\n\t\t\tindex: true,\n\t\t},\n\t\tuserId: {\n\t\t\ttype: String,\n\t\t\trequired: true,\n\t\t\tindex: true,\n\t\t},\n\t\ttitle: {\n\t\t\ttype: String,\n\t\t\trequired: false,\n\t\t},\n\t\tcreated_at: {\n\t\t\ttype: Number,\n\t\t\trequired: true,\n\t\t},\n\t\tupdated_at: {\n\t\t\ttype: Number,\n\t\t\trequired: true,\n\t\t}\n\t},\n);\n\nexport interface SessionDocument extends Document {\n\tsessionId: string;\n\tuserId: string;\n\ttitle?: string;\n\tcreated_at: number;\n\tupdated_at: number;\n}\n\n// ChatContentObject schema\nexport const ChatContentObjectSchema = new Schema(\n\t{\n\t\ttype: { type: String, required: true },\n\t\tparts: { type: [Schema.Types.Mixed], required: true },\n\t},\n\t{ _id: false },\n);\n\n// ChatObject schema - 개별 문서로 저장\nexport const ChatObjectSchema = new Schema(\n\t{\n\t\tsessionId: {\n\t\t\ttype: String,\n\t\t\trequired: true,\n\t\t\tindex: true,\n\t\t},\n\t\trole: {\n\t\t\ttype: String,\n\t\t\tenum: Object.values(ChatRole),\n\t\t\trequired: true,\n\t\t},\n\t\tcontent: {\n\t\t\ttype: ChatContentObjectSchema,\n\t\t\trequired: true,\n\t\t},\n\t\ttimestamp: {\n\t\t\ttype: Number,\n\t\t\trequired: true,\n\t\t},\n\t\tmetadata: {\n\t\t\ttype: Schema.Types.Mixed,\n\t\t\tdefault: {},\n\t\t},\n\t},\n);\n\n// Chat Document interface\nexport interface ChatDocument extends Document {\n\tsessionId: string;\n\trole: ChatRole;\n\tcontent: {\n\t\ttype: string;\n\t\tparts: any[];\n\t};\n\ttimestamp: number;\n\tmetadata?: { [key: string]: unknown };\n\tcreatedAt: Date;\n\tupdatedAt: Date;\n}\n","import { randomUUID } from \"node:crypto\";\nimport type { Intent } from \"@ainetwork/adk/types/memory\";\nimport { IIntentMemory } from \"@ainetwork/adk/modules\";\nimport { MongoDBMemory } from \"./base.memory\";\nimport { IntentModel } from \"../models/intent.model\";\n\nexport class MongoDBIntent extends MongoDBMemory implements IIntentMemory {\n  public async getIntent(intentId: string): Promise<Intent | undefined> {\n    const intent = await IntentModel.findById(intentId);\n    if (intent) {\n      return {\n        name: intent.name,\n        description: intent.description,\n        prompt: intent.prompt,\n        llm: intent.llm,\n      } as Intent;\n    }\n    return undefined;\n  };\n\n\tpublic async saveIntent(intent: Intent): Promise<void> {\n    const newId = randomUUID();\n    await IntentModel.create({\n      _id: newId,\n      name: intent.name,\n      description: intent.description,\n      prompt: intent.prompt,\n      llm: intent.llm,\n    });\n  };\n\n\tpublic async updateIntent(intentId: string, intent: Intent): Promise<void> {\n    await IntentModel.updateOne({\n      _id: intentId,\n    },{\n      name: intent.name,\n      description: intent.description,\n      prompt: intent.prompt,\n      llm: intent.llm,\n    });\n  };\n\n\tpublic async deleteIntent(intentId: string): Promise<void> {\n    await IntentModel.deleteOne({ _id: intentId });\n  };\n\n\tpublic async listIntents(): Promise<Intent[]> {\n    const intents = await IntentModel.find();\n    return intents.map(intent => ({\n      name: intent.name,\n      description: intent.description,\n      prompt: intent.prompt,\n      llm: intent.llm,\n    } as Intent));\n  };\n}","import mongoose, { type Document, Schema } from \"mongoose\";\n\nconst IntentObjectSchema = new Schema(\n\t{\n\t\tname: {\n\t\t\ttype: String,\n\t\t\trequired: true,\n\t\t\tindex: true,\n\t\t},\n\t\tdescription: {\n\t\t\ttype: String,\n\t\t\trequired: true,\n\t\t},\n\t\tprompt: {\n\t\t\ttype: String,\n\t\t\trequired: false,\n\t\t},\n\t\tllm: {\n\t\t\ttype: String,\n\t\t\trequired: false,\n\t\t},\n\t}\n);\n\nexport interface IntentDocument extends Document {\n\tname: string;\n\tdescription: string;\n\tprompt?: string;\n\tllm?: string;\n}\n\nexport const IntentModel = mongoose.model<IntentDocument>(\"Intent\", IntentObjectSchema);"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACAA,yBAA2B;;;ACC3B,sBAAmC;AACnC,oBAAwB;AAEjB,IAAM,gBAAN,MAAuC;AAAA,EACpC,eAAwB;AAAA,EACxB;AAAA,EACA;AAAA,EAER,YAAY,KAAa;AACvB,SAAK,OAAO;AACZ,SAAK,YAAY,IAAI,yBAAS;AAAA,EAChC;AAAA,EAEO,cAAwB;AAC7B,WAAO,KAAK;AAAA,EACd;AAAA,EAEA,MAAa,UAAyB;AACtC,QAAI,KAAK,cAAc;AACtB;AAAA,IACD;AAEA,QAAI;AACA,YAAM,KAAK,UAAU,QAAQ,KAAK,IAAI;AACzC,WAAK,eAAe;AACpB,4BAAQ,MAAM,KAAK,gCAAgC;AAAA,IACpD,SAAS,OAAO;AACf,4BAAQ,MAAM,MAAM,iCAAiC,KAAK;AAC1D,YAAM;AAAA,IACP;AAAA,EACA;AAAA,EAEA,MAAa,aAA4B;AACzC,QAAI,CAAC,KAAK,aAAa;AACtB;AAAA,IACD;AAEA,QAAI;AACH,YAAM,KAAK,WAAW,WAAW;AACjC,WAAK,eAAe;AACpB,4BAAQ,MAAM,KAAK,mCAAmC;AAAA,IACvD,SAAS,OAAO;AACf,4BAAQ,MAAM,MAAM,sCAAsC,KAAK;AAC/D,YAAM;AAAA,IACP;AAAA,EACA;AAAA,EAEO,cAAuB;AAC5B,WAAO,KAAK;AAAA,EACd;AACF;;;ACnDA,IAAAA,mBAAsC;AAG/B,IAAK,WAAL,kBAAKC,cAAL;AACN,EAAAA,UAAA,UAAO;AACP,EAAAA,UAAA,YAAS;AACT,EAAAA,UAAA,WAAQ;AAHG,SAAAA;AAAA,GAAA;AAML,IAAM,sBAAsB,IAAI;AAAA,EACtC;AAAA,IACC,WAAW;AAAA,MACV,MAAM;AAAA,MACN,UAAU;AAAA,MACV,OAAO;AAAA,IACR;AAAA,IACA,QAAQ;AAAA,MACP,MAAM;AAAA,MACN,UAAU;AAAA,MACV,OAAO;AAAA,IACR;AAAA,IACA,OAAO;AAAA,MACN,MAAM;AAAA,MACN,UAAU;AAAA,IACX;AAAA,IACA,YAAY;AAAA,MACX,MAAM;AAAA,MACN,UAAU;AAAA,IACX;AAAA,IACA,YAAY;AAAA,MACX,MAAM;AAAA,MACN,UAAU;AAAA,IACX;AAAA,EACD;AACD;AAWO,IAAM,0BAA0B,IAAI;AAAA,EAC1C;AAAA,IACC,MAAM,EAAE,MAAM,QAAQ,UAAU,KAAK;AAAA,IACrC,OAAO,EAAE,MAAM,CAAC,wBAAO,MAAM,KAAK,GAAG,UAAU,KAAK;AAAA,EACrD;AAAA,EACA,EAAE,KAAK,MAAM;AACd;AAGO,IAAM,mBAAmB,IAAI;AAAA,EACnC;AAAA,IACC,WAAW;AAAA,MACV,MAAM;AAAA,MACN,UAAU;AAAA,MACV,OAAO;AAAA,IACR;AAAA,IACA,MAAM;AAAA,MACL,MAAM;AAAA,MACN,MAAM,OAAO,OAAO,QAAQ;AAAA,MAC5B,UAAU;AAAA,IACX;AAAA,IACA,SAAS;AAAA,MACR,MAAM;AAAA,MACN,UAAU;AAAA,IACX;AAAA,IACA,WAAW;AAAA,MACV,MAAM;AAAA,MACN,UAAU;AAAA,IACX;AAAA,IACA,UAAU;AAAA,MACT,MAAM,wBAAO,MAAM;AAAA,MACnB,SAAS,CAAC;AAAA,IACX;AAAA,EACD;AACD;;;AFpEA,IAAAC,iBAAwB;AAGjB,IAAM,iBAAN,cAA6B,cAAwC;AAAA,EAClE;AAAA,EACA;AAAA,EAER,YAAY,KAAa;AACvB,UAAM,GAAG;AACT,UAAM,YAAY,MAAM,YAAY;AACpC,SAAK,YAAY,UAAU,MAAoB,QAAQ,gBAAgB;AACvE,SAAK,eAAe,UAAU,MAAuB,WAAW,mBAAmB;AAAA,EACrF;AAAA,EAEA,MAAa,WAAW,WAAmB,QAAqD;AAChG,UAAM,QAAQ,MAAM,KAAK,UAAU,KAAK,EAAE,UAAU,CAAC,EAAE,KAAK;AAAA,MAC3D,WAAW;AAAA,IACZ,CAAC;AAED,2BAAQ,MAAM,MAAM,SAAS,MAAM,MAAM,sBAAsB,SAAS,EAAE;AAE1E,UAAM,gBAA+B,EAAE,OAAO,CAAC,EAAE;AACjD,UAAM,QAAQ,CAAC,SAAuB;AACrC,YAAM,SAAS,KAAK,KAAK,SAAS,KAAK,KAAK;AAC5C,oBAAc,MAAM,MAAM,IAAI;AAAA,QAC7B,MAAM,KAAK;AAAA,QACX,SAAS,KAAK;AAAA,QACd,WAAW,KAAK;AAAA,QAChB,UAAU,KAAK;AAAA,MAChB;AAAA,IACD,CAAC;AAED,WAAO;AAAA,EACP;AAAA,EAED,MAAa,cAAc,QAAgB,WAAkC;AAC1E,UAAM,KAAK,aAAa,OAAO;AAAA,MAC7B;AAAA,MACA;AAAA,MACA,YAAY,KAAK,IAAI;AAAA,MACrB,YAAY,KAAK,IAAI;AAAA,IACvB,CAAC;AAAA,EACH;AAAA,EAED,MAAa,iBAAiB,QAAgB,WAAmB,MAAiC;AAC/F,UAAM,YAAQ,+BAAW;AACzB,UAAM,UAAU,MAAM,KAAK,aAAa,QAAQ,EAAE,WAAW,OAAO,CAAC;AACrE,QAAI,CAAC,SAAS;AACZ,YAAM,KAAK,cAAc,QAAQ,SAAS;AAAA,IAC5C,OAAO;AACL,YAAM,KAAK,aAAa,UAAU,EAAE,WAAW,OAAO,GAAG;AAAA,QACvD,YAAY,KAAK,IAAI;AAAA,MACvB,CAAC;AAAA,IACH;AACF,UAAM,KAAK,UAAU,OAAO;AAAA,MAC3B;AAAA,MACG,QAAQ;AAAA,MACR;AAAA,MACH,MAAM,KAAK;AAAA,MACX,SAAS,KAAK;AAAA,MACd,WAAW,KAAK;AAAA,MAChB,UAAU,KAAK;AAAA,IAChB,CAAC;AAAA,EACD;AAAA,EAED,MAAa,cAAc,QAAgB,WAAkC;AAC5E,UAAM,QAAQ,MAAM,KAAK,UAAU,KAAK,EAAE,QAAQ,UAAU,CAAC,EAAE,KAAK;AAAA,MACnE,WAAW;AAAA,IACZ,CAAC;AAED,WAAO,QAAQ,CAAC,SAAuB;AACnC,WAAK,UAAU;AAAA,IACnB,CAAC;AAEC,UAAM,UAAU,MAAM,KAAK,aAAa,QAAQ,EAAE,WAAW,OAAO,CAAC;AACrE,aAAS,UAAU;AAAA,EACrB;AAAA,EAED,MAAa,aAAa,QAA4C;AACnE,UAAM,WAAW,MAAM,KAAK,aAAa,KAAK,EAAE,OAAO,CAAC,EAAE,KAAK;AAAA,MAC7D,YAAY;AAAA,IACd,CAAC;AACD,UAAM,OAA0B,SAAS,IAAI,CAAC,YAA6B;AACzE,aAAO;AAAA,QACL,WAAW,QAAQ;AAAA,QACnB,OAAO,QAAQ;AAAA,QACf,WAAW,QAAQ;AAAA,MACrB;AAAA,IACF,CAAC;AACD,WAAO;AAAA,EACT;AACF;;;AGtGA,IAAAC,sBAA2B;;;ACA3B,IAAAC,mBAAgD;AAEhD,IAAM,qBAAqB,IAAI;AAAA,EAC9B;AAAA,IACC,MAAM;AAAA,MACL,MAAM;AAAA,MACN,UAAU;AAAA,MACV,OAAO;AAAA,IACR;AAAA,IACA,aAAa;AAAA,MACZ,MAAM;AAAA,MACN,UAAU;AAAA,IACX;AAAA,IACA,QAAQ;AAAA,MACP,MAAM;AAAA,MACN,UAAU;AAAA,IACX;AAAA,IACA,KAAK;AAAA,MACJ,MAAM;AAAA,MACN,UAAU;AAAA,IACX;AAAA,EACD;AACD;AASO,IAAM,cAAc,iBAAAC,QAAS,MAAsB,UAAU,kBAAkB;;;ADzB/E,IAAM,gBAAN,cAA4B,cAAuC;AAAA,EACxE,MAAa,UAAU,UAA+C;AACpE,UAAM,SAAS,MAAM,YAAY,SAAS,QAAQ;AAClD,QAAI,QAAQ;AACV,aAAO;AAAA,QACL,MAAM,OAAO;AAAA,QACb,aAAa,OAAO;AAAA,QACpB,QAAQ,OAAO;AAAA,QACf,KAAK,OAAO;AAAA,MACd;AAAA,IACF;AACA,WAAO;AAAA,EACT;AAAA,EAED,MAAa,WAAW,QAA+B;AACpD,UAAM,YAAQ,gCAAW;AACzB,UAAM,YAAY,OAAO;AAAA,MACvB,KAAK;AAAA,MACL,MAAM,OAAO;AAAA,MACb,aAAa,OAAO;AAAA,MACpB,QAAQ,OAAO;AAAA,MACf,KAAK,OAAO;AAAA,IACd,CAAC;AAAA,EACH;AAAA,EAED,MAAa,aAAa,UAAkB,QAA+B;AACxE,UAAM,YAAY,UAAU;AAAA,MAC1B,KAAK;AAAA,IACP,GAAE;AAAA,MACA,MAAM,OAAO;AAAA,MACb,aAAa,OAAO;AAAA,MACpB,QAAQ,OAAO;AAAA,MACf,KAAK,OAAO;AAAA,IACd,CAAC;AAAA,EACH;AAAA,EAED,MAAa,aAAa,UAAiC;AACxD,UAAM,YAAY,UAAU,EAAE,KAAK,SAAS,CAAC;AAAA,EAC/C;AAAA,EAED,MAAa,cAAiC;AAC3C,UAAM,UAAU,MAAM,YAAY,KAAK;AACvC,WAAO,QAAQ,IAAI,aAAW;AAAA,MAC5B,MAAM,OAAO;AAAA,MACb,aAAa,OAAO;AAAA,MACpB,QAAQ,OAAO;AAAA,MACf,KAAK,OAAO;AAAA,IACd,EAAY;AAAA,EACd;AACF;","names":["import_mongoose","ChatRole","import_logger","import_node_crypto","import_mongoose","mongoose"]}

package/dist/index.d.cts

@@ -1,14 +1,36 @@
-import { BaseMemory } from '@ainetwork/adk/modules';
-import { SessionObject, ChatObject } from '@ainetwork/adk/types/memory';
+import { SessionObject, ChatObject, SessionMetadata, Intent } from '@ainetwork/adk/types/memory';
+import { ISessionMemory, IIntentMemory } from '@ainetwork/adk/modules';
+import { IMemory } from 'node_modules/@ainetwork/adk/dist/esm/modules/memory/base.memory';
+import { Mongoose } from 'mongoose';
 
-declare class MongoDBMemory extends BaseMemory {
-    private isConnected;
+declare class MongoDBMemory implements IMemory {
+    private _isConnected;
+    private _uri;
+    private _mongoose;
     constructor(uri: string);
-    connect(uri: string): Promise<void>;
+    getInstance(): Mongoose;
+    connect(): 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>;
+    isConnected(): boolean;
 }
 
-export { MongoDBMemory };
+declare class MongoDBSession extends MongoDBMemory implements ISessionMemory {
+    private chatModel;
+    private sessionModel;
+    constructor(uri: string);
+    getSession(sessionId: string, userId?: string): Promise<SessionObject | undefined>;
+    createSession(userId: string, sessionId: string): Promise<void>;
+    addChatToSession(userId: string, sessionId: string, chat: ChatObject): Promise<void>;
+    deleteSession(userId: string, sessionId: string): Promise<void>;
+    listSessions(userId: string): Promise<SessionMetadata[]>;
+}
+
+declare class MongoDBIntent extends MongoDBMemory implements IIntentMemory {
+    getIntent(intentId: string): Promise<Intent | undefined>;
+    saveIntent(intent: Intent): Promise<void>;
+    updateIntent(intentId: string, intent: Intent): Promise<void>;
+    deleteIntent(intentId: string): Promise<void>;
+    listIntents(): Promise<Intent[]>;
+}
+
+export { MongoDBIntent, MongoDBSession };

package/dist/index.d.ts

@@ -1,14 +1,36 @@
-import { BaseMemory } from '@ainetwork/adk/modules';
-import { SessionObject, ChatObject } from '@ainetwork/adk/types/memory';
+import { SessionObject, ChatObject, SessionMetadata, Intent } from '@ainetwork/adk/types/memory';
+import { ISessionMemory, IIntentMemory } from '@ainetwork/adk/modules';
+import { IMemory } from 'node_modules/@ainetwork/adk/dist/esm/modules/memory/base.memory';
+import { Mongoose } from 'mongoose';
 
-declare class MongoDBMemory extends BaseMemory {
-    private isConnected;
+declare class MongoDBMemory implements IMemory {
+    private _isConnected;
+    private _uri;
+    private _mongoose;
     constructor(uri: string);
-    connect(uri: string): Promise<void>;
+    getInstance(): Mongoose;
+    connect(): 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>;
+    isConnected(): boolean;
 }
 
-export { MongoDBMemory };
+declare class MongoDBSession extends MongoDBMemory implements ISessionMemory {
+    private chatModel;
+    private sessionModel;
+    constructor(uri: string);
+    getSession(sessionId: string, userId?: string): Promise<SessionObject | undefined>;
+    createSession(userId: string, sessionId: string): Promise<void>;
+    addChatToSession(userId: string, sessionId: string, chat: ChatObject): Promise<void>;
+    deleteSession(userId: string, sessionId: string): Promise<void>;
+    listSessions(userId: string): Promise<SessionMetadata[]>;
+}
+
+declare class MongoDBIntent extends MongoDBMemory implements IIntentMemory {
+    getIntent(intentId: string): Promise<Intent | undefined>;
+    saveIntent(intent: Intent): Promise<void>;
+    updateIntent(intentId: string, intent: Intent): Promise<void>;
+    deleteIntent(intentId: string): Promise<void>;
+    listIntents(): Promise<Intent[]>;
+}
+
+export { MongoDBIntent, MongoDBSession };