@mastra/pg 1.31.0-alpha.3 → 1.31.0

package/dist/index.js

@@ -1,6 +1,6 @@
 import { ErrorCategory, ErrorDomain, MastraError } from "@mastra/core/error";
 import * as coreStorage from "@mastra/core/storage";
-import { AgentsStorage, BRANCH_SPAN_TYPES, BackgroundTasksStorage, BlobStore, ChannelsStorage, DATASETS_SCHEMA, DATASET_ITEMS_SCHEMA, DATASET_VERSIONS_SCHEMA, DatasetsStorage, EXPERIMENTS_SCHEMA, EXPERIMENT_RESULTS_SCHEMA, EntityType, ExperimentsStorage, FactoryStorage, FavoritesStorage, KNOWLEDGE_SEMANTIC_OUTBOX_SCHEMA, KnowledgeConflictError, KnowledgeNotFoundError, KnowledgeStorage, MCPClientsStorage, MCPServersStorage, METRIC_DISTINCT_COLUMNS, MastraCompositeStore, MemoryStorage, NotificationsStorage, OBSERVATIONAL_MEMORY_TABLE_SCHEMA, ObservabilityStorage, PromptBlocksStorage, SchedulesStorage, ScorerDefinitionsStorage, ScoresStorage, SkillsStorage, TABLE_AGENTS, TABLE_AGENT_VERSIONS, TABLE_BACKGROUND_TASKS, TABLE_CHANNEL_CONFIG, TABLE_CHANNEL_INSTALLATIONS, TABLE_CONFIGS, TABLE_DATASETS, TABLE_DATASET_ITEMS, TABLE_DATASET_VERSIONS, TABLE_EXPERIMENTS, TABLE_EXPERIMENT_RESULTS, TABLE_FAVORITES, 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_SCHEMAS, TABLE_SCORERS, TABLE_SCORER_DEFINITIONS, TABLE_SCORER_DEFINITION_VERSIONS, TABLE_SKILLS, TABLE_SKILL_BLOBS, TABLE_SKILL_VERSIONS, TABLE_SPANS, TABLE_THREADS, TABLE_THREAD_STATE, TABLE_TOOL_PROVIDER_CONNECTIONS, TABLE_WORKFLOW_DEFINITIONS, TABLE_WORKFLOW_SNAPSHOT, TABLE_WORKSPACES, TABLE_WORKSPACE_VERSIONS, ThreadStateStorage, ToolProviderConnectionsStorage, TraceStatus, UniqueViolationError, WorkflowDefinitionsStorage, WorkflowsStorage, WorkspacesStorage, calculatePagination, canonicalizeKnowledgeScope, createKnowledgeUlid, createStorageErrorId, createVectorErrorId, ensureDate, executeRetentionPrune, getDefaultValue, getSqlType, hasErrorCode, isKnowledgeScopeVisible, knowledgeScopeKey, knowledgeSemanticDocumentId, knowledgeSemanticIdempotencyKey, listBranchesArgsSchema, listFeedbackArgsSchema, listLogsArgsSchema, listMetricsArgsSchema, listScoresArgsSchema, listTracesArgsSchema, matchesExpectedWorkflowStatus, mergeWorkflowStepResult, normalizePerPage, normalizeScheduleTarget, parseDuration, parseKnowledgeNodeCursor, parseKnowledgeWikilinks, resolveListOrderBy, resolveRetentionTargets, retentionCutoffMs, runRetentionBatches, safelyParseJSON, storageMessageMatchesMetadataFilter, toTraceSpans, transformScoreRow, validateStorageMetadataFilter } from "@mastra/core/storage";
+import { AgentsStorage, BRANCH_SPAN_TYPES, BackgroundTasksStorage, BlobStore, ChannelsStorage, DATASETS_SCHEMA, DATASET_ITEMS_SCHEMA, DATASET_VERSIONS_SCHEMA, DatasetsStorage, EXPERIMENTS_SCHEMA, EXPERIMENT_RESULTS_SCHEMA, EntityType, ExperimentsStorage, FactoryStorage, FavoritesStorage, KNOWLEDGE_SEMANTIC_OUTBOX_SCHEMA, KnowledgeConflictError, KnowledgeNotFoundError, KnowledgeStorage, MCPClientsStorage, MCPServersStorage, METRIC_DISTINCT_COLUMNS, MastraCompositeStore, MemoryStorage, NotificationsStorage, OBSERVATIONAL_MEMORY_TABLE_SCHEMA, ObservabilityStorage, PromptBlocksStorage, RUN_FENCING_TABLE_SCHEMAS, RunFenceConflictError, SchedulesStorage, ScorerDefinitionsStorage, ScoresStorage, SkillsStorage, TABLE_AGENTS, TABLE_AGENT_VERSIONS, TABLE_BACKGROUND_TASKS, TABLE_CHANNEL_CONFIG, TABLE_CHANNEL_INSTALLATIONS, TABLE_CONFIGS, TABLE_DATASETS, TABLE_DATASET_ITEMS, TABLE_DATASET_VERSIONS, TABLE_EXPERIMENTS, TABLE_EXPERIMENT_RESULTS, TABLE_FAVORITES, 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_SCHEMAS, TABLE_SCORERS, TABLE_SCORER_DEFINITIONS, TABLE_SCORER_DEFINITION_VERSIONS, TABLE_SKILLS, TABLE_SKILL_BLOBS, TABLE_SKILL_VERSIONS, TABLE_SPANS, TABLE_THREADS, TABLE_THREAD_STATE, TABLE_TOOL_PROVIDER_CONNECTIONS, TABLE_WORKFLOW_DEFINITIONS, TABLE_WORKFLOW_RUN_OWNERS, TABLE_WORKFLOW_SNAPSHOT, TABLE_WORKSPACES, TABLE_WORKSPACE_VERSIONS, ThreadStateStorage, ToolProviderConnectionsStorage, TraceStatus, UniqueViolationError, WorkflowDefinitionsStorage, WorkflowsStorage, WorkspacesStorage, calculatePagination, canonicalizeKnowledgeScope, createKnowledgeUlid, createStorageErrorId, createVectorErrorId, ensureDate, executeRetentionPrune, getDefaultValue, getSqlType, hasErrorCode, isKnowledgeScopeVisible, isRunFenceConflictError, knowledgeScopeKey, knowledgeSemanticDocumentId, knowledgeSemanticIdempotencyKey, listBranchesArgsSchema, listFeedbackArgsSchema, listLogsArgsSchema, listMetricsArgsSchema, listScoresArgsSchema, listTracesArgsSchema, matchesExpectedWorkflowStatus, matchesRunFence, mergeWorkflowStepResult, normalizePerPage, normalizeScheduleTarget, parseDuration, parseKnowledgeNodeCursor, parseKnowledgeWikilinks, resolveListOrderBy, resolveRetentionTargets, resolveRunFence, retentionCutoffMs, runRetentionBatches, safelyParseJSON, storageMessageMatchesMetadataFilter, toTraceSpans, transformScoreRow, validateStorageMetadataFilter } from "@mastra/core/storage";
 import { parseFieldKey, parseSqlIdentifier } from "@mastra/core/utils";
 import { MastraVector, validateTopK, validateUpsertInput } from "@mastra/core/vector";
 import { Mutex } from "async-mutex";
@@ -3189,9 +3189,10 @@ var PgDB = class extends MastraBase {
 		}
 		await flush();
 	}
