@mastra/dynamodb 0.11.1-alpha.0 → 0.11.1-alpha.2

package/dist/index.cjs

@@ -3,6 +3,7 @@
 var clientDynamodb = require('@aws-sdk/client-dynamodb');
 var libDynamodb = require('@aws-sdk/lib-dynamodb');
 var agent = require('@mastra/core/agent');
+var error = require('@mastra/core/error');
 var storage = require('@mastra/core/storage');
 var electrodb = require('electrodb');
 
@@ -566,22 +567,33 @@ var DynamoDBStore = class extends storage.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 clientDynamodb.DynamoDBClient({
+        region: config.region || "us-east-1",
+        endpoint: config.endpoint,
+        credentials: config.credentials
+      });
+      this.tableName = config.tableName;
+      this.client = libDynamodb.DynamoDBDocumentClient.from(dynamoClient);
+      this.service = getElectroDbService(this.client, this.tableName);
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "STORAGE_DYNAMODB_STORE_CONSTRUCTOR_FAILED",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.USER
+        },
+        error$1
       );
     }
-    const dynamoClient = new clientDynamodb.DynamoDBClient({
-      region: config.region || "us-east-1",
-      endpoint: config.endpoint,
-      credentials: config.credentials
-    });
-    this.tableName = config.tableName;
-    this.client = libDynamodb.DynamoDBDocumentClient.from(dynamoClient);
-    this.service = getElectroDbService(this.client, this.tableName);
   }
   /**
    * This method is modified for DynamoDB with ElectroDB single-table design.
@@ -603,9 +615,17 @@ var DynamoDBStore = class extends storage.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;
+    } catch (error$1) {
+      this.logger.error("Error validating table access", { tableName: this.tableName, error: error$1 });
+      throw new error.MastraError(
+        {
+          id: "STORAGE_DYNAMODB_STORE_VALIDATE_TABLE_ACCESS_FAILED",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY,
+          details: { tableName: this.tableName }
+        },
+        error$1
+      );
     }
   }
   /**
@@ -620,11 +640,19 @@ var DynamoDBStore = class extends storage.MastraStorage {
       });
       await this.client.send(command);
       return true;
-    } catch (error) {
-      if (error.name === "ResourceNotFoundException") {
+    } catch (error$1) {
+      if (error$1.name === "ResourceNotFoundException") {
         return false;
       }
-      throw error;
+      throw new error.MastraError(
+        {
+          id: "STORAGE_DYNAMODB_STORE_VALIDATE_TABLE_EXISTS_FAILED",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY,
+          details: { tableName: this.tableName }
+        },
+        error$1
+      );
     }
   }
   /**
@@ -638,8 +666,16 @@ var DynamoDBStore = class extends storage.MastraStorage {
     }
     try {
       await this.hasInitialized;
-    } catch (error) {
-      throw error;
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "STORAGE_DYNAMODB_STORE_INIT_FAILED",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY,
+          details: { tableName: this.tableName }
+        },
+        error$1
+      );
     }
   }
   /**
@@ -685,7 +721,13 @@ var DynamoDBStore = class extends storage.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 error.MastraError({
+        id: "STORAGE_DYNAMODB_STORE_CLEAR_TABLE_INVALID_ARGS",
+        domain: error.ErrorDomain.STORAGE,
+        category: error.ErrorCategory.USER,
+        text: "No entity defined for tableName",
+        details: { tableName }
+      });
     }
     try {
       const result = await this.service.entities[entityName].scan.go({ pages: "all" });
@@ -732,36 +774,68 @@ var DynamoDBStore = class extends storage.MastraStorage {
         await this.service.entities[entityName].delete(batchKeys).go();
       }
       this.logger.debug(`Successfully cleared all records for ${tableName}`);
-    } catch (error) {
-      this.logger.error("Failed to clear table", { tableName, error });
-      throw error;
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "STORAGE_DYNAMODB_STORE_CLEAR_TABLE_FAILED",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY,
+          details: { tableName }
+        },
+        error$1
+      );
     }
   }
   /**
    * Insert a record into the specified "table" (entity)
    */
-  async insert({ tableName, record }) {
+  async insert({
+    tableName,
+    record
+  }) {
     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 error.MastraError({
+        id: "STORAGE_DYNAMODB_STORE_INSERT_INVALID_ARGS",
+        domain: error.ErrorDomain.STORAGE,
+        category: error.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;
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "STORAGE_DYNAMODB_STORE_INSERT_FAILED",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY,
+          details: { tableName }
+        },
+        error$1
+      );
     }
   }
   /**
    * Insert multiple records as a batch
    */
-  async batchInsert({ tableName, records }) {
+  async batchInsert({
+    tableName,
+    records
+  }) {
     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 error.MastraError({
+        id: "STORAGE_DYNAMODB_STORE_BATCH_INSERT_INVALID_ARGS",
+        domain: error.ErrorDomain.STORAGE,
+        category: error.ErrorCategory.USER,
+        text: "No entity defined for tableName",
+        details: { tableName }
+      });
     }
     const recordsToSave = records.map((rec) => ({ entity: entityName, ...this.preprocessRecord(rec) }));
     const batchSize = 25;
@@ -781,19 +855,35 @@ var DynamoDBStore = class extends storage.MastraStorage {
           await this.service.entities[entityName].create(recordData).go();
         }
       }
