@mastra/dynamodb 0.0.0-workflow-deno-20250616132510 → 0.0.0-working-memory-per-user-20250620163010

package/dist/_tsup-dts-rollup.d.cts

@@ -1,6 +1,7 @@
 import type { DynamoDBDocumentClient } from '@aws-sdk/lib-dynamodb';
 import { Entity } from 'electrodb';
 import type { EvalRow } from '@mastra/core/storage';
+import type { MastraMessageContentV2 } from '@mastra/core/agent';
 import type { MastraMessageV1 } from '@mastra/core/memory';
 import type { MastraMessageV2 } from '@mastra/core/memory';
 import { MastraStorage } from '@mastra/core/storage';
@@ -196,6 +197,15 @@ declare class DynamoDBStore extends MastraStorage {
      * Should be called when the store is no longer needed, e.g., at the end of tests or application shutdown.
      */
     close(): Promise<void>;
+    updateMessages(_args: {
+        messages: Partial<Omit<MastraMessageV2, 'createdAt'>> & {
+            id: string;
+            content?: {
+                metadata?: MastraMessageContentV2['metadata'];
+                content?: MastraMessageContentV2['content'];
+            };
+        }[];
+    }): Promise<MastraMessageV2[]>;
 }
 export { DynamoDBStore }
 export { DynamoDBStore as DynamoDBStore_alias_1 }

package/dist/_tsup-dts-rollup.d.ts

@@ -1,6 +1,7 @@
 import type { DynamoDBDocumentClient } from '@aws-sdk/lib-dynamodb';
 import { Entity } from 'electrodb';
 import type { EvalRow } from '@mastra/core/storage';
+import type { MastraMessageContentV2 } from '@mastra/core/agent';
 import type { MastraMessageV1 } from '@mastra/core/memory';
 import type { MastraMessageV2 } from '@mastra/core/memory';
 import { MastraStorage } from '@mastra/core/storage';
@@ -196,6 +197,15 @@ declare class DynamoDBStore extends MastraStorage {
      * Should be called when the store is no longer needed, e.g., at the end of tests or application shutdown.
      */
     close(): Promise<void>;
+    updateMessages(_args: {
+        messages: Partial<Omit<MastraMessageV2, 'createdAt'>> & {
+            id: string;
+            content?: {
+                metadata?: MastraMessageContentV2['metadata'];
+                content?: MastraMessageContentV2['content'];
+            };
+        }[];
+    }): Promise<MastraMessageV2[]>;
 }
 export { DynamoDBStore }
 export { DynamoDBStore as DynamoDBStore_alias_1 }

package/dist/index.cjs

@@ -928,8 +928,9 @@ var DynamoDBStore = class extends storage.MastraStorage {
     this.logger.debug("Getting messages", { threadId, selectBy });
     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, order: "desc" });
+      const limit = this.resolveMessageLimit({ last: selectBy?.last, defaultLimit: Number.MAX_SAFE_INTEGER });
+      if (limit !== Number.MAX_SAFE_INTEGER) {
+        const results2 = await query.go({ limit, order: "desc" });
         const list2 = new agent.MessageList({ threadId, resourceId }).add(
           results2.data.map((data) => this.parseMessageData(data)),
           "memory"
@@ -1249,7 +1250,8 @@ var DynamoDBStore = class extends storage.MastraStorage {
       [storage.TABLE_MESSAGES]: "message",
       [storage.TABLE_WORKFLOW_SNAPSHOT]: "workflowSnapshot",
       [storage.TABLE_EVALS]: "eval",
-      [storage.TABLE_TRACES]: "trace"
+      [storage.TABLE_TRACES]: "trace",
+      [storage.TABLE_RESOURCES]: "resource"
     };
     return mapping[tableName] || null;
   }
@@ -1333,6 +1335,10 @@ var DynamoDBStore = class extends storage.MastraStorage {
       throw error;
     }
   }
+  async updateMessages(_args) {
+    this.logger.error("updateMessages is not yet implemented in DynamoDBStore");
+    throw new Error("Method not implemented");
+  }
 };
 
 exports.DynamoDBStore = DynamoDBStore;

package/dist/index.js

@@ -1,7 +1,7 @@
 import { DynamoDBClient, DescribeTableCommand } from '@aws-sdk/client-dynamodb';
 import { DynamoDBDocumentClient } from '@aws-sdk/lib-dynamodb';
 import { MessageList } from '@mastra/core/agent';
-import { MastraStorage, TABLE_TRACES, TABLE_EVALS, TABLE_WORKFLOW_SNAPSHOT, TABLE_MESSAGES, TABLE_THREADS } from '@mastra/core/storage';
+import { MastraStorage, TABLE_TRACES, TABLE_RESOURCES, TABLE_EVALS, TABLE_WORKFLOW_SNAPSHOT, TABLE_MESSAGES, TABLE_THREADS } from '@mastra/core/storage';
 import { Entity, Service } from 'electrodb';
 
 // src/storage/index.ts
@@ -926,8 +926,9 @@ var DynamoDBStore = class extends MastraStorage {
     this.logger.debug("Getting messages", { threadId, selectBy });
     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, order: "desc" });
