@mastra/dynamodb 0.10.1-alpha.2 → 0.10.2

package/dist/index.cjs

@@ -560,9 +560,12 @@ function getElectroDbService(client, tableName) {
 
 // src/storage/index.ts
 var DynamoDBStore = class extends storage.MastraStorage {
+  tableName;
+  client;
+  service;
+  hasInitialized = null;
   constructor({ name, config }) {
     super({ name });
-    this.hasInitialized = null;
     if (!config.tableName || typeof config.tableName !== "string" || config.tableName.trim() === "") {
       throw new Error("DynamoDBStore: config.tableName must be provided and cannot be empty.");
     }
@@ -901,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.d.cts

@@ -0,0 +1,2 @@
+export { DynamoDBStoreConfig } from './_tsup-dts-rollup.cjs';
+export { DynamoDBStore } from './_tsup-dts-rollup.cjs';

package/dist/index.d.ts

@@ -0,0 +1,2 @@
+export { DynamoDBStoreConfig } from './_tsup-dts-rollup.js';
+export { DynamoDBStore } from './_tsup-dts-rollup.js';

package/dist/index.js

@@ -558,9 +558,12 @@ function getElectroDbService(client, tableName) {
 
 // src/storage/index.ts
 var DynamoDBStore = class extends MastraStorage {
+  tableName;
+  client;
+  service;
+  hasInitialized = null;
   constructor({ name, config }) {
     super({ name });
-    this.hasInitialized = null;
     if (!config.tableName || typeof config.tableName !== "string" || config.tableName.trim() === "") {
       throw new Error("DynamoDBStore: config.tableName must be provided and cannot be empty.");
     }
@@ -899,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,10 +1,23 @@
 {
   "name": "@mastra/dynamodb",
-  "version": "0.10.1-alpha.2",
+  "version": "0.10.2",
   "description": "DynamoDB storage adapter for Mastra",
   "type": "module",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
+  "exports": {
+    ".": {
+      "import": {
+        "types": "./dist/index.d.ts",
+        "default": "./dist/index.js"
+      },
+      "require": {
+        "types": "./dist/index.d.cts",
+        "default": "./dist/index.cjs"
+      }
+    },
+    "./package.json": "./package.json"
+  },
   "files": [
     "dist",
     "src"
@@ -15,7 +28,7 @@
     "electrodb": "^3.4.1"
   },
   "peerDependencies": {
-    "@mastra/core": "^0.10.0-alpha.0"
+    "@mastra/core": "^0.10.2-alpha.0"
   },
   "devDependencies": {
     "@microsoft/api-extractor": "^7.52.1",
@@ -27,11 +40,11 @@
     "tsup": "^8.4.0",
     "typescript": "^5.8.2",
     "vitest": "^3.0.9",
-    "@internal/lint": "0.0.7",
-    "@mastra/core": "0.10.2-alpha.4"
+    "@internal/lint": "0.0.9",
+    "@mastra/core": "0.10.2"
   },
   "scripts": {
-    "build": "tsup src/index.ts --format esm,cjs --clean --treeshake=smallest --splitting",
+    "build": "tsup src/index.ts --format esm,cjs --experimental-dts --clean --treeshake=smallest --splitting",
     "dev": "tsup --watch",
     "clean": "rm -rf dist",
     "lint": "eslint .",

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)),