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

package/dist/index.js

@@ -6,13 +6,13 @@ import {
 } from "./chunk-YFW7JXII.js";
 import {
   MessageModel
-} from "./chunk-5CCEN7NK.js";
+} from "./chunk-T5WRFXBY.js";
 import {
   ThreadModel
-} from "./chunk-GPOFS7ZT.js";
+} from "./chunk-EPJZT2YE.js";
 import {
   WorkflowModel
-} from "./chunk-NGLXQZLX.js";
+} from "./chunk-ENGJ6NTO.js";
 
 // implements/base.memory.ts
 import mongoose from "mongoose";
@@ -179,6 +179,7 @@ var MongoDBThread = class {
         userId: thread.userId,
         type: thread.type,
         title: thread.title || "New thread",
+        isPinned: thread.isPinned ?? false,
         messages: []
       };
       messages.forEach((message) => {
@@ -195,23 +196,18 @@ var MongoDBThread = class {
   }
   async createThread(type, userId, threadId, title) {
     return this.executeWithRetry(async () => {
-      const now = Date.now();
       await ThreadModel.create({
         type,
         userId,
         threadId,
-        title,
-        updated_at: now,
-        created_at: now
+        title
       });
       return { type, userId, threadId, title, messages: [] };
     }, `createThread(${userId}, ${threadId})`);
   }
   async addMessagesToThread(userId, threadId, messages) {
     return this.executeWithRetry(async () => {
-      await ThreadModel.updateOne({ threadId, userId }, {
-        updated_at: Date.now()
-      });
+      await ThreadModel.updateOne({ threadId, userId }, { $set: {} });
       for (const message of messages) {
         await MessageModel.create({
           threadId,
@@ -235,19 +231,29 @@ var MongoDBThread = class {
   async listThreads(userId) {
     return this.executeWithRetry(async () => {
       const timeout = this.getOperationTimeout();
-      const threads = await ThreadModel.find({ userId }).sort({ updated_at: -1 }).maxTimeMS(timeout);
+      const threads = await ThreadModel.find({ userId }).sort({ updatedAt: -1 }).maxTimeMS(timeout);
       const data = threads.map((thread) => {
         return {
           type: thread.type,
           userId,
           threadId: thread.threadId,
           title: thread.title,
-          updatedAt: thread.updated_at
+          isPinned: thread.isPinned ?? false,
+          updatedAt: thread.updatedAt
         };
       });
       return data;
     }, `listThreads(${userId})`);
   }
+  async updateThreadPin(userId, threadId, isPinned) {
+    return this.executeWithRetry(async () => {
+      const timeout = this.getOperationTimeout();
+      await ThreadModel.updateOne(
+        { threadId, userId },
+        { $set: { isPinned } }
+      ).maxTimeMS(timeout);
+    }, `updateThreadPin(${userId}, ${threadId})`);
+  }
 };
 
 // implements/workflow.memory.ts
@@ -314,6 +320,8 @@ var MongoDBMemory = class _MongoDBMemory {
   connectionConfig;
   eventListenersSetup = false;
   operationTimeoutMS;
+  threadTTLSeconds;
+  orphanCleanupTimer;
   agentMemory;
   intentMemory;
   threadMemory;
@@ -324,6 +332,9 @@ var MongoDBMemory = class _MongoDBMemory {
     this.maxReconnectAttempts = cfg.maxReconnectAttempts ?? 5;
     this.reconnectInterval = cfg.reconnectInterval ?? 5e3;
     this.operationTimeoutMS = cfg.operationTimeoutMS ?? 1e4;
+    if (cfg.threadTTLSeconds !== void 0 && cfg.threadTTLSeconds > 0) {
+      this.threadTTLSeconds = cfg.threadTTLSeconds;
+    }
     this.connectionConfig = {
       maxPoolSize: cfg.maxPoolSize ?? 1,
       serverSelectionTimeoutMS: cfg.serverSelectionTimeoutMS ?? 3e4,
@@ -437,6 +448,8 @@ var MongoDBMemory = class _MongoDBMemory {
       await mongoose.connect(this.uri, this.connectionConfig);
       this.connected = true;
       this.reconnectAttempts = 0;
+      await this.setupTTLIndex();
+      this.startOrphanCleanup();
     } catch (error) {
       loggers2.agent.error("Failed to connect to MongoDB:", error);
       throw error;
@@ -447,6 +460,10 @@ var MongoDBMemory = class _MongoDBMemory {
       return;
     }
     try {
+      if (this.orphanCleanupTimer) {
+        clearInterval(this.orphanCleanupTimer);
+        this.orphanCleanupTimer = void 0;
+      }
       await mongoose.disconnect();
       this.connected = false;
     } catch (error) {
@@ -470,6 +487,62 @@ var MongoDBMemory = class _MongoDBMemory {
       throw new Error("MongoDB is not connected and reconnection failed");
     }
   }
+  async setupTTLIndex() {
+    if (this.threadTTLSeconds === void 0) return;
+    try {
+      const db = mongoose.connection.db;
+      if (!db) return;
+      const collection = db.collection("threads");
+      const indexes = await collection.indexes();
+      const existingTTL = indexes.find(
+        (idx) => idx.key?.updatedAt !== void 0 && idx.expireAfterSeconds !== void 0
+      );
+      if (existingTTL) {
+        if (existingTTL.expireAfterSeconds !== this.threadTTLSeconds) {
+          await db.command({
+            collMod: "threads",
+            index: { keyPattern: { updatedAt: 1 }, expireAfterSeconds: this.threadTTLSeconds }
+          });
+          loggers2.agent.info(`Thread TTL index updated to ${this.threadTTLSeconds} seconds`);
+        }
+      } else {
+        await collection.createIndex(
+          { updatedAt: 1 },
+          { expireAfterSeconds: this.threadTTLSeconds }
+        );
+        loggers2.agent.info(`Thread TTL index created with ${this.threadTTLSeconds} seconds`);
+      }
+    } catch (error) {
+      loggers2.agent.error("Failed to setup TTL index:", error);
+    }
+  }
+  startOrphanCleanup() {
+    if (this.threadTTLSeconds === void 0) return;
+    if (this.orphanCleanupTimer) return;
+    const intervalMs = Math.max(6e4, Math.min(this.threadTTLSeconds * 500, 36e5));
+    this.orphanCleanupTimer = setInterval(() => {
+      this.cleanupOrphanedMessages().catch((error) => {
+        loggers2.agent.error("Orphaned message cleanup failed:", error);
+      });
+    }, intervalMs);
+    loggers2.agent.info(`Orphaned message cleanup scheduled every ${Math.round(intervalMs / 1e3)}s`);
+  }
+  async cleanupOrphanedMessages() {
+    if (!this.connected) return;
+    try {
+      const db = mongoose.connection.db;
+      if (!db) return;
+      const existingThreadIds = await db.collection("threads").distinct("threadId");
+      const result = await MessageModel.deleteMany({
+        threadId: { $nin: existingThreadIds }
+      });
+      if (result.deletedCount > 0) {
+        loggers2.agent.info(`Cleaned up ${result.deletedCount} orphaned messages`);
+      }
+    } catch (error) {
+      loggers2.agent.error("Failed to cleanup orphaned messages:", error);
+    }
+  }
   /**
    * Get the operation timeout in milliseconds
    */

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\";\n\nexport interface MongoDBMemoryConfig {\n  uri: string;\n  maxReconnectAttempts?: number;\n  reconnectInterval?: number;\n  maxPoolSize?: number;\n  serverSelectionTimeoutMS?: number;\n  socketTimeoutMS?: number;\n  connectTimeoutMS?: number;\n  operationTimeoutMS?: number; // Timeout for database operations\n}\n\nexport class MongoDBMemory implements IMemory {\n  private static instance: MongoDBMemory;\n  private uri: string;\n  private connected: boolean = false;\n  private reconnectAttempts: number = 0;\n  private maxReconnectAttempts: number;\n  private reconnectInterval: number;\n  private reconnecting: boolean = false;\n  private connectionConfig: mongoose.ConnectOptions;\n  private eventListenersSetup: boolean = false;\n  private operationTimeoutMS: number;\n\n  private agentMemory: MongoDBAgent;\n  private intentMemory: MongoDBIntent;\n  private threadMemory: MongoDBThread;\n  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    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    } catch (error) {\n      loggers.agent.error(\"Failed to connect to MongoDB:\", error);\n      throw error;\n    }\n  }\n\n  public async disconnect(): Promise<void> {\n    if (!this.isConnected) {\n      return;\n    }\n\n    try {\n      await mongoose.disconnect();\n      this.connected = false;\n    } catch (error) {\n      loggers.agent.error(\"Failed to disconnect from MongoDB:\", error);\n      throw error;\n    }\n  }\n\n  public isConnected(): boolean {\n    return this.connected;\n  }\n\n  private async ensureConnection(): Promise<void> {\n    if (!this.isConnected && !this.reconnecting) {\n      await this.connect();\n    }\n\n    // Wait for reconnection if in progress\n    const maxWaitTime = 30000; // 30 seconds\n    const startTime = Date.now();\n    while (this.reconnecting && Date.now() - startTime < maxWaitTime) {\n      await new Promise((resolve) => setTimeout(resolve, 100));\n    }\n\n    if (!this.isConnected) {\n      throw new Error(\"MongoDB is not connected and reconnection failed\");\n    }\n  }\n\n  /**\n   * Get the operation timeout in milliseconds\n   */\n  protected getOperationTimeout(): number {\n    return this.operationTimeoutMS;\n  }\n\n  /**\n   * Execute a database operation with automatic retry on connection errors\n   * Note: Use mongoose's maxTimeMS option in queries for timeout control\n   */\n  protected async executeWithRetry<T>(\n    operation: () => Promise<T>,\n    operationName: string = \"Database operation\"\n  ): Promise<T> {\n    await this.ensureConnection();\n\n    try {\n      return await operation();\n    } catch (error: any) {\n      // Check if it's a timeout error from MongoDB\n      if (error.code === 50 || error.message?.includes(\"operation exceeded time limit\")) {\n        loggers.agent.error(`${operationName} exceeded time limit`);\n        throw error;\n      }\n\n      // Check if it's a connection-related error\n      if (\n        error.name === \"MongoNetworkError\" ||\n        error.name === \"MongoServerError\" ||\n        error.message?.includes(\"connection\") ||\n        error.message?.includes(\"disconnect\")\n      ) {\n        loggers.agent.warn(\n          `${operationName} failed due to connection issue, attempting reconnection...`\n        );\n\n        await this.ensureConnection();\n\n        // Retry the operation once after reconnection\n        try {\n          return await operation();\n        } catch (retryError: any) {\n          loggers.agent.error(`${operationName} failed after retry:`, retryError);\n          throw retryError;\n        }\n      }\n\n      // If it's not a connection error, just throw it\n      throw error;\n    }\n  }\n}\n","import { 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        messages: []\n      };\n      messages.forEach((message: MessageDocument) => {\n        threadObject.messages.push({\n          messageId: message.messageId,\n          role: message.role as MessageRole,\n          content: message.content,\n          timestamp: message.timestamp,\n          metadata: message.metadata,\n        });\n      });\n\n      return threadObject;\n    }, `getThread(${userId}, ${threadId})`);\n  };\n\n  public async createThread(\n    type: ThreadType,\n    userId: string,\n    threadId: string,\n    title: string,\n  ): Promise<ThreadObject> {\n    return this.executeWithRetry(async () => {\n      const now = Date.now();\n      await ThreadModel.create({\n        type,\n        userId,\n        threadId,\n        title,\n        updated_at: now,\n        created_at: now,\n      });\n\n      return { type, userId, threadId, title, messages: []};\n    }, `createThread(${userId}, ${threadId})`);\n  };\n\n  public async addMessagesToThread(\n    userId: string,\n    threadId: string,\n    messages: MessageObject[]\n  ): Promise<void> {\n    return this.executeWithRetry(async () => {\n      await ThreadModel.updateOne({ threadId, userId }, {\n        updated_at: Date.now(),\n      });\n      for (const message of messages) {\n        await MessageModel.create({\n          threadId,\n          messageId: message.messageId,\n          userId,\n          role: message.role,\n          content: message.content,\n          timestamp: message.timestamp,\n          metadata: message.metadata,\n        });\n      }\n    }, `addMessagesToThread(${userId}, ${threadId})`);\n  };\n\n  public async deleteThread(userId: string, threadId: string): Promise<void> {\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n\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({ updated_at: -1 })\n        .maxTimeMS(timeout);\n      const data: ThreadMetadata[] = threads.map((thread: ThreadDocument) => {\n        return {\n          type: thread.type,\n          userId,\n          threadId: thread.threadId,\n          title: thread.title,\n          updatedAt: thread.updated_at\n        } as ThreadMetadata;\n      })\n      return data;\n    }, `listThreads(${userId})`);\n  };\n}\n","import 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,CAAC;AAAA,MACb;AACA,eAAS,QAAQ,CAAC,YAA6B;AAC7C,qBAAa,SAAS,KAAK;AAAA,UACzB,WAAW,QAAQ;AAAA,UACnB,MAAM,QAAQ;AAAA,UACd,SAAS,QAAQ;AAAA,UACjB,WAAW,QAAQ;AAAA,UACnB,UAAU,QAAQ;AAAA,QACpB,CAAC;AAAA,MACH,CAAC;AAED,aAAO;AAAA,IACT,GAAG,aAAa,MAAM,KAAK,QAAQ,GAAG;AAAA,EACxC;AAAA,EAEA,MAAa,aACX,MACA,QACA,UACA,OACuB;AACvB,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,MAAM,KAAK,IAAI;AACrB,YAAM,YAAY,OAAO;AAAA,QACvB;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA,YAAY;AAAA,QACZ,YAAY;AAAA,MACd,CAAC;AAED,aAAO,EAAE,MAAM,QAAQ,UAAU,OAAO,UAAU,CAAC,EAAC;AAAA,IACtD,GAAG,gBAAgB,MAAM,KAAK,QAAQ,GAAG;AAAA,EAC3C;AAAA,EAEA,MAAa,oBACX,QACA,UACA,UACe;AACf,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,YAAY,UAAU,EAAE,UAAU,OAAO,GAAG;AAAA,QAChD,YAAY,KAAK,IAAI;AAAA,MACvB,CAAC;AACD,iBAAW,WAAW,UAAU;AAC9B,cAAM,aAAa,OAAO;AAAA,UACxB;AAAA,UACA,WAAW,QAAQ;AAAA,UACnB;AAAA,UACA,MAAM,QAAQ;AAAA,UACd,SAAS,QAAQ;AAAA,UACjB,WAAW,QAAQ;AAAA,UACnB,UAAU,QAAQ;AAAA,QACpB,CAAC;AAAA,MACH;AAAA,IACF,GAAG,uBAAuB,MAAM,KAAK,QAAQ,GAAG;AAAA,EAClD;AAAA,EAEA,MAAa,aAAa,QAAgB,UAAiC;AACzE,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,UAAU,KAAK,oBAAoB;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,YAAY,GAAG,CAAC,EACvB,UAAU,OAAO;AACpB,YAAM,OAAyB,QAAQ,IAAI,CAAC,WAA2B;AACrE,eAAO;AAAA,UACL,MAAM,OAAO;AAAA,UACb;AAAA,UACA,UAAU,OAAO;AAAA,UACjB,OAAO,OAAO;AAAA,UACd,WAAW,OAAO;AAAA,QACpB;AAAA,MACF,CAAC;AACD,aAAO;AAAA,IACT,GAAG,eAAe,MAAM,GAAG;AAAA,EAC7B;AACF;;;AC7HO,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;;;AJ1DO,IAAM,gBAAN,MAAM,eAAiC;AAAA,EAC5C,OAAe;AAAA,EACP;AAAA,EACA,YAAqB;AAAA,EACrB,oBAA4B;AAAA,EAC5B;AAAA,EACA;AAAA,EACA,eAAwB;AAAA,EACxB;AAAA,EACA,sBAA+B;AAAA,EAC/B;AAAA,EAEA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EAER,YAAY,QAAsC;AAChD,UAAM,MAAM,OAAO,WAAW,WAAW,EAAE,KAAK,OAAO,IAAI;AAE3D,SAAK,MAAM,IAAI;AACf,SAAK,uBAAuB,IAAI,wBAAwB;AACxD,SAAK,oBAAoB,IAAI,qBAAqB;AAClD,SAAK,qBAAqB,IAAI,sBAAsB;AACpD,SAAK,mBAAmB;AAAA,MACtB,aAAa,IAAI,eAAe;AAAA,MAChC,0BAA0B,IAAI,4BAA4B;AAAA,MAC1D,iBAAiB,IAAI,mBAAmB;AAAA,MACxC,kBAAkB,IAAI,oBAAoB;AAAA,MAC1C,gBAAgB;AAAA,IAClB;AAEA,QAAI,CAAC,eAAc,UAAU;AAC3B,qBAAc,WAAW;AACzB,WAAK,4BAA4B;AAAA,IACnC,OAAO;AAEL,WAAK,YAAY,eAAc,SAAS;AACxC,WAAK,qBAAqB,eAAc,SAAS;AAAA,IACnD;AAEF,SAAK,cAAc,IAAI;AAAA,MACtB,KAAK,iBAAiB,KAAK,IAAI;AAAA,MAC/B,KAAK,oBAAoB,KAAK,IAAI;AAAA,IACnC;AAEA,SAAK,eAAe,IAAI;AAAA,MACvB,KAAK,iBAAiB,KAAK,IAAI;AAAA,MAC/B,KAAK,oBAAoB,KAAK,IAAI;AAAA,IACnC;AAEA,SAAK,eAAe,IAAI;AAAA,MACvB,KAAK,iBAAiB,KAAK,IAAI;AAAA,MAC/B,KAAK,oBAAoB,KAAK,IAAI;AAAA,IACnC;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;AAAA,IAC3B,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,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;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/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"]}

package/dist/models/messages.model.cjs

@@ -79,6 +79,9 @@ var MessageObjectSchema = new import_mongoose.Schema(
       type: import_mongoose.Schema.Types.Mixed,
       default: {}
     }
+  },
+  {
+    timestamps: true
   }
 );
 var MessageModel = import_mongoose2.default.model("Message", MessageObjectSchema);

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);\n\n// Message Document interface\nexport interface MessageDocument extends Document {\n\tmessageId: string;\n\tthreadId: string;\n\tuserId: string;\n\trole: MessageRole;\n\tcontent: {\n\t\ttype: string;\n\t\tparts: any[];\n\t};\n\ttimestamp: number;\n\tmetadata?: { [key: string]: unknown };\n}\n\nexport const MessageModel = mongoose.model<MessageDocument>(\"Message\", MessageObjectSchema);"],"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;AACD;AAgBO,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\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"]}

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

@@ -17,7 +17,12 @@ declare const MessageContentObjectSchema: Schema<any, mongoose.Model<any, any, a
 } & {
     __v: number;
 }>;
-declare const MessageObjectSchema: Schema<any, mongoose.Model<any, any, any, any, any, any>, {}, {}, {}, {}, mongoose.DefaultSchemaOptions, {
+declare const MessageObjectSchema: Schema<any, mongoose.Model<any, any, any, any, any, any>, {}, {}, {}, {}, {
+    timestamps: true;
+}, {
+    createdAt: NativeDate;
+    updatedAt: NativeDate;
+} & {
     threadId: string;
     userId: string;
     messageId: string;
@@ -29,6 +34,9 @@ declare const MessageObjectSchema: Schema<any, mongoose.Model<any, any, any, any
     timestamp: number;
     metadata: any;
 }, Document<unknown, {}, mongoose.FlatRecord<{
+    createdAt: NativeDate;
+    updatedAt: NativeDate;
+} & {
     threadId: string;
     userId: string;
     messageId: string;
@@ -40,6 +48,9 @@ declare const MessageObjectSchema: Schema<any, mongoose.Model<any, any, any, any
     timestamp: number;
     metadata: any;
 }>, {}> & mongoose.FlatRecord<{
+    createdAt: NativeDate;
+    updatedAt: NativeDate;
+} & {
     threadId: string;
     userId: string;
     messageId: string;
@@ -65,6 +76,8 @@ interface MessageDocument extends Document {
         parts: any[];
     };
     timestamp: number;
+    createdAt: Date;
+    updatedAt: Date;
     metadata?: {
         [key: string]: unknown;
     };

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

@@ -17,7 +17,12 @@ declare const MessageContentObjectSchema: Schema<any, mongoose.Model<any, any, a
 } & {
     __v: number;
 }>;
-declare const MessageObjectSchema: Schema<any, mongoose.Model<any, any, any, any, any, any>, {}, {}, {}, {}, mongoose.DefaultSchemaOptions, {
+declare const MessageObjectSchema: Schema<any, mongoose.Model<any, any, any, any, any, any>, {}, {}, {}, {}, {
+    timestamps: true;
+}, {
+    createdAt: NativeDate;
+    updatedAt: NativeDate;
+} & {
     threadId: string;
     userId: string;
     messageId: string;
@@ -29,6 +34,9 @@ declare const MessageObjectSchema: Schema<any, mongoose.Model<any, any, any, any
     timestamp: number;
     metadata: any;
 }, Document<unknown, {}, mongoose.FlatRecord<{
+    createdAt: NativeDate;
+    updatedAt: NativeDate;
+} & {
     threadId: string;
     userId: string;
     messageId: string;
@@ -40,6 +48,9 @@ declare const MessageObjectSchema: Schema<any, mongoose.Model<any, any, any, any
     timestamp: number;
     metadata: any;
 }>, {}> & mongoose.FlatRecord<{
+    createdAt: NativeDate;
+    updatedAt: NativeDate;
+} & {
     threadId: string;
     userId: string;
     messageId: string;
@@ -65,6 +76,8 @@ interface MessageDocument extends Document {
         parts: any[];
     };
     timestamp: number;
+    createdAt: Date;
+    updatedAt: Date;
     metadata?: {
         [key: string]: unknown;
     };

package/dist/models/messages.model.js

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

package/dist/models/threads.model.cjs

@@ -58,14 +58,14 @@ var ThreadObjectSchema = new import_mongoose.Schema(
       type: String,
       required: false
     },
-    created_at: {
-      type: Number,
-      required: true
-    },
-    updated_at: {
-      type: Number,
-      required: true
+    isPinned: {
+      type: Boolean,
+      required: false,
+      default: false
     }
+  },
+  {
+    timestamps: true
   }
 );
 var ThreadModel = import_mongoose2.default.model("Thread", ThreadObjectSchema);

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\tcreated_at: {\n\t\t\ttype: Number,\n\t\t\trequired: true,\n\t\t},\n\t\tupdated_at: {\n\t\t\ttype: Number,\n\t\t\trequired: true,\n\t\t}\n\t},\n);\n\nexport interface ThreadDocument extends Document {\n\ttype: ThreadType;\n\tthreadId: string;\n\tuserId: string;\n\ttitle: string;\n\tcreated_at: number;\n\tupdated_at: number;\n}\n\nexport const ThreadModel = mongoose.model<ThreadDocument>(\"Thread\", ThreadObjectSchema);\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,oBAA2B;AAC3B,sBAAsC;AACtC,IAAAA,mBAAqB;AAEd,IAAM,qBAAqB,IAAI;AAAA,EACrC;AAAA,IACC,MAAM;AAAA,MACL,MAAM;AAAA,MACN,MAAM,OAAO,OAAO,wBAAU;AAAA,MAC9B,UAAU;AAAA,IACX;AAAA,IACA,UAAU;AAAA,MACT,MAAM;AAAA,MACN,UAAU;AAAA,MACV,OAAO;AAAA,IACR;AAAA,IACA,QAAQ;AAAA,MACP,MAAM;AAAA,MACN,UAAU;AAAA,MACV,OAAO;AAAA,IACR;AAAA,IACA,OAAO;AAAA,MACN,MAAM;AAAA,MACN,UAAU;AAAA,IACX;AAAA,IACA,YAAY;AAAA,MACX,MAAM;AAAA,MACN,UAAU;AAAA,IACX;AAAA,IACA,YAAY;AAAA,MACX,MAAM;AAAA,MACN,UAAU;AAAA,IACX;AAAA,EACD;AACD;AAWO,IAAM,cAAc,iBAAAC,QAAS,MAAsB,UAAU,kBAAkB;","names":["import_mongoose","mongoose"]}
+{"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"]}

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

@@ -1,27 +1,35 @@
 import { ThreadType } from '@ainetwork/adk/types/memory';
 import mongoose, { Schema, Document } from 'mongoose';
 
-declare const ThreadObjectSchema: Schema<any, mongoose.Model<any, any, any, any, any, any>, {}, {}, {}, {}, mongoose.DefaultSchemaOptions, {
+declare const ThreadObjectSchema: Schema<any, mongoose.Model<any, any, any, any, any, any>, {}, {}, {}, {}, {
+    timestamps: true;
+}, {
+    createdAt: NativeDate;
+    updatedAt: NativeDate;
+} & {
     type: ThreadType;
     threadId: string;
     userId: string;
-    created_at: number;
-    updated_at: number;
     title?: string | null | undefined;
+    isPinned?: boolean | null | undefined;
 }, Document<unknown, {}, mongoose.FlatRecord<{
+    createdAt: NativeDate;
+    updatedAt: NativeDate;
+} & {
     type: ThreadType;
     threadId: string;
     userId: string;
-    created_at: number;
-    updated_at: number;
     title?: string | null | undefined;
+    isPinned?: boolean | null | undefined;
 }>, {}> & mongoose.FlatRecord<{
+    createdAt: NativeDate;
+    updatedAt: NativeDate;
+} & {
     type: ThreadType;
     threadId: string;
     userId: string;
-    created_at: number;
-    updated_at: number;
     title?: string | null | undefined;
+    isPinned?: boolean | null | undefined;
 }> & {
     _id: mongoose.Types.ObjectId;
 } & {
@@ -32,8 +40,9 @@ interface ThreadDocument extends Document {
     threadId: string;
     userId: string;
     title: string;
-    created_at: number;
-    updated_at: number;
+    isPinned: boolean;
+    createdAt: Date;
+    updatedAt: Date;
 }
 declare const ThreadModel: mongoose.Model<ThreadDocument, {}, {}, {}, Document<unknown, {}, ThreadDocument, {}> & ThreadDocument & Required<{
     _id: unknown;

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

@@ -1,27 +1,35 @@
 import { ThreadType } from '@ainetwork/adk/types/memory';
 import mongoose, { Schema, Document } from 'mongoose';
 
-declare const ThreadObjectSchema: Schema<any, mongoose.Model<any, any, any, any, any, any>, {}, {}, {}, {}, mongoose.DefaultSchemaOptions, {
+declare const ThreadObjectSchema: Schema<any, mongoose.Model<any, any, any, any, any, any>, {}, {}, {}, {}, {
+    timestamps: true;
+}, {
+    createdAt: NativeDate;
+    updatedAt: NativeDate;
+} & {
     type: ThreadType;
     threadId: string;
     userId: string;
-    created_at: number;
-    updated_at: number;
     title?: string | null | undefined;
+    isPinned?: boolean | null | undefined;
 }, Document<unknown, {}, mongoose.FlatRecord<{
+    createdAt: NativeDate;
+    updatedAt: NativeDate;
+} & {
     type: ThreadType;
     threadId: string;
     userId: string;
-    created_at: number;
-    updated_at: number;
     title?: string | null | undefined;
+    isPinned?: boolean | null | undefined;
 }>, {}> & mongoose.FlatRecord<{
+    createdAt: NativeDate;
+    updatedAt: NativeDate;
+} & {
     type: ThreadType;
     threadId: string;
     userId: string;
-    created_at: number;
-    updated_at: number;
     title?: string | null | undefined;
+    isPinned?: boolean | null | undefined;
 }> & {
     _id: mongoose.Types.ObjectId;
 } & {
@@ -32,8 +40,9 @@ interface ThreadDocument extends Document {
     threadId: string;
     userId: string;
     title: string;
-    created_at: number;
-    updated_at: number;
+    isPinned: boolean;
+    createdAt: Date;
+    updatedAt: Date;
 }
 declare const ThreadModel: mongoose.Model<ThreadDocument, {}, {}, {}, Document<unknown, {}, ThreadDocument, {}> & ThreadDocument & Required<{
     _id: unknown;

package/dist/models/threads.model.js

@@ -1,7 +1,7 @@
 import {
   ThreadModel,
   ThreadObjectSchema
-} from "../chunk-GPOFS7ZT.js";
+} from "../chunk-EPJZT2YE.js";
 export {
   ThreadModel,
   ThreadObjectSchema

package/dist/models/workflow.model.cjs

@@ -62,6 +62,9 @@ var WorkflowObjectSchema = new import_mongoose.Schema(
     content: {
       type: String,
       required: true
+    },
+    variables: {
+      type: import_mongoose.Schema.Types.Mixed
     }
   },
   {