@mastra/dynamodb 0.11.0 → 0.11.1-alpha.1

package/dist/index.js

@@ -1,6 +1,7 @@
 import { DynamoDBClient, DescribeTableCommand } from '@aws-sdk/client-dynamodb';
 import { DynamoDBDocumentClient } from '@aws-sdk/lib-dynamodb';
 import { MessageList } from '@mastra/core/agent';
+import { MastraError, ErrorCategory, ErrorDomain } from '@mastra/core/error';
 import { MastraStorage, TABLE_TRACES, TABLE_EVALS, TABLE_WORKFLOW_SNAPSHOT, TABLE_MESSAGES, TABLE_THREADS } from '@mastra/core/storage';
 import { Entity, Service } from 'electrodb';
 
@@ -564,22 +565,33 @@ var DynamoDBStore = class extends MastraStorage {
   hasInitialized = null;
   constructor({ name, config }) {
     super({ name });
-    if (!config.tableName || typeof config.tableName !== "string" || config.tableName.trim() === "") {
-      throw new Error("DynamoDBStore: config.tableName must be provided and cannot be empty.");
-    }
-    if (!/^[a-zA-Z0-9_.-]{3,255}$/.test(config.tableName)) {
-      throw new Error(
-        `DynamoDBStore: config.tableName "${config.tableName}" contains invalid characters or is not between 3 and 255 characters long.`
+    try {
+      if (!config.tableName || typeof config.tableName !== "string" || config.tableName.trim() === "") {
+        throw new Error("DynamoDBStore: config.tableName must be provided and cannot be empty.");
+      }
+      if (!/^[a-zA-Z0-9_.-]{3,255}$/.test(config.tableName)) {
+        throw new Error(
+          `DynamoDBStore: config.tableName "${config.tableName}" contains invalid characters or is not between 3 and 255 characters long.`
+        );
+      }
+      const dynamoClient = new DynamoDBClient({
+        region: config.region || "us-east-1",
+        endpoint: config.endpoint,
+        credentials: config.credentials
+      });
+      this.tableName = config.tableName;
+      this.client = DynamoDBDocumentClient.from(dynamoClient);
+      this.service = getElectroDbService(this.client, this.tableName);
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: "STORAGE_DYNAMODB_STORE_CONSTRUCTOR_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.USER
+        },
+        error
       );
     }
-    const dynamoClient = new DynamoDBClient({
-      region: config.region || "us-east-1",
-      endpoint: config.endpoint,
-      credentials: config.credentials
-    });
-    this.tableName = config.tableName;
-    this.client = DynamoDBDocumentClient.from(dynamoClient);
-    this.service = getElectroDbService(this.client, this.tableName);
   }
   /**
    * This method is modified for DynamoDB with ElectroDB single-table design.
@@ -603,7 +615,15 @@ var DynamoDBStore = class extends MastraStorage {
       this.logger.debug(`Table ${this.tableName} exists and is accessible`);
     } catch (error) {
       this.logger.error("Error validating table access", { tableName: this.tableName, error });
-      throw error;
+      throw new MastraError(
+        {
+          id: "STORAGE_DYNAMODB_STORE_VALIDATE_TABLE_ACCESS_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { tableName: this.tableName }
+        },
+        error
+      );
     }
   }
   /**
@@ -622,7 +642,15 @@ var DynamoDBStore = class extends MastraStorage {
       if (error.name === "ResourceNotFoundException") {
         return false;
       }
-      throw error;
+      throw new MastraError(
+        {
+          id: "STORAGE_DYNAMODB_STORE_VALIDATE_TABLE_EXISTS_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { tableName: this.tableName }
+        },
+        error
+      );
     }
   }
   /**
@@ -637,7 +665,15 @@ var DynamoDBStore = class extends MastraStorage {
     try {
       await this.hasInitialized;
     } catch (error) {
-      throw error;
+      throw new MastraError(
+        {
+          id: "STORAGE_DYNAMODB_STORE_INIT_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { tableName: this.tableName }
+        },
+        error
+      );
     }
   }
   /**
@@ -683,7 +719,13 @@ var DynamoDBStore = class extends MastraStorage {
     this.logger.debug("DynamoDB clearTable called", { tableName });
     const entityName = this.getEntityNameForTable(tableName);
     if (!entityName || !this.service.entities[entityName]) {
-      throw new Error(`No entity defined for ${tableName}`);
+      throw new MastraError({
+        id: "STORAGE_DYNAMODB_STORE_CLEAR_TABLE_INVALID_ARGS",
+        domain: ErrorDomain.STORAGE,
+        category: ErrorCategory.USER,
+        text: "No entity defined for tableName",
+        details: { tableName }
+      });
     }
     try {
       const result = await this.service.entities[entityName].scan.go({ pages: "all" });
@@ -731,8 +773,15 @@ var DynamoDBStore = class extends MastraStorage {
       }
       this.logger.debug(`Successfully cleared all records for ${tableName}`);
     } catch (error) {
-      this.logger.error("Failed to clear table", { tableName, error });
-      throw error;
+      throw new MastraError(
+        {
+          id: "STORAGE_DYNAMODB_STORE_CLEAR_TABLE_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { tableName }
+        },
+        error
+      );
     }
   }
   /**
@@ -742,14 +791,27 @@ var DynamoDBStore = class extends MastraStorage {
     this.logger.debug("DynamoDB insert called", { tableName });
     const entityName = this.getEntityNameForTable(tableName);
     if (!entityName || !this.service.entities[entityName]) {
-      throw new Error(`No entity defined for ${tableName}`);
+      throw new MastraError({
+        id: "STORAGE_DYNAMODB_STORE_INSERT_INVALID_ARGS",
+        domain: ErrorDomain.STORAGE,
+        category: ErrorCategory.USER,
+        text: "No entity defined for tableName",
+        details: { tableName }
+      });
     }
     try {
       const dataToSave = { entity: entityName, ...this.preprocessRecord(record) };
       await this.service.entities[entityName].create(dataToSave).go();
     } catch (error) {
-      this.logger.error("Failed to insert record", { tableName, error });
-      throw error;
+      throw new MastraError(
+        {
+          id: "STORAGE_DYNAMODB_STORE_INSERT_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { tableName }
+        },
+        error
+      );
     }
   }
   /**
@@ -759,7 +821,13 @@ var DynamoDBStore = class extends MastraStorage {
     this.logger.debug("DynamoDB batchInsert called", { tableName, count: records.length });
     const entityName = this.getEntityNameForTable(tableName);
     if (!entityName || !this.service.entities[entityName]) {
-      throw new Error(`No entity defined for ${tableName}`);
+      throw new MastraError({
+        id: "STORAGE_DYNAMODB_STORE_BATCH_INSERT_INVALID_ARGS",
+        domain: ErrorDomain.STORAGE,
+        category: ErrorCategory.USER,
+        text: "No entity defined for tableName",
+        details: { tableName }
+      });
     }
     const recordsToSave = records.map((rec) => ({ entity: entityName, ...this.preprocessRecord(rec) }));
     const batchSize = 25;
@@ -780,8 +848,15 @@ var DynamoDBStore = class extends MastraStorage {
         }
       }
     } catch (error) {
-      this.logger.error("Failed to batch insert records", { tableName, error });
-      throw error;
+      throw new MastraError(
+        {
+          id: "STORAGE_DYNAMODB_STORE_BATCH_INSERT_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { tableName }
+        },
+        error
+      );
     }
   }
   /**
@@ -791,7 +866,13 @@ var DynamoDBStore = class extends MastraStorage {
     this.logger.debug("DynamoDB load called", { tableName, keys });
     const entityName = this.getEntityNameForTable(tableName);
     if (!entityName || !this.service.entities[entityName]) {
-      throw new Error(`No entity defined for ${tableName}`);
+      throw new MastraError({
+        id: "STORAGE_DYNAMODB_STORE_LOAD_INVALID_ARGS",
+        domain: ErrorDomain.STORAGE,
+        category: ErrorCategory.USER,
+        text: "No entity defined for tableName",
+        details: { tableName }
+      });
     }
     try {
       const keyObject = { entity: entityName, ...keys };
@@ -802,8 +883,15 @@ var DynamoDBStore = class extends MastraStorage {
       let data = result.data;
       return data;
     } catch (error) {
-      this.logger.error("Failed to load record", { tableName, keys, error });
-      throw error;
+      throw new MastraError(
+        {
+          id: "STORAGE_DYNAMODB_STORE_LOAD_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { tableName }
+        },
+        error
+      );
     }
   }
   // Thread operations
@@ -824,8 +912,15 @@ var DynamoDBStore = class extends MastraStorage {
         // metadata is already transformed by the entity's getter
       };
     } catch (error) {
-      this.logger.error("Failed to get thread by ID", { threadId, error });
-      throw error;
+      throw new MastraError(
+        {
+          id: "STORAGE_DYNAMODB_STORE_GET_THREAD_BY_ID_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { threadId }
+        },
+        error
+      );
     }
   }
   async getThreadsByResourceId({ resourceId }) {
@@ -844,8 +939,15 @@ var DynamoDBStore = class extends MastraStorage {
         // metadata is already transformed by the entity's getter
       }));
     } catch (error) {
-      this.logger.error("Failed to get threads by resource ID", { resourceId, error });
-      throw error;
+      throw new MastraError(
+        {
+          id: "STORAGE_DYNAMODB_STORE_GET_THREADS_BY_RESOURCE_ID_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { resourceId }
+        },
+        error
+      );
     }
   }
   async saveThread({ thread }) {
@@ -871,8 +973,15 @@ var DynamoDBStore = class extends MastraStorage {
         metadata: thread.metadata
       };
     } catch (error) {
-      this.logger.error("Failed to save thread", { threadId: thread.id, error });
-      throw error;
+      throw new MastraError(
+        {
+          id: "STORAGE_DYNAMODB_STORE_SAVE_THREAD_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { threadId: thread.id }
+        },
+        error
+      );
     }
   }
   async updateThread({
@@ -904,8 +1013,15 @@ var DynamoDBStore = class extends MastraStorage {
         updatedAt: now
       };
     } catch (error) {
-      this.logger.error("Failed to update thread", { threadId: id, error });
-      throw error;
+      throw new MastraError(
+        {
+          id: "STORAGE_DYNAMODB_STORE_UPDATE_THREAD_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { threadId: id }
+        },
+        error
+      );
     }
   }
   async deleteThread({ threadId }) {
@@ -913,8 +1029,15 @@ var DynamoDBStore = class extends MastraStorage {
     try {
       await this.service.entities.thread.delete({ entity: "thread", id: threadId }).go();
     } catch (error) {
-      this.logger.error("Failed to delete thread", { threadId, error });
-      throw error;
+      throw new MastraError(
+        {
+          id: "STORAGE_DYNAMODB_STORE_DELETE_THREAD_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { threadId }
+        },
+        error
+      );
     }
   }
   async getMessages({
@@ -926,8 +1049,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"
@@ -943,8 +1067,15 @@ var DynamoDBStore = class extends MastraStorage {
       if (format === `v2`) return list.get.all.v2();
       return list.get.all.v1();
     } catch (error) {
-      this.logger.error("Failed to get messages", { threadId, error });
-      throw error;
+      throw new MastraError(
+        {
+          id: "STORAGE_DYNAMODB_STORE_GET_MESSAGES_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { threadId }
+        },
+        error
+      );
     }
   }
   async saveMessages(args) {
@@ -1004,8 +1135,15 @@ var DynamoDBStore = class extends MastraStorage {
       if (format === `v1`) return list.get.all.v1();
       return list.get.all.v2();
     } catch (error) {
-      this.logger.error("Failed to save messages", { error });
-      throw error;
+      throw new MastraError(
+        {
+          id: "STORAGE_DYNAMODB_STORE_SAVE_MESSAGES_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { count: messages.length }
+        },
+        error
+      );
     }
   }
   // Helper function to parse message data (handle JSON fields)
@@ -1051,8 +1189,14 @@ var DynamoDBStore = class extends MastraStorage {
       } while (cursor && pagesFetched < startPage);
       return items;
     } catch (error) {
-      this.logger.error("Failed to get traces", { error });
-      throw error;
+      throw new MastraError(
+        {
+          id: "STORAGE_DYNAMODB_STORE_GET_TRACES_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY
+        },
+        error
+      );
     }
   }
   async batchTraceInsert({ records }) {
@@ -1068,8 +1212,15 @@ var DynamoDBStore = class extends MastraStorage {
         // Pass records with 'entity' included
       });
     } catch (error) {
-      this.logger.error("Failed to batch insert traces", { error });
-      throw error;
+      throw new MastraError(
+        {
+          id: "STORAGE_DYNAMODB_STORE_BATCH_TRACE_INSERT_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { count: records.length }
+        },
+        error
+      );
     }
   }
   // Workflow operations
@@ -1095,8 +1246,15 @@ var DynamoDBStore = class extends MastraStorage {
       };
       await this.service.entities.workflowSnapshot.upsert(data).go();
     } catch (error) {
-      this.logger.error("Failed to persist workflow snapshot", { workflowName, runId, error });
-      throw error;
+      throw new MastraError(
+        {
+          id: "STORAGE_DYNAMODB_STORE_PERSIST_WORKFLOW_SNAPSHOT_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { workflowName, runId }
+        },
+        error
+      );
     }
   }
   async loadWorkflowSnapshot({
@@ -1116,8 +1274,15 @@ var DynamoDBStore = class extends MastraStorage {
       }
       return result.data.snapshot;
     } catch (error) {
-      this.logger.error("Failed to load workflow snapshot", { workflowName, runId, error });
-      throw error;
+      throw new MastraError(
+        {
+          id: "STORAGE_DYNAMODB_STORE_LOAD_WORKFLOW_SNAPSHOT_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { workflowName, runId }
+        },
+        error
+      );
     }
   }
   async getWorkflowRuns(args) {
@@ -1178,8 +1343,15 @@ var DynamoDBStore = class extends MastraStorage {
         total
       };
     } catch (error) {
-      this.logger.error("Failed to get workflow runs", { error });
-      throw error;
+      throw new MastraError(
+        {
+          id: "STORAGE_DYNAMODB_STORE_GET_WORKFLOW_RUNS_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { workflowName: args?.workflowName || "", resourceId: args?.resourceId || "" }
+        },
+        error
+      );
     }
   }
   async getWorkflowRunById(args) {
@@ -1225,8 +1397,15 @@ var DynamoDBStore = class extends MastraStorage {
         resourceId: matchingRunDbItem.resourceId
       };
     } catch (error) {
-      this.logger.error("Failed to get workflow run by ID", { runId, workflowName, error });
-      throw error;
+      throw new MastraError(
+        {
+          id: "STORAGE_DYNAMODB_STORE_GET_WORKFLOW_RUN_BY_ID_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { runId, workflowName: args?.workflowName || "" }
+        },
+        error
+      );
     }
   }
   // Helper function to format workflow run
@@ -1304,18 +1483,46 @@ var DynamoDBStore = class extends MastraStorage {
         }
       });
     } catch (error) {
-      this.logger.error("Failed to get evals by agent name", { agentName, type, error });
-      throw error;
+      throw new MastraError(
+        {
+          id: "STORAGE_DYNAMODB_STORE_GET_EVALS_BY_AGENT_NAME_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { agentName }
+        },
+        error
+      );
     }
   }
   async getTracesPaginated(_args) {
-    throw new Error("Method not implemented.");
+    throw new MastraError(
+      {
+        id: "STORAGE_DYNAMODB_STORE_GET_TRACES_PAGINATED_FAILED",
+        domain: ErrorDomain.STORAGE,
+        category: ErrorCategory.THIRD_PARTY
+      },
+      new Error("Method not implemented.")
+    );
   }
   async getThreadsByResourceIdPaginated(_args) {
-    throw new Error("Method not implemented.");
+    throw new MastraError(
+      {
+        id: "STORAGE_DYNAMODB_STORE_GET_THREADS_BY_RESOURCE_ID_PAGINATED_FAILED",
+        domain: ErrorDomain.STORAGE,
+        category: ErrorCategory.THIRD_PARTY
+      },
+      new Error("Method not implemented.")
+    );
   }
   async getMessagesPaginated(_args) {
-    throw new Error("Method not implemented.");
+    throw new MastraError(
+      {
+        id: "STORAGE_DYNAMODB_STORE_GET_MESSAGES_PAGINATED_FAILED",
+        domain: ErrorDomain.STORAGE,
+        category: ErrorCategory.THIRD_PARTY
+      },
+      new Error("Method not implemented.")
+    );
   }
   /**
    * Closes the DynamoDB client connection and cleans up resources.
@@ -1327,10 +1534,20 @@ var DynamoDBStore = class extends MastraStorage {
       this.client.destroy();
       this.logger.debug("DynamoDB client closed successfully for store:", { name: this.name });
     } catch (error) {
-      this.logger.error("Error closing DynamoDB client for store:", { name: this.name, error });
-      throw error;
+      throw new MastraError(
+        {
+          id: "STORAGE_DYNAMODB_STORE_CLOSE_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY
+        },
+        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.11.0",
+  "version": "0.11.1-alpha.1",
   "description": "DynamoDB storage adapter for Mastra",
   "type": "module",
   "main": "dist/index.js",
@@ -41,8 +41,8 @@
     "typescript": "^5.8.3",
     "vitest": "^3.2.3",
     "@internal/lint": "0.0.13",
-    "@internal/storage-test-utils": "0.0.9",
-    "@mastra/core": "0.10.6"
+    "@mastra/core": "0.10.7-alpha.1",
+    "@internal/storage-test-utils": "0.0.9"
   },
   "scripts": {
     "build": "tsup src/index.ts --format esm,cjs --experimental-dts --clean --treeshake=smallest --splitting",