@ainetwork/adk-provider-memory-mongodb 0.1.6 → 0.1.7

package/dist/{chunk-N5FAVJ4Z.js → chunk-S537KQMN.js}

@@ -1,14 +1,14 @@
 // models/chats.model.ts
+import { MessageRole, ThreadType } from "@ainetwork/adk/types/memory";
 import { Schema } from "mongoose";
-var ChatRole = /* @__PURE__ */ ((ChatRole2) => {
-  ChatRole2["USER"] = "USER";
-  ChatRole2["SYSTEM"] = "SYSTEM";
-  ChatRole2["MODEL"] = "MODEL";
-  return ChatRole2;
-})(ChatRole || {});
-var SessionObjectSchema = new Schema(
+var ThreadObjectSchema = new Schema(
   {
-    sessionId: {
+    type: {
+      type: String,
+      enum: Object.values(ThreadType),
+      required: true
+    },
+    threadId: {
       type: String,
       required: true,
       index: true
@@ -41,7 +41,7 @@ var ChatContentObjectSchema = new Schema(
 );
 var ChatObjectSchema = new Schema(
   {
-    sessionId: {
+    threadId: {
       type: String,
       required: true,
       index: true
@@ -53,7 +53,7 @@ var ChatObjectSchema = new Schema(
     },
     role: {
       type: String,
-      enum: Object.values(ChatRole),
+      enum: Object.values(MessageRole),
       required: true
     },
     content: {
@@ -72,9 +72,8 @@ var ChatObjectSchema = new Schema(
 );
 
 export {
-  ChatRole,
-  SessionObjectSchema,
+  ThreadObjectSchema,
   ChatContentObjectSchema,
   ChatObjectSchema
 };
-//# sourceMappingURL=chunk-N5FAVJ4Z.js.map
+//# sourceMappingURL=chunk-S537KQMN.js.map

package/dist/chunk-S537KQMN.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../models/chats.model.ts"],"sourcesContent":["import { MessageRole, ThreadType } from \"@ainetwork/adk/types/memory\";\nimport { type Document, Schema } from \"mongoose\";\n\nexport const ThreadObjectSchema = new Schema(\n\t{\n\t\ttype: {\n\t\t\ttype: String,\n\t\t\tenum: Object.values(ThreadType),\n\t\t\trequired: true,\n\t\t},\n\t\tthreadId: {\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 ThreadDocument extends Document {\n\ttype: ThreadType;\n\tthreadId: 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\tthreadId: {\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\trole: {\n\t\t\ttype: String,\n\t\t\tenum: Object.values(MessageRole),\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\tthreadId: string;\n\trole: MessageRole;\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,SAAS,aAAa,kBAAkB;AACxC,SAAwB,cAAc;AAE/B,IAAM,qBAAqB,IAAI;AAAA,EACrC;AAAA,IACC,MAAM;AAAA,MACL,MAAM;AAAA,MACN,MAAM,OAAO,OAAO,UAAU;AAAA,MAC9B,UAAU;AAAA,IACX;AAAA,IACA,UAAU;AAAA,MACT,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;AAYO,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,UAAU;AAAA,MACT,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,MAAM;AAAA,MACL,MAAM;AAAA,MACN,MAAM,OAAO,OAAO,WAAW;AAAA,MAC/B,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":[]}

package/dist/index.cjs

@@ -31,11 +31,11 @@ var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: tru
 var index_exports = {};
 __export(index_exports, {
   MongoDBIntent: () => MongoDBIntent,
-  MongoDBSession: () => MongoDBSession
+  MongoDBThread: () => MongoDBThread
 });
 module.exports = __toCommonJS(index_exports);
 
-// implements/session.memory.ts
+// implements/thread.memory.ts
 var import_node_crypto = require("crypto");
 
 // implements/base.memory.ts
@@ -84,16 +84,16 @@ var MongoDBMemory = class {
 };
 
 // models/chats.model.ts
+var import_memory = require("@ainetwork/adk/types/memory");
 var import_mongoose2 = require("mongoose");
-var ChatRole = /* @__PURE__ */ ((ChatRole3) => {
-  ChatRole3["USER"] = "USER";
-  ChatRole3["SYSTEM"] = "SYSTEM";
-  ChatRole3["MODEL"] = "MODEL";
-  return ChatRole3;
-})(ChatRole || {});
-var SessionObjectSchema = new import_mongoose2.Schema(
+var ThreadObjectSchema = new import_mongoose2.Schema(
   {
-    sessionId: {
+    type: {
+      type: String,
+      enum: Object.values(import_memory.ThreadType),
+      required: true
+    },
+    threadId: {
       type: String,
       required: true,
       index: true
@@ -126,7 +126,7 @@ var ChatContentObjectSchema = new import_mongoose2.Schema(
 );
 var ChatObjectSchema = new import_mongoose2.Schema(
   {
-    sessionId: {
+    threadId: {
       type: String,
       required: true,
       index: true
@@ -138,7 +138,7 @@ var ChatObjectSchema = new import_mongoose2.Schema(
     },
     role: {
       type: String,
-      enum: Object.values(ChatRole),
+      enum: Object.values(import_memory.MessageRole),
       required: true
     },
     content: {
@@ -156,81 +156,89 @@ var ChatObjectSchema = new import_mongoose2.Schema(
   }
 );
 
-// implements/session.memory.ts
+// implements/thread.memory.ts
 var import_logger2 = require("@ainetwork/adk/utils/logger");
-var MongoDBSession = class extends MongoDBMemory {
+var MongoDBThread = class extends MongoDBMemory {
   chatModel;
-  sessionModel;
+  threadModel;
   constructor(uri) {
     super(uri);
     const _mongoose = super.getInstance();
     this.chatModel = _mongoose.model("Chat", ChatObjectSchema);
-    this.sessionModel = _mongoose.model("Session", SessionObjectSchema);
+    this.threadModel = _mongoose.model("Thread", ThreadObjectSchema);
   }
-  async getSession(userId, sessionId) {
-    const chats = await this.chatModel.find({ sessionId, userId }).sort({
+  async getThread(type, userId, threadId) {
+    const thread = await this.threadModel.findOne({ type, threadId, userId });
+    const chats = await this.chatModel.find({ threadId, userId }).sort({
       timestamp: 1
     });
-    const session = await this.sessionModel.findOne({ sessionId, userId });
-    import_logger2.loggers.agent.debug(`Found ${chats.length} chats for session ${sessionId}`);
-    const sessionObject = { chats: {} };
-    sessionObject.title = session?.title;
+    if (!thread) return void 0;
+    import_logger2.loggers.agent.debug(`Found ${chats.length} chats for thread ${threadId}`);
+    const threadObject = {
+      type: thread.type,
+      title: thread.title || "New chats",
+      messages: {}
+    };
     chats.forEach((chat) => {
       const chatId = chat._id?.toString() || chat.id;
-      sessionObject.chats[chatId] = {
+      threadObject.messages[chatId] = {
         role: chat.role,
         content: chat.content,
         timestamp: chat.timestamp,
         metadata: chat.metadata
       };
     });
-    return sessionObject;
+    return threadObject;
   }
-  async createSession(userId, sessionId, title) {
+  async createThread(type, userId, threadId, title) {
     const now = Date.now();
-    await this.sessionModel.create({
-      sessionId,
+    await this.threadModel.create({
+      type,
       userId,
+      threadId,
       title,
       updated_at: now,
       created_at: now
     });
-    return { title, sessionId, updatedAt: now };
+    return { type, threadId, title, updatedAt: now };
   }
-  async addChatToSession(userId, sessionId, chat) {
-    const newId = (0, import_node_crypto.randomUUID)();
-    await this.sessionModel.updateOne({ sessionId, userId }, {
+  async addMessagesToThread(userId, threadId, messages) {
+    await this.threadModel.updateOne({ threadId, 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
-    });
+    for (const message of messages) {
+      const newId = (0, import_node_crypto.randomUUID)();
+      await this.chatModel.create({
+        threadId,
+        chatId: newId,
+        userId,
+        role: message.role,
+        content: message.content,
+        timestamp: message.timestamp,
+        metadata: message.metadata
+      });
+    }
   }
-  async deleteSession(userId, sessionId) {
-    const chats = await this.chatModel.find({ userId, sessionId }).sort({
+  async deleteThread(userId, threadId) {
+    const chats = await this.chatModel.find({ userId, threadId }).sort({
       timestamp: 1
     });
     chats?.forEach((chat) => {
       chat.deleteOne();
     });
-    const session = await this.sessionModel.findOne({ sessionId, userId });
-    session?.deleteOne();
+    const thread = await this.threadModel.findOne({ userId, threadId });
+    thread?.deleteOne();
   }
-  async listSessions(userId) {
-    const sessions = await this.sessionModel.find({ userId }).sort({
+  async listThreads(userId) {
+    const threads = await this.threadModel.find({ userId }).sort({
       updated_at: -1
     });
-    const data = sessions.map((session) => {
+    const data = threads.map((thread) => {
       return {
-        sessionId: session.sessionId,
-        title: session.title,
-        updatedAt: session.updated_at
+        type: thread.type,
+        threadId: thread.threadId,
+        title: thread.title,
+        updatedAt: thread.updated_at
       };
     });
     return data;
@@ -315,6 +323,6 @@ var MongoDBIntent = class extends MongoDBMemory {
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   MongoDBIntent,
-  MongoDBSession
+  MongoDBThread
 });
 //# sourceMappingURL=index.cjs.map

package/dist/index.cjs.map

@@ -1 +1 @@
-{"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(userId: string, sessionId: string): Promise<SessionObject | undefined> {\n\t\tconst chats = await this.chatModel.find({ sessionId, userId }).sort({\n\t\t\ttimestamp: 1,\n\t\t});\n    const session = await this.sessionModel.findOne({ sessionId, userId });\n\n\t\tloggers.agent.debug(`Found ${chats.length} chats for session ${sessionId}`);\n\n\t\tconst sessionObject: SessionObject = { chats: {} };\n    sessionObject.title = session?.title;\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, title: string): Promise<SessionMetadata> {\n    const now = Date.now();\n    await this.sessionModel.create({\n      sessionId,\n      userId,\n      title,\n      updated_at: now,\n      created_at: now,\n    });\n\n    return { title, sessionId, updatedAt: now };\n  };\n\n\tpublic async addChatToSession(userId: string, sessionId: string, chat: ChatObject): Promise<void> {\n    const newId = randomUUID();\n    await this.sessionModel.updateOne({ sessionId, userId }, {\n      updated_at: Date.now(),\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\tuserId: {\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,QAAQ;AAAA,MACP,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;;;AFzEA,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,QAAgB,WAAuD;AAC/F,UAAM,QAAQ,MAAM,KAAK,UAAU,KAAK,EAAE,WAAW,OAAO,CAAC,EAAE,KAAK;AAAA,MACnE,WAAW;AAAA,IACZ,CAAC;AACC,UAAM,UAAU,MAAM,KAAK,aAAa,QAAQ,EAAE,WAAW,OAAO,CAAC;AAEvE,2BAAQ,MAAM,MAAM,SAAS,MAAM,MAAM,sBAAsB,SAAS,EAAE;AAE1E,UAAM,gBAA+B,EAAE,OAAO,CAAC,EAAE;AAC/C,kBAAc,QAAQ,SAAS;AACjC,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,WAAmB,OAAyC;AACpG,UAAM,MAAM,KAAK,IAAI;AACrB,UAAM,KAAK,aAAa,OAAO;AAAA,MAC7B;AAAA,MACA;AAAA,MACA;AAAA,MACA,YAAY;AAAA,MACZ,YAAY;AAAA,IACd,CAAC;AAED,WAAO,EAAE,OAAO,WAAW,WAAW,IAAI;AAAA,EAC5C;AAAA,EAED,MAAa,iBAAiB,QAAgB,WAAmB,MAAiC;AAC/F,UAAM,YAAQ,+BAAW;AACzB,UAAM,KAAK,aAAa,UAAU,EAAE,WAAW,OAAO,GAAG;AAAA,MACvD,YAAY,KAAK,IAAI;AAAA,IACvB,CAAC;AACH,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;;;AGvGA,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"]}
+{"version":3,"sources":["../index.ts","../implements/thread.memory.ts","../implements/base.memory.ts","../models/chats.model.ts","../implements/intent.memory.ts","../models/intent.model.ts"],"sourcesContent":["export { MongoDBThread } from \"./implements/thread.memory\";\nexport { MongoDBIntent } from \"./implements/intent.memory\";","import { randomUUID } from \"node:crypto\";\nimport type { MessageObject, ThreadMetadata, ThreadObject, ThreadType } from \"@ainetwork/adk/types/memory\";\nimport { MessageRole } from \"@ainetwork/adk/types/memory\";\nimport { IThreadMemory } from \"@ainetwork/adk/modules\";\nimport { MongoDBMemory } from \"./base.memory\";\nimport {\n\tChatDocument,\n  ChatObjectSchema,\n  ThreadObjectSchema,\n  ThreadDocument\n} from \"../models/chats.model\";\nimport { loggers } from \"@ainetwork/adk/utils/logger\";\nimport { Model } from \"mongoose\";\n\nexport class MongoDBThread extends MongoDBMemory implements IThreadMemory {\n  private chatModel: Model<ChatDocument>;\n  private threadModel: Model<ThreadDocument>;\n\n  constructor(uri: string) {\n    super(uri);\n    const _mongoose = super.getInstance();\n    this.chatModel = _mongoose.model<ChatDocument>(\"Chat\", ChatObjectSchema);\n    this.threadModel = _mongoose.model<ThreadDocument>(\"Thread\", ThreadObjectSchema);\n  }\n\n  public async getThread(\n    type: ThreadType,\n    userId: string,\n    threadId: string\n  ): Promise<ThreadObject | undefined> {\n    const thread = await this.threadModel.findOne({ type, threadId, userId });\n\t\tconst chats = await this.chatModel.find({ threadId, userId }).sort({\n\t\t\ttimestamp: 1,\n\t\t});\n\n    if (!thread) return undefined;\n\n\t\tloggers.agent.debug(`Found ${chats.length} chats for thread ${threadId}`);\n\n\t\tconst threadObject: ThreadObject = { \n      type: thread.type as ThreadType,\n      title: thread.title || \"New chats\",\n      messages: {}\n    };\n\t\tchats.forEach((chat: ChatDocument) => {\n\t\t\tconst chatId = chat._id?.toString() || chat.id;\n\t\t\tthreadObject.messages[chatId] = {\n\t\t\t\trole: chat.role as MessageRole,\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 threadObject;\n  };\n\n\tpublic async createThread(\n\t\ttype: ThreadType,\n\t\tuserId: string,\n\t\tthreadId: string,\n\t\ttitle: string,\n  ): Promise<ThreadMetadata> {\n    const now = Date.now();\n    await this.threadModel.create({\n      type,\n      userId,\n      threadId,\n      title,\n      updated_at: now,\n      created_at: now,\n    });\n\n    return { type, threadId, title, updatedAt: now };\n  };\n\n\tpublic async addMessagesToThread(\n    userId: string,\n    threadId: string,\n    messages: MessageObject[]\n  ): Promise<void> {\n    await this.threadModel.updateOne({ threadId, userId }, {\n      updated_at: Date.now(),\n    });\n    for (const message of messages) {\n      const newId = randomUUID();\n      await this.chatModel.create({\n        threadId,\n        chatId: newId,\n        userId,\n        role: message.role,\n        content: message.content,\n        timestamp: message.timestamp,\n        metadata: message.metadata,\n      });\n    }\n  };\n\n\tpublic async deleteThread(userId: string, threadId: string): Promise<void> {\n\t\tconst chats = await this.chatModel.find({ userId, threadId }).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 thread = await this.threadModel.findOne({ userId, threadId });\n    thread?.deleteOne();\n  };\n\n\tpublic async listThreads(userId: string): Promise<ThreadMetadata[]> {\n    const threads = await this.threadModel.find({ userId }).sort({\n      updated_at: -1,\n    });\n    const data: ThreadMetadata[] = threads.map((thread: ThreadDocument) => {\n      return {\n        type: thread.type,\n        threadId: thread.threadId,\n        title: thread.title,\n        updatedAt: thread.updated_at\n      } as ThreadMetadata;\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 { MessageRole, ThreadType } from \"@ainetwork/adk/types/memory\";\nimport { type Document, Schema } from \"mongoose\";\n\nexport const ThreadObjectSchema = new Schema(\n\t{\n\t\ttype: {\n\t\t\ttype: String,\n\t\t\tenum: Object.values(ThreadType),\n\t\t\trequired: true,\n\t\t},\n\t\tthreadId: {\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 ThreadDocument extends Document {\n\ttype: ThreadType;\n\tthreadId: 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\tthreadId: {\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\trole: {\n\t\t\ttype: String,\n\t\t\tenum: Object.values(MessageRole),\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\tthreadId: string;\n\trole: MessageRole;\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,oBAAwC;AACxC,IAAAA,mBAAsC;AAE/B,IAAM,qBAAqB,IAAI;AAAA,EACrC;AAAA,IACC,MAAM;AAAA,MACL,MAAM;AAAA,MACN,MAAM,OAAO,OAAO,wBAAU;AAAA,MAC9B,UAAU;AAAA,IACX;AAAA,IACA,UAAU;AAAA,MACT,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;AAYO,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,UAAU;AAAA,MACT,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,MAAM;AAAA,MACL,MAAM;AAAA,MACN,MAAM,OAAO,OAAO,yBAAW;AAAA,MAC/B,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;;;AFzEA,IAAAC,iBAAwB;AAGjB,IAAM,gBAAN,cAA4B,cAAuC;AAAA,EAChE;AAAA,EACA;AAAA,EAER,YAAY,KAAa;AACvB,UAAM,GAAG;AACT,UAAM,YAAY,MAAM,YAAY;AACpC,SAAK,YAAY,UAAU,MAAoB,QAAQ,gBAAgB;AACvE,SAAK,cAAc,UAAU,MAAsB,UAAU,kBAAkB;AAAA,EACjF;AAAA,EAEA,MAAa,UACX,MACA,QACA,UACmC;AACnC,UAAM,SAAS,MAAM,KAAK,YAAY,QAAQ,EAAE,MAAM,UAAU,OAAO,CAAC;AAC1E,UAAM,QAAQ,MAAM,KAAK,UAAU,KAAK,EAAE,UAAU,OAAO,CAAC,EAAE,KAAK;AAAA,MAClE,WAAW;AAAA,IACZ,CAAC;AAEC,QAAI,CAAC,OAAQ,QAAO;AAEtB,2BAAQ,MAAM,MAAM,SAAS,MAAM,MAAM,qBAAqB,QAAQ,EAAE;AAExE,UAAM,eAA6B;AAAA,MAC/B,MAAM,OAAO;AAAA,MACb,OAAO,OAAO,SAAS;AAAA,MACvB,UAAU,CAAC;AAAA,IACb;AACF,UAAM,QAAQ,CAAC,SAAuB;AACrC,YAAM,SAAS,KAAK,KAAK,SAAS,KAAK,KAAK;AAC5C,mBAAa,SAAS,MAAM,IAAI;AAAA,QAC/B,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,aACZ,MACA,QACA,UACA,OAC2B;AACzB,UAAM,MAAM,KAAK,IAAI;AACrB,UAAM,KAAK,YAAY,OAAO;AAAA,MAC5B;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA,YAAY;AAAA,MACZ,YAAY;AAAA,IACd,CAAC;AAED,WAAO,EAAE,MAAM,UAAU,OAAO,WAAW,IAAI;AAAA,EACjD;AAAA,EAED,MAAa,oBACV,QACA,UACA,UACe;AACf,UAAM,KAAK,YAAY,UAAU,EAAE,UAAU,OAAO,GAAG;AAAA,MACrD,YAAY,KAAK,IAAI;AAAA,IACvB,CAAC;AACD,eAAW,WAAW,UAAU;AAC9B,YAAM,YAAQ,+BAAW;AACzB,YAAM,KAAK,UAAU,OAAO;AAAA,QAC1B;AAAA,QACA,QAAQ;AAAA,QACR;AAAA,QACA,MAAM,QAAQ;AAAA,QACd,SAAS,QAAQ;AAAA,QACjB,WAAW,QAAQ;AAAA,QACnB,UAAU,QAAQ;AAAA,MACpB,CAAC;AAAA,IACH;AAAA,EACF;AAAA,EAED,MAAa,aAAa,QAAgB,UAAiC;AAC1E,UAAM,QAAQ,MAAM,KAAK,UAAU,KAAK,EAAE,QAAQ,SAAS,CAAC,EAAE,KAAK;AAAA,MAClE,WAAW;AAAA,IACZ,CAAC;AAED,WAAO,QAAQ,CAAC,SAAuB;AACnC,WAAK,UAAU;AAAA,IACnB,CAAC;AAEC,UAAM,SAAS,MAAM,KAAK,YAAY,QAAQ,EAAE,QAAQ,SAAS,CAAC;AAClE,YAAQ,UAAU;AAAA,EACpB;AAAA,EAED,MAAa,YAAY,QAA2C;AACjE,UAAM,UAAU,MAAM,KAAK,YAAY,KAAK,EAAE,OAAO,CAAC,EAAE,KAAK;AAAA,MAC3D,YAAY;AAAA,IACd,CAAC;AACD,UAAM,OAAyB,QAAQ,IAAI,CAAC,WAA2B;AACrE,aAAO;AAAA,QACL,MAAM,OAAO;AAAA,QACb,UAAU,OAAO;AAAA,QACjB,OAAO,OAAO;AAAA,QACd,WAAW,OAAO;AAAA,MACpB;AAAA,IACF,CAAC;AACD,WAAO;AAAA,EACT;AACF;;;AG7HA,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","import_logger","import_node_crypto","import_mongoose","mongoose"]}

package/dist/index.d.cts

@@ -1,5 +1,5 @@
-import { SessionObject, SessionMetadata, ChatObject, Intent } from '@ainetwork/adk/types/memory';
-import { ISessionMemory, IIntentMemory } from '@ainetwork/adk/modules';
+import { ThreadType, ThreadObject, ThreadMetadata, MessageObject, Intent } from '@ainetwork/adk/types/memory';
+import { IThreadMemory, IIntentMemory } from '@ainetwork/adk/modules';
 import { IMemory } from 'node_modules/@ainetwork/adk/dist/esm/modules/memory/base.memory';
 import { Mongoose } from 'mongoose';
 
@@ -14,15 +14,15 @@ declare class MongoDBMemory implements IMemory {
     isConnected(): boolean;
 }
 
-declare class MongoDBSession extends MongoDBMemory implements ISessionMemory {
+declare class MongoDBThread extends MongoDBMemory implements IThreadMemory {
     private chatModel;
-    private sessionModel;
+    private threadModel;
     constructor(uri: string);
-    getSession(userId: string, sessionId: string): Promise<SessionObject | undefined>;
-    createSession(userId: string, sessionId: string, title: string): Promise<SessionMetadata>;
-    addChatToSession(userId: string, sessionId: string, chat: ChatObject): Promise<void>;
-    deleteSession(userId: string, sessionId: string): Promise<void>;
-    listSessions(userId: string): Promise<SessionMetadata[]>;
+    getThread(type: ThreadType, userId: string, threadId: string): Promise<ThreadObject | undefined>;
+    createThread(type: ThreadType, userId: string, threadId: string, title: string): Promise<ThreadMetadata>;
+    addMessagesToThread(userId: string, threadId: string, messages: MessageObject[]): Promise<void>;
+    deleteThread(userId: string, threadId: string): Promise<void>;
+    listThreads(userId: string): Promise<ThreadMetadata[]>;
 }
 
 declare class MongoDBIntent extends MongoDBMemory implements IIntentMemory {
@@ -33,4 +33,4 @@ declare class MongoDBIntent extends MongoDBMemory implements IIntentMemory {
     listIntents(): Promise<Intent[]>;
 }
 
-export { MongoDBIntent, MongoDBSession };
+export { MongoDBIntent, MongoDBThread };

package/dist/index.d.ts

@@ -1,5 +1,5 @@
-import { SessionObject, SessionMetadata, ChatObject, Intent } from '@ainetwork/adk/types/memory';
-import { ISessionMemory, IIntentMemory } from '@ainetwork/adk/modules';
+import { ThreadType, ThreadObject, ThreadMetadata, MessageObject, Intent } from '@ainetwork/adk/types/memory';
+import { IThreadMemory, IIntentMemory } from '@ainetwork/adk/modules';
 import { IMemory } from 'node_modules/@ainetwork/adk/dist/esm/modules/memory/base.memory';
 import { Mongoose } from 'mongoose';
 
@@ -14,15 +14,15 @@ declare class MongoDBMemory implements IMemory {
     isConnected(): boolean;
 }
 
-declare class MongoDBSession extends MongoDBMemory implements ISessionMemory {
+declare class MongoDBThread extends MongoDBMemory implements IThreadMemory {
     private chatModel;
-    private sessionModel;
+    private threadModel;
     constructor(uri: string);
-    getSession(userId: string, sessionId: string): Promise<SessionObject | undefined>;
-    createSession(userId: string, sessionId: string, title: string): Promise<SessionMetadata>;
-    addChatToSession(userId: string, sessionId: string, chat: ChatObject): Promise<void>;
-    deleteSession(userId: string, sessionId: string): Promise<void>;
-    listSessions(userId: string): Promise<SessionMetadata[]>;
+    getThread(type: ThreadType, userId: string, threadId: string): Promise<ThreadObject | undefined>;
+    createThread(type: ThreadType, userId: string, threadId: string, title: string): Promise<ThreadMetadata>;
+    addMessagesToThread(userId: string, threadId: string, messages: MessageObject[]): Promise<void>;
+    deleteThread(userId: string, threadId: string): Promise<void>;
+    listThreads(userId: string): Promise<ThreadMetadata[]>;
 }
 
 declare class MongoDBIntent extends MongoDBMemory implements IIntentMemory {
@@ -33,4 +33,4 @@ declare class MongoDBIntent extends MongoDBMemory implements IIntentMemory {
     listIntents(): Promise<Intent[]>;
 }
 
-export { MongoDBIntent, MongoDBSession };
+export { MongoDBIntent, MongoDBThread };

package/dist/index.js

@@ -1,12 +1,12 @@
 import {
   ChatObjectSchema,
-  SessionObjectSchema
-} from "./chunk-N5FAVJ4Z.js";
+  ThreadObjectSchema
+} from "./chunk-S537KQMN.js";
 import {
   IntentModel
 } from "./chunk-JCE5NAY5.js";
 
-// implements/session.memory.ts
+// implements/thread.memory.ts
 import { randomUUID } from "crypto";
 
 // implements/base.memory.ts
@@ -54,81 +54,89 @@ var MongoDBMemory = class {
   }
 };
 
-// implements/session.memory.ts
+// implements/thread.memory.ts
 import { loggers as loggers2 } from "@ainetwork/adk/utils/logger";
-var MongoDBSession = class extends MongoDBMemory {
+var MongoDBThread = class extends MongoDBMemory {
   chatModel;
-  sessionModel;
+  threadModel;
   constructor(uri) {
     super(uri);
     const _mongoose = super.getInstance();
     this.chatModel = _mongoose.model("Chat", ChatObjectSchema);
-    this.sessionModel = _mongoose.model("Session", SessionObjectSchema);
+    this.threadModel = _mongoose.model("Thread", ThreadObjectSchema);
   }
-  async getSession(userId, sessionId) {
-    const chats = await this.chatModel.find({ sessionId, userId }).sort({
+  async getThread(type, userId, threadId) {
+    const thread = await this.threadModel.findOne({ type, threadId, userId });
+    const chats = await this.chatModel.find({ threadId, userId }).sort({
       timestamp: 1
     });
-    const session = await this.sessionModel.findOne({ sessionId, userId });
-    loggers2.agent.debug(`Found ${chats.length} chats for session ${sessionId}`);
-    const sessionObject = { chats: {} };
-    sessionObject.title = session?.title;
+    if (!thread) return void 0;
+    loggers2.agent.debug(`Found ${chats.length} chats for thread ${threadId}`);
+    const threadObject = {
+      type: thread.type,
+      title: thread.title || "New chats",
+      messages: {}
+    };
     chats.forEach((chat) => {
       const chatId = chat._id?.toString() || chat.id;
-      sessionObject.chats[chatId] = {
+      threadObject.messages[chatId] = {
         role: chat.role,
         content: chat.content,
         timestamp: chat.timestamp,
         metadata: chat.metadata
       };
     });
-    return sessionObject;
+    return threadObject;
   }
-  async createSession(userId, sessionId, title) {
+  async createThread(type, userId, threadId, title) {
     const now = Date.now();
-    await this.sessionModel.create({
-      sessionId,
+    await this.threadModel.create({
+      type,
       userId,
+      threadId,
       title,
       updated_at: now,
       created_at: now
     });
-    return { title, sessionId, updatedAt: now };
+    return { type, threadId, title, updatedAt: now };
   }
-  async addChatToSession(userId, sessionId, chat) {
-    const newId = randomUUID();
-    await this.sessionModel.updateOne({ sessionId, userId }, {
+  async addMessagesToThread(userId, threadId, messages) {
+    await this.threadModel.updateOne({ threadId, 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
-    });
+    for (const message of messages) {
+      const newId = randomUUID();
+      await this.chatModel.create({
+        threadId,
+        chatId: newId,
+        userId,
+        role: message.role,
+        content: message.content,
+        timestamp: message.timestamp,
+        metadata: message.metadata
+      });
+    }
   }
-  async deleteSession(userId, sessionId) {
-    const chats = await this.chatModel.find({ userId, sessionId }).sort({
+  async deleteThread(userId, threadId) {
+    const chats = await this.chatModel.find({ userId, threadId }).sort({
       timestamp: 1
     });
     chats?.forEach((chat) => {
       chat.deleteOne();
     });
-    const session = await this.sessionModel.findOne({ sessionId, userId });
-    session?.deleteOne();
+    const thread = await this.threadModel.findOne({ userId, threadId });
+    thread?.deleteOne();
   }
-  async listSessions(userId) {
-    const sessions = await this.sessionModel.find({ userId }).sort({
+  async listThreads(userId) {
+    const threads = await this.threadModel.find({ userId }).sort({
       updated_at: -1
     });
-    const data = sessions.map((session) => {
+    const data = threads.map((thread) => {
       return {
-        sessionId: session.sessionId,
-        title: session.title,
-        updatedAt: session.updated_at
+        type: thread.type,
+        threadId: thread.threadId,
+        title: thread.title,
+        updatedAt: thread.updated_at
       };
     });
     return data;
@@ -185,6 +193,6 @@ var MongoDBIntent = class extends MongoDBMemory {
 };
 export {
   MongoDBIntent,
-  MongoDBSession
+  MongoDBThread
 };
 //# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../implements/session.memory.ts","../implements/base.memory.ts","../implements/intent.memory.ts"],"sourcesContent":["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(userId: string, sessionId: string): Promise<SessionObject | undefined> {\n\t\tconst chats = await this.chatModel.find({ sessionId, userId }).sort({\n\t\t\ttimestamp: 1,\n\t\t});\n    const session = await this.sessionModel.findOne({ sessionId, userId });\n\n\t\tloggers.agent.debug(`Found ${chats.length} chats for session ${sessionId}`);\n\n\t\tconst sessionObject: SessionObject = { chats: {} };\n    sessionObject.title = session?.title;\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, title: string): Promise<SessionMetadata> {\n    const now = Date.now();\n    await this.sessionModel.create({\n      sessionId,\n      userId,\n      title,\n      updated_at: now,\n      created_at: now,\n    });\n\n    return { title, sessionId, updatedAt: now };\n  };\n\n\tpublic async addChatToSession(userId: string, sessionId: string, chat: ChatObject): Promise<void> {\n    const newId = randomUUID();\n    await this.sessionModel.updateOne({ sessionId, userId }, {\n      updated_at: Date.now(),\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 { 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}"],"mappings":";;;;;;;;;AAAA,SAAS,kBAAkB;;;ACC3B,SAAmB,gBAAgB;AACnC,SAAS,eAAe;AAEjB,IAAM,gBAAN,MAAuC;AAAA,EACpC,eAAwB;AAAA,EACxB;AAAA,EACA;AAAA,EAER,YAAY,KAAa;AACvB,SAAK,OAAO;AACZ,SAAK,YAAY,IAAI,SAAS;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,cAAQ,MAAM,KAAK,gCAAgC;AAAA,IACpD,SAAS,OAAO;AACf,cAAQ,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,cAAQ,MAAM,KAAK,mCAAmC;AAAA,IACvD,SAAS,OAAO;AACf,cAAQ,MAAM,MAAM,sCAAsC,KAAK;AAC/D,YAAM;AAAA,IACP;AAAA,EACA;AAAA,EAEO,cAAuB;AAC5B,WAAO,KAAK;AAAA,EACd;AACF;;;ADxCA,SAAS,WAAAA,gBAAe;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,QAAgB,WAAuD;AAC/F,UAAM,QAAQ,MAAM,KAAK,UAAU,KAAK,EAAE,WAAW,OAAO,CAAC,EAAE,KAAK;AAAA,MACnE,WAAW;AAAA,IACZ,CAAC;AACC,UAAM,UAAU,MAAM,KAAK,aAAa,QAAQ,EAAE,WAAW,OAAO,CAAC;AAEvE,IAAAA,SAAQ,MAAM,MAAM,SAAS,MAAM,MAAM,sBAAsB,SAAS,EAAE;AAE1E,UAAM,gBAA+B,EAAE,OAAO,CAAC,EAAE;AAC/C,kBAAc,QAAQ,SAAS;AACjC,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,WAAmB,OAAyC;AACpG,UAAM,MAAM,KAAK,IAAI;AACrB,UAAM,KAAK,aAAa,OAAO;AAAA,MAC7B;AAAA,MACA;AAAA,MACA;AAAA,MACA,YAAY;AAAA,MACZ,YAAY;AAAA,IACd,CAAC;AAED,WAAO,EAAE,OAAO,WAAW,WAAW,IAAI;AAAA,EAC5C;AAAA,EAED,MAAa,iBAAiB,QAAgB,WAAmB,MAAiC;AAC/F,UAAM,QAAQ,WAAW;AACzB,UAAM,KAAK,aAAa,UAAU,EAAE,WAAW,OAAO,GAAG;AAAA,MACvD,YAAY,KAAK,IAAI;AAAA,IACvB,CAAC;AACH,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;;;AEvGA,SAAS,cAAAC,mBAAkB;AAMpB,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,QAAQC,YAAW;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":["loggers","randomUUID","randomUUID"]}
+{"version":3,"sources":["../implements/thread.memory.ts","../implements/base.memory.ts","../implements/intent.memory.ts"],"sourcesContent":["import { randomUUID } from \"node:crypto\";\nimport type { MessageObject, ThreadMetadata, ThreadObject, ThreadType } from \"@ainetwork/adk/types/memory\";\nimport { MessageRole } from \"@ainetwork/adk/types/memory\";\nimport { IThreadMemory } from \"@ainetwork/adk/modules\";\nimport { MongoDBMemory } from \"./base.memory\";\nimport {\n\tChatDocument,\n  ChatObjectSchema,\n  ThreadObjectSchema,\n  ThreadDocument\n} from \"../models/chats.model\";\nimport { loggers } from \"@ainetwork/adk/utils/logger\";\nimport { Model } from \"mongoose\";\n\nexport class MongoDBThread extends MongoDBMemory implements IThreadMemory {\n  private chatModel: Model<ChatDocument>;\n  private threadModel: Model<ThreadDocument>;\n\n  constructor(uri: string) {\n    super(uri);\n    const _mongoose = super.getInstance();\n    this.chatModel = _mongoose.model<ChatDocument>(\"Chat\", ChatObjectSchema);\n    this.threadModel = _mongoose.model<ThreadDocument>(\"Thread\", ThreadObjectSchema);\n  }\n\n  public async getThread(\n    type: ThreadType,\n    userId: string,\n    threadId: string\n  ): Promise<ThreadObject | undefined> {\n    const thread = await this.threadModel.findOne({ type, threadId, userId });\n\t\tconst chats = await this.chatModel.find({ threadId, userId }).sort({\n\t\t\ttimestamp: 1,\n\t\t});\n\n    if (!thread) return undefined;\n\n\t\tloggers.agent.debug(`Found ${chats.length} chats for thread ${threadId}`);\n\n\t\tconst threadObject: ThreadObject = { \n      type: thread.type as ThreadType,\n      title: thread.title || \"New chats\",\n      messages: {}\n    };\n\t\tchats.forEach((chat: ChatDocument) => {\n\t\t\tconst chatId = chat._id?.toString() || chat.id;\n\t\t\tthreadObject.messages[chatId] = {\n\t\t\t\trole: chat.role as MessageRole,\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 threadObject;\n  };\n\n\tpublic async createThread(\n\t\ttype: ThreadType,\n\t\tuserId: string,\n\t\tthreadId: string,\n\t\ttitle: string,\n  ): Promise<ThreadMetadata> {\n    const now = Date.now();\n    await this.threadModel.create({\n      type,\n      userId,\n      threadId,\n      title,\n      updated_at: now,\n      created_at: now,\n    });\n\n    return { type, threadId, title, updatedAt: now };\n  };\n\n\tpublic async addMessagesToThread(\n    userId: string,\n    threadId: string,\n    messages: MessageObject[]\n  ): Promise<void> {\n    await this.threadModel.updateOne({ threadId, userId }, {\n      updated_at: Date.now(),\n    });\n    for (const message of messages) {\n      const newId = randomUUID();\n      await this.chatModel.create({\n        threadId,\n        chatId: newId,\n        userId,\n        role: message.role,\n        content: message.content,\n        timestamp: message.timestamp,\n        metadata: message.metadata,\n      });\n    }\n  };\n\n\tpublic async deleteThread(userId: string, threadId: string): Promise<void> {\n\t\tconst chats = await this.chatModel.find({ userId, threadId }).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 thread = await this.threadModel.findOne({ userId, threadId });\n    thread?.deleteOne();\n  };\n\n\tpublic async listThreads(userId: string): Promise<ThreadMetadata[]> {\n    const threads = await this.threadModel.find({ userId }).sort({\n      updated_at: -1,\n    });\n    const data: ThreadMetadata[] = threads.map((thread: ThreadDocument) => {\n      return {\n        type: thread.type,\n        threadId: thread.threadId,\n        title: thread.title,\n        updatedAt: thread.updated_at\n      } as ThreadMetadata;\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 { 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}"],"mappings":";;;;;;;;;AAAA,SAAS,kBAAkB;;;ACC3B,SAAmB,gBAAgB;AACnC,SAAS,eAAe;AAEjB,IAAM,gBAAN,MAAuC;AAAA,EACpC,eAAwB;AAAA,EACxB;AAAA,EACA;AAAA,EAER,YAAY,KAAa;AACvB,SAAK,OAAO;AACZ,SAAK,YAAY,IAAI,SAAS;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,cAAQ,MAAM,KAAK,gCAAgC;AAAA,IACpD,SAAS,OAAO;AACf,cAAQ,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,cAAQ,MAAM,KAAK,mCAAmC;AAAA,IACvD,SAAS,OAAO;AACf,cAAQ,MAAM,MAAM,sCAAsC,KAAK;AAC/D,YAAM;AAAA,IACP;AAAA,EACA;AAAA,EAEO,cAAuB;AAC5B,WAAO,KAAK;AAAA,EACd;AACF;;;ADxCA,SAAS,WAAAA,gBAAe;AAGjB,IAAM,gBAAN,cAA4B,cAAuC;AAAA,EAChE;AAAA,EACA;AAAA,EAER,YAAY,KAAa;AACvB,UAAM,GAAG;AACT,UAAM,YAAY,MAAM,YAAY;AACpC,SAAK,YAAY,UAAU,MAAoB,QAAQ,gBAAgB;AACvE,SAAK,cAAc,UAAU,MAAsB,UAAU,kBAAkB;AAAA,EACjF;AAAA,EAEA,MAAa,UACX,MACA,QACA,UACmC;AACnC,UAAM,SAAS,MAAM,KAAK,YAAY,QAAQ,EAAE,MAAM,UAAU,OAAO,CAAC;AAC1E,UAAM,QAAQ,MAAM,KAAK,UAAU,KAAK,EAAE,UAAU,OAAO,CAAC,EAAE,KAAK;AAAA,MAClE,WAAW;AAAA,IACZ,CAAC;AAEC,QAAI,CAAC,OAAQ,QAAO;AAEtB,IAAAA,SAAQ,MAAM,MAAM,SAAS,MAAM,MAAM,qBAAqB,QAAQ,EAAE;AAExE,UAAM,eAA6B;AAAA,MAC/B,MAAM,OAAO;AAAA,MACb,OAAO,OAAO,SAAS;AAAA,MACvB,UAAU,CAAC;AAAA,IACb;AACF,UAAM,QAAQ,CAAC,SAAuB;AACrC,YAAM,SAAS,KAAK,KAAK,SAAS,KAAK,KAAK;AAC5C,mBAAa,SAAS,MAAM,IAAI;AAAA,QAC/B,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,aACZ,MACA,QACA,UACA,OAC2B;AACzB,UAAM,MAAM,KAAK,IAAI;AACrB,UAAM,KAAK,YAAY,OAAO;AAAA,MAC5B;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA,YAAY;AAAA,MACZ,YAAY;AAAA,IACd,CAAC;AAED,WAAO,EAAE,MAAM,UAAU,OAAO,WAAW,IAAI;AAAA,EACjD;AAAA,EAED,MAAa,oBACV,QACA,UACA,UACe;AACf,UAAM,KAAK,YAAY,UAAU,EAAE,UAAU,OAAO,GAAG;AAAA,MACrD,YAAY,KAAK,IAAI;AAAA,IACvB,CAAC;AACD,eAAW,WAAW,UAAU;AAC9B,YAAM,QAAQ,WAAW;AACzB,YAAM,KAAK,UAAU,OAAO;AAAA,QAC1B;AAAA,QACA,QAAQ;AAAA,QACR;AAAA,QACA,MAAM,QAAQ;AAAA,QACd,SAAS,QAAQ;AAAA,QACjB,WAAW,QAAQ;AAAA,QACnB,UAAU,QAAQ;AAAA,MACpB,CAAC;AAAA,IACH;AAAA,EACF;AAAA,EAED,MAAa,aAAa,QAAgB,UAAiC;AAC1E,UAAM,QAAQ,MAAM,KAAK,UAAU,KAAK,EAAE,QAAQ,SAAS,CAAC,EAAE,KAAK;AAAA,MAClE,WAAW;AAAA,IACZ,CAAC;AAED,WAAO,QAAQ,CAAC,SAAuB;AACnC,WAAK,UAAU;AAAA,IACnB,CAAC;AAEC,UAAM,SAAS,MAAM,KAAK,YAAY,QAAQ,EAAE,QAAQ,SAAS,CAAC;AAClE,YAAQ,UAAU;AAAA,EACpB;AAAA,EAED,MAAa,YAAY,QAA2C;AACjE,UAAM,UAAU,MAAM,KAAK,YAAY,KAAK,EAAE,OAAO,CAAC,EAAE,KAAK;AAAA,MAC3D,YAAY;AAAA,IACd,CAAC;AACD,UAAM,OAAyB,QAAQ,IAAI,CAAC,WAA2B;AACrE,aAAO;AAAA,QACL,MAAM,OAAO;AAAA,QACb,UAAU,OAAO;AAAA,QACjB,OAAO,OAAO;AAAA,QACd,WAAW,OAAO;AAAA,MACpB;AAAA,IACF,CAAC;AACD,WAAO;AAAA,EACT;AACF;;;AE7HA,SAAS,cAAAC,mBAAkB;AAMpB,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,QAAQC,YAAW;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":["loggers","randomUUID","randomUUID"]}

package/dist/models/chats.model.cjs

@@ -22,20 +22,19 @@ var chats_model_exports = {};
 __export(chats_model_exports, {
   ChatContentObjectSchema: () => ChatContentObjectSchema,
   ChatObjectSchema: () => ChatObjectSchema,
-  ChatRole: () => ChatRole,
-  SessionObjectSchema: () => SessionObjectSchema
+  ThreadObjectSchema: () => ThreadObjectSchema
 });
 module.exports = __toCommonJS(chats_model_exports);
+var import_memory = require("@ainetwork/adk/types/memory");
 var import_mongoose = require("mongoose");
-var ChatRole = /* @__PURE__ */ ((ChatRole2) => {
-  ChatRole2["USER"] = "USER";
-  ChatRole2["SYSTEM"] = "SYSTEM";
-  ChatRole2["MODEL"] = "MODEL";
-  return ChatRole2;
-})(ChatRole || {});
-var SessionObjectSchema = new import_mongoose.Schema(
+var ThreadObjectSchema = new import_mongoose.Schema(
   {
-    sessionId: {
+    type: {
+      type: String,
+      enum: Object.values(import_memory.ThreadType),
+      required: true
+    },
+    threadId: {
       type: String,
       required: true,
       index: true
@@ -68,7 +67,7 @@ var ChatContentObjectSchema = new import_mongoose.Schema(
 );
 var ChatObjectSchema = new import_mongoose.Schema(
   {
-    sessionId: {
+    threadId: {
       type: String,
       required: true,
       index: true
@@ -80,7 +79,7 @@ var ChatObjectSchema = new import_mongoose.Schema(
     },
     role: {
       type: String,
-      enum: Object.values(ChatRole),
+      enum: Object.values(import_memory.MessageRole),
       required: true
     },
     content: {
@@ -101,7 +100,6 @@ var ChatObjectSchema = new import_mongoose.Schema(
 0 && (module.exports = {
   ChatContentObjectSchema,
   ChatObjectSchema,
-  ChatRole,
-  SessionObjectSchema
+  ThreadObjectSchema
 });
 //# sourceMappingURL=chats.model.cjs.map

package/dist/models/chats.model.cjs.map

@@ -1 +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\tuserId: {\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;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,sBAAsC;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,uBAAO,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,QAAQ;AAAA,MACP,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;AACD;","names":["ChatRole"]}
+{"version":3,"sources":["../../models/chats.model.ts"],"sourcesContent":["import { MessageRole, ThreadType } from \"@ainetwork/adk/types/memory\";\nimport { type Document, Schema } from \"mongoose\";\n\nexport const ThreadObjectSchema = new Schema(\n\t{\n\t\ttype: {\n\t\t\ttype: String,\n\t\t\tenum: Object.values(ThreadType),\n\t\t\trequired: true,\n\t\t},\n\t\tthreadId: {\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 ThreadDocument extends Document {\n\ttype: ThreadType;\n\tthreadId: 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\tthreadId: {\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\trole: {\n\t\t\ttype: String,\n\t\t\tenum: Object.values(MessageRole),\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\tthreadId: string;\n\trole: MessageRole;\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;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,oBAAwC;AACxC,sBAAsC;AAE/B,IAAM,qBAAqB,IAAI;AAAA,EACrC;AAAA,IACC,MAAM;AAAA,MACL,MAAM;AAAA,MACN,MAAM,OAAO,OAAO,wBAAU;AAAA,MAC9B,UAAU;AAAA,IACX;AAAA,IACA,UAAU;AAAA,MACT,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;AAYO,IAAM,0BAA0B,IAAI;AAAA,EAC1C;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;AAGO,IAAM,mBAAmB,IAAI;AAAA,EACnC;AAAA,IACC,UAAU;AAAA,MACT,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,MAAM;AAAA,MACL,MAAM;AAAA,MACN,MAAM,OAAO,OAAO,yBAAW;AAAA,MAC/B,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;AACD;","names":[]}

package/dist/models/chats.model.d.cts

@@ -1,25 +1,24 @@
 import * as mongoose from 'mongoose';
 import { Schema, Document } from 'mongoose';
+import { ThreadType, MessageRole } from '@ainetwork/adk/types/memory';
 
-declare enum ChatRole {
-    USER = "USER",
-    SYSTEM = "SYSTEM",
-    MODEL = "MODEL"
-}
-declare const SessionObjectSchema: Schema<any, mongoose.Model<any, any, any, any, any, any>, {}, {}, {}, {}, mongoose.DefaultSchemaOptions, {
-    sessionId: string;
+declare const ThreadObjectSchema: Schema<any, mongoose.Model<any, any, any, any, any, any>, {}, {}, {}, {}, mongoose.DefaultSchemaOptions, {
+    type: ThreadType;
+    threadId: string;
     userId: string;
     created_at: number;
     updated_at: number;
     title?: string | null | undefined;
 }, Document<unknown, {}, mongoose.FlatRecord<{
-    sessionId: string;
+    type: ThreadType;
+    threadId: string;
     userId: string;
     created_at: number;
     updated_at: number;
     title?: string | null | undefined;
 }>, {}> & mongoose.FlatRecord<{
-    sessionId: string;
+    type: ThreadType;
+    threadId: string;
     userId: string;
     created_at: number;
     updated_at: number;
@@ -29,10 +28,11 @@ declare const SessionObjectSchema: Schema<any, mongoose.Model<any, any, any, any
 } & {
     __v: number;
 }>;
-interface SessionDocument extends Document {
-    sessionId: string;
+interface ThreadDocument extends Document {
+    type: ThreadType;
+    threadId: string;
     userId: string;
-    title?: string;
+    title: string;
     created_at: number;
     updated_at: number;
 }
@@ -53,9 +53,9 @@ declare const ChatContentObjectSchema: Schema<any, mongoose.Model<any, any, any,
     __v: number;
 }>;
 declare const ChatObjectSchema: Schema<any, mongoose.Model<any, any, any, any, any, any>, {}, {}, {}, {}, mongoose.DefaultSchemaOptions, {
-    sessionId: string;
+    threadId: string;
     userId: string;
-    role: ChatRole;
+    role: MessageRole;
     content: {
         type: string;
         parts: any[];
@@ -63,9 +63,9 @@ declare const ChatObjectSchema: Schema<any, mongoose.Model<any, any, any, any, a
     timestamp: number;
     metadata: any;
 }, Document<unknown, {}, mongoose.FlatRecord<{
-    sessionId: string;
+    threadId: string;
     userId: string;
-    role: ChatRole;
+    role: MessageRole;
     content: {
         type: string;
         parts: any[];
@@ -73,9 +73,9 @@ declare const ChatObjectSchema: Schema<any, mongoose.Model<any, any, any, any, a
     timestamp: number;
     metadata: any;
 }>, {}> & mongoose.FlatRecord<{
-    sessionId: string;
+    threadId: string;
     userId: string;
-    role: ChatRole;
+    role: MessageRole;
     content: {
         type: string;
         parts: any[];
@@ -88,8 +88,8 @@ declare const ChatObjectSchema: Schema<any, mongoose.Model<any, any, any, any, a
     __v: number;
 }>;
 interface ChatDocument extends Document {
-    sessionId: string;
-    role: ChatRole;
+    threadId: string;
+    role: MessageRole;
     content: {
         type: string;
         parts: any[];
@@ -102,4 +102,4 @@ interface ChatDocument extends Document {
     updatedAt: Date;
 }
 
-export { ChatContentObjectSchema, type ChatDocument, ChatObjectSchema, ChatRole, type SessionDocument, SessionObjectSchema };
+export { ChatContentObjectSchema, type ChatDocument, ChatObjectSchema, type ThreadDocument, ThreadObjectSchema };

package/dist/models/chats.model.d.ts

@@ -1,25 +1,24 @@
 import * as mongoose from 'mongoose';
 import { Schema, Document } from 'mongoose';
+import { ThreadType, MessageRole } from '@ainetwork/adk/types/memory';
 
-declare enum ChatRole {
-    USER = "USER",
-    SYSTEM = "SYSTEM",
-    MODEL = "MODEL"
-}
-declare const SessionObjectSchema: Schema<any, mongoose.Model<any, any, any, any, any, any>, {}, {}, {}, {}, mongoose.DefaultSchemaOptions, {
-    sessionId: string;
+declare const ThreadObjectSchema: Schema<any, mongoose.Model<any, any, any, any, any, any>, {}, {}, {}, {}, mongoose.DefaultSchemaOptions, {
+    type: ThreadType;
+    threadId: string;
     userId: string;
     created_at: number;
     updated_at: number;
     title?: string | null | undefined;
 }, Document<unknown, {}, mongoose.FlatRecord<{
-    sessionId: string;
+    type: ThreadType;
+    threadId: string;
     userId: string;
     created_at: number;
     updated_at: number;
     title?: string | null | undefined;
 }>, {}> & mongoose.FlatRecord<{
-    sessionId: string;
+    type: ThreadType;
+    threadId: string;
     userId: string;
     created_at: number;
     updated_at: number;
@@ -29,10 +28,11 @@ declare const SessionObjectSchema: Schema<any, mongoose.Model<any, any, any, any
 } & {
     __v: number;
 }>;
-interface SessionDocument extends Document {
-    sessionId: string;
+interface ThreadDocument extends Document {
+    type: ThreadType;
+    threadId: string;
     userId: string;
-    title?: string;
+    title: string;
     created_at: number;
     updated_at: number;
 }
@@ -53,9 +53,9 @@ declare const ChatContentObjectSchema: Schema<any, mongoose.Model<any, any, any,
     __v: number;
 }>;
 declare const ChatObjectSchema: Schema<any, mongoose.Model<any, any, any, any, any, any>, {}, {}, {}, {}, mongoose.DefaultSchemaOptions, {
-    sessionId: string;
+    threadId: string;
     userId: string;
-    role: ChatRole;
+    role: MessageRole;
     content: {
         type: string;
         parts: any[];
@@ -63,9 +63,9 @@ declare const ChatObjectSchema: Schema<any, mongoose.Model<any, any, any, any, a
     timestamp: number;
     metadata: any;
 }, Document<unknown, {}, mongoose.FlatRecord<{
-    sessionId: string;
+    threadId: string;
     userId: string;
-    role: ChatRole;
+    role: MessageRole;
     content: {
         type: string;
         parts: any[];
@@ -73,9 +73,9 @@ declare const ChatObjectSchema: Schema<any, mongoose.Model<any, any, any, any, a
     timestamp: number;
     metadata: any;
 }>, {}> & mongoose.FlatRecord<{
-    sessionId: string;
+    threadId: string;
     userId: string;
-    role: ChatRole;
+    role: MessageRole;
     content: {
         type: string;
         parts: any[];
@@ -88,8 +88,8 @@ declare const ChatObjectSchema: Schema<any, mongoose.Model<any, any, any, any, a
     __v: number;
 }>;
 interface ChatDocument extends Document {
-    sessionId: string;
-    role: ChatRole;
+    threadId: string;
+    role: MessageRole;
     content: {
         type: string;
         parts: any[];
@@ -102,4 +102,4 @@ interface ChatDocument extends Document {
     updatedAt: Date;
 }
 
-export { ChatContentObjectSchema, type ChatDocument, ChatObjectSchema, ChatRole, type SessionDocument, SessionObjectSchema };
+export { ChatContentObjectSchema, type ChatDocument, ChatObjectSchema, type ThreadDocument, ThreadObjectSchema };

package/dist/models/chats.model.js

@@ -1,13 +1,11 @@
 import {
   ChatContentObjectSchema,
   ChatObjectSchema,
-  ChatRole,
-  SessionObjectSchema
-} from "../chunk-N5FAVJ4Z.js";
+  ThreadObjectSchema
+} from "../chunk-S537KQMN.js";
 export {
   ChatContentObjectSchema,
   ChatObjectSchema,
-  ChatRole,
-  SessionObjectSchema
+  ThreadObjectSchema
 };
 //# sourceMappingURL=chats.model.js.map

package/implements/thread.memory.ts

@@ -0,0 +1,126 @@
+import { randomUUID } from "node:crypto";
+import type { MessageObject, ThreadMetadata, ThreadObject, ThreadType } from "@ainetwork/adk/types/memory";
+import { MessageRole } from "@ainetwork/adk/types/memory";
+import { IThreadMemory } from "@ainetwork/adk/modules";
+import { MongoDBMemory } from "./base.memory";
+import {
+	ChatDocument,
+  ChatObjectSchema,
+  ThreadObjectSchema,
+  ThreadDocument
+} from "../models/chats.model";
+import { loggers } from "@ainetwork/adk/utils/logger";
+import { Model } from "mongoose";
+
+export class MongoDBThread extends MongoDBMemory implements IThreadMemory {
+  private chatModel: Model<ChatDocument>;
+  private threadModel: Model<ThreadDocument>;
+
+  constructor(uri: string) {
+    super(uri);
+    const _mongoose = super.getInstance();
+    this.chatModel = _mongoose.model<ChatDocument>("Chat", ChatObjectSchema);
+    this.threadModel = _mongoose.model<ThreadDocument>("Thread", ThreadObjectSchema);
+  }
+
+  public async getThread(
+    type: ThreadType,
+    userId: string,
+    threadId: string
+  ): Promise<ThreadObject | undefined> {
+    const thread = await this.threadModel.findOne({ type, threadId, userId });
+		const chats = await this.chatModel.find({ threadId, userId }).sort({
+			timestamp: 1,
+		});
+
+    if (!thread) return undefined;
+
+		loggers.agent.debug(`Found ${chats.length} chats for thread ${threadId}`);
+
+		const threadObject: ThreadObject = { 
+      type: thread.type as ThreadType,
+      title: thread.title || "New chats",
+      messages: {}
+    };
+		chats.forEach((chat: ChatDocument) => {
+			const chatId = chat._id?.toString() || chat.id;
+			threadObject.messages[chatId] = {
+				role: chat.role as MessageRole,
+				content: chat.content,
+				timestamp: chat.timestamp,
+				metadata: chat.metadata,
+			};
+		});
+
+		return threadObject;
+  };
+
+	public async createThread(
+		type: ThreadType,
+		userId: string,
+		threadId: string,
+		title: string,
+  ): Promise<ThreadMetadata> {
+    const now = Date.now();
+    await this.threadModel.create({
+      type,
+      userId,
+      threadId,
+      title,
+      updated_at: now,
+      created_at: now,
+    });
+
+    return { type, threadId, title, updatedAt: now };
+  };
+
+	public async addMessagesToThread(
+    userId: string,
+    threadId: string,
+    messages: MessageObject[]
+  ): Promise<void> {
+    await this.threadModel.updateOne({ threadId, userId }, {
+      updated_at: Date.now(),
+    });
+    for (const message of messages) {
+      const newId = randomUUID();
+      await this.chatModel.create({
+        threadId,
+        chatId: newId,
+        userId,
+        role: message.role,
+        content: message.content,
+        timestamp: message.timestamp,
+        metadata: message.metadata,
+      });
+    }
+  };
+
+	public async deleteThread(userId: string, threadId: string): Promise<void> {
+		const chats = await this.chatModel.find({ userId, threadId }).sort({
+			timestamp: 1,
+		});
+
+		chats?.forEach((chat: ChatDocument) => {
+      chat.deleteOne();
+		});
+    
+    const thread = await this.threadModel.findOne({ userId, threadId });
+    thread?.deleteOne();
+  };
+
+	public async listThreads(userId: string): Promise<ThreadMetadata[]> {
+    const threads = await this.threadModel.find({ userId }).sort({
+      updated_at: -1,
+    });
+    const data: ThreadMetadata[] = threads.map((thread: ThreadDocument) => {
+      return {
+        type: thread.type,
+        threadId: thread.threadId,
+        title: thread.title,
+        updatedAt: thread.updated_at
+      } as ThreadMetadata;
+    })
+    return data;
+  };
+}

package/index.ts

@@ -1,2 +1,2 @@
-export { MongoDBSession } from "./implements/session.memory";
+export { MongoDBThread } from "./implements/thread.memory";
 export { MongoDBIntent } from "./implements/intent.memory";

package/models/chats.model.ts

@@ -1,15 +1,14 @@
+import { MessageRole, ThreadType } from "@ainetwork/adk/types/memory";
 import { type Document, Schema } from "mongoose";
 
-// ChatRole enum
-export enum ChatRole {
-	USER = "USER",
-	SYSTEM = "SYSTEM",
-	MODEL = "MODEL",
-}
-
-export const SessionObjectSchema = new Schema(
+export const ThreadObjectSchema = new Schema(
 	{
-		sessionId: {
+		type: {
+			type: String,
+			enum: Object.values(ThreadType),
+			required: true,
+		},
+		threadId: {
 			type: String,
 			required: true,
 			index: true,
@@ -34,10 +33,11 @@ export const SessionObjectSchema = new Schema(
 	},
 );
 
-export interface SessionDocument extends Document {
-	sessionId: string;
+export interface ThreadDocument extends Document {
+	type: ThreadType;
+	threadId: string;
 	userId: string;
-	title?: string;
+	title: string;
 	created_at: number;
 	updated_at: number;
 }
@@ -54,7 +54,7 @@ export const ChatContentObjectSchema = new Schema(
 // ChatObject schema - 개별 문서로 저장
 export const ChatObjectSchema = new Schema(
 	{
-		sessionId: {
+		threadId: {
 			type: String,
 			required: true,
 			index: true,
@@ -66,7 +66,7 @@ export const ChatObjectSchema = new Schema(
 		},
 		role: {
 			type: String,
-			enum: Object.values(ChatRole),
+			enum: Object.values(MessageRole),
 			required: true,
 		},
 		content: {
@@ -86,8 +86,8 @@ export const ChatObjectSchema = new Schema(
 
 // Chat Document interface
 export interface ChatDocument extends Document {
-	sessionId: string;
-	role: ChatRole;
+	threadId: string;
+	role: MessageRole;
 	content: {
 		type: string;
 		parts: any[];

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ainetwork/adk-provider-memory-mongodb",
-  "version": "0.1.6",
+  "version": "0.1.7",
   "author": "AI Network (https://ainetwork.ai)",
   "type": "module",
   "engines": {
@@ -13,12 +13,14 @@
     ".": {
       "types": "./dist/index.d.ts",
       "import": "./dist/index.js",
-      "require": "./dist/index.cjs"
+      "require": "./dist/index.cjs",
+      "default": "./dist/index.js"
     },
     "./models/*": {
       "types": "./dist/models/*.d.ts",
       "import": "./dist/models/*.js",
-      "require": "./dist/models/*.cjs"
+      "require": "./dist/models/*.cjs",
+      "default": "./dist/models/*.js"
     }
   },
   "scripts": {
@@ -26,7 +28,7 @@
     "clean": "rm -rf dist"
   },
   "dependencies": {
-    "@ainetwork/adk": "^0.1.9",
+    "@ainetwork/adk": "^0.1.13",
     "mongoose": "^8.16.5"
   },
   "devDependencies": {
@@ -36,5 +38,5 @@
   "publishConfig": {
     "access": "public"
   },
-  "gitHead": "14b5d3a6d314acc93de7263135f8a13ba70a22ff"
+  "gitHead": "d27f9542cbc4049593be4015b78e607e19ac22e6"
 }

package/dist/chunk-N5FAVJ4Z.js.map

@@ -1 +0,0 @@
-{"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\tuserId: {\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,QAAQ;AAAA,MACP,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/implements/session.memory.ts

@@ -1,104 +0,0 @@
-import { randomUUID } from "node:crypto";
-import type { ChatObject, SessionMetadata, SessionObject } from "@ainetwork/adk/types/memory";
-import { ISessionMemory } from "@ainetwork/adk/modules";
-import { MongoDBMemory } from "./base.memory";
-import {
-	ChatDocument,
-	ChatRole,
-  ChatObjectSchema,
-  SessionObjectSchema,
-  SessionDocument
-} from "../models/chats.model";
-import { loggers } from "@ainetwork/adk/utils/logger";
-import { Model } from "mongoose";
-
-export class MongoDBSession extends MongoDBMemory implements ISessionMemory {
-  private chatModel: Model<ChatDocument>;
-  private sessionModel: Model<SessionDocument>;
-
-  constructor(uri: string) {
-    super(uri);
-    const _mongoose = super.getInstance();
-    this.chatModel = _mongoose.model<ChatDocument>("Chat", ChatObjectSchema);
-    this.sessionModel = _mongoose.model<SessionDocument>("Session", SessionObjectSchema);
-  }
-
-  public async getSession(userId: string, sessionId: string): Promise<SessionObject | undefined> {
-		const chats = await this.chatModel.find({ sessionId, userId }).sort({
-			timestamp: 1,
-		});
-    const session = await this.sessionModel.findOne({ sessionId, userId });
-
-		loggers.agent.debug(`Found ${chats.length} chats for session ${sessionId}`);
-
-		const sessionObject: SessionObject = { chats: {} };
-    sessionObject.title = session?.title;
-		chats.forEach((chat: ChatDocument) => {
-			const chatId = chat._id?.toString() || chat.id;
-			sessionObject.chats[chatId] = {
-				role: chat.role as ChatRole,
-				content: chat.content,
-				timestamp: chat.timestamp,
-				metadata: chat.metadata,
-			};
-		});
-
-		return sessionObject;
-  };
-
-	public async createSession(userId: string, sessionId: string, title: string): Promise<SessionMetadata> {
-    const now = Date.now();
-    await this.sessionModel.create({
-      sessionId,
-      userId,
-      title,
-      updated_at: now,
-      created_at: now,
-    });
-
-    return { title, sessionId, updatedAt: now };
-  };
-
-	public async addChatToSession(userId: string, sessionId: string, chat: ChatObject): Promise<void> {
-    const newId = randomUUID();
-    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,
-		});
-  };
-
-	public async deleteSession(userId: string, sessionId: string): Promise<void> {
-		const chats = await this.chatModel.find({ userId, sessionId }).sort({
-			timestamp: 1,
-		});
-
-		chats?.forEach((chat: ChatDocument) => {
-      chat.deleteOne();
-		});
-    
-    const session = await this.sessionModel.findOne({ sessionId, userId });
-    session?.deleteOne();
-  };
-
-	public async listSessions(userId: string): Promise<SessionMetadata[]> {
-    const sessions = await this.sessionModel.find({ userId }).sort({
-      updated_at: -1,
-    });
-    const data: SessionMetadata[] = sessions.map((session: SessionDocument) => {
-      return {
-        sessionId: session.sessionId,
-        title: session.title,
-        updatedAt: session.updated_at
-      } as SessionMetadata;
-    })
-    return data;
-  };
-}