@mastra/dynamodb 0.10.4 → 0.11.0-alpha.1

package/dist/index.cjs

@@ -818,7 +818,10 @@ var DynamoDBStore = class extends storage.MastraStorage {
       }
       const data = result.data;
       return {
-        ...data
+        ...data,
+        // Convert date strings back to Date objects for consistency
+        createdAt: typeof data.createdAt === "string" ? new Date(data.createdAt) : data.createdAt,
+        updatedAt: typeof data.updatedAt === "string" ? new Date(data.updatedAt) : data.updatedAt
         // metadata: data.metadata ? JSON.parse(data.metadata) : undefined, // REMOVED by AI
         // metadata is already transformed by the entity's getter
       };
@@ -835,7 +838,10 @@ var DynamoDBStore = class extends storage.MastraStorage {
         return [];
       }
       return result.data.map((data) => ({
-        ...data
+        ...data,
+        // Convert date strings back to Date objects for consistency
+        createdAt: typeof data.createdAt === "string" ? new Date(data.createdAt) : data.createdAt,
+        updatedAt: typeof data.updatedAt === "string" ? new Date(data.updatedAt) : data.updatedAt
         // metadata: data.metadata ? JSON.parse(data.metadata) : undefined, // REMOVED by AI
         // metadata is already transformed by the entity's getter
       }));
