@ainetwork/adk-provider-memory-mongodb 0.2.4 → 0.3.2

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\";\nexport { MongoDBMemory } from \"./implements/base.memory\";\nexport type { MongoDBMemoryConfig } from \"./implements/base.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 interface MongoDBMemoryConfig {\n  uri: string;\n  maxReconnectAttempts?: number;\n  reconnectInterval?: number;\n  maxPoolSize?: number;\n  serverSelectionTimeoutMS?: number;\n  socketTimeoutMS?: number;\n  connectTimeoutMS?: number;\n  operationTimeoutMS?: number; // Timeout for database operations\n}\n\nexport class MongoDBMemory implements IMemory {\n  private static instance: MongoDBMemory;\n  private uri: string;\n  private connected: boolean = false;\n  private reconnectAttempts: number = 0;\n  private maxReconnectAttempts: number;\n  private reconnectInterval: number;\n  private reconnecting: boolean = false;\n  private connectionConfig: mongoose.ConnectOptions;\n  private eventListenersSetup: boolean = false;\n  private operationTimeoutMS: number;\n\n  constructor(config: string | MongoDBMemoryConfig) {\n    const cfg = typeof config === 'string' ? { uri: config } : config;\n\n    this.uri = cfg.uri;\n    this.maxReconnectAttempts = cfg.maxReconnectAttempts ?? 5;\n    this.reconnectInterval = cfg.reconnectInterval ?? 5000;\n    this.operationTimeoutMS = cfg.operationTimeoutMS ?? 10000; // Default 10 seconds\n    this.connectionConfig = {\n      maxPoolSize: cfg.maxPoolSize ?? 1,\n      serverSelectionTimeoutMS: cfg.serverSelectionTimeoutMS ?? 30000,\n      socketTimeoutMS: cfg.socketTimeoutMS ?? 45000,\n      connectTimeoutMS: cfg.connectTimeoutMS ?? 30000,\n      bufferCommands: false,\n    };\n\n    if (!MongoDBMemory.instance) {\n      MongoDBMemory.instance = this;\n      this.setupMongooseEventListeners();\n    } else {\n      // Use existing instance's connection state\n      this.connected = MongoDBMemory.instance.connected;\n      this.operationTimeoutMS = MongoDBMemory.instance.operationTimeoutMS;\n    }\n  }\n\n  private setupMongooseEventListeners(): void {\n    if (this.eventListenersSetup) return;\n\n    this.eventListenersSetup = true;\n\n    mongoose.connection.on(\"connected\", () => {\n      this.connected = true;\n      this.reconnectAttempts = 0;\n      this.reconnecting = false;\n      loggers.agent.info(\"MongoDB connected successfully\");\n    });\n\n    mongoose.connection.on(\"disconnected\", () => {\n      this.connected = false;\n      loggers.agent.warn(\"MongoDB disconnected\");\n      this.handleDisconnection();\n    });\n\n    mongoose.connection.on(\"error\", (error) => {\n      this.connected = false;\n      loggers.agent.error(\"MongoDB connection error:\", error);\n      this.handleDisconnection();\n    });\n\n    mongoose.connection.on(\"reconnected\", () => {\n      this.connected = true;\n      this.reconnectAttempts = 0;\n      this.reconnecting = false;\n      loggers.agent.info(\"MongoDB reconnected successfully\");\n    });\n  }\n\n  private async handleDisconnection(): Promise<void> {\n    if (this.reconnecting) {\n      return;\n    }\n\n    this.reconnecting = true;\n\n    while (this.reconnectAttempts < this.maxReconnectAttempts && !this.isConnected) {\n      this.reconnectAttempts++;\n      loggers.agent.info(\n        `Attempting to reconnect to MongoDB (${this.reconnectAttempts}/${this.maxReconnectAttempts})...`\n      );\n\n      try {\n        await mongoose.connect(this.uri, this.connectionConfig);\n        this.connected = true;\n        this.reconnectAttempts = 0;\n        this.reconnecting = false;\n        loggers.agent.info(\"MongoDB reconnection successful\");\n        return;\n      } catch (error) {\n        loggers.agent.error(\n          `Reconnection attempt ${this.reconnectAttempts} failed:`,\n          error\n        );\n\n        if (this.reconnectAttempts < this.maxReconnectAttempts) {\n          await new Promise((resolve) =>\n            setTimeout(resolve, this.reconnectInterval)\n          );\n        }\n      }\n    }\n\n    this.reconnecting = false;\n\n    if (!this.isConnected) {\n      loggers.agent.error(\n        `Failed to reconnect to MongoDB after ${this.maxReconnectAttempts} attempts`\n      );\n    }\n  }\n\n  public async connect(): Promise<void> {\n    if (this.connected) {\n      return;\n    }\n\n    try {\n      await mongoose.connect(this.uri, this.connectionConfig);\n      this.connected = true;\n      this.reconnectAttempts = 0;\n    } catch (error) {\n      loggers.agent.error(\"Failed to connect to MongoDB:\", error);\n      throw error;\n    }\n  }\n\n  public async disconnect(): Promise<void> {\n    if (!this.isConnected) {\n      return;\n    }\n\n    try {\n      await mongoose.disconnect();\n      this.connected = false;\n    } catch (error) {\n      loggers.agent.error(\"Failed to disconnect from MongoDB:\", error);\n      throw error;\n    }\n  }\n\n  public isConnected(): boolean {\n    return this.connected;\n  }\n\n  private async ensureConnection(): Promise<void> {\n    if (!this.isConnected && !this.reconnecting) {\n      await this.connect();\n    }\n\n    // Wait for reconnection if in progress\n    const maxWaitTime = 30000; // 30 seconds\n    const startTime = Date.now();\n    while (this.reconnecting && Date.now() - startTime < maxWaitTime) {\n      await new Promise((resolve) => setTimeout(resolve, 100));\n    }\n\n    if (!this.isConnected) {\n      throw new Error(\"MongoDB is not connected and reconnection failed\");\n    }\n  }\n\n  /**\n   * Get the operation timeout in milliseconds\n   */\n  protected getOperationTimeout(): number {\n    return this.operationTimeoutMS;\n  }\n\n  /**\n   * Execute a database operation with automatic retry on connection errors\n   * Note: Use mongoose's maxTimeMS option in queries for timeout control\n   */\n  protected async executeWithRetry<T>(\n    operation: () => Promise<T>,\n    operationName: string = \"Database operation\"\n  ): Promise<T> {\n    await this.ensureConnection();\n\n    try {\n      return await operation();\n    } catch (error: any) {\n      // Check if it's a timeout error from MongoDB\n      if (error.code === 50 || error.message?.includes(\"operation exceeded time limit\")) {\n        loggers.agent.error(`${operationName} exceeded time limit`);\n        throw error;\n      }\n\n      // Check if it's a connection-related error\n      if (\n        error.name === \"MongoNetworkError\" ||\n        error.name === \"MongoServerError\" ||\n        error.message?.includes(\"connection\") ||\n        error.message?.includes(\"disconnect\")\n      ) {\n        loggers.agent.warn(\n          `${operationName} failed due to connection issue, attempting reconnection...`\n        );\n\n        await this.ensureConnection();\n\n        // Retry the operation once after reconnection\n        try {\n          return await operation();\n        } catch (retryError: any) {\n          loggers.agent.error(`${operationName} failed after retry:`, retryError);\n          throw retryError;\n        }\n      }\n\n      // If it's not a connection error, just throw it\n      throw error;\n    }\n  }\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    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      const thread = await ThreadModel.findOne({ threadId, userId }).maxTimeMS(timeout);\n      const messages = await MessageModel.find({ threadId, userId })\n        .sort({ timestamp: 1 })\n        .maxTimeMS(timeout);\n\n      if (!thread) return undefined;\n\n      loggers.agent.debug(`Found ${messages.length} messages for thread ${threadId}`);\n\n      const 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      messages.forEach((message: MessageDocument) => {\n        threadObject.messages.push({\n          messageId: message.messageId,\n          role: message.role as MessageRole,\n          content: message.content,\n          timestamp: message.timestamp,\n          metadata: message.metadata,\n        });\n      });\n\n      return threadObject;\n    }, `getThread(${userId}, ${threadId})`);\n  };\n\n  public async createThread(\n    type: ThreadType,\n    userId: string,\n    threadId: string,\n    title: string,\n  ): Promise<ThreadObject> {\n    return this.executeWithRetry(async () => {\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    }, `createThread(${userId}, ${threadId})`);\n  };\n\n  public async addMessagesToThread(\n    userId: string,\n    threadId: string,\n    messages: MessageObject[]\n  ): Promise<void> {\n    return this.executeWithRetry(async () => {\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    }, `addMessagesToThread(${userId}, ${threadId})`);\n  };\n\n  public async deleteThread(userId: string, threadId: string): Promise<void> {\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      const messages = await MessageModel.find({ userId, threadId })\n        .sort({ timestamp: 1 })\n        .maxTimeMS(timeout);\n\n      messages?.forEach((message: MessageDocument) => {\n        message.deleteOne();\n      });\n\n      const thread = await ThreadModel.findOne({ userId, threadId }).maxTimeMS(timeout);\n      thread?.deleteOne();\n    }, `deleteThread(${userId}, ${threadId})`);\n  };\n\n  public async listThreads(userId: string): Promise<ThreadMetadata[]> {\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      const threads = await ThreadModel.find({ userId })\n        .sort({ updated_at: -1 })\n        .maxTimeMS(timeout);\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    }, `listThreads(${userId})`);\n  };\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    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      const intent = await IntentModel.findOne({ id: intentId })\n        .maxTimeMS(timeout)\n        .lean<Intent>();\n      return intent || undefined;\n    }, `getIntent(${intentId})`);\n  };\n\n  public async getIntentByName(intentName: string): Promise<Intent | undefined> {\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      const intent = await IntentModel.findOne({ name: intentName })\n        .maxTimeMS(timeout)\n        .lean<Intent>();\n      return intent || undefined;\n    }, `getIntentByName(${intentName})`);\n  }\n\n  public async saveIntent(intent: Intent): Promise<void> {\n    return this.executeWithRetry(async () => {\n      await IntentModel.create(intent);\n    }, `saveIntent(${intent.id})`);\n  };\n\n  public async updateIntent(intentId: string, intent: Intent): Promise<void> {\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      await IntentModel.updateOne({\n        id: intentId,\n      }, intent).maxTimeMS(timeout);\n    }, `updateIntent(${intentId})`);\n  };\n\n  public async deleteIntent(intentId: string): Promise<void> {\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      await IntentModel.deleteOne({ id: intentId }).maxTimeMS(timeout);\n    }, `deleteIntent(${intentId})`);\n  };\n\n  public async listIntents(): Promise<Intent[]> {\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      const intents = await IntentModel.find()\n        .maxTimeMS(timeout)\n        .lean<Intent[]>();\n      return intents;\n    }, `listIntents()`);\n  };\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACCA,sBAAqB;AACrB,oBAAwB;AAajB,IAAM,gBAAN,MAAM,eAAiC;AAAA,EAC5C,OAAe;AAAA,EACP;AAAA,EACA,YAAqB;AAAA,EACrB,oBAA4B;AAAA,EAC5B;AAAA,EACA;AAAA,EACA,eAAwB;AAAA,EACxB;AAAA,EACA,sBAA+B;AAAA,EAC/B;AAAA,EAER,YAAY,QAAsC;AAChD,UAAM,MAAM,OAAO,WAAW,WAAW,EAAE,KAAK,OAAO,IAAI;AAE3D,SAAK,MAAM,IAAI;AACf,SAAK,uBAAuB,IAAI,wBAAwB;AACxD,SAAK,oBAAoB,IAAI,qBAAqB;AAClD,SAAK,qBAAqB,IAAI,sBAAsB;AACpD,SAAK,mBAAmB;AAAA,MACtB,aAAa,IAAI,eAAe;AAAA,MAChC,0BAA0B,IAAI,4BAA4B;AAAA,MAC1D,iBAAiB,IAAI,mBAAmB;AAAA,MACxC,kBAAkB,IAAI,oBAAoB;AAAA,MAC1C,gBAAgB;AAAA,IAClB;AAEA,QAAI,CAAC,eAAc,UAAU;AAC3B,qBAAc,WAAW;AACzB,WAAK,4BAA4B;AAAA,IACnC,OAAO;AAEL,WAAK,YAAY,eAAc,SAAS;AACxC,WAAK,qBAAqB,eAAc,SAAS;AAAA,IACnD;AAAA,EACF;AAAA,EAEQ,8BAAoC;AAC1C,QAAI,KAAK,oBAAqB;AAE9B,SAAK,sBAAsB;AAE3B,oBAAAA,QAAS,WAAW,GAAG,aAAa,MAAM;AACxC,WAAK,YAAY;AACjB,WAAK,oBAAoB;AACzB,WAAK,eAAe;AACpB,4BAAQ,MAAM,KAAK,gCAAgC;AAAA,IACrD,CAAC;AAED,oBAAAA,QAAS,WAAW,GAAG,gBAAgB,MAAM;AAC3C,WAAK,YAAY;AACjB,4BAAQ,MAAM,KAAK,sBAAsB;AACzC,WAAK,oBAAoB;AAAA,IAC3B,CAAC;AAED,oBAAAA,QAAS,WAAW,GAAG,SAAS,CAAC,UAAU;AACzC,WAAK,YAAY;AACjB,4BAAQ,MAAM,MAAM,6BAA6B,KAAK;AACtD,WAAK,oBAAoB;AAAA,IAC3B,CAAC;AAED,oBAAAA,QAAS,WAAW,GAAG,eAAe,MAAM;AAC1C,WAAK,YAAY;AACjB,WAAK,oBAAoB;AACzB,WAAK,eAAe;AACpB,4BAAQ,MAAM,KAAK,kCAAkC;AAAA,IACvD,CAAC;AAAA,EACH;AAAA,EAEA,MAAc,sBAAqC;AACjD,QAAI,KAAK,cAAc;AACrB;AAAA,IACF;AAEA,SAAK,eAAe;AAEpB,WAAO,KAAK,oBAAoB,KAAK,wBAAwB,CAAC,KAAK,aAAa;AAC9E,WAAK;AACL,4BAAQ,MAAM;AAAA,QACZ,uCAAuC,KAAK,iBAAiB,IAAI,KAAK,oBAAoB;AAAA,MAC5F;AAEA,UAAI;AACF,cAAM,gBAAAA,QAAS,QAAQ,KAAK,KAAK,KAAK,gBAAgB;AACtD,aAAK,YAAY;AACjB,aAAK,oBAAoB;AACzB,aAAK,eAAe;AACpB,8BAAQ,MAAM,KAAK,iCAAiC;AACpD;AAAA,MACF,SAAS,OAAO;AACd,8BAAQ,MAAM;AAAA,UACZ,wBAAwB,KAAK,iBAAiB;AAAA,UAC9C;AAAA,QACF;AAEA,YAAI,KAAK,oBAAoB,KAAK,sBAAsB;AACtD,gBAAM,IAAI;AAAA,YAAQ,CAAC,YACjB,WAAW,SAAS,KAAK,iBAAiB;AAAA,UAC5C;AAAA,QACF;AAAA,MACF;AAAA,IACF;AAEA,SAAK,eAAe;AAEpB,QAAI,CAAC,KAAK,aAAa;AACrB,4BAAQ,MAAM;AAAA,QACZ,wCAAwC,KAAK,oBAAoB;AAAA,MACnE;AAAA,IACF;AAAA,EACF;AAAA,EAEA,MAAa,UAAyB;AACpC,QAAI,KAAK,WAAW;AAClB;AAAA,IACF;AAEA,QAAI;AACF,YAAM,gBAAAA,QAAS,QAAQ,KAAK,KAAK,KAAK,gBAAgB;AACtD,WAAK,YAAY;AACjB,WAAK,oBAAoB;AAAA,IAC3B,SAAS,OAAO;AACd,4BAAQ,MAAM,MAAM,iCAAiC,KAAK;AAC1D,YAAM;AAAA,IACR;AAAA,EACF;AAAA,EAEA,MAAa,aAA4B;AACvC,QAAI,CAAC,KAAK,aAAa;AACrB;AAAA,IACF;AAEA,QAAI;AACF,YAAM,gBAAAA,QAAS,WAAW;AAC1B,WAAK,YAAY;AAAA,IACnB,SAAS,OAAO;AACd,4BAAQ,MAAM,MAAM,sCAAsC,KAAK;AAC/D,YAAM;AAAA,IACR;AAAA,EACF;AAAA,EAEO,cAAuB;AAC5B,WAAO,KAAK;AAAA,EACd;AAAA,EAEA,MAAc,mBAAkC;AAC9C,QAAI,CAAC,KAAK,eAAe,CAAC,KAAK,cAAc;AAC3C,YAAM,KAAK,QAAQ;AAAA,IACrB;AAGA,UAAM,cAAc;AACpB,UAAM,YAAY,KAAK,IAAI;AAC3B,WAAO,KAAK,gBAAgB,KAAK,IAAI,IAAI,YAAY,aAAa;AAChE,YAAM,IAAI,QAAQ,CAAC,YAAY,WAAW,SAAS,GAAG,CAAC;AAAA,IACzD;AAEA,QAAI,CAAC,KAAK,aAAa;AACrB,YAAM,IAAI,MAAM,kDAAkD;AAAA,IACpE;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKU,sBAA8B;AACtC,WAAO,KAAK;AAAA,EACd;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,MAAgB,iBACd,WACA,gBAAwB,sBACZ;AACZ,UAAM,KAAK,iBAAiB;AAE5B,QAAI;AACF,aAAO,MAAM,UAAU;AAAA,IACzB,SAAS,OAAY;AAEnB,UAAI,MAAM,SAAS,MAAM,MAAM,SAAS,SAAS,+BAA+B,GAAG;AACjF,8BAAQ,MAAM,MAAM,GAAG,aAAa,sBAAsB;AAC1D,cAAM;AAAA,MACR;AAGA,UACE,MAAM,SAAS,uBACf,MAAM,SAAS,sBACf,MAAM,SAAS,SAAS,YAAY,KACpC,MAAM,SAAS,SAAS,YAAY,GACpC;AACA,8BAAQ,MAAM;AAAA,UACZ,GAAG,aAAa;AAAA,QAClB;AAEA,cAAM,KAAK,iBAAiB;AAG5B,YAAI;AACF,iBAAO,MAAM,UAAU;AAAA,QACzB,SAAS,YAAiB;AACxB,gCAAQ,MAAM,MAAM,GAAG,aAAa,wBAAwB,UAAU;AACtE,gBAAM;AAAA,QACR;AAAA,MACF;AAGA,YAAM;AAAA,IACR;AAAA,EACF;AACF;;;ACrOA,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,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,UAAU,KAAK,oBAAoB;AACzC,YAAM,SAAS,MAAM,YAAY,QAAQ,EAAE,UAAU,OAAO,CAAC,EAAE,UAAU,OAAO;AAChF,YAAM,WAAW,MAAM,aAAa,KAAK,EAAE,UAAU,OAAO,CAAC,EAC1D,KAAK,EAAE,WAAW,EAAE,CAAC,EACrB,UAAU,OAAO;AAEpB,UAAI,CAAC,OAAQ,QAAO;AAEpB,6BAAQ,MAAM,MAAM,SAAS,SAAS,MAAM,wBAAwB,QAAQ,EAAE;AAE9E,YAAM,eAA6B;AAAA,QACjC,UAAU,OAAO;AAAA,QACjB,QAAQ,OAAO;AAAA,QACf,MAAM,OAAO;AAAA,QACb,OAAO,OAAO,SAAS;AAAA,QACvB,UAAU,CAAC;AAAA,MACb;AACA,eAAS,QAAQ,CAAC,YAA6B;AAC7C,qBAAa,SAAS,KAAK;AAAA,UACzB,WAAW,QAAQ;AAAA,UACnB,MAAM,QAAQ;AAAA,UACd,SAAS,QAAQ;AAAA,UACjB,WAAW,QAAQ;AAAA,UACnB,UAAU,QAAQ;AAAA,QACpB,CAAC;AAAA,MACH,CAAC;AAED,aAAO;AAAA,IACT,GAAG,aAAa,MAAM,KAAK,QAAQ,GAAG;AAAA,EACxC;AAAA,EAEA,MAAa,aACX,MACA,QACA,UACA,OACuB;AACvB,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,MAAM,KAAK,IAAI;AACrB,YAAM,YAAY,OAAO;AAAA,QACvB;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA,YAAY;AAAA,QACZ,YAAY;AAAA,MACd,CAAC;AAED,aAAO,EAAE,MAAM,QAAQ,UAAU,OAAO,UAAU,CAAC,EAAC;AAAA,IACtD,GAAG,gBAAgB,MAAM,KAAK,QAAQ,GAAG;AAAA,EAC3C;AAAA,EAEA,MAAa,oBACX,QACA,UACA,UACe;AACf,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,YAAY,UAAU,EAAE,UAAU,OAAO,GAAG;AAAA,QAChD,YAAY,KAAK,IAAI;AAAA,MACvB,CAAC;AACD,iBAAW,WAAW,UAAU;AAC9B,cAAM,aAAa,OAAO;AAAA,UACxB;AAAA,UACA,WAAW,QAAQ;AAAA,UACnB;AAAA,UACA,MAAM,QAAQ;AAAA,UACd,SAAS,QAAQ;AAAA,UACjB,WAAW,QAAQ;AAAA,UACnB,UAAU,QAAQ;AAAA,QACpB,CAAC;AAAA,MACH;AAAA,IACF,GAAG,uBAAuB,MAAM,KAAK,QAAQ,GAAG;AAAA,EAClD;AAAA,EAEA,MAAa,aAAa,QAAgB,UAAiC;AACzE,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,UAAU,KAAK,oBAAoB;AACzC,YAAM,WAAW,MAAM,aAAa,KAAK,EAAE,QAAQ,SAAS,CAAC,EAC1D,KAAK,EAAE,WAAW,EAAE,CAAC,EACrB,UAAU,OAAO;AAEpB,gBAAU,QAAQ,CAAC,YAA6B;AAC9C,gBAAQ,UAAU;AAAA,MACpB,CAAC;AAED,YAAM,SAAS,MAAM,YAAY,QAAQ,EAAE,QAAQ,SAAS,CAAC,EAAE,UAAU,OAAO;AAChF,cAAQ,UAAU;AAAA,IACpB,GAAG,gBAAgB,MAAM,KAAK,QAAQ,GAAG;AAAA,EAC3C;AAAA,EAEA,MAAa,YAAY,QAA2C;AAClE,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,UAAU,KAAK,oBAAoB;AACzC,YAAM,UAAU,MAAM,YAAY,KAAK,EAAE,OAAO,CAAC,EAC9C,KAAK,EAAE,YAAY,GAAG,CAAC,EACvB,UAAU,OAAO;AACpB,YAAM,OAAyB,QAAQ,IAAI,CAAC,WAA2B;AACrE,eAAO;AAAA,UACL,MAAM,OAAO;AAAA,UACb;AAAA,UACA,UAAU,OAAO;AAAA,UACjB,OAAO,OAAO;AAAA,UACd,WAAW,OAAO;AAAA,QACpB;AAAA,MACF,CAAC;AACD,aAAO;AAAA,IACT,GAAG,eAAe,MAAM,GAAG;AAAA,EAC7B;AACF;;;AC9HA,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,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,UAAU,KAAK,oBAAoB;AACzC,YAAM,SAAS,MAAM,YAAY,QAAQ,EAAE,IAAI,SAAS,CAAC,EACtD,UAAU,OAAO,EACjB,KAAa;AAChB,aAAO,UAAU;AAAA,IACnB,GAAG,aAAa,QAAQ,GAAG;AAAA,EAC7B;AAAA,EAEA,MAAa,gBAAgB,YAAiD;AAC5E,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,UAAU,KAAK,oBAAoB;AACzC,YAAM,SAAS,MAAM,YAAY,QAAQ,EAAE,MAAM,WAAW,CAAC,EAC1D,UAAU,OAAO,EACjB,KAAa;AAChB,aAAO,UAAU;AAAA,IACnB,GAAG,mBAAmB,UAAU,GAAG;AAAA,EACrC;AAAA,EAEA,MAAa,WAAW,QAA+B;AACrD,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,YAAY,OAAO,MAAM;AAAA,IACjC,GAAG,cAAc,OAAO,EAAE,GAAG;AAAA,EAC/B;AAAA,EAEA,MAAa,aAAa,UAAkB,QAA+B;AACzE,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,UAAU,KAAK,oBAAoB;AACzC,YAAM,YAAY,UAAU;AAAA,QAC1B,IAAI;AAAA,MACN,GAAG,MAAM,EAAE,UAAU,OAAO;AAAA,IAC9B,GAAG,gBAAgB,QAAQ,GAAG;AAAA,EAChC;AAAA,EAEA,MAAa,aAAa,UAAiC;AACzD,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,UAAU,KAAK,oBAAoB;AACzC,YAAM,YAAY,UAAU,EAAE,IAAI,SAAS,CAAC,EAAE,UAAU,OAAO;AAAA,IACjE,GAAG,gBAAgB,QAAQ,GAAG;AAAA,EAChC;AAAA,EAEA,MAAa,cAAiC;AAC5C,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,UAAU,KAAK,oBAAoB;AACzC,YAAM,UAAU,MAAM,YAAY,KAAK,EACpC,UAAU,OAAO,EACjB,KAAe;AAClB,aAAO;AAAA,IACT,GAAG,eAAe;AAAA,EACpB;AACF;","names":["mongoose","import_mongoose","mongoose","import_memory","import_mongoose","mongoose","import_logger","import_mongoose","mongoose"]}
+{"version":3,"sources":["../index.ts","../implements/base.memory.ts","../models/agent.model.ts","../implements/agent.memory.ts","../models/intent.model.ts","../implements/intent.memory.ts","../models/threads.model.ts","../models/messages.model.ts","../implements/thread.memory.ts"],"sourcesContent":["export { MongoDBMemory } from \"./implements/base.memory\";\nexport type { MongoDBMemoryConfig } from \"./implements/base.memory\";","import { IAgentMemory, IIntentMemory, IMemory, IThreadMemory } from \"node_modules/@ainetwork/adk/dist/esm/modules/memory/base.memory\";\nimport mongoose from \"mongoose\";\nimport { loggers } from \"@ainetwork/adk/utils/logger\";\nimport { MongoDBAgent } from \"./agent.memory\";\nimport { MongoDBIntent } from \"./intent.memory\";\nimport { MongoDBThread } from \"./thread.memory\";\n\nexport interface MongoDBMemoryConfig {\n  uri: string;\n  maxReconnectAttempts?: number;\n  reconnectInterval?: number;\n  maxPoolSize?: number;\n  serverSelectionTimeoutMS?: number;\n  socketTimeoutMS?: number;\n  connectTimeoutMS?: number;\n  operationTimeoutMS?: number; // Timeout for database operations\n}\n\nexport class MongoDBMemory implements IMemory {\n  private static instance: MongoDBMemory;\n  private uri: string;\n  private connected: boolean = false;\n  private reconnectAttempts: number = 0;\n  private maxReconnectAttempts: number;\n  private reconnectInterval: number;\n  private reconnecting: boolean = false;\n  private connectionConfig: mongoose.ConnectOptions;\n  private eventListenersSetup: boolean = false;\n  private operationTimeoutMS: number;\n\n  private agentMemory: MongoDBAgent;\n  private intentMemory: MongoDBIntent;\n  private threadMemory: MongoDBThread;\n\n  constructor(config: string | MongoDBMemoryConfig) {\n    const cfg = typeof config === 'string' ? { uri: config } : config;\n\n    this.uri = cfg.uri;\n    this.maxReconnectAttempts = cfg.maxReconnectAttempts ?? 5;\n    this.reconnectInterval = cfg.reconnectInterval ?? 5000;\n    this.operationTimeoutMS = cfg.operationTimeoutMS ?? 10000; // Default 10 seconds\n    this.connectionConfig = {\n      maxPoolSize: cfg.maxPoolSize ?? 1,\n      serverSelectionTimeoutMS: cfg.serverSelectionTimeoutMS ?? 30000,\n      socketTimeoutMS: cfg.socketTimeoutMS ?? 45000,\n      connectTimeoutMS: cfg.connectTimeoutMS ?? 30000,\n      bufferCommands: false,\n    };\n\n    if (!MongoDBMemory.instance) {\n      MongoDBMemory.instance = this;\n      this.setupMongooseEventListeners();\n    } else {\n      // Use existing instance's connection state\n      this.connected = MongoDBMemory.instance.connected;\n      this.operationTimeoutMS = MongoDBMemory.instance.operationTimeoutMS;\n    }\n\n\t\tthis.agentMemory = new MongoDBAgent(\n\t\t\tthis.executeWithRetry.bind(this),\n\t\t\tthis.getOperationTimeout.bind(this)\n\t\t);\n\n\t\tthis.threadMemory = new MongoDBThread(\n\t\t\tthis.executeWithRetry.bind(this),\n\t\t\tthis.getOperationTimeout.bind(this)\n\t\t);\n\n\t\tthis.intentMemory = new MongoDBIntent(\n\t\t\tthis.executeWithRetry.bind(this),\n\t\t\tthis.getOperationTimeout.bind(this)\n\t\t);\n  }\n\n  public getAgentMemory(): IAgentMemory {\n    return this.agentMemory;\n  }\n\n  public getThreadMemory(): IThreadMemory {\n    return this.threadMemory;\n  }\n\n  public getIntentMemory(): IIntentMemory {\n    return this.intentMemory;\n  }\n\n  private setupMongooseEventListeners(): void {\n    if (this.eventListenersSetup) return;\n\n    this.eventListenersSetup = true;\n\n    mongoose.connection.on(\"connected\", () => {\n      this.connected = true;\n      this.reconnectAttempts = 0;\n      this.reconnecting = false;\n      loggers.agent.info(\"MongoDB connected successfully\");\n    });\n\n    mongoose.connection.on(\"disconnected\", () => {\n      this.connected = false;\n      loggers.agent.warn(\"MongoDB disconnected\");\n      this.handleDisconnection();\n    });\n\n    mongoose.connection.on(\"error\", (error) => {\n      this.connected = false;\n      loggers.agent.error(\"MongoDB connection error:\", error);\n      this.handleDisconnection();\n    });\n\n    mongoose.connection.on(\"reconnected\", () => {\n      this.connected = true;\n      this.reconnectAttempts = 0;\n      this.reconnecting = false;\n      loggers.agent.info(\"MongoDB reconnected successfully\");\n    });\n  }\n\n  private async handleDisconnection(): Promise<void> {\n    if (this.reconnecting) {\n      return;\n    }\n\n    this.reconnecting = true;\n\n    while (this.reconnectAttempts < this.maxReconnectAttempts && !this.isConnected) {\n      this.reconnectAttempts++;\n      loggers.agent.info(\n        `Attempting to reconnect to MongoDB (${this.reconnectAttempts}/${this.maxReconnectAttempts})...`\n      );\n\n      try {\n        await mongoose.connect(this.uri, this.connectionConfig);\n        this.connected = true;\n        this.reconnectAttempts = 0;\n        this.reconnecting = false;\n        loggers.agent.info(\"MongoDB reconnection successful\");\n        return;\n      } catch (error) {\n        loggers.agent.error(\n          `Reconnection attempt ${this.reconnectAttempts} failed:`,\n          error\n        );\n\n        if (this.reconnectAttempts < this.maxReconnectAttempts) {\n          await new Promise((resolve) =>\n            setTimeout(resolve, this.reconnectInterval)\n          );\n        }\n      }\n    }\n\n    this.reconnecting = false;\n\n    if (!this.isConnected) {\n      loggers.agent.error(\n        `Failed to reconnect to MongoDB after ${this.maxReconnectAttempts} attempts`\n      );\n    }\n  }\n\n  public async connect(): Promise<void> {\n    if (this.connected) {\n      return;\n    }\n\n    try {\n      await mongoose.connect(this.uri, this.connectionConfig);\n      this.connected = true;\n      this.reconnectAttempts = 0;\n    } catch (error) {\n      loggers.agent.error(\"Failed to connect to MongoDB:\", error);\n      throw error;\n    }\n  }\n\n  public async disconnect(): Promise<void> {\n    if (!this.isConnected) {\n      return;\n    }\n\n    try {\n      await mongoose.disconnect();\n      this.connected = false;\n    } catch (error) {\n      loggers.agent.error(\"Failed to disconnect from MongoDB:\", error);\n      throw error;\n    }\n  }\n\n  public isConnected(): boolean {\n    return this.connected;\n  }\n\n  private async ensureConnection(): Promise<void> {\n    if (!this.isConnected && !this.reconnecting) {\n      await this.connect();\n    }\n\n    // Wait for reconnection if in progress\n    const maxWaitTime = 30000; // 30 seconds\n    const startTime = Date.now();\n    while (this.reconnecting && Date.now() - startTime < maxWaitTime) {\n      await new Promise((resolve) => setTimeout(resolve, 100));\n    }\n\n    if (!this.isConnected) {\n      throw new Error(\"MongoDB is not connected and reconnection failed\");\n    }\n  }\n\n  /**\n   * Get the operation timeout in milliseconds\n   */\n  protected getOperationTimeout(): number {\n    return this.operationTimeoutMS;\n  }\n\n  /**\n   * Execute a database operation with automatic retry on connection errors\n   * Note: Use mongoose's maxTimeMS option in queries for timeout control\n   */\n  protected async executeWithRetry<T>(\n    operation: () => Promise<T>,\n    operationName: string = \"Database operation\"\n  ): Promise<T> {\n    await this.ensureConnection();\n\n    try {\n      return await operation();\n    } catch (error: any) {\n      // Check if it's a timeout error from MongoDB\n      if (error.code === 50 || error.message?.includes(\"operation exceeded time limit\")) {\n        loggers.agent.error(`${operationName} exceeded time limit`);\n        throw error;\n      }\n\n      // Check if it's a connection-related error\n      if (\n        error.name === \"MongoNetworkError\" ||\n        error.name === \"MongoServerError\" ||\n        error.message?.includes(\"connection\") ||\n        error.message?.includes(\"disconnect\")\n      ) {\n        loggers.agent.warn(\n          `${operationName} failed due to connection issue, attempting reconnection...`\n        );\n\n        await this.ensureConnection();\n\n        // Retry the operation once after reconnection\n        try {\n          return await operation();\n        } catch (retryError: any) {\n          loggers.agent.error(`${operationName} failed after retry:`, retryError);\n          throw retryError;\n        }\n      }\n\n      // If it's not a connection error, just throw it\n      throw error;\n    }\n  }\n}\n","import { type Document, Schema } from \"mongoose\";\nimport mongoose from \"mongoose\";\n\nexport const AgentObjectSchema = new Schema(\n\t{\n\t\tagent_prompt: {\n\t\t\ttype: String,\n\t\t},\n\t},\n);\n\nexport interface AgentDocument extends Document {\n\tagent_prompt: string;\n}\n\nexport const AgentModel = mongoose.model<AgentDocument>(\"Agent\", AgentObjectSchema);\n","import { IAgentMemory } from \"@ainetwork/adk/modules\";\nimport { AgentModel } from \"../models/agent.model\";\n\nexport type ExecuteWithRetryFn = <T>(\n  operation: () => Promise<T>,\n  operationName?: string\n) => Promise<T>;\n\nexport type GetOperationTimeoutFn = () => number;\n\ntype AgentMetadata = {\n  agent_prompt: string;\n}\n\nexport class MongoDBAgent implements IAgentMemory {\n  private executeWithRetry: ExecuteWithRetryFn;\n  private getOperationTimeout: GetOperationTimeoutFn;\n\n  constructor(\n    executeWithRetry: ExecuteWithRetryFn,\n    getOperationTimeout: GetOperationTimeoutFn\n  ) {\n    this.executeWithRetry = executeWithRetry;\n    this.getOperationTimeout = getOperationTimeout;\n  }\n\n  public async getAgentPrompt(): Promise<string> {\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      const metadata = await AgentModel.findOne({\n        id: \"agent_metadata\"\n      }).maxTimeMS(timeout)\n        .lean<AgentMetadata>();\n      return metadata?.agent_prompt || \"\";\n    }, \"getAgentPrompt()\");\n  };\n  \n  public async updateAgentPrompt(prompt: string): Promise<void> {\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      await AgentModel.updateOne({\n        id: \"agent_metadata\",\n      }, { \"agent_prompt\": prompt }).maxTimeMS(timeout);\n    }, \"updateAgentPrompt()\");\n  };\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 { IntentModel } from \"../models/intent.model\";\n\nexport type ExecuteWithRetryFn = <T>(\n  operation: () => Promise<T>,\n  operationName?: string\n) => Promise<T>;\n\nexport type GetOperationTimeoutFn = () => number;\n\nexport class MongoDBIntent implements IIntentMemory {\n  private executeWithRetry: ExecuteWithRetryFn;\n  private getOperationTimeout: GetOperationTimeoutFn;\n\n  constructor(\n    executeWithRetry: ExecuteWithRetryFn,\n    getOperationTimeout: GetOperationTimeoutFn\n  ) {\n    this.executeWithRetry = executeWithRetry;\n    this.getOperationTimeout = getOperationTimeout;\n  }\n\n  public async getIntent(intentId: string): Promise<Intent | undefined> {\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      const intent = await IntentModel.findOne({ id: intentId })\n        .maxTimeMS(timeout)\n        .lean<Intent>();\n      return intent || undefined;\n    }, `getIntent(${intentId})`);\n  };\n\n  public async getIntentByName(intentName: string): Promise<Intent | undefined> {\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      const intent = await IntentModel.findOne({ name: intentName })\n        .maxTimeMS(timeout)\n        .lean<Intent>();\n      return intent || undefined;\n    }, `getIntentByName(${intentName})`);\n  }\n\n  public async saveIntent(intent: Intent): Promise<void> {\n    return this.executeWithRetry(async () => {\n      await IntentModel.create(intent);\n    }, `saveIntent(${intent.id})`);\n  };\n\n  public async updateIntent(intentId: string, intent: Intent): Promise<void> {\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      await IntentModel.updateOne({\n        id: intentId,\n      }, intent).maxTimeMS(timeout);\n    }, `updateIntent(${intentId})`);\n  };\n\n  public async deleteIntent(intentId: string): Promise<void> {\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      await IntentModel.deleteOne({ id: intentId }).maxTimeMS(timeout);\n    }, `deleteIntent(${intentId})`);\n  };\n\n  public async listIntents(): Promise<Intent[]> {\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      const intents = await IntentModel.find()\n        .maxTimeMS(timeout)\n        .lean<Intent[]>();\n      return intents;\n    }, `listIntents()`);\n  };\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 { ThreadDocument, ThreadModel } from \"../models/threads.model\";\nimport { MessageDocument, MessageModel } from \"../models/messages.model\";\nimport { loggers } from \"@ainetwork/adk/utils/logger\";\n\nexport type ExecuteWithRetryFn = <T>(\n  operation: () => Promise<T>,\n  operationName?: string\n) => Promise<T>;\n\nexport type GetOperationTimeoutFn = () => number;\n\nexport class MongoDBThread implements IThreadMemory {\n  private executeWithRetry: ExecuteWithRetryFn;\n  private getOperationTimeout: GetOperationTimeoutFn;\n\n  constructor(\n    executeWithRetry: ExecuteWithRetryFn,\n    getOperationTimeout: GetOperationTimeoutFn\n  ) {\n    this.executeWithRetry = executeWithRetry;\n    this.getOperationTimeout = getOperationTimeout;\n  }\n\n  public async getThread(\n    userId: string,\n    threadId: string\n  ): Promise<ThreadObject | undefined> {\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      const thread = await ThreadModel.findOne({ threadId, userId }).maxTimeMS(timeout);\n      const messages = await MessageModel.find({ threadId, userId })\n        .sort({ timestamp: 1 })\n        .maxTimeMS(timeout);\n\n      if (!thread) return undefined;\n\n      loggers.agent.debug(`Found ${messages.length} messages for thread ${threadId}`);\n\n      const 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      messages.forEach((message: MessageDocument) => {\n        threadObject.messages.push({\n          messageId: message.messageId,\n          role: message.role as MessageRole,\n          content: message.content,\n          timestamp: message.timestamp,\n          metadata: message.metadata,\n        });\n      });\n\n      return threadObject;\n    }, `getThread(${userId}, ${threadId})`);\n  };\n\n  public async createThread(\n    type: ThreadType,\n    userId: string,\n    threadId: string,\n    title: string,\n  ): Promise<ThreadObject> {\n    return this.executeWithRetry(async () => {\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    }, `createThread(${userId}, ${threadId})`);\n  };\n\n  public async addMessagesToThread(\n    userId: string,\n    threadId: string,\n    messages: MessageObject[]\n  ): Promise<void> {\n    return this.executeWithRetry(async () => {\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    }, `addMessagesToThread(${userId}, ${threadId})`);\n  };\n\n  public async deleteThread(userId: string, threadId: string): Promise<void> {\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      const messages = await MessageModel.find({ userId, threadId })\n        .sort({ timestamp: 1 })\n        .maxTimeMS(timeout);\n\n      messages?.forEach((message: MessageDocument) => {\n        message.deleteOne();\n      });\n\n      const thread = await ThreadModel.findOne({ userId, threadId }).maxTimeMS(timeout);\n      thread?.deleteOne();\n    }, `deleteThread(${userId}, ${threadId})`);\n  };\n\n  public async listThreads(userId: string): Promise<ThreadMetadata[]> {\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      const threads = await ThreadModel.find({ userId })\n        .sort({ updated_at: -1 })\n        .maxTimeMS(timeout);\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    }, `listThreads(${userId})`);\n  };\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACCA,IAAAA,mBAAqB;AACrB,IAAAC,iBAAwB;;;ACFxB,sBAAsC;AACtC,IAAAC,mBAAqB;AAEd,IAAM,oBAAoB,IAAI;AAAA,EACpC;AAAA,IACC,cAAc;AAAA,MACb,MAAM;AAAA,IACP;AAAA,EACD;AACD;AAMO,IAAM,aAAa,iBAAAC,QAAS,MAAqB,SAAS,iBAAiB;;;ACD3E,IAAM,eAAN,MAA2C;AAAA,EACxC;AAAA,EACA;AAAA,EAER,YACE,kBACA,qBACA;AACA,SAAK,mBAAmB;AACxB,SAAK,sBAAsB;AAAA,EAC7B;AAAA,EAEA,MAAa,iBAAkC;AAC7C,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,UAAU,KAAK,oBAAoB;AACzC,YAAM,WAAW,MAAM,WAAW,QAAQ;AAAA,QACxC,IAAI;AAAA,MACN,CAAC,EAAE,UAAU,OAAO,EACjB,KAAoB;AACvB,aAAO,UAAU,gBAAgB;AAAA,IACnC,GAAG,kBAAkB;AAAA,EACvB;AAAA,EAEA,MAAa,kBAAkB,QAA+B;AAC5D,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,UAAU,KAAK,oBAAoB;AACzC,YAAM,WAAW,UAAU;AAAA,QACzB,IAAI;AAAA,MACN,GAAG,EAAE,gBAAgB,OAAO,CAAC,EAAE,UAAU,OAAO;AAAA,IAClD,GAAG,qBAAqB;AAAA,EAC1B;AACF;;;AC5CA,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;;;AChC/E,IAAM,gBAAN,MAA6C;AAAA,EAC1C;AAAA,EACA;AAAA,EAER,YACE,kBACA,qBACA;AACA,SAAK,mBAAmB;AACxB,SAAK,sBAAsB;AAAA,EAC7B;AAAA,EAEA,MAAa,UAAU,UAA+C;AACpE,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,UAAU,KAAK,oBAAoB;AACzC,YAAM,SAAS,MAAM,YAAY,QAAQ,EAAE,IAAI,SAAS,CAAC,EACtD,UAAU,OAAO,EACjB,KAAa;AAChB,aAAO,UAAU;AAAA,IACnB,GAAG,aAAa,QAAQ,GAAG;AAAA,EAC7B;AAAA,EAEA,MAAa,gBAAgB,YAAiD;AAC5E,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,UAAU,KAAK,oBAAoB;AACzC,YAAM,SAAS,MAAM,YAAY,QAAQ,EAAE,MAAM,WAAW,CAAC,EAC1D,UAAU,OAAO,EACjB,KAAa;AAChB,aAAO,UAAU;AAAA,IACnB,GAAG,mBAAmB,UAAU,GAAG;AAAA,EACrC;AAAA,EAEA,MAAa,WAAW,QAA+B;AACrD,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,YAAY,OAAO,MAAM;AAAA,IACjC,GAAG,cAAc,OAAO,EAAE,GAAG;AAAA,EAC/B;AAAA,EAEA,MAAa,aAAa,UAAkB,QAA+B;AACzE,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,UAAU,KAAK,oBAAoB;AACzC,YAAM,YAAY,UAAU;AAAA,QAC1B,IAAI;AAAA,MACN,GAAG,MAAM,EAAE,UAAU,OAAO;AAAA,IAC9B,GAAG,gBAAgB,QAAQ,GAAG;AAAA,EAChC;AAAA,EAEA,MAAa,aAAa,UAAiC;AACzD,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,UAAU,KAAK,oBAAoB;AACzC,YAAM,YAAY,UAAU,EAAE,IAAI,SAAS,CAAC,EAAE,UAAU,OAAO;AAAA,IACjE,GAAG,gBAAgB,QAAQ,GAAG;AAAA,EAChC;AAAA,EAEA,MAAa,cAAiC;AAC5C,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,UAAU,KAAK,oBAAoB;AACzC,YAAM,UAAU,MAAM,YAAY,KAAK,EACpC,UAAU,OAAO,EACjB,KAAe;AAClB,aAAO;AAAA,IACT,GAAG,eAAe;AAAA,EACpB;AACF;;;AC1EA,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;;;AC5D1F,oBAAwB;AASjB,IAAM,gBAAN,MAA6C;AAAA,EAC1C;AAAA,EACA;AAAA,EAER,YACE,kBACA,qBACA;AACA,SAAK,mBAAmB;AACxB,SAAK,sBAAsB;AAAA,EAC7B;AAAA,EAEA,MAAa,UACX,QACA,UACmC;AACnC,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,UAAU,KAAK,oBAAoB;AACzC,YAAM,SAAS,MAAM,YAAY,QAAQ,EAAE,UAAU,OAAO,CAAC,EAAE,UAAU,OAAO;AAChF,YAAM,WAAW,MAAM,aAAa,KAAK,EAAE,UAAU,OAAO,CAAC,EAC1D,KAAK,EAAE,WAAW,EAAE,CAAC,EACrB,UAAU,OAAO;AAEpB,UAAI,CAAC,OAAQ,QAAO;AAEpB,4BAAQ,MAAM,MAAM,SAAS,SAAS,MAAM,wBAAwB,QAAQ,EAAE;AAE9E,YAAM,eAA6B;AAAA,QACjC,UAAU,OAAO;AAAA,QACjB,QAAQ,OAAO;AAAA,QACf,MAAM,OAAO;AAAA,QACb,OAAO,OAAO,SAAS;AAAA,QACvB,UAAU,CAAC;AAAA,MACb;AACA,eAAS,QAAQ,CAAC,YAA6B;AAC7C,qBAAa,SAAS,KAAK;AAAA,UACzB,WAAW,QAAQ;AAAA,UACnB,MAAM,QAAQ;AAAA,UACd,SAAS,QAAQ;AAAA,UACjB,WAAW,QAAQ;AAAA,UACnB,UAAU,QAAQ;AAAA,QACpB,CAAC;AAAA,MACH,CAAC;AAED,aAAO;AAAA,IACT,GAAG,aAAa,MAAM,KAAK,QAAQ,GAAG;AAAA,EACxC;AAAA,EAEA,MAAa,aACX,MACA,QACA,UACA,OACuB;AACvB,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,MAAM,KAAK,IAAI;AACrB,YAAM,YAAY,OAAO;AAAA,QACvB;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA,YAAY;AAAA,QACZ,YAAY;AAAA,MACd,CAAC;AAED,aAAO,EAAE,MAAM,QAAQ,UAAU,OAAO,UAAU,CAAC,EAAC;AAAA,IACtD,GAAG,gBAAgB,MAAM,KAAK,QAAQ,GAAG;AAAA,EAC3C;AAAA,EAEA,MAAa,oBACX,QACA,UACA,UACe;AACf,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,YAAY,UAAU,EAAE,UAAU,OAAO,GAAG;AAAA,QAChD,YAAY,KAAK,IAAI;AAAA,MACvB,CAAC;AACD,iBAAW,WAAW,UAAU;AAC9B,cAAM,aAAa,OAAO;AAAA,UACxB;AAAA,UACA,WAAW,QAAQ;AAAA,UACnB;AAAA,UACA,MAAM,QAAQ;AAAA,UACd,SAAS,QAAQ;AAAA,UACjB,WAAW,QAAQ;AAAA,UACnB,UAAU,QAAQ;AAAA,QACpB,CAAC;AAAA,MACH;AAAA,IACF,GAAG,uBAAuB,MAAM,KAAK,QAAQ,GAAG;AAAA,EAClD;AAAA,EAEA,MAAa,aAAa,QAAgB,UAAiC;AACzE,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,UAAU,KAAK,oBAAoB;AACzC,YAAM,WAAW,MAAM,aAAa,KAAK,EAAE,QAAQ,SAAS,CAAC,EAC1D,KAAK,EAAE,WAAW,EAAE,CAAC,EACrB,UAAU,OAAO;AAEpB,gBAAU,QAAQ,CAAC,YAA6B;AAC9C,gBAAQ,UAAU;AAAA,MACpB,CAAC;AAED,YAAM,SAAS,MAAM,YAAY,QAAQ,EAAE,QAAQ,SAAS,CAAC,EAAE,UAAU,OAAO;AAChF,cAAQ,UAAU;AAAA,IACpB,GAAG,gBAAgB,MAAM,KAAK,QAAQ,GAAG;AAAA,EAC3C;AAAA,EAEA,MAAa,YAAY,QAA2C;AAClE,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,UAAU,KAAK,oBAAoB;AACzC,YAAM,UAAU,MAAM,YAAY,KAAK,EAAE,OAAO,CAAC,EAC9C,KAAK,EAAE,YAAY,GAAG,CAAC,EACvB,UAAU,OAAO;AACpB,YAAM,OAAyB,QAAQ,IAAI,CAAC,WAA2B;AACrE,eAAO;AAAA,UACL,MAAM,OAAO;AAAA,UACb;AAAA,UACA,UAAU,OAAO;AAAA,UACjB,OAAO,OAAO;AAAA,UACd,WAAW,OAAO;AAAA,QACpB;AAAA,MACF,CAAC;AACD,aAAO;AAAA,IACT,GAAG,eAAe,MAAM,GAAG;AAAA,EAC7B;AACF;;;AP1HO,IAAM,gBAAN,MAAM,eAAiC;AAAA,EAC5C,OAAe;AAAA,EACP;AAAA,EACA,YAAqB;AAAA,EACrB,oBAA4B;AAAA,EAC5B;AAAA,EACA;AAAA,EACA,eAAwB;AAAA,EACxB;AAAA,EACA,sBAA+B;AAAA,EAC/B;AAAA,EAEA;AAAA,EACA;AAAA,EACA;AAAA,EAER,YAAY,QAAsC;AAChD,UAAM,MAAM,OAAO,WAAW,WAAW,EAAE,KAAK,OAAO,IAAI;AAE3D,SAAK,MAAM,IAAI;AACf,SAAK,uBAAuB,IAAI,wBAAwB;AACxD,SAAK,oBAAoB,IAAI,qBAAqB;AAClD,SAAK,qBAAqB,IAAI,sBAAsB;AACpD,SAAK,mBAAmB;AAAA,MACtB,aAAa,IAAI,eAAe;AAAA,MAChC,0BAA0B,IAAI,4BAA4B;AAAA,MAC1D,iBAAiB,IAAI,mBAAmB;AAAA,MACxC,kBAAkB,IAAI,oBAAoB;AAAA,MAC1C,gBAAgB;AAAA,IAClB;AAEA,QAAI,CAAC,eAAc,UAAU;AAC3B,qBAAc,WAAW;AACzB,WAAK,4BAA4B;AAAA,IACnC,OAAO;AAEL,WAAK,YAAY,eAAc,SAAS;AACxC,WAAK,qBAAqB,eAAc,SAAS;AAAA,IACnD;AAEF,SAAK,cAAc,IAAI;AAAA,MACtB,KAAK,iBAAiB,KAAK,IAAI;AAAA,MAC/B,KAAK,oBAAoB,KAAK,IAAI;AAAA,IACnC;AAEA,SAAK,eAAe,IAAI;AAAA,MACvB,KAAK,iBAAiB,KAAK,IAAI;AAAA,MAC/B,KAAK,oBAAoB,KAAK,IAAI;AAAA,IACnC;AAEA,SAAK,eAAe,IAAI;AAAA,MACvB,KAAK,iBAAiB,KAAK,IAAI;AAAA,MAC/B,KAAK,oBAAoB,KAAK,IAAI;AAAA,IACnC;AAAA,EACA;AAAA,EAEO,iBAA+B;AACpC,WAAO,KAAK;AAAA,EACd;AAAA,EAEO,kBAAiC;AACtC,WAAO,KAAK;AAAA,EACd;AAAA,EAEO,kBAAiC;AACtC,WAAO,KAAK;AAAA,EACd;AAAA,EAEQ,8BAAoC;AAC1C,QAAI,KAAK,oBAAqB;AAE9B,SAAK,sBAAsB;AAE3B,qBAAAC,QAAS,WAAW,GAAG,aAAa,MAAM;AACxC,WAAK,YAAY;AACjB,WAAK,oBAAoB;AACzB,WAAK,eAAe;AACpB,6BAAQ,MAAM,KAAK,gCAAgC;AAAA,IACrD,CAAC;AAED,qBAAAA,QAAS,WAAW,GAAG,gBAAgB,MAAM;AAC3C,WAAK,YAAY;AACjB,6BAAQ,MAAM,KAAK,sBAAsB;AACzC,WAAK,oBAAoB;AAAA,IAC3B,CAAC;AAED,qBAAAA,QAAS,WAAW,GAAG,SAAS,CAAC,UAAU;AACzC,WAAK,YAAY;AACjB,6BAAQ,MAAM,MAAM,6BAA6B,KAAK;AACtD,WAAK,oBAAoB;AAAA,IAC3B,CAAC;AAED,qBAAAA,QAAS,WAAW,GAAG,eAAe,MAAM;AAC1C,WAAK,YAAY;AACjB,WAAK,oBAAoB;AACzB,WAAK,eAAe;AACpB,6BAAQ,MAAM,KAAK,kCAAkC;AAAA,IACvD,CAAC;AAAA,EACH;AAAA,EAEA,MAAc,sBAAqC;AACjD,QAAI,KAAK,cAAc;AACrB;AAAA,IACF;AAEA,SAAK,eAAe;AAEpB,WAAO,KAAK,oBAAoB,KAAK,wBAAwB,CAAC,KAAK,aAAa;AAC9E,WAAK;AACL,6BAAQ,MAAM;AAAA,QACZ,uCAAuC,KAAK,iBAAiB,IAAI,KAAK,oBAAoB;AAAA,MAC5F;AAEA,UAAI;AACF,cAAM,iBAAAA,QAAS,QAAQ,KAAK,KAAK,KAAK,gBAAgB;AACtD,aAAK,YAAY;AACjB,aAAK,oBAAoB;AACzB,aAAK,eAAe;AACpB,+BAAQ,MAAM,KAAK,iCAAiC;AACpD;AAAA,MACF,SAAS,OAAO;AACd,+BAAQ,MAAM;AAAA,UACZ,wBAAwB,KAAK,iBAAiB;AAAA,UAC9C;AAAA,QACF;AAEA,YAAI,KAAK,oBAAoB,KAAK,sBAAsB;AACtD,gBAAM,IAAI;AAAA,YAAQ,CAAC,YACjB,WAAW,SAAS,KAAK,iBAAiB;AAAA,UAC5C;AAAA,QACF;AAAA,MACF;AAAA,IACF;AAEA,SAAK,eAAe;AAEpB,QAAI,CAAC,KAAK,aAAa;AACrB,6BAAQ,MAAM;AAAA,QACZ,wCAAwC,KAAK,oBAAoB;AAAA,MACnE;AAAA,IACF;AAAA,EACF;AAAA,EAEA,MAAa,UAAyB;AACpC,QAAI,KAAK,WAAW;AAClB;AAAA,IACF;AAEA,QAAI;AACF,YAAM,iBAAAA,QAAS,QAAQ,KAAK,KAAK,KAAK,gBAAgB;AACtD,WAAK,YAAY;AACjB,WAAK,oBAAoB;AAAA,IAC3B,SAAS,OAAO;AACd,6BAAQ,MAAM,MAAM,iCAAiC,KAAK;AAC1D,YAAM;AAAA,IACR;AAAA,EACF;AAAA,EAEA,MAAa,aAA4B;AACvC,QAAI,CAAC,KAAK,aAAa;AACrB;AAAA,IACF;AAEA,QAAI;AACF,YAAM,iBAAAA,QAAS,WAAW;AAC1B,WAAK,YAAY;AAAA,IACnB,SAAS,OAAO;AACd,6BAAQ,MAAM,MAAM,sCAAsC,KAAK;AAC/D,YAAM;AAAA,IACR;AAAA,EACF;AAAA,EAEO,cAAuB;AAC5B,WAAO,KAAK;AAAA,EACd;AAAA,EAEA,MAAc,mBAAkC;AAC9C,QAAI,CAAC,KAAK,eAAe,CAAC,KAAK,cAAc;AAC3C,YAAM,KAAK,QAAQ;AAAA,IACrB;AAGA,UAAM,cAAc;AACpB,UAAM,YAAY,KAAK,IAAI;AAC3B,WAAO,KAAK,gBAAgB,KAAK,IAAI,IAAI,YAAY,aAAa;AAChE,YAAM,IAAI,QAAQ,CAAC,YAAY,WAAW,SAAS,GAAG,CAAC;AAAA,IACzD;AAEA,QAAI,CAAC,KAAK,aAAa;AACrB,YAAM,IAAI,MAAM,kDAAkD;AAAA,IACpE;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKU,sBAA8B;AACtC,WAAO,KAAK;AAAA,EACd;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,MAAgB,iBACd,WACA,gBAAwB,sBACZ;AACZ,UAAM,KAAK,iBAAiB;AAE5B,QAAI;AACF,aAAO,MAAM,UAAU;AAAA,IACzB,SAAS,OAAY;AAEnB,UAAI,MAAM,SAAS,MAAM,MAAM,SAAS,SAAS,+BAA+B,GAAG;AACjF,+BAAQ,MAAM,MAAM,GAAG,aAAa,sBAAsB;AAC1D,cAAM;AAAA,MACR;AAGA,UACE,MAAM,SAAS,uBACf,MAAM,SAAS,sBACf,MAAM,SAAS,SAAS,YAAY,KACpC,MAAM,SAAS,SAAS,YAAY,GACpC;AACA,+BAAQ,MAAM;AAAA,UACZ,GAAG,aAAa;AAAA,QAClB;AAEA,cAAM,KAAK,iBAAiB;AAG5B,YAAI;AACF,iBAAO,MAAM,UAAU;AAAA,QACzB,SAAS,YAAiB;AACxB,iCAAQ,MAAM,MAAM,GAAG,aAAa,wBAAwB,UAAU;AACtE,gBAAM;AAAA,QACR;AAAA,MACF;AAGA,YAAM;AAAA,IACR;AAAA,EACF;AACF;","names":["import_mongoose","import_logger","import_mongoose","mongoose","import_mongoose","mongoose","import_mongoose","mongoose","import_memory","import_mongoose","mongoose","mongoose"]}

package/dist/index.d.cts

@@ -1,6 +1,4 @@
-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';
+import { IMemory, IAgentMemory, IThreadMemory, IIntentMemory } from 'node_modules/@ainetwork/adk/dist/esm/modules/memory/base.memory';
 
 interface MongoDBMemoryConfig {
     uri: string;
@@ -23,7 +21,13 @@ declare class MongoDBMemory implements IMemory {
     private connectionConfig;
     private eventListenersSetup;
     private operationTimeoutMS;
+    private agentMemory;
+    private intentMemory;
+    private threadMemory;
     constructor(config: string | MongoDBMemoryConfig);
+    getAgentMemory(): IAgentMemory;
+    getThreadMemory(): IThreadMemory;
+    getIntentMemory(): IIntentMemory;
     private setupMongooseEventListeners;
     private handleDisconnection;
     connect(): Promise<void>;
@@ -41,22 +45,4 @@ declare class MongoDBMemory implements IMemory {
     protected executeWithRetry<T>(operation: () => Promise<T>, operationName?: string): Promise<T>;
 }
 
-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<ThreadObject>;
-    addMessagesToThread(userId: string, threadId: string, messages: MessageObject[]): Promise<void>;
-    deleteThread(userId: string, threadId: string): Promise<void>;
-    listThreads(userId: string): Promise<ThreadMetadata[]>;
-}
-
-declare class MongoDBIntent extends MongoDBMemory implements IIntentMemory {
-    getIntent(intentId: string): Promise<Intent | undefined>;
-    getIntentByName(intentName: string): Promise<Intent | undefined>;
-    saveIntent(intent: Intent): Promise<void>;
-    updateIntent(intentId: string, intent: Intent): Promise<void>;
-    deleteIntent(intentId: string): Promise<void>;
-    listIntents(): Promise<Intent[]>;
-}
-
-export { MongoDBIntent, MongoDBMemory, type MongoDBMemoryConfig, MongoDBThread };
+export { MongoDBMemory, type MongoDBMemoryConfig };

package/dist/index.d.ts

@@ -1,6 +1,4 @@
-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';
+import { IMemory, IAgentMemory, IThreadMemory, IIntentMemory } from 'node_modules/@ainetwork/adk/dist/esm/modules/memory/base.memory';
 
 interface MongoDBMemoryConfig {
     uri: string;
@@ -23,7 +21,13 @@ declare class MongoDBMemory implements IMemory {
     private connectionConfig;
     private eventListenersSetup;
     private operationTimeoutMS;
+    private agentMemory;
+    private intentMemory;
+    private threadMemory;
     constructor(config: string | MongoDBMemoryConfig);
+    getAgentMemory(): IAgentMemory;
+    getThreadMemory(): IThreadMemory;
+    getIntentMemory(): IIntentMemory;
     private setupMongooseEventListeners;
     private handleDisconnection;
     connect(): Promise<void>;
@@ -41,22 +45,4 @@ declare class MongoDBMemory implements IMemory {
     protected executeWithRetry<T>(operation: () => Promise<T>, operationName?: string): Promise<T>;
 }
 
-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<ThreadObject>;
-    addMessagesToThread(userId: string, threadId: string, messages: MessageObject[]): Promise<void>;
-    deleteThread(userId: string, threadId: string): Promise<void>;
-    listThreads(userId: string): Promise<ThreadMetadata[]>;
-}
-
-declare class MongoDBIntent extends MongoDBMemory implements IIntentMemory {
-    getIntent(intentId: string): Promise<Intent | undefined>;
-    getIntentByName(intentName: string): Promise<Intent | undefined>;
-    saveIntent(intent: Intent): Promise<void>;
-    updateIntent(intentId: string, intent: Intent): Promise<void>;
-    deleteIntent(intentId: string): Promise<void>;
-    listIntents(): Promise<Intent[]>;
-}
-
-export { MongoDBIntent, MongoDBMemory, type MongoDBMemoryConfig, MongoDBThread };
+export { MongoDBMemory, type MongoDBMemoryConfig };