@mastra/pg 1.33.0-alpha.0 → 1.33.0-alpha.2

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, 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, computeTraceStatus, 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 { 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, assertWorkflowStepGuard, calculatePagination, canonicalizeKnowledgeScope, computeTraceStatus, createKnowledgeUlid, createStorageErrorId, createVectorErrorId, ensureDate, executeRetentionPrune, getDefaultValue, getSqlType, hasErrorCode, isKnowledgeScopeVisible, isRunFenceConflictError, isWorkflowStepConflictError, 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";
@@ -16,7 +16,7 @@ import { MessageList } from "@mastra/core/agent";
 import { saveScorePayloadSchema } from "@mastra/core/evals";
 import { skillSnapshotFieldValuesEqual } from "@mastra/core/storage/domains/skills";
 import { isDeepStrictEqual } from "util";
-//#region src/shared/config.ts
+// src/shared/config.ts
 /**
 * Type guard for pre-configured pg.Pool config
 */
@@ -63,8 +63,7 @@ const validateConfig = (name, config) => {
 		]) if (!config[key] || typeof config[key] !== "string" || config[key].trim() === "") throw new Error(`${name}: ${key} must be provided and cannot be empty. Passing an empty string may cause fallback to local Postgres defaults.`);
 	} else throw new Error(`${name}: invalid config. Provide either {pool}, {connectionString}, {host,port,database,user,password}, or a pg ClientConfig (e.g., Cloud SQL connector with \`stream\`).`);
 };