-    } catch (error) {
-      this.logger.error("Failed to batch insert records", { tableName, error });
-      throw error;
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "STORAGE_DYNAMODB_STORE_BATCH_INSERT_FAILED",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY,
+          details: { tableName }
+        },
+        error$1
+      );
     }
   }
   /**
    * Load a record by its keys
    */
-  async load({ tableName, keys }) {
+  async load({
+    tableName,
+    keys
+  }) {
     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 error.MastraError({
+        id: "STORAGE_DYNAMODB_STORE_LOAD_INVALID_ARGS",
+        domain: error.ErrorDomain.STORAGE,
+        category: error.ErrorCategory.USER,
+        text: "No entity defined for tableName",
+        details: { tableName }
+      });
     }
     try {
       const keyObject = { entity: entityName, ...keys };
@@ -803,9 +893,16 @@ var DynamoDBStore = class extends storage.MastraStorage {
       }
       let data = result.data;
       return data;
-    } catch (error) {
-      this.logger.error("Failed to load record", { tableName, keys, error });
-      throw error;
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "STORAGE_DYNAMODB_STORE_LOAD_FAILED",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY,
+          details: { tableName }
+        },
+        error$1
+      );
     }
   }
   // Thread operations
@@ -825,9 +922,16 @@ var DynamoDBStore = class extends storage.MastraStorage {
         // metadata: data.metadata ? JSON.parse(data.metadata) : undefined, // REMOVED by AI
         // metadata is already transformed by the entity's getter
       };
-    } catch (error) {
-      this.logger.error("Failed to get thread by ID", { threadId, error });
-      throw error;
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "STORAGE_DYNAMODB_STORE_GET_THREAD_BY_ID_FAILED",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY,
+          details: { threadId }
+        },
+        error$1
+      );
     }
   }
   async getThreadsByResourceId({ resourceId }) {
@@ -845,9 +949,16 @@ var DynamoDBStore = class extends storage.MastraStorage {
         // metadata: data.metadata ? JSON.parse(data.metadata) : undefined, // REMOVED by AI
         // 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;
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "STORAGE_DYNAMODB_STORE_GET_THREADS_BY_RESOURCE_ID_FAILED",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY,
+          details: { resourceId }
+        },
+        error$1
+      );
     }
   }
   async saveThread({ thread }) {
@@ -872,9 +983,16 @@ var DynamoDBStore = class extends storage.MastraStorage {
         updatedAt: now,
         metadata: thread.metadata
       };
-    } catch (error) {
-      this.logger.error("Failed to save thread", { threadId: thread.id, error });
-      throw error;
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "STORAGE_DYNAMODB_STORE_SAVE_THREAD_FAILED",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY,
+          details: { threadId: thread.id }
+        },
+        error$1
+      );
     }
   }
   async updateThread({
@@ -905,18 +1023,32 @@ var DynamoDBStore = class extends storage.MastraStorage {
         metadata: metadata || existingThread.metadata,
         updatedAt: now
       };
-    } catch (error) {
-      this.logger.error("Failed to update thread", { threadId: id, error });
-      throw error;
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "STORAGE_DYNAMODB_STORE_UPDATE_THREAD_FAILED",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY,
+          details: { threadId: id }
+        },
+        error$1
+      );
     }
   }
   async deleteThread({ threadId }) {
     this.logger.debug("Deleting thread", { threadId });
     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;
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "STORAGE_DYNAMODB_STORE_DELETE_THREAD_FAILED",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY,
+          details: { threadId }
+        },
+        error$1
+      );
     }
   }
   async getMessages({
@@ -928,8 +1060,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"
@@ -944,9 +1077,16 @@ var DynamoDBStore = class extends storage.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;
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "STORAGE_DYNAMODB_STORE_GET_MESSAGES_FAILED",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY,
+          details: { threadId }
+        },
+        error$1
+      );
     }
   }
   async saveMessages(args) {
@@ -994,7 +1134,7 @@ var DynamoDBStore = class extends storage.MastraStorage {
               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.put(messageData).go();
           }
         }),
         // Update thread's updatedAt timestamp
@@ -1005,9 +1145,16 @@ var DynamoDBStore = class extends storage.MastraStorage {
       const list = new agent.MessageList().add(messages, "memory");
       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;
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "STORAGE_DYNAMODB_STORE_SAVE_MESSAGES_FAILED",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY,
+          details: { count: messages.length }
+        },
+        error$1
+      );
     }
   }
   // Helper function to parse message data (handle JSON fields)
@@ -1052,9 +1199,15 @@ var DynamoDBStore = class extends storage.MastraStorage {
         }
       } while (cursor && pagesFetched < startPage);
       return items;
