@ainetwork/adk-provider-memory-mongodb 0.3.4 → 0.3.5

package/dist/index.cjs

@@ -410,24 +410,27 @@ var MongoDBWorkflow = class {
     }, "getWorkflow()");
   }
   async updateWorkflow(workflowId, updates) {
+    if (!updates.userId) {
+      throw new Error("userId is required for updateWorkflow");
+    }
     return this.executeWithRetry(async () => {
       const timeout = this.getOperationTimeout();
       await WorkflowModel.updateOne(
-        { workflowId },
+        { workflowId, userId: updates.userId },
         { $set: updates }
       ).maxTimeMS(timeout);
     }, "updateWorkflow()");
   }
-  async deleteWorkflow(workflowId) {
+  async deleteWorkflow(workflowId, userId) {
     return this.executeWithRetry(async () => {
       const timeout = this.getOperationTimeout();
-      await WorkflowModel.deleteOne({ workflowId }).maxTimeMS(timeout);
+      await WorkflowModel.deleteOne({ workflowId, userId }).maxTimeMS(timeout);
     }, "deleteWorkflow()");
   }
   async listWorkflows(userId) {
     return this.executeWithRetry(async () => {
       const timeout = this.getOperationTimeout();
-      const query = userId ? { userId } : {};
+      const query = userId ? { $or: [{ userId }, { userId: { $exists: false } }, { userId: null }] } : { $or: [{ userId: { $exists: false } }, { userId: null }] };
       const workflows = await WorkflowModel.find(query).maxTimeMS(timeout).lean();
       return workflows;
     }, "listWorkflows()");

package/dist/index.cjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../index.ts","../implements/base.memory.ts","../models/agent.model.ts","../implements/agent.memory.ts","../models/intent.model.ts","../implements/intent.memory.ts","../models/threads.model.ts","../models/messages.model.ts","../implements/thread.memory.ts","../models/workflow.model.ts","../implements/workflow.memory.ts"],"sourcesContent":["export { MongoDBMemory } from \"./implements/base.memory\";\nexport type { MongoDBMemoryConfig } from \"./implements/base.memory\";","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 { type Document, Schema } from \"mongoose\";\nimport mongoose from \"mongoose\";\n\nexport const AgentObjectSchema = new Schema(\n\t{\n\t\tagent_prompt: {\n\t\t\ttype: String,\n\t\t},\n\t},\n);\n\nexport interface AgentDocument extends Document {\n\tagent_prompt: string;\n}\n\nexport const AgentModel = mongoose.model<AgentDocument>(\"Agent\", AgentObjectSchema);\n","import { IAgentMemory } from \"@ainetwork/adk/modules\";\nimport { AgentModel } from \"../models/agent.model\";\n\nexport type ExecuteWithRetryFn = <T>(\n  operation: () => Promise<T>,\n  operationName?: string\n) => Promise<T>;\n\nexport type GetOperationTimeoutFn = () => number;\n\ntype AgentMetadata = {\n  agent_prompt: string;\n}\n\nexport class MongoDBAgent implements IAgentMemory {\n  private executeWithRetry: ExecuteWithRetryFn;\n  private getOperationTimeout: GetOperationTimeoutFn;\n\n  constructor(\n    executeWithRetry: ExecuteWithRetryFn,\n    getOperationTimeout: GetOperationTimeoutFn\n  ) {\n    this.executeWithRetry = executeWithRetry;\n    this.getOperationTimeout = getOperationTimeout;\n  }\n\n  public async getAgentPrompt(): Promise<string> {\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      const metadata = await AgentModel.findOne({\n        id: \"agent_metadata\"\n      }).maxTimeMS(timeout)\n        .lean<AgentMetadata>();\n      return metadata?.agent_prompt || \"\";\n    }, \"getAgentPrompt()\");\n  };\n  \n  public async updateAgentPrompt(prompt: string): Promise<void> {\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      await AgentModel.updateOne({\n        id: \"agent_metadata\",\n      }, { \"agent_prompt\": prompt }).maxTimeMS(timeout);\n    }, \"updateAgentPrompt()\");\n  };\n}\n","import { Intent } from \"@ainetwork/adk/types/memory\";\nimport mongoose, { type Document, Schema } from \"mongoose\";\n\nconst IntentObjectSchema = new Schema(\n\t{\n\t\tid: { \n\t\t\ttype: String,\n\t\t\trequired: true,\n\t\t\tindex: true,\n\t\t\tunique: true,\n\t\t},\n\t\tname: {\n\t\t\ttype: String,\n\t\t\trequired: true,\n\t\t\tindex: true,\n\t\t},\n\t\tdescription: {\n\t\t\ttype: String,\n\t\t\trequired: true,\n\t\t},\n\t\tprompt: {\n\t\t\ttype: String,\n\t\t\trequired: false,\n\t\t},\n\t\tstatus: {\n\t\t\ttype: String,\n\t\t\trequired: true,\n\t\t},\n\t\ttriggeringSentences: {\n\t\t\ttype: [String],\n\t\t\trequired: false,\n\t\t},\n\t\ttags: {\n\t\t\ttype: [String],\n\t\t\trequired: false,\n\t\t},\n\t},\n);\n\nexport interface IntentDocument extends Omit<Document, 'id'>, Omit<Intent, 'id'> {\n\tid: string;\n}\n\nexport const IntentModel = mongoose.model<IntentDocument>(\"Intent\", IntentObjectSchema);","import type { Intent } from \"@ainetwork/adk/types/memory\";\nimport { IIntentMemory } from \"@ainetwork/adk/modules\";\nimport { IntentModel } from \"../models/intent.model\";\n\nexport type ExecuteWithRetryFn = <T>(\n  operation: () => Promise<T>,\n  operationName?: string\n) => Promise<T>;\n\nexport type GetOperationTimeoutFn = () => number;\n\nexport class MongoDBIntent implements IIntentMemory {\n  private executeWithRetry: ExecuteWithRetryFn;\n  private getOperationTimeout: GetOperationTimeoutFn;\n\n  constructor(\n    executeWithRetry: ExecuteWithRetryFn,\n    getOperationTimeout: GetOperationTimeoutFn\n  ) {\n    this.executeWithRetry = executeWithRetry;\n    this.getOperationTimeout = getOperationTimeout;\n  }\n\n  public async getIntent(intentId: string): Promise<Intent | undefined> {\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      const intent = await IntentModel.findOne({ id: intentId })\n        .maxTimeMS(timeout)\n        .lean<Intent>();\n      return intent || undefined;\n    }, `getIntent(${intentId})`);\n  };\n\n  public async getIntentByName(intentName: string): Promise<Intent | undefined> {\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      const intent = await IntentModel.findOne({ name: intentName })\n        .maxTimeMS(timeout)\n        .lean<Intent>();\n      return intent || undefined;\n    }, `getIntentByName(${intentName})`);\n  }\n\n  public async saveIntent(intent: Intent): Promise<void> {\n    return this.executeWithRetry(async () => {\n      await IntentModel.create(intent);\n    }, `saveIntent(${intent.id})`);\n  };\n\n  public async updateIntent(intentId: string, intent: Intent): Promise<void> {\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      await IntentModel.updateOne({\n        id: intentId,\n      }, intent).maxTimeMS(timeout);\n    }, `updateIntent(${intentId})`);\n  };\n\n  public async deleteIntent(intentId: string): Promise<void> {\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      await IntentModel.deleteOne({ id: intentId }).maxTimeMS(timeout);\n    }, `deleteIntent(${intentId})`);\n  };\n\n  public async listIntents(): Promise<Intent[]> {\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      const intents = await IntentModel.find()\n        .maxTimeMS(timeout)\n        .lean<Intent[]>();\n      return intents;\n    }, `listIntents()`);\n  };\n}\n","import { ThreadType } from \"@ainetwork/adk/types/memory\";\nimport { type Document, Schema } from \"mongoose\";\nimport mongoose from \"mongoose\";\n\nexport const ThreadObjectSchema = new Schema(\n\t{\n\t\ttype: {\n\t\t\ttype: String,\n\t\t\tenum: Object.values(ThreadType),\n\t\t\trequired: true,\n\t\t},\n\t\tthreadId: {\n\t\t\ttype: String,\n\t\t\trequired: true,\n\t\t\tindex: true,\n\t\t},\n\t\tuserId: {\n\t\t\ttype: String,\n\t\t\trequired: true,\n\t\t\tindex: true,\n\t\t},\n\t\ttitle: {\n\t\t\ttype: String,\n\t\t\trequired: false,\n\t\t},\n\t\tcreated_at: {\n\t\t\ttype: Number,\n\t\t\trequired: true,\n\t\t},\n\t\tupdated_at: {\n\t\t\ttype: Number,\n\t\t\trequired: true,\n\t\t}\n\t},\n);\n\nexport interface ThreadDocument extends Document {\n\ttype: ThreadType;\n\tthreadId: string;\n\tuserId: string;\n\ttitle: string;\n\tcreated_at: number;\n\tupdated_at: number;\n}\n\nexport const ThreadModel = mongoose.model<ThreadDocument>(\"Thread\", ThreadObjectSchema);\n","import { MessageRole } from \"@ainetwork/adk/types/memory\";\nimport { type Document, Schema } from \"mongoose\";\nimport mongoose from \"mongoose\";\n\n// MessageContentObject schema\nexport const MessageContentObjectSchema = new Schema(\n\t{\n\t\ttype: { type: String, required: true },\n\t\tparts: { type: [Schema.Types.Mixed], required: true },\n\t},\n\t{ _id: false },\n);\n\n// MessageObject schema - 개별 문서로 저장\nexport const MessageObjectSchema = new Schema(\n\t{\n\t\tmessageId: {\n\t\t\ttype: String,\n\t\t\trequired: true,\n\t\t\tindex: true,\n\t\t},\n\t\tthreadId: {\n\t\t\ttype: String,\n\t\t\trequired: true,\n\t\t\tindex: true,\n\t\t},\n\t\tuserId: {\n\t\t\ttype: String,\n\t\t\trequired: true,\n\t\t\tindex: true,\n\t\t},\n\t\trole: {\n\t\t\ttype: String,\n\t\t\tenum: Object.values(MessageRole),\n\t\t\trequired: true,\n\t\t},\n\t\tcontent: {\n\t\t\ttype: MessageContentObjectSchema,\n\t\t\trequired: true,\n\t\t},\n\t\ttimestamp: {\n\t\t\ttype: Number,\n\t\t\trequired: true,\n\t\t},\n\t\tmetadata: {\n\t\t\ttype: Schema.Types.Mixed,\n\t\t\tdefault: {},\n\t\t},\n\t},\n);\n\n// Message Document interface\nexport interface MessageDocument extends Document {\n\tmessageId: string;\n\tthreadId: string;\n\tuserId: string;\n\trole: MessageRole;\n\tcontent: {\n\t\ttype: string;\n\t\tparts: any[];\n\t};\n\ttimestamp: number;\n\tmetadata?: { [key: string]: unknown };\n}\n\nexport const MessageModel = mongoose.model<MessageDocument>(\"Message\", MessageObjectSchema);","import type { MessageObject, ThreadMetadata, ThreadObject, ThreadType } from \"@ainetwork/adk/types/memory\";\nimport { MessageRole } from \"@ainetwork/adk/types/memory\";\nimport { IThreadMemory } from \"@ainetwork/adk/modules\";\nimport { ThreadDocument, ThreadModel } from \"../models/threads.model\";\nimport { MessageDocument, MessageModel } from \"../models/messages.model\";\nimport { loggers } from \"@ainetwork/adk/utils/logger\";\n\nexport type ExecuteWithRetryFn = <T>(\n  operation: () => Promise<T>,\n  operationName?: string\n) => Promise<T>;\n\nexport type GetOperationTimeoutFn = () => number;\n\nexport class MongoDBThread implements IThreadMemory {\n  private executeWithRetry: ExecuteWithRetryFn;\n  private getOperationTimeout: GetOperationTimeoutFn;\n\n  constructor(\n    executeWithRetry: ExecuteWithRetryFn,\n    getOperationTimeout: GetOperationTimeoutFn\n  ) {\n    this.executeWithRetry = executeWithRetry;\n    this.getOperationTimeout = getOperationTimeout;\n  }\n\n  public async getThread(\n    userId: string,\n    threadId: string\n  ): Promise<ThreadObject | undefined> {\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      const thread = await ThreadModel.findOne({ threadId, userId }).maxTimeMS(timeout);\n      const messages = await MessageModel.find({ threadId, userId })\n        .sort({ timestamp: 1 })\n        .maxTimeMS(timeout);\n\n      if (!thread) return undefined;\n\n      loggers.agent.debug(`Found ${messages.length} messages for thread ${threadId}`);\n\n      const threadObject: ThreadObject = {\n        threadId: thread.threadId,\n        userId: thread.userId,\n        type: thread.type as ThreadType,\n        title: thread.title || \"New thread\",\n        messages: []\n      };\n      messages.forEach((message: MessageDocument) => {\n        threadObject.messages.push({\n          messageId: message.messageId,\n          role: message.role as MessageRole,\n          content: message.content,\n          timestamp: message.timestamp,\n          metadata: message.metadata,\n        });\n      });\n\n      return threadObject;\n    }, `getThread(${userId}, ${threadId})`);\n  };\n\n  public async createThread(\n    type: ThreadType,\n    userId: string,\n    threadId: string,\n    title: string,\n  ): Promise<ThreadObject> {\n    return this.executeWithRetry(async () => {\n      const now = Date.now();\n      await ThreadModel.create({\n        type,\n        userId,\n        threadId,\n        title,\n        updated_at: now,\n        created_at: now,\n      });\n\n      return { type, userId, threadId, title, messages: []};\n    }, `createThread(${userId}, ${threadId})`);\n  };\n\n  public async addMessagesToThread(\n    userId: string,\n    threadId: string,\n    messages: MessageObject[]\n  ): Promise<void> {\n    return this.executeWithRetry(async () => {\n      await ThreadModel.updateOne({ threadId, userId }, {\n        updated_at: Date.now(),\n      });\n      for (const message of messages) {\n        await MessageModel.create({\n          threadId,\n          messageId: message.messageId,\n          userId,\n          role: message.role,\n          content: message.content,\n          timestamp: message.timestamp,\n          metadata: message.metadata,\n        });\n      }\n    }, `addMessagesToThread(${userId}, ${threadId})`);\n  };\n\n  public async deleteThread(userId: string, threadId: string): Promise<void> {\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      const messages = await MessageModel.find({ userId, threadId })\n        .sort({ timestamp: 1 })\n        .maxTimeMS(timeout);\n\n      messages?.forEach((message: MessageDocument) => {\n        message.deleteOne();\n      });\n\n      const thread = await ThreadModel.findOne({ userId, threadId }).maxTimeMS(timeout);\n      thread?.deleteOne();\n    }, `deleteThread(${userId}, ${threadId})`);\n  };\n\n  public async listThreads(userId: string): Promise<ThreadMetadata[]> {\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      const threads = await ThreadModel.find({ userId })\n        .sort({ updated_at: -1 })\n        .maxTimeMS(timeout);\n      const data: ThreadMetadata[] = threads.map((thread: ThreadDocument) => {\n        return {\n          type: thread.type,\n          userId,\n          threadId: thread.threadId,\n          title: thread.title,\n          updatedAt: thread.updated_at\n        } as ThreadMetadata;\n      })\n      return data;\n    }, `listThreads(${userId})`);\n  };\n}\n","import { type Document, Schema } from \"mongoose\";\nimport mongoose from \"mongoose\";\n\nexport const WorkflowObjectSchema = new Schema(\n\t{\n\t\tworkflowId: {\n\t\t\ttype: String,\n\t\t\trequired: true,\n\t\t\tunique: true,\n\t\t},\n\t\tuserId: {\n\t\t\ttype: String,\n\t\t},\n\t\ttitle: {\n\t\t\ttype: String,\n\t\t\trequired: true,\n\t\t},\n\t\tdescription: {\n\t\t\ttype: String,\n\t\t\trequired: true,\n\t\t},\n\t\tactive: {\n\t\t\ttype: Boolean,\n\t\t\trequired: true,\n\t\t\tdefault: false,\n\t\t},\n\t\tcontent: {\n\t\t\ttype: String,\n\t\t\trequired: true,\n\t\t},\n\t},\n\t{\n\t\ttimestamps: true,\n\t}\n);\n\nexport interface WorkflowDocument extends Document {\n\tworkflowId: string;\n\tuserId?: string;\n\ttitle: string;\n\tdescription: string;\n\tactive: boolean;\n\tcontent: string;\n}\n\nexport const WorkflowModel = mongoose.model<WorkflowDocument>(\"Workflow\", WorkflowObjectSchema);\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    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      await WorkflowModel.updateOne(\n        { workflowId },\n        { $set: updates }\n      ).maxTimeMS(timeout);\n    }, \"updateWorkflow()\");\n  }\n\n  public async deleteWorkflow(workflowId: string): Promise<void> {\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      await WorkflowModel.deleteOne({ workflowId }).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      const query = userId ? { userId } : {};\n      const workflows = await WorkflowModel.find(query)\n        .maxTimeMS(timeout)\n        .lean<Workflow[]>();\n      return workflows;\n    }, \"listWorkflows()\");\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACCA,IAAAA,oBAAqB;AACrB,IAAAC,iBAAwB;;;ACFxB,sBAAsC;AACtC,IAAAC,mBAAqB;AAEd,IAAM,oBAAoB,IAAI;AAAA,EACpC;AAAA,IACC,cAAc;AAAA,MACb,MAAM;AAAA,IACP;AAAA,EACD;AACD;AAMO,IAAM,aAAa,iBAAAC,QAAS,MAAqB,SAAS,iBAAiB;;;ACD3E,IAAM,eAAN,MAA2C;AAAA,EACxC;AAAA,EACA;AAAA,EAER,YACE,kBACA,qBACA;AACA,SAAK,mBAAmB;AACxB,SAAK,sBAAsB;AAAA,EAC7B;AAAA,EAEA,MAAa,iBAAkC;AAC7C,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,UAAU,KAAK,oBAAoB;AACzC,YAAM,WAAW,MAAM,WAAW,QAAQ;AAAA,QACxC,IAAI;AAAA,MACN,CAAC,EAAE,UAAU,OAAO,EACjB,KAAoB;AACvB,aAAO,UAAU,gBAAgB;AAAA,IACnC,GAAG,kBAAkB;AAAA,EACvB;AAAA,EAEA,MAAa,kBAAkB,QAA+B;AAC5D,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,UAAU,KAAK,oBAAoB;AACzC,YAAM,WAAW,UAAU;AAAA,QACzB,IAAI;AAAA,MACN,GAAG,EAAE,gBAAgB,OAAO,CAAC,EAAE,UAAU,OAAO;AAAA,IAClD,GAAG,qBAAqB;AAAA,EAC1B;AACF;;;AC5CA,IAAAC,mBAAgD;AAEhD,IAAM,qBAAqB,IAAI;AAAA,EAC9B;AAAA,IACC,IAAI;AAAA,MACH,MAAM;AAAA,MACN,UAAU;AAAA,MACV,OAAO;AAAA,MACP,QAAQ;AAAA,IACT;AAAA,IACA,MAAM;AAAA,MACL,MAAM;AAAA,MACN,UAAU;AAAA,MACV,OAAO;AAAA,IACR;AAAA,IACA,aAAa;AAAA,MACZ,MAAM;AAAA,MACN,UAAU;AAAA,IACX;AAAA,IACA,QAAQ;AAAA,MACP,MAAM;AAAA,MACN,UAAU;AAAA,IACX;AAAA,IACA,QAAQ;AAAA,MACP,MAAM;AAAA,MACN,UAAU;AAAA,IACX;AAAA,IACA,qBAAqB;AAAA,MACpB,MAAM,CAAC,MAAM;AAAA,MACb,UAAU;AAAA,IACX;AAAA,IACA,MAAM;AAAA,MACL,MAAM,CAAC,MAAM;AAAA,MACb,UAAU;AAAA,IACX;AAAA,EACD;AACD;AAMO,IAAM,cAAc,iBAAAC,QAAS,MAAsB,UAAU,kBAAkB;;;AChC/E,IAAM,gBAAN,MAA6C;AAAA,EAC1C;AAAA,EACA;AAAA,EAER,YACE,kBACA,qBACA;AACA,SAAK,mBAAmB;AACxB,SAAK,sBAAsB;AAAA,EAC7B;AAAA,EAEA,MAAa,UAAU,UAA+C;AACpE,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,UAAU,KAAK,oBAAoB;AACzC,YAAM,SAAS,MAAM,YAAY,QAAQ,EAAE,IAAI,SAAS,CAAC,EACtD,UAAU,OAAO,EACjB,KAAa;AAChB,aAAO,UAAU;AAAA,IACnB,GAAG,aAAa,QAAQ,GAAG;AAAA,EAC7B;AAAA,EAEA,MAAa,gBAAgB,YAAiD;AAC5E,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,UAAU,KAAK,oBAAoB;AACzC,YAAM,SAAS,MAAM,YAAY,QAAQ,EAAE,MAAM,WAAW,CAAC,EAC1D,UAAU,OAAO,EACjB,KAAa;AAChB,aAAO,UAAU;AAAA,IACnB,GAAG,mBAAmB,UAAU,GAAG;AAAA,EACrC;AAAA,EAEA,MAAa,WAAW,QAA+B;AACrD,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,YAAY,OAAO,MAAM;AAAA,IACjC,GAAG,cAAc,OAAO,EAAE,GAAG;AAAA,EAC/B;AAAA,EAEA,MAAa,aAAa,UAAkB,QAA+B;AACzE,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,UAAU,KAAK,oBAAoB;AACzC,YAAM,YAAY,UAAU;AAAA,QAC1B,IAAI;AAAA,MACN,GAAG,MAAM,EAAE,UAAU,OAAO;AAAA,IAC9B,GAAG,gBAAgB,QAAQ,GAAG;AAAA,EAChC;AAAA,EAEA,MAAa,aAAa,UAAiC;AACzD,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,UAAU,KAAK,oBAAoB;AACzC,YAAM,YAAY,UAAU,EAAE,IAAI,SAAS,CAAC,EAAE,UAAU,OAAO;AAAA,IACjE,GAAG,gBAAgB,QAAQ,GAAG;AAAA,EAChC;AAAA,EAEA,MAAa,cAAiC;AAC5C,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,UAAU,KAAK,oBAAoB;AACzC,YAAM,UAAU,MAAM,YAAY,KAAK,EACpC,UAAU,OAAO,EACjB,KAAe;AAClB,aAAO;AAAA,IACT,GAAG,eAAe;AAAA,EACpB;AACF;;;AC1EA,oBAA2B;AAC3B,IAAAC,mBAAsC;AACtC,IAAAA,mBAAqB;AAEd,IAAM,qBAAqB,IAAI;AAAA,EACrC;AAAA,IACC,MAAM;AAAA,MACL,MAAM;AAAA,MACN,MAAM,OAAO,OAAO,wBAAU;AAAA,MAC9B,UAAU;AAAA,IACX;AAAA,IACA,UAAU;AAAA,MACT,MAAM;AAAA,MACN,UAAU;AAAA,MACV,OAAO;AAAA,IACR;AAAA,IACA,QAAQ;AAAA,MACP,MAAM;AAAA,MACN,UAAU;AAAA,MACV,OAAO;AAAA,IACR;AAAA,IACA,OAAO;AAAA,MACN,MAAM;AAAA,MACN,UAAU;AAAA,IACX;AAAA,IACA,YAAY;AAAA,MACX,MAAM;AAAA,MACN,UAAU;AAAA,IACX;AAAA,IACA,YAAY;AAAA,MACX,MAAM;AAAA,MACN,UAAU;AAAA,IACX;AAAA,EACD;AACD;AAWO,IAAM,cAAc,iBAAAC,QAAS,MAAsB,UAAU,kBAAkB;;;AC7CtF,IAAAC,iBAA4B;AAC5B,IAAAC,mBAAsC;AACtC,IAAAA,mBAAqB;AAGd,IAAM,6BAA6B,IAAI;AAAA,EAC7C;AAAA,IACC,MAAM,EAAE,MAAM,QAAQ,UAAU,KAAK;AAAA,IACrC,OAAO,EAAE,MAAM,CAAC,wBAAO,MAAM,KAAK,GAAG,UAAU,KAAK;AAAA,EACrD;AAAA,EACA,EAAE,KAAK,MAAM;AACd;AAGO,IAAM,sBAAsB,IAAI;AAAA,EACtC;AAAA,IACC,WAAW;AAAA,MACV,MAAM;AAAA,MACN,UAAU;AAAA,MACV,OAAO;AAAA,IACR;AAAA,IACA,UAAU;AAAA,MACT,MAAM;AAAA,MACN,UAAU;AAAA,MACV,OAAO;AAAA,IACR;AAAA,IACA,QAAQ;AAAA,MACP,MAAM;AAAA,MACN,UAAU;AAAA,MACV,OAAO;AAAA,IACR;AAAA,IACA,MAAM;AAAA,MACL,MAAM;AAAA,MACN,MAAM,OAAO,OAAO,0BAAW;AAAA,MAC/B,UAAU;AAAA,IACX;AAAA,IACA,SAAS;AAAA,MACR,MAAM;AAAA,MACN,UAAU;AAAA,IACX;AAAA,IACA,WAAW;AAAA,MACV,MAAM;AAAA,MACN,UAAU;AAAA,IACX;AAAA,IACA,UAAU;AAAA,MACT,MAAM,wBAAO,MAAM;AAAA,MACnB,SAAS,CAAC;AAAA,IACX;AAAA,EACD;AACD;AAgBO,IAAM,eAAe,iBAAAC,QAAS,MAAuB,WAAW,mBAAmB;;;AC5D1F,oBAAwB;AASjB,IAAM,gBAAN,MAA6C;AAAA,EAC1C;AAAA,EACA;AAAA,EAER,YACE,kBACA,qBACA;AACA,SAAK,mBAAmB;AACxB,SAAK,sBAAsB;AAAA,EAC7B;AAAA,EAEA,MAAa,UACX,QACA,UACmC;AACnC,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,UAAU,KAAK,oBAAoB;AACzC,YAAM,SAAS,MAAM,YAAY,QAAQ,EAAE,UAAU,OAAO,CAAC,EAAE,UAAU,OAAO;AAChF,YAAM,WAAW,MAAM,aAAa,KAAK,EAAE,UAAU,OAAO,CAAC,EAC1D,KAAK,EAAE,WAAW,EAAE,CAAC,EACrB,UAAU,OAAO;AAEpB,UAAI,CAAC,OAAQ,QAAO;AAEpB,4BAAQ,MAAM,MAAM,SAAS,SAAS,MAAM,wBAAwB,QAAQ,EAAE;AAE9E,YAAM,eAA6B;AAAA,QACjC,UAAU,OAAO;AAAA,QACjB,QAAQ,OAAO;AAAA,QACf,MAAM,OAAO;AAAA,QACb,OAAO,OAAO,SAAS;AAAA,QACvB,UAAU,CAAC;AAAA,MACb;AACA,eAAS,QAAQ,CAAC,YAA6B;AAC7C,qBAAa,SAAS,KAAK;AAAA,UACzB,WAAW,QAAQ;AAAA,UACnB,MAAM,QAAQ;AAAA,UACd,SAAS,QAAQ;AAAA,UACjB,WAAW,QAAQ;AAAA,UACnB,UAAU,QAAQ;AAAA,QACpB,CAAC;AAAA,MACH,CAAC;AAED,aAAO;AAAA,IACT,GAAG,aAAa,MAAM,KAAK,QAAQ,GAAG;AAAA,EACxC;AAAA,EAEA,MAAa,aACX,MACA,QACA,UACA,OACuB;AACvB,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,MAAM,KAAK,IAAI;AACrB,YAAM,YAAY,OAAO;AAAA,QACvB;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA,YAAY;AAAA,QACZ,YAAY;AAAA,MACd,CAAC;AAED,aAAO,EAAE,MAAM,QAAQ,UAAU,OAAO,UAAU,CAAC,EAAC;AAAA,IACtD,GAAG,gBAAgB,MAAM,KAAK,QAAQ,GAAG;AAAA,EAC3C;AAAA,EAEA,MAAa,oBACX,QACA,UACA,UACe;AACf,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,YAAY,UAAU,EAAE,UAAU,OAAO,GAAG;AAAA,QAChD,YAAY,KAAK,IAAI;AAAA,MACvB,CAAC;AACD,iBAAW,WAAW,UAAU;AAC9B,cAAM,aAAa,OAAO;AAAA,UACxB;AAAA,UACA,WAAW,QAAQ;AAAA,UACnB;AAAA,UACA,MAAM,QAAQ;AAAA,UACd,SAAS,QAAQ;AAAA,UACjB,WAAW,QAAQ;AAAA,UACnB,UAAU,QAAQ;AAAA,QACpB,CAAC;AAAA,MACH;AAAA,IACF,GAAG,uBAAuB,MAAM,KAAK,QAAQ,GAAG;AAAA,EAClD;AAAA,EAEA,MAAa,aAAa,QAAgB,UAAiC;AACzE,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,UAAU,KAAK,oBAAoB;AACzC,YAAM,WAAW,MAAM,aAAa,KAAK,EAAE,QAAQ,SAAS,CAAC,EAC1D,KAAK,EAAE,WAAW,EAAE,CAAC,EACrB,UAAU,OAAO;AAEpB,gBAAU,QAAQ,CAAC,YAA6B;AAC9C,gBAAQ,UAAU;AAAA,MACpB,CAAC;AAED,YAAM,SAAS,MAAM,YAAY,QAAQ,EAAE,QAAQ,SAAS,CAAC,EAAE,UAAU,OAAO;AAChF,cAAQ,UAAU;AAAA,IACpB,GAAG,gBAAgB,MAAM,KAAK,QAAQ,GAAG;AAAA,EAC3C;AAAA,EAEA,MAAa,YAAY,QAA2C;AAClE,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,UAAU,KAAK,oBAAoB;AACzC,YAAM,UAAU,MAAM,YAAY,KAAK,EAAE,OAAO,CAAC,EAC9C,KAAK,EAAE,YAAY,GAAG,CAAC,EACvB,UAAU,OAAO;AACpB,YAAM,OAAyB,QAAQ,IAAI,CAAC,WAA2B;AACrE,eAAO;AAAA,UACL,MAAM,OAAO;AAAA,UACb;AAAA,UACA,UAAU,OAAO;AAAA,UACjB,OAAO,OAAO;AAAA,UACd,WAAW,OAAO;AAAA,QACpB;AAAA,MACF,CAAC;AACD,aAAO;AAAA,IACT,GAAG,eAAe,MAAM,GAAG;AAAA,EAC7B;AACF;;;AC5IA,IAAAC,mBAAsC;AACtC,IAAAA,mBAAqB;AAEd,IAAM,uBAAuB,IAAI;AAAA,EACvC;AAAA,IACC,YAAY;AAAA,MACX,MAAM;AAAA,MACN,UAAU;AAAA,MACV,QAAQ;AAAA,IACT;AAAA,IACA,QAAQ;AAAA,MACP,MAAM;AAAA,IACP;AAAA,IACA,OAAO;AAAA,MACN,MAAM;AAAA,MACN,UAAU;AAAA,IACX;AAAA,IACA,aAAa;AAAA,MACZ,MAAM;AAAA,MACN,UAAU;AAAA,IACX;AAAA,IACA,QAAQ;AAAA,MACP,MAAM;AAAA,MACN,UAAU;AAAA,MACV,SAAS;AAAA,IACV;AAAA,IACA,SAAS;AAAA,MACR,MAAM;AAAA,MACN,UAAU;AAAA,IACX;AAAA,EACD;AAAA,EACA;AAAA,IACC,YAAY;AAAA,EACb;AACD;AAWO,IAAM,gBAAgB,iBAAAC,QAAS,MAAwB,YAAY,oBAAoB;;;AClCvF,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,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,UAAU,KAAK,oBAAoB;AACzC,YAAM,cAAc;AAAA,QAClB,EAAE,WAAW;AAAA,QACb,EAAE,MAAM,QAAQ;AAAA,MAClB,EAAE,UAAU,OAAO;AAAA,IACrB,GAAG,kBAAkB;AAAA,EACvB;AAAA,EAEA,MAAa,eAAe,YAAmC;AAC7D,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,UAAU,KAAK,oBAAoB;AACzC,YAAM,cAAc,UAAU,EAAE,WAAW,CAAC,EAAE,UAAU,OAAO;AAAA,IACjE,GAAG,kBAAkB;AAAA,EACvB;AAAA,EAEA,MAAa,cAAc,QAAsC;AAC/D,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,UAAU,KAAK,oBAAoB;AACzC,YAAM,QAAQ,SAAS,EAAE,OAAO,IAAI,CAAC;AACrC,YAAM,YAAY,MAAM,cAAc,KAAK,KAAK,EAC7C,UAAU,OAAO,EACjB,KAAiB;AACpB,aAAO;AAAA,IACT,GAAG,iBAAiB;AAAA,EACtB;AACF;;;ATjDO,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,sBAAAC,QAAS,WAAW,GAAG,aAAa,MAAM;AACxC,WAAK,YAAY;AACjB,WAAK,oBAAoB;AACzB,WAAK,eAAe;AACpB,6BAAQ,MAAM,KAAK,gCAAgC;AAAA,IACrD,CAAC;AAED,sBAAAA,QAAS,WAAW,GAAG,gBAAgB,MAAM;AAC3C,WAAK,YAAY;AACjB,6BAAQ,MAAM,KAAK,sBAAsB;AACzC,WAAK,oBAAoB;AAAA,IAC3B,CAAC;AAED,sBAAAA,QAAS,WAAW,GAAG,SAAS,CAAC,UAAU;AACzC,WAAK,YAAY;AACjB,6BAAQ,MAAM,MAAM,6BAA6B,KAAK;AACtD,WAAK,oBAAoB;AAAA,IAC3B,CAAC;AAED,sBAAAA,QAAS,WAAW,GAAG,eAAe,MAAM;AAC1C,WAAK,YAAY;AACjB,WAAK,oBAAoB;AACzB,WAAK,eAAe;AACpB,6BAAQ,MAAM,KAAK,kCAAkC;AAAA,IACvD,CAAC;AAAA,EACH;AAAA,EAEA,MAAc,sBAAqC;AACjD,QAAI,KAAK,cAAc;AACrB;AAAA,IACF;AAEA,SAAK,eAAe;AAEpB,WAAO,KAAK,oBAAoB,KAAK,wBAAwB,CAAC,KAAK,aAAa;AAC9E,WAAK;AACL,6BAAQ,MAAM;AAAA,QACZ,uCAAuC,KAAK,iBAAiB,IAAI,KAAK,oBAAoB;AAAA,MAC5F;AAEA,UAAI;AACF,cAAM,kBAAAA,QAAS,QAAQ,KAAK,KAAK,KAAK,gBAAgB;AACtD,aAAK,YAAY;AACjB,aAAK,oBAAoB;AACzB,aAAK,eAAe;AACpB,+BAAQ,MAAM,KAAK,iCAAiC;AACpD;AAAA,MACF,SAAS,OAAO;AACd,+BAAQ,MAAM;AAAA,UACZ,wBAAwB,KAAK,iBAAiB;AAAA,UAC9C;AAAA,QACF;AAEA,YAAI,KAAK,oBAAoB,KAAK,sBAAsB;AACtD,gBAAM,IAAI;AAAA,YAAQ,CAAC,YACjB,WAAW,SAAS,KAAK,iBAAiB;AAAA,UAC5C;AAAA,QACF;AAAA,MACF;AAAA,IACF;AAEA,SAAK,eAAe;AAEpB,QAAI,CAAC,KAAK,aAAa;AACrB,6BAAQ,MAAM;AAAA,QACZ,wCAAwC,KAAK,oBAAoB;AAAA,MACnE;AAAA,IACF;AAAA,EACF;AAAA,EAEA,MAAa,UAAyB;AACpC,QAAI,KAAK,WAAW;AAClB;AAAA,IACF;AAEA,QAAI;AACF,YAAM,kBAAAA,QAAS,QAAQ,KAAK,KAAK,KAAK,gBAAgB;AACtD,WAAK,YAAY;AACjB,WAAK,oBAAoB;AAAA,IAC3B,SAAS,OAAO;AACd,6BAAQ,MAAM,MAAM,iCAAiC,KAAK;AAC1D,YAAM;AAAA,IACR;AAAA,EACF;AAAA,EAEA,MAAa,aAA4B;AACvC,QAAI,CAAC,KAAK,aAAa;AACrB;AAAA,IACF;AAEA,QAAI;AACF,YAAM,kBAAAA,QAAS,WAAW;AAC1B,WAAK,YAAY;AAAA,IACnB,SAAS,OAAO;AACd,6BAAQ,MAAM,MAAM,sCAAsC,KAAK;AAC/D,YAAM;AAAA,IACR;AAAA,EACF;AAAA,EAEO,cAAuB;AAC5B,WAAO,KAAK;AAAA,EACd;AAAA,EAEA,MAAc,mBAAkC;AAC9C,QAAI,CAAC,KAAK,eAAe,CAAC,KAAK,cAAc;AAC3C,YAAM,KAAK,QAAQ;AAAA,IACrB;AAGA,UAAM,cAAc;AACpB,UAAM,YAAY,KAAK,IAAI;AAC3B,WAAO,KAAK,gBAAgB,KAAK,IAAI,IAAI,YAAY,aAAa;AAChE,YAAM,IAAI,QAAQ,CAAC,YAAY,WAAW,SAAS,GAAG,CAAC;AAAA,IACzD;AAEA,QAAI,CAAC,KAAK,aAAa;AACrB,YAAM,IAAI,MAAM,kDAAkD;AAAA,IACpE;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKU,sBAA8B;AACtC,WAAO,KAAK;AAAA,EACd;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,MAAgB,iBACd,WACA,gBAAwB,sBACZ;AACZ,UAAM,KAAK,iBAAiB;AAE5B,QAAI;AACF,aAAO,MAAM,UAAU;AAAA,IACzB,SAAS,OAAY;AAEnB,UAAI,MAAM,SAAS,MAAM,MAAM,SAAS,SAAS,+BAA+B,GAAG;AACjF,+BAAQ,MAAM,MAAM,GAAG,aAAa,sBAAsB;AAC1D,cAAM;AAAA,MACR;AAGA,UACE,MAAM,SAAS,uBACf,MAAM,SAAS,sBACf,MAAM,SAAS,SAAS,YAAY,KACpC,MAAM,SAAS,SAAS,YAAY,GACpC;AACA,+BAAQ,MAAM;AAAA,UACZ,GAAG,aAAa;AAAA,QAClB;AAEA,cAAM,KAAK,iBAAiB;AAG5B,YAAI;AACF,iBAAO,MAAM,UAAU;AAAA,QACzB,SAAS,YAAiB;AACxB,iCAAQ,MAAM,MAAM,GAAG,aAAa,wBAAwB,UAAU;AACtE,gBAAM;AAAA,QACR;AAAA,MACF;AAGA,YAAM;AAAA,IACR;AAAA,EACF;AACF;","names":["import_mongoose","import_logger","import_mongoose","mongoose","import_mongoose","mongoose","import_mongoose","mongoose","import_memory","import_mongoose","mongoose","import_mongoose","mongoose","mongoose"]}
+{"version":3,"sources":["../index.ts","../implements/base.memory.ts","../models/agent.model.ts","../implements/agent.memory.ts","../models/intent.model.ts","../implements/intent.memory.ts","../models/threads.model.ts","../models/messages.model.ts","../implements/thread.memory.ts","../models/workflow.model.ts","../implements/workflow.memory.ts"],"sourcesContent":["export { MongoDBMemory } from \"./implements/base.memory\";\nexport type { MongoDBMemoryConfig } from \"./implements/base.memory\";","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 { type Document, Schema } from \"mongoose\";\nimport mongoose from \"mongoose\";\n\nexport const AgentObjectSchema = new Schema(\n\t{\n\t\tagent_prompt: {\n\t\t\ttype: String,\n\t\t},\n\t},\n);\n\nexport interface AgentDocument extends Document {\n\tagent_prompt: string;\n}\n\nexport const AgentModel = mongoose.model<AgentDocument>(\"Agent\", AgentObjectSchema);\n","import { IAgentMemory } from \"@ainetwork/adk/modules\";\nimport { AgentModel } from \"../models/agent.model\";\n\nexport type ExecuteWithRetryFn = <T>(\n  operation: () => Promise<T>,\n  operationName?: string\n) => Promise<T>;\n\nexport type GetOperationTimeoutFn = () => number;\n\ntype AgentMetadata = {\n  agent_prompt: string;\n}\n\nexport class MongoDBAgent implements IAgentMemory {\n  private executeWithRetry: ExecuteWithRetryFn;\n  private getOperationTimeout: GetOperationTimeoutFn;\n\n  constructor(\n    executeWithRetry: ExecuteWithRetryFn,\n    getOperationTimeout: GetOperationTimeoutFn\n  ) {\n    this.executeWithRetry = executeWithRetry;\n    this.getOperationTimeout = getOperationTimeout;\n  }\n\n  public async getAgentPrompt(): Promise<string> {\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      const metadata = await AgentModel.findOne({\n        id: \"agent_metadata\"\n      }).maxTimeMS(timeout)\n        .lean<AgentMetadata>();\n      return metadata?.agent_prompt || \"\";\n    }, \"getAgentPrompt()\");\n  };\n  \n  public async updateAgentPrompt(prompt: string): Promise<void> {\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      await AgentModel.updateOne({\n        id: \"agent_metadata\",\n      }, { \"agent_prompt\": prompt }).maxTimeMS(timeout);\n    }, \"updateAgentPrompt()\");\n  };\n}\n","import { Intent } from \"@ainetwork/adk/types/memory\";\nimport mongoose, { type Document, Schema } from \"mongoose\";\n\nconst IntentObjectSchema = new Schema(\n\t{\n\t\tid: { \n\t\t\ttype: String,\n\t\t\trequired: true,\n\t\t\tindex: true,\n\t\t\tunique: true,\n\t\t},\n\t\tname: {\n\t\t\ttype: String,\n\t\t\trequired: true,\n\t\t\tindex: true,\n\t\t},\n\t\tdescription: {\n\t\t\ttype: String,\n\t\t\trequired: true,\n\t\t},\n\t\tprompt: {\n\t\t\ttype: String,\n\t\t\trequired: false,\n\t\t},\n\t\tstatus: {\n\t\t\ttype: String,\n\t\t\trequired: true,\n\t\t},\n\t\ttriggeringSentences: {\n\t\t\ttype: [String],\n\t\t\trequired: false,\n\t\t},\n\t\ttags: {\n\t\t\ttype: [String],\n\t\t\trequired: false,\n\t\t},\n\t},\n);\n\nexport interface IntentDocument extends Omit<Document, 'id'>, Omit<Intent, 'id'> {\n\tid: string;\n}\n\nexport const IntentModel = mongoose.model<IntentDocument>(\"Intent\", IntentObjectSchema);","import type { Intent } from \"@ainetwork/adk/types/memory\";\nimport { IIntentMemory } from \"@ainetwork/adk/modules\";\nimport { IntentModel } from \"../models/intent.model\";\n\nexport type ExecuteWithRetryFn = <T>(\n  operation: () => Promise<T>,\n  operationName?: string\n) => Promise<T>;\n\nexport type GetOperationTimeoutFn = () => number;\n\nexport class MongoDBIntent implements IIntentMemory {\n  private executeWithRetry: ExecuteWithRetryFn;\n  private getOperationTimeout: GetOperationTimeoutFn;\n\n  constructor(\n    executeWithRetry: ExecuteWithRetryFn,\n    getOperationTimeout: GetOperationTimeoutFn\n  ) {\n    this.executeWithRetry = executeWithRetry;\n    this.getOperationTimeout = getOperationTimeout;\n  }\n\n  public async getIntent(intentId: string): Promise<Intent | undefined> {\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      const intent = await IntentModel.findOne({ id: intentId })\n        .maxTimeMS(timeout)\n        .lean<Intent>();\n      return intent || undefined;\n    }, `getIntent(${intentId})`);\n  };\n\n  public async getIntentByName(intentName: string): Promise<Intent | undefined> {\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      const intent = await IntentModel.findOne({ name: intentName })\n        .maxTimeMS(timeout)\n        .lean<Intent>();\n      return intent || undefined;\n    }, `getIntentByName(${intentName})`);\n  }\n\n  public async saveIntent(intent: Intent): Promise<void> {\n    return this.executeWithRetry(async () => {\n      await IntentModel.create(intent);\n    }, `saveIntent(${intent.id})`);\n  };\n\n  public async updateIntent(intentId: string, intent: Intent): Promise<void> {\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      await IntentModel.updateOne({\n        id: intentId,\n      }, intent).maxTimeMS(timeout);\n    }, `updateIntent(${intentId})`);\n  };\n\n  public async deleteIntent(intentId: string): Promise<void> {\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      await IntentModel.deleteOne({ id: intentId }).maxTimeMS(timeout);\n    }, `deleteIntent(${intentId})`);\n  };\n\n  public async listIntents(): Promise<Intent[]> {\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      const intents = await IntentModel.find()\n        .maxTimeMS(timeout)\n        .lean<Intent[]>();\n      return intents;\n    }, `listIntents()`);\n  };\n}\n","import { ThreadType } from \"@ainetwork/adk/types/memory\";\nimport { type Document, Schema } from \"mongoose\";\nimport mongoose from \"mongoose\";\n\nexport const ThreadObjectSchema = new Schema(\n\t{\n\t\ttype: {\n\t\t\ttype: String,\n\t\t\tenum: Object.values(ThreadType),\n\t\t\trequired: true,\n\t\t},\n\t\tthreadId: {\n\t\t\ttype: String,\n\t\t\trequired: true,\n\t\t\tindex: true,\n\t\t},\n\t\tuserId: {\n\t\t\ttype: String,\n\t\t\trequired: true,\n\t\t\tindex: true,\n\t\t},\n\t\ttitle: {\n\t\t\ttype: String,\n\t\t\trequired: false,\n\t\t},\n\t\tcreated_at: {\n\t\t\ttype: Number,\n\t\t\trequired: true,\n\t\t},\n\t\tupdated_at: {\n\t\t\ttype: Number,\n\t\t\trequired: true,\n\t\t}\n\t},\n);\n\nexport interface ThreadDocument extends Document {\n\ttype: ThreadType;\n\tthreadId: string;\n\tuserId: string;\n\ttitle: string;\n\tcreated_at: number;\n\tupdated_at: number;\n}\n\nexport const ThreadModel = mongoose.model<ThreadDocument>(\"Thread\", ThreadObjectSchema);\n","import { MessageRole } from \"@ainetwork/adk/types/memory\";\nimport { type Document, Schema } from \"mongoose\";\nimport mongoose from \"mongoose\";\n\n// MessageContentObject schema\nexport const MessageContentObjectSchema = new Schema(\n\t{\n\t\ttype: { type: String, required: true },\n\t\tparts: { type: [Schema.Types.Mixed], required: true },\n\t},\n\t{ _id: false },\n);\n\n// MessageObject schema - 개별 문서로 저장\nexport const MessageObjectSchema = new Schema(\n\t{\n\t\tmessageId: {\n\t\t\ttype: String,\n\t\t\trequired: true,\n\t\t\tindex: true,\n\t\t},\n\t\tthreadId: {\n\t\t\ttype: String,\n\t\t\trequired: true,\n\t\t\tindex: true,\n\t\t},\n\t\tuserId: {\n\t\t\ttype: String,\n\t\t\trequired: true,\n\t\t\tindex: true,\n\t\t},\n\t\trole: {\n\t\t\ttype: String,\n\t\t\tenum: Object.values(MessageRole),\n\t\t\trequired: true,\n\t\t},\n\t\tcontent: {\n\t\t\ttype: MessageContentObjectSchema,\n\t\t\trequired: true,\n\t\t},\n\t\ttimestamp: {\n\t\t\ttype: Number,\n\t\t\trequired: true,\n\t\t},\n\t\tmetadata: {\n\t\t\ttype: Schema.Types.Mixed,\n\t\t\tdefault: {},\n\t\t},\n\t},\n);\n\n// Message Document interface\nexport interface MessageDocument extends Document {\n\tmessageId: string;\n\tthreadId: string;\n\tuserId: string;\n\trole: MessageRole;\n\tcontent: {\n\t\ttype: string;\n\t\tparts: any[];\n\t};\n\ttimestamp: number;\n\tmetadata?: { [key: string]: unknown };\n}\n\nexport const MessageModel = mongoose.model<MessageDocument>(\"Message\", MessageObjectSchema);","import type { MessageObject, ThreadMetadata, ThreadObject, ThreadType } from \"@ainetwork/adk/types/memory\";\nimport { MessageRole } from \"@ainetwork/adk/types/memory\";\nimport { IThreadMemory } from \"@ainetwork/adk/modules\";\nimport { ThreadDocument, ThreadModel } from \"../models/threads.model\";\nimport { MessageDocument, MessageModel } from \"../models/messages.model\";\nimport { loggers } from \"@ainetwork/adk/utils/logger\";\n\nexport type ExecuteWithRetryFn = <T>(\n  operation: () => Promise<T>,\n  operationName?: string\n) => Promise<T>;\n\nexport type GetOperationTimeoutFn = () => number;\n\nexport class MongoDBThread implements IThreadMemory {\n  private executeWithRetry: ExecuteWithRetryFn;\n  private getOperationTimeout: GetOperationTimeoutFn;\n\n  constructor(\n    executeWithRetry: ExecuteWithRetryFn,\n    getOperationTimeout: GetOperationTimeoutFn\n  ) {\n    this.executeWithRetry = executeWithRetry;\n    this.getOperationTimeout = getOperationTimeout;\n  }\n\n  public async getThread(\n    userId: string,\n    threadId: string\n  ): Promise<ThreadObject | undefined> {\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      const thread = await ThreadModel.findOne({ threadId, userId }).maxTimeMS(timeout);\n      const messages = await MessageModel.find({ threadId, userId })\n        .sort({ timestamp: 1 })\n        .maxTimeMS(timeout);\n\n      if (!thread) return undefined;\n\n      loggers.agent.debug(`Found ${messages.length} messages for thread ${threadId}`);\n\n      const threadObject: ThreadObject = {\n        threadId: thread.threadId,\n        userId: thread.userId,\n        type: thread.type as ThreadType,\n        title: thread.title || \"New thread\",\n        messages: []\n      };\n      messages.forEach((message: MessageDocument) => {\n        threadObject.messages.push({\n          messageId: message.messageId,\n          role: message.role as MessageRole,\n          content: message.content,\n          timestamp: message.timestamp,\n          metadata: message.metadata,\n        });\n      });\n\n      return threadObject;\n    }, `getThread(${userId}, ${threadId})`);\n  };\n\n  public async createThread(\n    type: ThreadType,\n    userId: string,\n    threadId: string,\n    title: string,\n  ): Promise<ThreadObject> {\n    return this.executeWithRetry(async () => {\n      const now = Date.now();\n      await ThreadModel.create({\n        type,\n        userId,\n        threadId,\n        title,\n        updated_at: now,\n        created_at: now,\n      });\n\n      return { type, userId, threadId, title, messages: []};\n    }, `createThread(${userId}, ${threadId})`);\n  };\n\n  public async addMessagesToThread(\n    userId: string,\n    threadId: string,\n    messages: MessageObject[]\n  ): Promise<void> {\n    return this.executeWithRetry(async () => {\n      await ThreadModel.updateOne({ threadId, userId }, {\n        updated_at: Date.now(),\n      });\n      for (const message of messages) {\n        await MessageModel.create({\n          threadId,\n          messageId: message.messageId,\n          userId,\n          role: message.role,\n          content: message.content,\n          timestamp: message.timestamp,\n          metadata: message.metadata,\n        });\n      }\n    }, `addMessagesToThread(${userId}, ${threadId})`);\n  };\n\n  public async deleteThread(userId: string, threadId: string): Promise<void> {\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      const messages = await MessageModel.find({ userId, threadId })\n        .sort({ timestamp: 1 })\n        .maxTimeMS(timeout);\n\n      messages?.forEach((message: MessageDocument) => {\n        message.deleteOne();\n      });\n\n      const thread = await ThreadModel.findOne({ userId, threadId }).maxTimeMS(timeout);\n      thread?.deleteOne();\n    }, `deleteThread(${userId}, ${threadId})`);\n  };\n\n  public async listThreads(userId: string): Promise<ThreadMetadata[]> {\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      const threads = await ThreadModel.find({ userId })\n        .sort({ updated_at: -1 })\n        .maxTimeMS(timeout);\n      const data: ThreadMetadata[] = threads.map((thread: ThreadDocument) => {\n        return {\n          type: thread.type,\n          userId,\n          threadId: thread.threadId,\n          title: thread.title,\n          updatedAt: thread.updated_at\n        } as ThreadMetadata;\n      })\n      return data;\n    }, `listThreads(${userId})`);\n  };\n}\n","import { type Document, Schema } from \"mongoose\";\nimport mongoose from \"mongoose\";\n\nexport const WorkflowObjectSchema = new Schema(\n\t{\n\t\tworkflowId: {\n\t\t\ttype: String,\n\t\t\trequired: true,\n\t\t\tunique: true,\n\t\t},\n\t\tuserId: {\n\t\t\ttype: String,\n\t\t},\n\t\ttitle: {\n\t\t\ttype: String,\n\t\t\trequired: true,\n\t\t},\n\t\tdescription: {\n\t\t\ttype: String,\n\t\t\trequired: true,\n\t\t},\n\t\tactive: {\n\t\t\ttype: Boolean,\n\t\t\trequired: true,\n\t\t\tdefault: false,\n\t\t},\n\t\tcontent: {\n\t\t\ttype: String,\n\t\t\trequired: true,\n\t\t},\n\t},\n\t{\n\t\ttimestamps: true,\n\t}\n);\n\nexport interface WorkflowDocument extends Document {\n\tworkflowId: string;\n\tuserId?: string;\n\ttitle: string;\n\tdescription: string;\n\tactive: boolean;\n\tcontent: string;\n}\n\nexport const WorkflowModel = mongoose.model<WorkflowDocument>(\"Workflow\", WorkflowObjectSchema);\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":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACCA,IAAAA,oBAAqB;AACrB,IAAAC,iBAAwB;;;ACFxB,sBAAsC;AACtC,IAAAC,mBAAqB;AAEd,IAAM,oBAAoB,IAAI;AAAA,EACpC;AAAA,IACC,cAAc;AAAA,MACb,MAAM;AAAA,IACP;AAAA,EACD;AACD;AAMO,IAAM,aAAa,iBAAAC,QAAS,MAAqB,SAAS,iBAAiB;;;ACD3E,IAAM,eAAN,MAA2C;AAAA,EACxC;AAAA,EACA;AAAA,EAER,YACE,kBACA,qBACA;AACA,SAAK,mBAAmB;AACxB,SAAK,sBAAsB;AAAA,EAC7B;AAAA,EAEA,MAAa,iBAAkC;AAC7C,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,UAAU,KAAK,oBAAoB;AACzC,YAAM,WAAW,MAAM,WAAW,QAAQ;AAAA,QACxC,IAAI;AAAA,MACN,CAAC,EAAE,UAAU,OAAO,EACjB,KAAoB;AACvB,aAAO,UAAU,gBAAgB;AAAA,IACnC,GAAG,kBAAkB;AAAA,EACvB;AAAA,EAEA,MAAa,kBAAkB,QAA+B;AAC5D,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,UAAU,KAAK,oBAAoB;AACzC,YAAM,WAAW,UAAU;AAAA,QACzB,IAAI;AAAA,MACN,GAAG,EAAE,gBAAgB,OAAO,CAAC,EAAE,UAAU,OAAO;AAAA,IAClD,GAAG,qBAAqB;AAAA,EAC1B;AACF;;;AC5CA,IAAAC,mBAAgD;AAEhD,IAAM,qBAAqB,IAAI;AAAA,EAC9B;AAAA,IACC,IAAI;AAAA,MACH,MAAM;AAAA,MACN,UAAU;AAAA,MACV,OAAO;AAAA,MACP,QAAQ;AAAA,IACT;AAAA,IACA,MAAM;AAAA,MACL,MAAM;AAAA,MACN,UAAU;AAAA,MACV,OAAO;AAAA,IACR;AAAA,IACA,aAAa;AAAA,MACZ,MAAM;AAAA,MACN,UAAU;AAAA,IACX;AAAA,IACA,QAAQ;AAAA,MACP,MAAM;AAAA,MACN,UAAU;AAAA,IACX;AAAA,IACA,QAAQ;AAAA,MACP,MAAM;AAAA,MACN,UAAU;AAAA,IACX;AAAA,IACA,qBAAqB;AAAA,MACpB,MAAM,CAAC,MAAM;AAAA,MACb,UAAU;AAAA,IACX;AAAA,IACA,MAAM;AAAA,MACL,MAAM,CAAC,MAAM;AAAA,MACb,UAAU;AAAA,IACX;AAAA,EACD;AACD;AAMO,IAAM,cAAc,iBAAAC,QAAS,MAAsB,UAAU,kBAAkB;;;AChC/E,IAAM,gBAAN,MAA6C;AAAA,EAC1C;AAAA,EACA;AAAA,EAER,YACE,kBACA,qBACA;AACA,SAAK,mBAAmB;AACxB,SAAK,sBAAsB;AAAA,EAC7B;AAAA,EAEA,MAAa,UAAU,UAA+C;AACpE,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,UAAU,KAAK,oBAAoB;AACzC,YAAM,SAAS,MAAM,YAAY,QAAQ,EAAE,IAAI,SAAS,CAAC,EACtD,UAAU,OAAO,EACjB,KAAa;AAChB,aAAO,UAAU;AAAA,IACnB,GAAG,aAAa,QAAQ,GAAG;AAAA,EAC7B;AAAA,EAEA,MAAa,gBAAgB,YAAiD;AAC5E,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,UAAU,KAAK,oBAAoB;AACzC,YAAM,SAAS,MAAM,YAAY,QAAQ,EAAE,MAAM,WAAW,CAAC,EAC1D,UAAU,OAAO,EACjB,KAAa;AAChB,aAAO,UAAU;AAAA,IACnB,GAAG,mBAAmB,UAAU,GAAG;AAAA,EACrC;AAAA,EAEA,MAAa,WAAW,QAA+B;AACrD,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,YAAY,OAAO,MAAM;AAAA,IACjC,GAAG,cAAc,OAAO,EAAE,GAAG;AAAA,EAC/B;AAAA,EAEA,MAAa,aAAa,UAAkB,QAA+B;AACzE,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,UAAU,KAAK,oBAAoB;AACzC,YAAM,YAAY,UAAU;AAAA,QAC1B,IAAI;AAAA,MACN,GAAG,MAAM,EAAE,UAAU,OAAO;AAAA,IAC9B,GAAG,gBAAgB,QAAQ,GAAG;AAAA,EAChC;AAAA,EAEA,MAAa,aAAa,UAAiC;AACzD,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,UAAU,KAAK,oBAAoB;AACzC,YAAM,YAAY,UAAU,EAAE,IAAI,SAAS,CAAC,EAAE,UAAU,OAAO;AAAA,IACjE,GAAG,gBAAgB,QAAQ,GAAG;AAAA,EAChC;AAAA,EAEA,MAAa,cAAiC;AAC5C,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,UAAU,KAAK,oBAAoB;AACzC,YAAM,UAAU,MAAM,YAAY,KAAK,EACpC,UAAU,OAAO,EACjB,KAAe;AAClB,aAAO;AAAA,IACT,GAAG,eAAe;AAAA,EACpB;AACF;;;AC1EA,oBAA2B;AAC3B,IAAAC,mBAAsC;AACtC,IAAAA,mBAAqB;AAEd,IAAM,qBAAqB,IAAI;AAAA,EACrC;AAAA,IACC,MAAM;AAAA,MACL,MAAM;AAAA,MACN,MAAM,OAAO,OAAO,wBAAU;AAAA,MAC9B,UAAU;AAAA,IACX;AAAA,IACA,UAAU;AAAA,MACT,MAAM;AAAA,MACN,UAAU;AAAA,MACV,OAAO;AAAA,IACR;AAAA,IACA,QAAQ;AAAA,MACP,MAAM;AAAA,MACN,UAAU;AAAA,MACV,OAAO;AAAA,IACR;AAAA,IACA,OAAO;AAAA,MACN,MAAM;AAAA,MACN,UAAU;AAAA,IACX;AAAA,IACA,YAAY;AAAA,MACX,MAAM;AAAA,MACN,UAAU;AAAA,IACX;AAAA,IACA,YAAY;AAAA,MACX,MAAM;AAAA,MACN,UAAU;AAAA,IACX;AAAA,EACD;AACD;AAWO,IAAM,cAAc,iBAAAC,QAAS,MAAsB,UAAU,kBAAkB;;;AC7CtF,IAAAC,iBAA4B;AAC5B,IAAAC,mBAAsC;AACtC,IAAAA,mBAAqB;AAGd,IAAM,6BAA6B,IAAI;AAAA,EAC7C;AAAA,IACC,MAAM,EAAE,MAAM,QAAQ,UAAU,KAAK;AAAA,IACrC,OAAO,EAAE,MAAM,CAAC,wBAAO,MAAM,KAAK,GAAG,UAAU,KAAK;AAAA,EACrD;AAAA,EACA,EAAE,KAAK,MAAM;AACd;AAGO,IAAM,sBAAsB,IAAI;AAAA,EACtC;AAAA,IACC,WAAW;AAAA,MACV,MAAM;AAAA,MACN,UAAU;AAAA,MACV,OAAO;AAAA,IACR;AAAA,IACA,UAAU;AAAA,MACT,MAAM;AAAA,MACN,UAAU;AAAA,MACV,OAAO;AAAA,IACR;AAAA,IACA,QAAQ;AAAA,MACP,MAAM;AAAA,MACN,UAAU;AAAA,MACV,OAAO;AAAA,IACR;AAAA,IACA,MAAM;AAAA,MACL,MAAM;AAAA,MACN,MAAM,OAAO,OAAO,0BAAW;AAAA,MAC/B,UAAU;AAAA,IACX;AAAA,IACA,SAAS;AAAA,MACR,MAAM;AAAA,MACN,UAAU;AAAA,IACX;AAAA,IACA,WAAW;AAAA,MACV,MAAM;AAAA,MACN,UAAU;AAAA,IACX;AAAA,IACA,UAAU;AAAA,MACT,MAAM,wBAAO,MAAM;AAAA,MACnB,SAAS,CAAC;AAAA,IACX;AAAA,EACD;AACD;AAgBO,IAAM,eAAe,iBAAAC,QAAS,MAAuB,WAAW,mBAAmB;;;AC5D1F,oBAAwB;AASjB,IAAM,gBAAN,MAA6C;AAAA,EAC1C;AAAA,EACA;AAAA,EAER,YACE,kBACA,qBACA;AACA,SAAK,mBAAmB;AACxB,SAAK,sBAAsB;AAAA,EAC7B;AAAA,EAEA,MAAa,UACX,QACA,UACmC;AACnC,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,UAAU,KAAK,oBAAoB;AACzC,YAAM,SAAS,MAAM,YAAY,QAAQ,EAAE,UAAU,OAAO,CAAC,EAAE,UAAU,OAAO;AAChF,YAAM,WAAW,MAAM,aAAa,KAAK,EAAE,UAAU,OAAO,CAAC,EAC1D,KAAK,EAAE,WAAW,EAAE,CAAC,EACrB,UAAU,OAAO;AAEpB,UAAI,CAAC,OAAQ,QAAO;AAEpB,4BAAQ,MAAM,MAAM,SAAS,SAAS,MAAM,wBAAwB,QAAQ,EAAE;AAE9E,YAAM,eAA6B;AAAA,QACjC,UAAU,OAAO;AAAA,QACjB,QAAQ,OAAO;AAAA,QACf,MAAM,OAAO;AAAA,QACb,OAAO,OAAO,SAAS;AAAA,QACvB,UAAU,CAAC;AAAA,MACb;AACA,eAAS,QAAQ,CAAC,YAA6B;AAC7C,qBAAa,SAAS,KAAK;AAAA,UACzB,WAAW,QAAQ;AAAA,UACnB,MAAM,QAAQ;AAAA,UACd,SAAS,QAAQ;AAAA,UACjB,WAAW,QAAQ;AAAA,UACnB,UAAU,QAAQ;AAAA,QACpB,CAAC;AAAA,MACH,CAAC;AAED,aAAO;AAAA,IACT,GAAG,aAAa,MAAM,KAAK,QAAQ,GAAG;AAAA,EACxC;AAAA,EAEA,MAAa,aACX,MACA,QACA,UACA,OACuB;AACvB,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,MAAM,KAAK,IAAI;AACrB,YAAM,YAAY,OAAO;AAAA,QACvB;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA,YAAY;AAAA,QACZ,YAAY;AAAA,MACd,CAAC;AAED,aAAO,EAAE,MAAM,QAAQ,UAAU,OAAO,UAAU,CAAC,EAAC;AAAA,IACtD,GAAG,gBAAgB,MAAM,KAAK,QAAQ,GAAG;AAAA,EAC3C;AAAA,EAEA,MAAa,oBACX,QACA,UACA,UACe;AACf,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,YAAY,UAAU,EAAE,UAAU,OAAO,GAAG;AAAA,QAChD,YAAY,KAAK,IAAI;AAAA,MACvB,CAAC;AACD,iBAAW,WAAW,UAAU;AAC9B,cAAM,aAAa,OAAO;AAAA,UACxB;AAAA,UACA,WAAW,QAAQ;AAAA,UACnB;AAAA,UACA,MAAM,QAAQ;AAAA,UACd,SAAS,QAAQ;AAAA,UACjB,WAAW,QAAQ;AAAA,UACnB,UAAU,QAAQ;AAAA,QACpB,CAAC;AAAA,MACH;AAAA,IACF,GAAG,uBAAuB,MAAM,KAAK,QAAQ,GAAG;AAAA,EAClD;AAAA,EAEA,MAAa,aAAa,QAAgB,UAAiC;AACzE,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,UAAU,KAAK,oBAAoB;AACzC,YAAM,WAAW,MAAM,aAAa,KAAK,EAAE,QAAQ,SAAS,CAAC,EAC1D,KAAK,EAAE,WAAW,EAAE,CAAC,EACrB,UAAU,OAAO;AAEpB,gBAAU,QAAQ,CAAC,YAA6B;AAC9C,gBAAQ,UAAU;AAAA,MACpB,CAAC;AAED,YAAM,SAAS,MAAM,YAAY,QAAQ,EAAE,QAAQ,SAAS,CAAC,EAAE,UAAU,OAAO;AAChF,cAAQ,UAAU;AAAA,IACpB,GAAG,gBAAgB,MAAM,KAAK,QAAQ,GAAG;AAAA,EAC3C;AAAA,EAEA,MAAa,YAAY,QAA2C;AAClE,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,UAAU,KAAK,oBAAoB;AACzC,YAAM,UAAU,MAAM,YAAY,KAAK,EAAE,OAAO,CAAC,EAC9C,KAAK,EAAE,YAAY,GAAG,CAAC,EACvB,UAAU,OAAO;AACpB,YAAM,OAAyB,QAAQ,IAAI,CAAC,WAA2B;AACrE,eAAO;AAAA,UACL,MAAM,OAAO;AAAA,UACb;AAAA,UACA,UAAU,OAAO;AAAA,UACjB,OAAO,OAAO;AAAA,UACd,WAAW,OAAO;AAAA,QACpB;AAAA,MACF,CAAC;AACD,aAAO;AAAA,IACT,GAAG,eAAe,MAAM,GAAG;AAAA,EAC7B;AACF;;;AC5IA,IAAAC,mBAAsC;AACtC,IAAAA,mBAAqB;AAEd,IAAM,uBAAuB,IAAI;AAAA,EACvC;AAAA,IACC,YAAY;AAAA,MACX,MAAM;AAAA,MACN,UAAU;AAAA,MACV,QAAQ;AAAA,IACT;AAAA,IACA,QAAQ;AAAA,MACP,MAAM;AAAA,IACP;AAAA,IACA,OAAO;AAAA,MACN,MAAM;AAAA,MACN,UAAU;AAAA,IACX;AAAA,IACA,aAAa;AAAA,MACZ,MAAM;AAAA,MACN,UAAU;AAAA,IACX;AAAA,IACA,QAAQ;AAAA,MACP,MAAM;AAAA,MACN,UAAU;AAAA,MACV,SAAS;AAAA,IACV;AAAA,IACA,SAAS;AAAA,MACR,MAAM;AAAA,MACN,UAAU;AAAA,IACX;AAAA,EACD;AAAA,EACA;AAAA,IACC,YAAY;AAAA,EACb;AACD;AAWO,IAAM,gBAAgB,iBAAAC,QAAS,MAAwB,YAAY,oBAAoB;;;AClCvF,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;;;AT1DO,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,sBAAAC,QAAS,WAAW,GAAG,aAAa,MAAM;AACxC,WAAK,YAAY;AACjB,WAAK,oBAAoB;AACzB,WAAK,eAAe;AACpB,6BAAQ,MAAM,KAAK,gCAAgC;AAAA,IACrD,CAAC;AAED,sBAAAA,QAAS,WAAW,GAAG,gBAAgB,MAAM;AAC3C,WAAK,YAAY;AACjB,6BAAQ,MAAM,KAAK,sBAAsB;AACzC,WAAK,oBAAoB;AAAA,IAC3B,CAAC;AAED,sBAAAA,QAAS,WAAW,GAAG,SAAS,CAAC,UAAU;AACzC,WAAK,YAAY;AACjB,6BAAQ,MAAM,MAAM,6BAA6B,KAAK;AACtD,WAAK,oBAAoB;AAAA,IAC3B,CAAC;AAED,sBAAAA,QAAS,WAAW,GAAG,eAAe,MAAM;AAC1C,WAAK,YAAY;AACjB,WAAK,oBAAoB;AACzB,WAAK,eAAe;AACpB,6BAAQ,MAAM,KAAK,kCAAkC;AAAA,IACvD,CAAC;AAAA,EACH;AAAA,EAEA,MAAc,sBAAqC;AACjD,QAAI,KAAK,cAAc;AACrB;AAAA,IACF;AAEA,SAAK,eAAe;AAEpB,WAAO,KAAK,oBAAoB,KAAK,wBAAwB,CAAC,KAAK,aAAa;AAC9E,WAAK;AACL,6BAAQ,MAAM;AAAA,QACZ,uCAAuC,KAAK,iBAAiB,IAAI,KAAK,oBAAoB;AAAA,MAC5F;AAEA,UAAI;AACF,cAAM,kBAAAA,QAAS,QAAQ,KAAK,KAAK,KAAK,gBAAgB;AACtD,aAAK,YAAY;AACjB,aAAK,oBAAoB;AACzB,aAAK,eAAe;AACpB,+BAAQ,MAAM,KAAK,iCAAiC;AACpD;AAAA,MACF,SAAS,OAAO;AACd,+BAAQ,MAAM;AAAA,UACZ,wBAAwB,KAAK,iBAAiB;AAAA,UAC9C;AAAA,QACF;AAEA,YAAI,KAAK,oBAAoB,KAAK,sBAAsB;AACtD,gBAAM,IAAI;AAAA,YAAQ,CAAC,YACjB,WAAW,SAAS,KAAK,iBAAiB;AAAA,UAC5C;AAAA,QACF;AAAA,MACF;AAAA,IACF;AAEA,SAAK,eAAe;AAEpB,QAAI,CAAC,KAAK,aAAa;AACrB,6BAAQ,MAAM;AAAA,QACZ,wCAAwC,KAAK,oBAAoB;AAAA,MACnE;AAAA,IACF;AAAA,EACF;AAAA,EAEA,MAAa,UAAyB;AACpC,QAAI,KAAK,WAAW;AAClB;AAAA,IACF;AAEA,QAAI;AACF,YAAM,kBAAAA,QAAS,QAAQ,KAAK,KAAK,KAAK,gBAAgB;AACtD,WAAK,YAAY;AACjB,WAAK,oBAAoB;AAAA,IAC3B,SAAS,OAAO;AACd,6BAAQ,MAAM,MAAM,iCAAiC,KAAK;AAC1D,YAAM;AAAA,IACR;AAAA,EACF;AAAA,EAEA,MAAa,aAA4B;AACvC,QAAI,CAAC,KAAK,aAAa;AACrB;AAAA,IACF;AAEA,QAAI;AACF,YAAM,kBAAAA,QAAS,WAAW;AAC1B,WAAK,YAAY;AAAA,IACnB,SAAS,OAAO;AACd,6BAAQ,MAAM,MAAM,sCAAsC,KAAK;AAC/D,YAAM;AAAA,IACR;AAAA,EACF;AAAA,EAEO,cAAuB;AAC5B,WAAO,KAAK;AAAA,EACd;AAAA,EAEA,MAAc,mBAAkC;AAC9C,QAAI,CAAC,KAAK,eAAe,CAAC,KAAK,cAAc;AAC3C,YAAM,KAAK,QAAQ;AAAA,IACrB;AAGA,UAAM,cAAc;AACpB,UAAM,YAAY,KAAK,IAAI;AAC3B,WAAO,KAAK,gBAAgB,KAAK,IAAI,IAAI,YAAY,aAAa;AAChE,YAAM,IAAI,QAAQ,CAAC,YAAY,WAAW,SAAS,GAAG,CAAC;AAAA,IACzD;AAEA,QAAI,CAAC,KAAK,aAAa;AACrB,YAAM,IAAI,MAAM,kDAAkD;AAAA,IACpE;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKU,sBAA8B;AACtC,WAAO,KAAK;AAAA,EACd;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,MAAgB,iBACd,WACA,gBAAwB,sBACZ;AACZ,UAAM,KAAK,iBAAiB;AAE5B,QAAI;AACF,aAAO,MAAM,UAAU;AAAA,IACzB,SAAS,OAAY;AAEnB,UAAI,MAAM,SAAS,MAAM,MAAM,SAAS,SAAS,+BAA+B,GAAG;AACjF,+BAAQ,MAAM,MAAM,GAAG,aAAa,sBAAsB;AAC1D,cAAM;AAAA,MACR;AAGA,UACE,MAAM,SAAS,uBACf,MAAM,SAAS,sBACf,MAAM,SAAS,SAAS,YAAY,KACpC,MAAM,SAAS,SAAS,YAAY,GACpC;AACA,+BAAQ,MAAM;AAAA,UACZ,GAAG,aAAa;AAAA,QAClB;AAEA,cAAM,KAAK,iBAAiB;AAG5B,YAAI;AACF,iBAAO,MAAM,UAAU;AAAA,QACzB,SAAS,YAAiB;AACxB,iCAAQ,MAAM,MAAM,GAAG,aAAa,wBAAwB,UAAU;AACtE,gBAAM;AAAA,QACR;AAAA,MACF;AAGA,YAAM;AAAA,IACR;AAAA,EACF;AACF;","names":["import_mongoose","import_logger","import_mongoose","mongoose","import_mongoose","mongoose","import_mongoose","mongoose","import_memory","import_mongoose","mongoose","import_mongoose","mongoose","mongoose"]}

package/dist/index.js

@@ -216,24 +216,27 @@ var MongoDBWorkflow = class {
     }, "getWorkflow()");
   }
   async updateWorkflow(workflowId, updates) {
+    if (!updates.userId) {
+      throw new Error("userId is required for updateWorkflow");
+    }
     return this.executeWithRetry(async () => {
       const timeout = this.getOperationTimeout();
       await WorkflowModel.updateOne(
-        { workflowId },
+        { workflowId, userId: updates.userId },
         { $set: updates }
       ).maxTimeMS(timeout);
     }, "updateWorkflow()");
   }
-  async deleteWorkflow(workflowId) {
+  async deleteWorkflow(workflowId, userId) {
     return this.executeWithRetry(async () => {
       const timeout = this.getOperationTimeout();
-      await WorkflowModel.deleteOne({ workflowId }).maxTimeMS(timeout);
+      await WorkflowModel.deleteOne({ workflowId, userId }).maxTimeMS(timeout);
     }, "deleteWorkflow()");
   }
   async listWorkflows(userId) {
     return this.executeWithRetry(async () => {
       const timeout = this.getOperationTimeout();
-      const query = userId ? { userId } : {};
+      const query = userId ? { $or: [{ userId }, { userId: { $exists: false } }, { userId: null }] } : { $or: [{ userId: { $exists: false } }, { userId: null }] };
       const workflows = await WorkflowModel.find(query).maxTimeMS(timeout).lean();
       return workflows;
     }, "listWorkflows()");

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 AgentMetadata = {\n  agent_prompt: string;\n}\n\nexport class MongoDBAgent implements IAgentMemory {\n  private executeWithRetry: ExecuteWithRetryFn;\n  private getOperationTimeout: GetOperationTimeoutFn;\n\n  constructor(\n    executeWithRetry: ExecuteWithRetryFn,\n    getOperationTimeout: GetOperationTimeoutFn\n  ) {\n    this.executeWithRetry = executeWithRetry;\n    this.getOperationTimeout = getOperationTimeout;\n  }\n\n  public async getAgentPrompt(): Promise<string> {\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      const metadata = await AgentModel.findOne({\n        id: \"agent_metadata\"\n      }).maxTimeMS(timeout)\n        .lean<AgentMetadata>();\n      return metadata?.agent_prompt || \"\";\n    }, \"getAgentPrompt()\");\n  };\n  \n  public async updateAgentPrompt(prompt: string): Promise<void> {\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      await AgentModel.updateOne({\n        id: \"agent_metadata\",\n      }, { \"agent_prompt\": prompt }).maxTimeMS(timeout);\n    }, \"updateAgentPrompt()\");\n  };\n}\n","import 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      const messages = await MessageModel.find({ userId, threadId })\n        .sort({ timestamp: 1 })\n        .maxTimeMS(timeout);\n\n      messages?.forEach((message: MessageDocument) => {\n        message.deleteOne();\n      });\n\n      const thread = await ThreadModel.findOne({ userId, threadId }).maxTimeMS(timeout);\n      thread?.deleteOne();\n    }, `deleteThread(${userId}, ${threadId})`);\n  };\n\n  public async listThreads(userId: string): Promise<ThreadMetadata[]> {\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      const threads = await ThreadModel.find({ userId })\n        .sort({ updated_at: -1 })\n        .maxTimeMS(timeout);\n      const data: ThreadMetadata[] = threads.map((thread: ThreadDocument) => {\n        return {\n          type: thread.type,\n          userId,\n          threadId: thread.threadId,\n          title: thread.title,\n          updatedAt: thread.updated_at\n        } as ThreadMetadata;\n      })\n      return data;\n    }, `listThreads(${userId})`);\n  };\n}\n","import 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    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      await WorkflowModel.updateOne(\n        { workflowId },\n        { $set: updates }\n      ).maxTimeMS(timeout);\n    }, \"updateWorkflow()\");\n  }\n\n  public async deleteWorkflow(workflowId: string): Promise<void> {\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      await WorkflowModel.deleteOne({ workflowId }).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      const query = userId ? { userId } : {};\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;;;ACYjB,IAAM,eAAN,MAA2C;AAAA,EACxC;AAAA,EACA;AAAA,EAER,YACE,kBACA,qBACA;AACA,SAAK,mBAAmB;AACxB,SAAK,sBAAsB;AAAA,EAC7B;AAAA,EAEA,MAAa,iBAAkC;AAC7C,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,UAAU,KAAK,oBAAoB;AACzC,YAAM,WAAW,MAAM,WAAW,QAAQ;AAAA,QACxC,IAAI;AAAA,MACN,CAAC,EAAE,UAAU,OAAO,EACjB,KAAoB;AACvB,aAAO,UAAU,gBAAgB;AAAA,IACnC,GAAG,kBAAkB;AAAA,EACvB;AAAA,EAEA,MAAa,kBAAkB,QAA+B;AAC5D,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,UAAU,KAAK,oBAAoB;AACzC,YAAM,WAAW,UAAU;AAAA,QACzB,IAAI;AAAA,MACN,GAAG,EAAE,gBAAgB,OAAO,CAAC,EAAE,UAAU,OAAO;AAAA,IAClD,GAAG,qBAAqB;AAAA,EAC1B;AACF;;;AClCO,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;AACzC,YAAM,WAAW,MAAM,aAAa,KAAK,EAAE,QAAQ,SAAS,CAAC,EAC1D,KAAK,EAAE,WAAW,EAAE,CAAC,EACrB,UAAU,OAAO;AAEpB,gBAAU,QAAQ,CAAC,YAA6B;AAC9C,gBAAQ,UAAU;AAAA,MACpB,CAAC;AAED,YAAM,SAAS,MAAM,YAAY,QAAQ,EAAE,QAAQ,SAAS,CAAC,EAAE,UAAU,OAAO;AAChF,cAAQ,UAAU;AAAA,IACpB,GAAG,gBAAgB,MAAM,KAAK,QAAQ,GAAG;AAAA,EAC3C;AAAA,EAEA,MAAa,YAAY,QAA2C;AAClE,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,UAAU,KAAK,oBAAoB;AACzC,YAAM,UAAU,MAAM,YAAY,KAAK,EAAE,OAAO,CAAC,EAC9C,KAAK,EAAE,YAAY,GAAG,CAAC,EACvB,UAAU,OAAO;AACpB,YAAM,OAAyB,QAAQ,IAAI,CAAC,WAA2B;AACrE,eAAO;AAAA,UACL,MAAM,OAAO;AAAA,UACb;AAAA,UACA,UAAU,OAAO;AAAA,UACjB,OAAO,OAAO;AAAA,UACd,WAAW,OAAO;AAAA,QACpB;AAAA,MACF,CAAC;AACD,aAAO;AAAA,IACT,GAAG,eAAe,MAAM,GAAG;AAAA,EAC7B;AACF;;;ACjIO,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,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,UAAU,KAAK,oBAAoB;AACzC,YAAM,cAAc;AAAA,QAClB,EAAE,WAAW;AAAA,QACb,EAAE,MAAM,QAAQ;AAAA,MAClB,EAAE,UAAU,OAAO;AAAA,IACrB,GAAG,kBAAkB;AAAA,EACvB;AAAA,EAEA,MAAa,eAAe,YAAmC;AAC7D,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,UAAU,KAAK,oBAAoB;AACzC,YAAM,cAAc,UAAU,EAAE,WAAW,CAAC,EAAE,UAAU,OAAO;AAAA,IACjE,GAAG,kBAAkB;AAAA,EACvB;AAAA,EAEA,MAAa,cAAc,QAAsC;AAC/D,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,UAAU,KAAK,oBAAoB;AACzC,YAAM,QAAQ,SAAS,EAAE,OAAO,IAAI,CAAC;AACrC,YAAM,YAAY,MAAM,cAAc,KAAK,KAAK,EAC7C,UAAU,OAAO,EACjB,KAAiB;AACpB,aAAO;AAAA,IACT,GAAG,iBAAiB;AAAA,EACtB;AACF;;;AJjDO,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\";\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 AgentMetadata = {\n  agent_prompt: string;\n}\n\nexport class MongoDBAgent implements IAgentMemory {\n  private executeWithRetry: ExecuteWithRetryFn;\n  private getOperationTimeout: GetOperationTimeoutFn;\n\n  constructor(\n    executeWithRetry: ExecuteWithRetryFn,\n    getOperationTimeout: GetOperationTimeoutFn\n  ) {\n    this.executeWithRetry = executeWithRetry;\n    this.getOperationTimeout = getOperationTimeout;\n  }\n\n  public async getAgentPrompt(): Promise<string> {\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      const metadata = await AgentModel.findOne({\n        id: \"agent_metadata\"\n      }).maxTimeMS(timeout)\n        .lean<AgentMetadata>();\n      return metadata?.agent_prompt || \"\";\n    }, \"getAgentPrompt()\");\n  };\n  \n  public async updateAgentPrompt(prompt: string): Promise<void> {\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      await AgentModel.updateOne({\n        id: \"agent_metadata\",\n      }, { \"agent_prompt\": prompt }).maxTimeMS(timeout);\n    }, \"updateAgentPrompt()\");\n  };\n}\n","import 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      const messages = await MessageModel.find({ userId, threadId })\n        .sort({ timestamp: 1 })\n        .maxTimeMS(timeout);\n\n      messages?.forEach((message: MessageDocument) => {\n        message.deleteOne();\n      });\n\n      const thread = await ThreadModel.findOne({ userId, threadId }).maxTimeMS(timeout);\n      thread?.deleteOne();\n    }, `deleteThread(${userId}, ${threadId})`);\n  };\n\n  public async listThreads(userId: string): Promise<ThreadMetadata[]> {\n    return this.executeWithRetry(async () => {\n      const timeout = this.getOperationTimeout();\n      const threads = await ThreadModel.find({ userId })\n        .sort({ updated_at: -1 })\n        .maxTimeMS(timeout);\n      const data: ThreadMetadata[] = threads.map((thread: ThreadDocument) => {\n        return {\n          type: thread.type,\n          userId,\n          threadId: thread.threadId,\n          title: thread.title,\n          updatedAt: thread.updated_at\n        } as ThreadMetadata;\n      })\n      return data;\n    }, `listThreads(${userId})`);\n  };\n}\n","import 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;;;ACYjB,IAAM,eAAN,MAA2C;AAAA,EACxC;AAAA,EACA;AAAA,EAER,YACE,kBACA,qBACA;AACA,SAAK,mBAAmB;AACxB,SAAK,sBAAsB;AAAA,EAC7B;AAAA,EAEA,MAAa,iBAAkC;AAC7C,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,UAAU,KAAK,oBAAoB;AACzC,YAAM,WAAW,MAAM,WAAW,QAAQ;AAAA,QACxC,IAAI;AAAA,MACN,CAAC,EAAE,UAAU,OAAO,EACjB,KAAoB;AACvB,aAAO,UAAU,gBAAgB;AAAA,IACnC,GAAG,kBAAkB;AAAA,EACvB;AAAA,EAEA,MAAa,kBAAkB,QAA+B;AAC5D,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,UAAU,KAAK,oBAAoB;AACzC,YAAM,WAAW,UAAU;AAAA,QACzB,IAAI;AAAA,MACN,GAAG,EAAE,gBAAgB,OAAO,CAAC,EAAE,UAAU,OAAO;AAAA,IAClD,GAAG,qBAAqB;AAAA,EAC1B;AACF;;;AClCO,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;AACzC,YAAM,WAAW,MAAM,aAAa,KAAK,EAAE,QAAQ,SAAS,CAAC,EAC1D,KAAK,EAAE,WAAW,EAAE,CAAC,EACrB,UAAU,OAAO;AAEpB,gBAAU,QAAQ,CAAC,YAA6B;AAC9C,gBAAQ,UAAU;AAAA,MACpB,CAAC;AAED,YAAM,SAAS,MAAM,YAAY,QAAQ,EAAE,QAAQ,SAAS,CAAC,EAAE,UAAU,OAAO;AAChF,cAAQ,UAAU;AAAA,IACpB,GAAG,gBAAgB,MAAM,KAAK,QAAQ,GAAG;AAAA,EAC3C;AAAA,EAEA,MAAa,YAAY,QAA2C;AAClE,WAAO,KAAK,iBAAiB,YAAY;AACvC,YAAM,UAAU,KAAK,oBAAoB;AACzC,YAAM,UAAU,MAAM,YAAY,KAAK,EAAE,OAAO,CAAC,EAC9C,KAAK,EAAE,YAAY,GAAG,CAAC,EACvB,UAAU,OAAO;AACpB,YAAM,OAAyB,QAAQ,IAAI,CAAC,WAA2B;AACrE,eAAO;AAAA,UACL,MAAM,OAAO;AAAA,UACb;AAAA,UACA,UAAU,OAAO;AAAA,UACjB,OAAO,OAAO;AAAA,UACd,WAAW,OAAO;AAAA,QACpB;AAAA,MACF,CAAC;AACD,aAAO;AAAA,IACT,GAAG,eAAe,MAAM,GAAG;AAAA,EAC7B;AACF;;;ACjIO,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"]}

package/implements/workflow.memory.ts

@@ -40,26 +40,35 @@ export class MongoDBWorkflow implements IWorkflowMemory {
   }
 
   public async updateWorkflow(workflowId: string, updates: Partial<Workflow>): Promise<void> {
+    if (!updates.userId) {
+      throw new Error("userId is required for updateWorkflow");
+    }
+
     return this.executeWithRetry(async () => {
       const timeout = this.getOperationTimeout();
+      // userId를 조건에 포함하여 소유자만 수정 가능하도록 함
       await WorkflowModel.updateOne(
-        { workflowId },
+        { workflowId, userId: updates.userId },
         { $set: updates }
       ).maxTimeMS(timeout);
     }, "updateWorkflow()");
   }
 
-  public async deleteWorkflow(workflowId: string): Promise<void> {
+  public async deleteWorkflow(workflowId: string, userId: string): Promise<void> {
     return this.executeWithRetry(async () => {
       const timeout = this.getOperationTimeout();
-      await WorkflowModel.deleteOne({ workflowId }).maxTimeMS(timeout);
+      await WorkflowModel.deleteOne({ workflowId, userId }).maxTimeMS(timeout);
     }, "deleteWorkflow()");
   }
 
   public async listWorkflows(userId?: string): Promise<Workflow[]> {
     return this.executeWithRetry(async () => {
       const timeout = this.getOperationTimeout();
-      const query = userId ? { userId } : {};
+      // userId가 있으면: 해당 유저 소유 + 템플릿(userId 없음)
+      // userId가 없으면: 템플릿만 반환
+      const query = userId
+        ? { $or: [{ userId }, { userId: { $exists: false } }, { userId: null }] }
+        : { $or: [{ userId: { $exists: false } }, { userId: null }] };
       const workflows = await WorkflowModel.find(query)
         .maxTimeMS(timeout)
         .lean<Workflow[]>();

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ainetwork/adk-provider-memory-mongodb",
-  "version": "0.3.4",
+  "version": "0.3.5",
   "author": "AI Network (https://ainetwork.ai)",
   "type": "module",
   "engines": {
@@ -28,7 +28,7 @@
     "clean": "rm -rf dist"
   },
   "dependencies": {
-    "@ainetwork/adk": "^0.3.4",
+    "@ainetwork/adk": "^0.3.5",
     "mongoose": "^8.16.5"
   },
   "devDependencies": {
@@ -38,5 +38,5 @@
   "publishConfig": {
     "access": "public"
   },
-  "gitHead": "a1e19f88b1255e8b5d3fd6f60edb82139b324123"
+  "gitHead": "0361ececa3c7404035d1b67c4830f52fa904d3d8"
 }