-//#endregion
-//#region src/shared/pool-config.ts
+// src/shared/pool-config.ts
 /**
 * Builds the `pg.Pool` options for a connection-string based config.
 *
@@ -93,8 +92,40 @@ function buildConnectionStringPoolConfig(config, defaults) {
 		idleTimeoutMillis: config.idleTimeoutMillis ?? defaults.idleTimeoutMillis
 	};
 }
-//#endregion
-//#region src/shared/schema-name.ts
+// src/shared/pool-error-guard.ts
+/**
+* A pool-level `error` listener only covers *idle* clients: pg hands ownership
+* of a client to the borrower for the duration of a checkout and takes its own
+* listener off. So a backend restart or network blip that lands on a client
+* mid-transaction reaches an emitter with no listener, and Node escalates that
+* to an uncaughtException that kills the process — the pool handler logs the
+* idle siblings and the borrowed one still brings everything down.
+*
+* Attach a listener once per physical connection instead. `connect` fires when
+* the pool establishes a client, so the client stays covered for its whole life
+* and there is nothing to remove on release.
+*
+* That listener outlives the checkout, so it also sees the failures the pool is
+* already reporting through its own idle listener. Follow `acquire`/`release`
+* to tell the two apart and stay quiet while the client is idle — otherwise one
+* dropped connection is announced twice, the second time as the wrong thing.
+*
+* The pool still discards the failed connection; this only keeps the failure
+* reportable instead of fatal.
+*/
+function guardCheckedOutClientErrors(pool, warn = console.warn) {
+	const borrowed = /* @__PURE__ */ new WeakSet();
+	pool.on("connect", (client) => {
+		borrowed.add(client);
+		client.on("error", (err) => {
+			if (!borrowed.has(client)) return;
+			warn(`PostgreSQL: client error while checked out (the pool discards this connection): ${err instanceof Error ? err.message : String(err)}`);
+		});
+	});
+	pool.on("acquire", (client) => borrowed.add(client));
+	pool.on("release", (_err, client) => borrowed.delete(client));
+}
+// src/shared/schema-name.ts
 const POSTGRES_IDENTIFIER_MAX_BYTES = 63;
 const UNSAFE_SCHEMA_NAME_CHARS = /["'\\$\u0000-\u001f\u007f]/;
 /**
@@ -121,8 +152,7 @@ function schemaNamePrefix(name) {
 	const sanitized = parseSchemaName(name).replace(/[^A-Za-z0-9_]/g, "_");
 	return /^[0-9]/.test(sanitized) ? `_${sanitized}` : sanitized;
 }
-//#endregion
-//#region src/vector/filter.ts
+// src/vector/filter.ts
 /**
 * Translates MongoDB-style filters to PG compatible filters.
 *
@@ -178,8 +208,7 @@ var PGFilterTranslator = class extends BaseFilterTranslator {
 		return { $regex: flags ? `(?${flags})${pattern}` : pattern };
 	}
 };
-//#endregion
-//#region src/vector/sql-builder.ts
+// src/vector/sql-builder.ts
 const getTextExtractExpr = (key) => {
 	const jsonPathKey = parseJsonPathKey(key);
 	if (!key.includes(".")) return `metadata->>'${jsonPathKey}'`;
@@ -557,8 +586,7 @@ function buildFilterQuery(filter, minScore, topK) {
 		values
 	};
 }
-//#endregion
-//#region src/vector/index.ts
+// src/vector/index.ts
 const DEFAULT_NAMESPACE = "default";
 const MAX_UPSERT_ROWS_PER_STATEMENT = Math.floor(65535 / 4);
 var PgVector = class extends MastraVector {
@@ -625,6 +653,7 @@ var PgVector = class extends MastraVector {
 			};
 			else throw new Error("PgVector: invalid configuration provided");
 			this.pool = new pg$1.Pool(poolConfig);
+			guardCheckedOutClientErrors(this.pool, (message) => this.logger?.warn?.(message));
 			this.pool.on("error", (err) => {
 				this.logger?.warn?.("PgVector: idle pool client error (pool discards the client and reconnects on next checkout)", { err: err instanceof Error ? err.message : err });
 			});
@@ -2023,8 +2052,7 @@ var PgVector = class extends MastraVector {
 		}
 	}
 };
-//#endregion
-//#region src/storage/client.ts
+// src/storage/client.ts
 /**
 * Truncate a query string for error messages.
 */
@@ -2406,8 +2434,7 @@ function buildConstraintName({ baseName, schemaName, maxLength = 63, hashWhenTru
 	}
 	return truncateIdentifier(fullName, maxLength);
 }
-//#endregion
-//#region src/storage/db/pg-errors.ts
+// src/storage/db/pg-errors.ts
 function asPgError(error) {
 	return error ?? {};
 }
@@ -2435,8 +2462,7 @@ function isDuplicateSchemaError(error) {
 	if (code === "23505" && constraint === "pg_namespace_nspname_index") return true;
 	return /schema .* already exists/i.test(message);
 }
-//#endregion
-//#region src/storage/db/sanitize-json.ts
+// src/storage/db/sanitize-json.ts
 const repair = (value) => value.replace(/[\uD800-\uDBFF](?![\uDC00-\uDFFF])|(?<![\uD800-\uDBFF])[\uDC00-\uDFFF]/g, "�").replaceAll("\0", "");
 function repairJson(value) {
 	if (typeof value === "string") return repair(value);
@@ -2462,8 +2488,7 @@ function toPgJson(value) {
 	if (json === void 0 || !/\\u(?:0000|[dD][89a-fA-F][0-9a-fA-F]{2})/.test(json)) return json;
 	return JSON.stringify(repairJson(JSON.parse(json)));
 }
-//#endregion
-//#region src/storage/db/schema-snapshot.ts
+// src/storage/db/schema-snapshot.ts
 /**
 * Returns the snapshot currently installed on `client`, but only when it was
 * taken from the schema the caller operates on. A store configured for another
@@ -2545,8 +2570,7 @@ async function loadSchemaSnapshot(client, schemaName) {
 		invalidIndexes
 	};
 }
-//#endregion
-//#region src/storage/db/index.ts
+// src/storage/db/index.ts
 const POSTGRES_MAX_BIND_PARAMETERS$1 = 65535;
 /**
 * Resolves PgDomainConfig to a database client and schema.
@@ -2589,6 +2613,7 @@ function resolvePgConfig(config) {
 	pool.on("error", (err) => {
 		console.warn(`resolvePgConfig: idle pool client error (pool discards the client and reconnects on next checkout): ${err instanceof Error ? err.message : String(err)}`);
 	});
+	guardCheckedOutClientErrors(pool);
 	const client = new PoolAdapter(pool);
 	return {
 		client,
@@ -4117,8 +4142,7 @@ MIGRATION REQUIRED: Duplicate spans detected in ${duplicateInfo.tableName}\n====
 		await client.none(sql, values);
 	}
 };
-//#endregion
-//#region src/storage/domains/utils.ts
+// src/storage/domains/utils.ts
 function getSchemaName$5(schema) {
 	return schema ? `"${parseSchemaName(schema)}"` : void 0;
 }
@@ -4199,8 +4223,7 @@ function tenancyWhere(filters, startIndex) {
 		nextIndex: idx
 	};
 }
-//#endregion
-//#region src/storage/domains/agents/index.ts
+// src/storage/domains/agents/index.ts
 var AgentsPG = class AgentsPG extends AgentsStorage {
 	#db;
 	#schema;
@@ -5047,8 +5070,7 @@ var AgentsPG = class AgentsPG extends AgentsStorage {
 		};
 	}
 };
-//#endregion
-//#region src/storage/retention.ts
+// src/storage/retention.ts
 function cutoffFor(policy, anchorType, now = Date.now()) {
 	const cutoffMs = retentionCutoffMs(policy, now);
 	return anchorType === "epoch-ms" ? cutoffMs : new Date(cutoffMs);
@@ -5075,8 +5097,7 @@ function resolveTargets({ policies, descriptor, order }) {
 		order
 	});
 }
-//#endregion
-//#region src/storage/domains/background-tasks/index.ts
+// src/storage/domains/background-tasks/index.ts
 function getSchemaName$4(schema) {
 	return schema ? `"${schema}"` : "\"public\"";
 }
@@ -5490,8 +5511,7 @@ var BackgroundTasksPG = class BackgroundTasksPG extends BackgroundTasksStorage {
 		return Number(result?.count ?? 0);
 	}
 };
-//#endregion
-//#region src/storage/domains/blobs/index.ts
+// src/storage/domains/blobs/index.ts
 var BlobsPG = class extends BlobStore {
 	#db;
 	#schema;
@@ -5593,8 +5613,7 @@ var BlobsPG = class extends BlobStore {
 		};
 	}
 };
-//#endregion
-//#region src/storage/domains/channels/index.ts
+// src/storage/domains/channels/index.ts
 var ChannelsPG = class ChannelsPG extends ChannelsStorage {
 	#db;
 	#schema;
@@ -5808,8 +5827,7 @@ var ChannelsPG = class ChannelsPG extends ChannelsStorage {
 		};
 	}
 };
-//#endregion
-//#region src/storage/domains/datasets/index.ts
+// src/storage/domains/datasets/index.ts
 /** Serialize a value for a jsonb column. Returns null for null/undefined. */
 function jsonbArg(value) {
 	return value === void 0 || value === null ? null : toPgJson(value);
@@ -7080,8 +7098,7 @@ var DatasetsPG = class DatasetsPG extends DatasetsStorage {
 		await this.#db.clearTable({ tableName: TABLE_DATASETS });
 	}
 };
-//#endregion
-//#region src/storage/domains/experiments/index.ts
+// src/storage/domains/experiments/index.ts
 const DEFAULT_PRUNE_BATCH_SIZE = 1e3;
 var ExperimentsPG = class ExperimentsPG extends ExperimentsStorage {
 	#db;
@@ -8027,8 +8044,7 @@ var ExperimentsPG = class ExperimentsPG extends ExperimentsStorage {
 		await this.#db.clearTable({ tableName: TABLE_EXPERIMENTS });
 	}
 };
-//#endregion
-//#region src/storage/domains/favorites/index.ts
+// src/storage/domains/favorites/index.ts
 /**
 * Maps a favorite entity type to its parent entity table.
 */
@@ -8300,8 +8316,7 @@ var FavoritesPG = class FavoritesPG extends FavoritesStorage {
 		}
 	}
 };