@@ -949,6 +955,10 @@ var DynamoDBStore = class extends storage.MastraStorage {
     if (!messages.length) {
       return [];
     }
+    const threadId = messages[0]?.threadId;
+    if (!threadId) {
+      throw new Error("Thread ID is required");
+    }
     const messagesToSave = messages.map((msg) => {
       const now = (/* @__PURE__ */ new Date()).toISOString();
       return {
@@ -976,15 +986,22 @@ var DynamoDBStore = class extends storage.MastraStorage {
         const batch = messagesToSave.slice(i, i + batchSize);
         batches.push(batch);
       }
-      for (const batch of batches) {
-        for (const messageData of batch) {
-          if (!messageData.entity) {
-            this.logger.error("Missing entity property in message data for create", { messageData });
-            throw new Error("Internal error: Missing entity property during saveMessages");
+      await Promise.all([
+        // Process message batches
+        ...batches.map(async (batch) => {
+          for (const messageData of batch) {
+            if (!messageData.entity) {
+              this.logger.error("Missing entity property in message data for create", { messageData });
+              throw new Error("Internal error: Missing entity property during saveMessages");
+            }
+            await this.service.entities.message.create(messageData).go();
           }
-          await this.service.entities.message.create(messageData).go();
-        }
-      }
+        }),
+        // Update thread's updatedAt timestamp
+        this.service.entities.thread.update({ entity: "thread", id: threadId }).set({
+          updatedAt: (/* @__PURE__ */ new Date()).toISOString()
+        }).go()
+      ]);
       const list = new agent.MessageList().add(messages, "memory");
       if (format === `v1`) return list.get.all.v1();
       return list.get.all.v2();

package/dist/index.js

@@ -816,7 +816,10 @@ var DynamoDBStore = class extends MastraStorage {
       }
       const data = result.data;
       return {
-        ...data
+        ...data,
+        // Convert date strings back to Date objects for consistency
+        createdAt: typeof data.createdAt === "string" ? new Date(data.createdAt) : data.createdAt,
+        updatedAt: typeof data.updatedAt === "string" ? new Date(data.updatedAt) : data.updatedAt
         // metadata: data.metadata ? JSON.parse(data.metadata) : undefined, // REMOVED by AI
         // metadata is already transformed by the entity's getter
       };
@@ -833,7 +836,10 @@ var DynamoDBStore = class extends MastraStorage {
         return [];
       }
       return result.data.map((data) => ({
-        ...data
+        ...data,
+        // Convert date strings back to Date objects for consistency
+        createdAt: typeof data.createdAt === "string" ? new Date(data.createdAt) : data.createdAt,
+        updatedAt: typeof data.updatedAt === "string" ? new Date(data.updatedAt) : data.updatedAt
         // metadata: data.metadata ? JSON.parse(data.metadata) : undefined, // REMOVED by AI
         // metadata is already transformed by the entity's getter
       }));
@@ -947,6 +953,10 @@ var DynamoDBStore = class extends MastraStorage {
     if (!messages.length) {
       return [];
     }
+    const threadId = messages[0]?.threadId;
+    if (!threadId) {
+      throw new Error("Thread ID is required");
+    }
     const messagesToSave = messages.map((msg) => {
       const now = (/* @__PURE__ */ new Date()).toISOString();
       return {
@@ -974,15 +984,22 @@ var DynamoDBStore = class extends MastraStorage {
         const batch = messagesToSave.slice(i, i + batchSize);
         batches.push(batch);
       }
-      for (const batch of batches) {
-        for (const messageData of batch) {
-          if (!messageData.entity) {
-            this.logger.error("Missing entity property in message data for create", { messageData });
-            throw new Error("Internal error: Missing entity property during saveMessages");
+      await Promise.all([
+        // Process message batches
+        ...batches.map(async (batch) => {
+          for (const messageData of batch) {
+            if (!messageData.entity) {
+              this.logger.error("Missing entity property in message data for create", { messageData });
+              throw new Error("Internal error: Missing entity property during saveMessages");
+            }
+            await this.service.entities.message.create(messageData).go();
           }
-          await this.service.entities.message.create(messageData).go();
-        }
-      }
+        }),
+        // Update thread's updatedAt timestamp
+        this.service.entities.thread.update({ entity: "thread", id: threadId }).set({
+          updatedAt: (/* @__PURE__ */ new Date()).toISOString()
+        }).go()
+      ]);
       const list = new MessageList().add(messages, "memory");
       if (format === `v1`) return list.get.all.v1();
       return list.get.all.v2();

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mastra/dynamodb",
-  "version": "0.10.4",
+  "version": "0.11.0-alpha.1",
   "description": "DynamoDB storage adapter for Mastra",
   "type": "module",
   "main": "dist/index.js",
@@ -23,26 +23,26 @@
     "src"
   ],
   "dependencies": {
-    "@aws-sdk/client-dynamodb": "^3.823.0",
-    "@aws-sdk/lib-dynamodb": "^3.823.0",
-    "electrodb": "^3.4.1"
+    "@aws-sdk/client-dynamodb": "^3.826.0",
+    "@aws-sdk/lib-dynamodb": "^3.826.0",
+    "electrodb": "^3.4.3"
   },
   "peerDependencies": {
     "@mastra/core": ">=0.10.4-0 <0.11.0"
   },
   "devDependencies": {
     "@microsoft/api-extractor": "^7.52.8",
-    "@types/node": "^20.17.57",
+    "@types/node": "^20.19.0",
     "@vitest/coverage-v8": "3.2.2",
     "@vitest/ui": "3.2.2",
     "axios": "^1.9.0",
     "eslint": "^9.28.0",
     "tsup": "^8.5.0",
-    "typescript": "^5.8.2",
-    "vitest": "^3.2.2",
-    "@internal/lint": "0.0.11",
-    "@mastra/core": "0.10.4",
-    "@internal/storage-test-utils": "0.0.7"
+    "typescript": "^5.8.3",
+    "vitest": "^3.2.3",
+    "@internal/lint": "0.0.12",
+    "@internal/storage-test-utils": "0.0.8",
+    "@mastra/core": "0.10.6-alpha.0"
   },
   "scripts": {
     "build": "tsup src/index.ts --format esm,cjs --experimental-dts --clean --treeshake=smallest --splitting",

package/src/storage/index.test.ts

@@ -325,6 +325,7 @@ describe('DynamoDBStore Integration Tests', () => {
           suspendedPaths: { test: [1] },
           runId: 'test-run-large', // Use unique runId
           timestamp: now,
+          status: 'success',
         };
 
         await expect(
@@ -442,6 +443,43 @@ describe('DynamoDBStore Integration Tests', () => {
         expect(last3).toHaveLength(3);
         expect(last3.map(m => m.content)).toEqual(['msg-7', 'msg-8', 'msg-9']);
       });
+
+      test('should update thread updatedAt when a message is saved to it', async () => {
+        const thread: StorageThreadType = {
+          id: 'thread-update-test',
+          resourceId: 'resource-update',
+          title: 'Update Test Thread',
+          createdAt: new Date(),
+          updatedAt: new Date(),
+          metadata: { test: true },
+        };
+        await store.saveThread({ thread });
+
+        // Get the initial thread to capture the original updatedAt
+        const initialThread = await store.getThreadById({ threadId: thread.id });
+        expect(initialThread).toBeDefined();
+        const originalUpdatedAt = initialThread!.updatedAt;
+
+        // Wait a small amount to ensure different timestamp
+        await new Promise(resolve => setTimeout(resolve, 100));
+
+        // Create and save a message to the thread
+        const message: MastraMessageV1 = {
+          id: 'msg-update-test',
+          threadId: thread.id,
+          resourceId: 'resource-update',
+          content: 'Test message for update',
+          createdAt: new Date(),
+          role: 'user',
+          type: 'text',
+        };
+        await store.saveMessages({ messages: [message] });
+
+        // Retrieve the thread again and check that updatedAt was updated
+        const updatedThread = await store.getThreadById({ threadId: thread.id });
+        expect(updatedThread).toBeDefined();
+        expect(updatedThread!.updatedAt.getTime()).toBeGreaterThan(originalUpdatedAt.getTime());
+      });
     });
 
     describe('Batch Operations', () => {
@@ -547,6 +585,7 @@ describe('DynamoDBStore Integration Tests', () => {
           suspendedPaths: { test: [1] },
           runId: 'mixed-run',
           timestamp: Date.now(),
+          status: 'success',
         };
         await store.persistWorkflowSnapshot({ workflowName, runId: 'mixed-run', snapshot: workflowSnapshot });
 
@@ -861,6 +900,7 @@ describe('DynamoDBStore Integration Tests', () => {
         suspendedPaths: {},
         runId: runId,
         timestamp: createdAt.getTime(),
+        status: 'success',
         ...(resourceId && { resourceId: resourceId }), // Conditionally add resourceId to snapshot
       };
       return {

package/src/storage/index.ts

@@ -422,6 +422,9 @@ export class DynamoDBStore extends MastraStorage {
       const data = result.data;
       return {
         ...data,
+        // Convert date strings back to Date objects for consistency
+        createdAt: typeof data.createdAt === 'string' ? new Date(data.createdAt) : data.createdAt,
+        updatedAt: typeof data.updatedAt === 'string' ? new Date(data.updatedAt) : data.updatedAt,
         // metadata: data.metadata ? JSON.parse(data.metadata) : undefined, // REMOVED by AI
         // metadata is already transformed by the entity's getter
       } as StorageThreadType;
@@ -443,6 +446,9 @@ export class DynamoDBStore extends MastraStorage {
       // ElectroDB handles the transformation with attribute getters
       return result.data.map((data: any) => ({
         ...data,
+        // Convert date strings back to Date objects for consistency
+        createdAt: typeof data.createdAt === 'string' ? new Date(data.createdAt) : data.createdAt,
+        updatedAt: typeof data.updatedAt === 'string' ? new Date(data.updatedAt) : data.updatedAt,
         // metadata: data.metadata ? JSON.parse(data.metadata) : undefined, // REMOVED by AI
         // metadata is already transformed by the entity's getter
       })) as StorageThreadType[];
@@ -614,6 +620,11 @@ export class DynamoDBStore extends MastraStorage {
       return [];
     }
 
+    const threadId = messages[0]?.threadId;
+    if (!threadId) {
+      throw new Error('Thread ID is required');
+    }
+
     // Ensure 'entity' is added and complex fields are handled
     const messagesToSave = messages.map(msg => {
       const now = new Date().toISOString();
@@ -644,19 +655,27 @@ export class DynamoDBStore extends MastraStorage {
         batches.push(batch);
       }
 
-      // Process each batch
-      for (const batch of batches) {
-        // Try creating each item individually instead of passing the whole batch
-        for (const messageData of batch) {
-          // Ensure each item has the entity property before sending
-          if (!messageData.entity) {
-            this.logger.error('Missing entity property in message data for create', { messageData });
-            throw new Error('Internal error: Missing entity property during saveMessages');
+      // Process each batch and update thread's updatedAt in parallel for better performance
+      await Promise.all([
+        // Process message batches
+        ...batches.map(async batch => {
+          for (const messageData of batch) {
+            // Ensure each item has the entity property before sending
+            if (!messageData.entity) {
+              this.logger.error('Missing entity property in message data for create', { messageData });
+              throw new Error('Internal error: Missing entity property during saveMessages');
+            }
+            await this.service.entities.message.create(messageData).go();
           }
-          await this.service.entities.message.create(messageData).go();
-        }
-        // Original batch call: await this.service.entities.message.create(batch).go();
-      }
+        }),
+        // Update thread's updatedAt timestamp
+        this.service.entities.thread
+          .update({ entity: 'thread', id: threadId })
+          .set({
+            updatedAt: new Date().toISOString(),
+          })
+          .go(),
+      ]);
 
       const list = new MessageList().add(messages, 'memory');
       if (format === `v1`) return list.get.all.v1();