@mastra/mongodb 1.23.1-alpha.0 → 1.23.1-alpha.1

package/dist/index.js

@@ -1,6 +1,6 @@
 import { v4 } from "@lukeed/uuid";
 import { ErrorCategory, ErrorDomain, MastraError } from "@mastra/core/error";
-import { AgentsStorage, BackgroundTasksStorage, BlobStore, DatasetsStorage, ExperimentsStorage, KnowledgeConflictError, KnowledgeNotFoundError, KnowledgeStorage, MCPClientsStorage, MCPServersStorage, MastraCompositeStore, MemoryStorage, NotificationsStorage, ObservabilityStorage, PromptBlocksStorage, SchedulesStorage, ScorerDefinitionsStorage, ScoresStorage, SkillsStorage, TABLE_AGENTS, TABLE_AGENT_VERSIONS, TABLE_BACKGROUND_TASKS, TABLE_DATASETS, TABLE_DATASET_ITEMS, TABLE_DATASET_VERSIONS, TABLE_EXPERIMENTS, TABLE_EXPERIMENT_RESULTS, TABLE_KNOWLEDGE_ACTIVITY, TABLE_KNOWLEDGE_MENTIONS, TABLE_KNOWLEDGE_NODES, TABLE_KNOWLEDGE_RECORDS, TABLE_KNOWLEDGE_SEMANTIC_OUTBOX, TABLE_MCP_CLIENTS, TABLE_MCP_CLIENT_VERSIONS, TABLE_MCP_SERVERS, TABLE_MCP_SERVER_VERSIONS, TABLE_MESSAGES, TABLE_NOTIFICATIONS, TABLE_PROMPT_BLOCKS, TABLE_PROMPT_BLOCK_VERSIONS, TABLE_RESOURCES, TABLE_SCHEDULES, TABLE_SCHEDULE_TRIGGERS, TABLE_SCORERS, TABLE_SCORER_DEFINITIONS, TABLE_SCORER_DEFINITION_VERSIONS, TABLE_SKILLS, TABLE_SKILL_BLOBS, TABLE_SKILL_VERSIONS, TABLE_SPANS, TABLE_THREADS, TABLE_WORKFLOW_DEFINITIONS, TABLE_WORKFLOW_SNAPSHOT, TABLE_WORKSPACES, TABLE_WORKSPACE_VERSIONS, TraceStatus, WorkflowDefinitionsStorage, WorkflowsStorage, WorkspacesStorage, calculatePagination, canonicalizeKnowledgeScope, createKnowledgeUlid, createStorageErrorId, createVectorErrorId, ensureDate, hasErrorCode, isKnowledgeScopeVisible, knowledgeScopeKey, knowledgeSemanticDocumentId, knowledgeSemanticIdempotencyKey, listTracesArgsSchema, normalizePerPage, normalizeScheduleTarget, parseDuration, parseKnowledgeNodeCursor, parseKnowledgeWikilinks, resolveListOrderBy, safelyParseJSON, storageMessageMatchesMetadataFilter, toTraceSpans, transformScoreRow, validateStorageMetadataFilter } from "@mastra/core/storage";
+import { AgentsStorage, BackgroundTasksStorage, BlobStore, DatasetsStorage, ExperimentsStorage, KnowledgeConflictError, KnowledgeNotFoundError, KnowledgeStorage, MCPClientsStorage, MCPServersStorage, MastraCompositeStore, MemoryStorage, NotificationsStorage, ObservabilityStorage, PromptBlocksStorage, RunFenceConflictError, SchedulesStorage, ScorerDefinitionsStorage, ScoresStorage, SkillsStorage, TABLE_AGENTS, TABLE_AGENT_VERSIONS, TABLE_BACKGROUND_TASKS, TABLE_DATASETS, TABLE_DATASET_ITEMS, TABLE_DATASET_VERSIONS, TABLE_EXPERIMENTS, TABLE_EXPERIMENT_RESULTS, TABLE_KNOWLEDGE_ACTIVITY, TABLE_KNOWLEDGE_MENTIONS, TABLE_KNOWLEDGE_NODES, TABLE_KNOWLEDGE_RECORDS, TABLE_KNOWLEDGE_SEMANTIC_OUTBOX, TABLE_MCP_CLIENTS, TABLE_MCP_CLIENT_VERSIONS, TABLE_MCP_SERVERS, TABLE_MCP_SERVER_VERSIONS, TABLE_MEMORY_RUN_FENCES, TABLE_MESSAGES, TABLE_NOTIFICATIONS, TABLE_PROMPT_BLOCKS, TABLE_PROMPT_BLOCK_VERSIONS, TABLE_RESOURCES, TABLE_SCHEDULES, TABLE_SCHEDULE_TRIGGERS, TABLE_SCORERS, TABLE_SCORER_DEFINITIONS, TABLE_SCORER_DEFINITION_VERSIONS, TABLE_SKILLS, TABLE_SKILL_BLOBS, TABLE_SKILL_VERSIONS, TABLE_SPANS, TABLE_THREADS, TABLE_WORKFLOW_DEFINITIONS, TABLE_WORKFLOW_RUN_OWNERS, TABLE_WORKFLOW_SNAPSHOT, TABLE_WORKSPACES, TABLE_WORKSPACE_VERSIONS, TraceStatus, WorkflowDefinitionsStorage, WorkflowsStorage, WorkspacesStorage, calculatePagination, canonicalizeKnowledgeScope, createKnowledgeUlid, createStorageErrorId, createVectorErrorId, ensureDate, hasErrorCode, isKnowledgeScopeVisible, isRunFenceConflictError, knowledgeScopeKey, knowledgeSemanticDocumentId, knowledgeSemanticIdempotencyKey, listTracesArgsSchema, normalizePerPage, normalizeScheduleTarget, parseDuration, parseKnowledgeNodeCursor, parseKnowledgeWikilinks, resolveListOrderBy, resolveRunFence, safelyParseJSON, storageMessageMatchesMetadataFilter, toTraceSpans, transformScoreRow, validateStorageMetadataFilter } from "@mastra/core/storage";
 import { MastraVector, validateUpsertInput, validateVectorValues } from "@mastra/core/vector";
 import { MongoBulkWriteError, MongoClient, ObjectId } from "mongodb";
 import { BaseFilterTranslator } from "@mastra/core/vector/filter";
@@ -9,7 +9,7 @@ import { MessageList } from "@mastra/core/agent";
 import { saveScorePayloadSchema } from "@mastra/core/evals";
 import { skillSnapshotFieldValuesEqual } from "@mastra/core/storage/domains/skills";
 //#region package.json
