@mastra/upstash 0.14.1 → 0.14.2

package/dist/index.js

@@ -622,6 +622,15 @@ var StoreMemoryUpstash = class extends MemoryStorage {
     }
     return [];
   }
+  parseStoredMessage(storedMessage) {
+    const defaultMessageContent = { format: 2, parts: [{ type: "text", text: "" }] };
+    const { _index, ...rest } = storedMessage;
+    return {
+      ...rest,
+      createdAt: new Date(rest.createdAt),
+      content: rest.content || defaultMessageContent
+    };
+  }
   async getMessages({
     threadId,
     selectBy,
@@ -697,6 +706,40 @@ var StoreMemoryUpstash = class extends MemoryStorage {
       );
     }
   }
+  async getMessagesById({
+    messageIds,
+    format
+  }) {
+    if (messageIds.length === 0) return [];
+    try {
+      const threadKeys = await this.client.keys("thread:*");
+      const result = await Promise.all(
+        threadKeys.map((threadKey) => {
+          const threadId = threadKey.split(":")[1];
+          if (!threadId) throw new Error(`Failed to parse thread ID from thread key "${threadKey}"`);
+          return this.client.mget(
+            messageIds.map((id) => getMessageKey(threadId, id))
+          );
+        })
+      );
+      const rawMessages = result.flat(1).filter((msg) => !!msg);
+      const list = new MessageList().add(rawMessages.map(this.parseStoredMessage), "memory");
+      if (format === `v1`) return list.get.all.v1();
+      return list.get.all.v2();
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: "STORAGE_UPSTASH_STORAGE_GET_MESSAGES_BY_ID_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: {
+            messageIds: JSON.stringify(messageIds)
+          }
+        },
+        error
+      );
+    }
+  }
   async getMessagesPaginated(args) {
     const { threadId, selectBy, format } = args;
     const { page = 0, perPage = 40, dateRange } = selectBy?.pagination || {};
@@ -1777,6 +1820,12 @@ var UpstashStore = class extends MastraStorage {
   }) {
     return this.stores.memory.getMessages({ threadId, selectBy, format });
   }
+  async getMessagesById({
+    messageIds,
+    format
+  }) {
+    return this.stores.memory.getMessagesById({ messageIds, format });
+  }
   async getMessagesPaginated(args) {
     return this.stores.memory.getMessagesPaginated(args);
   }