-	async insert({ tableName, record }) {
+	/** `client` lets a caller run the insert inside its own transaction. */
+	async insert({ tableName, record }, client = this.client) {
 		try {
-			await this.executeInsert(this.client, {
+			await this.executeInsert(client, {
 				tableName,
 				record
 			});
@@ -4025,13 +4026,15 @@ MIGRATION REQUIRED: Duplicate spans detected in ${duplicateInfo.tableName}\n====
 			indexName: tableName,
 			schemaName: getSchemaName$6(this.schemaName)
 		});
+		const parsedColumn = `"${parseSqlIdentifier(column, "column name")}"`;
 		const sql = `
       DELETE FROM ${fullTableName}
       WHERE ctid IN (
         SELECT ctid FROM ${fullTableName}
-        WHERE ${`"${parseSqlIdentifier(column, "column name")}"`} < $1
+        WHERE ${parsedColumn} < $1
         LIMIT $2
       )
+      AND ${parsedColumn} < $1
     `;
 		return (await this.client.query(sql, [cutoff, limit])).rowCount ?? 0;
 	}
@@ -4494,6 +4497,7 @@ var AgentsPG = class AgentsPG extends AgentsStorage {
 	}
 	async create(input) {
 		const { agent } = input;
+		let draftInserted = false;
 		try {
 			const agentsTable = getTableName$5({
 				indexName: TABLE_AGENTS,
@@ -4520,6 +4524,7 @@ var AgentsPG = class AgentsPG extends AgentsStorage {
 				nowIso,
 				nowIso
 			]);
+			draftInserted = true;
 			const { id: _id, authorId: _authorId, visibility: _visibility, metadata: _metadata, ...snapshotConfig } = agent;
 			const versionId = crypto.randomUUID();
 			await this.createVersion({
@@ -4542,14 +4547,14 @@ var AgentsPG = class AgentsPG extends AgentsStorage {
 				updatedAt: now
 			};
 		} catch (error) {
-			if (error instanceof MastraError) throw error;
-			try {
+			if (draftInserted) try {
 				const agentsTable = getTableName$5({
 					indexName: TABLE_AGENTS,
 					schemaName: getSchemaName$5(this.#schema)
 				});
 				await this.#db.client.none(`DELETE FROM ${agentsTable} WHERE id = $1 AND status = 'draft' AND "activeVersionId" IS NULL`, [agent.id]);
 			} catch {}
+			if (error instanceof MastraError) throw error;
 			throw new MastraError({
 				id: createStorageErrorId("PG", "CREATE_AGENT", "FAILED"),
 				domain: ErrorDomain.STORAGE,
@@ -10545,6 +10550,7 @@ var MCPClientsPG = class MCPClientsPG extends MCPClientsStorage {
 	}
 	async create(input) {
 		const { mcpClient } = input;
+		let draftInserted = false;
 		try {
 			const tableName = getTableName$5({
 				indexName: TABLE_MCP_CLIENTS,
@@ -10567,6 +10573,7 @@ var MCPClientsPG = class MCPClientsPG extends MCPClientsStorage {
 				nowIso,
 				nowIso
 			]);
+			draftInserted = true;
 			const { id: _id, authorId: _authorId, metadata: _metadata, ...snapshotConfig } = mcpClient;
 			const versionId = crypto.randomUUID();
 			await this.createVersion({
@@ -10587,7 +10594,7 @@ var MCPClientsPG = class MCPClientsPG extends MCPClientsStorage {
 				updatedAt: now
 			};
 		} catch (error) {
-			try {
+			if (draftInserted) try {
 				const tableName = getTableName$5({
 					indexName: TABLE_MCP_CLIENTS,
 					schemaName: getSchemaName$5(this.#schema)
@@ -11113,6 +11120,7 @@ var MCPServersPG = class MCPServersPG extends MCPServersStorage {
 	}
 	async create(input) {
 		const { mcpServer } = input;
+		let draftInserted = false;
 		try {
 			const tableName = getTableName$5({
 				indexName: TABLE_MCP_SERVERS,
@@ -11134,6 +11142,7 @@ var MCPServersPG = class MCPServersPG extends MCPServersStorage {
 				nowIso,
 				nowIso
 			]);
+			draftInserted = true;
 			const { id: _id, authorId: _authorId, metadata: _metadata, ...snapshotConfig } = mcpServer;
 			const versionId = crypto.randomUUID();
 			await this.createVersion({
@@ -11154,7 +11163,7 @@ var MCPServersPG = class MCPServersPG extends MCPServersStorage {
 				updatedAt: now
 			};
 		} catch (error) {
-			try {
+			if (draftInserted) try {
 				const tableName = getTableName$5({
 					indexName: TABLE_MCP_SERVERS,
 					schemaName: getSchemaName$5(this.#schema)
@@ -11578,6 +11587,38 @@ var MCPServersPG = class MCPServersPG extends MCPServersStorage {
 	}
 };
 //#endregion
+//#region src/storage/db/run-fencing.ts
+/**
+* Epoch milliseconds on the database clock. `clock_timestamp()` rather than
+* `now()`: a statement can wait on a row lock, and lease math must use the
+* time it actually runs, not when its transaction started.
+*/
+const DB_NOW_MS = `(EXTRACT(EPOCH FROM clock_timestamp()) * 1000)::bigint`;
+/**
+* The check half of a fenced write; call it first inside the write's
+* transaction. `FOR SHARE` holds the claim row until commit, so a claim that
+* updates the row waits for the write to finish, and every later write sees
+* the new claim.
+*/
+async function assertRunFence(t, claimsTable, fence, operation) {
+	const row = await t.oneOrNone(`SELECT generation, "ownerId" FROM ${claimsTable} WHERE "runId" = $1 FOR SHARE`, [fence.runId]);
+	if (!matchesRunFence(row && {
+		generation: Number(row.generation),
+		ownerId: row.ownerId
+	}, fence)) throw new RunFenceConflictError(fence, operation);
+}
+/**
+* Runs `write` directly when it carries no fence, or in a transaction behind
+* the fence check when it does.
+*/
+function withRunFence(client, claimsTable, fence, operation, write) {
+	if (!fence) return write(client);
+	return client.tx(async (t) => {
+		await assertRunFence(t, claimsTable, fence, operation);
+		return write(t);
+	});
+}
+//#endregion
 //#region src/storage/domains/memory/index.ts
 /**
 * Local constant for the observational memory table name.
@@ -11678,6 +11719,10 @@ var MemoryPG = class MemoryPG extends MemoryStorage {
 	* and are indexed for fast batched deletes. Cascade order is enforced in
 	* `prune()` (children before threads), not here. Observational memory has no
 	* timestamp anchor and is deliberately excluded.
+	*
+	* Run fences anchor on `retiredAt` (epoch ms), which stays NULL until the
+	* execution holding the fence settles, so a running execution's fence is
+	* never pruned.
 	*/
 	static retentionTables = {
 		messages: {
@@ -11694,6 +11739,12 @@ var MemoryPG = class MemoryPG extends MemoryStorage {
 			table: TABLE_THREADS,
 			column: "createdAtZ",
 			indexed: true
+		},
+		runFences: {
+			table: TABLE_MEMORY_RUN_FENCES,
+			column: "retiredAt",
+			indexed: true,
+			anchorType: "epoch-ms"
 		}
 	};
 	#db;
@@ -11757,6 +11808,10 @@ var MemoryPG = class MemoryPG extends MemoryStorage {
 			});
 			await this.#db.createIndexFromStatement("idx_om_lookup_key", `CREATE INDEX IF NOT EXISTS idx_om_lookup_key ON ${omTableName} ("lookupKey")`);
 		}
+		await this.#db.createTable({
+			tableName: TABLE_MEMORY_RUN_FENCES,
+			schema: RUN_FENCING_TABLE_SCHEMAS[TABLE_MEMORY_RUN_FENCES]
+		});
 		await this.createDefaultIndexes();
 		await this.createCustomIndexes();
 	}
@@ -11834,6 +11889,12 @@ var MemoryPG = class MemoryPG extends MemoryStorage {
 			statements.push(`CREATE INDEX IF NOT EXISTS "${idxPrefix}idx_om_lookup_key" ON ${fullOmTableName} ("lookupKey");`);
 		}
 		for (const idx of MemoryPG.getDefaultIndexDefs(schemaPrefix)) statements.push(generateIndexSQL(idx, schemaName));
+		statements.push(generateTableSQL({
+			tableName: TABLE_MEMORY_RUN_FENCES,
+			schema: RUN_FENCING_TABLE_SCHEMAS[TABLE_MEMORY_RUN_FENCES],
+			schemaName,
+			includeAllConstraints: true
+		}));
 		return statements;
 	}
 	/**
@@ -11869,6 +11930,61 @@ var MemoryPG = class MemoryPG extends MemoryStorage {
 		await this.#db.clearTable({ tableName: TABLE_MESSAGES });
 		await this.#db.clearTable({ tableName: TABLE_THREADS });
 		await this.#db.clearTable({ tableName: TABLE_RESOURCES });
+		await this.#db.clearTable({ tableName: TABLE_MEMORY_RUN_FENCES });
+	}
+	supportsRunFencing() {
+		return true;
+	}
+	#runFencesTable() {
+		return getTableName$3({
+			indexName: TABLE_MEMORY_RUN_FENCES,
+			schemaName: getSchemaName$3(this.#schema)
+		});
+	}
+	/** Runs `write` behind the run's fence check when it carries a fence. */
+	#fenced(fence, operation, write) {
+		return withRunFence(this.#db.client, this.#runFencesTable(), resolveRunFence(this, fence), operation, write);
+	}
+	async raiseRunFence(fence) {
+		try {
+			return await this.#db.client.oneOrNone(`INSERT INTO ${this.#runFencesTable()} AS f ("runId", generation, "ownerId")
+         VALUES ($1, $2, $3)
+         ON CONFLICT ("runId") DO UPDATE
+           SET generation = EXCLUDED.generation, "ownerId" = EXCLUDED."ownerId", "retiredAt" = NULL
+         WHERE f.generation < EXCLUDED.generation
+            OR (f.generation = EXCLUDED.generation AND f."ownerId" = EXCLUDED."ownerId")
+         RETURNING "runId"`, [
+				fence.runId,
+				fence.generation,
+				fence.ownerId
+			]) !== null;
+		} catch (error) {
+			throw new MastraError({
+				id: createStorageErrorId("PG", "RAISE_RUN_FENCE", "FAILED"),
+				domain: ErrorDomain.STORAGE,
+				category: ErrorCategory.THIRD_PARTY,
+				details: { runId: fence.runId }
+			}, error);
+		}
+	}
+	async retireRunFence(fence) {
+		try {
+			return await this.#db.client.oneOrNone(`UPDATE ${this.#runFencesTable()}
+         SET "retiredAt" = ${DB_NOW_MS}
+         WHERE "runId" = $1 AND generation = $2 AND "ownerId" = $3
+         RETURNING "runId"`, [
+				fence.runId,
+				fence.generation,
+				fence.ownerId
+			]) !== null;
+		} catch (error) {
+			throw new MastraError({
+				id: createStorageErrorId("PG", "RETIRE_RUN_FENCE", "FAILED"),
+				domain: ErrorDomain.STORAGE,
+				category: ErrorCategory.THIRD_PARTY,
+				details: { runId: fence.runId }
+			}, error);
+		}
 	}
 	/**
 	* Deletes rows older than the configured `maxAge` per table, in bounded,
@@ -11889,7 +12005,8 @@ var MemoryPG = class MemoryPG extends MemoryStorage {
 			order: [
 				"messages",
 				"resources",
-				"threads"
+				"threads",
+				"runFences"
 			]
 		});
 		const results = await runPrune({
@@ -12147,7 +12264,7 @@ var MemoryPG = class MemoryPG extends MemoryStorage {
 			throw mastraError;
 		}
 	}
-	async saveThread({ thread }) {
+	async saveThread({ thread, fence }) {
 		try {
 			const tableName = getTableName$3({
 				indexName: TABLE_THREADS,
@@ -12156,7 +12273,7 @@ var MemoryPG = class MemoryPG extends MemoryStorage {
 			const createdAt = toUtcISOString(thread.createdAt);
 			const updatedAt = toUtcISOString(thread.updatedAt);
 			const metadataJson = thread.metadata ? toPgJson(thread.metadata) : null;
-			await this.#db.client.none(`INSERT INTO ${tableName} (
+			await this.#fenced(fence, "saveThread", (q) => q.none(`INSERT INTO ${tableName} (
           id,
           "resourceId",
           title,
@@ -12182,12 +12299,13 @@ var MemoryPG = class MemoryPG extends MemoryStorage {
 				createdAt,
 				updatedAt,
 				updatedAt
-			]);
+			]));
 			return {
 				...thread,
 				metadata: metadataJson ? JSON.parse(metadataJson) : thread.metadata
 			};
 		} catch (error) {
+			if (isRunFenceConflictError(error)) throw error;
 			throw new MastraError({
 				id: createStorageErrorId("PG", "SAVE_THREAD", "FAILED"),
 				domain: ErrorDomain.STORAGE,
@@ -12196,29 +12314,30 @@ var MemoryPG = class MemoryPG extends MemoryStorage {
 			}, error);
 		}
 	}
-	async updateThread({ id, title, metadata }) {
+	async updateThread({ id, title, metadata, fence }) {
 		const threadTableName = getTableName$3({
 			indexName: TABLE_THREADS,
 			schemaName: getSchemaName$3(this.#schema)
 		});
-		const existingThread = await this.#getThreadById(this.#db.client, { threadId: id });
-		if (!existingThread) throw new MastraError({
-			id: createStorageErrorId("PG", "UPDATE_THREAD", "FAILED"),
-			domain: ErrorDomain.STORAGE,
-			category: ErrorCategory.USER,
-			text: `Thread ${id} not found`,
-			details: {
-				threadId: id,
-				title: title ?? null
-			}
-		});
-		const mergedMetadata = {
-			...existingThread.metadata,
-			...metadata
-		};
 		try {
-			const nowStr = toUtcISOString(/* @__PURE__ */ new Date());
-			const thread = await this.#db.client.one(`UPDATE ${threadTableName}
+			return await this.#fenced(fence, "updateThread", async (q) => {
+				const existingThread = await this.#getThreadById(q, { threadId: id });
+				if (!existingThread) throw new MastraError({
+					id: createStorageErrorId("PG", "UPDATE_THREAD", "FAILED"),
+					domain: ErrorDomain.STORAGE,
+					category: ErrorCategory.USER,
+					text: `Thread ${id} not found`,
+					details: {
+						threadId: id,
+						title: title ?? null
+					}
+				});
+				const mergedMetadata = {
+					...existingThread.metadata,
+					...metadata
+				};
+				const nowStr = toUtcISOString(/* @__PURE__ */ new Date());
+				const thread = await q.one(`UPDATE ${threadTableName}
                     SET
                         title = COALESCE($1, title),
                         metadata = $2,
@@ -12227,21 +12346,23 @@ var MemoryPG = class MemoryPG extends MemoryStorage {
                     WHERE id = $5
                     RETURNING *
                 `, [
-				title ?? null,
-				mergedMetadata,
-				nowStr,
-				nowStr,
-				id
-			]);
-			return {
-				id: thread.id,
-				resourceId: thread.resourceId,
-				title: thread.title,
-				metadata: typeof thread.metadata === "string" ? JSON.parse(thread.metadata) : thread.metadata,
-				createdAt: thread.createdAtZ || thread.createdAt,
-				updatedAt: thread.updatedAtZ || thread.updatedAt
-			};
+					title ?? null,
+					mergedMetadata,
+					nowStr,
+					nowStr,
+					id
+				]);
+				return {
+					id: thread.id,
+					resourceId: thread.resourceId,
+					title: thread.title,
+					metadata: typeof thread.metadata === "string" ? JSON.parse(thread.metadata) : thread.metadata,
+					createdAt: thread.createdAtZ || thread.createdAt,
+					updatedAt: thread.updatedAtZ || thread.updatedAt
+				};
+			});
 		} catch (error) {
+			if (error instanceof MastraError || isRunFenceConflictError(error)) throw error;
 			throw new MastraError({
 				id: createStorageErrorId("PG", "UPDATE_THREAD", "FAILED"),
 				domain: ErrorDomain.STORAGE,
@@ -12794,8 +12915,9 @@ var MemoryPG = class MemoryPG extends MemoryStorage {
 			throw mastraError;
 		}
 	}
-	async saveMessages({ messages }) {
+	async saveMessages({ messages, fence: explicitFence }) {
 		if (messages.length === 0) return { messages: [] };
+		const fence = resolveRunFence(this, explicitFence);
 		const threadId = messages[0]?.threadId;
 		if (!threadId) throw new MastraError({
 			id: createStorageErrorId("PG", "SAVE_MESSAGES", "FAILED"),
@@ -12823,6 +12945,7 @@ var MemoryPG = class MemoryPG extends MemoryStorage {
 			});
 			const messagesToSave = dedupeMessagesForSave(messages);
 			await this.#db.client.tx(async (t) => {
+				if (fence) await assertRunFence(t, this.#runFencesTable(), fence, "saveMessages");
 				for (let offset = 0; offset < messagesToSave.length; offset += MAX_MESSAGES_PER_INSERT) {
 					const batch = messagesToSave.slice(offset, offset + MAX_MESSAGES_PER_INSERT);
 					const values = [];
@@ -12869,7 +12992,7 @@ var MemoryPG = class MemoryPG extends MemoryStorage {
 			});
 			return { messages: new MessageList().add(messagesWithParsedContent, "memory").get.all.db() };
 		} catch (error) {
-			if (error instanceof MastraError) throw error;
+			if (error instanceof MastraError || isRunFenceConflictError(error)) throw error;
 			throw new MastraError({
 				id: createStorageErrorId("PG", "SAVE_MESSAGES", "FAILED"),
 				domain: ErrorDomain.STORAGE,
@@ -12878,8 +13001,9 @@ var MemoryPG = class MemoryPG extends MemoryStorage {
 			}, error);
 		}
 	}
-	async updateMessages({ messages }) {
+	async updateMessages({ messages, fence: explicitFence }) {
 		if (messages.length === 0) return [];
+		const fence = resolveRunFence(this, explicitFence);
 		const messageIds = messages.map((m) => m.id);
 		const selectQuery = `SELECT id, content, role, type, "createdAt", "createdAtZ", thread_id AS "threadId", "resourceId" FROM ${getTableName$3({
 			indexName: TABLE_MESSAGES,
@@ -12895,6 +13019,7 @@ var MemoryPG = class MemoryPG extends MemoryStorage {
 		});
 		const threadIdsToUpdate = /* @__PURE__ */ new Set();
 		await this.#db.client.tx(async (t) => {
+			if (fence) await assertRunFence(t, this.#runFencesTable(), fence, "updateMessages");
 			const queries = [];
 			const columnMapping = { threadId: "thread_id" };
 			for (const existingMessage of existingMessages) {
@@ -12955,8 +13080,9 @@ var MemoryPG = class MemoryPG extends MemoryStorage {
 			return message;
 		});
 	}
-	async deleteMessages(messageIds) {
+	async deleteMessages(messageIds, options) {
 		if (!messageIds || messageIds.length === 0) return;
+		const fence = resolveRunFence(this, options?.fence);
 		try {
 			const messageTableName = getTableName$3({
 				indexName: TABLE_MESSAGES,
@@ -12967,12 +13093,14 @@ var MemoryPG = class MemoryPG extends MemoryStorage {
 				schemaName: getSchemaName$3(this.#schema)
 			});
 			await this.#db.client.tx(async (t) => {
+				if (fence) await assertRunFence(t, this.#runFencesTable(), fence, "deleteMessages");
 				const placeholders = messageIds.map((_, idx) => `$${idx + 1}`).join(",");
 				const threadIds = (await t.manyOrNone(`SELECT DISTINCT thread_id FROM ${messageTableName} WHERE id IN (${placeholders})`, messageIds))?.map((msg) => msg.thread_id).filter(Boolean) || [];
 				await t.none(`DELETE FROM ${messageTableName} WHERE id IN (${placeholders})`, messageIds);
 				if (threadIds.length > 0) await t.none(`UPDATE ${threadTableName} SET "updatedAt" = NOW(), "updatedAtZ" = NOW() WHERE id IN (${inPlaceholders(threadIds.length)})`, threadIds);
 			});
 		} catch (error) {
+			if (isRunFenceConflictError(error)) throw error;
 			throw new MastraError({
 				id: createStorageErrorId("PG", "DELETE_MESSAGES", "FAILED"),
 				domain: ErrorDomain.STORAGE,
@@ -13000,6 +13128,9 @@ var MemoryPG = class MemoryPG extends MemoryStorage {
 		};
 	}
 	async saveResource({ resource }) {
+		return this.#insertResource(this.#db.client, resource);
+	}
+	async #insertResource(q, resource) {
 		const createdAt = toUtcISOString(resource.createdAt);
 		const updatedAt = toUtcISOString(resource.updatedAt);
 		const metadataJson = toPgJson(resource.metadata);
@@ -13011,7 +13142,7 @@ var MemoryPG = class MemoryPG extends MemoryStorage {
 				createdAt,
 				updatedAt
 			}
-		});
+		}, q);
 		return {
 			...resource,
 			metadata: metadataJson ? JSON.parse(metadataJson) : resource.metadata
@@ -13022,9 +13153,11 @@ var MemoryPG = class MemoryPG extends MemoryStorage {
 			indexName: TABLE_RESOURCES,
 			schemaName: getSchemaName$3(this.#schema)
 		});
+		const fence = resolveRunFence(this, void 0);
 		const callbackError = {};
 		try {
 			return await this.#db.client.tx(async (t) => {
+				if (fence) await assertRunFence(t, this.#runFencesTable(), fence, "mergeResourceWorkingMemory");
 				const now = (/* @__PURE__ */ new Date()).toISOString();
 				await t.none(`INSERT INTO ${tableName} (id, metadata, "createdAt", "createdAtZ", "updatedAt", "updatedAtZ")
          VALUES ($1, $2, $3, $4, $5, $6) ON CONFLICT (id) DO NOTHING`, [
@@ -13068,8 +13201,15 @@ var MemoryPG = class MemoryPG extends MemoryStorage {
 			}, error);
 		}
 	}
-	async updateResource({ resourceId, workingMemory, metadata }) {
-		const existingResource = await this.#getResourceById(this.#db.client, resourceId);
+	async updateResource({ resourceId, workingMemory, metadata, fence }) {
+		return this.#fenced(fence, "updateResource", (q) => this.#updateResource(q, {
+			resourceId,
+			workingMemory,
+			metadata
+		}));
+	}
+	async #updateResource(q, { resourceId, workingMemory, metadata }) {
+		const existingResource = await this.#getResourceById(q, resourceId);
 		if (!existingResource) {
 			const newResource = {
 				id: resourceId,
@@ -13078,7 +13218,7 @@ var MemoryPG = class MemoryPG extends MemoryStorage {
 				createdAt: /* @__PURE__ */ new Date(),
 				updatedAt: /* @__PURE__ */ new Date()
 			};
-			return this.saveResource({ resource: newResource });
+			return this.#insertResource(q, newResource);
 		}
 		const updatedResource = {
 			...existingResource,
@@ -13114,7 +13254,7 @@ var MemoryPG = class MemoryPG extends MemoryStorage {
 		updates.push(`"updatedAtZ" = $${paramIndex++}`);
 		values.push(updatedAtStr);
 		values.push(resourceId);
-		await this.#db.client.none(`UPDATE ${tableName} SET ${updates.join(", ")} WHERE id = $${paramIndex}`, values);
+		await q.none(`UPDATE ${tableName} SET ${updates.join(", ")} WHERE id = $${paramIndex}`, values);
 		return metadataJson ? {
 			...updatedResource,
 			metadata: JSON.parse(metadataJson)
@@ -13253,9 +13393,11 @@ var MemoryPG = class MemoryPG extends MemoryStorage {
 	* serializes generation creation across processes instead. It is
 	* transaction-scoped so it also works behind transaction-pooling proxies.
 	*/
-	async #withOMLookupKeyLock(tableName, lookupKey, fn) {
+	async #withOMLookupKeyLock(tableName, lookupKey, fn, fencedOperation) {
+		const fence = fencedOperation ? resolveRunFence(this, void 0) : void 0;
 		return this.#db.client.tx(async (t) => {
 			await t.query("SELECT pg_advisory_xact_lock(hashtextextended($1, 0))", [`${tableName}:${lookupKey}`]);
+			if (fence && fencedOperation) await assertRunFence(t, this.#runFencesTable(), fence, fencedOperation);
 			return fn(t);
 		});
 	}
@@ -13552,7 +13694,7 @@ var MemoryPG = class MemoryPG extends MemoryStorage {
 			const lastObservedAtStr = input.lastObservedAt.toISOString();
 			const nowStr = now.toISOString();
 			const observedMessageIdsJson = input.observedMessageIds ? toPgJson(input.observedMessageIds) : null;
-			if ((await this.#db.client.query(`UPDATE ${tableName} SET
+			if ((await this.#fenced(void 0, "updateActiveObservations", (q) => q.query(`UPDATE ${tableName} SET
           "activeObservations" = $1,
           "lastObservedAt" = $2,
           "lastObservedAtZ" = $3,
@@ -13572,7 +13714,7 @@ var MemoryPG = class MemoryPG extends MemoryStorage {
 				nowStr,
 				nowStr,
 				input.id
-			])).rowCount === 0) throw new MastraError({
+			]))).rowCount === 0) throw new MastraError({
 				id: createStorageErrorId("PG", "UPDATE_ACTIVE_OBSERVATIONS", "NOT_FOUND"),
 				text: `Observational memory record not found: ${input.id}`,
 				domain: ErrorDomain.STORAGE,
@@ -13580,7 +13722,7 @@ var MemoryPG = class MemoryPG extends MemoryStorage {
 				details: { id: input.id }
 			});
 		} catch (error) {
-			if (error instanceof MastraError) throw error;
+			if (error instanceof MastraError || isRunFenceConflictError(error)) throw error;
 			throw new MastraError({
 				id: createStorageErrorId("PG", "UPDATE_ACTIVE_OBSERVATIONS", "FAILED"),
 				domain: ErrorDomain.STORAGE,
@@ -13600,8 +13742,9 @@ var MemoryPG = class MemoryPG extends MemoryStorage {
 				const newer = await this.#getNewerOMGeneration(t, tableName, lookupKey, input.currentRecord.generationCount);
 				if (newer) return newer;
 				return this.#insertReflectionGeneration(t, tableName, lookupKey, input);
-			});
+			}, "createReflectionGeneration");
 		} catch (error) {
+			if (isRunFenceConflictError(error)) throw error;
 			throw new MastraError({
 				id: createStorageErrorId("PG", "CREATE_REFLECTION_GENERATION", "FAILED"),
 				domain: ErrorDomain.STORAGE,
@@ -13961,7 +14104,7 @@ var MemoryPG = class MemoryPG extends MemoryStorage {
 				extractionFailures: input.chunk.extractionFailures
 			};
 			const lastBufferedAtTime = input.lastBufferedAtTime ? input.lastBufferedAtTime.toISOString() : null;
-			if ((await this.#db.client.query(`UPDATE ${tableName} SET
+			if ((await this.#fenced(void 0, "updateBufferedObservations", (q) => q.query(`UPDATE ${tableName} SET
           "bufferedObservationChunks" = CASE
             WHEN EXISTS (
               SELECT 1
@@ -13980,7 +14123,7 @@ var MemoryPG = class MemoryPG extends MemoryStorage {
 				nowStr,
 				nowStr,
 				input.id
-			])).rowCount === 0) throw new MastraError({
+			]))).rowCount === 0) throw new MastraError({
 				id: createStorageErrorId("PG", "UPDATE_BUFFERED_OBSERVATIONS", "NOT_FOUND"),
 				text: `Observational memory record not found: ${input.id}`,
 				domain: ErrorDomain.STORAGE,
@@ -13988,7 +14131,7 @@ var MemoryPG = class MemoryPG extends MemoryStorage {
 				details: { id: input.id }
 			});
 		} catch (error) {
-			if (error instanceof MastraError) throw error;
+			if (error instanceof MastraError || isRunFenceConflictError(error)) throw error;
 			throw new MastraError({
 				id: createStorageErrorId("PG", "UPDATE_BUFFERED_OBSERVATIONS", "FAILED"),
 				domain: ErrorDomain.STORAGE,
@@ -14070,7 +14213,7 @@ var MemoryPG = class MemoryPG extends MemoryStorage {
 			const lastObservedAt = input.lastObservedAt ?? (latestChunk?.lastObservedAt ? new Date(latestChunk.lastObservedAt) : /* @__PURE__ */ new Date());
 			const lastObservedAtStr = lastObservedAt.toISOString();
 			const boundary = `\n\n--- message boundary (${lastObservedAt.toISOString()}) ---\n\n`;
-			if ((await this.#db.client.query(`UPDATE ${tableName} SET
+			if ((await this.#fenced(void 0, "swapBufferedToActive", (q) => q.query(`UPDATE ${tableName} SET
           "activeObservations" = CASE
             WHEN "activeObservations" IS NOT NULL AND "activeObservations" != ''
             THEN "activeObservations" || $10 || $1
@@ -14096,7 +14239,7 @@ var MemoryPG = class MemoryPG extends MemoryStorage {
 				nowStr,
 				input.id,
 				boundary
-			])).rowCount === 0) return {
+			]))).rowCount === 0) return {
 				chunksActivated: 0,
 				messageTokensActivated: 0,
 				observationTokensActivated: 0,
@@ -14124,7 +14267,7 @@ var MemoryPG = class MemoryPG extends MemoryStorage {
 				currentTask: latestChunkHints?.currentTask ?? void 0
 			};
 		} catch (error) {
-			if (error instanceof MastraError) throw error;
+			if (error instanceof MastraError || isRunFenceConflictError(error)) throw error;
 			throw new MastraError({
 				id: createStorageErrorId("PG", "SWAP_BUFFERED_TO_ACTIVE", "FAILED"),
 				domain: ErrorDomain.STORAGE,
@@ -14140,7 +14283,7 @@ var MemoryPG = class MemoryPG extends MemoryStorage {
 				schemaName: getSchemaName$3(this.#schema)
 			});
 			const nowStr = (/* @__PURE__ */ new Date()).toISOString();
-			if ((await this.#db.client.query(`UPDATE ${tableName} SET
+			if ((await this.#fenced(void 0, "updateBufferedReflection", (q) => q.query(`UPDATE ${tableName} SET
           "bufferedReflection" = CASE 
             WHEN "bufferedReflection" IS NOT NULL AND "bufferedReflection" != '' 
             THEN "bufferedReflection" || E'\\n\\n' || $1
@@ -14159,7 +14302,7 @@ var MemoryPG = class MemoryPG extends MemoryStorage {
 				nowStr,
 				nowStr,
 				input.id
-			])).rowCount === 0) throw new MastraError({
+			]))).rowCount === 0) throw new MastraError({
 				id: createStorageErrorId("PG", "UPDATE_BUFFERED_REFLECTION", "NOT_FOUND"),
 				text: `Observational memory record not found: ${input.id}`,
 				domain: ErrorDomain.STORAGE,
@@ -14167,7 +14310,7 @@ var MemoryPG = class MemoryPG extends MemoryStorage {
 				details: { id: input.id }
 			});
 		} catch (error) {
-			if (error instanceof MastraError) throw error;
+			if (error instanceof MastraError || isRunFenceConflictError(error)) throw error;
 			throw new MastraError({
 				id: createStorageErrorId("PG", "UPDATE_BUFFERED_REFLECTION", "FAILED"),
 				domain: ErrorDomain.STORAGE,
@@ -14224,9 +14367,9 @@ var MemoryPG = class MemoryPG extends MemoryStorage {
 					input.currentRecord.id
 				]);
 				return newRecord;
-			});
+			}, "swapBufferedReflectionToActive");
 		} catch (error) {
-			if (error instanceof MastraError) throw error;
+			if (error instanceof MastraError || isRunFenceConflictError(error)) throw error;
 			throw new MastraError({
 				id: createStorageErrorId("PG", "SWAP_BUFFERED_REFLECTION_TO_ACTIVE", "FAILED"),
 				domain: ErrorDomain.STORAGE,
@@ -21222,6 +21365,7 @@ var PromptBlocksPG = class PromptBlocksPG extends PromptBlocksStorage {
 	}
 	async create(input) {
 		const { promptBlock } = input;
+		let draftInserted = false;
 		try {
 			const tableName = getTableName$5({
 				indexName: TABLE_PROMPT_BLOCKS,
@@ -21243,6 +21387,7 @@ var PromptBlocksPG = class PromptBlocksPG extends PromptBlocksStorage {
 				nowIso,
 				nowIso
 			]);
+			draftInserted = true;
 			const { id: _id, authorId: _authorId, metadata: _metadata, ...snapshotConfig } = promptBlock;
 			const versionId = crypto.randomUUID();
 			await this.createVersion({
@@ -21263,14 +21408,14 @@ var PromptBlocksPG = class PromptBlocksPG extends PromptBlocksStorage {
 				updatedAt: now
 			};
 		} catch (error) {
-			if (error instanceof MastraError) throw error;
-			try {
+			if (draftInserted) try {
 				const tableName = getTableName$5({
 					indexName: TABLE_PROMPT_BLOCKS,
 					schemaName: getSchemaName$5(this.#schema)
 				});
 				await this.#db.client.none(`DELETE FROM ${tableName} WHERE id = $1 AND status = 'draft' AND "activeVersionId" IS NULL`, [promptBlock.id]);
 			} catch {}
+			if (error instanceof MastraError) throw error;
 			throw new MastraError({
 				id: createStorageErrorId("PG", "CREATE_PROMPT_BLOCK", "FAILED"),
 				domain: ErrorDomain.STORAGE,
@@ -22166,6 +22311,7 @@ var ScorerDefinitionsPG = class ScorerDefinitionsPG extends ScorerDefinitionsSto
 	}
 	async create(input) {
 		const { scorerDefinition } = input;
+		let draftInserted = false;
 		try {
 			const tableName = getTableName$5({
 				indexName: TABLE_SCORER_DEFINITIONS,
@@ -22189,6 +22335,7 @@ var ScorerDefinitionsPG = class ScorerDefinitionsPG extends ScorerDefinitionsSto
 				nowIso,
 				nowIso
 			]);
+			draftInserted = true;
 			const { id: _id, authorId: _authorId, organizationId: _organizationId, projectId: _projectId, metadata: _metadata, ...snapshotConfig } = scorerDefinition;
 			const versionId = crypto.randomUUID();
 			await this.createVersion({
@@ -22211,7 +22358,7 @@ var ScorerDefinitionsPG = class ScorerDefinitionsPG extends ScorerDefinitionsSto
 				updatedAt: now
 			};
 		} catch (error) {
-			try {
+			if (draftInserted) try {
 				const tableName = getTableName$5({
 					indexName: TABLE_SCORER_DEFINITIONS,
 					schemaName: getSchemaName$5(this.#schema)
@@ -23229,6 +23376,7 @@ var SkillsPG = class SkillsPG extends SkillsStorage {
 	}
 	async create(input) {
 		const { skill } = input;
+		let draftInserted = false;
 		try {
 			const tableName = getTableName$5({
 				indexName: TABLE_SKILLS,
@@ -23252,6 +23400,7 @@ var SkillsPG = class SkillsPG extends SkillsStorage {
 				nowIso,
 				nowIso
 			]);
+			draftInserted = true;
 			const { id: _id, authorId: _authorId, visibility: _visibility, ...snapshotConfig } = skill;
 			const versionId = crypto.randomUUID();
 			await this.createVersion({
@@ -23273,7 +23422,7 @@ var SkillsPG = class SkillsPG extends SkillsStorage {
 				updatedAt: now
 			};
 		} catch (error) {
-			try {
+			if (draftInserted) try {
 				const tableName = getTableName$5({
 					indexName: TABLE_SKILLS,
 					schemaName: getSchemaName$5(this.#schema)
@@ -24418,6 +24567,17 @@ function getTableName({ indexName, schemaName }) {
 	const quotedIndexName = `"${indexName}"`;
 	return schemaName ? `${schemaName}.${quotedIndexName}` : quotedIndexName;
 }
+const RUN_OWNER_COLUMNS = `generation, "ownerId", "leaseExpiresAt", ${DB_NOW_MS} AS "nowMs"`;
+function toRunOwnershipRecord(runId, row) {
+	const leaseExpiresAt = row.leaseExpiresAt === null ? null : Number(row.leaseExpiresAt);
+	return {
+		runId,
+		generation: Number(row.generation),
+		ownerId: row.ownerId,
+		leaseExpiresAt: leaseExpiresAt === null ? null : new Date(leaseExpiresAt),
+		live: leaseExpiresAt !== null && leaseExpiresAt > Number(row.nowMs)
+	};
+}
 /** Base name (before any schema prefix) of the expression index backing the status filter. */
 const WORKFLOW_SNAPSHOT_STATUS_INDEX = "mastra_workflow_snapshot_name_status_createdat_idx";
 /**
@@ -24502,12 +24662,24 @@ var WorkflowsPG = class WorkflowsPG extends WorkflowsStorage {
 	* Workflow run snapshots accumulate as runs execute. Anchored on the
 	* timezone-aware `updatedAtZ` mirror column (last activity) so suspended or
 	* long-running runs are not pruned by start age.
+	*
+	* Run ownership rows are anchored on `updatedAt` (epoch ms), which every
+	* claim, renewal and release refreshes, so a live execution's row is never
+	* older than its renew interval.
 	*/
-	static retentionTables = { workflowSnapshot: {
-		table: TABLE_WORKFLOW_SNAPSHOT,
-		column: "updatedAtZ",
-		indexed: true
-	} };
+	static retentionTables = {
+		workflowSnapshot: {
+			table: TABLE_WORKFLOW_SNAPSHOT,
+			column: "updatedAtZ",
+			indexed: true
+		},
+		runOwnership: {
+			table: TABLE_WORKFLOW_RUN_OWNERS,
+			column: "updatedAt",
+			indexed: true,
+			anchorType: "epoch-ms"
+		}
+	};
 	constructor(config) {
 		super();
 		const { client, readClient, schemaName, skipDefaultIndexes, indexes } = resolvePgConfig(config);
@@ -24564,6 +24736,12 @@ var WorkflowsPG = class WorkflowsPG extends WorkflowsStorage {
 		for (const idx of WorkflowsPG.getDefaultIndexDefs(schemaPrefix)) statements.push(generateIndexSQL(idx, schemaName));
 		statements.push(`${workflowSnapshotStatusIndexSQL(workflowSnapshotStatusIndexName(parsedSchema), schemaName)};`);
 		statements.push(`${workflowSnapshotThreadIdIndexSQL(workflowSnapshotThreadIdIndexName(parsedSchema), schemaName)};`);
+		statements.push(generateTableSQL({
+			tableName: TABLE_WORKFLOW_RUN_OWNERS,
+			schema: RUN_FENCING_TABLE_SCHEMAS[TABLE_WORKFLOW_RUN_OWNERS],
+			schemaName,
+			includeAllConstraints: true
+		}));
 		return statements;
 	}
 	/**
@@ -24607,6 +24785,10 @@ var WorkflowsPG = class WorkflowsPG extends WorkflowsStorage {
 			schema: TABLE_SCHEMAS[TABLE_WORKFLOW_SNAPSHOT],
 			ifNotExists: ["resourceId"]
 		});
+		await this.#db.createTable({
+			tableName: TABLE_WORKFLOW_RUN_OWNERS,
+			schema: RUN_FENCING_TABLE_SCHEMAS[TABLE_WORKFLOW_RUN_OWNERS]
+		});
 		await this.createDefaultIndexes();
 		await this.createCustomIndexes();
 	}
@@ -24634,13 +24816,13 @@ var WorkflowsPG = class WorkflowsPG extends WorkflowsStorage {
 			}
 		}
 	}
-	/** Delete workflow run snapshots older than the `workflowSnapshot` policy's `maxAge`, batched. */
+	/** Delete workflow run snapshots and run ownership rows older than their policy's `maxAge`, batched. */
 	async prune(policies, options) {
 		await this.ensureRetentionIndexes(policies);
 		const targets = resolveTargets({
 			policies,
 			descriptor: WorkflowsPG.retentionTables,
-			order: ["workflowSnapshot"]
+			order: ["workflowSnapshot", "runOwnership"]
 		});
 		return runPrune({
 			db: this.#db,
@@ -24662,10 +24844,126 @@ var WorkflowsPG = class WorkflowsPG extends WorkflowsStorage {
 	}
 	async dangerouslyClearAll() {
 		await this.#db.clearTable({ tableName: TABLE_WORKFLOW_SNAPSHOT });
+		await this.#db.clearTable({ tableName: TABLE_WORKFLOW_RUN_OWNERS });
 	}
-	async updateWorkflowResults({ workflowName, runId, stepId, result, requestContext, state }) {
+	supportsRunFencing() {
+		return true;
+	}
+	#runOwnersTable() {
+		return getTableName({
+			indexName: TABLE_WORKFLOW_RUN_OWNERS,
+			schemaName: getSchemaName(this.#schema)
+		});
+	}
+	async #readRunOwner(q, runId) {
+		const row = await q.oneOrNone(`SELECT ${RUN_OWNER_COLUMNS} FROM ${this.#runOwnersTable()} WHERE "runId" = $1`, [runId]);
+		return row ? toRunOwnershipRecord(runId, row) : null;
+	}
+	#ownershipError(operation, runId, error) {
+		return new MastraError({
+			id: createStorageErrorId("PG", operation, "FAILED"),
+			domain: ErrorDomain.STORAGE,
+			category: ErrorCategory.THIRD_PARTY,
+			details: { runId }
+		}, error);
+	}
+	async claimRunOwnership({ runId, ownerId, leaseMs, force, expectedGeneration }) {
+		const table = this.#runOwnersTable();
+		try {
+			return await this.#db.client.tx(async (t) => {
+				if (!await t.oneOrNone(`SELECT 1 FROM ${table} WHERE "runId" = $1 FOR UPDATE`, [runId])) {
+					if (expectedGeneration !== void 0 && expectedGeneration !== 0) return {
+						acquired: false,
+						record: null
+					};
+					const inserted = await t.oneOrNone(`INSERT INTO ${table} ("runId", generation, "ownerId", "leaseExpiresAt", "updatedAt")
+             VALUES ($1, GREATEST(1, ${DB_NOW_MS}), $2, ${DB_NOW_MS} + $3, ${DB_NOW_MS})
+             ON CONFLICT ("runId") DO NOTHING
+             RETURNING ${RUN_OWNER_COLUMNS}`, [
+						runId,
+						ownerId,
+						leaseMs
+					]);
+					if (inserted) return {
+						acquired: true,
+						record: toRunOwnershipRecord(runId, inserted)
+					};
+				}
+				const claimed = await t.oneOrNone(`UPDATE ${table}
+           SET generation = generation + 1, "ownerId" = $2, "leaseExpiresAt" = ${DB_NOW_MS} + $3,
+               "updatedAt" = ${DB_NOW_MS}
+           WHERE "runId" = $1
+             AND ($4::bigint IS NULL OR generation = $4)
+             AND ($5::boolean OR "leaseExpiresAt" IS NULL OR "leaseExpiresAt" <= ${DB_NOW_MS})
+           RETURNING ${RUN_OWNER_COLUMNS}`, [
+					runId,
+					ownerId,
+					leaseMs,
+					expectedGeneration ?? null,
+					force === true
+				]);
+				if (claimed) return {
+					acquired: true,
+					record: toRunOwnershipRecord(runId, claimed)
+				};
+				return {
+					acquired: false,
+					record: await this.#readRunOwner(t, runId)
+				};
+			});
+		} catch (error) {
+			throw this.#ownershipError("CLAIM_RUN_OWNERSHIP", runId, error);
+		}
+	}
+	async renewRunOwnership({ leaseMs, ...fence }) {
+		try {
+			const renewed = await this.#db.client.oneOrNone(`UPDATE ${this.#runOwnersTable()}
+         SET "leaseExpiresAt" = ${DB_NOW_MS} + $4, "updatedAt" = ${DB_NOW_MS}
+         WHERE "runId" = $1 AND generation = $2 AND "ownerId" = $3 AND "leaseExpiresAt" IS NOT NULL
+         RETURNING ${RUN_OWNER_COLUMNS}`, [
+				fence.runId,
+				fence.generation,
+				fence.ownerId,
+				leaseMs
+			]);
+			if (renewed) return {
+				renewed: true,
+				record: toRunOwnershipRecord(fence.runId, renewed)
+			};
+			return {
+				renewed: false,
+				record: await this.#readRunOwner(this.#db.client, fence.runId)
+			};
+		} catch (error) {
+			throw this.#ownershipError("RENEW_RUN_OWNERSHIP", fence.runId, error);
+		}
+	}
+	async releaseRunOwnership(fence) {
+		try {
+			return await this.#db.client.oneOrNone(`UPDATE ${this.#runOwnersTable()}
+         SET "leaseExpiresAt" = NULL, "updatedAt" = ${DB_NOW_MS}
+         WHERE "runId" = $1 AND generation = $2 AND "ownerId" = $3
+         RETURNING "runId"`, [
+				fence.runId,
+				fence.generation,
+				fence.ownerId
+			]) !== null;
+		} catch (error) {
+			throw this.#ownershipError("RELEASE_RUN_OWNERSHIP", fence.runId, error);
+		}
+	}
+	async getRunOwnership({ runId }) {
+		try {
+			return await this.#readRunOwner(this.#db.client, runId);
+		} catch (error) {
+			throw this.#ownershipError("GET_RUN_OWNERSHIP", runId, error);
+		}
+	}
+	async updateWorkflowResults({ workflowName, runId, stepId, result, requestContext, state, fence: explicitFence }) {
+		const fence = resolveRunFence(this, explicitFence, runId);
 		try {
 			return await this.#db.client.tx(async (t) => {
+				if (fence) await assertRunFence(t, this.#runOwnersTable(), fence, "updateWorkflowResults");
 				const tableName = getTableName({
 					indexName: TABLE_WORKFLOW_SNAPSHOT,
 					schemaName: getSchemaName(this.#schema)
@@ -24715,6 +25013,7 @@ var WorkflowsPG = class WorkflowsPG extends WorkflowsStorage {
 				return snapshot.context;
 			});
 		} catch (error) {
+			if (isRunFenceConflictError(error)) throw error;
 			throw new MastraError({
 				id: createStorageErrorId("PG", "UPDATE_WORKFLOW_RESULTS", "FAILED"),
 				domain: ErrorDomain.STORAGE,
@@ -24727,9 +25026,11 @@ var WorkflowsPG = class WorkflowsPG extends WorkflowsStorage {
 			}, error);
 		}
 	}
-	async updateWorkflowState({ workflowName, runId, opts }) {
+	async updateWorkflowState({ workflowName, runId, opts, fence: explicitFence }) {
+		const fence = resolveRunFence(this, explicitFence, runId);
 		try {
 			return await this.#db.client.tx(async (t) => {
+				if (fence) await assertRunFence(t, this.#runOwnersTable(), fence, "updateWorkflowState");
 				const tableName = getTableName({
 					indexName: TABLE_WORKFLOW_SNAPSHOT,
 					schemaName: getSchemaName(this.#schema)
@@ -24759,6 +25060,7 @@ var WorkflowsPG = class WorkflowsPG extends WorkflowsStorage {
 				return updatedSnapshot;
 			});
 		} catch (error) {
+			if (isRunFenceConflictError(error)) throw error;
 			throw new MastraError({
 				id: createStorageErrorId("PG", "UPDATE_WORKFLOW_STATE", "FAILED"),
 				domain: ErrorDomain.STORAGE,
@@ -24770,13 +25072,13 @@ var WorkflowsPG = class WorkflowsPG extends WorkflowsStorage {
 			}, error);
 		}
 	}
-	async persistWorkflowSnapshot({ workflowName, runId, resourceId, snapshot, createdAt, updatedAt }) {
+	async persistWorkflowSnapshot({ workflowName, runId, resourceId, snapshot, createdAt, updatedAt, fence }) {
 		try {
 			const now = /* @__PURE__ */ new Date();
 			const createdAtValue = createdAt ? createdAt : now;
 			const updatedAtValue = updatedAt ? updatedAt : now;
 			const sanitizedSnapshot = toPgJson(snapshot);
-			await this.#db.client.none(`INSERT INTO ${getTableName({
+			await withRunFence(this.#db.client, this.#runOwnersTable(), resolveRunFence(this, fence, runId), "persistWorkflowSnapshot", (q) => q.none(`INSERT INTO ${getTableName({
 				indexName: TABLE_WORKFLOW_SNAPSHOT,
 				schemaName: getSchemaName(this.#schema)
 			})} AS t
@@ -24792,8 +25094,9 @@ var WorkflowsPG = class WorkflowsPG extends WorkflowsStorage {
 				updatedAtValue,
 				createdAtValue,
 				updatedAtValue
-			]);
+			]));
 		} catch (error) {
+			if (isRunFenceConflictError(error)) throw error;
 			throw new MastraError({
 				id: createStorageErrorId("PG", "PERSIST_WORKFLOW_SNAPSHOT", "FAILED"),
 				domain: ErrorDomain.STORAGE,
@@ -24859,13 +25162,14 @@ var WorkflowsPG = class WorkflowsPG extends WorkflowsStorage {
 			}, error);
 		}
 	}
-	async deleteWorkflowRunById({ runId, workflowName }) {
+	async deleteWorkflowRunById({ runId, workflowName, fence }) {
 		try {
-			await this.#db.client.none(`DELETE FROM ${getTableName({
+			await withRunFence(this.#db.client, this.#runOwnersTable(), resolveRunFence(this, fence, runId), "deleteWorkflowRunById", (q) => q.none(`DELETE FROM ${getTableName({
 				indexName: TABLE_WORKFLOW_SNAPSHOT,
 				schemaName: getSchemaName(this.#schema)
-			})} WHERE run_id = $1 AND workflow_name = $2`, [runId, workflowName]);
+			})} WHERE run_id = $1 AND workflow_name = $2`, [runId, workflowName]));
 		} catch (error) {
+			if (isRunFenceConflictError(error)) throw error;
 			throw new MastraError({
 				id: createStorageErrorId("PG", "DELETE_WORKFLOW_RUN_BY_ID", "FAILED"),
 				domain: ErrorDomain.STORAGE,
@@ -25077,6 +25381,7 @@ var WorkspacesPG = class WorkspacesPG extends WorkspacesStorage {
 	}
 	async create(input) {
 		const { workspace } = input;
+		let draftInserted = false;
 		try {
 			const tableName = getTableName$5({
 				indexName: TABLE_WORKSPACES,
@@ -25098,6 +25403,7 @@ var WorkspacesPG = class WorkspacesPG extends WorkspacesStorage {
 				nowIso,
 				nowIso
 			]);
+			draftInserted = true;
 			const { id: _id, authorId: _authorId, metadata: _metadata, ...snapshotConfig } = workspace;
 			const versionId = crypto.randomUUID();
 			await this.createVersion({
@@ -25118,7 +25424,7 @@ var WorkspacesPG = class WorkspacesPG extends WorkspacesStorage {
 				updatedAt: now
 			};
 		} catch (error) {
-			try {
+			if (draftInserted) try {
 				const tableName = getTableName$5({
 					indexName: TABLE_WORKSPACES,
 					schemaName: getSchemaName$5(this.#schema)