-//#endregion
-//#region ../../packages/_internal-core/dist/knowledge-compat/index.js
+// ../../packages/_internal-core/dist/knowledge-compat/index.js
 const KNOWLEDGE_V2_CORE_FEATURE = "knowledge-v2";
 /** Hard cap on scope nodes returned by one `listScopeNodes` read. */
 const MAX_KNOWLEDGE_SCOPE_NODES = 1e3;
@@ -8935,8 +8950,7 @@ function createKnowledgeV2CoreLoader(features, loadStorageModule) {
 		return knowledgeV2Core;
 	};
 }
-//#endregion
-//#region src/storage/domains/knowledge/index.ts
+// src/storage/domains/knowledge/index.ts
 const loadKnowledgeV2Core = createKnowledgeV2CoreLoader(coreFeatures, () => import("@mastra/core/storage"));
 async function assertKnowledgeDescriptionWithinBoundCompat(description) {
 	const { assertKnowledgeDescriptionWithinBound } = await loadKnowledgeV2Core();
@@ -10460,8 +10474,7 @@ var KnowledgePG = class extends KnowledgeStorage {
 		});
 	}
 };
-//#endregion
-//#region src/storage/domains/mcp-clients/index.ts
+// src/storage/domains/mcp-clients/index.ts
 const SNAPSHOT_FIELDS$5 = [
 	"name",
 	"description",
@@ -11022,8 +11035,7 @@ var MCPClientsPG = class MCPClientsPG extends MCPClientsStorage {
 		};
 	}
 };
-//#endregion
-//#region src/storage/domains/mcp-servers/index.ts
+// src/storage/domains/mcp-servers/index.ts
 const SNAPSHOT_FIELDS$4 = [
 	"name",
 	"version",
@@ -11605,8 +11617,7 @@ var MCPServersPG = class MCPServersPG extends MCPServersStorage {
 		};
 	}
 };
