@mastra/dynamodb 1.1.0 → 1.1.1

package/dist/index.js

@@ -2780,17 +2780,35 @@ var WorkflowStorageDynamoDB = class extends WorkflowsStorage {
     this.logger.debug("Persisting workflow snapshot", { workflowName, runId });
     try {
       const now = /* @__PURE__ */ new Date();
+      const createdAtValue = (createdAt ?? now).toISOString();
+      const updatedAtValue = (updatedAt ?? now).toISOString();
       const data = {
         entity: "workflow_snapshot",
         workflow_name: workflowName,
         run_id: runId,
         snapshot: JSON.stringify(snapshot),
-        createdAt: (createdAt ?? now).toISOString(),
-        updatedAt: (updatedAt ?? now).toISOString(),
+        createdAt: createdAtValue,
+        updatedAt: updatedAtValue,
         resourceId,
         ...getTtlProps("workflow_snapshot", this.ttlConfig)
       };
-      await this.service.entities.workflow_snapshot.upsert(data).go();
+      try {
+        await this.service.entities.workflow_snapshot.create(data).where((attr, op) => op.notExists(attr.run_id)).go();
+      } catch (error) {
+        if (!this.isConditionalCheckFailed(error)) {
+          throw error;
+        }
+        await this.service.entities.workflow_snapshot.update({
+          entity: "workflow_snapshot",
+          workflow_name: workflowName,
+          run_id: runId
+        }).set({
+          snapshot: JSON.stringify(snapshot),
+          updatedAt: updatedAtValue,
+          resourceId,
+          ...getTtlProps("workflow_snapshot", this.ttlConfig)
+        }).go();
+      }
     } catch (error) {
       throw new MastraError(
         {