@mastra/pg 0.14.1 → 0.14.2

package/.turbo/turbo-build.log

@@ -1,4 +1,4 @@
 
-> @mastra/pg@0.14.1-alpha.1 build /home/runner/work/mastra/mastra/stores/pg
+> @mastra/pg@0.14.2-alpha.0 build /home/runner/work/mastra/mastra/stores/pg
 > tsup --silent --config tsup.config.ts
 

package/CHANGELOG.md

@@ -1,5 +1,23 @@
 # @mastra/pg
 
+## 0.14.2
+
+### Patch Changes
+
+- [#6700](https://github.com/mastra-ai/mastra/pull/6700) [`a5a23d9`](https://github.com/mastra-ai/mastra/commit/a5a23d981920d458dc6078919992a5338931ef02) Thanks [@gpanakkal](https://github.com/gpanakkal)! - Add `getMessagesById` method to `MastraStorage` adapters
+
+- Updated dependencies [[`6e7e120`](https://github.com/mastra-ai/mastra/commit/6e7e1207d6e8d8b838f9024f90bd10df1181ba27), [`0f00e17`](https://github.com/mastra-ai/mastra/commit/0f00e172953ccdccadb35ed3d70f5e4d89115869), [`217cd7a`](https://github.com/mastra-ai/mastra/commit/217cd7a4ce171e9a575c41bb8c83300f4db03236), [`a5a23d9`](https://github.com/mastra-ai/mastra/commit/a5a23d981920d458dc6078919992a5338931ef02)]:
+  - @mastra/core@0.14.1
+
+## 0.14.2-alpha.0
+
+### Patch Changes
+
+- [#6700](https://github.com/mastra-ai/mastra/pull/6700) [`a5a23d9`](https://github.com/mastra-ai/mastra/commit/a5a23d981920d458dc6078919992a5338931ef02) Thanks [@gpanakkal](https://github.com/gpanakkal)! - Add `getMessagesById` method to `MastraStorage` adapters
+
+- Updated dependencies [[`6e7e120`](https://github.com/mastra-ai/mastra/commit/6e7e1207d6e8d8b838f9024f90bd10df1181ba27), [`a5a23d9`](https://github.com/mastra-ai/mastra/commit/a5a23d981920d458dc6078919992a5338931ef02)]:
+  - @mastra/core@0.14.1-alpha.0
+
 ## 0.14.1
 
 ### Patch Changes

package/dist/index.cjs

@@ -1551,6 +1551,22 @@ var MemoryPG = class extends storage.MemoryStorage {
     });
     return dedupedRows;
   }
+  parseRow(row) {
+    let content = row.content;
+    try {
+      content = JSON.parse(row.content);
+    } catch {
+    }
+    return {
+      id: row.id,
+      content,
+      role: row.role,
+      createdAt: new Date(row.createdAt),
+      threadId: row.threadId,
+      resourceId: row.resourceId,
+      ...row.type && row.type !== "v2" ? { type: row.type } : {}
+    };
+  }
   async getMessages(args) {
     const { threadId, format, selectBy } = args;
     const selectStatement = `SELECT id, content, role, type, "createdAt", thread_id AS "threadId", "resourceId"`;
@@ -1609,6 +1625,40 @@ var MemoryPG = class extends storage.MemoryStorage {
       return [];
     }
   }
+  async getMessagesById({
+    messageIds,
+    format
+  }) {
+    if (messageIds.length === 0) return [];
+    const selectStatement = `SELECT id, content, role, type, "createdAt", thread_id AS "threadId", "resourceId"`;
+    try {
+      const tableName = getTableName({ indexName: storage.TABLE_MESSAGES, schemaName: getSchemaName(this.schema) });
+      const query = `
+        ${selectStatement} FROM ${tableName} 
+        WHERE id IN (${messageIds.map((_, i) => `$${i + 1}`).join(", ")})
+        ORDER BY "createdAt" DESC
+      `;
+      const resultRows = await this.client.manyOrNone(query, messageIds);
+      const list = new agent.MessageList().add(resultRows.map(this.parseRow), "memory");
+      if (format === `v1`) return list.get.all.v1();
+      return list.get.all.v2();
+    } catch (error$1) {
+      const mastraError = new error.MastraError(
+        {
+          id: "MASTRA_STORAGE_PG_STORE_GET_MESSAGES_BY_ID_FAILED",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY,
+          details: {
+            messageIds: JSON.stringify(messageIds)
+          }
+        },
+        error$1
+      );
+      this.logger?.error?.(mastraError.toString());
+      this.logger?.trackException(mastraError);
+      return [];
+    }
+  }
   async getMessagesPaginated(args) {
     const { threadId, format, selectBy } = args;
     const { page = 0, perPage: perPageInput, dateRange } = selectBy?.pagination || {};
@@ -3062,6 +3112,12 @@ var PostgresStore = class extends storage.MastraStorage {
   async getMessages(args) {
     return this.stores.memory.getMessages(args);
   }
+  async getMessagesById({
+    messageIds,
+    format
+  }) {
+    return this.stores.memory.getMessagesById({ messageIds, format });
+  }
   async getMessagesPaginated(args) {
     return this.stores.memory.getMessagesPaginated(args);
   }