-//#endregion
-//#region ../../packages/_internal-core/dist/memory/index.js
+// ../../packages/_internal-core/dist/memory/index.js
 /**
 * Shared rules for observational memory lifecycle writes. Every storage adapter applies
 * these inside its own atomic section (lock, transaction, or conditional update), so the
@@ -11691,8 +11702,7 @@ function isAppendOnlySince(storedObservations, snapshotObservations) {
 	const snapshot = snapshotObservations ?? "";
 	return stored === snapshot || stored.startsWith(snapshot);
 }
-//#endregion
-//#region src/storage/db/run-fencing.ts
+// 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
@@ -11724,8 +11734,7 @@ function withRunFence(client, claimsTable, fence, operation, write) {
 		return write(t);
 	});
 }
-//#endregion
-//#region src/storage/domains/memory/index.ts
+// src/storage/domains/memory/index.ts
 /**
 * Local constant for the observational memory table name.
 * Defined locally to avoid a static import that crashes on older @mastra/core
@@ -14557,8 +14566,7 @@ var MemoryPG = class MemoryPG extends MemoryStorage {
 		}
 	}
 };
-//#endregion
-//#region src/storage/domains/notifications/index.ts
+// src/storage/domains/notifications/index.ts
 const COMPOSITE_PRIMARY_KEY$1 = TABLE_CONFIGS[TABLE_NOTIFICATIONS].compositePrimaryKey;
 const statusTimestamp = (status, now) => {
 	if (status === "delivered") return { deliveredAt: now };
@@ -15061,8 +15069,7 @@ var NotificationsPG = class NotificationsPG extends NotificationsStorage {
 		return row ? rowToNotification(row) : void 0;
 	}
 };
-//#endregion
-//#region src/storage/domains/observability/index.ts
+// src/storage/domains/observability/index.ts
 var ObservabilityPG = class ObservabilityPG extends ObservabilityStorage {
 	/**
 	* Retention-eligible tables. The observability domain has a single physical
@@ -15733,8 +15740,7 @@ var ObservabilityPG = class ObservabilityPG extends ObservabilityStorage {
 		}
 	}
 };
-//#endregion
-//#region src/storage/domains/observability/v-next/signal-schema.ts
+// src/storage/domains/observability/v-next/signal-schema.ts
 const CURSOR_ID_COLUMN = {
 	name: "cursorId",
 	type: "bigserial"
@@ -16385,8 +16391,7 @@ const TEXT_ARRAY_COLUMNS = new Set(columnNamesByType(ALL_EVENT_COLUMNS, "text[]"
 const METRIC_TYPED_COLUMNS = typedColumnNames(METRIC_EVENT_COLUMNS);
 const SCORE_TYPED_COLUMNS = typedColumnNames(SCORE_EVENT_COLUMNS);
 const FEEDBACK_TYPED_COLUMNS = typedColumnNames(FEEDBACK_EVENT_COLUMNS);
-//#endregion
-//#region src/storage/domains/observability/v-next/ddl.ts
+// src/storage/domains/observability/v-next/ddl.ts
 /**
 * Raw DDL for Postgres v-next observability tables.
 *
@@ -16783,8 +16788,7 @@ function additiveColumns(schema) {
 function allIndexDDL(schema) {
 	return tableIndexes().map((spec) => indexDDL(schema, spec));
 }
-//#endregion
-//#region src/storage/domains/observability/v-next/discovery.ts
+// src/storage/domains/observability/v-next/discovery.ts
 const DEFAULT_TTL_SECONDS = 300;
 const DEFAULT_LOOKBACK_SECONDS = 2592e3;
 /** All signal tables that contain a column. Used by cross-signal discovery. */
@@ -17019,8 +17023,7 @@ async function getMetricLabelValues(client, schema, args, config) {
 	if (args.limit) filtered = filtered.slice(0, args.limit);
 	return { values: filtered };
 }
