@mastra/dynamodb 0.10.1 → 0.10.2

package/dist/index.cjs

@@ -904,7 +904,7 @@ var DynamoDBStore = class extends storage.MastraStorage {
     try {
       const query = this.service.entities.message.query.byThread({ entity: "message", threadId });
       if (selectBy?.last && typeof selectBy.last === "number") {
-        const results2 = await query.go({ limit: selectBy.last, reverse: true });
+        const results2 = await query.go({ limit: selectBy.last, order: "desc" });
         const list2 = new agent.MessageList({ threadId, resourceId }).add(
           results2.data.map((data) => this.parseMessageData(data)),
           "memory"

package/dist/index.js

@@ -902,7 +902,7 @@ var DynamoDBStore = class extends MastraStorage {
     try {
       const query = this.service.entities.message.query.byThread({ entity: "message", threadId });
       if (selectBy?.last && typeof selectBy.last === "number") {
-        const results2 = await query.go({ limit: selectBy.last, reverse: true });
+        const results2 = await query.go({ limit: selectBy.last, order: "desc" });
         const list2 = new MessageList({ threadId, resourceId }).add(
           results2.data.map((data) => this.parseMessageData(data)),
           "memory"

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mastra/dynamodb",
-  "version": "0.10.1",
+  "version": "0.10.2",
   "description": "DynamoDB storage adapter for Mastra",
   "type": "module",
   "main": "dist/index.js",
@@ -40,7 +40,7 @@
     "tsup": "^8.4.0",
     "typescript": "^5.8.2",
     "vitest": "^3.0.9",
-    "@internal/lint": "0.0.8",
+    "@internal/lint": "0.0.9",
     "@mastra/core": "0.10.2"
   },
   "scripts": {

package/src/storage/index.test.ts

@@ -387,6 +387,61 @@ describe('DynamoDBStore Integration Tests', () => {
         expect(retrieved?.title).toBe('Updated Thread 2');
         expect(retrieved?.metadata?.update).toBe(2);
       });
+
+      test('getMessages should return the N most recent messages [v2 storage]', async () => {
+        const threadId = 'last-selector-thread';
+        const start = Date.now();
+
+        // Insert 10 messages with increasing timestamps
+        const messages: MastraMessageV2[] = Array.from({ length: 10 }, (_, i) => ({
+          id: `m-${i}`,
+          threadId,
+          resourceId: 'r',
+          content: { format: 2, parts: [{ type: 'text', text: `msg-${i}` }] },
+          createdAt: new Date(start + i), // 0..9 ms apart
+          role: 'user',
+          type: 'text',
+        }));
+        await store.saveMessages({ messages, format: 'v2' });
+
+        const last3 = await store.getMessages({
+          format: 'v2',
+          threadId,
+          selectBy: { last: 3 },
+        });
+
+        expect(last3).toHaveLength(3);
+        expect(last3.map(m => (m.content.parts[0] as { type: string; text: string }).text)).toEqual([
+          'msg-7',
+          'msg-8',
+          'msg-9',
+        ]);
+      });
+
+      test('getMessages should return the N most recent messages [v1 storage]', async () => {
+        const threadId = 'last-selector-thread';
+        const start = Date.now();
+
+        // Insert 10 messages with increasing timestamps
+        const messages: MastraMessageV1[] = Array.from({ length: 10 }, (_, i) => ({
+          id: `m-${i}`,
+          threadId,
+          resourceId: 'r',
+          content: `msg-${i}`,
+          createdAt: new Date(start + i), // 0..9 ms apart
+          role: 'user',
+          type: 'text',
+        }));
+        await store.saveMessages({ messages });
+
+        const last3 = await store.getMessages({
+          threadId,
+          selectBy: { last: 3 },
+        });
+
+        expect(last3).toHaveLength(3);
+        expect(last3.map(m => m.content)).toEqual(['msg-7', 'msg-8', 'msg-9']);
+      });
     });
 
     describe('Batch Operations', () => {

package/src/storage/index.ts

@@ -532,10 +532,10 @@ export class DynamoDBStore extends MastraStorage {
 
       // Apply the 'last' limit if provided
       if (selectBy?.last && typeof selectBy.last === 'number') {
-        // Use ElectroDB's limit parameter (descending sort assumed on GSI SK)
-        // Ensure GSI sk (createdAt) is sorted descending for 'last' to work correctly
-        // Assuming default sort is ascending on SK, use reverse: true for descending
-        const results = await query.go({ limit: selectBy.last, reverse: true });
+        // Use ElectroDB's limit parameter
+        // DDB GSIs are sorted in ascending order
+        // Use ElectroDB's order parameter to sort in descending order to retrieve 'latest' messages
+        const results = await query.go({ limit: selectBy.last, order: 'desc' });
         // Use arrow function in map to preserve 'this' context for parseMessageData
         const list = new MessageList({ threadId, resourceId }).add(
           results.data.map((data: any) => this.parseMessageData(data)),