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

package/dist/index.js

@@ -1,34 +1,37 @@
-import {
-  ChatObjectSchema,
-  ThreadObjectSchema
-} from "./chunk-S537KQMN.js";
 import {
   IntentModel
 } from "./chunk-JCE5NAY5.js";
+import {
+  MessageModel
+} from "./chunk-MLPXPX2L.js";
+import {
+  ThreadModel
+} from "./chunk-GPOFS7ZT.js";
 
 // implements/thread.memory.ts
 import { randomUUID } from "crypto";
 
 // implements/base.memory.ts
-import { Mongoose } from "mongoose";
+import mongoose from "mongoose";
 import { loggers } from "@ainetwork/adk/utils/logger";
 var MongoDBMemory = class {
   _isConnected = false;
   _uri;
-  _mongoose;
   constructor(uri) {
     this._uri = uri;
-    this._mongoose = new Mongoose();
-  }
-  getInstance() {
-    return this._mongoose;
   }
   async connect() {
     if (this._isConnected) {
       return;
     }
     try {
-      await this._mongoose.connect(this._uri);
+      await mongoose.connect(this._uri, {
+        maxPoolSize: 1,
+        serverSelectionTimeoutMS: 3e4,
+        socketTimeoutMS: 45e3,
+        connectTimeoutMS: 3e4,
+        bufferCommands: false
+      });
       this._isConnected = true;
       loggers.agent.info("MongoDB connected successfully");
     } catch (error) {
@@ -41,7 +44,7 @@ var MongoDBMemory = class {
       return;
     }
     try {
-      await this._mongoose?.disconnect();
+      await mongoose.disconnect();
       this._isConnected = false;
       loggers.agent.info("MongoDB disconnected successfully");
     } catch (error) {
@@ -57,40 +60,35 @@ var MongoDBMemory = class {
 // implements/thread.memory.ts
 import { loggers as loggers2 } from "@ainetwork/adk/utils/logger";
 var MongoDBThread = class extends MongoDBMemory {
-  chatModel;
-  threadModel;
   constructor(uri) {
     super(uri);
-    const _mongoose = super.getInstance();
-    this.chatModel = _mongoose.model("Chat", ChatObjectSchema);
-    this.threadModel = _mongoose.model("Thread", ThreadObjectSchema);
   }
-  async getThread(type, userId, threadId) {
-    const thread = await this.threadModel.findOne({ type, threadId, userId });
-    const chats = await this.chatModel.find({ threadId, userId }).sort({
+  async getThread(userId, threadId) {
+    const thread = await ThreadModel.findOne({ threadId, userId });
+    const messages = await MessageModel.find({ threadId, userId }).sort({
       timestamp: 1
     });
     if (!thread) return void 0;
-    loggers2.agent.debug(`Found ${chats.length} chats for thread ${threadId}`);
+    loggers2.agent.debug(`Found ${messages.length} messages for thread ${threadId}`);
     const threadObject = {
       type: thread.type,
-      title: thread.title || "New chats",
+      title: thread.title || "New thread",
       messages: {}
     };
-    chats.forEach((chat) => {
-      const chatId = chat._id?.toString() || chat.id;
-      threadObject.messages[chatId] = {
-        role: chat.role,
-        content: chat.content,
-        timestamp: chat.timestamp,
-        metadata: chat.metadata
+    messages.forEach((message) => {
+      const messageId = message._id?.toString() || message.id;
+      threadObject.messages[messageId] = {
+        role: message.role,
+        content: message.content,
+        timestamp: message.timestamp,
+        metadata: message.metadata
       };
     });
     return threadObject;
   }
   async createThread(type, userId, threadId, title) {
     const now = Date.now();
-    await this.threadModel.create({
+    await ThreadModel.create({
       type,
       userId,
       threadId,
@@ -101,34 +99,37 @@ var MongoDBThread = class extends MongoDBMemory {
     return { type, threadId, title, updatedAt: now };
   }
   async addMessagesToThread(userId, threadId, messages) {
-    await this.threadModel.updateOne({ threadId, userId }, {
+    await ThreadModel.updateOne({ threadId, userId }, {
       updated_at: Date.now()
     });
+    const messageIds = [];
     for (const message of messages) {
       const newId = randomUUID();
-      await this.chatModel.create({
+      await MessageModel.create({
         threadId,
-        chatId: newId,
+        messageId: newId,
         userId,
         role: message.role,
         content: message.content,
         timestamp: message.timestamp,
         metadata: message.metadata
       });
+      messageIds.push(newId);
     }
+    return messageIds;
   }
   async deleteThread(userId, threadId) {
-    const chats = await this.chatModel.find({ userId, threadId }).sort({
+    const messages = await MessageModel.find({ userId, threadId }).sort({
       timestamp: 1
     });
-    chats?.forEach((chat) => {
-      chat.deleteOne();
+    messages?.forEach((message) => {
+      message.deleteOne();
     });
-    const thread = await this.threadModel.findOne({ userId, threadId });
+    const thread = await ThreadModel.findOne({ userId, threadId });
     thread?.deleteOne();
   }
   async listThreads(userId) {
-    const threads = await this.threadModel.find({ userId }).sort({
+    const threads = await ThreadModel.find({ userId }).sort({
       updated_at: -1
     });
     const data = threads.map((thread) => {
@@ -144,10 +145,22 @@ var MongoDBThread = class extends MongoDBMemory {
 };
 
 // implements/intent.memory.ts
-import { randomUUID as randomUUID2 } from "crypto";
+import { Types } from "mongoose";
 var MongoDBIntent = class extends MongoDBMemory {
   async getIntent(intentId) {
-    const intent = await IntentModel.findById(intentId);
+    const intent = await IntentModel.findById(new Types.ObjectId(intentId));
+    if (intent) {
+      return {
+        name: intent.name,
+        description: intent.description,
+        prompt: intent.prompt,
+        llm: intent.llm
+      };
+    }
+    return void 0;
+  }
+  async getIntentByName(intentName) {
+    const intent = await IntentModel.findOne({ name: intentName });
     if (intent) {
       return {
         name: intent.name,
@@ -159,9 +172,7 @@ var MongoDBIntent = class extends MongoDBMemory {
     return void 0;
   }
   async saveIntent(intent) {
-    const newId = randomUUID2();
     await IntentModel.create({
-      _id: newId,
       name: intent.name,
       description: intent.description,
       prompt: intent.prompt,
@@ -170,7 +181,7 @@ var MongoDBIntent = class extends MongoDBMemory {
   }
   async updateIntent(intentId, intent) {
     await IntentModel.updateOne({
-      _id: intentId
+      _id: new Types.ObjectId(intentId)
     }, {
       name: intent.name,
       description: intent.description,
@@ -179,7 +190,7 @@ var MongoDBIntent = class extends MongoDBMemory {
     });
   }
   async deleteIntent(intentId) {
-    await IntentModel.deleteOne({ _id: intentId });
+    await IntentModel.deleteOne({ _id: new Types.ObjectId(intentId) });
   }
   async listIntents() {
     const intents = await IntentModel.find();

package/dist/index.js.map

@@ -1 +1 @@
-{"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"]}
+{"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 { ThreadDocument, ThreadModel } from \"../models/threads.model\";\nimport { MessageDocument, MessageModel } from \"../models/messages.model\";\nimport { loggers } from \"@ainetwork/adk/utils/logger\";\n\nexport class MongoDBThread extends MongoDBMemory implements IThreadMemory {\n  constructor(uri: string) {\n    super(uri);\n  }\n\n  public async getThread(\n    userId: string,\n    threadId: string\n  ): Promise<ThreadObject | undefined> {\n    const thread = await ThreadModel.findOne({ threadId, userId });\n\t\tconst messages = await MessageModel.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 ${messages.length} messages for thread ${threadId}`);\n\n\t\tconst threadObject: ThreadObject = { \n      type: thread.type as ThreadType,\n      title: thread.title || \"New thread\",\n      messages: {}\n    };\n\t\tmessages.forEach((message: MessageDocument) => {\n\t\t\tconst messageId = message._id?.toString() || message.id;\n\t\t\tthreadObject.messages[messageId] = {\n\t\t\t\trole: message.role as MessageRole,\n\t\t\t\tcontent: message.content,\n\t\t\t\ttimestamp: message.timestamp,\n\t\t\t\tmetadata: message.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 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<string[]> {\n    await ThreadModel.updateOne({ threadId, userId }, {\n      updated_at: Date.now(),\n    });\n    const messageIds: string[] = [];\n    for (const message of messages) {\n      const newId = randomUUID();\n      await MessageModel.create({\n        threadId,\n        messageId: newId,\n        userId,\n        role: message.role,\n        content: message.content,\n        timestamp: message.timestamp,\n        metadata: message.metadata,\n      });\n      messageIds.push(newId);\n    }\n    return messageIds;\n  };\n\n\tpublic async deleteThread(userId: string, threadId: string): Promise<void> {\n\t\tconst messages = await MessageModel.find({ userId, threadId }).sort({\n\t\t\ttimestamp: 1,\n\t\t});\n\n\t\tmessages?.forEach((message: MessageDocument) => {\n      message.deleteOne();\n\t\t});\n    \n    const thread = await ThreadModel.findOne({ userId, threadId });\n    thread?.deleteOne();\n  };\n\n\tpublic async listThreads(userId: string): Promise<ThreadMetadata[]> {\n    const threads = await 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 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\n  constructor(uri: string) {\n    this._uri = uri;\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 mongoose.connect(this._uri, {\n        maxPoolSize: 1,\n        serverSelectionTimeoutMS: 30000,\n        socketTimeoutMS: 45000,\n        connectTimeoutMS: 30000,\n        bufferCommands: false,\n      });\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 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 { Intent } from \"@ainetwork/adk/types/memory\";\nimport { IIntentMemory } from \"@ainetwork/adk/modules\";\nimport { MongoDBMemory } from \"./base.memory\";\nimport { IntentModel } from \"../models/intent.model\";\nimport { Types } from \"mongoose\";\n\nexport class MongoDBIntent extends MongoDBMemory implements IIntentMemory {\n  public async getIntent(intentId: string): Promise<Intent | undefined> {\n    const intent = await IntentModel.findById(new Types.ObjectId(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 getIntentByName(intentName: string): Promise<Intent | undefined> {\n\t\tconst intent = await IntentModel.findOne({ name: intentName });\n\t\tif (intent) {\n\t\t\treturn {\n\t\t\t\tname: intent.name,\n\t\t\t\tdescription: intent.description,\n\t\t\t\tprompt: intent.prompt,\n\t\t\t\tllm: intent.llm,\n\t\t\t} as Intent;\n\t\t}\n\t\treturn undefined;\n\t}\n\n\tpublic async saveIntent(intent: Intent): Promise<void> {\n    // ObjectId automatically generated (MongoDB automatically generates)\n    await IntentModel.create({\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: new Types.ObjectId(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: new Types.ObjectId(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,OAAO,cAAc;AACrB,SAAS,eAAe;AAEjB,IAAM,gBAAN,MAAuC;AAAA,EACpC,eAAwB;AAAA,EACxB;AAAA,EAER,YAAY,KAAa;AACvB,SAAK,OAAO;AAAA,EACd;AAAA,EAEA,MAAa,UAAyB;AACtC,QAAI,KAAK,cAAc;AACtB;AAAA,IACD;AAEA,QAAI;AACA,YAAM,SAAS,QAAQ,KAAK,MAAM;AAAA,QAChC,aAAa;AAAA,QACb,0BAA0B;AAAA,QAC1B,iBAAiB;AAAA,QACjB,kBAAkB;AAAA,QAClB,gBAAgB;AAAA,MAClB,CAAC;AACJ,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,SAAS,WAAW;AAC1B,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;;;AD5CA,SAAS,WAAAA,gBAAe;AAEjB,IAAM,gBAAN,cAA4B,cAAuC;AAAA,EACxE,YAAY,KAAa;AACvB,UAAM,GAAG;AAAA,EACX;AAAA,EAEA,MAAa,UACX,QACA,UACmC;AACnC,UAAM,SAAS,MAAM,YAAY,QAAQ,EAAE,UAAU,OAAO,CAAC;AAC/D,UAAM,WAAW,MAAM,aAAa,KAAK,EAAE,UAAU,OAAO,CAAC,EAAE,KAAK;AAAA,MACnE,WAAW;AAAA,IACZ,CAAC;AAEC,QAAI,CAAC,OAAQ,QAAO;AAEtB,IAAAA,SAAQ,MAAM,MAAM,SAAS,SAAS,MAAM,wBAAwB,QAAQ,EAAE;AAE9E,UAAM,eAA6B;AAAA,MAC/B,MAAM,OAAO;AAAA,MACb,OAAO,OAAO,SAAS;AAAA,MACvB,UAAU,CAAC;AAAA,IACb;AACF,aAAS,QAAQ,CAAC,YAA6B;AAC9C,YAAM,YAAY,QAAQ,KAAK,SAAS,KAAK,QAAQ;AACrD,mBAAa,SAAS,SAAS,IAAI;AAAA,QAClC,MAAM,QAAQ;AAAA,QACd,SAAS,QAAQ;AAAA,QACjB,WAAW,QAAQ;AAAA,QACnB,UAAU,QAAQ;AAAA,MACnB;AAAA,IACD,CAAC;AAED,WAAO;AAAA,EACP;AAAA,EAED,MAAa,aACZ,MACA,QACA,UACA,OAC2B;AACzB,UAAM,MAAM,KAAK,IAAI;AACrB,UAAM,YAAY,OAAO;AAAA,MACvB;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,UACmB;AACnB,UAAM,YAAY,UAAU,EAAE,UAAU,OAAO,GAAG;AAAA,MAChD,YAAY,KAAK,IAAI;AAAA,IACvB,CAAC;AACD,UAAM,aAAuB,CAAC;AAC9B,eAAW,WAAW,UAAU;AAC9B,YAAM,QAAQ,WAAW;AACzB,YAAM,aAAa,OAAO;AAAA,QACxB;AAAA,QACA,WAAW;AAAA,QACX;AAAA,QACA,MAAM,QAAQ;AAAA,QACd,SAAS,QAAQ;AAAA,QACjB,WAAW,QAAQ;AAAA,QACnB,UAAU,QAAQ;AAAA,MACpB,CAAC;AACD,iBAAW,KAAK,KAAK;AAAA,IACvB;AACA,WAAO;AAAA,EACT;AAAA,EAED,MAAa,aAAa,QAAgB,UAAiC;AAC1E,UAAM,WAAW,MAAM,aAAa,KAAK,EAAE,QAAQ,SAAS,CAAC,EAAE,KAAK;AAAA,MACnE,WAAW;AAAA,IACZ,CAAC;AAED,cAAU,QAAQ,CAAC,YAA6B;AAC5C,cAAQ,UAAU;AAAA,IACtB,CAAC;AAEC,UAAM,SAAS,MAAM,YAAY,QAAQ,EAAE,QAAQ,SAAS,CAAC;AAC7D,YAAQ,UAAU;AAAA,EACpB;AAAA,EAED,MAAa,YAAY,QAA2C;AACjE,UAAM,UAAU,MAAM,YAAY,KAAK,EAAE,OAAO,CAAC,EAAE,KAAK;AAAA,MACtD,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;;;AEhHA,SAAS,aAAa;AAEf,IAAM,gBAAN,cAA4B,cAAuC;AAAA,EACxE,MAAa,UAAU,UAA+C;AACpE,UAAM,SAAS,MAAM,YAAY,SAAS,IAAI,MAAM,SAAS,QAAQ,CAAC;AACtE,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,gBAAgB,YAAiD;AAC7E,UAAM,SAAS,MAAM,YAAY,QAAQ,EAAE,MAAM,WAAW,CAAC;AAC7D,QAAI,QAAQ;AACX,aAAO;AAAA,QACN,MAAM,OAAO;AAAA,QACb,aAAa,OAAO;AAAA,QACpB,QAAQ,OAAO;AAAA,QACf,KAAK,OAAO;AAAA,MACb;AAAA,IACD;AACA,WAAO;AAAA,EACR;AAAA,EAEA,MAAa,WAAW,QAA+B;AAEpD,UAAM,YAAY,OAAO;AAAA,MACvB,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,IAAI,MAAM,SAAS,QAAQ;AAAA,IAClC,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,IAAI,MAAM,SAAS,QAAQ,EAAE,CAAC;AAAA,EACnE;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"]}

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

@@ -1,4 +1,4 @@
-import mongoose__default, { Document } from 'mongoose';
+import mongoose, { Document } from 'mongoose';
 
 interface IntentDocument extends Document {
     name: string;
@@ -6,7 +6,7 @@ interface IntentDocument extends Document {
     prompt?: string;
     llm?: string;
 }
-declare const IntentModel: mongoose__default.Model<IntentDocument, {}, {}, {}, mongoose__default.Document<unknown, {}, IntentDocument, {}> & IntentDocument & Required<{
+declare const IntentModel: mongoose.Model<IntentDocument, {}, {}, {}, mongoose.Document<unknown, {}, IntentDocument, {}> & IntentDocument & Required<{
     _id: unknown;
 }> & {
     __v: number;

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

@@ -1,4 +1,4 @@
-import mongoose__default, { Document } from 'mongoose';
+import mongoose, { Document } from 'mongoose';
 
 interface IntentDocument extends Document {
     name: string;
@@ -6,7 +6,7 @@ interface IntentDocument extends Document {
     prompt?: string;
     llm?: string;
 }
-declare const IntentModel: mongoose__default.Model<IntentDocument, {}, {}, {}, mongoose__default.Document<unknown, {}, IntentDocument, {}> & IntentDocument & Required<{
+declare const IntentModel: mongoose.Model<IntentDocument, {}, {}, {}, mongoose.Document<unknown, {}, IntentDocument, {}> & IntentDocument & Required<{
     _id: unknown;
 }> & {
     __v: number;

package/dist/models/messages.model.cjs

@@ -0,0 +1,86 @@
+"use strict";
+var __create = Object.create;
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __getProtoOf = Object.getPrototypeOf;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__getProtoOf(mod)) : {}, __copyProps(
+  // If the importer is in node compatibility mode or this is not an ESM
+  // file that has been converted to a CommonJS file using a Babel-
+  // compatible transform (i.e. "__esModule" has not been set), then set
+  // "default" to the CommonJS "module.exports" for node compatibility.
+  isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", { value: mod, enumerable: true }) : target,
+  mod
+));
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+
+// models/messages.model.ts
+var messages_model_exports = {};
+__export(messages_model_exports, {
+  MessageContentObjectSchema: () => MessageContentObjectSchema,
+  MessageModel: () => MessageModel,
+  MessageObjectSchema: () => MessageObjectSchema
+});
+module.exports = __toCommonJS(messages_model_exports);
+var import_memory = require("@ainetwork/adk/types/memory");
+var import_mongoose = require("mongoose");
+var import_mongoose2 = __toESM(require("mongoose"), 1);
+var MessageContentObjectSchema = new import_mongoose.Schema(
+  {
+    type: { type: String, required: true },
+    parts: { type: [import_mongoose.Schema.Types.Mixed], required: true }
+  },
+  { _id: false }
+);
+var MessageObjectSchema = new import_mongoose.Schema(
+  {
+    threadId: {
+      type: String,
+      required: true,
+      index: true
+    },
+    userId: {
+      type: String,
+      required: true,
+      index: true
+    },
+    role: {
+      type: String,
+      enum: Object.values(import_memory.MessageRole),
+      required: true
+    },
+    content: {
+      type: MessageContentObjectSchema,
+      required: true
+    },
+    timestamp: {
+      type: Number,
+      required: true
+    },
+    metadata: {
+      type: import_mongoose.Schema.Types.Mixed,
+      default: {}
+    }
+  }
+);
+var MessageModel = import_mongoose2.default.model("Message", MessageObjectSchema);
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  MessageContentObjectSchema,
+  MessageModel,
+  MessageObjectSchema
+});
+//# sourceMappingURL=messages.model.cjs.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["../../models/messages.model.ts"],"sourcesContent":["import { MessageRole } from \"@ainetwork/adk/types/memory\";\nimport { type Document, Schema } from \"mongoose\";\nimport mongoose from \"mongoose\";\n\n// MessageContentObject schema\nexport const MessageContentObjectSchema = 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// MessageObject schema - 개별 문서로 저장\nexport const MessageObjectSchema = 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: MessageContentObjectSchema,\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// Message Document interface\nexport interface MessageDocument 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\nexport const MessageModel = mongoose.model<MessageDocument>(\"Message\", MessageObjectSchema);"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,oBAA4B;AAC5B,sBAAsC;AACtC,IAAAA,mBAAqB;AAGd,IAAM,6BAA6B,IAAI;AAAA,EAC7C;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,sBAAsB,IAAI;AAAA,EACtC;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;AAgBO,IAAM,eAAe,iBAAAC,QAAS,MAAuB,WAAW,mBAAmB;","names":["import_mongoose","mongoose"]}

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

@@ -0,0 +1,75 @@
+import { MessageRole } from '@ainetwork/adk/types/memory';
+import mongoose, { Schema, Document } from 'mongoose';
+
+declare const MessageContentObjectSchema: Schema<any, mongoose.Model<any, any, any, any, any, any>, {}, {}, {}, {}, {
+    _id: false;
+}, {
+    type: string;
+    parts: any[];
+}, Document<unknown, {}, mongoose.FlatRecord<{
+    type: string;
+    parts: any[];
+}>, {}> & mongoose.FlatRecord<{
+    type: string;
+    parts: any[];
+}> & {
+    _id: mongoose.Types.ObjectId;
+} & {
+    __v: number;
+}>;
+declare const MessageObjectSchema: Schema<any, mongoose.Model<any, any, any, any, any, any>, {}, {}, {}, {}, mongoose.DefaultSchemaOptions, {
+    threadId: string;
+    userId: string;
+    role: MessageRole;
+    content: {
+        type: string;
+        parts: any[];
+    };
+    timestamp: number;
+    metadata: any;
+}, Document<unknown, {}, mongoose.FlatRecord<{
+    threadId: string;
+    userId: string;
+    role: MessageRole;
+    content: {
+        type: string;
+        parts: any[];
+    };
+    timestamp: number;
+    metadata: any;
+}>, {}> & mongoose.FlatRecord<{
+    threadId: string;
+    userId: string;
+    role: MessageRole;
+    content: {
+        type: string;
+        parts: any[];
+    };
+    timestamp: number;
+    metadata: any;
+}> & {
+    _id: mongoose.Types.ObjectId;
+} & {
+    __v: number;
+}>;
+interface MessageDocument extends Document {
+    threadId: string;
+    role: MessageRole;
+    content: {
+        type: string;
+        parts: any[];
+    };
+    timestamp: number;
+    metadata?: {
+        [key: string]: unknown;
+    };
+    createdAt: Date;
+    updatedAt: Date;
+}
+declare const MessageModel: mongoose.Model<MessageDocument, {}, {}, {}, Document<unknown, {}, MessageDocument, {}> & MessageDocument & Required<{
+    _id: unknown;
+}> & {
+    __v: number;
+}, any>;
+
+export { MessageContentObjectSchema, type MessageDocument, MessageModel, MessageObjectSchema };

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

@@ -0,0 +1,75 @@
+import { MessageRole } from '@ainetwork/adk/types/memory';
+import mongoose, { Schema, Document } from 'mongoose';
+
+declare const MessageContentObjectSchema: Schema<any, mongoose.Model<any, any, any, any, any, any>, {}, {}, {}, {}, {
+    _id: false;
+}, {
+    type: string;
+    parts: any[];
+}, Document<unknown, {}, mongoose.FlatRecord<{
+    type: string;
+    parts: any[];
+}>, {}> & mongoose.FlatRecord<{
+    type: string;
+    parts: any[];
+}> & {
+    _id: mongoose.Types.ObjectId;
+} & {
+    __v: number;
+}>;
+declare const MessageObjectSchema: Schema<any, mongoose.Model<any, any, any, any, any, any>, {}, {}, {}, {}, mongoose.DefaultSchemaOptions, {
+    threadId: string;
+    userId: string;
+    role: MessageRole;
+    content: {
+        type: string;
+        parts: any[];
+    };
+    timestamp: number;
+    metadata: any;
+}, Document<unknown, {}, mongoose.FlatRecord<{
+    threadId: string;
+    userId: string;
+    role: MessageRole;
+    content: {
+        type: string;
+        parts: any[];
+    };
+    timestamp: number;
+    metadata: any;
+}>, {}> & mongoose.FlatRecord<{
+    threadId: string;
+    userId: string;
+    role: MessageRole;
+    content: {
+        type: string;
+        parts: any[];
+    };
+    timestamp: number;
+    metadata: any;
+}> & {
+    _id: mongoose.Types.ObjectId;
+} & {
+    __v: number;
+}>;
+interface MessageDocument extends Document {
+    threadId: string;
+    role: MessageRole;
+    content: {
+        type: string;
+        parts: any[];
+    };
+    timestamp: number;
+    metadata?: {
+        [key: string]: unknown;
+    };
+    createdAt: Date;
+    updatedAt: Date;
+}
+declare const MessageModel: mongoose.Model<MessageDocument, {}, {}, {}, Document<unknown, {}, MessageDocument, {}> & MessageDocument & Required<{
+    _id: unknown;
+}> & {
+    __v: number;
+}, any>;
+
+export { MessageContentObjectSchema, type MessageDocument, MessageModel, MessageObjectSchema };

package/dist/models/messages.model.js

@@ -0,0 +1,11 @@
+import {
+  MessageContentObjectSchema,
+  MessageModel,
+  MessageObjectSchema
+} from "../chunk-MLPXPX2L.js";
+export {
+  MessageContentObjectSchema,
+  MessageModel,
+  MessageObjectSchema
+};
+//# sourceMappingURL=messages.model.js.map

package/dist/models/{chats.model.cjs → threads.model.cjs}

@@ -1,7 +1,9 @@
 "use strict";
+var __create = Object.create;
 var __defProp = Object.defineProperty;
 var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
 var __getOwnPropNames = Object.getOwnPropertyNames;
+var __getProtoOf = Object.getPrototypeOf;
 var __hasOwnProp = Object.prototype.hasOwnProperty;
 var __export = (target, all) => {
   for (var name in all)
@@ -15,18 +17,26 @@ var __copyProps = (to, from, except, desc) => {
   }
   return to;
 };
+var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__getProtoOf(mod)) : {}, __copyProps(
+  // If the importer is in node compatibility mode or this is not an ESM
+  // file that has been converted to a CommonJS file using a Babel-
+  // compatible transform (i.e. "__esModule" has not been set), then set
+  // "default" to the CommonJS "module.exports" for node compatibility.
+  isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", { value: mod, enumerable: true }) : target,
+  mod
+));
 var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
 
-// models/chats.model.ts
-var chats_model_exports = {};
-__export(chats_model_exports, {
-  ChatContentObjectSchema: () => ChatContentObjectSchema,
-  ChatObjectSchema: () => ChatObjectSchema,
+// models/threads.model.ts
+var threads_model_exports = {};
+__export(threads_model_exports, {
+  ThreadModel: () => ThreadModel,
   ThreadObjectSchema: () => ThreadObjectSchema
 });
-module.exports = __toCommonJS(chats_model_exports);
+module.exports = __toCommonJS(threads_model_exports);
 var import_memory = require("@ainetwork/adk/types/memory");
 var import_mongoose = require("mongoose");
+var import_mongoose2 = __toESM(require("mongoose"), 1);
 var ThreadObjectSchema = new import_mongoose.Schema(
   {
     type: {
@@ -58,48 +68,10 @@ var ThreadObjectSchema = new import_mongoose.Schema(
     }
   }
 );
-var ChatContentObjectSchema = new import_mongoose.Schema(
-  {
-    type: { type: String, required: true },
-    parts: { type: [import_mongoose.Schema.Types.Mixed], required: true }
-  },
-  { _id: false }
-);
-var ChatObjectSchema = new import_mongoose.Schema(
-  {
-    threadId: {
-      type: String,
-      required: true,
-      index: true
-    },
-    userId: {
-      type: String,
-      required: true,
-      index: true
-    },
-    role: {
-      type: String,
-      enum: Object.values(import_memory.MessageRole),
-      required: true
-    },
-    content: {
-      type: ChatContentObjectSchema,
-      required: true
-    },
-    timestamp: {
-      type: Number,
-      required: true
-    },
-    metadata: {
-      type: import_mongoose.Schema.Types.Mixed,
-      default: {}
-    }
-  }
-);
+var ThreadModel = import_mongoose2.default.model("Thread", ThreadObjectSchema);
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
-  ChatContentObjectSchema,
-  ChatObjectSchema,
+  ThreadModel,
   ThreadObjectSchema
 });
-//# sourceMappingURL=chats.model.cjs.map
+//# sourceMappingURL=threads.model.cjs.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["../../models/threads.model.ts"],"sourcesContent":["import { ThreadType } from \"@ainetwork/adk/types/memory\";\nimport { type Document, Schema } from \"mongoose\";\nimport mongoose 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\nexport const ThreadModel = mongoose.model<ThreadDocument>(\"Thread\", ThreadObjectSchema);\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,oBAA2B;AAC3B,sBAAsC;AACtC,IAAAA,mBAAqB;AAEd,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;AAWO,IAAM,cAAc,iBAAAC,QAAS,MAAsB,UAAU,kBAAkB;","names":["import_mongoose","mongoose"]}