@mastra/dynamodb 1.0.1 → 1.0.2-alpha.0

package/CHANGELOG.md

@@ -1,5 +1,14 @@
 # @mastra/dynamodb
 
+## 1.0.2-alpha.0
+
+### Patch Changes
+
+- Added atomic `updateWorkflowResults` and `updateWorkflowState` to safely merge concurrent step results into workflow snapshots. ([#12575](https://github.com/mastra-ai/mastra/pull/12575))
+
+- Updated dependencies [[`504fc8b`](https://github.com/mastra-ai/mastra/commit/504fc8b9d0ddab717577ad3bf9c95ea4bd5377bd), [`f9c150b`](https://github.com/mastra-ai/mastra/commit/f9c150b7595ad05ad9cc9a11098e2944361e8c22), [`88de7e8`](https://github.com/mastra-ai/mastra/commit/88de7e8dfe4b7e1951a9e441bb33136e705ce24e), [`edee4b3`](https://github.com/mastra-ai/mastra/commit/edee4b37dff0af515fc7cc0e8d71ee39e6a762f0), [`3790c75`](https://github.com/mastra-ai/mastra/commit/3790c7578cc6a47d854eb12d89e6b1912867fe29), [`e7a235b`](https://github.com/mastra-ai/mastra/commit/e7a235be6472e0c870ed6c791ddb17c492dc188b), [`d51d298`](https://github.com/mastra-ai/mastra/commit/d51d298953967aab1f58ec965b644d109214f085), [`6dbeeb9`](https://github.com/mastra-ai/mastra/commit/6dbeeb94a8b1eebb727300d1a98961f882180794), [`d5f0d8d`](https://github.com/mastra-ai/mastra/commit/d5f0d8d6a03e515ddaa9b5da19b7e44b8357b07b), [`09c3b18`](https://github.com/mastra-ai/mastra/commit/09c3b1802ff14e243a8a8baea327440bc8cc2e32), [`b896379`](https://github.com/mastra-ai/mastra/commit/b8963791c6afa79484645fcec596a201f936b9a2), [`85c84eb`](https://github.com/mastra-ai/mastra/commit/85c84ebb78aebfcba9d209c8e152b16d7a00cb71), [`a89272a`](https://github.com/mastra-ai/mastra/commit/a89272a5d71939b9fcd284e6a6dc1dd091a6bdcf), [`ee9c8df`](https://github.com/mastra-ai/mastra/commit/ee9c8df644f19d055af5f496bf4942705f5a47b7), [`77b4a25`](https://github.com/mastra-ai/mastra/commit/77b4a254e51907f8ff3a3ba95596a18e93ae4b35), [`276246e`](https://github.com/mastra-ai/mastra/commit/276246e0b9066a1ea48bbc70df84dbe528daaf99), [`08ecfdb`](https://github.com/mastra-ai/mastra/commit/08ecfdbdad6fb8285deef86a034bdf4a6047cfca), [`d5f628c`](https://github.com/mastra-ai/mastra/commit/d5f628ca86c6f6f3ff1035d52f635df32dd81cab), [`524c0f3`](https://github.com/mastra-ai/mastra/commit/524c0f3c434c3d9d18f66338dcef383d6161b59c), [`c18a0e9`](https://github.com/mastra-ai/mastra/commit/c18a0e9cef1e4ca004b2963d35e4cfc031971eac), [`4bd21ea`](https://github.com/mastra-ai/mastra/commit/4bd21ea43d44d0a0427414fc047577f9f0aa3bec), [`115a7a4`](https://github.com/mastra-ai/mastra/commit/115a7a47db5e9896fec12ae6507501adb9ec89bf), [`22a48ae`](https://github.com/mastra-ai/mastra/commit/22a48ae2513eb54d8d79dad361fddbca97a155e8), [`3c6ef79`](https://github.com/mastra-ai/mastra/commit/3c6ef798481e00d6d22563be2de98818fd4dd5e0), [`9311c17`](https://github.com/mastra-ai/mastra/commit/9311c17d7a0640d9c4da2e71b814dc67c57c6369), [`7edf78f`](https://github.com/mastra-ai/mastra/commit/7edf78f80422c43e84585f08ba11df0d4d0b73c5), [`1c4221c`](https://github.com/mastra-ai/mastra/commit/1c4221cf6032ec98d0e094d4ee11da3e48490d96), [`d25b9ea`](https://github.com/mastra-ai/mastra/commit/d25b9eabd400167255a97b690ffbc4ee4097ded5), [`fe1ce5c`](https://github.com/mastra-ai/mastra/commit/fe1ce5c9211c03d561606fda95cbfe7df1d9a9b5), [`b03c0e0`](https://github.com/mastra-ai/mastra/commit/b03c0e0389a799523929a458b0509c9e4244d562), [`0a8366b`](https://github.com/mastra-ai/mastra/commit/0a8366b0a692fcdde56c4d526e4cf03c502ae4ac), [`85664e9`](https://github.com/mastra-ai/mastra/commit/85664e9fd857320fbc245e301f764f45f66f32a3), [`bc79650`](https://github.com/mastra-ai/mastra/commit/bc796500c6e0334faa158a96077e3fb332274869), [`9257d01`](https://github.com/mastra-ai/mastra/commit/9257d01d1366d81f84c582fe02b5e200cf9621f4), [`3a3a59e`](https://github.com/mastra-ai/mastra/commit/3a3a59e8ffaa6a985fe3d9a126a3f5ade11a6724), [`3108d4e`](https://github.com/mastra-ai/mastra/commit/3108d4e649c9fddbf03253a6feeb388a5fa9fa5a), [`0c33b2c`](https://github.com/mastra-ai/mastra/commit/0c33b2c9db537f815e1c59e2c898ffce2e395a79), [`191e5bd`](https://github.com/mastra-ai/mastra/commit/191e5bd29b82f5bda35243945790da7bc7b695c2), [`f77cd94`](https://github.com/mastra-ai/mastra/commit/f77cd94c44eabed490384e7d19232a865e13214c), [`e8135c7`](https://github.com/mastra-ai/mastra/commit/e8135c7e300dac5040670eec7eab896ac6092e30), [`daca48f`](https://github.com/mastra-ai/mastra/commit/daca48f0fb17b7ae0b62a2ac40cf0e491b2fd0b7), [`257d14f`](https://github.com/mastra-ai/mastra/commit/257d14faca5931f2e4186fc165b6f0b1f915deee), [`352f25d`](https://github.com/mastra-ai/mastra/commit/352f25da316b24cdd5b410fd8dddf6a8b763da2a), [`93477d0`](https://github.com/mastra-ai/mastra/commit/93477d0769b8a13ea5ed73d508d967fb23eaeed9), [`31c78b3`](https://github.com/mastra-ai/mastra/commit/31c78b3eb28f58a8017f1dcc795c33214d87feac), [`0bc0720`](https://github.com/mastra-ai/mastra/commit/0bc07201095791858087cc56f353fcd65e87ab54), [`36516ac`](https://github.com/mastra-ai/mastra/commit/36516aca1021cbeb42e74751b46a2614101f37c8), [`e947652`](https://github.com/mastra-ai/mastra/commit/e9476527fdecb4449e54570e80dfaf8466901254), [`3c6ef79`](https://github.com/mastra-ai/mastra/commit/3c6ef798481e00d6d22563be2de98818fd4dd5e0), [`9257d01`](https://github.com/mastra-ai/mastra/commit/9257d01d1366d81f84c582fe02b5e200cf9621f4), [`ec248f6`](https://github.com/mastra-ai/mastra/commit/ec248f6b56e8a037c066c49b2178e2507471d988)]:
+  - @mastra/core@1.9.0-alpha.0
+
 ## 1.0.1
 
 ### Patch Changes

package/LICENSE.md

@@ -1,3 +1,18 @@
+Portions of this software are licensed as follows:
+
+- All content that resides under any directory named "ee/" within this
+  repository, including but not limited to:
+  - `packages/core/src/auth/ee/`
+  - `packages/server/src/server/auth/ee/`
+    is licensed under the license defined in `ee/LICENSE`.
+
+- All third-party components incorporated into the Mastra Software are
+  licensed under the original license provided by the owner of the
+  applicable component.
+
+- Content outside of the above-mentioned directories or restrictions is
+  available under the "Apache License 2.0" as defined below.
+
 # Apache License 2.0
 
 Copyright (c) 2025 Kepler Software, Inc.

package/dist/docs/SKILL.md

@@ -3,7 +3,7 @@ name: mastra-dynamodb
 description: Documentation for @mastra/dynamodb. Use when working with @mastra/dynamodb APIs, configuration, or implementation.
 metadata:
   package: "@mastra/dynamodb"
-  version: "1.0.1"
+  version: "1.0.2-alpha.0"
 ---
 
 ## When to use

package/dist/docs/assets/SOURCE_MAP.json

@@ -1,5 +1,5 @@
 {
-  "version": "1.0.1",
+  "version": "1.0.2-alpha.0",
   "package": "@mastra/dynamodb",
   "exports": {},
   "modules": {}

package/dist/docs/references/reference-storage-dynamodb.md

@@ -53,19 +53,19 @@ Detailed instructions for setting up the table using AWS CloudFormation or AWS C
 ### Basic Usage
 
 ```typescript
-import { Memory } from "@mastra/memory";
-import { DynamoDBStore } from "@mastra/dynamodb";
+import { Memory } from '@mastra/memory'
+import { DynamoDBStore } from '@mastra/dynamodb'
 
 // Initialize the DynamoDB storage
 const storage = new DynamoDBStore({
-  id: "dynamodb", // Unique identifier for this storage instance
+  id: 'dynamodb', // Unique identifier for this storage instance
   config: {
-    tableName: "mastra-single-table", // Name of your DynamoDB table
-    region: "us-east-1", // Optional: AWS region, defaults to 'us-east-1'
+    tableName: 'mastra-single-table', // Name of your DynamoDB table
+    region: 'us-east-1', // Optional: AWS region, defaults to 'us-east-1'
     // endpoint: "http://localhost:8000", // Optional: For local DynamoDB
     // credentials: { accessKeyId: "YOUR_ACCESS_KEY", secretAccessKey: "YOUR_SECRET_KEY" } // Optional
   },
-});
+})
 
 // Example: Initialize Memory with DynamoDB storage
 const memory = new Memory({
@@ -73,7 +73,7 @@ const memory = new Memory({
   options: {
     lastMessages: 10,
   },
-});
+})
 ```
 
 ### Local Development with DynamoDB Local
@@ -89,19 +89,19 @@ For local development, you can use [DynamoDB Local](https://docs.aws.amazon.com/
 2. **Configure `DynamoDBStore` to use the local endpoint:**
 
    ```typescript
-   import { DynamoDBStore } from "@mastra/dynamodb";
+   import { DynamoDBStore } from '@mastra/dynamodb'
 
    const storage = new DynamoDBStore({
-     id: "dynamodb-local",
+     id: 'dynamodb-local',
      config: {
-       tableName: "mastra-single-table", // Ensure this table is created in your local DynamoDB
-       region: "localhost", // Can be any string for local, 'localhost' is common
-       endpoint: "http://localhost:8000",
+       tableName: 'mastra-single-table', // Ensure this table is created in your local DynamoDB
+       region: 'localhost', // Can be any string for local, 'localhost' is common
+       endpoint: 'http://localhost:8000',
        // For DynamoDB Local, credentials are not typically required unless configured.
        // If you've configured local credentials:
        // credentials: { accessKeyId: "fakeMyKeyId", secretAccessKey: "fakeSecretAccessKey" }
      },
-   });
+   })
    ```
 
    You will still need to create the table and GSIs in your local DynamoDB instance, for example, using the AWS CLI pointed to your local endpoint.
@@ -137,13 +137,13 @@ To use TTL, you must:
 2. **Enable TTL on your DynamoDB table** via AWS Console or CLI, specifying the attribute name (default: `ttl`)
 
 ```typescript
-import { DynamoDBStore } from "@mastra/dynamodb";
+import { DynamoDBStore } from '@mastra/dynamodb'
 
 const storage = new DynamoDBStore({
-  name: "dynamodb",
+  name: 'dynamodb',
   config: {
-    tableName: "mastra-single-table",
-    region: "us-east-1",
+    tableName: 'mastra-single-table',
+    region: 'us-east-1',
     ttl: {
       // Messages expire after 30 days
       message: {
@@ -158,7 +158,7 @@ const storage = new DynamoDBStore({
       // Traces expire after 7 days with custom attribute name
       trace: {
         enabled: true,
-        attributeName: "expiresAt", // Custom TTL attribute
+        attributeName: 'expiresAt', // Custom TTL attribute
         defaultTtlSeconds: 7 * 24 * 60 * 60, // 7 days
       },
       // Workflow snapshots don't expire
@@ -167,7 +167,7 @@ const storage = new DynamoDBStore({
       },
     },
   },
-});
+})
 ```
 
 ### Supported Entity Types
@@ -214,7 +214,7 @@ aws dynamodb update-time-to-live \
 4. Under "Time to Live (TTL)", click "Manage TTL"
 5. Enable TTL and specify the attribute name (default: `ttl`)
 
-> **Note**: DynamoDB deletes expired items within 48 hours after expiration. Items remain queryable until actually deleted.
+> **Note:** DynamoDB deletes expired items within 48 hours after expiration. Items remain queryable until actually deleted.
 
 ## AWS IAM Permissions
 

package/dist/index.cjs

@@ -2144,6 +2144,8 @@ function formatWorkflowRun(snapshotData) {
     resourceId: snapshotData.resourceId
   };
 }
+var MAX_RETRIES = 5;
+var BASE_DELAY_MS = 50;
 var WorkflowStorageDynamoDB = class extends storage.WorkflowsStorage {
   service;
   ttlConfig;
@@ -2153,9 +2155,41 @@ var WorkflowStorageDynamoDB = class extends storage.WorkflowsStorage {
     this.service = resolved.service;
     this.ttlConfig = resolved.ttl;
   }
+  supportsConcurrentUpdates() {
+    return true;
+  }
   async dangerouslyClearAll() {
     await deleteTableData(this.service, storage.TABLE_WORKFLOW_SNAPSHOT);
   }
+  /**
+   * Helper to check if an error is a DynamoDB conditional check failure
+   */
+  isConditionalCheckFailed(error) {
+    if (error && typeof error === "object") {
+      const err = error;
+      if (err.name === "ConditionalCheckFailedException" || err.code === "ConditionalCheckFailedException" || (err.__type?.includes("ConditionalCheckFailedException") ?? false)) {
+        return true;
+      }
+      if (err.message?.includes("conditional request failed")) {
+        return true;
+      }
+      if (err.cause && typeof err.cause === "object") {
+        const cause = err.cause;
+        if (cause.name === "ConditionalCheckFailedException" || cause.code === "ConditionalCheckFailedException") {
+          return true;
+        }
+      }
+    }
+    return false;
+  }
+  /**
+   * Helper to delay with exponential backoff and jitter
+   */
+  async delay(attempt) {
+    const backoff = BASE_DELAY_MS * Math.pow(2, attempt);
+    const jitter = Math.random() * backoff * 0.5;
+    await new Promise((resolve) => setTimeout(resolve, backoff + jitter));
+  }
   async updateWorkflowResults({
     workflowName,
     runId,
@@ -2163,69 +2197,147 @@ var WorkflowStorageDynamoDB = class extends storage.WorkflowsStorage {
     result,
     requestContext
   }) {
-    try {
-      const existingSnapshot = await this.loadWorkflowSnapshot({ workflowName, runId });
-      let snapshot;
-      if (!existingSnapshot) {
-        snapshot = {
-          context: {},
-          activePaths: [],
-          timestamp: Date.now(),
-          suspendedPaths: {},
-          activeStepsPath: {},
-          resumeLabels: {},
-          serializedStepGraph: [],
-          status: "pending",
-          value: {},
-          waitingPaths: {},
-          runId,
-          requestContext: {}
+    for (let attempt = 0; attempt < MAX_RETRIES; attempt++) {
+      try {
+        const existingRecord = await this.service.entities.workflow_snapshot.get({
+          entity: "workflow_snapshot",
+          workflow_name: workflowName,
+          run_id: runId
+        }).go();
+        const now = /* @__PURE__ */ new Date();
+        let snapshot;
+        let previousUpdatedAt;
+        if (!existingRecord.data) {
+          snapshot = {
+            context: {},
+            activePaths: [],
+            timestamp: Date.now(),
+            suspendedPaths: {},
+            activeStepsPath: {},
+            resumeLabels: {},
+            serializedStepGraph: [],
+            status: "pending",
+            value: {},
+            waitingPaths: {},
+            runId,
+            requestContext: {}
+          };
+        } else {
+          snapshot = existingRecord.data.snapshot;
+          previousUpdatedAt = existingRecord.data.updatedAt;
+        }
+        snapshot.context[stepId] = result;
+        snapshot.requestContext = { ...snapshot.requestContext, ...requestContext };
+        const data = {
+          entity: "workflow_snapshot",
+          workflow_name: workflowName,
+          run_id: runId,
+          snapshot: JSON.stringify(snapshot),
+          createdAt: existingRecord.data?.createdAt ?? now.toISOString(),
+          updatedAt: now.toISOString(),
+          ...getTtlProps("workflow_snapshot", this.ttlConfig)
         };
-      } else {
-        snapshot = existingSnapshot;
+        if (previousUpdatedAt) {
+          await this.service.entities.workflow_snapshot.upsert(data).where((attr, op) => op.eq(attr.updatedAt, previousUpdatedAt)).go();
+        } else {
+          await this.service.entities.workflow_snapshot.create(data).where((attr, op) => op.notExists(attr.run_id)).go();
+        }
+        return snapshot.context;
+      } catch (error$1) {
+        if (this.isConditionalCheckFailed(error$1)) {
+          if (attempt < MAX_RETRIES - 1) {
+            this.logger.debug(
+              `Optimistic locking conflict in updateWorkflowResults, retrying (attempt ${attempt + 1})`
+            );
+            await this.delay(attempt);
+            continue;
+          }
+        }
+        if (error$1 instanceof error.MastraError) throw error$1;
+        throw new error.MastraError(
+          {
+            id: storage.createStorageErrorId("DYNAMODB", "UPDATE_WORKFLOW_RESULTS", "FAILED"),
+            domain: error.ErrorDomain.STORAGE,
+            category: error.ErrorCategory.THIRD_PARTY,
+            details: { workflowName, runId, stepId }
+          },
+          error$1
+        );
       }
-      snapshot.context[stepId] = result;
-      snapshot.requestContext = { ...snapshot.requestContext, ...requestContext };
-      await this.persistWorkflowSnapshot({ workflowName, runId, snapshot });
-      return snapshot.context;
-    } catch (error$1) {
-      if (error$1 instanceof error.MastraError) throw error$1;
-      throw new error.MastraError(
-        {
-          id: storage.createStorageErrorId("DYNAMODB", "UPDATE_WORKFLOW_RESULTS", "FAILED"),
-          domain: error.ErrorDomain.STORAGE,
-          category: error.ErrorCategory.THIRD_PARTY,
-          details: { workflowName, runId, stepId }
-        },
-        error$1
-      );
     }
+    throw new error.MastraError(
+      {
+        id: storage.createStorageErrorId("DYNAMODB", "UPDATE_WORKFLOW_RESULTS", "MAX_RETRIES_EXCEEDED"),
+        domain: error.ErrorDomain.STORAGE,
+        category: error.ErrorCategory.THIRD_PARTY,
+        details: { workflowName, runId, stepId }
+      },
+      new Error("Max retries exceeded for optimistic locking")
+    );
   }
   async updateWorkflowState({
     workflowName,
     runId,
     opts
   }) {
-    try {
-      const existingSnapshot = await this.loadWorkflowSnapshot({ workflowName, runId });
-      if (!existingSnapshot || !existingSnapshot.context) {
-        return void 0;
+    for (let attempt = 0; attempt < MAX_RETRIES; attempt++) {
+      try {
+        const existingRecord = await this.service.entities.workflow_snapshot.get({
+          entity: "workflow_snapshot",
+          workflow_name: workflowName,
+          run_id: runId
+        }).go();
+        if (!existingRecord.data) {
+          return void 0;
+        }
+        const existingSnapshot = existingRecord.data.snapshot;
+        if (!existingSnapshot || !existingSnapshot.context) {
+          return void 0;
+        }
+        const previousUpdatedAt = existingRecord.data.updatedAt;
+        const updatedSnapshot = { ...existingSnapshot, ...opts };
+        const now = /* @__PURE__ */ new Date();
+        const data = {
+          entity: "workflow_snapshot",
+          workflow_name: workflowName,
+          run_id: runId,
+          snapshot: JSON.stringify(updatedSnapshot),
+          createdAt: existingRecord.data.createdAt,
+          updatedAt: now.toISOString(),
+          resourceId: existingRecord.data.resourceId,
+          ...getTtlProps("workflow_snapshot", this.ttlConfig)
+        };
+        await this.service.entities.workflow_snapshot.upsert(data).where((attr, op) => op.eq(attr.updatedAt, previousUpdatedAt)).go();
+        return updatedSnapshot;
+      } catch (error$1) {
+        if (this.isConditionalCheckFailed(error$1)) {
+          if (attempt < MAX_RETRIES - 1) {
+            this.logger.debug(`Optimistic locking conflict in updateWorkflowState, retrying (attempt ${attempt + 1})`);
+            await this.delay(attempt);
+            continue;
+          }
+        }
+        if (error$1 instanceof error.MastraError) throw error$1;
+        throw new error.MastraError(
+          {
+            id: storage.createStorageErrorId("DYNAMODB", "UPDATE_WORKFLOW_STATE", "FAILED"),
+            domain: error.ErrorDomain.STORAGE,
+            category: error.ErrorCategory.THIRD_PARTY,
+            details: { workflowName, runId }
+          },
+          error$1
+        );
       }
-      const updatedSnapshot = { ...existingSnapshot, ...opts };
-      await this.persistWorkflowSnapshot({ workflowName, runId, snapshot: updatedSnapshot });
-      return updatedSnapshot;
-    } catch (error$1) {
-      if (error$1 instanceof error.MastraError) throw error$1;
-      throw new error.MastraError(
-        {
-          id: storage.createStorageErrorId("DYNAMODB", "UPDATE_WORKFLOW_STATE", "FAILED"),
-          domain: error.ErrorDomain.STORAGE,
-          category: error.ErrorCategory.THIRD_PARTY,
-          details: { workflowName, runId }
-        },
-        error$1
-      );
     }
+    throw new error.MastraError(
+      {
+        id: storage.createStorageErrorId("DYNAMODB", "UPDATE_WORKFLOW_STATE", "MAX_RETRIES_EXCEEDED"),
+        domain: error.ErrorDomain.STORAGE,
+        category: error.ErrorCategory.THIRD_PARTY,
+        details: { workflowName, runId }
+      },
+      new Error("Max retries exceeded for optimistic locking")
+    );
   }
   // Workflow operations
   async persistWorkflowSnapshot({