@ainetwork/adk-provider-memory-mongodb 0.2.1 → 0.2.3

package/dist/{chunk-JCE5NAY5.js → chunk-YFW7JXII.js}

@@ -2,6 +2,12 @@
 import mongoose, { Schema } from "mongoose";
 var IntentObjectSchema = new Schema(
   {
+    id: {
+      type: String,
+      required: true,
+      index: true,
+      unique: true
+    },
     name: {
       type: String,
       required: true,
@@ -15,8 +21,16 @@ var IntentObjectSchema = new Schema(
       type: String,
       required: false
     },
-    llm: {
+    status: {
       type: String,
+      required: true
+    },
+    triggeringSentences: {
+      type: [String],
+      required: false
+    },
+    tags: {
+      type: [String],
       required: false
     }
   }
@@ -26,4 +40,4 @@ var IntentModel = mongoose.model("Intent", IntentObjectSchema);
 export {
   IntentModel
 };
-//# sourceMappingURL=chunk-JCE5NAY5.js.map
+//# sourceMappingURL=chunk-YFW7JXII.js.map

package/dist/chunk-YFW7JXII.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../models/intent.model.ts"],"sourcesContent":["import { Intent } from \"@ainetwork/adk/types/memory\";\nimport mongoose, { type Document, Schema } from \"mongoose\";\n\nconst IntentObjectSchema = new Schema(\n\t{\n\t\tid: { \n\t\t\ttype: String,\n\t\t\trequired: true,\n\t\t\tindex: true,\n\t\t\tunique: true,\n\t\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\tstatus: {\n\t\t\ttype: String,\n\t\t\trequired: true,\n\t\t},\n\t\ttriggeringSentences: {\n\t\t\ttype: [String],\n\t\t\trequired: false,\n\t\t},\n\t\ttags: {\n\t\t\ttype: [String],\n\t\t\trequired: false,\n\t\t},\n\t},\n);\n\nexport interface IntentDocument extends Omit<Document, 'id'>, Omit<Intent, 'id'> {\n\tid: string;\n}\n\nexport const IntentModel = mongoose.model<IntentDocument>(\"Intent\", IntentObjectSchema);"],"mappings":";AACA,OAAO,YAA2B,cAAc;AAEhD,IAAM,qBAAqB,IAAI;AAAA,EAC9B;AAAA,IACC,IAAI;AAAA,MACH,MAAM;AAAA,MACN,UAAU;AAAA,MACV,OAAO;AAAA,MACP,QAAQ;AAAA,IACT;AAAA,IACA,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,QAAQ;AAAA,MACP,MAAM;AAAA,MACN,UAAU;AAAA,IACX;AAAA,IACA,qBAAqB;AAAA,MACpB,MAAM,CAAC,MAAM;AAAA,MACb,UAAU;AAAA,IACX;AAAA,IACA,MAAM;AAAA,MACL,MAAM,CAAC,MAAM;AAAA,MACb,UAAU;AAAA,IACX;AAAA,EACD;AACD;AAMO,IAAM,cAAc,SAAS,MAAsB,UAAU,kBAAkB;","names":[]}

package/dist/index.cjs

@@ -181,6 +181,8 @@ var MongoDBThread = class extends MongoDBMemory {
     if (!thread) return void 0;
     import_logger2.loggers.agent.debug(`Found ${messages.length} messages for thread ${threadId}`);
     const threadObject = {
+      threadId: thread.threadId,
+      userId: thread.userId,
       type: thread.type,
       title: thread.title || "New thread",
       messages: []
@@ -206,7 +208,7 @@ var MongoDBThread = class extends MongoDBMemory {
       updated_at: now,
       created_at: now
     });
-    return { type, threadId, title, updatedAt: now };
+    return { type, userId, threadId, title, messages: [] };
   }
   async addMessagesToThread(userId, threadId, messages) {
     await ThreadModel.updateOne({ threadId, userId }, {
@@ -241,6 +243,7 @@ var MongoDBThread = class extends MongoDBMemory {
     const data = threads.map((thread) => {
       return {
         type: thread.type,
+        userId,
         threadId: thread.threadId,
         title: thread.title,
         updatedAt: thread.updated_at
@@ -254,6 +257,12 @@ var MongoDBThread = class extends MongoDBMemory {
 var import_mongoose6 = __toESM(require("mongoose"), 1);
 var IntentObjectSchema = new import_mongoose6.Schema(
   {
+    id: {
+      type: String,
+      required: true,
+      index: true,
+      unique: true
+    },
     name: {
       type: String,
       required: true,
@@ -267,8 +276,16 @@ var IntentObjectSchema = new import_mongoose6.Schema(
       type: String,
       required: false
     },
-    llm: {
+    status: {
       type: String,
+      required: true
+    },
+    triggeringSentences: {
+      type: [String],
+      required: false
+    },
+    tags: {
+      type: [String],
       required: false
     }
   }
@@ -276,61 +293,29 @@ var IntentObjectSchema = new import_mongoose6.Schema(
 var IntentModel = import_mongoose6.default.model("Intent", IntentObjectSchema);
 
 // implements/intent.memory.ts
-var import_mongoose7 = require("mongoose");
 var MongoDBIntent = class extends MongoDBMemory {
   async getIntent(intentId) {
-    const intent = await IntentModel.findById(new import_mongoose7.Types.ObjectId(intentId));
-    if (intent) {
-      return {
-        name: intent.name,
-        description: intent.description,
-        prompt: intent.prompt,
-        llm: intent.llm
-      };
-    }
-    return void 0;
+    const intent = await IntentModel.findOne({ id: intentId }).lean();
+    return intent || void 0;
   }
   async getIntentByName(intentName) {
-    const intent = await IntentModel.findOne({ name: intentName });
-    if (intent) {
-      return {
-        name: intent.name,
-        description: intent.description,
-        prompt: intent.prompt,
-        llm: intent.llm
-      };
-    }
-    return void 0;
+    const intent = await IntentModel.findOne({ name: intentName }).lean();
+    return intent || void 0;
   }
   async saveIntent(intent) {
-    await IntentModel.create({
-      name: intent.name,
-      description: intent.description,
-      prompt: intent.prompt,
-      llm: intent.llm
-    });
+    await IntentModel.create(intent);
   }
   async updateIntent(intentId, intent) {
     await IntentModel.updateOne({
-      _id: new import_mongoose7.Types.ObjectId(intentId)
-    }, {
-      name: intent.name,
-      description: intent.description,
-      prompt: intent.prompt,
-      llm: intent.llm
-    });
+      id: intentId
+    }, intent);
   }
   async deleteIntent(intentId) {
-    await IntentModel.deleteOne({ _id: new import_mongoose7.Types.ObjectId(intentId) });
+    await IntentModel.deleteOne({ id: intentId });
   }
   async listIntents() {
-    const intents = await IntentModel.find();
-    return intents.map((intent) => ({
-      name: intent.name,
-      description: intent.description,
-      prompt: intent.prompt,
-      llm: intent.llm
-    }));
+    const intents = await IntentModel.find().lean();
+    return intents;
   }
 };
 // Annotate the CommonJS export names for ESM import in node:

package/dist/index.cjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../index.ts","../implements/base.memory.ts","../models/threads.model.ts","../models/messages.model.ts","../implements/thread.memory.ts","../models/intent.model.ts","../implements/intent.memory.ts"],"sourcesContent":["export { MongoDBThread } from \"./implements/thread.memory\";\nexport { MongoDBIntent } from \"./implements/intent.memory\";","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 { 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","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\tmessageId: {\n\t\t\ttype: String,\n\t\t\trequired: true,\n\t\t\tindex: 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\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\tmessageId: string;\n\tthreadId: string;\n\tuserId: 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}\n\nexport const MessageModel = mongoose.model<MessageDocument>(\"Message\", MessageObjectSchema);","import 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\tthreadObject.messages.push({\n        messageId: message.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<void> {\n    await ThreadModel.updateOne({ threadId, userId }, {\n      updated_at: Date.now(),\n    });\n    for (const message of messages) {\n      await MessageModel.create({\n        threadId,\n        messageId: message.messageId,\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 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 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);","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;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACCA,sBAAqB;AACrB,oBAAwB;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,gBAAAA,QAAS,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,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,gBAAAA,QAAS,WAAW;AAC1B,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,oBAA2B;AAC3B,IAAAC,mBAAsC;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;;;AC7CtF,IAAAC,iBAA4B;AAC5B,IAAAC,mBAAsC;AACtC,IAAAA,mBAAqB;AAGd,IAAM,6BAA6B,IAAI;AAAA,EAC7C;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,sBAAsB,IAAI;AAAA,EACtC;AAAA,IACC,WAAW;AAAA,MACV,MAAM;AAAA,MACN,UAAU;AAAA,MACV,OAAO;AAAA,IACR;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,MAAM;AAAA,MACL,MAAM;AAAA,MACN,MAAM,OAAO,OAAO,0BAAW;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;AAgBO,IAAM,eAAe,iBAAAC,QAAS,MAAuB,WAAW,mBAAmB;;;AC3D1F,IAAAC,iBAAwB;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,2BAAQ,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,mBAAa,SAAS,KAAK;AAAA,QACtB,WAAW,QAAQ;AAAA,QACvB,MAAM,QAAQ;AAAA,QACd,SAAS,QAAQ;AAAA,QACjB,WAAW,QAAQ;AAAA,QACnB,UAAU,QAAQ;AAAA,MACnB,CAAC;AAAA,IACF,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,UACe;AACf,UAAM,YAAY,UAAU,EAAE,UAAU,OAAO,GAAG;AAAA,MAChD,YAAY,KAAK,IAAI;AAAA,IACvB,CAAC;AACD,eAAW,WAAW,UAAU;AAC9B,YAAM,aAAa,OAAO;AAAA,QACxB;AAAA,QACA,WAAW,QAAQ;AAAA,QACnB;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,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;;;AC/GA,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;;;AC3BtF,IAAAC,mBAAsB;AAEf,IAAM,gBAAN,cAA4B,cAAuC;AAAA,EACxE,MAAa,UAAU,UAA+C;AACpE,UAAM,SAAS,MAAM,YAAY,SAAS,IAAI,uBAAM,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,uBAAM,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,uBAAM,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":["mongoose","import_mongoose","mongoose","import_memory","import_mongoose","mongoose","import_logger","import_mongoose","mongoose","import_mongoose"]}
+{"version":3,"sources":["../index.ts","../implements/base.memory.ts","../models/threads.model.ts","../models/messages.model.ts","../implements/thread.memory.ts","../models/intent.model.ts","../implements/intent.memory.ts"],"sourcesContent":["export { MongoDBThread } from \"./implements/thread.memory\";\nexport { MongoDBIntent } from \"./implements/intent.memory\";","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 { 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","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\tmessageId: {\n\t\t\ttype: String,\n\t\t\trequired: true,\n\t\t\tindex: 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\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\tmessageId: string;\n\tthreadId: string;\n\tuserId: 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}\n\nexport const MessageModel = mongoose.model<MessageDocument>(\"Message\", MessageObjectSchema);","import 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      threadId: thread.threadId, \n      userId: thread.userId,\n      type: thread.type as ThreadType,\n      title: thread.title || \"New thread\",\n      messages: []\n    };\n\t\tmessages.forEach((message: MessageDocument) => {\n\t\t\tthreadObject.messages.push({\n        messageId: message.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<ThreadObject> {\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, userId, threadId, title, messages: []};\n  };\n\n\tpublic async addMessagesToThread(\n    userId: string,\n    threadId: string,\n    messages: MessageObject[]\n  ): Promise<void> {\n    await ThreadModel.updateOne({ threadId, userId }, {\n      updated_at: Date.now(),\n    });\n    for (const message of messages) {\n      await MessageModel.create({\n        threadId,\n        messageId: message.messageId,\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 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        userId,\n        threadId: thread.threadId,\n        title: thread.title,\n        updatedAt: thread.updated_at\n      } as ThreadMetadata;\n    })\n    return data;\n  };\n}","import { Intent } from \"@ainetwork/adk/types/memory\";\nimport mongoose, { type Document, Schema } from \"mongoose\";\n\nconst IntentObjectSchema = new Schema(\n\t{\n\t\tid: { \n\t\t\ttype: String,\n\t\t\trequired: true,\n\t\t\tindex: true,\n\t\t\tunique: true,\n\t\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\tstatus: {\n\t\t\ttype: String,\n\t\t\trequired: true,\n\t\t},\n\t\ttriggeringSentences: {\n\t\t\ttype: [String],\n\t\t\trequired: false,\n\t\t},\n\t\ttags: {\n\t\t\ttype: [String],\n\t\t\trequired: false,\n\t\t},\n\t},\n);\n\nexport interface IntentDocument extends Omit<Document, 'id'>, Omit<Intent, 'id'> {\n\tid: string;\n}\n\nexport const IntentModel = mongoose.model<IntentDocument>(\"Intent\", IntentObjectSchema);","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\";\n\nexport class MongoDBIntent extends MongoDBMemory implements IIntentMemory {\n  public async getIntent(intentId: string): Promise<Intent | undefined> {\n    const intent = await IntentModel.findOne({ id: intentId }).lean<Intent>();\n    return intent || undefined;\n  };\n\n\tpublic async getIntentByName(intentName: string): Promise<Intent | undefined> {\n\t\tconst intent = await IntentModel.findOne({ name: intentName }).lean<Intent>();\n    return intent || undefined;\n\t}\n\n\tpublic async saveIntent(intent: Intent): Promise<void> {\n    await IntentModel.create(intent);\n  };\n\n\tpublic async updateIntent(intentId: string, intent: Intent): Promise<void> {\n    await IntentModel.updateOne({\n      id: intentId,\n    }, intent);\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().lean<Intent[]>();\n    return intents;\n  };\n}"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACCA,sBAAqB;AACrB,oBAAwB;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,gBAAAA,QAAS,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,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,gBAAAA,QAAS,WAAW;AAC1B,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,oBAA2B;AAC3B,IAAAC,mBAAsC;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;;;AC7CtF,IAAAC,iBAA4B;AAC5B,IAAAC,mBAAsC;AACtC,IAAAA,mBAAqB;AAGd,IAAM,6BAA6B,IAAI;AAAA,EAC7C;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,sBAAsB,IAAI;AAAA,EACtC;AAAA,IACC,WAAW;AAAA,MACV,MAAM;AAAA,MACN,UAAU;AAAA,MACV,OAAO;AAAA,IACR;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,MAAM;AAAA,MACL,MAAM;AAAA,MACN,MAAM,OAAO,OAAO,0BAAW;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;AAgBO,IAAM,eAAe,iBAAAC,QAAS,MAAuB,WAAW,mBAAmB;;;AC3D1F,IAAAC,iBAAwB;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,2BAAQ,MAAM,MAAM,SAAS,SAAS,MAAM,wBAAwB,QAAQ,EAAE;AAE9E,UAAM,eAA6B;AAAA,MAC/B,UAAU,OAAO;AAAA,MACjB,QAAQ,OAAO;AAAA,MACf,MAAM,OAAO;AAAA,MACb,OAAO,OAAO,SAAS;AAAA,MACvB,UAAU,CAAC;AAAA,IACb;AACF,aAAS,QAAQ,CAAC,YAA6B;AAC9C,mBAAa,SAAS,KAAK;AAAA,QACtB,WAAW,QAAQ;AAAA,QACvB,MAAM,QAAQ;AAAA,QACd,SAAS,QAAQ;AAAA,QACjB,WAAW,QAAQ;AAAA,QACnB,UAAU,QAAQ;AAAA,MACnB,CAAC;AAAA,IACF,CAAC;AAED,WAAO;AAAA,EACP;AAAA,EAED,MAAa,aACZ,MACA,QACA,UACA,OACyB;AACvB,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,QAAQ,UAAU,OAAO,UAAU,CAAC,EAAC;AAAA,EACtD;AAAA,EAED,MAAa,oBACV,QACA,UACA,UACe;AACf,UAAM,YAAY,UAAU,EAAE,UAAU,OAAO,GAAG;AAAA,MAChD,YAAY,KAAK,IAAI;AAAA,IACvB,CAAC;AACD,eAAW,WAAW,UAAU;AAC9B,YAAM,aAAa,OAAO;AAAA,QACxB;AAAA,QACA,WAAW,QAAQ;AAAA,QACnB;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,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;AAAA,QACA,UAAU,OAAO;AAAA,QACjB,OAAO,OAAO;AAAA,QACd,WAAW,OAAO;AAAA,MACpB;AAAA,IACF,CAAC;AACD,WAAO;AAAA,EACT;AACF;;;ACjHA,IAAAC,mBAAgD;AAEhD,IAAM,qBAAqB,IAAI;AAAA,EAC9B;AAAA,IACC,IAAI;AAAA,MACH,MAAM;AAAA,MACN,UAAU;AAAA,MACV,OAAO;AAAA,MACP,QAAQ;AAAA,IACT;AAAA,IACA,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,QAAQ;AAAA,MACP,MAAM;AAAA,MACN,UAAU;AAAA,IACX;AAAA,IACA,qBAAqB;AAAA,MACpB,MAAM,CAAC,MAAM;AAAA,MACb,UAAU;AAAA,IACX;AAAA,IACA,MAAM;AAAA,MACL,MAAM,CAAC,MAAM;AAAA,MACb,UAAU;AAAA,IACX;AAAA,EACD;AACD;AAMO,IAAM,cAAc,iBAAAC,QAAS,MAAsB,UAAU,kBAAkB;;;ACtC/E,IAAM,gBAAN,cAA4B,cAAuC;AAAA,EACxE,MAAa,UAAU,UAA+C;AACpE,UAAM,SAAS,MAAM,YAAY,QAAQ,EAAE,IAAI,SAAS,CAAC,EAAE,KAAa;AACxE,WAAO,UAAU;AAAA,EACnB;AAAA,EAED,MAAa,gBAAgB,YAAiD;AAC7E,UAAM,SAAS,MAAM,YAAY,QAAQ,EAAE,MAAM,WAAW,CAAC,EAAE,KAAa;AAC1E,WAAO,UAAU;AAAA,EACpB;AAAA,EAEA,MAAa,WAAW,QAA+B;AACpD,UAAM,YAAY,OAAO,MAAM;AAAA,EACjC;AAAA,EAED,MAAa,aAAa,UAAkB,QAA+B;AACxE,UAAM,YAAY,UAAU;AAAA,MAC1B,IAAI;AAAA,IACN,GAAG,MAAM;AAAA,EACX;AAAA,EAED,MAAa,aAAa,UAAiC;AACxD,UAAM,YAAY,UAAU,EAAE,IAAI,SAAS,CAAC;AAAA,EAC9C;AAAA,EAED,MAAa,cAAiC;AAC3C,UAAM,UAAU,MAAM,YAAY,KAAK,EAAE,KAAe;AACxD,WAAO;AAAA,EACT;AACF;","names":["mongoose","import_mongoose","mongoose","import_memory","import_mongoose","mongoose","import_logger","import_mongoose","mongoose"]}

package/dist/index.d.cts

@@ -1,4 +1,4 @@
-import { ThreadObject, ThreadType, ThreadMetadata, MessageObject, Intent } from '@ainetwork/adk/types/memory';
+import { ThreadObject, ThreadType, MessageObject, ThreadMetadata, 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';
 
@@ -14,7 +14,7 @@ declare class MongoDBMemory implements IMemory {
 declare class MongoDBThread extends MongoDBMemory implements IThreadMemory {
     constructor(uri: string);
     getThread(userId: string, threadId: string): Promise<ThreadObject | undefined>;
-    createThread(type: ThreadType, userId: string, threadId: string, title: string): Promise<ThreadMetadata>;
+    createThread(type: ThreadType, userId: string, threadId: string, title: string): Promise<ThreadObject>;
     addMessagesToThread(userId: string, threadId: string, messages: MessageObject[]): Promise<void>;
     deleteThread(userId: string, threadId: string): Promise<void>;
     listThreads(userId: string): Promise<ThreadMetadata[]>;

package/dist/index.d.ts

@@ -1,4 +1,4 @@
-import { ThreadObject, ThreadType, ThreadMetadata, MessageObject, Intent } from '@ainetwork/adk/types/memory';
+import { ThreadObject, ThreadType, MessageObject, ThreadMetadata, 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';
 
@@ -14,7 +14,7 @@ declare class MongoDBMemory implements IMemory {
 declare class MongoDBThread extends MongoDBMemory implements IThreadMemory {
     constructor(uri: string);
     getThread(userId: string, threadId: string): Promise<ThreadObject | undefined>;
-    createThread(type: ThreadType, userId: string, threadId: string, title: string): Promise<ThreadMetadata>;
+    createThread(type: ThreadType, userId: string, threadId: string, title: string): Promise<ThreadObject>;
     addMessagesToThread(userId: string, threadId: string, messages: MessageObject[]): Promise<void>;
     deleteThread(userId: string, threadId: string): Promise<void>;
     listThreads(userId: string): Promise<ThreadMetadata[]>;

package/dist/index.js

@@ -1,6 +1,6 @@
 import {
   IntentModel
-} from "./chunk-JCE5NAY5.js";
+} from "./chunk-YFW7JXII.js";
 import {
   MessageModel
 } from "./chunk-5CCEN7NK.js";
@@ -68,6 +68,8 @@ var MongoDBThread = class extends MongoDBMemory {
     if (!thread) return void 0;
     loggers2.agent.debug(`Found ${messages.length} messages for thread ${threadId}`);
     const threadObject = {
+      threadId: thread.threadId,
+      userId: thread.userId,
       type: thread.type,
       title: thread.title || "New thread",
       messages: []
@@ -93,7 +95,7 @@ var MongoDBThread = class extends MongoDBMemory {
       updated_at: now,
       created_at: now
     });
-    return { type, threadId, title, updatedAt: now };
+    return { type, userId, threadId, title, messages: [] };
   }
   async addMessagesToThread(userId, threadId, messages) {
     await ThreadModel.updateOne({ threadId, userId }, {
@@ -128,6 +130,7 @@ var MongoDBThread = class extends MongoDBMemory {
     const data = threads.map((thread) => {
       return {
         type: thread.type,
+        userId,
         threadId: thread.threadId,
         title: thread.title,
         updatedAt: thread.updated_at
@@ -138,61 +141,29 @@ var MongoDBThread = class extends MongoDBMemory {
 };
 
 // implements/intent.memory.ts
-import { Types } from "mongoose";
 var MongoDBIntent = class extends MongoDBMemory {
   async getIntent(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;
+    const intent = await IntentModel.findOne({ id: intentId }).lean();
+    return intent || void 0;
   }
   async getIntentByName(intentName) {
-    const intent = await IntentModel.findOne({ name: intentName });
-    if (intent) {
-      return {
-        name: intent.name,
-        description: intent.description,
-        prompt: intent.prompt,
-        llm: intent.llm
-      };
-    }
-    return void 0;
+    const intent = await IntentModel.findOne({ name: intentName }).lean();
+    return intent || void 0;
   }
   async saveIntent(intent) {
-    await IntentModel.create({
-      name: intent.name,
-      description: intent.description,
-      prompt: intent.prompt,
-      llm: intent.llm
-    });
+    await IntentModel.create(intent);
   }
   async updateIntent(intentId, intent) {
     await IntentModel.updateOne({
-      _id: new Types.ObjectId(intentId)
-    }, {
-      name: intent.name,
-      description: intent.description,
-      prompt: intent.prompt,
-      llm: intent.llm
-    });
+      id: intentId
+    }, intent);
   }
   async deleteIntent(intentId) {
-    await IntentModel.deleteOne({ _id: new Types.ObjectId(intentId) });
+    await IntentModel.deleteOne({ id: intentId });
   }
   async listIntents() {
-    const intents = await IntentModel.find();
-    return intents.map((intent) => ({
-      name: intent.name,
-      description: intent.description,
-      prompt: intent.prompt,
-      llm: intent.llm
-    }));
+    const intents = await IntentModel.find().lean();
+    return intents;
   }
 };
 export {

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../implements/base.memory.ts","../implements/thread.memory.ts","../implements/intent.memory.ts"],"sourcesContent":["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 { 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\tthreadObject.messages.push({\n        messageId: message.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<void> {\n    await ThreadModel.updateOne({ threadId, userId }, {\n      updated_at: Date.now(),\n    });\n    for (const message of messages) {\n      await MessageModel.create({\n        threadId,\n        messageId: message.messageId,\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 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 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":";;;;;;;;;;;AACA,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;;;AC7CA,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,mBAAa,SAAS,KAAK;AAAA,QACtB,WAAW,QAAQ;AAAA,QACvB,MAAM,QAAQ;AAAA,QACd,SAAS,QAAQ;AAAA,QACjB,WAAW,QAAQ;AAAA,QACnB,UAAU,QAAQ;AAAA,MACnB,CAAC;AAAA,IACF,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,UACe;AACf,UAAM,YAAY,UAAU,EAAE,UAAU,OAAO,GAAG;AAAA,MAChD,YAAY,KAAK,IAAI;AAAA,IACvB,CAAC;AACD,eAAW,WAAW,UAAU;AAC9B,YAAM,aAAa,OAAO;AAAA,QACxB;AAAA,QACA,WAAW,QAAQ;AAAA,QACnB;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,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;;;AC3GA,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"]}
+{"version":3,"sources":["../implements/base.memory.ts","../implements/thread.memory.ts","../implements/intent.memory.ts"],"sourcesContent":["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 { 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      threadId: thread.threadId, \n      userId: thread.userId,\n      type: thread.type as ThreadType,\n      title: thread.title || \"New thread\",\n      messages: []\n    };\n\t\tmessages.forEach((message: MessageDocument) => {\n\t\t\tthreadObject.messages.push({\n        messageId: message.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<ThreadObject> {\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, userId, threadId, title, messages: []};\n  };\n\n\tpublic async addMessagesToThread(\n    userId: string,\n    threadId: string,\n    messages: MessageObject[]\n  ): Promise<void> {\n    await ThreadModel.updateOne({ threadId, userId }, {\n      updated_at: Date.now(),\n    });\n    for (const message of messages) {\n      await MessageModel.create({\n        threadId,\n        messageId: message.messageId,\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 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        userId,\n        threadId: thread.threadId,\n        title: thread.title,\n        updatedAt: thread.updated_at\n      } as ThreadMetadata;\n    })\n    return data;\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\";\n\nexport class MongoDBIntent extends MongoDBMemory implements IIntentMemory {\n  public async getIntent(intentId: string): Promise<Intent | undefined> {\n    const intent = await IntentModel.findOne({ id: intentId }).lean<Intent>();\n    return intent || undefined;\n  };\n\n\tpublic async getIntentByName(intentName: string): Promise<Intent | undefined> {\n\t\tconst intent = await IntentModel.findOne({ name: intentName }).lean<Intent>();\n    return intent || undefined;\n\t}\n\n\tpublic async saveIntent(intent: Intent): Promise<void> {\n    await IntentModel.create(intent);\n  };\n\n\tpublic async updateIntent(intentId: string, intent: Intent): Promise<void> {\n    await IntentModel.updateOne({\n      id: intentId,\n    }, intent);\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().lean<Intent[]>();\n    return intents;\n  };\n}"],"mappings":";;;;;;;;;;;AACA,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;;;AC7CA,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,UAAU,OAAO;AAAA,MACjB,QAAQ,OAAO;AAAA,MACf,MAAM,OAAO;AAAA,MACb,OAAO,OAAO,SAAS;AAAA,MACvB,UAAU,CAAC;AAAA,IACb;AACF,aAAS,QAAQ,CAAC,YAA6B;AAC9C,mBAAa,SAAS,KAAK;AAAA,QACtB,WAAW,QAAQ;AAAA,QACvB,MAAM,QAAQ;AAAA,QACd,SAAS,QAAQ;AAAA,QACjB,WAAW,QAAQ;AAAA,QACnB,UAAU,QAAQ;AAAA,MACnB,CAAC;AAAA,IACF,CAAC;AAED,WAAO;AAAA,EACP;AAAA,EAED,MAAa,aACZ,MACA,QACA,UACA,OACyB;AACvB,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,QAAQ,UAAU,OAAO,UAAU,CAAC,EAAC;AAAA,EACtD;AAAA,EAED,MAAa,oBACV,QACA,UACA,UACe;AACf,UAAM,YAAY,UAAU,EAAE,UAAU,OAAO,GAAG;AAAA,MAChD,YAAY,KAAK,IAAI;AAAA,IACvB,CAAC;AACD,eAAW,WAAW,UAAU;AAC9B,YAAM,aAAa,OAAO;AAAA,QACxB;AAAA,QACA,WAAW,QAAQ;AAAA,QACnB;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,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;AAAA,QACA,UAAU,OAAO;AAAA,QACjB,OAAO,OAAO;AAAA,QACd,WAAW,OAAO;AAAA,MACpB;AAAA,IACF,CAAC;AACD,WAAO;AAAA,EACT;AACF;;;AC7GO,IAAM,gBAAN,cAA4B,cAAuC;AAAA,EACxE,MAAa,UAAU,UAA+C;AACpE,UAAM,SAAS,MAAM,YAAY,QAAQ,EAAE,IAAI,SAAS,CAAC,EAAE,KAAa;AACxE,WAAO,UAAU;AAAA,EACnB;AAAA,EAED,MAAa,gBAAgB,YAAiD;AAC7E,UAAM,SAAS,MAAM,YAAY,QAAQ,EAAE,MAAM,WAAW,CAAC,EAAE,KAAa;AAC1E,WAAO,UAAU;AAAA,EACpB;AAAA,EAEA,MAAa,WAAW,QAA+B;AACpD,UAAM,YAAY,OAAO,MAAM;AAAA,EACjC;AAAA,EAED,MAAa,aAAa,UAAkB,QAA+B;AACxE,UAAM,YAAY,UAAU;AAAA,MAC1B,IAAI;AAAA,IACN,GAAG,MAAM;AAAA,EACX;AAAA,EAED,MAAa,aAAa,UAAiC;AACxD,UAAM,YAAY,UAAU,EAAE,IAAI,SAAS,CAAC;AAAA,EAC9C;AAAA,EAED,MAAa,cAAiC;AAC3C,UAAM,UAAU,MAAM,YAAY,KAAK,EAAE,KAAe;AACxD,WAAO;AAAA,EACT;AACF;","names":["loggers"]}

package/dist/models/intent.model.cjs

@@ -36,6 +36,12 @@ module.exports = __toCommonJS(intent_model_exports);
 var import_mongoose = __toESM(require("mongoose"), 1);
 var IntentObjectSchema = new import_mongoose.Schema(
   {
+    id: {
+      type: String,
+      required: true,
+      index: true,
+      unique: true
+    },
     name: {
       type: String,
       required: true,
@@ -49,8 +55,16 @@ var IntentObjectSchema = new import_mongoose.Schema(
       type: String,
       required: false
     },
-    llm: {
+    status: {
       type: String,
+      required: true
+    },
+    triggeringSentences: {
+      type: [String],
+      required: false
+    },
+    tags: {
+      type: [String],
       required: false
     }
   }

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

@@ -1 +1 @@
-{"version":3,"sources":["../../models/intent.model.ts"],"sourcesContent":["import mongoose, { type Document, Schema } from \"mongoose\";\n\nconst IntentObjectSchema = new Schema(\n\t{\n\t\tname: {\n\t\t\ttype: String,\n\t\t\trequired: true,\n\t\t\tindex: true,\n\t\t},\n\t\tdescription: {\n\t\t\ttype: String,\n\t\t\trequired: true,\n\t\t},\n\t\tprompt: {\n\t\t\ttype: String,\n\t\t\trequired: false,\n\t\t},\n\t\tllm: {\n\t\t\ttype: String,\n\t\t\trequired: false,\n\t\t},\n\t}\n);\n\nexport interface IntentDocument extends Document {\n\tname: string;\n\tdescription: string;\n\tprompt?: string;\n\tllm?: string;\n}\n\nexport const IntentModel = mongoose.model<IntentDocument>(\"Intent\", IntentObjectSchema);"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,sBAAgD;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,gBAAAA,QAAS,MAAsB,UAAU,kBAAkB;","names":["mongoose"]}
+{"version":3,"sources":["../../models/intent.model.ts"],"sourcesContent":["import { Intent } from \"@ainetwork/adk/types/memory\";\nimport mongoose, { type Document, Schema } from \"mongoose\";\n\nconst IntentObjectSchema = new Schema(\n\t{\n\t\tid: { \n\t\t\ttype: String,\n\t\t\trequired: true,\n\t\t\tindex: true,\n\t\t\tunique: true,\n\t\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\tstatus: {\n\t\t\ttype: String,\n\t\t\trequired: true,\n\t\t},\n\t\ttriggeringSentences: {\n\t\t\ttype: [String],\n\t\t\trequired: false,\n\t\t},\n\t\ttags: {\n\t\t\ttype: [String],\n\t\t\trequired: false,\n\t\t},\n\t},\n);\n\nexport interface IntentDocument extends Omit<Document, 'id'>, Omit<Intent, 'id'> {\n\tid: string;\n}\n\nexport const IntentModel = mongoose.model<IntentDocument>(\"Intent\", IntentObjectSchema);"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AACA,sBAAgD;AAEhD,IAAM,qBAAqB,IAAI;AAAA,EAC9B;AAAA,IACC,IAAI;AAAA,MACH,MAAM;AAAA,MACN,UAAU;AAAA,MACV,OAAO;AAAA,MACP,QAAQ;AAAA,IACT;AAAA,IACA,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,QAAQ;AAAA,MACP,MAAM;AAAA,MACN,UAAU;AAAA,IACX;AAAA,IACA,qBAAqB;AAAA,MACpB,MAAM,CAAC,MAAM;AAAA,MACb,UAAU;AAAA,IACX;AAAA,IACA,MAAM;AAAA,MACL,MAAM,CAAC,MAAM;AAAA,MACb,UAAU;AAAA,IACX;AAAA,EACD;AACD;AAMO,IAAM,cAAc,gBAAAA,QAAS,MAAsB,UAAU,kBAAkB;","names":["mongoose"]}

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

@@ -1,10 +1,8 @@
+import { Intent } from '@ainetwork/adk/types/memory';
 import mongoose, { Document } from 'mongoose';
 
-interface IntentDocument extends Document {
-    name: string;
-    description: string;
-    prompt?: string;
-    llm?: string;
+interface IntentDocument extends Omit<Document, 'id'>, Omit<Intent, 'id'> {
+    id: string;
 }
 declare const IntentModel: mongoose.Model<IntentDocument, {}, {}, {}, mongoose.Document<unknown, {}, IntentDocument, {}> & IntentDocument & Required<{
     _id: unknown;

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

@@ -1,10 +1,8 @@
+import { Intent } from '@ainetwork/adk/types/memory';
 import mongoose, { Document } from 'mongoose';
 
-interface IntentDocument extends Document {
-    name: string;
-    description: string;
-    prompt?: string;
-    llm?: string;
+interface IntentDocument extends Omit<Document, 'id'>, Omit<Intent, 'id'> {
+    id: string;
 }
 declare const IntentModel: mongoose.Model<IntentDocument, {}, {}, {}, mongoose.Document<unknown, {}, IntentDocument, {}> & IntentDocument & Required<{
     _id: unknown;

package/dist/models/intent.model.js

@@ -1,6 +1,6 @@
 import {
   IntentModel
-} from "../chunk-JCE5NAY5.js";
+} from "../chunk-YFW7JXII.js";
 export {
   IntentModel
 };

package/implements/intent.memory.ts

@@ -2,67 +2,34 @@ import type { Intent } from "@ainetwork/adk/types/memory";
 import { IIntentMemory } from "@ainetwork/adk/modules";
 import { MongoDBMemory } from "./base.memory";
 import { IntentModel } from "../models/intent.model";
-import { Types } from "mongoose";
 
 export class MongoDBIntent extends MongoDBMemory implements IIntentMemory {
   public async getIntent(intentId: string): Promise<Intent | undefined> {
-    const intent = await IntentModel.findById(new Types.ObjectId(intentId));
-    if (intent) {
-      return {
-        name: intent.name,
-        description: intent.description,
-        prompt: intent.prompt,
-        llm: intent.llm,
-      } as Intent;
-    }
-    return undefined;
+    const intent = await IntentModel.findOne({ id: intentId }).lean<Intent>();
+    return intent || undefined;
   };
 
 	public async getIntentByName(intentName: string): Promise<Intent | undefined> {
-		const intent = await IntentModel.findOne({ name: intentName });
-		if (intent) {
-			return {
-				name: intent.name,
-				description: intent.description,
-				prompt: intent.prompt,
-				llm: intent.llm,
-			} as Intent;
-		}
-		return undefined;
+		const intent = await IntentModel.findOne({ name: intentName }).lean<Intent>();
+    return intent || undefined;
 	}
 
 	public async saveIntent(intent: Intent): Promise<void> {
-    // ObjectId automatically generated (MongoDB automatically generates)
-    await IntentModel.create({
-      name: intent.name,
-      description: intent.description,
-      prompt: intent.prompt,
-      llm: intent.llm,
-    });
+    await IntentModel.create(intent);
   };
 
 	public async updateIntent(intentId: string, intent: Intent): Promise<void> {
     await IntentModel.updateOne({
-      _id: new Types.ObjectId(intentId),
-    },{
-      name: intent.name,
-      description: intent.description,
-      prompt: intent.prompt,
-      llm: intent.llm,
-    });
+      id: intentId,
+    }, intent);
   };
 
 	public async deleteIntent(intentId: string): Promise<void> {
-    await IntentModel.deleteOne({ _id: new Types.ObjectId(intentId) });
+    await IntentModel.deleteOne({ id: intentId });
   };
 
 	public async listIntents(): Promise<Intent[]> {
-    const intents = await IntentModel.find();
-    return intents.map(intent => ({
-      name: intent.name,
-      description: intent.description,
-      prompt: intent.prompt,
-      llm: intent.llm,
-    } as Intent));
+    const intents = await IntentModel.find().lean<Intent[]>();
+    return intents;
   };
 }

package/implements/thread.memory.ts

@@ -25,6 +25,8 @@ export class MongoDBThread extends MongoDBMemory implements IThreadMemory {
 		loggers.agent.debug(`Found ${messages.length} messages for thread ${threadId}`);
 
 		const threadObject: ThreadObject = { 
+      threadId: thread.threadId, 
+      userId: thread.userId,
       type: thread.type as ThreadType,
       title: thread.title || "New thread",
       messages: []
@@ -47,7 +49,7 @@ export class MongoDBThread extends MongoDBMemory implements IThreadMemory {
 		userId: string,
 		threadId: string,
 		title: string,
-  ): Promise<ThreadMetadata> {
+  ): Promise<ThreadObject> {
     const now = Date.now();
     await ThreadModel.create({
       type,
@@ -58,7 +60,7 @@ export class MongoDBThread extends MongoDBMemory implements IThreadMemory {
       created_at: now,
     });
 
-    return { type, threadId, title, updatedAt: now };
+    return { type, userId, threadId, title, messages: []};
   };
 
 	public async addMessagesToThread(
@@ -102,6 +104,7 @@ export class MongoDBThread extends MongoDBMemory implements IThreadMemory {
     const data: ThreadMetadata[] = threads.map((thread: ThreadDocument) => {
       return {
         type: thread.type,
+        userId,
         threadId: thread.threadId,
         title: thread.title,
         updatedAt: thread.updated_at

package/models/intent.model.ts

@@ -1,7 +1,14 @@
+import { Intent } from "@ainetwork/adk/types/memory";
 import mongoose, { type Document, Schema } from "mongoose";
 
 const IntentObjectSchema = new Schema(
 	{
+		id: { 
+			type: String,
+			required: true,
+			index: true,
+			unique: true,
+		},
 		name: {
 			type: String,
 			required: true,
@@ -15,18 +22,23 @@ const IntentObjectSchema = new Schema(
 			type: String,
 			required: false,
 		},
-		llm: {
+		status: {
 			type: String,
+			required: true,
+		},
+		triggeringSentences: {
+			type: [String],
+			required: false,
+		},
+		tags: {
+			type: [String],
 			required: false,
 		},
-	}
+	},
 );
 
-export interface IntentDocument extends Document {
-	name: string;
-	description: string;
-	prompt?: string;
-	llm?: string;
+export interface IntentDocument extends Omit<Document, 'id'>, Omit<Intent, 'id'> {
+	id: string;
 }
 
 export const IntentModel = mongoose.model<IntentDocument>("Intent", IntentObjectSchema);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ainetwork/adk-provider-memory-mongodb",
-  "version": "0.2.1",
+  "version": "0.2.3",
   "author": "AI Network (https://ainetwork.ai)",
   "type": "module",
   "engines": {
@@ -28,7 +28,7 @@
     "clean": "rm -rf dist"
   },
   "dependencies": {
-    "@ainetwork/adk": "^0.2.5",
+    "@ainetwork/adk": "0.2.9",
     "mongoose": "^8.16.5"
   },
   "devDependencies": {
@@ -38,5 +38,5 @@
   "publishConfig": {
     "access": "public"
   },
-  "gitHead": "65f5c77dad2ba16ee695a88f9b1b3ca2a5a886a3"
+  "gitHead": "da5af7747935eeb8cedeb932cb2e918da6dc36c3"
 }

package/dist/chunk-JCE5NAY5.js.map

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