@ainetwork/adk-provider-memory-mongodb 0.4.1 → 0.5.0

package/dist/index.js

@@ -6,13 +6,16 @@ import {
 } from "./chunk-YFW7JXII.js";
 import {
   MessageModel
-} from "./chunk-T5WRFXBY.js";
+} from "./chunk-QULDFKGZ.js";
 import {
   ThreadModel
-} from "./chunk-EPJZT2YE.js";
+} from "./chunk-2XJ6S2W5.js";
 import {
-  WorkflowModel
-} from "./chunk-ENGJ6NTO.js";
+  UserWorkflowModel
+} from "./chunk-RC275GLE.js";
+import {
+  WorkflowTemplateModel
+} from "./chunk-SJ2FHHN6.js";
 
 // implements/base.memory.ts
 import mongoose from "mongoose";
@@ -180,6 +183,7 @@ var MongoDBThread = class {
         type: thread.type,
         title: thread.title || "New thread",
         isPinned: thread.isPinned ?? false,
+        workflowId: thread.workflowId,
         messages: []
       };
       messages.forEach((message) => {
@@ -194,31 +198,39 @@ var MongoDBThread = class {
       return threadObject;
     }, `getThread(${userId}, ${threadId})`);
   }
-  async createThread(type, userId, threadId, title) {
+  async createThread(type, userId, threadId, title, workflowId) {
     return this.executeWithRetry(async () => {
       await ThreadModel.create({
         type,
         userId,
         threadId,
-        title
+        title,
+        workflowId
       });
-      return { type, userId, threadId, title, messages: [] };
+      return { type, userId, threadId, title, workflowId, messages: [] };
     }, `createThread(${userId}, ${threadId})`);
   }
   async addMessagesToThread(userId, threadId, messages) {
     return this.executeWithRetry(async () => {
-      await ThreadModel.updateOne({ threadId, userId }, { $set: {} });
-      for (const message of messages) {
-        await MessageModel.create({
-          threadId,
-          messageId: message.messageId,
-          userId,
-          role: message.role,
-          content: message.content,
-          timestamp: message.timestamp,
-          metadata: message.metadata
-        });
+      if (messages.length > 0) {
+        const messageIds = messages.map((m) => m.messageId);
+        await MessageModel.deleteMany({ threadId, userId, messageId: { $in: messageIds } });
+        await MessageModel.insertMany(
+          messages.map((message) => ({
+            threadId,
+            messageId: message.messageId,
+            userId,
+            role: message.role,
+            content: message.content,
+            timestamp: message.timestamp,
+            metadata: message.metadata
+          }))
+        );
       }
+      await ThreadModel.updateOne(
+        { threadId, userId },
+        { $set: { updatedAt: /* @__PURE__ */ new Date() } }
+      );
     }, `addMessagesToThread(${userId}, ${threadId})`);
   }
   async deleteThread(userId, threadId) {
@@ -228,10 +240,13 @@ var MongoDBThread = class {
       await ThreadModel.deleteOne({ userId, threadId }).maxTimeMS(timeout);
     }, `deleteThread(${userId}, ${threadId})`);
   }
-  async listThreads(userId) {
+  async listThreads(userId, filter) {
     return this.executeWithRetry(async () => {
       const timeout = this.getOperationTimeout();
-      const threads = await ThreadModel.find({ userId }).sort({ updatedAt: -1 }).maxTimeMS(timeout);
+      const query = { userId };
+      if (filter?.workflowId) query.workflowId = filter.workflowId;
+      if (filter?.type) query.type = filter.type;
+      const threads = await ThreadModel.find(query).sort({ updatedAt: -1 }).maxTimeMS(timeout);
       const data = threads.map((thread) => {
         return {
           type: thread.type,
@@ -239,7 +254,9 @@ var MongoDBThread = class {
           threadId: thread.threadId,
           title: thread.title,
           isPinned: thread.isPinned ?? false,
-          updatedAt: thread.updatedAt
+          workflowId: thread.workflowId,
+          createdAt: thread.createdAt?.toISOString(),
+          updatedAt: thread.updatedAt?.toISOString()
         };
       });
       return data;
@@ -256,55 +273,109 @@ var MongoDBThread = class {
   }
 };
 
-// implements/workflow.memory.ts
-var MongoDBWorkflow = class {
+// implements/user-workflow.memory.ts
+var MongoDBUserWorkflow = class {
   executeWithRetry;
   getOperationTimeout;
   constructor(executeWithRetry, getOperationTimeout) {
     this.executeWithRetry = executeWithRetry;
     this.getOperationTimeout = getOperationTimeout;
   }
-  async createWorkflow(workflow) {
+  async createUserWorkflow(workflow) {
     return this.executeWithRetry(async () => {
-      const timeout = this.getOperationTimeout();
-      const created = await WorkflowModel.create(workflow);
+      const created = await UserWorkflowModel.create(workflow);
       return created.toObject();
-    }, "createWorkflow()");
+    }, "createUserWorkflow()");
   }
-  async getWorkflow(workflowId) {
+  async getUserWorkflow(workflowId) {
     return this.executeWithRetry(async () => {
       const timeout = this.getOperationTimeout();
-      const workflow = await WorkflowModel.findOne({
-        workflowId
-      }).maxTimeMS(timeout).lean();
+      const workflow = await UserWorkflowModel.findOne({ workflowId }).maxTimeMS(timeout).lean();
       return workflow || void 0;
-    }, "getWorkflow()");
+    }, "getUserWorkflow()");
   }
-  async updateWorkflow(workflowId, updates) {
+  async updateUserWorkflow(workflowId, updates) {
     if (!updates.userId) {
-      throw new Error("userId is required for updateWorkflow");
+      throw new Error("userId is required for updateUserWorkflow");
     }
+    const { userId, workflowId: _workflowId, ...mutableUpdates } = updates;
     return this.executeWithRetry(async () => {
       const timeout = this.getOperationTimeout();
-      await WorkflowModel.updateOne(
-        { workflowId, userId: updates.userId },
-        { $set: updates }
+      await UserWorkflowModel.updateOne(
+        { workflowId, userId },
+        { $set: mutableUpdates }
       ).maxTimeMS(timeout);
-    }, "updateWorkflow()");
+    }, "updateUserWorkflow()");
+  }
+  async deleteUserWorkflow(workflowId, userId) {
+    return this.executeWithRetry(async () => {
+      const timeout = this.getOperationTimeout();
+      await UserWorkflowModel.deleteOne({ workflowId, userId }).maxTimeMS(timeout);
+    }, "deleteUserWorkflow()");
   }
-  async deleteWorkflow(workflowId, userId) {
+  async listUserWorkflows(userId) {
     return this.executeWithRetry(async () => {
       const timeout = this.getOperationTimeout();
-      await WorkflowModel.deleteOne({ workflowId, userId }).maxTimeMS(timeout);
-    }, "deleteWorkflow()");
+      const query = userId ? { userId } : {};
+      const workflows = await UserWorkflowModel.find(query).maxTimeMS(timeout).lean();
+      return workflows;
+    }, "listUserWorkflows()");
   }
-  async listWorkflows(userId) {
+  async listActiveScheduledWorkflows() {
     return this.executeWithRetry(async () => {
       const timeout = this.getOperationTimeout();
-      const query = userId ? { $or: [{ userId }, { userId: { $exists: false } }, { userId: null }] } : { $or: [{ userId: { $exists: false } }, { userId: null }] };
-      const workflows = await WorkflowModel.find(query).maxTimeMS(timeout).lean();
+      const workflows = await UserWorkflowModel.find({
+        active: true,
+        schedule: { $exists: true, $ne: null }
+      }).maxTimeMS(timeout).lean();
       return workflows;
-    }, "listWorkflows()");
+    }, "listActiveScheduledWorkflows()");
+  }
+};
+
+// implements/workflow-template.memory.ts
+var MongoDBWorkflowTemplate = class {
+  executeWithRetry;
+  getOperationTimeout;
+  constructor(executeWithRetry, getOperationTimeout) {
+    this.executeWithRetry = executeWithRetry;
+    this.getOperationTimeout = getOperationTimeout;
+  }
+  async createTemplate(template) {
+    return this.executeWithRetry(async () => {
+      const created = await WorkflowTemplateModel.create(template);
+      return created.toObject();
+    }, "createTemplate()");
+  }
+  async getTemplate(templateId) {
+    return this.executeWithRetry(async () => {
+      const timeout = this.getOperationTimeout();
+      const template = await WorkflowTemplateModel.findOne({ templateId }).maxTimeMS(timeout).lean();
+      return template || void 0;
+    }, "getTemplate()");
+  }
+  async updateTemplate(templateId, updates) {
+    const { templateId: _templateId, ...mutableUpdates } = updates;
+    return this.executeWithRetry(async () => {
+      const timeout = this.getOperationTimeout();
+      await WorkflowTemplateModel.updateOne(
+        { templateId },
+        { $set: mutableUpdates }
+      ).maxTimeMS(timeout);
+    }, "updateTemplate()");
+  }
+  async deleteTemplate(templateId) {
+    return this.executeWithRetry(async () => {
+      const timeout = this.getOperationTimeout();
+      await WorkflowTemplateModel.deleteOne({ templateId }).maxTimeMS(timeout);
+    }, "deleteTemplate()");
+  }
+  async listTemplates() {
+    return this.executeWithRetry(async () => {
+      const timeout = this.getOperationTimeout();
+      const templates = await WorkflowTemplateModel.find().maxTimeMS(timeout).lean();
+      return templates;
+    }, "listTemplates()");
   }
 };
 
@@ -325,7 +396,8 @@ var MongoDBMemory = class _MongoDBMemory {
   agentMemory;
   intentMemory;
   threadMemory;
-  workflowMemory;
+  workflowTemplateMemory;
+  userWorkflowMemory;
   constructor(config) {
     const cfg = typeof config === "string" ? { uri: config } : config;
     this.uri = cfg.uri;
@@ -336,7 +408,9 @@ var MongoDBMemory = class _MongoDBMemory {
       this.threadTTLSeconds = cfg.threadTTLSeconds;
     }
     this.connectionConfig = {
-      maxPoolSize: cfg.maxPoolSize ?? 1,
+      maxPoolSize: cfg.maxPoolSize ?? 10,
+      minPoolSize: 0,
+      maxIdleTimeMS: 3e4,
       serverSelectionTimeoutMS: cfg.serverSelectionTimeoutMS ?? 3e4,
       socketTimeoutMS: cfg.socketTimeoutMS ?? 45e3,
       connectTimeoutMS: cfg.connectTimeoutMS ?? 3e4,
@@ -361,7 +435,11 @@ var MongoDBMemory = class _MongoDBMemory {
       this.executeWithRetry.bind(this),
       this.getOperationTimeout.bind(this)
     );
-    this.workflowMemory = new MongoDBWorkflow(
+    this.workflowTemplateMemory = new MongoDBWorkflowTemplate(
+      this.executeWithRetry.bind(this),
+      this.getOperationTimeout.bind(this)
+    );
+    this.userWorkflowMemory = new MongoDBUserWorkflow(
       this.executeWithRetry.bind(this),
       this.getOperationTimeout.bind(this)
     );
@@ -375,8 +453,11 @@ var MongoDBMemory = class _MongoDBMemory {
   getIntentMemory() {
     return this.intentMemory;
   }
-  getWorkflowMemory() {
-    return this.workflowMemory;
+  getWorkflowTemplateMemory() {
+    return this.workflowTemplateMemory;
+  }
+  getUserWorkflowMemory() {
+    return this.userWorkflowMemory;
   }
   setupMongooseEventListeners() {
     if (this.eventListenersSetup) return;
@@ -409,7 +490,7 @@ var MongoDBMemory = class _MongoDBMemory {
       return;
     }
     this.reconnecting = true;
-    while (this.reconnectAttempts < this.maxReconnectAttempts && !this.isConnected) {
+    while (this.reconnectAttempts < this.maxReconnectAttempts && !this.connected) {
       this.reconnectAttempts++;
       loggers2.agent.info(
         `Attempting to reconnect to MongoDB (${this.reconnectAttempts}/${this.maxReconnectAttempts})...`
@@ -434,7 +515,7 @@ var MongoDBMemory = class _MongoDBMemory {
       }
     }
     this.reconnecting = false;
-    if (!this.isConnected) {
+    if (!this.connected) {
       loggers2.agent.error(
         `Failed to reconnect to MongoDB after ${this.maxReconnectAttempts} attempts`
       );
@@ -456,7 +537,7 @@ var MongoDBMemory = class _MongoDBMemory {
     }
   }
   async disconnect() {
-    if (!this.isConnected) {
+    if (!this.connected) {
       return;
     }
     try {
@@ -475,7 +556,7 @@ var MongoDBMemory = class _MongoDBMemory {
     return this.connected;
   }
   async ensureConnection() {
-    if (!this.isConnected && !this.reconnecting) {
+    if (!this.connected && !this.reconnecting) {
       await this.connect();
     }
     const maxWaitTime = 3e4;
@@ -483,7 +564,7 @@ var MongoDBMemory = class _MongoDBMemory {
     while (this.reconnecting && Date.now() - startTime < maxWaitTime) {
       await new Promise((resolve) => setTimeout(resolve, 100));
     }
-    if (!this.isConnected) {
+    if (!this.connected) {
       throw new Error("MongoDB is not connected and reconnection failed");
     }
   }
@@ -562,6 +643,25 @@ var MongoDBMemory = class _MongoDBMemory {
         loggers2.agent.error(`${operationName} exceeded time limit`);
         throw error;
       }
+      if (error.code === 261 || error.codeName === "TooManyLogicalSessions") {
+        loggers2.agent.warn(
+          `${operationName} failed due to too many sessions, disconnecting to release sessions...`
+        );
+        try {
+          await mongoose.disconnect();
+          this.connected = false;
+        } catch (disconnectError) {
+          loggers2.agent.error("Failed to disconnect during session cleanup:", disconnectError);
+        }
+        await new Promise((resolve) => setTimeout(resolve, 5e3));
+        await this.ensureConnection();
+        try {
+          return await operation();
+        } catch (retryError) {
+          loggers2.agent.error(`${operationName} failed after session cleanup retry:`, retryError);
+          throw retryError;
+        }
+      }
       if (error.name === "MongoNetworkError" || error.name === "MongoServerError" || error.message?.includes("connection") || error.message?.includes("disconnect")) {
         loggers2.agent.warn(
           `${operationName} failed due to connection issue, attempting reconnection...`

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../implements/base.memory.ts","../implements/agent.memory.ts","../implements/intent.memory.ts","../implements/thread.memory.ts","../implements/workflow.memory.ts"],"sourcesContent":["import { IAgentMemory, IIntentMemory, IMemory, IThreadMemory, IWorkflowMemory } from \"@ainetwork/adk/modules\";\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\";\nimport { MongoDBWorkflow } from \"./workflow.memory\";\nimport { MessageModel } from \"../models/messages.model\";\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  threadTTLSeconds?: number; // TTL for thread documents (in seconds). Orphaned messages are periodically cleaned up.\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  private threadTTLSeconds?: number;\n  private orphanCleanupTimer?: ReturnType<typeof setInterval>;\n\n  private agentMemory: MongoDBAgent;\n  private intentMemory: MongoDBIntent;\n  private threadMemory: MongoDBThread;\n  private workflowMemory: MongoDBWorkflow;\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    if (cfg.threadTTLSeconds !== undefined && cfg.threadTTLSeconds > 0) {\n      this.threadTTLSeconds = cfg.threadTTLSeconds;\n    }\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\t\tthis.workflowMemory = new MongoDBWorkflow(\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  public getWorkflowMemory(): IWorkflowMemory {\n    return this.workflowMemory;\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      await this.setupTTLIndex();\n      this.startOrphanCleanup();\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      if (this.orphanCleanupTimer) {\n        clearInterval(this.orphanCleanupTimer);\n        this.orphanCleanupTimer = undefined;\n      }\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  private async setupTTLIndex(): Promise<void> {\n    if (this.threadTTLSeconds === undefined) return;\n\n    try {\n      const db = mongoose.connection.db;\n      if (!db) return;\n\n      const collection = db.collection('threads');\n      const indexes = await collection.indexes();\n      const existingTTL = indexes.find(\n        (idx) => idx.key?.updatedAt !== undefined && idx.expireAfterSeconds !== undefined\n      );\n\n      if (existingTTL) {\n        if (existingTTL.expireAfterSeconds !== this.threadTTLSeconds) {\n          await db.command({\n            collMod: 'threads',\n            index: { keyPattern: { updatedAt: 1 }, expireAfterSeconds: this.threadTTLSeconds },\n          });\n          loggers.agent.info(`Thread TTL index updated to ${this.threadTTLSeconds} seconds`);\n        }\n      } else {\n        await collection.createIndex(\n          { updatedAt: 1 },\n          { expireAfterSeconds: this.threadTTLSeconds },\n        );\n        loggers.agent.info(`Thread TTL index created with ${this.threadTTLSeconds} seconds`);\n      }\n    } catch (error) {\n      loggers.agent.error('Failed to setup TTL index:', error);\n    }\n  }\n\n  private startOrphanCleanup(): void {\n    if (this.threadTTLSeconds === undefined) return;\n    if (this.orphanCleanupTimer) return;\n\n    // Run cleanup at half the TTL interval, with a minimum of 60s and maximum of 1 hour\n    const intervalMs = Math.max(60_000, Math.min(this.threadTTLSeconds * 500, 3_600_000));\n    this.orphanCleanupTimer = setInterval(() => {\n      this.cleanupOrphanedMessages().catch((error) => {\n        loggers.agent.error('Orphaned message cleanup failed:', error);\n      });\n    }, intervalMs);\n\n    loggers.agent.info(`Orphaned message cleanup scheduled every ${Math.round(intervalMs / 1000)}s`);\n  }\n\n  private async cleanupOrphanedMessages(): Promise<void> {\n    if (!this.connected) return;\n\n    try {\n      const db = mongoose.connection.db;\n      if (!db) return;\n\n      const existingThreadIds = await db.collection('threads').distinct('threadId');\n      const result = await MessageModel.deleteMany({\n        threadId: { $nin: existingThreadIds },\n      });\n\n      if (result.deletedCount > 0) {\n        loggers.agent.info(`Cleaned up ${result.deletedCount} orphaned messages`);\n      }\n    } catch (error) {\n      loggers.agent.error('Failed to cleanup orphaned messages:', error);\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 { 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 PromptDocument = {\n  id: string;\n  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_prompt\"\n      }).maxTimeMS(timeout)\n        .lean<PromptDocument>();\n      return metadata?.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_prompt\",\n      }, { \"prompt\": prompt }, { upsert: true }).maxTimeMS(timeout);\n    }, \"updateAgentPrompt()\");\n  };\n\n  public async getAggregatePrompt(): Promise<string> {\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      const metadata = await AgentModel.findOne({\n        id: \"aggregate_prompt\"\n      }).maxTimeMS(timeout)\n        .lean<PromptDocument>();\n      return metadata?.prompt || \"\";\n    }, \"getAggregatePrompt()\");\n  };\n\n  public async getGenerateTitlePrompt(): Promise<string> {\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      const metadata = await AgentModel.findOne({\n        id: \"generate_title_prompt\"\n      }).maxTimeMS(timeout)\n        .lean<PromptDocument>();\n      return metadata?.prompt || \"\";\n    }, \"getGenerateTitlePrompt()\");\n  };\n\n  public async getSingleTriggerPrompt(): Promise<string> {\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      const metadata = await AgentModel.findOne({\n        id: \"single_trigger_prompt\"\n      }).maxTimeMS(timeout)\n        .lean<PromptDocument>();\n      return metadata?.prompt || \"\";\n    }, \"getSingleTriggerPrompt()\");\n  };\n\n  public async getMultiTriggerPrompt(): Promise<string> {\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      const metadata = await AgentModel.findOne({\n        id: \"multi_trigger_prompt\"\n      }).maxTimeMS(timeout)\n        .lean<PromptDocument>();\n      return metadata?.prompt || \"\";\n    }, \"getMultiTriggerPrompt()\");\n  };\n\n  public async getToolSelectPrompt(): Promise<string> {\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      const metadata = await AgentModel.findOne({\n        id: \"tool_select_prompt\"\n      }).maxTimeMS(timeout)\n        .lean<PromptDocument>();\n      return metadata?.prompt || \"\";\n    }, \"getToolSelectPrompt()\");\n  };\n\n  public async getPIIDetectPrompt(): Promise<string> {\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      const metadata = await AgentModel.findOne({\n        id: \"pii_detect_prompt\"\n      }).maxTimeMS(timeout)\n        .lean<PromptDocument>();\n      return metadata?.prompt || \"\";\n    }, \"getPIIDetectPrompt()\");\n  };\n\n  public async getPIIFilterPrompt(): Promise<string> {\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      const metadata = await AgentModel.findOne({\n        id: \"pii_filter_prompt\"\n      }).maxTimeMS(timeout)\n        .lean<PromptDocument>();\n      return metadata?.prompt || \"\";\n    }, \"getPIIFilterPrompt()\");\n  };\n}\n","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 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        isPinned: thread.isPinned ?? false,\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      await ThreadModel.create({\n        type,\n        userId,\n        threadId,\n        title,\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 }, { $set: {} });\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\n      // Delete all messages for this thread\n      await MessageModel.deleteMany({ userId, threadId }).maxTimeMS(timeout);\n\n      // Delete the thread itself\n      await ThreadModel.deleteOne({ userId, threadId }).maxTimeMS(timeout);\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({ updatedAt: -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          isPinned: thread.isPinned ?? false,\n          updatedAt: thread.updatedAt,\n        } as ThreadMetadata;\n      })\n      return data;\n    }, `listThreads(${userId})`);\n  };\n\n  public async updateThreadPin(\n    userId: string,\n    threadId: string,\n    isPinned: boolean\n  ): Promise<void> {\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      await ThreadModel.updateOne(\n        { threadId, userId },\n        { $set: { isPinned } }\n      ).maxTimeMS(timeout);\n    }, `updateThreadPin(${userId}, ${threadId})`);\n  };\n}\n","import type { IWorkflowMemory } from \"@ainetwork/adk/modules\";\nimport type { Workflow } from \"@ainetwork/adk/types/memory\";\nimport { WorkflowModel } from \"../models/workflow.model\";\n\nexport type ExecuteWithRetryFn = <T>(\n  operation: () => Promise<T>,\n  operationName?: string\n) => Promise<T>;\n\nexport type GetOperationTimeoutFn = () => number;\n\nexport class MongoDBWorkflow implements IWorkflowMemory {\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 createWorkflow(workflow: Workflow): Promise<Workflow> {\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      const created = await WorkflowModel.create(workflow);\n      return created.toObject() as Workflow;\n    }, \"createWorkflow()\");\n  }\n\n  public async getWorkflow(workflowId: string): Promise<Workflow | undefined> {\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      const workflow = await WorkflowModel.findOne({\n        workflowId\n      }).maxTimeMS(timeout).lean<Workflow>();\n      return workflow || undefined;\n    }, \"getWorkflow()\");\n  }\n\n  public async updateWorkflow(workflowId: string, updates: Partial<Workflow>): Promise<void> {\n    if (!updates.userId) {\n      throw new Error(\"userId is required for updateWorkflow\");\n    }\n\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      // userId를 조건에 포함하여 소유자만 수정 가능하도록 함\n      await WorkflowModel.updateOne(\n        { workflowId, userId: updates.userId },\n        { $set: updates }\n      ).maxTimeMS(timeout);\n    }, \"updateWorkflow()\");\n  }\n\n  public async deleteWorkflow(workflowId: string, userId: string): Promise<void> {\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      await WorkflowModel.deleteOne({ workflowId, userId }).maxTimeMS(timeout);\n    }, \"deleteWorkflow()\");\n  }\n\n  public async listWorkflows(userId?: string): Promise<Workflow[]> {\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      // userId가 있으면: 해당 유저 소유 + 템플릿(userId 없음)\n      // userId가 없으면: 템플릿만 반환\n      const query = userId\n        ? { $or: [{ userId }, { userId: { $exists: false } }, { userId: null }] }\n        : { $or: [{ userId: { $exists: false } }, { userId: null }] };\n      const workflows = await WorkflowModel.find(query)\n        .maxTimeMS(timeout)\n        .lean<Workflow[]>();\n      return workflows;\n    }, \"listWorkflows()\");\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;AACA,OAAO,cAAc;AACrB,SAAS,WAAAA,gBAAe;;;ACajB,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,KAAqB;AACxB,aAAO,UAAU,UAAU;AAAA,IAC7B,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,UAAU,OAAO,GAAG,EAAE,QAAQ,KAAK,CAAC,EAAE,UAAU,OAAO;AAAA,IAC9D,GAAG,qBAAqB;AAAA,EAC1B;AAAA,EAEA,MAAa,qBAAsC;AACjD,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,KAAqB;AACxB,aAAO,UAAU,UAAU;AAAA,IAC7B,GAAG,sBAAsB;AAAA,EAC3B;AAAA,EAEA,MAAa,yBAA0C;AACrD,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,KAAqB;AACxB,aAAO,UAAU,UAAU;AAAA,IAC7B,GAAG,0BAA0B;AAAA,EAC/B;AAAA,EAEA,MAAa,yBAA0C;AACrD,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,KAAqB;AACxB,aAAO,UAAU,UAAU;AAAA,IAC7B,GAAG,0BAA0B;AAAA,EAC/B;AAAA,EAEA,MAAa,wBAAyC;AACpD,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,KAAqB;AACxB,aAAO,UAAU,UAAU;AAAA,IAC7B,GAAG,yBAAyB;AAAA,EAC9B;AAAA,EAEA,MAAa,sBAAuC;AAClD,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,KAAqB;AACxB,aAAO,UAAU,UAAU;AAAA,IAC7B,GAAG,uBAAuB;AAAA,EAC5B;AAAA,EAEA,MAAa,qBAAsC;AACjD,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,KAAqB;AACxB,aAAO,UAAU,UAAU;AAAA,IAC7B,GAAG,sBAAsB;AAAA,EAC3B;AAAA,EAEA,MAAa,qBAAsC;AACjD,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,KAAqB;AACxB,aAAO,UAAU,UAAU;AAAA,IAC7B,GAAG,sBAAsB;AAAA,EAC3B;AACF;;;AChHO,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;;;ACrEA,SAAS,eAAe;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,cAAQ,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,OAAO,YAAY;AAAA,QAC7B,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,YAAY,OAAO;AAAA,QACvB;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,MACF,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,EAAE,MAAM,CAAC,EAAE,CAAC;AAC9D,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;AAGzC,YAAM,aAAa,WAAW,EAAE,QAAQ,SAAS,CAAC,EAAE,UAAU,OAAO;AAGrE,YAAM,YAAY,UAAU,EAAE,QAAQ,SAAS,CAAC,EAAE,UAAU,OAAO;AAAA,IACrE,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,WAAW,GAAG,CAAC,EACtB,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,UAAU,OAAO,YAAY;AAAA,UAC7B,WAAW,OAAO;AAAA,QACpB;AAAA,MACF,CAAC;AACD,aAAO;AAAA,IACT,GAAG,eAAe,MAAM,GAAG;AAAA,EAC7B;AAAA,EAEA,MAAa,gBACX,QACA,UACA,UACe;AACf,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,UAAU,KAAK,oBAAoB;AACzC,YAAM,YAAY;AAAA,QAChB,EAAE,UAAU,OAAO;AAAA,QACnB,EAAE,MAAM,EAAE,SAAS,EAAE;AAAA,MACvB,EAAE,UAAU,OAAO;AAAA,IACrB,GAAG,mBAAmB,MAAM,KAAK,QAAQ,GAAG;AAAA,EAC9C;AACF;;;ACxIO,IAAM,kBAAN,MAAiD;AAAA,EAC9C;AAAA,EACA;AAAA,EAER,YACE,kBACA,qBACA;AACA,SAAK,mBAAmB;AACxB,SAAK,sBAAsB;AAAA,EAC7B;AAAA,EAEA,MAAa,eAAe,UAAuC;AACjE,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,UAAU,KAAK,oBAAoB;AACzC,YAAM,UAAU,MAAM,cAAc,OAAO,QAAQ;AACnD,aAAO,QAAQ,SAAS;AAAA,IAC1B,GAAG,kBAAkB;AAAA,EACvB;AAAA,EAEA,MAAa,YAAY,YAAmD;AAC1E,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,UAAU,KAAK,oBAAoB;AACzC,YAAM,WAAW,MAAM,cAAc,QAAQ;AAAA,QAC3C;AAAA,MACF,CAAC,EAAE,UAAU,OAAO,EAAE,KAAe;AACrC,aAAO,YAAY;AAAA,IACrB,GAAG,eAAe;AAAA,EACpB;AAAA,EAEA,MAAa,eAAe,YAAoB,SAA2C;AACzF,QAAI,CAAC,QAAQ,QAAQ;AACnB,YAAM,IAAI,MAAM,uCAAuC;AAAA,IACzD;AAEA,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,UAAU,KAAK,oBAAoB;AAEzC,YAAM,cAAc;AAAA,QAClB,EAAE,YAAY,QAAQ,QAAQ,OAAO;AAAA,QACrC,EAAE,MAAM,QAAQ;AAAA,MAClB,EAAE,UAAU,OAAO;AAAA,IACrB,GAAG,kBAAkB;AAAA,EACvB;AAAA,EAEA,MAAa,eAAe,YAAoB,QAA+B;AAC7E,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,UAAU,KAAK,oBAAoB;AACzC,YAAM,cAAc,UAAU,EAAE,YAAY,OAAO,CAAC,EAAE,UAAU,OAAO;AAAA,IACzE,GAAG,kBAAkB;AAAA,EACvB;AAAA,EAEA,MAAa,cAAc,QAAsC;AAC/D,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,UAAU,KAAK,oBAAoB;AAGzC,YAAM,QAAQ,SACV,EAAE,KAAK,CAAC,EAAE,OAAO,GAAG,EAAE,QAAQ,EAAE,SAAS,MAAM,EAAE,GAAG,EAAE,QAAQ,KAAK,CAAC,EAAE,IACtE,EAAE,KAAK,CAAC,EAAE,QAAQ,EAAE,SAAS,MAAM,EAAE,GAAG,EAAE,QAAQ,KAAK,CAAC,EAAE;AAC9D,YAAM,YAAY,MAAM,cAAc,KAAK,KAAK,EAC7C,UAAU,OAAO,EACjB,KAAiB;AACpB,aAAO;AAAA,IACT,GAAG,iBAAiB;AAAA,EACtB;AACF;;;AJxDO,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,EACA;AAAA,EACA;AAAA,EAEA;AAAA,EACA;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,QAAI,IAAI,qBAAqB,UAAa,IAAI,mBAAmB,GAAG;AAClE,WAAK,mBAAmB,IAAI;AAAA,IAC9B;AACA,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;AAEA,SAAK,iBAAiB,IAAI;AAAA,MACzB,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,EAEO,oBAAqC;AAC1C,WAAO,KAAK;AAAA,EACd;AAAA,EAEQ,8BAAoC;AAC1C,QAAI,KAAK,oBAAqB;AAE9B,SAAK,sBAAsB;AAE3B,aAAS,WAAW,GAAG,aAAa,MAAM;AACxC,WAAK,YAAY;AACjB,WAAK,oBAAoB;AACzB,WAAK,eAAe;AACpB,MAAAC,SAAQ,MAAM,KAAK,gCAAgC;AAAA,IACrD,CAAC;AAED,aAAS,WAAW,GAAG,gBAAgB,MAAM;AAC3C,WAAK,YAAY;AACjB,MAAAA,SAAQ,MAAM,KAAK,sBAAsB;AACzC,WAAK,oBAAoB;AAAA,IAC3B,CAAC;AAED,aAAS,WAAW,GAAG,SAAS,CAAC,UAAU;AACzC,WAAK,YAAY;AACjB,MAAAA,SAAQ,MAAM,MAAM,6BAA6B,KAAK;AACtD,WAAK,oBAAoB;AAAA,IAC3B,CAAC;AAED,aAAS,WAAW,GAAG,eAAe,MAAM;AAC1C,WAAK,YAAY;AACjB,WAAK,oBAAoB;AACzB,WAAK,eAAe;AACpB,MAAAA,SAAQ,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,MAAAA,SAAQ,MAAM;AAAA,QACZ,uCAAuC,KAAK,iBAAiB,IAAI,KAAK,oBAAoB;AAAA,MAC5F;AAEA,UAAI;AACF,cAAM,SAAS,QAAQ,KAAK,KAAK,KAAK,gBAAgB;AACtD,aAAK,YAAY;AACjB,aAAK,oBAAoB;AACzB,aAAK,eAAe;AACpB,QAAAA,SAAQ,MAAM,KAAK,iCAAiC;AACpD;AAAA,MACF,SAAS,OAAO;AACd,QAAAA,SAAQ,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,MAAAA,SAAQ,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,SAAS,QAAQ,KAAK,KAAK,KAAK,gBAAgB;AACtD,WAAK,YAAY;AACjB,WAAK,oBAAoB;AACzB,YAAM,KAAK,cAAc;AACzB,WAAK,mBAAmB;AAAA,IAC1B,SAAS,OAAO;AACd,MAAAA,SAAQ,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,UAAI,KAAK,oBAAoB;AAC3B,sBAAc,KAAK,kBAAkB;AACrC,aAAK,qBAAqB;AAAA,MAC5B;AACA,YAAM,SAAS,WAAW;AAC1B,WAAK,YAAY;AAAA,IACnB,SAAS,OAAO;AACd,MAAAA,SAAQ,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,EAEA,MAAc,gBAA+B;AAC3C,QAAI,KAAK,qBAAqB,OAAW;AAEzC,QAAI;AACF,YAAM,KAAK,SAAS,WAAW;AAC/B,UAAI,CAAC,GAAI;AAET,YAAM,aAAa,GAAG,WAAW,SAAS;AAC1C,YAAM,UAAU,MAAM,WAAW,QAAQ;AACzC,YAAM,cAAc,QAAQ;AAAA,QAC1B,CAAC,QAAQ,IAAI,KAAK,cAAc,UAAa,IAAI,uBAAuB;AAAA,MAC1E;AAEA,UAAI,aAAa;AACf,YAAI,YAAY,uBAAuB,KAAK,kBAAkB;AAC5D,gBAAM,GAAG,QAAQ;AAAA,YACf,SAAS;AAAA,YACT,OAAO,EAAE,YAAY,EAAE,WAAW,EAAE,GAAG,oBAAoB,KAAK,iBAAiB;AAAA,UACnF,CAAC;AACD,UAAAA,SAAQ,MAAM,KAAK,+BAA+B,KAAK,gBAAgB,UAAU;AAAA,QACnF;AAAA,MACF,OAAO;AACL,cAAM,WAAW;AAAA,UACf,EAAE,WAAW,EAAE;AAAA,UACf,EAAE,oBAAoB,KAAK,iBAAiB;AAAA,QAC9C;AACA,QAAAA,SAAQ,MAAM,KAAK,iCAAiC,KAAK,gBAAgB,UAAU;AAAA,MACrF;AAAA,IACF,SAAS,OAAO;AACd,MAAAA,SAAQ,MAAM,MAAM,8BAA8B,KAAK;AAAA,IACzD;AAAA,EACF;AAAA,EAEQ,qBAA2B;AACjC,QAAI,KAAK,qBAAqB,OAAW;AACzC,QAAI,KAAK,mBAAoB;AAG7B,UAAM,aAAa,KAAK,IAAI,KAAQ,KAAK,IAAI,KAAK,mBAAmB,KAAK,IAAS,CAAC;AACpF,SAAK,qBAAqB,YAAY,MAAM;AAC1C,WAAK,wBAAwB,EAAE,MAAM,CAAC,UAAU;AAC9C,QAAAA,SAAQ,MAAM,MAAM,oCAAoC,KAAK;AAAA,MAC/D,CAAC;AAAA,IACH,GAAG,UAAU;AAEb,IAAAA,SAAQ,MAAM,KAAK,4CAA4C,KAAK,MAAM,aAAa,GAAI,CAAC,GAAG;AAAA,EACjG;AAAA,EAEA,MAAc,0BAAyC;AACrD,QAAI,CAAC,KAAK,UAAW;AAErB,QAAI;AACF,YAAM,KAAK,SAAS,WAAW;AAC/B,UAAI,CAAC,GAAI;AAET,YAAM,oBAAoB,MAAM,GAAG,WAAW,SAAS,EAAE,SAAS,UAAU;AAC5E,YAAM,SAAS,MAAM,aAAa,WAAW;AAAA,QAC3C,UAAU,EAAE,MAAM,kBAAkB;AAAA,MACtC,CAAC;AAED,UAAI,OAAO,eAAe,GAAG;AAC3B,QAAAA,SAAQ,MAAM,KAAK,cAAc,OAAO,YAAY,oBAAoB;AAAA,MAC1E;AAAA,IACF,SAAS,OAAO;AACd,MAAAA,SAAQ,MAAM,MAAM,wCAAwC,KAAK;AAAA,IACnE;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,QAAAA,SAAQ,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,QAAAA,SAAQ,MAAM;AAAA,UACZ,GAAG,aAAa;AAAA,QAClB;AAEA,cAAM,KAAK,iBAAiB;AAG5B,YAAI;AACF,iBAAO,MAAM,UAAU;AAAA,QACzB,SAAS,YAAiB;AACxB,UAAAA,SAAQ,MAAM,MAAM,GAAG,aAAa,wBAAwB,UAAU;AACtE,gBAAM;AAAA,QACR;AAAA,MACF;AAGA,YAAM;AAAA,IACR;AAAA,EACF;AACF;","names":["loggers","loggers"]}
+{"version":3,"sources":["../implements/base.memory.ts","../implements/agent.memory.ts","../implements/intent.memory.ts","../implements/thread.memory.ts","../implements/user-workflow.memory.ts","../implements/workflow-template.memory.ts"],"sourcesContent":["import { IAgentMemory, IIntentMemory, IMemory, IUserWorkflowMemory, IThreadMemory, IWorkflowTemplateMemory } from \"@ainetwork/adk/modules\";\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\";\nimport { MongoDBUserWorkflow } from \"./user-workflow.memory\";\nimport { MongoDBWorkflowTemplate } from \"./workflow-template.memory\";\nimport { MessageModel } from \"../models/messages.model\";\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  threadTTLSeconds?: number; // TTL for thread documents (in seconds). Orphaned messages are periodically cleaned up.\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  private threadTTLSeconds?: number;\n  private orphanCleanupTimer?: ReturnType<typeof setInterval>;\n\n  private agentMemory: MongoDBAgent;\n  private intentMemory: MongoDBIntent;\n  private threadMemory: MongoDBThread;\n  private workflowTemplateMemory: MongoDBWorkflowTemplate;\n  private userWorkflowMemory: MongoDBUserWorkflow;\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    if (cfg.threadTTLSeconds !== undefined && cfg.threadTTLSeconds > 0) {\n      this.threadTTLSeconds = cfg.threadTTLSeconds;\n    }\n    this.connectionConfig = {\n      maxPoolSize: cfg.maxPoolSize ?? 10,\n      minPoolSize: 0,\n      maxIdleTimeMS: 30000,\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\t\tthis.workflowTemplateMemory = new MongoDBWorkflowTemplate(\n\t\t\tthis.executeWithRetry.bind(this),\n\t\t\tthis.getOperationTimeout.bind(this)\n\t\t);\n\n\t\tthis.userWorkflowMemory = new MongoDBUserWorkflow(\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  public getWorkflowTemplateMemory(): IWorkflowTemplateMemory {\n    return this.workflowTemplateMemory;\n  }\n\n  public getUserWorkflowMemory(): IUserWorkflowMemory {\n    return this.userWorkflowMemory;\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.connected) {\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.connected) {\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      await this.setupTTLIndex();\n      this.startOrphanCleanup();\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.connected) {\n      return;\n    }\n\n    try {\n      if (this.orphanCleanupTimer) {\n        clearInterval(this.orphanCleanupTimer);\n        this.orphanCleanupTimer = undefined;\n      }\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.connected && !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.connected) {\n      throw new Error(\"MongoDB is not connected and reconnection failed\");\n    }\n  }\n\n  private async setupTTLIndex(): Promise<void> {\n    if (this.threadTTLSeconds === undefined) return;\n\n    try {\n      const db = mongoose.connection.db;\n      if (!db) return;\n\n      const collection = db.collection('threads');\n      const indexes = await collection.indexes();\n      const existingTTL = indexes.find(\n        (idx) => idx.key?.updatedAt !== undefined && idx.expireAfterSeconds !== undefined\n      );\n\n      if (existingTTL) {\n        if (existingTTL.expireAfterSeconds !== this.threadTTLSeconds) {\n          await db.command({\n            collMod: 'threads',\n            index: { keyPattern: { updatedAt: 1 }, expireAfterSeconds: this.threadTTLSeconds },\n          });\n          loggers.agent.info(`Thread TTL index updated to ${this.threadTTLSeconds} seconds`);\n        }\n      } else {\n        await collection.createIndex(\n          { updatedAt: 1 },\n          { expireAfterSeconds: this.threadTTLSeconds },\n        );\n        loggers.agent.info(`Thread TTL index created with ${this.threadTTLSeconds} seconds`);\n      }\n    } catch (error) {\n      loggers.agent.error('Failed to setup TTL index:', error);\n    }\n  }\n\n  private startOrphanCleanup(): void {\n    if (this.threadTTLSeconds === undefined) return;\n    if (this.orphanCleanupTimer) return;\n\n    // Run cleanup at half the TTL interval, with a minimum of 60s and maximum of 1 hour\n    const intervalMs = Math.max(60_000, Math.min(this.threadTTLSeconds * 500, 3_600_000));\n    this.orphanCleanupTimer = setInterval(() => {\n      this.cleanupOrphanedMessages().catch((error) => {\n        loggers.agent.error('Orphaned message cleanup failed:', error);\n      });\n    }, intervalMs);\n\n    loggers.agent.info(`Orphaned message cleanup scheduled every ${Math.round(intervalMs / 1000)}s`);\n  }\n\n  private async cleanupOrphanedMessages(): Promise<void> {\n    if (!this.connected) return;\n\n    try {\n      const db = mongoose.connection.db;\n      if (!db) return;\n\n      const existingThreadIds = await db.collection('threads').distinct('threadId');\n      const result = await MessageModel.deleteMany({\n        threadId: { $nin: existingThreadIds },\n      });\n\n      if (result.deletedCount > 0) {\n        loggers.agent.info(`Cleaned up ${result.deletedCount} orphaned messages`);\n      }\n    } catch (error) {\n      loggers.agent.error('Failed to cleanup orphaned messages:', error);\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 TooManyLogicalSessions error\n      if (error.code === 261 || error.codeName === \"TooManyLogicalSessions\") {\n        loggers.agent.warn(\n          `${operationName} failed due to too many sessions, disconnecting to release sessions...`\n        );\n\n        try {\n          await mongoose.disconnect();\n          this.connected = false;\n        } catch (disconnectError) {\n          loggers.agent.error(\"Failed to disconnect during session cleanup:\", disconnectError);\n        }\n\n        await new Promise((resolve) => setTimeout(resolve, 5000));\n        await this.ensureConnection();\n\n        try {\n          return await operation();\n        } catch (retryError: any) {\n          loggers.agent.error(`${operationName} failed after session cleanup retry:`, retryError);\n          throw retryError;\n        }\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 { 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 PromptDocument = {\n  id: string;\n  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_prompt\"\n      }).maxTimeMS(timeout)\n        .lean<PromptDocument>();\n      return metadata?.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_prompt\",\n      }, { \"prompt\": prompt }, { upsert: true }).maxTimeMS(timeout);\n    }, \"updateAgentPrompt()\");\n  };\n\n  public async getAggregatePrompt(): Promise<string> {\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      const metadata = await AgentModel.findOne({\n        id: \"aggregate_prompt\"\n      }).maxTimeMS(timeout)\n        .lean<PromptDocument>();\n      return metadata?.prompt || \"\";\n    }, \"getAggregatePrompt()\");\n  };\n\n  public async getGenerateTitlePrompt(): Promise<string> {\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      const metadata = await AgentModel.findOne({\n        id: \"generate_title_prompt\"\n      }).maxTimeMS(timeout)\n        .lean<PromptDocument>();\n      return metadata?.prompt || \"\";\n    }, \"getGenerateTitlePrompt()\");\n  };\n\n  public async getSingleTriggerPrompt(): Promise<string> {\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      const metadata = await AgentModel.findOne({\n        id: \"single_trigger_prompt\"\n      }).maxTimeMS(timeout)\n        .lean<PromptDocument>();\n      return metadata?.prompt || \"\";\n    }, \"getSingleTriggerPrompt()\");\n  };\n\n  public async getMultiTriggerPrompt(): Promise<string> {\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      const metadata = await AgentModel.findOne({\n        id: \"multi_trigger_prompt\"\n      }).maxTimeMS(timeout)\n        .lean<PromptDocument>();\n      return metadata?.prompt || \"\";\n    }, \"getMultiTriggerPrompt()\");\n  };\n\n  public async getToolSelectPrompt(): Promise<string> {\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      const metadata = await AgentModel.findOne({\n        id: \"tool_select_prompt\"\n      }).maxTimeMS(timeout)\n        .lean<PromptDocument>();\n      return metadata?.prompt || \"\";\n    }, \"getToolSelectPrompt()\");\n  };\n\n  public async getPIIDetectPrompt(): Promise<string> {\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      const metadata = await AgentModel.findOne({\n        id: \"pii_detect_prompt\"\n      }).maxTimeMS(timeout)\n        .lean<PromptDocument>();\n      return metadata?.prompt || \"\";\n    }, \"getPIIDetectPrompt()\");\n  };\n\n  public async getPIIFilterPrompt(): Promise<string> {\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      const metadata = await AgentModel.findOne({\n        id: \"pii_filter_prompt\"\n      }).maxTimeMS(timeout)\n        .lean<PromptDocument>();\n      return metadata?.prompt || \"\";\n    }, \"getPIIFilterPrompt()\");\n  };\n}\n","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 type { MessageObject, ThreadMetadata, ThreadObject, ThreadType, ThreadFilter } 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        isPinned: thread.isPinned ?? false,\n        workflowId: thread.workflowId,\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    workflowId?: string,\n  ): Promise<ThreadObject> {\n    return this.executeWithRetry(async () => {\n      await ThreadModel.create({\n        type,\n        userId,\n        threadId,\n        title,\n        workflowId,\n      });\n\n      return { type, userId, threadId, title, workflowId, 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      if (messages.length > 0) {\n        const messageIds = messages.map((m) => m.messageId);\n        await MessageModel.deleteMany({ threadId, userId, messageId: { $in: messageIds } });\n        await MessageModel.insertMany(\n          messages.map((message) => ({\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      await ThreadModel.updateOne(\n        { threadId, userId },\n        { $set: { updatedAt: new Date() } }\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\n      // Delete all messages for this thread\n      await MessageModel.deleteMany({ userId, threadId }).maxTimeMS(timeout);\n\n      // Delete the thread itself\n      await ThreadModel.deleteOne({ userId, threadId }).maxTimeMS(timeout);\n    }, `deleteThread(${userId}, ${threadId})`);\n  };\n\n  public async listThreads(userId: string, filter?: ThreadFilter): Promise<ThreadMetadata[]> {\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      const query: Record<string, any> = { userId };\n      if (filter?.workflowId) query.workflowId = filter.workflowId;\n      if (filter?.type) query.type = filter.type;\n      const threads = await ThreadModel.find(query)\n        .sort({ updatedAt: -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          isPinned: thread.isPinned ?? false,\n          workflowId: thread.workflowId,\n          createdAt: thread.createdAt?.toISOString(),\n          updatedAt: thread.updatedAt?.toISOString(),\n        } as ThreadMetadata;\n      })\n      return data;\n    }, `listThreads(${userId})`);\n  };\n\n  public async updateThreadPin(\n    userId: string,\n    threadId: string,\n    isPinned: boolean\n  ): Promise<void> {\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      await ThreadModel.updateOne(\n        { threadId, userId },\n        { $set: { isPinned } }\n      ).maxTimeMS(timeout);\n    }, `updateThreadPin(${userId}, ${threadId})`);\n  };\n}\n","import type { IUserWorkflowMemory } from \"@ainetwork/adk/modules\";\nimport type { UserWorkflow } from \"@ainetwork/adk/types/memory\";\nimport { UserWorkflowModel } from \"../models/user-workflow.model\";\n\nexport type ExecuteWithRetryFn = <T>(\n  operation: () => Promise<T>,\n  operationName?: string\n) => Promise<T>;\n\nexport type GetOperationTimeoutFn = () => number;\n\nexport class MongoDBUserWorkflow implements IUserWorkflowMemory {\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 createUserWorkflow(workflow: UserWorkflow): Promise<UserWorkflow> {\n    return this.executeWithRetry(async () => {\n      const created = await UserWorkflowModel.create(workflow);\n      return created.toObject() as UserWorkflow;\n    }, \"createUserWorkflow()\");\n  }\n\n  public async getUserWorkflow(workflowId: string): Promise<UserWorkflow | undefined> {\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      const workflow = await UserWorkflowModel.findOne({ workflowId })\n        .maxTimeMS(timeout)\n        .lean<UserWorkflow>();\n      return workflow || undefined;\n    }, \"getUserWorkflow()\");\n  }\n\n  public async updateUserWorkflow(workflowId: string, updates: Partial<UserWorkflow>): Promise<void> {\n    if (!updates.userId) {\n      throw new Error(\"userId is required for updateUserWorkflow\");\n    }\n\n    const { userId, workflowId: _workflowId, ...mutableUpdates } = updates;\n\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      await UserWorkflowModel.updateOne(\n        { workflowId, userId },\n        { $set: mutableUpdates }\n      ).maxTimeMS(timeout);\n    }, \"updateUserWorkflow()\");\n  }\n\n  public async deleteUserWorkflow(workflowId: string, userId: string): Promise<void> {\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      await UserWorkflowModel.deleteOne({ workflowId, userId }).maxTimeMS(timeout);\n    }, \"deleteUserWorkflow()\");\n  }\n\n  public async listUserWorkflows(userId?: string): Promise<UserWorkflow[]> {\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      const query = userId ? { userId } : {};\n      const workflows = await UserWorkflowModel.find(query)\n        .maxTimeMS(timeout)\n        .lean<UserWorkflow[]>();\n      return workflows;\n    }, \"listUserWorkflows()\");\n  }\n\n  public async listActiveScheduledWorkflows(): Promise<UserWorkflow[]> {\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      const workflows = await UserWorkflowModel.find({\n        active: true,\n        schedule: { $exists: true, $ne: null },\n      })\n        .maxTimeMS(timeout)\n        .lean<UserWorkflow[]>();\n      return workflows;\n    }, \"listActiveScheduledWorkflows()\");\n  }\n}\n","import type { IWorkflowTemplateMemory } from \"@ainetwork/adk/modules\";\nimport type { WorkflowTemplate } from \"@ainetwork/adk/types/memory\";\nimport { WorkflowTemplateModel } from \"../models/workflow-template.model\";\n\nexport type ExecuteWithRetryFn = <T>(\n  operation: () => Promise<T>,\n  operationName?: string\n) => Promise<T>;\n\nexport type GetOperationTimeoutFn = () => number;\n\nexport class MongoDBWorkflowTemplate implements IWorkflowTemplateMemory {\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 createTemplate(template: WorkflowTemplate): Promise<WorkflowTemplate> {\n    return this.executeWithRetry(async () => {\n      const created = await WorkflowTemplateModel.create(template);\n      return created.toObject() as WorkflowTemplate;\n    }, \"createTemplate()\");\n  }\n\n  public async getTemplate(templateId: string): Promise<WorkflowTemplate | undefined> {\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      const template = await WorkflowTemplateModel.findOne({ templateId })\n        .maxTimeMS(timeout)\n        .lean<WorkflowTemplate>();\n      return template || undefined;\n    }, \"getTemplate()\");\n  }\n\n  public async updateTemplate(templateId: string, updates: Partial<WorkflowTemplate>): Promise<void> {\n    const { templateId: _templateId, ...mutableUpdates } = updates;\n\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      await WorkflowTemplateModel.updateOne(\n        { templateId },\n        { $set: mutableUpdates }\n      ).maxTimeMS(timeout);\n    }, \"updateTemplate()\");\n  }\n\n  public async deleteTemplate(templateId: string): Promise<void> {\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      await WorkflowTemplateModel.deleteOne({ templateId }).maxTimeMS(timeout);\n    }, \"deleteTemplate()\");\n  }\n\n  public async listTemplates(): Promise<WorkflowTemplate[]> {\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      const templates = await WorkflowTemplateModel.find()\n        .maxTimeMS(timeout)\n        .lean<WorkflowTemplate[]>();\n      return templates;\n    }, \"listTemplates()\");\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AACA,OAAO,cAAc;AACrB,SAAS,WAAAA,gBAAe;;;ACajB,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,KAAqB;AACxB,aAAO,UAAU,UAAU;AAAA,IAC7B,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,UAAU,OAAO,GAAG,EAAE,QAAQ,KAAK,CAAC,EAAE,UAAU,OAAO;AAAA,IAC9D,GAAG,qBAAqB;AAAA,EAC1B;AAAA,EAEA,MAAa,qBAAsC;AACjD,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,KAAqB;AACxB,aAAO,UAAU,UAAU;AAAA,IAC7B,GAAG,sBAAsB;AAAA,EAC3B;AAAA,EAEA,MAAa,yBAA0C;AACrD,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,KAAqB;AACxB,aAAO,UAAU,UAAU;AAAA,IAC7B,GAAG,0BAA0B;AAAA,EAC/B;AAAA,EAEA,MAAa,yBAA0C;AACrD,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,KAAqB;AACxB,aAAO,UAAU,UAAU;AAAA,IAC7B,GAAG,0BAA0B;AAAA,EAC/B;AAAA,EAEA,MAAa,wBAAyC;AACpD,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,KAAqB;AACxB,aAAO,UAAU,UAAU;AAAA,IAC7B,GAAG,yBAAyB;AAAA,EAC9B;AAAA,EAEA,MAAa,sBAAuC;AAClD,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,KAAqB;AACxB,aAAO,UAAU,UAAU;AAAA,IAC7B,GAAG,uBAAuB;AAAA,EAC5B;AAAA,EAEA,MAAa,qBAAsC;AACjD,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,KAAqB;AACxB,aAAO,UAAU,UAAU;AAAA,IAC7B,GAAG,sBAAsB;AAAA,EAC3B;AAAA,EAEA,MAAa,qBAAsC;AACjD,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,KAAqB;AACxB,aAAO,UAAU,UAAU;AAAA,IAC7B,GAAG,sBAAsB;AAAA,EAC3B;AACF;;;AChHO,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;;;ACrEA,SAAS,eAAe;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,cAAQ,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,OAAO,YAAY;AAAA,QAC7B,YAAY,OAAO;AAAA,QACnB,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,OACA,YACuB;AACvB,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,YAAY,OAAO;AAAA,QACvB;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,MACF,CAAC;AAED,aAAO,EAAE,MAAM,QAAQ,UAAU,OAAO,YAAY,UAAU,CAAC,EAAC;AAAA,IAClE,GAAG,gBAAgB,MAAM,KAAK,QAAQ,GAAG;AAAA,EAC3C;AAAA,EAEA,MAAa,oBACX,QACA,UACA,UACe;AACf,WAAO,KAAK,iBAAiB,YAAY;AACvC,UAAI,SAAS,SAAS,GAAG;AACvB,cAAM,aAAa,SAAS,IAAI,CAAC,MAAM,EAAE,SAAS;AAClD,cAAM,aAAa,WAAW,EAAE,UAAU,QAAQ,WAAW,EAAE,KAAK,WAAW,EAAE,CAAC;AAClF,cAAM,aAAa;AAAA,UACjB,SAAS,IAAI,CAAC,aAAa;AAAA,YACzB;AAAA,YACA,WAAW,QAAQ;AAAA,YACnB;AAAA,YACA,MAAM,QAAQ;AAAA,YACd,SAAS,QAAQ;AAAA,YACjB,WAAW,QAAQ;AAAA,YACnB,UAAU,QAAQ;AAAA,UACpB,EAAE;AAAA,QACJ;AAAA,MACF;AACA,YAAM,YAAY;AAAA,QAChB,EAAE,UAAU,OAAO;AAAA,QACnB,EAAE,MAAM,EAAE,WAAW,oBAAI,KAAK,EAAE,EAAE;AAAA,MACpC;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;AAGzC,YAAM,aAAa,WAAW,EAAE,QAAQ,SAAS,CAAC,EAAE,UAAU,OAAO;AAGrE,YAAM,YAAY,UAAU,EAAE,QAAQ,SAAS,CAAC,EAAE,UAAU,OAAO;AAAA,IACrE,GAAG,gBAAgB,MAAM,KAAK,QAAQ,GAAG;AAAA,EAC3C;AAAA,EAEA,MAAa,YAAY,QAAgB,QAAkD;AACzF,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,UAAU,KAAK,oBAAoB;AACzC,YAAM,QAA6B,EAAE,OAAO;AAC5C,UAAI,QAAQ,WAAY,OAAM,aAAa,OAAO;AAClD,UAAI,QAAQ,KAAM,OAAM,OAAO,OAAO;AACtC,YAAM,UAAU,MAAM,YAAY,KAAK,KAAK,EACzC,KAAK,EAAE,WAAW,GAAG,CAAC,EACtB,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,UAAU,OAAO,YAAY;AAAA,UAC7B,YAAY,OAAO;AAAA,UACnB,WAAW,OAAO,WAAW,YAAY;AAAA,UACzC,WAAW,OAAO,WAAW,YAAY;AAAA,QAC3C;AAAA,MACF,CAAC;AACD,aAAO;AAAA,IACT,GAAG,eAAe,MAAM,GAAG;AAAA,EAC7B;AAAA,EAEA,MAAa,gBACX,QACA,UACA,UACe;AACf,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,UAAU,KAAK,oBAAoB;AACzC,YAAM,YAAY;AAAA,QAChB,EAAE,UAAU,OAAO;AAAA,QACnB,EAAE,MAAM,EAAE,SAAS,EAAE;AAAA,MACvB,EAAE,UAAU,OAAO;AAAA,IACrB,GAAG,mBAAmB,MAAM,KAAK,QAAQ,GAAG;AAAA,EAC9C;AACF;;;ACvJO,IAAM,sBAAN,MAAyD;AAAA,EACtD;AAAA,EACA;AAAA,EAER,YACE,kBACA,qBACA;AACA,SAAK,mBAAmB;AACxB,SAAK,sBAAsB;AAAA,EAC7B;AAAA,EAEA,MAAa,mBAAmB,UAA+C;AAC7E,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,UAAU,MAAM,kBAAkB,OAAO,QAAQ;AACvD,aAAO,QAAQ,SAAS;AAAA,IAC1B,GAAG,sBAAsB;AAAA,EAC3B;AAAA,EAEA,MAAa,gBAAgB,YAAuD;AAClF,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,UAAU,KAAK,oBAAoB;AACzC,YAAM,WAAW,MAAM,kBAAkB,QAAQ,EAAE,WAAW,CAAC,EAC5D,UAAU,OAAO,EACjB,KAAmB;AACtB,aAAO,YAAY;AAAA,IACrB,GAAG,mBAAmB;AAAA,EACxB;AAAA,EAEA,MAAa,mBAAmB,YAAoB,SAA+C;AACjG,QAAI,CAAC,QAAQ,QAAQ;AACnB,YAAM,IAAI,MAAM,2CAA2C;AAAA,IAC7D;AAEA,UAAM,EAAE,QAAQ,YAAY,aAAa,GAAG,eAAe,IAAI;AAE/D,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,UAAU,KAAK,oBAAoB;AACzC,YAAM,kBAAkB;AAAA,QACtB,EAAE,YAAY,OAAO;AAAA,QACrB,EAAE,MAAM,eAAe;AAAA,MACzB,EAAE,UAAU,OAAO;AAAA,IACrB,GAAG,sBAAsB;AAAA,EAC3B;AAAA,EAEA,MAAa,mBAAmB,YAAoB,QAA+B;AACjF,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,UAAU,KAAK,oBAAoB;AACzC,YAAM,kBAAkB,UAAU,EAAE,YAAY,OAAO,CAAC,EAAE,UAAU,OAAO;AAAA,IAC7E,GAAG,sBAAsB;AAAA,EAC3B;AAAA,EAEA,MAAa,kBAAkB,QAA0C;AACvE,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,UAAU,KAAK,oBAAoB;AACzC,YAAM,QAAQ,SAAS,EAAE,OAAO,IAAI,CAAC;AACrC,YAAM,YAAY,MAAM,kBAAkB,KAAK,KAAK,EACjD,UAAU,OAAO,EACjB,KAAqB;AACxB,aAAO;AAAA,IACT,GAAG,qBAAqB;AAAA,EAC1B;AAAA,EAEA,MAAa,+BAAwD;AACnE,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,UAAU,KAAK,oBAAoB;AACzC,YAAM,YAAY,MAAM,kBAAkB,KAAK;AAAA,QAC7C,QAAQ;AAAA,QACR,UAAU,EAAE,SAAS,MAAM,KAAK,KAAK;AAAA,MACvC,CAAC,EACE,UAAU,OAAO,EACjB,KAAqB;AACxB,aAAO;AAAA,IACT,GAAG,gCAAgC;AAAA,EACrC;AACF;;;AC3EO,IAAM,0BAAN,MAAiE;AAAA,EAC9D;AAAA,EACA;AAAA,EAER,YACE,kBACA,qBACA;AACA,SAAK,mBAAmB;AACxB,SAAK,sBAAsB;AAAA,EAC7B;AAAA,EAEA,MAAa,eAAe,UAAuD;AACjF,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,UAAU,MAAM,sBAAsB,OAAO,QAAQ;AAC3D,aAAO,QAAQ,SAAS;AAAA,IAC1B,GAAG,kBAAkB;AAAA,EACvB;AAAA,EAEA,MAAa,YAAY,YAA2D;AAClF,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,UAAU,KAAK,oBAAoB;AACzC,YAAM,WAAW,MAAM,sBAAsB,QAAQ,EAAE,WAAW,CAAC,EAChE,UAAU,OAAO,EACjB,KAAuB;AAC1B,aAAO,YAAY;AAAA,IACrB,GAAG,eAAe;AAAA,EACpB;AAAA,EAEA,MAAa,eAAe,YAAoB,SAAmD;AACjG,UAAM,EAAE,YAAY,aAAa,GAAG,eAAe,IAAI;AAEvD,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,UAAU,KAAK,oBAAoB;AACzC,YAAM,sBAAsB;AAAA,QAC1B,EAAE,WAAW;AAAA,QACb,EAAE,MAAM,eAAe;AAAA,MACzB,EAAE,UAAU,OAAO;AAAA,IACrB,GAAG,kBAAkB;AAAA,EACvB;AAAA,EAEA,MAAa,eAAe,YAAmC;AAC7D,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,UAAU,KAAK,oBAAoB;AACzC,YAAM,sBAAsB,UAAU,EAAE,WAAW,CAAC,EAAE,UAAU,OAAO;AAAA,IACzE,GAAG,kBAAkB;AAAA,EACvB;AAAA,EAEA,MAAa,gBAA6C;AACxD,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,UAAU,KAAK,oBAAoB;AACzC,YAAM,YAAY,MAAM,sBAAsB,KAAK,EAChD,UAAU,OAAO,EACjB,KAAyB;AAC5B,aAAO;AAAA,IACT,GAAG,iBAAiB;AAAA,EACtB;AACF;;;AL9CO,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,EACA;AAAA,EACA;AAAA,EAEA;AAAA,EACA;AAAA,EACA;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,QAAI,IAAI,qBAAqB,UAAa,IAAI,mBAAmB,GAAG;AAClE,WAAK,mBAAmB,IAAI;AAAA,IAC9B;AACA,SAAK,mBAAmB;AAAA,MACtB,aAAa,IAAI,eAAe;AAAA,MAChC,aAAa;AAAA,MACb,eAAe;AAAA,MACf,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;AAEA,SAAK,yBAAyB,IAAI;AAAA,MACjC,KAAK,iBAAiB,KAAK,IAAI;AAAA,MAC/B,KAAK,oBAAoB,KAAK,IAAI;AAAA,IACnC;AAEA,SAAK,qBAAqB,IAAI;AAAA,MAC7B,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,EAEO,4BAAqD;AAC1D,WAAO,KAAK;AAAA,EACd;AAAA,EAEO,wBAA6C;AAClD,WAAO,KAAK;AAAA,EACd;AAAA,EAEQ,8BAAoC;AAC1C,QAAI,KAAK,oBAAqB;AAE9B,SAAK,sBAAsB;AAE3B,aAAS,WAAW,GAAG,aAAa,MAAM;AACxC,WAAK,YAAY;AACjB,WAAK,oBAAoB;AACzB,WAAK,eAAe;AACpB,MAAAC,SAAQ,MAAM,KAAK,gCAAgC;AAAA,IACrD,CAAC;AAED,aAAS,WAAW,GAAG,gBAAgB,MAAM;AAC3C,WAAK,YAAY;AACjB,MAAAA,SAAQ,MAAM,KAAK,sBAAsB;AACzC,WAAK,oBAAoB;AAAA,IAC3B,CAAC;AAED,aAAS,WAAW,GAAG,SAAS,CAAC,UAAU;AACzC,WAAK,YAAY;AACjB,MAAAA,SAAQ,MAAM,MAAM,6BAA6B,KAAK;AACtD,WAAK,oBAAoB;AAAA,IAC3B,CAAC;AAED,aAAS,WAAW,GAAG,eAAe,MAAM;AAC1C,WAAK,YAAY;AACjB,WAAK,oBAAoB;AACzB,WAAK,eAAe;AACpB,MAAAA,SAAQ,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,WAAW;AAC5E,WAAK;AACL,MAAAA,SAAQ,MAAM;AAAA,QACZ,uCAAuC,KAAK,iBAAiB,IAAI,KAAK,oBAAoB;AAAA,MAC5F;AAEA,UAAI;AACF,cAAM,SAAS,QAAQ,KAAK,KAAK,KAAK,gBAAgB;AACtD,aAAK,YAAY;AACjB,aAAK,oBAAoB;AACzB,aAAK,eAAe;AACpB,QAAAA,SAAQ,MAAM,KAAK,iCAAiC;AACpD;AAAA,MACF,SAAS,OAAO;AACd,QAAAA,SAAQ,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,WAAW;AACnB,MAAAA,SAAQ,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,SAAS,QAAQ,KAAK,KAAK,KAAK,gBAAgB;AACtD,WAAK,YAAY;AACjB,WAAK,oBAAoB;AACzB,YAAM,KAAK,cAAc;AACzB,WAAK,mBAAmB;AAAA,IAC1B,SAAS,OAAO;AACd,MAAAA,SAAQ,MAAM,MAAM,iCAAiC,KAAK;AAC1D,YAAM;AAAA,IACR;AAAA,EACF;AAAA,EAEA,MAAa,aAA4B;AACvC,QAAI,CAAC,KAAK,WAAW;AACnB;AAAA,IACF;AAEA,QAAI;AACF,UAAI,KAAK,oBAAoB;AAC3B,sBAAc,KAAK,kBAAkB;AACrC,aAAK,qBAAqB;AAAA,MAC5B;AACA,YAAM,SAAS,WAAW;AAC1B,WAAK,YAAY;AAAA,IACnB,SAAS,OAAO;AACd,MAAAA,SAAQ,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,aAAa,CAAC,KAAK,cAAc;AACzC,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,WAAW;AACnB,YAAM,IAAI,MAAM,kDAAkD;AAAA,IACpE;AAAA,EACF;AAAA,EAEA,MAAc,gBAA+B;AAC3C,QAAI,KAAK,qBAAqB,OAAW;AAEzC,QAAI;AACF,YAAM,KAAK,SAAS,WAAW;AAC/B,UAAI,CAAC,GAAI;AAET,YAAM,aAAa,GAAG,WAAW,SAAS;AAC1C,YAAM,UAAU,MAAM,WAAW,QAAQ;AACzC,YAAM,cAAc,QAAQ;AAAA,QAC1B,CAAC,QAAQ,IAAI,KAAK,cAAc,UAAa,IAAI,uBAAuB;AAAA,MAC1E;AAEA,UAAI,aAAa;AACf,YAAI,YAAY,uBAAuB,KAAK,kBAAkB;AAC5D,gBAAM,GAAG,QAAQ;AAAA,YACf,SAAS;AAAA,YACT,OAAO,EAAE,YAAY,EAAE,WAAW,EAAE,GAAG,oBAAoB,KAAK,iBAAiB;AAAA,UACnF,CAAC;AACD,UAAAA,SAAQ,MAAM,KAAK,+BAA+B,KAAK,gBAAgB,UAAU;AAAA,QACnF;AAAA,MACF,OAAO;AACL,cAAM,WAAW;AAAA,UACf,EAAE,WAAW,EAAE;AAAA,UACf,EAAE,oBAAoB,KAAK,iBAAiB;AAAA,QAC9C;AACA,QAAAA,SAAQ,MAAM,KAAK,iCAAiC,KAAK,gBAAgB,UAAU;AAAA,MACrF;AAAA,IACF,SAAS,OAAO;AACd,MAAAA,SAAQ,MAAM,MAAM,8BAA8B,KAAK;AAAA,IACzD;AAAA,EACF;AAAA,EAEQ,qBAA2B;AACjC,QAAI,KAAK,qBAAqB,OAAW;AACzC,QAAI,KAAK,mBAAoB;AAG7B,UAAM,aAAa,KAAK,IAAI,KAAQ,KAAK,IAAI,KAAK,mBAAmB,KAAK,IAAS,CAAC;AACpF,SAAK,qBAAqB,YAAY,MAAM;AAC1C,WAAK,wBAAwB,EAAE,MAAM,CAAC,UAAU;AAC9C,QAAAA,SAAQ,MAAM,MAAM,oCAAoC,KAAK;AAAA,MAC/D,CAAC;AAAA,IACH,GAAG,UAAU;AAEb,IAAAA,SAAQ,MAAM,KAAK,4CAA4C,KAAK,MAAM,aAAa,GAAI,CAAC,GAAG;AAAA,EACjG;AAAA,EAEA,MAAc,0BAAyC;AACrD,QAAI,CAAC,KAAK,UAAW;AAErB,QAAI;AACF,YAAM,KAAK,SAAS,WAAW;AAC/B,UAAI,CAAC,GAAI;AAET,YAAM,oBAAoB,MAAM,GAAG,WAAW,SAAS,EAAE,SAAS,UAAU;AAC5E,YAAM,SAAS,MAAM,aAAa,WAAW;AAAA,QAC3C,UAAU,EAAE,MAAM,kBAAkB;AAAA,MACtC,CAAC;AAED,UAAI,OAAO,eAAe,GAAG;AAC3B,QAAAA,SAAQ,MAAM,KAAK,cAAc,OAAO,YAAY,oBAAoB;AAAA,MAC1E;AAAA,IACF,SAAS,OAAO;AACd,MAAAA,SAAQ,MAAM,MAAM,wCAAwC,KAAK;AAAA,IACnE;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,QAAAA,SAAQ,MAAM,MAAM,GAAG,aAAa,sBAAsB;AAC1D,cAAM;AAAA,MACR;AAGA,UAAI,MAAM,SAAS,OAAO,MAAM,aAAa,0BAA0B;AACrE,QAAAA,SAAQ,MAAM;AAAA,UACZ,GAAG,aAAa;AAAA,QAClB;AAEA,YAAI;AACF,gBAAM,SAAS,WAAW;AAC1B,eAAK,YAAY;AAAA,QACnB,SAAS,iBAAiB;AACxB,UAAAA,SAAQ,MAAM,MAAM,gDAAgD,eAAe;AAAA,QACrF;AAEA,cAAM,IAAI,QAAQ,CAAC,YAAY,WAAW,SAAS,GAAI,CAAC;AACxD,cAAM,KAAK,iBAAiB;AAE5B,YAAI;AACF,iBAAO,MAAM,UAAU;AAAA,QACzB,SAAS,YAAiB;AACxB,UAAAA,SAAQ,MAAM,MAAM,GAAG,aAAa,wCAAwC,UAAU;AACtF,gBAAM;AAAA,QACR;AAAA,MACF;AAGA,UACE,MAAM,SAAS,uBACf,MAAM,SAAS,sBACf,MAAM,SAAS,SAAS,YAAY,KACpC,MAAM,SAAS,SAAS,YAAY,GACpC;AACA,QAAAA,SAAQ,MAAM;AAAA,UACZ,GAAG,aAAa;AAAA,QAClB;AAEA,cAAM,KAAK,iBAAiB;AAG5B,YAAI;AACF,iBAAO,MAAM,UAAU;AAAA,QACzB,SAAS,YAAiB;AACxB,UAAAA,SAAQ,MAAM,MAAM,GAAG,aAAa,wBAAwB,UAAU;AACtE,gBAAM;AAAA,QACR;AAAA,MACF;AAGA,YAAM;AAAA,IACR;AAAA,EACF;AACF;","names":["loggers","loggers"]}

package/dist/models/messages.model.cjs

@@ -84,6 +84,7 @@ var MessageObjectSchema = new import_mongoose.Schema(
     timestamps: true
   }
 );
+MessageObjectSchema.index({ threadId: 1, messageId: 1 }, { unique: true });
 var MessageModel = import_mongoose2.default.model("Message", MessageObjectSchema);
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {

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

@@ -1 +1 @@
-{"version":3,"sources":["../../models/messages.model.ts"],"sourcesContent":["import { MessageRole } from \"@ainetwork/adk/types/memory\";\nimport { type Document, Schema } from \"mongoose\";\nimport mongoose from \"mongoose\";\n\n// MessageContentObject schema\nexport const MessageContentObjectSchema = new Schema(\n\t{\n\t\ttype: { type: String, required: true },\n\t\tparts: { type: [Schema.Types.Mixed], required: true },\n\t},\n\t{ _id: false },\n);\n\n// MessageObject schema - 개별 문서로 저장\nexport const MessageObjectSchema = new Schema(\n\t{\n\t\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\t{\n\t\ttimestamps: true,\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\tcreatedAt: Date;\n\tupdatedAt: Date;\n\tmetadata?: { [key: string]: unknown };\n}\n\nexport const MessageModel = mongoose.model<MessageDocument>(\"Message\", MessageObjectSchema);"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,oBAA4B;AAC5B,sBAAsC;AACtC,IAAAA,mBAAqB;AAGd,IAAM,6BAA6B,IAAI;AAAA,EAC7C;AAAA,IACC,MAAM,EAAE,MAAM,QAAQ,UAAU,KAAK;AAAA,IACrC,OAAO,EAAE,MAAM,CAAC,uBAAO,MAAM,KAAK,GAAG,UAAU,KAAK;AAAA,EACrD;AAAA,EACA,EAAE,KAAK,MAAM;AACd;AAGO,IAAM,sBAAsB,IAAI;AAAA,EACtC;AAAA,IACC,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,yBAAW;AAAA,MAC/B,UAAU;AAAA,IACX;AAAA,IACA,SAAS;AAAA,MACR,MAAM;AAAA,MACN,UAAU;AAAA,IACX;AAAA,IACA,WAAW;AAAA,MACV,MAAM;AAAA,MACN,UAAU;AAAA,IACX;AAAA,IACA,UAAU;AAAA,MACT,MAAM,uBAAO,MAAM;AAAA,MACnB,SAAS,CAAC;AAAA,IACX;AAAA,EACD;AAAA,EACA;AAAA,IACC,YAAY;AAAA,EACb;AACD;AAkBO,IAAM,eAAe,iBAAAC,QAAS,MAAuB,WAAW,mBAAmB;","names":["import_mongoose","mongoose"]}
+{"version":3,"sources":["../../models/messages.model.ts"],"sourcesContent":["import { MessageRole } from \"@ainetwork/adk/types/memory\";\nimport { type Document, Schema } from \"mongoose\";\nimport mongoose from \"mongoose\";\n\n// MessageContentObject schema\nexport const MessageContentObjectSchema = new Schema(\n\t{\n\t\ttype: { type: String, required: true },\n\t\tparts: { type: [Schema.Types.Mixed], required: true },\n\t},\n\t{ _id: false },\n);\n\n// MessageObject schema - 개별 문서로 저장\nexport const MessageObjectSchema = new Schema(\n\t{\n\t\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\t{\n\t\ttimestamps: true,\n\t},\n);\n\nMessageObjectSchema.index({ threadId: 1, messageId: 1 }, { unique: true });\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\tcreatedAt: Date;\n\tupdatedAt: Date;\n\tmetadata?: { [key: string]: unknown };\n}\n\nexport const MessageModel = mongoose.model<MessageDocument>(\"Message\", MessageObjectSchema);"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,oBAA4B;AAC5B,sBAAsC;AACtC,IAAAA,mBAAqB;AAGd,IAAM,6BAA6B,IAAI;AAAA,EAC7C;AAAA,IACC,MAAM,EAAE,MAAM,QAAQ,UAAU,KAAK;AAAA,IACrC,OAAO,EAAE,MAAM,CAAC,uBAAO,MAAM,KAAK,GAAG,UAAU,KAAK;AAAA,EACrD;AAAA,EACA,EAAE,KAAK,MAAM;AACd;AAGO,IAAM,sBAAsB,IAAI;AAAA,EACtC;AAAA,IACC,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,yBAAW;AAAA,MAC/B,UAAU;AAAA,IACX;AAAA,IACA,SAAS;AAAA,MACR,MAAM;AAAA,MACN,UAAU;AAAA,IACX;AAAA,IACA,WAAW;AAAA,MACV,MAAM;AAAA,MACN,UAAU;AAAA,IACX;AAAA,IACA,UAAU;AAAA,MACT,MAAM,uBAAO,MAAM;AAAA,MACnB,SAAS,CAAC;AAAA,IACX;AAAA,EACD;AAAA,EACA;AAAA,IACC,YAAY;AAAA,EACb;AACD;AAEA,oBAAoB,MAAM,EAAE,UAAU,GAAG,WAAW,EAAE,GAAG,EAAE,QAAQ,KAAK,CAAC;AAkBlE,IAAM,eAAe,iBAAAC,QAAS,MAAuB,WAAW,mBAAmB;","names":["import_mongoose","mongoose"]}

package/dist/models/messages.model.js

@@ -2,7 +2,7 @@ import {
   MessageContentObjectSchema,
   MessageModel,
   MessageObjectSchema
-} from "../chunk-T5WRFXBY.js";
+} from "../chunk-QULDFKGZ.js";
 export {
   MessageContentObjectSchema,
   MessageModel,

package/dist/models/threads.model.cjs

@@ -62,6 +62,11 @@ var ThreadObjectSchema = new import_mongoose.Schema(
       type: Boolean,
       required: false,
       default: false
+    },
+    workflowId: {
+      type: String,
+      required: false,
+      index: true
     }
   },
   {

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

@@ -1 +1 @@
-{"version":3,"sources":["../../models/threads.model.ts"],"sourcesContent":["import { ThreadType } from \"@ainetwork/adk/types/memory\";\nimport { type Document, Schema } from \"mongoose\";\nimport mongoose from \"mongoose\";\n\nexport const ThreadObjectSchema = new Schema(\n\t{\n\t\ttype: {\n\t\t\ttype: String,\n\t\t\tenum: Object.values(ThreadType),\n\t\t\trequired: true,\n\t\t},\n\t\tthreadId: {\n\t\t\ttype: String,\n\t\t\trequired: true,\n\t\t\tindex: true,\n\t\t},\n\t\tuserId: {\n\t\t\ttype: String,\n\t\t\trequired: true,\n\t\t\tindex: true,\n\t\t},\n\t\ttitle: {\n\t\t\ttype: String,\n\t\t\trequired: false,\n\t\t},\n\t\tisPinned: {\n\t\t\ttype: Boolean,\n\t\t\trequired: false,\n\t\t\tdefault: false,\n\t\t},\n\t},\n\t{\n\t\ttimestamps: true,\n\t},\n);\n\nexport interface ThreadDocument extends Document {\n\ttype: ThreadType;\n\tthreadId: string;\n\tuserId: string;\n\ttitle: string;\n\tisPinned: boolean;\n\tcreatedAt: Date;\n\tupdatedAt: Date;\n}\n\nexport const ThreadModel = mongoose.model<ThreadDocument>(\"Thread\", ThreadObjectSchema);\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,oBAA2B;AAC3B,sBAAsC;AACtC,IAAAA,mBAAqB;AAEd,IAAM,qBAAqB,IAAI;AAAA,EACrC;AAAA,IACC,MAAM;AAAA,MACL,MAAM;AAAA,MACN,MAAM,OAAO,OAAO,wBAAU;AAAA,MAC9B,UAAU;AAAA,IACX;AAAA,IACA,UAAU;AAAA,MACT,MAAM;AAAA,MACN,UAAU;AAAA,MACV,OAAO;AAAA,IACR;AAAA,IACA,QAAQ;AAAA,MACP,MAAM;AAAA,MACN,UAAU;AAAA,MACV,OAAO;AAAA,IACR;AAAA,IACA,OAAO;AAAA,MACN,MAAM;AAAA,MACN,UAAU;AAAA,IACX;AAAA,IACA,UAAU;AAAA,MACT,MAAM;AAAA,MACN,UAAU;AAAA,MACV,SAAS;AAAA,IACV;AAAA,EACD;AAAA,EACA;AAAA,IACC,YAAY;AAAA,EACb;AACD;AAYO,IAAM,cAAc,iBAAAC,QAAS,MAAsB,UAAU,kBAAkB;","names":["import_mongoose","mongoose"]}
+{"version":3,"sources":["../../models/threads.model.ts"],"sourcesContent":["import { ThreadType } from \"@ainetwork/adk/types/memory\";\nimport { type Document, Schema } from \"mongoose\";\nimport mongoose from \"mongoose\";\n\nexport const ThreadObjectSchema = new Schema(\n\t{\n\t\ttype: {\n\t\t\ttype: String,\n\t\t\tenum: Object.values(ThreadType),\n\t\t\trequired: true,\n\t\t},\n\t\tthreadId: {\n\t\t\ttype: String,\n\t\t\trequired: true,\n\t\t\tindex: true,\n\t\t},\n\t\tuserId: {\n\t\t\ttype: String,\n\t\t\trequired: true,\n\t\t\tindex: true,\n\t\t},\n\t\ttitle: {\n\t\t\ttype: String,\n\t\t\trequired: false,\n\t\t},\n\t\tisPinned: {\n\t\t\ttype: Boolean,\n\t\t\trequired: false,\n\t\t\tdefault: false,\n\t\t},\n\t\tworkflowId: {\n\t\t\ttype: String,\n\t\t\trequired: false,\n\t\t\tindex: true,\n\t\t},\n\t},\n\t{\n\t\ttimestamps: true,\n\t},\n);\n\nexport interface ThreadDocument extends Document {\n\ttype: ThreadType;\n\tthreadId: string;\n\tuserId: string;\n\ttitle: string;\n\tisPinned: boolean;\n\tworkflowId?: string;\n\tcreatedAt: Date;\n\tupdatedAt: Date;\n}\n\nexport const ThreadModel = mongoose.model<ThreadDocument>(\"Thread\", ThreadObjectSchema);\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,oBAA2B;AAC3B,sBAAsC;AACtC,IAAAA,mBAAqB;AAEd,IAAM,qBAAqB,IAAI;AAAA,EACrC;AAAA,IACC,MAAM;AAAA,MACL,MAAM;AAAA,MACN,MAAM,OAAO,OAAO,wBAAU;AAAA,MAC9B,UAAU;AAAA,IACX;AAAA,IACA,UAAU;AAAA,MACT,MAAM;AAAA,MACN,UAAU;AAAA,MACV,OAAO;AAAA,IACR;AAAA,IACA,QAAQ;AAAA,MACP,MAAM;AAAA,MACN,UAAU;AAAA,MACV,OAAO;AAAA,IACR;AAAA,IACA,OAAO;AAAA,MACN,MAAM;AAAA,MACN,UAAU;AAAA,IACX;AAAA,IACA,UAAU;AAAA,MACT,MAAM;AAAA,MACN,UAAU;AAAA,MACV,SAAS;AAAA,IACV;AAAA,IACA,YAAY;AAAA,MACX,MAAM;AAAA,MACN,UAAU;AAAA,MACV,OAAO;AAAA,IACR;AAAA,EACD;AAAA,EACA;AAAA,IACC,YAAY;AAAA,EACb;AACD;AAaO,IAAM,cAAc,iBAAAC,QAAS,MAAsB,UAAU,kBAAkB;","names":["import_mongoose","mongoose"]}