-//#endregion
-//#region src/storage/domains/observability/v-next/filters.ts
+// src/storage/domains/observability/v-next/filters.ts
 function newFilterAccumulator(startIndex = 1) {
 	return {
 		conditions: [],
@@ -17095,8 +17098,7 @@ function applySingleOrArrayFilter(acc, column, value, prefix = "") {
 function whereOrEmpty(acc) {
 	return acc.conditions.length ? `WHERE ${acc.conditions.join(" AND ")}` : "";
 }
-//#endregion
-//#region src/storage/domains/observability/v-next/preview-compat.ts
+// src/storage/domains/observability/v-next/preview-compat.ts
 const generator = coreStorage.buildStoredPreview;
 const canGeneratePreviews = typeof generator === "function";
 /** SQL detection only; preview formatting stays in TypeScript. Empty strings are intentional. */
@@ -17153,8 +17155,7 @@ function buildSpanQueryResponse(...args) {
 	}
 	return response;
 }
-//#endregion
-//#region src/storage/domains/observability/v-next/review-status.ts
+// src/storage/domains/observability/v-next/review-status.ts
 const FEEDBACK_REVIEW_STATUSES = ["needs-review", "reviewed"];
 function isFeedbackReviewStatus(value) {
 	return FEEDBACK_REVIEW_STATUSES.some((status) => status === value);
@@ -17178,8 +17179,7 @@ function parseUpdateFeedbackReviewStatusArgs(args) {
 		reviewStatus: args.reviewStatus
 	};
 }
-//#endregion
-//#region src/storage/domains/observability/v-next/sql.ts
+// src/storage/domains/observability/v-next/sql.ts
 /**
 * SQL helpers for the v-next Postgres observability domain.
 *
@@ -17309,8 +17309,7 @@ const SCORE_SELECT_COLUMNS = `
 ${buildSelectColumns(SCORE_EVENT_COLUMNS)}`;
 const FEEDBACK_SELECT_COLUMNS = `
 ${buildSelectColumns(FEEDBACK_EVENT_COLUMNS)}`;
-//#endregion
-//#region src/storage/domains/observability/v-next/helpers.ts
+// src/storage/domains/observability/v-next/helpers.ts
 const PROMOTED_KEYS = /* @__PURE__ */ new Set([
 	"experimentId",
 	"entityType",
@@ -17747,8 +17746,7 @@ function rowToFeedbackRecord(row) {
 		scope: parsedJson(row.scope) ?? void 0
 	};
 }
-//#endregion
-//#region src/storage/domains/observability/v-next/polling.ts
+// src/storage/domains/observability/v-next/polling.ts
 /**
 * Delta-polling helpers for the v-next Postgres observability domain.
 *
@@ -17817,8 +17815,7 @@ function decodeDeltaCursor(cursor) {
 async function readSafeXactHorizon(client) {
 	return validatePgInteger((await client.one(`SELECT pg_snapshot_xmin(pg_current_snapshot())::text AS "xactId"`)).xactId);
 }
-//#endregion
-//#region src/storage/domains/observability/v-next/listing.ts
+// src/storage/domains/observability/v-next/listing.ts
 function responseItems(key, items) {
 	return { [key]: items };
 }
@@ -17905,8 +17902,7 @@ async function listSignalDelta(config) {
 		deltaCursor
 	};
 }
-//#endregion
-//#region src/storage/domains/observability/v-next/olap.ts
+// src/storage/domains/observability/v-next/olap.ts
 /**
 * Shared OLAP helpers for the v-next Postgres observability domain.
 *
@@ -18110,8 +18106,7 @@ const COMPLEX_GROUP_BY_EXCLUDED = /* @__PURE__ */ new Set([
 	"tags",
 	"requestContext"
 ]);
-//#endregion
-//#region src/storage/domains/observability/v-next/feedback.ts
+// src/storage/domains/observability/v-next/feedback.ts
 /**
 * Feedback operations for the v-next Postgres observability domain.
 *
@@ -18349,8 +18344,7 @@ async function getFeedbackPercentiles(client, schema, args) {
   `;
 	return { series: percentileSeriesFromRows(await client.manyOrNone(sql, acc.params), args.percentiles) };
 }
-//#endregion
-//#region src/storage/domains/observability/v-next/logs.ts
+// src/storage/domains/observability/v-next/logs.ts
 /**
 * Log operations for the v-next Postgres observability domain.
 */
@@ -18402,8 +18396,7 @@ async function listLogsDelta(client, table, filters, after, limit) {
 		mapRow: rowToLogRecord
 	});
 }
-//#endregion
-//#region src/storage/domains/observability/v-next/metrics.ts
+// src/storage/domains/observability/v-next/metrics.ts
 /**
 * Metric operations for the v-next Postgres observability domain.
 *
@@ -18640,8 +18633,7 @@ async function getMetricPercentiles(client, schema, args) {
   `;
 	return { series: percentileSeriesFromRows(await client.manyOrNone(sql, acc.params), args.percentiles) };
 }
-//#endregion
-//#region src/storage/domains/observability/v-next/partitioning.ts
+// src/storage/domains/observability/v-next/partitioning.ts
 const DEFAULT_FUTURE_DAYS = 14;
 async function detectTimescale(client) {
 	const row = await client.oneOrNone(`SELECT EXISTS (SELECT 1 FROM pg_extension WHERE extname = 'timescaledb') AS "exists"`);
@@ -18800,8 +18792,7 @@ async function setupPartitioning(client, schema, options = {}) {
 			return "native";
 	}
 }
-//#endregion
-//#region src/storage/domains/observability/v-next/retention.ts
+// src/storage/domains/observability/v-next/retention.ts
 /**
 * Age-based retention for the v-next Postgres observability domain.
 *
@@ -18921,8 +18912,7 @@ async function pruneTimescaleTable({ client, schema, table, cutoff, options }) {
 		done: true
 	};
 }
-//#endregion
-//#region src/storage/domains/observability/v-next/scores.ts
+// src/storage/domains/observability/v-next/scores.ts
 /**
 * Score operations for the v-next Postgres observability domain.
 *
@@ -19201,8 +19191,7 @@ async function getScorePercentiles(client, schema, args) {
   `;
 	return { series: percentileSeriesFromRows(await client.manyOrNone(sql, acc.params), args.percentiles) };
 }
-//#endregion
-//#region src/storage/domains/observability/v-next/trace-query.ts
+// src/storage/domains/observability/v-next/trace-query.ts
 const TRACE_STATUS_SQL = `CASE WHEN r."error" IS NOT NULL THEN 'error' ELSE 'success' END`;
 /** Core versions without the field reject it in strict query results, so it is only returned when core knows it. */
 const timeToFirstChunkSupported = "timeToFirstChunkMs" in coreStorage.TRACE_QUERY_FIELD_REGISTRY.trace;
@@ -20194,8 +20183,7 @@ function compileSpanQueryPredicate(predicate, offset) {
 function compileThreadQueryPredicate(predicate, offset) {
 	return compileScalarPredicate(predicate, Object.fromEntries(Object.keys(coreStorage.THREAD_QUERY_FIELD_REGISTRY).map((field) => [field, `s."${field}"`])), offset);
 }
-//#endregion
-//#region src/storage/domains/observability/v-next/span-query.ts
+// src/storage/domains/observability/v-next/span-query.ts
 const identityFields = [
 	"organizationId",
 	"resourceId",
@@ -20408,8 +20396,7 @@ async function querySpans(client, schema, plan, timeoutMs) {
 		return buildSpanQueryResponse(plan, hydrated, payloads, metrics);
 	}, { repeatableRead: true });
 }
-//#endregion
-//#region src/storage/domains/observability/v-next/thread-query.ts
+// src/storage/domains/observability/v-next/thread-query.ts
 const contextFields = [
 	"resourceId",
 	"userId",
@@ -20640,8 +20627,7 @@ async function queryThreadRows(client, schema, plan, timeoutMs, strategy = "auto
 		}) : null }
 	};
 }