-var version = "1.23.1-alpha.0";
+var version = "1.23.1-alpha.1";
 //#endregion
 //#region src/vector/filter.ts
 /**
@@ -1618,20 +1618,29 @@ var MongoDBConnector = class MongoDBConnector {
 		return (await this.getConnection()).listCollections({ name: collectionName }).hasNext();
 	}
 	/**
-	* Returns true when the deployment supports multi-document transactions
-	* (replica set or sharded cluster). Standalone servers and custom connector
-	* handlers return false. Probed once and cached.
+	* Whether the deployment supports multi-document transactions (replica set
+	* or sharded cluster). Standalone servers and custom connector handlers
+	* don't. The answer is cached once a probe completes; a probe that fails
+	* rejects and is retried on the next call, so the answer never changes.
 	*/
-	async supportsTransactions() {
+	async probeTransactions() {
 		if (this.#supportsTransactions !== void 0) return this.#supportsTransactions;
 		if (!this.#client) {
 			this.#supportsTransactions = false;
 			return false;
 		}
+		const hello = await (await this.getConnection()).admin().command({ hello: 1 });
+		this.#supportsTransactions = Boolean(hello.setName) || hello.msg === "isdbgrid";
+		return this.#supportsTransactions;
+	}
+	/**
+	* Like {@link probeTransactions}, but a failed probe reads as no
+	* transaction support (and is retried on the next call), so a momentary
+	* outage degrades writes to non-atomic instead of failing them.
+	*/
+	async supportsTransactions() {
 		try {
-			const hello = await (await this.getConnection()).admin().command({ hello: 1 });
-			this.#supportsTransactions = Boolean(hello.setName) || hello.msg === "isdbgrid";
-			return this.#supportsTransactions;
+			return await this.probeTransactions();
 		} catch {
 			return false;
 		}
@@ -2278,14 +2287,19 @@ function cutoffFor(policy, encoding, now = Date.now()) {
 * Delete up to `limit` documents whose anchor field is older than `cutoff`.
 *
 * `deleteMany` has no limit clause, so each batch first collects `_id`s with a
-* bounded `find`, then deletes exactly that set — the Mongo equivalent of the
-* SQL stores' `DELETE ... WHERE rowid/ctid IN (SELECT ... LIMIT n)` pattern.
+* bounded `find`, then deletes that set — the Mongo equivalent of the SQL
+* stores' `DELETE ... WHERE rowid/ctid IN (SELECT ... LIMIT n)` pattern. The
+* delete re-checks the anchor, so a document updated after the `find` (a
+* renewed run ownership, a touched snapshot) survives.
 */
 async function pruneCollectionBatch({ connector, table, column, cutoff, limit }) {
 	const collection = await connector.getCollection(table);
 	const docs = await collection.find({ [column]: { $lt: cutoff } }).project({ _id: 1 }).limit(limit).toArray();
 	if (docs.length === 0) return 0;
-	return (await collection.deleteMany({ _id: { $in: docs.map((doc) => doc._id) } })).deletedCount;
+	return (await collection.deleteMany({
+		_id: { $in: docs.map((doc) => doc._id) },
+		[column]: { $lt: cutoff }
+	})).deletedCount;
 }
 /**
 * Run the bounded/cancellable batched-delete loop for a single logical target,
@@ -6560,6 +6574,40 @@ function isAppendOnlySince(storedObservations, snapshotObservations) {
 	return stored === snapshot || stored.startsWith(snapshot);
 }
 //#endregion
+//#region src/storage/domains/run-fencing.ts
+async function getRunClaims(connector, name) {
+	return await connector.getCollection(name);
+}
+/** Duplicate-key error: an upsert lost the race to insert the document. */
+function isDuplicateKeyError(error) {
+	return error?.code === 11e3;
+}
+/**
+* Runs a write, fenced when `check` is set. A fenced write runs in a
+* transaction that first updates the run's claim document, matching it on
+* the fence. A fence that is no longer current matches nothing, and the write
+* throws `RunFenceConflictError` before anything is written. A claim that
+* changes while the transaction runs makes the update conflict, so the
+* transaction retries and sees the new claim; a claim attempted after the
+* update waits until the transaction commits. Updating the claim document,
+* rather than reading it, is what makes concurrent claims conflict.
+*
+* Fencing is declared only on deployments that support transactions.
+*/
+async function withRunFence(connector, check, write) {
+	if (!check) return write();
+	const { claims, fence, operation } = check;
+	const collection = await getRunClaims(connector, claims);
+	return connector.withTransaction(async (session) => {
+		if ((await collection.updateOne({
+			_id: fence.runId,
+			generation: fence.generation,
+			ownerId: fence.ownerId
+		}, { $inc: { fencedWrites: 1 } }, { session })).matchedCount === 0) throw new RunFenceConflictError(fence, operation);
+		return write(session);
+	});
+}
+//#endregion
 //#region src/storage/domains/memory/index.ts
 /**
 * Local constant for the observational memory table name.
@@ -6629,9 +6677,6 @@ function buildSuccessorDocument(old, pending) {
 function matchStoredFields(doc, fields) {
 	return Object.fromEntries(fields.map((f) => [f, doc[f] === void 0 ? { $exists: false } : doc[f]]));
 }
-function isDuplicateKeyError(error) {
-	return error?.code === 11e3;
-}
 function omNotFound(operation, id) {
 	return new MastraError({
 		id: createStorageErrorId("MONGODB", operation, "NOT_FOUND"),
@@ -6671,11 +6716,14 @@ var MemoryStorageMongoDB = class MemoryStorageMongoDB extends MemoryStorage {
 		TABLE_THREADS,
 		TABLE_MESSAGES,
 		TABLE_RESOURCES,
-		OM_TABLE
+		OM_TABLE,
+		TABLE_MEMORY_RUN_FENCES
 	];
 	/**
 	* Retention-eligible collections. The observational-memory collection is
 	* excluded: it has no timestamp anchor to age on. All anchors are BSON dates.
+	* A run fence's `retiredAt` stays null until the execution holding it
+	* settles, and null is never pruned.
 	*/
 	static retentionTables = {
 		messages: {
@@ -6692,6 +6740,11 @@ var MemoryStorageMongoDB = class MemoryStorageMongoDB extends MemoryStorage {
 			table: TABLE_THREADS,
 			column: "createdAt",
 			indexed: true
+		},
+		runFences: {
+			table: TABLE_MEMORY_RUN_FENCES,
+			column: "retiredAt",
+			indexed: true
 		}
 	};
 	constructor(config) {
@@ -6710,11 +6763,76 @@ var MemoryStorageMongoDB = class MemoryStorageMongoDB extends MemoryStorage {
 		await this.#maintainSupersededBy();
 	}
 	/**
+	* Fenced writes need multi-document transactions, so only replica sets and
+	* sharded clusters fence. Rejects while the deployment can't be probed.
+	*/
+	supportsRunFencing() {
+		return this.#connector.probeTransactions();
+	}
+	async raiseRunFence(fence) {
+		try {
+			const fences = await getRunClaims(this.#connector, TABLE_MEMORY_RUN_FENCES);
+			for (;;) {
+				try {
+					await fences.updateOne({
+						_id: fence.runId,
+						$or: [{ generation: { $lt: fence.generation } }, {
+							generation: fence.generation,
+							ownerId: fence.ownerId
+						}]
+					}, { $set: {
+						generation: fence.generation,
+						ownerId: fence.ownerId,
+						retiredAt: null
+					} }, { upsert: true });
+					return true;
+				} catch (error) {
+					if (!isDuplicateKeyError(error)) throw error;
+				}
+				const current = await fences.findOne({ _id: fence.runId });
+				if (current && current.generation >= fence.generation) return current.generation === fence.generation && current.ownerId === fence.ownerId;
+			}
+		} catch (error) {
+			throw new MastraError({
+				id: createStorageErrorId("MONGODB", "RAISE_RUN_FENCE", "FAILED"),
+				domain: ErrorDomain.STORAGE,
+				category: ErrorCategory.THIRD_PARTY,
+				details: { runId: fence.runId }
+			}, error);
+		}
+	}
+	async retireRunFence(fence) {
+		try {
+			return (await (await getRunClaims(this.#connector, TABLE_MEMORY_RUN_FENCES)).updateOne({
+				_id: fence.runId,
+				generation: fence.generation,
+				ownerId: fence.ownerId
+			}, [{ $set: { retiredAt: "$$NOW" } }])).matchedCount > 0;
+		} catch (error) {
+			throw new MastraError({
+				id: createStorageErrorId("MONGODB", "RETIRE_RUN_FENCE", "FAILED"),
+				domain: ErrorDomain.STORAGE,
+				category: ErrorCategory.THIRD_PARTY,
+				details: { runId: fence.runId }
+			}, error);
+		}
+	}
+	/** The check a write must pass: its own fence, otherwise the one in scope. */
+	#runFenceCheck(fence, operation) {
+		const resolved = resolveRunFence(this, fence);
+		return resolved && {
+			claims: TABLE_MEMORY_RUN_FENCES,
+			fence: resolved,
+			operation
+		};
+	}
+	/**
 	* Delete memory rows older than each table's `maxAge`, batched. Order is
 	* messages → resources → threads so child rows never outlive the delete of
-	* their thread. Like `deleteThread()`, this does not sweep vector-store
-	* embeddings — semantic-recall vectors live in a separate vector store the
-	* memory domain cannot reach; cleaning those up is the operator's concern.
+	* their thread; run fences are independent and go last. Like
+	* `deleteThread()`, this does not sweep vector-store embeddings —
+	* semantic-recall vectors live in a separate vector store the memory domain
+	* cannot reach; cleaning those up is the operator's concern.
 	*/
 	async prune(policies, options) {
 		const targets = resolveTargets({
@@ -6723,7 +6841,8 @@ var MemoryStorageMongoDB = class MemoryStorageMongoDB extends MemoryStorage {
 			order: [
 				"messages",
 				"resources",
-				"threads"
+				"threads",
+				"runFences"
 			]
 		});
 		return runPrune({
@@ -6829,17 +6948,19 @@ var MemoryStorageMongoDB = class MemoryStorageMongoDB extends MemoryStorage {
 		}
 	}
 	async dangerouslyClearAll() {
-		const [threadsCollection, messagesCollection, resourcesCollection, omCollection] = await Promise.all([
+		const [threadsCollection, messagesCollection, resourcesCollection, omCollection, fencesCollection] = await Promise.all([
 			this.getCollection(TABLE_THREADS),
 			this.getCollection(TABLE_MESSAGES),
 			this.getCollection(TABLE_RESOURCES),
-			this.getCollection(OM_TABLE)
+			this.getCollection(OM_TABLE),
+			this.getCollection(TABLE_MEMORY_RUN_FENCES)
 		]);
 		await Promise.all([
 			threadsCollection.deleteMany({}),
 			messagesCollection.deleteMany({}),
 			resourcesCollection.deleteMany({}),
-			omCollection.deleteMany({})
+			omCollection.deleteMany({}),
+			fencesCollection.deleteMany({})
 		]);
 	}
 	parseRow(row) {
@@ -7195,7 +7316,7 @@ var MemoryStorageMongoDB = class MemoryStorageMongoDB extends MemoryStorage {
 			throw mastraError;
 		}
 	}
-	async saveMessages({ messages }) {
+	async saveMessages({ messages, fence }) {
 		if (messages.length === 0) return { messages: [] };
 		try {
 			if (!messages[0]?.threadId) throw new Error("Thread ID is required");
@@ -7223,12 +7344,16 @@ var MemoryStorageMongoDB = class MemoryStorageMongoDB extends MemoryStorage {
 			});
 			const allThreadIds = new Set(messages.map((m) => m.threadId));
 			const now = /* @__PURE__ */ new Date();
-			await this.#connector.withTransaction(async (session) => {
+			const writeMessages = async (session) => {
 				await collection.bulkWrite(messagesToInsert, { session });
 				for (const tid of allThreadIds) await threadsCollection.updateOne({ id: tid }, { $set: { updatedAt: now } }, { session });
-			});
+			};
+			const check = this.#runFenceCheck(fence, "saveMessages");
+			if (check) await withRunFence(this.#connector, check, writeMessages);
+			else await this.#connector.withTransaction(writeMessages);
 			return { messages: new MessageList().add(messages, "memory").get.all.db() };
 		} catch (error) {
+			if (isRunFenceConflictError(error)) throw error;
 			throw new MastraError({
 				id: createStorageErrorId("MONGODB", "SAVE_MESSAGES", "FAILED"),
 				domain: ErrorDomain.STORAGE,
@@ -7236,49 +7361,51 @@ var MemoryStorageMongoDB = class MemoryStorageMongoDB extends MemoryStorage {
 			}, error);
 		}
 	}
-	async updateMessages({ messages }) {
+	async updateMessages({ messages, fence }) {
 		if (messages.length === 0) return [];
 		const messageIds = messages.map((m) => m.id);
 		const collection = await this.getCollection(TABLE_MESSAGES);
-		const existingMessagesParsed = (await collection.find({ id: { $in: messageIds } }).toArray()).map((msg) => this.parseRow(msg));
-		if (existingMessagesParsed.length === 0) return [];
-		const threadIdsToUpdate = /* @__PURE__ */ new Set();
-		const bulkOps = [];
-		for (const existingMessage of existingMessagesParsed) {
-			const updatePayload = messages.find((m) => m.id === existingMessage.id);
-			if (!updatePayload) continue;
-			const { id, ...fieldsToUpdate } = updatePayload;
-			if (Object.keys(fieldsToUpdate).length === 0) continue;
-			threadIdsToUpdate.add(existingMessage.threadId);
-			if (updatePayload.threadId && updatePayload.threadId !== existingMessage.threadId) threadIdsToUpdate.add(updatePayload.threadId);
-			const updateDoc = {};
-			const updatableFields = { ...fieldsToUpdate };
-			if (updatableFields.content) {
-				const newContent = {
-					...existingMessage.content,
-					...updatableFields.content,
-					...existingMessage.content?.metadata && updatableFields.content.metadata ? { metadata: {
-						...existingMessage.content.metadata,
-						...updatableFields.content.metadata
-					} } : {}
-				};
-				updateDoc.content = JSON.stringify(newContent);
-				delete updatableFields.content;
-			}
-			for (const key in updatableFields) if (Object.prototype.hasOwnProperty.call(updatableFields, key)) {
-				const dbKey = key === "threadId" ? "thread_id" : key;
-				let value = updatableFields[key];
-				if (typeof value === "object" && value !== null) value = JSON.stringify(value);
-				updateDoc[dbKey] = value;
+		return withRunFence(this.#connector, this.#runFenceCheck(fence, "updateMessages"), async (session) => {
+			const existingMessagesParsed = (await collection.find({ id: { $in: messageIds } }, { session }).toArray()).map((msg) => this.parseRow(msg));
+			if (existingMessagesParsed.length === 0) return [];
+			const threadIdsToUpdate = /* @__PURE__ */ new Set();
+			const bulkOps = [];
+			for (const existingMessage of existingMessagesParsed) {
+				const updatePayload = messages.find((m) => m.id === existingMessage.id);
+				if (!updatePayload) continue;
+				const { id, ...fieldsToUpdate } = updatePayload;
+				if (Object.keys(fieldsToUpdate).length === 0) continue;
+				threadIdsToUpdate.add(existingMessage.threadId);
+				if (updatePayload.threadId && updatePayload.threadId !== existingMessage.threadId) threadIdsToUpdate.add(updatePayload.threadId);
+				const updateDoc = {};
+				const updatableFields = { ...fieldsToUpdate };
+				if (updatableFields.content) {
+					const newContent = {
+						...existingMessage.content,
+						...updatableFields.content,
+						...existingMessage.content?.metadata && updatableFields.content.metadata ? { metadata: {
+							...existingMessage.content.metadata,
+							...updatableFields.content.metadata
+						} } : {}
+					};
+					updateDoc.content = JSON.stringify(newContent);
+					delete updatableFields.content;
+				}
+				for (const key in updatableFields) if (Object.prototype.hasOwnProperty.call(updatableFields, key)) {
+					const dbKey = key === "threadId" ? "thread_id" : key;
+					let value = updatableFields[key];
+					if (typeof value === "object" && value !== null) value = JSON.stringify(value);
+					updateDoc[dbKey] = value;
+				}
+				if (Object.keys(updateDoc).length > 0) bulkOps.push({ updateOne: {
+					filter: { id },
+					update: { $set: updateDoc }
+				} });
 			}
-			if (Object.keys(updateDoc).length > 0) bulkOps.push({ updateOne: {
-				filter: { id },
-				update: { $set: updateDoc }
-			} });
-		}
-		if (bulkOps.length > 0) await collection.bulkWrite(bulkOps);
-		if (threadIdsToUpdate.size > 0) await (await this.getCollection(TABLE_THREADS)).updateMany({ id: { $in: Array.from(threadIdsToUpdate) } }, { $set: { updatedAt: /* @__PURE__ */ new Date() } });
-		return (await collection.find({ id: { $in: messageIds } }).toArray()).map((row) => this.parseRow(row));
+			if (bulkOps.length > 0) await collection.bulkWrite(bulkOps, { session });
+			if (threadIdsToUpdate.size > 0) await (await this.getCollection(TABLE_THREADS)).updateMany({ id: { $in: Array.from(threadIdsToUpdate) } }, { $set: { updatedAt: /* @__PURE__ */ new Date() } }, { session });
+			return (await collection.find({ id: { $in: messageIds } }, { session }).toArray()).map((row) => this.parseRow(row));
+		});
 	}
 	async getResourceById({ resourceId }) {
 		try {
@@ -7313,9 +7440,11 @@ var MemoryStorageMongoDB = class MemoryStorageMongoDB extends MemoryStorage {
 			}, error);
 		}
 	}
-	async updateResource({ resourceId, workingMemory, metadata }) {
+	async updateResource({ resourceId, workingMemory, metadata, fence }) {
 		try {
+			const check = this.#runFenceCheck(fence, "updateResource");
 			const existingResource = await this.getResourceById({ resourceId });
+			const collection = await this.getCollection(TABLE_RESOURCES);
 			if (!existingResource) {
 				const newResource = {
 					id: resourceId,
@@ -7324,7 +7453,11 @@ var MemoryStorageMongoDB = class MemoryStorageMongoDB extends MemoryStorage {
 					createdAt: /* @__PURE__ */ new Date(),
 					updatedAt: /* @__PURE__ */ new Date()
 				};
-				return this.saveResource({ resource: newResource });
+				await withRunFence(this.#connector, check, (session) => collection.updateOne({ id: resourceId }, { $set: { ...newResource } }, {
+					upsert: true,
+					session
+				}));
+				return newResource;
 			}
 			const updatedResource = {
 				...existingResource,
@@ -7335,13 +7468,13 @@ var MemoryStorageMongoDB = class MemoryStorageMongoDB extends MemoryStorage {
 				} : existingResource.metadata,
 				updatedAt: /* @__PURE__ */ new Date()
 			};
-			const collection = await this.getCollection(TABLE_RESOURCES);
 			const updateDoc = { updatedAt: updatedResource.updatedAt };
 			if (workingMemory !== void 0) updateDoc.workingMemory = workingMemory;
 			if (metadata) updateDoc.metadata = updatedResource.metadata;
-			await collection.updateOne({ id: resourceId }, { $set: updateDoc });
+			await withRunFence(this.#connector, check, (session) => collection.updateOne({ id: resourceId }, { $set: updateDoc }, { session }));
 			return updatedResource;
 		} catch (error) {
+			if (isRunFenceConflictError(error)) throw error;
 			throw new MastraError({
 				id: createStorageErrorId("MONGODB", "UPDATE_RESOURCE", "FAILED"),
 				domain: ErrorDomain.STORAGE,
@@ -7441,14 +7574,19 @@ var MemoryStorageMongoDB = class MemoryStorageMongoDB extends MemoryStorage {
 			throw mastraError;
 		}
 	}
-	async saveThread({ thread }) {
+	async saveThread({ thread, fence }) {
 		try {
-			await (await this.getCollection(TABLE_THREADS)).updateOne({ id: thread.id }, { $set: {
+			const collection = await this.getCollection(TABLE_THREADS);
+			await withRunFence(this.#connector, this.#runFenceCheck(fence, "saveThread"), (session) => collection.updateOne({ id: thread.id }, { $set: {
 				...thread,
 				metadata: thread.metadata
-			} }, { upsert: true });
+			} }, {
+				upsert: true,
+				session
+			}));
 			return thread;
 		} catch (error) {
+			if (isRunFenceConflictError(error)) throw error;
 			throw new MastraError({
 				id: createStorageErrorId("MONGODB", "SAVE_THREAD", "FAILED"),
 				domain: ErrorDomain.STORAGE,
@@ -7457,7 +7595,8 @@ var MemoryStorageMongoDB = class MemoryStorageMongoDB extends MemoryStorage {
 			}, error);
 		}
 	}
-	async updateThread({ id, title, metadata }) {
+	async updateThread({ id, title, metadata, fence }) {
+		const check = this.#runFenceCheck(fence, "updateThread");
 		const thread = await this.getThreadById({ threadId: id });
 		if (!thread) throw new MastraError({
 			id: createStorageErrorId("MONGODB", "UPDATE_THREAD", "NOT_FOUND"),
@@ -7480,12 +7619,14 @@ var MemoryStorageMongoDB = class MemoryStorageMongoDB extends MemoryStorage {
 			updatedAt: now
 		};
 		try {
-			await (await this.getCollection(TABLE_THREADS)).updateOne({ id }, { $set: {
+			const collection = await this.getCollection(TABLE_THREADS);
+			await withRunFence(this.#connector, check, (session) => collection.updateOne({ id }, { $set: {
 				title: updatedThread.title,
 				metadata: updatedThread.metadata,
 				updatedAt: now
-			} });
+			} }, { session }));
 		} catch (error) {
+			if (isRunFenceConflictError(error)) throw error;
 			throw new MastraError({
 				id: createStorageErrorId("MONGODB", "UPDATE_THREAD", "FAILED"),
 				domain: ErrorDomain.STORAGE,
@@ -7508,16 +7649,19 @@ var MemoryStorageMongoDB = class MemoryStorageMongoDB extends MemoryStorage {
 			}, error);
 		}
 	}
-	async deleteMessages(messageIds) {
+	async deleteMessages(messageIds, options) {
 		if (messageIds.length === 0) return;
 		try {
 			const messagesCollection = await this.getCollection(TABLE_MESSAGES);
 			const threadsCollection = await this.getCollection(TABLE_THREADS);
-			const messagesToDelete = await messagesCollection.find({ id: { $in: messageIds } }).toArray();
-			const threadIds = [...new Set(messagesToDelete.map((m) => m.thread_id))];
-			await messagesCollection.deleteMany({ id: { $in: messageIds } });
-			if (threadIds.length > 0) await threadsCollection.updateMany({ id: { $in: threadIds } }, { $set: { updatedAt: /* @__PURE__ */ new Date() } });
+			await withRunFence(this.#connector, this.#runFenceCheck(options?.fence, "deleteMessages"), async (session) => {
+				const messagesToDelete = await messagesCollection.find({ id: { $in: messageIds } }, { session }).toArray();
+				const threadIds = [...new Set(messagesToDelete.map((m) => m.thread_id))];
+				await messagesCollection.deleteMany({ id: { $in: messageIds } }, { session });
+				if (threadIds.length > 0) await threadsCollection.updateMany({ id: { $in: threadIds } }, { $set: { updatedAt: /* @__PURE__ */ new Date() } }, { session });
+			});
 		} catch (error) {
+			if (isRunFenceConflictError(error)) throw error;
 			throw new MastraError({
 				id: createStorageErrorId("MONGODB", "DELETE_MESSAGES", "FAILED"),
 				domain: ErrorDomain.STORAGE,
@@ -7884,7 +8028,7 @@ var MemoryStorageMongoDB = class MemoryStorageMongoDB extends MemoryStorage {
 					applied: false,
 					reason: "conflict"
 				};
-				if ((await collection.updateOne({
+				if ((await withRunFence(this.#connector, this.#runFenceCheck(void 0, "updateActiveObservations"), (session) => collection.updateOne({
 					id: input.id,
 					supersededBy: null,
 					...matchStoredFields(doc, ["activeObservations", "lastObservedAt"])
@@ -7898,7 +8042,7 @@ var MemoryStorageMongoDB = class MemoryStorageMongoDB extends MemoryStorage {
 						updatedAt: /* @__PURE__ */ new Date()
 					},
 					$inc: { totalTokensObserved: safeTokenCount }
-				})).matchedCount === 1) return { applied: true };
+				}, { session }))).matchedCount === 1) return { applied: true };
 			}
 			throw omConflict("UPDATE_ACTIVE_OBSERVATIONS", input.id);
 		} catch (error) {
@@ -7914,7 +8058,7 @@ var MemoryStorageMongoDB = class MemoryStorageMongoDB extends MemoryStorage {
 	async createReflectionGeneration(input) {
 		try {
 			const { currentRecord } = input;
-			return await this.#rollOver(currentRecord, (doc) => {
+			return await this.#rollOver(currentRecord, "createReflectionGeneration", (doc) => {
 				const stored = this.parseOMDocument(doc);
 				const plan = planReflectionGenerationText({
 					storedObservations: stored.activeObservations,
@@ -8151,7 +8295,7 @@ var MemoryStorageMongoDB = class MemoryStorageMongoDB extends MemoryStorage {
 			let hops = 0;
 			let retries = 0;
 			for (;;) {
-				if ((await collection.updateOne({
+				if ((await withRunFence(this.#connector, this.#runFenceCheck(void 0, "updateBufferedObservations"), (session) => collection.updateOne({
 					id: targetId,
 					supersededBy: null,
 					"bufferedObservationChunks.cycleId": { $ne: input.chunk.cycleId },
@@ -8165,7 +8309,7 @@ var MemoryStorageMongoDB = class MemoryStorageMongoDB extends MemoryStorage {
 					bufferedObservationChunks: { $concatArrays: [{ $ifNull: ["$bufferedObservationChunks", []] }, { $literal: [newChunk] }] },
 					...lastBufferedAtTime ? { lastBufferedAtTime: { $max: ["$lastBufferedAtTime", { $literal: lastBufferedAtTime }] } } : {},
 					updatedAt: { $literal: /* @__PURE__ */ new Date() }
-				} }])).matchedCount === 1) return {
+				} }], { session }))).matchedCount === 1) return {
 					persisted: true,
 					recordId: targetId
 				};
@@ -8188,7 +8332,7 @@ var MemoryStorageMongoDB = class MemoryStorageMongoDB extends MemoryStorage {
 					recordId: targetId
 				};
 				if (stored !== void 0 && stored !== null && !Array.isArray(stored)) {
-					if ((await collection.updateOne({
+					if ((await withRunFence(this.#connector, this.#runFenceCheck(void 0, "updateBufferedObservations"), (session) => collection.updateOne({
 						id: targetId,
 						supersededBy: null,
 						bufferedObservationChunks: stored,
@@ -8197,7 +8341,7 @@ var MemoryStorageMongoDB = class MemoryStorageMongoDB extends MemoryStorage {
 						bufferedObservationChunks: [newChunk],
 						...lastBufferedAtTime ? { lastBufferedAtTime: maxObservationCursor(doc.lastBufferedAtTime, lastBufferedAtTime) } : {},
 						updatedAt: /* @__PURE__ */ new Date()
-					} })).matchedCount === 1) return {
+					} }, { session }))).matchedCount === 1) return {
 						persisted: true,
 						recordId: targetId
 					};
@@ -8287,7 +8431,7 @@ var MemoryStorageMongoDB = class MemoryStorageMongoDB extends MemoryStorage {
 				const newTokenCount = existingTokenCount + activatedTokens;
 				const existingPending = Number(doc.pendingMessageTokens || 0);
 				const newPending = Math.max(0, existingPending - activatedMessageTokens);
-				if ((await collection.updateOne({
+				if ((await withRunFence(this.#connector, this.#runFenceCheck(void 0, "swapBufferedToActive"), (session) => collection.updateOne({
 					id: input.id,
 					supersededBy: null,
 					...matchStoredFields(doc, [
@@ -8304,7 +8448,7 @@ var MemoryStorageMongoDB = class MemoryStorageMongoDB extends MemoryStorage {
 					bufferedObservationChunks: remainingChunks,
 					lastObservedAt: maxObservationCursor(doc.lastObservedAt, lastObservedAt),
 					updatedAt: /* @__PURE__ */ new Date()
-				} })).matchedCount === 0) continue;
+				} }, { session }))).matchedCount === 0) continue;
 				const latestChunkHints = activatedChunks[activatedChunks.length - 1];
 				return {
 					chunksActivated: activatedChunks.length,
@@ -8353,13 +8497,13 @@ var MemoryStorageMongoDB = class MemoryStorageMongoDB extends MemoryStorage {
 			const newContent = existingContent ? `${existingContent}\n\n${input.reflection}` : input.reflection;
 			const newTokens = existingTokens + input.tokenCount;
 			const newInputTokens = existingInputTokens + input.inputTokenCount;
-			if ((await collection.updateOne({ id: input.id }, { $set: {
+			if ((await withRunFence(this.#connector, this.#runFenceCheck(void 0, "updateBufferedReflection"), (session) => collection.updateOne({ id: input.id }, { $set: {
 				bufferedReflection: newContent,
 				bufferedReflectionTokens: newTokens,
 				bufferedReflectionInputTokens: newInputTokens,
 				reflectedObservationLineCount: input.reflectedObservationLineCount,
 				updatedAt: /* @__PURE__ */ new Date()
-			} })).matchedCount === 0) throw new MastraError({
+			} }, { session }))).matchedCount === 0) throw new MastraError({
 				id: createStorageErrorId("MONGODB", "UPDATE_BUFFERED_REFLECTION", "NOT_FOUND"),
 				text: `Observational memory record not found: ${input.id}`,
 				domain: ErrorDomain.STORAGE,
@@ -8379,7 +8523,7 @@ var MemoryStorageMongoDB = class MemoryStorageMongoDB extends MemoryStorage {
 	async swapBufferedReflectionToActive(input) {
 		try {
 			const { currentRecord } = input;
-			return await this.#rollOver(currentRecord, (doc) => {
+			return await this.#rollOver(currentRecord, "swapBufferedReflectionToActive", (doc) => {
 				const stored = this.parseOMDocument(doc);
 				const bufferedReflection = stored.bufferedReflection || "";
 				if (!bufferedReflection) throw new MastraError({
@@ -8465,7 +8609,7 @@ var MemoryStorageMongoDB = class MemoryStorageMongoDB extends MemoryStorage {
 	*
 	* A crash between 2 and 3 is finished by the next head read (`#getHeadDoc`) or startup.
 	*/
-	async #rollOver(currentRecord, decide) {
+	async #rollOver(currentRecord, operation, decide) {
 		const collection = await this.getCollection(OM_TABLE);
 		for (let attempt = 1; attempt <= OM_MAX_CONDITIONAL_ATTEMPTS; attempt++) {
 			const doc = await collection.findOne({ id: currentRecord.id });
@@ -8476,7 +8620,7 @@ var MemoryStorageMongoDB = class MemoryStorageMongoDB extends MemoryStorage {
 			}
 			const pending = decide(doc);
 			if (!pending) return this.parseOMDocument(doc);
-			const frozen = await collection.findOneAndUpdate({
+			const frozen = await withRunFence(this.#connector, this.#runFenceCheck(void 0, operation), (session) => collection.findOneAndUpdate({
 				id: doc.id,
 				supersededBy: null,
 				...matchStoredFields(doc, [
@@ -8488,7 +8632,10 @@ var MemoryStorageMongoDB = class MemoryStorageMongoDB extends MemoryStorage {
 			}, { $set: {
 				supersededBy: pending.newId,
 				pendingSuccessor: pending
-			} }, { returnDocument: "before" });
+			} }, {
+				returnDocument: "before",
+				session
+			}));
 			if (!frozen) continue;
 			return this.parseOMDocument(await this.#completeRollover(collection, frozen, pending));
 		}
@@ -11803,33 +11950,58 @@ var MongoDBWorkflowDefinitionsStore = class MongoDBWorkflowDefinitionsStore exte
 };
 //#endregion
 //#region src/storage/domains/workflows/index.ts
+const RUN_OWNER_PROJECTION = {
+	generation: 1,
+	ownerId: 1,
+	leaseExpiresAt: 1,
+	live: { $gt: ["$leaseExpiresAt", "$$NOW"] }
+};
+const NOT_LIVE = { $or: [{ leaseExpiresAt: null }, { $expr: { $lte: ["$leaseExpiresAt", "$$NOW"] } }] };
+function toRunOwnershipRecord(doc) {
+	return {
+		runId: doc._id,
+		generation: Number(doc.generation),
+		ownerId: doc.ownerId,
+		leaseExpiresAt: doc.leaseExpiresAt ?? null,
+		live: Boolean(doc.live)
+	};
+}
 var WorkflowsStorageMongoDB = class WorkflowsStorageMongoDB extends WorkflowsStorage {
 	#connector;
 	#skipDefaultIndexes;
 	#indexes;
 	/** Collections managed by this domain */
-	static MANAGED_COLLECTIONS = [TABLE_WORKFLOW_SNAPSHOT];
+	static MANAGED_COLLECTIONS = [TABLE_WORKFLOW_SNAPSHOT, TABLE_WORKFLOW_RUN_OWNERS];
 	/**
-	* Anchor is `updatedAt` (BSON date), so the policy reads as inactivity:
-	* a run is pruned only after its snapshot has not been touched for `maxAge`.
+	* Both anchors are `updatedAt` (BSON date), so the policies read as
+	* inactivity: a run is pruned only after its snapshot has not been touched
+	* for `maxAge`, and an ownership document only after no claim, renewal or
+	* release has touched it for `maxAge`.
 	*/
-	static retentionTables = { workflowSnapshot: {
-		table: TABLE_WORKFLOW_SNAPSHOT,
-		column: "updatedAt",
-		indexed: true
-	} };
+	static retentionTables = {
+		workflowSnapshot: {
+			table: TABLE_WORKFLOW_SNAPSHOT,
+			column: "updatedAt",
+			indexed: true
+		},
+		runOwnership: {
+			table: TABLE_WORKFLOW_RUN_OWNERS,
+			column: "updatedAt",
+			indexed: true
+		}
+	};
 	constructor(config) {
 		super();
 		this.#connector = resolveMongoDBConfig(config);
 		this.#skipDefaultIndexes = config.skipDefaultIndexes;
 		this.#indexes = config.indexes?.filter((idx) => WorkflowsStorageMongoDB.MANAGED_COLLECTIONS.includes(idx.collection));
 	}
-	/** Delete workflow snapshots idle for longer than the policy's `maxAge`, batched. */
+	/** Delete workflow snapshots and ownership documents idle for longer than each policy's `maxAge`, batched. */
 	async prune(policies, options) {
 		const targets = resolveTargets({
 			policies,
 			descriptor: WorkflowsStorageMongoDB.retentionTables,
-			order: ["workflowSnapshot"]
+			order: ["workflowSnapshot", "runOwnership"]
 		});
 		return runPrune({
 			connector: this.#connector,
@@ -11842,6 +12014,13 @@ var WorkflowsStorageMongoDB = class WorkflowsStorageMongoDB extends WorkflowsSto
 	supportsConcurrentUpdates() {
 		return true;
 	}
+	/**
+	* Fenced writes need multi-document transactions, so only replica sets and
+	* sharded clusters fence. Rejects while the deployment can't be probed.
+	*/
+	supportsRunFencing() {
+		return this.#connector.probeTransactions();
+	}
 	async getCollection(name) {
 		return this.#connector.getCollection(name);
 	}
@@ -11849,6 +12028,124 @@ var WorkflowsStorageMongoDB = class WorkflowsStorageMongoDB extends WorkflowsSto
 		await this.createDefaultIndexes();
 		await this.createCustomIndexes();
 	}
+	#runFenceCheck(fence, runId, operation) {
+		const resolved = resolveRunFence(this, fence, runId);
+		return resolved && {
+			claims: TABLE_WORKFLOW_RUN_OWNERS,
+			fence: resolved,
+			operation
+		};
+	}
+	async #readRunOwner(runId) {
+		const doc = await (await getRunClaims(this.#connector, TABLE_WORKFLOW_RUN_OWNERS)).findOne({ _id: runId }, { projection: RUN_OWNER_PROJECTION });
+		return doc ? toRunOwnershipRecord(doc) : null;
+	}
+	#ownershipError(operation, runId, error) {
+		return new MastraError({
+			id: createStorageErrorId("MONGODB", operation, "FAILED"),
+			domain: ErrorDomain.STORAGE,
+			category: ErrorCategory.THIRD_PARTY,
+			details: { runId }
+		}, error);
+	}
+	async claimRunOwnership({ runId, ownerId, leaseMs, force, expectedGeneration }) {
+		try {
+			const owners = await getRunClaims(this.#connector, TABLE_WORKFLOW_RUN_OWNERS);
+			const claim = (generation) => [{ $set: {
+				generation,
+				ownerId: { $literal: ownerId },
+				leaseExpiresAt: { $add: ["$$NOW", leaseMs] },
+				updatedAt: "$$NOW"
+			} }];
+			if (!expectedGeneration) {
+				try {
+					const created = await owners.findOneAndUpdate({
+						_id: runId,
+						generation: { $exists: false }
+					}, claim({ $max: [1, { $toLong: "$$NOW" }] }), {
+						upsert: true,
+						returnDocument: "after",
+						projection: RUN_OWNER_PROJECTION
+					});
+					if (created) return {
+						acquired: true,
+						record: toRunOwnershipRecord(created)
+					};
+				} catch (error) {
+					if (!isDuplicateKeyError(error)) throw error;
+				}
+				if (expectedGeneration === 0) return {
+					acquired: false,
+					record: await this.#readRunOwner(runId)
+				};
+			}
+			const advanced = await owners.findOneAndUpdate({
+				_id: runId,
+				...expectedGeneration !== void 0 ? { generation: expectedGeneration } : {},
+				...force ? {} : NOT_LIVE
+			}, claim({ $add: ["$generation", 1] }), {
+				returnDocument: "after",
+				projection: RUN_OWNER_PROJECTION
+			});
+			if (advanced) return {
+				acquired: true,
+				record: toRunOwnershipRecord(advanced)
+			};
+			return {
+				acquired: false,
+				record: await this.#readRunOwner(runId)
+			};
+		} catch (error) {
+			throw this.#ownershipError("CLAIM_RUN_OWNERSHIP", runId, error);
+		}
+	}
+	async renewRunOwnership({ leaseMs, ...fence }) {
+		try {
+			const renewed = await (await getRunClaims(this.#connector, TABLE_WORKFLOW_RUN_OWNERS)).findOneAndUpdate({
+				_id: fence.runId,
+				generation: fence.generation,
+				ownerId: fence.ownerId,
+				leaseExpiresAt: { $ne: null }
+			}, [{ $set: {
+				leaseExpiresAt: { $add: ["$$NOW", leaseMs] },
+				updatedAt: "$$NOW"
+			} }], {
+				returnDocument: "after",
+				projection: RUN_OWNER_PROJECTION
+			});
+			if (renewed) return {
+				renewed: true,
+				record: toRunOwnershipRecord(renewed)
+			};
+			return {
+				renewed: false,
+				record: await this.#readRunOwner(fence.runId)
+			};
+		} catch (error) {
+			throw this.#ownershipError("RENEW_RUN_OWNERSHIP", fence.runId, error);
+		}
+	}
+	async releaseRunOwnership(fence) {
+		try {
+			return (await (await getRunClaims(this.#connector, TABLE_WORKFLOW_RUN_OWNERS)).updateOne({
+				_id: fence.runId,
+				generation: fence.generation,
+				ownerId: fence.ownerId
+			}, [{ $set: {
+				leaseExpiresAt: null,
+				updatedAt: "$$NOW"
+			} }])).matchedCount > 0;
+		} catch (error) {
+			throw this.#ownershipError("RELEASE_RUN_OWNERSHIP", fence.runId, error);
+		}
+	}
+	async getRunOwnership({ runId }) {
+		try {
+			return await this.#readRunOwner(runId);
+		} catch (error) {
+			throw this.#ownershipError("GET_RUN_OWNERSHIP", runId, error);
+		}
+	}
 	/**
 	* Returns default index definitions for the workflows domain collections.
 	*/
@@ -11908,8 +12205,10 @@ var WorkflowsStorageMongoDB = class WorkflowsStorageMongoDB extends WorkflowsSto
 	}
 	async dangerouslyClearAll() {
 		await (await this.getCollection(TABLE_WORKFLOW_SNAPSHOT)).deleteMany({});
+		await (await this.getCollection(TABLE_WORKFLOW_RUN_OWNERS)).deleteMany({});
 	}
-	async updateWorkflowResults({ workflowName, runId, stepId, result, requestContext, state }) {
+	async updateWorkflowResults({ workflowName, runId, stepId, result, requestContext, state, fence }) {
+		const check = this.#runFenceCheck(fence, runId, "updateWorkflowResults");
 		try {
 			const collection = await this.getCollection(TABLE_WORKFLOW_SNAPSHOT);
 			const now = /* @__PURE__ */ new Date();
@@ -11929,7 +12228,7 @@ var WorkflowsStorageMongoDB = class WorkflowsStorageMongoDB extends WorkflowsSto
 				runId,
 				requestContext: {}
 			};
-			const updatedDoc = await collection.findOneAndUpdate({
+			const updatedDoc = await withRunFence(this.#connector, check, (session) => collection.findOneAndUpdate({
 				workflow_name: workflowName,
 				run_id: runId
 			}, [{ $set: {
@@ -11944,10 +12243,12 @@ var WorkflowsStorageMongoDB = class WorkflowsStorageMongoDB extends WorkflowsSto
 				createdAt: { $ifNull: ["$createdAt", now] }
 			} }], {
 				upsert: true,
-				returnDocument: "after"
-			});
+				returnDocument: "after",
+				session
+			}));
 			return (typeof updatedDoc?.snapshot === "string" ? JSON.parse(updatedDoc.snapshot) : updatedDoc?.snapshot)?.context || {};
 		} catch (error) {
+			if (isRunFenceConflictError(error)) throw error;
 			throw new MastraError({
 				id: createStorageErrorId("MONGODB", "UPDATE_WORKFLOW_RESULTS", "FAILED"),
 				domain: ErrorDomain.STORAGE,
@@ -11960,7 +12261,8 @@ var WorkflowsStorageMongoDB = class WorkflowsStorageMongoDB extends WorkflowsSto
 			}, error);
 		}
 	}
-	async updateWorkflowState({ workflowName, runId, opts }) {
+	async updateWorkflowState({ workflowName, runId, opts, fence }) {
+		const check = this.#runFenceCheck(fence, runId, "updateWorkflowState");
 		try {
 			const collection = await this.getCollection(TABLE_WORKFLOW_SNAPSHOT);
 			const { expectedStatus, ...state } = opts;
@@ -11969,10 +12271,13 @@ var WorkflowsStorageMongoDB = class WorkflowsStorageMongoDB extends WorkflowsSto
 				run_id: runId
 			};
 			if (expectedStatus !== void 0) filter["snapshot.status"] = { $in: Array.isArray(expectedStatus) ? expectedStatus : [expectedStatus] };
-			const updatedDoc = await collection.findOneAndUpdate(filter, [{ $set: {
+			const updatedDoc = await withRunFence(this.#connector, check, (session) => collection.findOneAndUpdate(filter, [{ $set: {
 				snapshot: { $mergeObjects: ["$snapshot", state] },
 				updatedAt: /* @__PURE__ */ new Date()
-			} }], { returnDocument: "after" });
+			} }], {
+				returnDocument: "after",
+				session
+			}));
 			if (!updatedDoc) return;
 			const snapshot = typeof updatedDoc.snapshot === "string" ? JSON.parse(updatedDoc.snapshot) : updatedDoc.snapshot;
 			if (!snapshot?.context) throw new MastraError({
@@ -11987,7 +12292,7 @@ var WorkflowsStorageMongoDB = class WorkflowsStorageMongoDB extends WorkflowsSto
 			});
 			return snapshot;
 		} catch (error) {
-			if (error instanceof MastraError) throw error;
+			if (error instanceof MastraError || isRunFenceConflictError(error)) throw error;
 			throw new MastraError({
 				id: createStorageErrorId("MONGODB", "UPDATE_WORKFLOW_STATE", "FAILED"),
 				domain: ErrorDomain.STORAGE,
@@ -11999,10 +12304,12 @@ var WorkflowsStorageMongoDB = class WorkflowsStorageMongoDB extends WorkflowsSto
 			}, error);
 		}
 	}
-	async persistWorkflowSnapshot({ workflowName, runId, resourceId, snapshot, createdAt, updatedAt }) {
+	async persistWorkflowSnapshot({ workflowName, runId, resourceId, snapshot, createdAt, updatedAt, fence }) {
+		const check = this.#runFenceCheck(fence, runId, "persistWorkflowSnapshot");
 		try {
 			const now = /* @__PURE__ */ new Date();
-			await (await this.getCollection(TABLE_WORKFLOW_SNAPSHOT)).updateOne({
+			const collection = await this.getCollection(TABLE_WORKFLOW_SNAPSHOT);
+			await withRunFence(this.#connector, check, (session) => collection.updateOne({
 				workflow_name: workflowName,
 				run_id: runId
 			}, {
@@ -12014,8 +12321,12 @@ var WorkflowsStorageMongoDB = class WorkflowsStorageMongoDB extends WorkflowsSto
 					updatedAt: updatedAt ?? now
 				},
 				$setOnInsert: { createdAt: createdAt ?? now }
-			}, { upsert: true });
+			}, {
+				upsert: true,
+				session
+			}));
 		} catch (error) {
+			if (isRunFenceConflictError(error)) throw error;
 			throw new MastraError({
 				id: createStorageErrorId("MONGODB", "PERSIST_WORKFLOW_SNAPSHOT", "FAILED"),
 				domain: ErrorDomain.STORAGE,
@@ -12108,13 +12419,16 @@ var WorkflowsStorageMongoDB = class WorkflowsStorageMongoDB extends WorkflowsSto
 			}, error);
 		}
 	}
-	async deleteWorkflowRunById({ runId, workflowName }) {
+	async deleteWorkflowRunById({ runId, workflowName, fence }) {
+		const check = this.#runFenceCheck(fence, runId, "deleteWorkflowRunById");
 		try {
-			await (await this.getCollection(TABLE_WORKFLOW_SNAPSHOT)).deleteOne({
+			const collection = await this.getCollection(TABLE_WORKFLOW_SNAPSHOT);
+			await withRunFence(this.#connector, check, (session) => collection.deleteOne({
 				workflow_name: workflowName,
 				run_id: runId
-			});
+			}, { session }));
 		} catch (error) {
+			if (isRunFenceConflictError(error)) throw error;
 			throw new MastraError({
 				id: createStorageErrorId("MONGODB", "DELETE_WORKFLOW_RUN_BY_ID", "FAILED"),
 				domain: ErrorDomain.STORAGE,