@mastra/mongodb 0.13.2 → 0.13.3

package/dist/index.js

@@ -923,6 +923,29 @@ var MemoryStorageMongoDB = class extends MemoryStorage {
       );
     }
   }
+  async getMessagesById({
+    messageIds,
+    format
+  }) {
+    if (messageIds.length === 0) return [];
+    try {
+      const collection = await this.operations.getCollection(TABLE_MESSAGES);
+      const rawMessages = await collection.find({ id: { $in: messageIds } }).sort({ createdAt: -1 }).toArray();
+      const list = new MessageList().add(rawMessages.map(this.parseRow), "memory");
+      if (format === `v1`) return list.get.all.v1();
+      return list.get.all.v2();
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: "MONGODB_STORE_GET_MESSAGES_BY_ID_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { messageIds: JSON.stringify(messageIds) }
+        },
+        error
+      );
+    }
+  }
   async getMessagesPaginated(args) {
     const { threadId, format, selectBy } = args;
     const { page = 0, perPage: perPageInput, dateRange } = selectBy?.pagination || {};
@@ -2233,6 +2256,12 @@ var MongoDBStore = class extends MastraStorage {
   }) {
     return this.stores.memory.getMessages({ threadId, selectBy, format });
   }
+  async getMessagesById({
+    messageIds,
+    format
+  }) {
+    return this.stores.memory.getMessagesById({ messageIds, format });
+  }
   async saveMessages(args) {
     return this.stores.memory.saveMessages(args);
   }