+      const limit = this.resolveMessageLimit({ last: selectBy?.last, defaultLimit: Number.MAX_SAFE_INTEGER });
+      if (limit !== Number.MAX_SAFE_INTEGER) {
+        const results2 = await query.go({ limit, order: "desc" });
         const list2 = new MessageList({ threadId, resourceId }).add(
           results2.data.map((data) => this.parseMessageData(data)),
           "memory"
@@ -1247,7 +1248,8 @@ var DynamoDBStore = class extends MastraStorage {
       [TABLE_MESSAGES]: "message",
       [TABLE_WORKFLOW_SNAPSHOT]: "workflowSnapshot",
       [TABLE_EVALS]: "eval",
-      [TABLE_TRACES]: "trace"
+      [TABLE_TRACES]: "trace",
+      [TABLE_RESOURCES]: "resource"
     };
     return mapping[tableName] || null;
   }
@@ -1331,6 +1333,10 @@ var DynamoDBStore = class extends MastraStorage {
       throw error;
     }
   }
+  async updateMessages(_args) {
+    this.logger.error("updateMessages is not yet implemented in DynamoDBStore");
+    throw new Error("Method not implemented");
+  }
 };
 
 export { DynamoDBStore };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mastra/dynamodb",
-  "version": "0.0.0-workflow-deno-20250616132510",
+  "version": "0.0.0-working-memory-per-user-20250620163010",
   "description": "DynamoDB storage adapter for Mastra",
   "type": "module",
   "main": "dist/index.js",
@@ -23,26 +23,26 @@
     "src"
   ],
   "dependencies": {
-    "@aws-sdk/client-dynamodb": "^3.826.0",
-    "@aws-sdk/lib-dynamodb": "^3.826.0",
+    "@aws-sdk/client-dynamodb": "^3.828.0",
+    "@aws-sdk/lib-dynamodb": "^3.828.0",
     "electrodb": "^3.4.3"
   },
   "peerDependencies": {
-    "@mastra/core": "0.0.0-workflow-deno-20250616132510"
+    "@mastra/core": "0.0.0-working-memory-per-user-20250620163010"
   },
   "devDependencies": {
     "@microsoft/api-extractor": "^7.52.8",
     "@types/node": "^20.19.0",
-    "@vitest/coverage-v8": "3.2.2",
-    "@vitest/ui": "3.2.2",
-    "axios": "^1.9.0",
+    "@vitest/coverage-v8": "3.2.3",
+    "@vitest/ui": "3.2.3",
+    "axios": "^1.10.0",
     "eslint": "^9.28.0",
     "tsup": "^8.5.0",
     "typescript": "^5.8.3",
     "vitest": "^3.2.3",
-    "@internal/storage-test-utils": "0.0.8",
-    "@internal/lint": "0.0.0-workflow-deno-20250616132510",
-    "@mastra/core": "0.0.0-workflow-deno-20250616132510"
+    "@internal/lint": "0.0.0-working-memory-per-user-20250620163010",
+    "@internal/storage-test-utils": "0.0.9",
+    "@mastra/core": "0.0.0-working-memory-per-user-20250620163010"
   },
   "scripts": {
     "build": "tsup src/index.ts --format esm,cjs --experimental-dts --clean --treeshake=smallest --splitting",

package/src/storage/index.ts

@@ -1,5 +1,6 @@
 import { DynamoDBClient, DescribeTableCommand } from '@aws-sdk/client-dynamodb';
 import { DynamoDBDocumentClient } from '@aws-sdk/lib-dynamodb';
+import type { MastraMessageContentV2 } from '@mastra/core/agent';
 import { MessageList } from '@mastra/core/agent';
 import type { StorageThreadType, MastraMessageV2, MastraMessageV1 } from '@mastra/core/memory';
 
@@ -10,6 +11,7 @@ import {
   TABLE_WORKFLOW_SNAPSHOT,
   TABLE_EVALS,
   TABLE_TRACES,
+  TABLE_RESOURCES,
 } from '@mastra/core/storage';
 import type {
   EvalRow,
@@ -579,12 +581,13 @@ export class DynamoDBStore extends MastraStorage {
       // Provide *all* composite key components for the 'byThread' index ('entity', 'threadId')
       const query = this.service.entities.message.query.byThread({ entity: 'message', threadId });
 
+      const limit = this.resolveMessageLimit({ last: selectBy?.last, defaultLimit: Number.MAX_SAFE_INTEGER });
       // Apply the 'last' limit if provided
-      if (selectBy?.last && typeof selectBy.last === 'number') {
+      if (limit !== Number.MAX_SAFE_INTEGER) {
         // 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' });
+        const results = await query.go({ limit, 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)),
@@ -1019,6 +1022,7 @@ export class DynamoDBStore extends MastraStorage {
       [TABLE_WORKFLOW_SNAPSHOT]: 'workflowSnapshot',
       [TABLE_EVALS]: 'eval',
       [TABLE_TRACES]: 'trace',
+      [TABLE_RESOURCES]: 'resource',
     };
     return mapping[tableName] || null;
   }
@@ -1135,4 +1139,15 @@ export class DynamoDBStore extends MastraStorage {
       throw error;
     }
   }
+
+  async updateMessages(_args: {
+    messages: Partial<Omit<MastraMessageV2, 'createdAt'>> &
+      {
+        id: string;
+        content?: { metadata?: MastraMessageContentV2['metadata']; content?: MastraMessageContentV2['content'] };
+      }[];
+  }): Promise<MastraMessageV2[]> {
+    this.logger.error('updateMessages is not yet implemented in DynamoDBStore');
+    throw new Error('Method not implemented');
+  }
 }