@mastra/dynamodb 1.4.4 → 1.4.5-alpha.1

package/dist/index.js

@@ -5,7 +5,7 @@ import { BackgroundTasksStorage, MastraCompositeStore, MemoryStorage, RunFenceCo
 import { Entity, Service } from "electrodb";
 import { MessageList } from "@mastra/core/agent";
 import { saveScorePayloadSchema } from "@mastra/core/evals";
-//#region src/entities/utils.ts
+// src/entities/utils.ts
 const baseAttributes = {
 	createdAt: {
 		type: "string",
@@ -61,8 +61,7 @@ const baseAttributes = {
 		required: false
 	}
 };
-//#endregion
-//#region src/entities/background-task.ts
+// src/entities/background-task.ts
 /**
 * JSON field helpers — DynamoDB stores JSON as stringified strings,
 * and parses them on get.
@@ -225,8 +224,7 @@ const backgroundTaskEntity = new Entity({
 		}
 	}
 });
-//#endregion
-//#region src/entities/eval.ts
+// src/entities/eval.ts
 const evalEntity = new Entity({
 	model: {
 		entity: "eval",
@@ -324,8 +322,7 @@ const evalEntity = new Entity({
 		}
 	}
 });
-//#endregion
-//#region src/entities/message.ts
+// src/entities/message.ts
 const messageEntity = new Entity({
 	model: {
 		entity: "message",
@@ -447,8 +444,7 @@ const messageEntity = new Entity({
 		}
 	}
 });
-//#endregion
-//#region src/entities/resource.ts
+// src/entities/resource.ts
 const resourceEntity = new Entity({
 	model: {
 		entity: "resource",
@@ -497,8 +493,7 @@ const resourceEntity = new Entity({
 		}
 	} }
 });
-//#endregion
-//#region src/entities/run-fencing.ts
+// src/entities/run-fencing.ts
 /**
 * The current claim on a durable workflow run. `leaseExpiresAt` holds the
 * lease's expiry in epoch milliseconds and is removed when the owner releases
@@ -571,8 +566,7 @@ const memoryRunFenceEntity = new Entity({
 		}
 	} }
 });
-//#endregion
-//#region src/entities/score.ts
+// src/entities/score.ts
 const scoreEntity = new Entity({
 	model: {
 		entity: "score",
@@ -932,8 +926,7 @@ const scoreEntity = new Entity({
 		}
 	}
 });
-//#endregion
-//#region src/entities/thread.ts
+// src/entities/thread.ts
 const threadEntity = new Entity({
 	model: {
 		entity: "thread",
@@ -999,8 +992,7 @@ const threadEntity = new Entity({
 		}
 	}
 });
-//#endregion
-//#region src/entities/trace.ts
+// src/entities/trace.ts
 const traceEntity = new Entity({
 	model: {
 		entity: "trace",
@@ -1129,8 +1121,7 @@ const traceEntity = new Entity({
 		}
 	}
 });
-//#endregion
-//#region src/entities/workflow-snapshot.ts
+// src/entities/workflow-snapshot.ts
 const workflowSnapshotEntity = new Entity({
 	model: {
 		entity: "workflow_snapshot",
@@ -1191,8 +1182,7 @@ const workflowSnapshotEntity = new Entity({
 		}
 	}
 });
-//#endregion
-//#region src/entities/index.ts
+// src/entities/index.ts
 function getElectroDbService(client, tableName) {
 	return new Service({
 		thread: threadEntity,
@@ -1210,8 +1200,7 @@ function getElectroDbService(client, tableName) {
 		table: tableName
 	});
 }
-//#endregion
-//#region src/storage/db/index.ts
+// src/storage/db/index.ts
 /**
 * Resolves DynamoDBDomainConfig to an ElectroDB service and TTL config.
 * Handles creating a new service if config is provided.
@@ -1231,8 +1220,7 @@ function resolveDynamoDBConfig(config) {
 		ttl: config.ttl
 	};
 }
-//#endregion
-//#region src/storage/domains/utils.ts
+// src/storage/domains/utils.ts
 /**
 * Maps table names to ElectroDB entity names
 */
@@ -1286,8 +1274,7 @@ async function deleteTableData(service, tableName) {
 		await entity.delete(keysToDelete).go();
 	}
 }
-//#endregion
-//#region src/storage/domains/background-tasks/index.ts
+// src/storage/domains/background-tasks/index.ts
 const ENTITY = "background_task";
 function serializeJson(v) {
 	if (typeof v === "object" && v != null) return JSON.stringify(v);
@@ -1562,8 +1549,7 @@ var BackgroundTasksStorageDynamoDB = class extends BackgroundTasksStorage {
 		return total;
 	}
 };
-//#endregion
-//#region src/storage/ttl.ts
+// src/storage/ttl.ts
 /**
 * Calculate TTL value for a given entity type.
 *
@@ -1617,8 +1603,7 @@ function getTtlProps(entityName, ttlConfig, customTtlSeconds) {
 	if (ttlValue === void 0) return {};
 	return { [getTtlAttributeName(entityName, ttlConfig)]: ttlValue };
 }
-//#endregion
-//#region src/storage/domains/run-fencing.ts
+// src/storage/domains/run-fencing.ts
 const MAX_CONFLICT_RETRIES = 8;
 const BASE_DELAY_MS$1 = 20;
 async function backoff(attempt) {
@@ -1734,8 +1719,7 @@ async function pruneRunClaims(service, entityName, attribute, cutoff, limit) {
 	} while (cursor);
 	return deleted;
 }
-//#endregion
-//#region src/storage/domains/memory/index.ts
+// src/storage/domains/memory/index.ts
 var MemoryStorageDynamoDB = class MemoryStorageDynamoDB extends MemoryStorage {
 	/**
 	* Tables `prune()` can clean up. Run fences anchor on `retiredAt` (epoch ms), which stays unset until the
@@ -2582,8 +2566,7 @@ var MemoryStorageDynamoDB = class MemoryStorageDynamoDB extends MemoryStorage {
 		}
 	}
 };
-//#endregion
-//#region src/storage/domains/scores/index.ts
+// src/storage/domains/scores/index.ts
 /** Returns true when a score matches the multi-tenant scope filters (or none provided). */
 function matchesTenancy(score, filters) {
 	if (filters?.organizationId !== void 0 && score.organizationId !== filters.organizationId) return false;
@@ -2882,8 +2865,7 @@ var ScoresStorageDynamoDB = class extends ScoresStorage {
 		}
 	}
 };
-//#endregion
-//#region src/storage/domains/workflows/index.ts
+// src/storage/domains/workflows/index.ts
 function formatWorkflowRun(snapshotData) {
 	return {
 		workflowName: snapshotData.workflow_name,
@@ -3093,7 +3075,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;
@@ -3171,7 +3153,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)) {
@@ -3283,7 +3265,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) {
@@ -3380,7 +3362,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 {
@@ -3445,8 +3427,7 @@ var WorkflowStorageDynamoDB = class WorkflowStorageDynamoDB extends WorkflowsSto
 		}
 	}
 };
-//#endregion
-//#region src/storage/index.ts
+// src/storage/index.ts
 /**
 * Type guard for pre-configured client config
 */
@@ -3591,7 +3572,6 @@ var DynamoDBStore = class extends MastraCompositeStore {
 		}
 	}
 };
-//#endregion
 export { BackgroundTasksStorageDynamoDB, DynamoDBStore, MemoryStorageDynamoDB, ScoresStorageDynamoDB, WorkflowStorageDynamoDB, calculateTtl, getTtlAttributeName, getTtlProps, isTtlEnabled };
 
 //# sourceMappingURL=index.js.map