-//#endregion
-//#region src/storage/domains/observability/v-next/trace-aggregate.ts
+// src/storage/domains/observability/v-next/trace-aggregate.ts
 /** Trace-root column expressions (alias `r`) for each groupable dimension; same values `where` sees. */
 const DIMENSION_SQL = {
 	entityType: "r.\"entityType\"",
@@ -20967,8 +20953,7 @@ async function runAggregate(client, plan, compiled, timeoutMs) {
 		truncated
 	};
 }
-//#endregion
-//#region src/storage/domains/observability/v-next/traces.ts
+// src/storage/domains/observability/v-next/traces.ts
 /**
 * Trace / branch list reads for the v-next Postgres observability domain.
 *
@@ -21582,8 +21567,7 @@ async function listBranchesDelta(client, span, filters, after, limit) {
 async function readBranchesStreamHeadCursor(client, span, filters) {
 	return encodeDeltaCursor(await readSafeXactHorizon(client), 0);
 }
-//#endregion
-//#region src/storage/domains/observability/v-next/tracing.ts
+// src/storage/domains/observability/v-next/tracing.ts
 async function writeSpanRows(client, schema, rows) {
 	const deduped = dedupeSpanRows(rows);
 	const insert = buildInsert(schema, TABLE_SPAN_EVENTS, deduped, spanConflictClause(Object.keys(deduped[0] ?? {})));
@@ -21712,8 +21696,7 @@ async function dangerouslyClearTracing(client, schema) {
 	const span = qualifiedTable(schema, TABLE_SPAN_EVENTS);
 	await client.none(`TRUNCATE TABLE ${span} RESTART IDENTITY`);
 }
-//#endregion
-//#region src/storage/domains/observability/v-next/index.ts
+// src/storage/domains/observability/v-next/index.ts
 /**
 * Postgres v-next observability storage domain.
 *
@@ -21801,7 +21784,7 @@ var ObservabilityStoragePostgresVNext = class ObservabilityStoragePostgresVNext
 			else if (await detectPartman(this.#client)) mode = "partman";
 			else mode = "native";
 			const ddlMode = mode === "timescale" ? "timescale" : "partitioned";
-			try {
+			if (!await this.#client.oneOrNone("SELECT 1 FROM information_schema.schemata WHERE schema_name = $1", [this.#schema])) try {
 				await this.#client.none(schemaDDL(this.#schema));
 			} catch (error) {
 				if (!isDuplicateSchemaError(error)) throw error;
@@ -22152,8 +22135,7 @@ var ObservabilityStoragePostgresVNext = class ObservabilityStoragePostgresVNext
 		}
 	}
 };
-//#endregion
-//#region src/storage/domains/prompt-blocks/index.ts
+// src/storage/domains/prompt-blocks/index.ts
 const SNAPSHOT_FIELDS$3 = [
 	"name",
 	"description",
@@ -22730,8 +22712,7 @@ var PromptBlocksPG = class PromptBlocksPG extends PromptBlocksStorage {
 		};
 	}
 };
-//#endregion
-//#region src/storage/domains/schedules/index.ts
+// src/storage/domains/schedules/index.ts
 function getSchemaName$2(schema) {
 	return schema ? `"${schema}"` : "\"public\"";
 }
@@ -23099,8 +23080,7 @@ var SchedulesPG = class SchedulesPG extends SchedulesStorage {
        ${limitClause}`, params)).map(rowToTrigger);
 	}
 };
-//#endregion
-//#region src/storage/domains/scorer-definitions/index.ts
+// src/storage/domains/scorer-definitions/index.ts
 const SNAPSHOT_FIELDS$2 = [
 	"name",
 	"description",
@@ -23700,8 +23680,7 @@ var ScorerDefinitionsPG = class ScorerDefinitionsPG extends ScorerDefinitionsSto
 		};
 	}
 };
-//#endregion
-//#region src/storage/domains/scores/index.ts
+// src/storage/domains/scores/index.ts
 /**
 * Appends multi-tenant scope conditions to a query. Mutates `conditions`/`queryParams`.
 * Returns the next positional parameter index.
@@ -24163,8 +24142,7 @@ var ScoresPG = class ScoresPG extends ScoresStorage {
 		}
 	}
 };
-//#endregion
-//#region src/storage/domains/skills/index.ts
+// src/storage/domains/skills/index.ts
 const SNAPSHOT_FIELDS$1 = [
 	"name",
 	"description",
@@ -24860,8 +24838,7 @@ var SkillsPG = class SkillsPG extends SkillsStorage {
 		};
 	}
 };
-//#endregion
-//#region src/storage/domains/thread-state/index.ts
+// src/storage/domains/thread-state/index.ts
 const COMPOSITE_PRIMARY_KEY = ["threadId", "type"];
 /**
 * PostgreSQL implementation of {@link ThreadStateStorage}.
@@ -25026,8 +25003,7 @@ var ThreadStatePG = class ThreadStatePG extends ThreadStateStorage {
 		}
 	}
 };
-//#endregion
-//#region src/storage/domains/tool-provider-connections/index.ts
+// src/storage/domains/tool-provider-connections/index.ts
 function normaliseScope(raw) {
 	const value = raw == null ? "per-author" : String(raw);
 	if (value === "shared") return "shared";
@@ -25282,8 +25258,7 @@ var ToolProviderConnectionsPG = class ToolProviderConnectionsPG extends ToolProv
 		}
 	}
 };
-//#endregion
-//#region src/storage/domains/workflow-definitions/index.ts
+// src/storage/domains/workflow-definitions/index.ts
 function rowToDefinition(row) {
 	const inputSchema = parseJsonResilient(row.inputSchema);
 	const outputSchema = parseJsonResilient(row.outputSchema);
@@ -25483,8 +25458,7 @@ var WorkflowDefinitionsPG = class WorkflowDefinitionsPG extends WorkflowDefiniti
 		await this.#db.client.none(`DELETE FROM ${tableName} WHERE "id" = $1`, [id]);
 	}
 };
-//#endregion
-//#region src/storage/domains/workflows/index.ts
+// src/storage/domains/workflows/index.ts
 function getSchemaName(schema) {
 	return schema ? `"${schema}"` : "\"public\"";
 }
@@ -25884,7 +25858,10 @@ var WorkflowsPG = class WorkflowsPG extends WorkflowsStorage {
 			throw this.#ownershipError("GET_RUN_OWNERSHIP", runId, error);
 		}
 	}
-	async updateWorkflowResults({ workflowName, runId, stepId, result, requestContext, state, fence: explicitFence }) {
+	supportsWorkflowStepGuards() {
+		return true;
+	}
+	async updateWorkflowResults({ workflowName, runId, stepId, result, requestContext, state, fence: explicitFence, startGuard, completionGuard }) {
 		const fence = resolveRunFence(this, explicitFence, runId);
 		try {
 			return await this.#db.client.tx(async (t) => {
@@ -25894,16 +25871,48 @@ var WorkflowsPG = class WorkflowsPG extends WorkflowsStorage {
 					schemaName: getSchemaName(this.#schema)
 				});
 				const existingSnapshotResult = await t.oneOrNone(`SELECT snapshot FROM ${tableName} WHERE workflow_name = $1 AND run_id = $2 FOR UPDATE`, [workflowName, runId]);
-				if (!existingSnapshotResult) return {};
+				if (!existingSnapshotResult) {
+					assertWorkflowStepGuard({
+						snapshot: null,
+						stepId,
+						startGuard,
+						completionGuard
+					});
+					return {};
+				}
 				const existingSnapshot = existingSnapshotResult.snapshot;
 				const snapshot = typeof existingSnapshot === "string" ? JSON.parse(existingSnapshot) : existingSnapshot;
-				mergeWorkflowStepResult({
+				assertWorkflowStepGuard({
+					snapshot,
+					stepId,
+					startGuard,
+					completionGuard
+				});
+				if (completionGuard?.pruneOnly) snapshot.context[stepId] = {
+					...result,
+					metadata: {
+						...snapshot.context[stepId]?.metadata,
+						...result.metadata,
+						__continuationStarted: true
+					}
+				};
+				else mergeWorkflowStepResult({
 					snapshot,
 					stepId,
 					result,
 					requestContext,
 					state
 				});
+				const previous = startGuard?.previousStep;
+				const previousResult = previous && snapshot.context[previous.stepId];
+				if (previous && previous.stepId !== stepId && previousResult?.status === "success" && previousResult.metadata?.startedBy === previous.eventId) snapshot.context[previous.stepId] = {
+					...previous.result,
+					metadata: {
+						...previousResult.metadata,
+						...previous.result.metadata,
+						__continuationStarted: true
+					}
+				};
 				const now = /* @__PURE__ */ new Date();
 				const sanitizedSnapshot = toPgJson(snapshot);
 				await t.none(`INSERT INTO ${tableName}
@@ -25922,7 +25931,7 @@ var WorkflowsPG = class WorkflowsPG extends WorkflowsStorage {
 				return snapshot.context;
 			});
 		} catch (error) {
-			if (isRunFenceConflictError(error)) throw error;
+			if (isRunFenceConflictError(error) || isWorkflowStepConflictError(error)) throw error;
 			throw new MastraError({
 				id: createStorageErrorId("PG", "UPDATE_WORKFLOW_RESULTS", "FAILED"),
 				domain: ErrorDomain.STORAGE,
@@ -26178,8 +26187,7 @@ var WorkflowsPG = class WorkflowsPG extends WorkflowsStorage {
 		}
 	}
 };
-//#endregion
-//#region src/storage/domains/workspaces/index.ts
+// src/storage/domains/workspaces/index.ts
 const SNAPSHOT_FIELDS = [
 	"name",
 	"description",
@@ -26800,8 +26808,7 @@ var WorkspacesPG = class WorkspacesPG extends WorkspacesStorage {
 		};
 	}
 };
-//#endregion
-//#region src/storage/factory-storage.ts
+// src/storage/factory-storage.ts
 const IDENTIFIER_RE = /^[A-Za-z_][A-Za-z0-9_]*$/;
 const PG_UNIQUE_VIOLATION = "23505";
 const COLUMN_DDL = {
@@ -27093,38 +27100,6 @@ var PgFactoryStorageOps = class {
 * Supports serializable app-table transactions for cross-replica invariant
 * enforcement and `authDatabase()` exposing the shared pool.
 */
-/**
-* A pool-level `error` listener only covers *idle* clients: pg hands ownership
-* of a client to the borrower for the duration of a checkout and takes its own
-* listener off. So a backend restart or network blip that lands on a client
-* mid-transaction reaches an emitter with no listener, and Node escalates that
-* to an uncaughtException that kills the process — the pool handler logs the
-* idle siblings and the borrowed one still brings everything down.
-*
-* Attach a listener once per physical connection instead. `connect` fires when
-* the pool establishes a client, so the client stays covered for its whole life
-* and there is nothing to remove on release.
-*
-* That listener outlives the checkout, so it also sees the failures the pool is
-* already reporting through its own idle listener. Follow `acquire`/`release`
-* to tell the two apart and stay quiet while the client is idle — otherwise one
-* dropped connection is announced twice, the second time as the wrong thing.
-*
-* The pool still discards the failed connection; this only keeps the failure
-* reportable instead of fatal.
-*/
-function guardCheckedOutClientErrors(pool, warn = console.warn) {
-	const borrowed = /* @__PURE__ */ new WeakSet();
-	pool.on("connect", (client) => {
-		borrowed.add(client);
-		client.on("error", (err) => {
-			if (!borrowed.has(client)) return;
-			warn(`PgFactoryStorage: client error while checked out (the pool discards this connection): ${err instanceof Error ? err.message : String(err)}`);
-		});
-	});
-	pool.on("acquire", (client) => borrowed.add(client));
-	pool.on("release", (_err, client) => borrowed.delete(client));
-}
 var PgFactoryStorage = class extends FactoryStorage {
 	ops;
 	#pool;
@@ -27250,8 +27225,7 @@ var PgFactoryStorage = class extends FactoryStorage {
 		}
 	}
 };
-//#endregion
-//#region src/storage/index.ts
+// src/storage/index.ts
 /** Default maximum number of connections in the pool */
 const DEFAULT_MAX_CONNECTIONS = 20;
 /** Default idle timeout in milliseconds */
@@ -27423,6 +27397,7 @@ var PostgresStore = class extends MastraCompositeStore {
 		pool.on("error", (err) => {
 			this.logger?.warn?.("PostgresStore: idle pool client error (pool discards the client and reconnects on next checkout)", { err: err instanceof Error ? err.message : err });
 		});
+		guardCheckedOutClientErrors(pool, (message) => this.logger?.warn?.(message));
 		return pool;
 	}
 	#buildPool(config) {
@@ -27562,9 +27537,12 @@ var PostgresStoreVNext = class extends PostgresStore {
 		const observabilityClient = built.client;
 		this.#observabilityPool = built.pool;
 		this.#ownsObservabilityPool = built.owned;
-		if (built.owned) built.pool.on("error", (err) => {
-			this.logger?.warn?.("PostgresStoreVNext: idle observability pool client error (pool discards the client and reconnects on next checkout)", { err: err instanceof Error ? err.message : err });
-		});
+		if (built.owned) {
+			guardCheckedOutClientErrors(built.pool, (message) => this.logger?.warn?.(message));
+			built.pool.on("error", (err) => {
+				this.logger?.warn?.("PostgresStoreVNext: idle observability pool client error (pool discards the client and reconnects on next checkout)", { err: err instanceof Error ? err.message : err });
+			});
+		}
 		const observability = new ObservabilityStoragePostgresVNext({
 			client: observabilityClient,
 			schemaName: obsConfig.schemaName ?? config.schemaName,
@@ -27618,8 +27596,7 @@ var PostgresStoreVNext = class extends PostgresStore {
 		}
 	}
 };
-//#endregion
-//#region src/vector/prompt.ts
+// src/vector/prompt.ts
 /**
 * Vector store specific prompt that details supported operators and examples.
 * This prompt helps users construct valid filters for PG Vector.
@@ -27721,7 +27698,6 @@ Example Complex Query:
     ]}
   ]
 }`;
-//#endregion
 export { AgentsPG, BackgroundTasksPG, BlobsPG, ChannelsPG, DatasetsPG, ExperimentsPG, FavoritesPG, KnowledgePG, MCPClientsPG, MCPServersPG, MemoryPG, NotificationsPG, ObservabilityPG, ObservabilityStoragePostgresVNext, PGVECTOR_PROMPT, PgFactoryStorage, PgVector, PoolAdapter, PostgresStore, PostgresStoreVNext, PromptBlocksPG, SchedulesPG, ScorerDefinitionsPG, ScoresPG, SkillsPG, ThreadStatePG, ToolProviderConnectionsPG, WorkflowDefinitionsPG, WorkflowsPG, WorkspacesPG, exportSchemas };
 
 //# sourceMappingURL=index.js.map