-    } catch (error) {
-      this.logger.error("Failed to get traces", { error });
-      throw error;
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "STORAGE_DYNAMODB_STORE_GET_TRACES_FAILED",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY
+        },
+        error$1
+      );
     }
   }
   async batchTraceInsert({ records }) {
@@ -1069,9 +1222,16 @@ var DynamoDBStore = class extends storage.MastraStorage {
         records: recordsToSave
         // Pass records with 'entity' included
       });
-    } catch (error) {
-      this.logger.error("Failed to batch insert traces", { error });
-      throw error;
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "STORAGE_DYNAMODB_STORE_BATCH_TRACE_INSERT_FAILED",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY,
+          details: { count: records.length }
+        },
+        error$1
+      );
     }
   }
   // Workflow operations
@@ -1096,9 +1256,16 @@ var DynamoDBStore = class extends storage.MastraStorage {
         resourceId
       };
       await this.service.entities.workflowSnapshot.upsert(data).go();
-    } catch (error) {
-      this.logger.error("Failed to persist workflow snapshot", { workflowName, runId, error });
-      throw error;
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "STORAGE_DYNAMODB_STORE_PERSIST_WORKFLOW_SNAPSHOT_FAILED",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY,
+          details: { workflowName, runId }
+        },
+        error$1
+      );
     }
   }
   async loadWorkflowSnapshot({
@@ -1117,9 +1284,16 @@ var DynamoDBStore = class extends storage.MastraStorage {
         return null;
       }
       return result.data.snapshot;
-    } catch (error) {
-      this.logger.error("Failed to load workflow snapshot", { workflowName, runId, error });
-      throw error;
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "STORAGE_DYNAMODB_STORE_LOAD_WORKFLOW_SNAPSHOT_FAILED",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY,
+          details: { workflowName, runId }
+        },
+        error$1
+      );
     }
   }
   async getWorkflowRuns(args) {
@@ -1179,9 +1353,16 @@ var DynamoDBStore = class extends storage.MastraStorage {
         runs,
         total
       };
-    } catch (error) {
-      this.logger.error("Failed to get workflow runs", { error });
-      throw error;
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "STORAGE_DYNAMODB_STORE_GET_WORKFLOW_RUNS_FAILED",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY,
+          details: { workflowName: args?.workflowName || "", resourceId: args?.resourceId || "" }
+        },
+        error$1
+      );
     }
   }
   async getWorkflowRunById(args) {
@@ -1226,9 +1407,16 @@ var DynamoDBStore = class extends storage.MastraStorage {
         updatedAt: new Date(matchingRunDbItem.updatedAt),
         resourceId: matchingRunDbItem.resourceId
       };
-    } catch (error) {
-      this.logger.error("Failed to get workflow run by ID", { runId, workflowName, error });
-      throw error;
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "STORAGE_DYNAMODB_STORE_GET_WORKFLOW_RUN_BY_ID_FAILED",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY,
+          details: { runId, workflowName: args?.workflowName || "" }
+        },
+        error$1
+      );
     }
   }
   // Helper function to format workflow run
@@ -1305,19 +1493,47 @@ var DynamoDBStore = class extends storage.MastraStorage {
           };
         }
       });
-    } catch (error) {
-      this.logger.error("Failed to get evals by agent name", { agentName, type, error });
-      throw error;
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "STORAGE_DYNAMODB_STORE_GET_EVALS_BY_AGENT_NAME_FAILED",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY,
+          details: { agentName }
+        },
+        error$1
+      );
     }
   }
   async getTracesPaginated(_args) {
-    throw new Error("Method not implemented.");
+    throw new error.MastraError(
+      {
+        id: "STORAGE_DYNAMODB_STORE_GET_TRACES_PAGINATED_FAILED",
+        domain: error.ErrorDomain.STORAGE,
+        category: error.ErrorCategory.THIRD_PARTY
+      },
+      new Error("Method not implemented.")
+    );
   }
   async getThreadsByResourceIdPaginated(_args) {
-    throw new Error("Method not implemented.");
+    throw new error.MastraError(
+      {
+        id: "STORAGE_DYNAMODB_STORE_GET_THREADS_BY_RESOURCE_ID_PAGINATED_FAILED",
+        domain: error.ErrorDomain.STORAGE,
+        category: error.ErrorCategory.THIRD_PARTY
+      },
+      new Error("Method not implemented.")
+    );
   }
   async getMessagesPaginated(_args) {
-    throw new Error("Method not implemented.");
+    throw new error.MastraError(
+      {
+        id: "STORAGE_DYNAMODB_STORE_GET_MESSAGES_PAGINATED_FAILED",
+        domain: error.ErrorDomain.STORAGE,
+        category: error.ErrorCategory.THIRD_PARTY
+      },
+      new Error("Method not implemented.")
+    );
   }
   /**
    * Closes the DynamoDB client connection and cleans up resources.
@@ -1328,9 +1544,15 @@ var DynamoDBStore = class extends storage.MastraStorage {
     try {
       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;
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "STORAGE_DYNAMODB_STORE_CLOSE_FAILED",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY
+        },
+        error$1
+      );
     }
   }
   async updateMessages(_args) {