@mastra/dynamodb 1.4.4-alpha.0 → 1.4.5-alpha.0

package/dist/index.js

@@ -1382,22 +1382,38 @@ var BackgroundTasksStorageDynamoDB = class extends BackgroundTasksStorage {
 			const removeFields = [];
 			if ("status" in update && update.status !== void 0) setFields.status = update.status;
 			if ("retryCount" in update && update.retryCount !== void 0) setFields.retryCount = update.retryCount;
-			if ("result" in update) if (update.result === void 0 || update.result === null) removeFields.push("result");
-			else setFields.result = serializeJson(update.result);
-			if ("error" in update) if (update.error === void 0 || update.error === null) removeFields.push("error");
-			else setFields.error = serializeJson(update.error);
-			if ("suspendPayload" in update) if (update.suspendPayload === void 0 || update.suspendPayload === null) removeFields.push("suspendPayload");
-			else setFields.suspendPayload = serializeJson(update.suspendPayload);
-			if ("startedAt" in update) if (update.startedAt === void 0 || update.startedAt === null) removeFields.push("startedAtIso");
-			else setFields.startedAtIso = update.startedAt.toISOString();
-			if ("suspendedAt" in update) if (update.suspendedAt === void 0 || update.suspendedAt === null) removeFields.push("suspendedAtIso");
-			else setFields.suspendedAtIso = update.suspendedAt.toISOString();
-			if ("completedAt" in update) if (update.completedAt === void 0 || update.completedAt === null) removeFields.push("completedAtIso");
-			else setFields.completedAtIso = update.completedAt.toISOString();
-			if ("ownerId" in update) if (update.ownerId === void 0 || update.ownerId === null) removeFields.push("ownerId");
-			else setFields.ownerId = update.ownerId;
-			if ("leaseExpiresAt" in update) if (update.leaseExpiresAt === void 0 || update.leaseExpiresAt === null) removeFields.push("leaseExpiresAtIso");
-			else setFields.leaseExpiresAtIso = update.leaseExpiresAt.toISOString();
+			if ("result" in update) {
+				if (update.result === void 0 || update.result === null) removeFields.push("result");
+				else setFields.result = serializeJson(update.result);
+			}
+			if ("error" in update) {
+				if (update.error === void 0 || update.error === null) removeFields.push("error");
+				else setFields.error = serializeJson(update.error);
+			}
+			if ("suspendPayload" in update) {
+				if (update.suspendPayload === void 0 || update.suspendPayload === null) removeFields.push("suspendPayload");
+				else setFields.suspendPayload = serializeJson(update.suspendPayload);
+			}
+			if ("startedAt" in update) {
+				if (update.startedAt === void 0 || update.startedAt === null) removeFields.push("startedAtIso");
+				else setFields.startedAtIso = update.startedAt.toISOString();
+			}
+			if ("suspendedAt" in update) {
+				if (update.suspendedAt === void 0 || update.suspendedAt === null) removeFields.push("suspendedAtIso");
+				else setFields.suspendedAtIso = update.suspendedAt.toISOString();
+			}
+			if ("completedAt" in update) {
+				if (update.completedAt === void 0 || update.completedAt === null) removeFields.push("completedAtIso");
+				else setFields.completedAtIso = update.completedAt.toISOString();
+			}
+			if ("ownerId" in update) {
+				if (update.ownerId === void 0 || update.ownerId === null) removeFields.push("ownerId");
+				else setFields.ownerId = update.ownerId;
+			}
+			if ("leaseExpiresAt" in update) {
+				if (update.leaseExpiresAt === void 0 || update.leaseExpiresAt === null) removeFields.push("leaseExpiresAtIso");
+				else setFields.leaseExpiresAtIso = update.leaseExpiresAt.toISOString();
+			}
 			if (Object.keys(setFields).length === 0 && removeFields.length === 0) return false;
 			let op = this.service.entities.background_task.patch({
 				entity: ENTITY,
@@ -1745,13 +1761,14 @@ var MemoryStorageDynamoDB = class MemoryStorageDynamoDB extends MemoryStorage {
 		return true;
 	}
 	async prune(policies, options) {
+		const targets = resolveRetentionTargets({
+			policies,
+			descriptor: MemoryStorageDynamoDB.retentionTables,
+			order: ["runFences"]
+		});
 		return executeRetentionPrune({
 			domain: "memory",
-			targets: resolveRetentionTargets({
-				policies,
-				descriptor: MemoryStorageDynamoDB.retentionTables,
-				order: ["runFences"]
-			}),
+			targets,
 			options,
 			cutoffFor: (target, now) => retentionCutoffMs(target.policy, now),
 			deleteBatch: (_target, cutoff, limit) => pruneRunClaims(this.service, "memory_run_fence", "retiredAt", Number(cutoff), limit)
@@ -2915,13 +2932,14 @@ var WorkflowStorageDynamoDB = class WorkflowStorageDynamoDB extends WorkflowsSto
 		return true;
 	}
 	async prune(policies, options) {
+		const targets = resolveRetentionTargets({
+			policies,
+			descriptor: WorkflowStorageDynamoDB.retentionTables,
+			order: ["runOwnership"]
+		});
 		return executeRetentionPrune({
 			domain: "workflows",
-			targets: resolveRetentionTargets({
-				policies,
-				descriptor: WorkflowStorageDynamoDB.retentionTables,
-				order: ["runOwnership"]
-			}),
+			targets,
 			options,
 			cutoffFor: (target, now) => retentionCutoffMs(target.policy, now),
 			deleteBatch: (_target, cutoff, limit) => pruneRunClaims(this.service, "workflow_run_owner", "updatedAt", Number(cutoff), limit)
@@ -3075,7 +3093,7 @@ var WorkflowStorageDynamoDB = class WorkflowStorageDynamoDB extends WorkflowsSto
 				entity: "workflow_snapshot",
 				workflow_name: workflowName,
 				run_id: runId
-			}).go();
+			}).go({ consistent: true });
 			const now = /* @__PURE__ */ new Date();
 			let snapshot;
 			let previousUpdatedAt;
@@ -3117,7 +3135,7 @@ var WorkflowStorageDynamoDB = class WorkflowStorageDynamoDB extends WorkflowsSto
 			return snapshot.context;
 		} catch (error) {
 			if (this.isConditionalCheckFailed(error)) {
-				if (attempt < MAX_RETRIES - 1) {
+				if (attempt < 4) {
 					this.logger.debug(`Optimistic locking conflict in updateWorkflowResults, retrying (attempt ${attempt + 1})`);
 					await this.delay(attempt);
 					continue;
@@ -3153,7 +3171,7 @@ var WorkflowStorageDynamoDB = class WorkflowStorageDynamoDB extends WorkflowsSto
 				entity: "workflow_snapshot",
 				workflow_name: workflowName,
 				run_id: runId
-			}).go();
+			}).go({ consistent: true });
 			const existingSnapshot = existingRecord.data?.snapshot;
 			const { expectedStatus, ...state } = opts;
 			if (!existingSnapshot?.context || !matchesExpectedWorkflowStatus(existingSnapshot.status, expectedStatus)) {
@@ -3180,7 +3198,7 @@ var WorkflowStorageDynamoDB = class WorkflowStorageDynamoDB extends WorkflowsSto
 			return updatedSnapshot;
 		} catch (error) {
 			if (this.isConditionalCheckFailed(error)) {
-				if (attempt < MAX_RETRIES - 1) {
+				if (attempt < 4) {
 					this.logger.debug(`Optimistic locking conflict in updateWorkflowState, retrying (attempt ${attempt + 1})`);
 					await this.delay(attempt);
 					continue;
@@ -3265,7 +3283,7 @@ var WorkflowStorageDynamoDB = class WorkflowStorageDynamoDB extends WorkflowsSto
 				entity: "workflow_snapshot",
 				workflow_name: workflowName,
 				run_id: runId
-			}).go();
+			}).go({ consistent: true });
 			if (!result.data?.snapshot) return null;
 			return result.data.snapshot;
 		} catch (error) {
@@ -3362,7 +3380,7 @@ var WorkflowStorageDynamoDB = class WorkflowStorageDynamoDB extends WorkflowsSto
 					entity: "workflow_snapshot",
 					workflow_name: workflowName,
 					run_id: runId
-				}).go();
+				}).go({ consistent: true });
 				if (!result.data) return null;
 				const snapshot = result.data.snapshot;
 				return {