@mastra/pg 1.36.0-alpha.4 → 1.36.0-alpha.6

package/dist/docs/SKILL.md

@@ -3,7 +3,7 @@ name: mastra-pg
 description: Documentation for @mastra/pg. Use when working with @mastra/pg APIs, configuration, or implementation.
 metadata:
   package: "@mastra/pg"
-  version: "1.36.0-alpha.4"
+  version: "1.36.0-alpha.6"
 ---
 
 ## When to use

package/dist/docs/assets/SOURCE_MAP.json

@@ -1,86 +1,86 @@
 {
-  "version": "1.36.0-alpha.4",
+  "version": "1.36.0-alpha.6",
   "package": "@mastra/pg",
   "exports": {
     "AgentsPG": {
       "types": "dist/storage/domains/agents/index.d.ts",
       "implementation": "dist/index.js",
-      "line": 4373
+      "line": 4404
     },
     "BackgroundTasksPG": {
       "types": "dist/storage/domains/background-tasks/index.d.ts",
       "implementation": "dist/index.js",
-      "line": 5304
+      "line": 5335
     },
     "BlobsPG": {
       "types": "dist/storage/domains/blobs/index.d.ts",
       "implementation": "dist/index.js",
-      "line": 5695
+      "line": 5726
     },
     "ChannelsPG": {
       "types": "dist/storage/domains/channels/index.d.ts",
       "implementation": "dist/index.js",
-      "line": 5797
+      "line": 5828
     },
     "DatasetsPG": {
       "types": "dist/storage/domains/datasets/index.d.ts",
       "implementation": "dist/index.js",
-      "line": 6039
+      "line": 6070
     },
     "ExperimentsPG": {
       "types": "dist/storage/domains/experiments/index.d.ts",
       "implementation": "dist/index.js",
-      "line": 7296
+      "line": 7327
     },
     "FavoritesPG": {
       "types": "dist/storage/domains/favorites/index.d.ts",
       "implementation": "dist/index.js",
-      "line": 8268
+      "line": 8299
     },
     "KnowledgePG": {
       "types": "dist/storage/domains/knowledge/index.d.ts",
       "implementation": "dist/index.js",
-      "line": 10146
+      "line": 10177
     },
     "MCPClientsPG": {
       "types": "dist/storage/domains/mcp-clients/index.d.ts",
       "implementation": "dist/index.js",
-      "line": 10303
+      "line": 10334
     },
     "MCPServersPG": {
       "types": "dist/storage/domains/mcp-servers/index.d.ts",
       "implementation": "dist/index.js",
-      "line": 10872
+      "line": 10903
     },
     "MemoryPG": {
       "types": "dist/storage/domains/memory/index.d.ts",
       "implementation": "dist/index.js",
-      "line": 11667
+      "line": 11698
     },
     "NotificationsPG": {
       "types": "dist/storage/domains/notifications/index.d.ts",
       "implementation": "dist/index.js",
-      "line": 14574
+      "line": 14605
     },
     "ObservabilityPG": {
       "types": "dist/storage/domains/observability/index.d.ts",
       "implementation": "dist/index.js",
-      "line": 14996
+      "line": 15027
     },
     "ObservabilityStoragePostgresVNext": {
       "types": "dist/storage/domains/observability/v-next/index.d.ts",
       "implementation": "dist/index.js",
-      "line": 21954
+      "line": 21993
     },
     "PGVECTOR_PROMPT": {
       "types": "dist/vector/prompt.d.ts",
       "implementation": "dist/index.js",
-      "line": 28641
+      "line": 28697
     },
     "PgFactoryStorage": {
       "types": "dist/storage/factory-storage.d.ts",
       "implementation": "dist/index.js",
-      "line": 28127
+      "line": 28177
     },
     "PgVector": {
       "types": "dist/vector/index.d.ts",
@@ -95,67 +95,67 @@
     "PostgresStore": {
       "types": "dist/storage/index.d.ts",
       "implementation": "dist/index.js",
-      "line": 28342
+      "line": 28397
     },
     "PostgresStoreVNext": {
       "types": "dist/storage/index.d.ts",
       "implementation": "dist/index.js",
-      "line": 28560
+      "line": 28616
     },
     "PromptBlocksPG": {
       "types": "dist/storage/domains/prompt-blocks/index.d.ts",
       "implementation": "dist/index.js",
-      "line": 22404
+      "line": 22443
     },
     "SchedulesPG": {
       "types": "dist/storage/domains/schedules/index.d.ts",
       "implementation": "dist/index.js",
-      "line": 23033
+      "line": 23072
     },
     "ScorerDefinitionsPG": {
       "types": "dist/storage/domains/scorer-definitions/index.d.ts",
       "implementation": "dist/index.js",
-      "line": 23363
+      "line": 23402
     },
     "ScoresPG": {
       "types": "dist/storage/domains/scores/index.d.ts",
       "implementation": "dist/index.js",
-      "line": 23986
+      "line": 24025
     },
     "SkillsPG": {
       "types": "dist/storage/domains/skills/index.d.ts",
       "implementation": "dist/index.js",
-      "line": 24429
+      "line": 24468
     },
     "ThreadStatePG": {
       "types": "dist/storage/domains/thread-state/index.d.ts",
       "implementation": "dist/index.js",
-      "line": 25120
+      "line": 25159
     },
     "ToolProviderConnectionsPG": {
       "types": "dist/storage/domains/tool-provider-connections/index.d.ts",
       "implementation": "dist/index.js",
-      "line": 25294
+      "line": 25333
     },
     "WorkflowDefinitionsPG": {
       "types": "dist/storage/domains/workflow-definitions/index.d.ts",
       "implementation": "dist/index.js",
-      "line": 25558
+      "line": 25597
     },
     "WorkflowsPG": {
       "types": "dist/storage/domains/workflows/index.d.ts",
       "implementation": "dist/index.js",
-      "line": 25941
+      "line": 25980
     },
     "WorkspacesPG": {
       "types": "dist/storage/domains/workspaces/index.d.ts",
       "implementation": "dist/index.js",
-      "line": 27227
+      "line": 27266
     },
     "exportSchemas": {
       "types": "dist/storage/index.d.ts",
       "implementation": "dist/index.js",
-      "line": 28309
+      "line": 28364
     }
   },
   "modules": {}

package/dist/index.cjs

@@ -2089,8 +2089,10 @@ function truncateQuery(query, maxLength = 100) {
 */
 var PoolAdapter = class {
 	$pool;
-	constructor($pool) {
+	onRollbackError;
+	constructor($pool, onRollbackError) {
 		this.$pool = $pool;
+		this.onRollbackError = onRollbackError;
 	}
 	connect() {
 		return this.$pool.connect();
@@ -2127,8 +2129,14 @@ var PoolAdapter = class {
 	}
 	async tx(callback) {
 		const client = await this.$pool.connect();
+		let destroyClient = false;
 		try {
-			await client.query("BEGIN");
+			try {
+				await client.query("BEGIN");
+			} catch (error) {
+				destroyClient = true;
+				throw error;
+			}
 			const txClient = new TransactionClient(client);
 			try {
 				const result = await callback(txClient);
@@ -2140,12 +2148,16 @@ var PoolAdapter = class {
 				try {
 					await client.query("ROLLBACK");
 				} catch (rollbackError) {
-					console.error("Transaction rollback failed:", rollbackError);
+					destroyClient = true;
+					try {
+						if (this.onRollbackError) await this.onRollbackError(rollbackError);
+						else console.error("Failed to rollback transaction:", rollbackError);
+					} catch {}
 				}
 				throw error;
 			}
 		} finally {
-			client.release();
+			client.release(destroyClient || void 0);
 		}
 	}
 };
@@ -2252,6 +2264,7 @@ var TransactionClient = class {
 var PinnedClientAdapter = class {
 	$pool;
 	pinnedClient;
+	onRollbackError;
 	/**
 	* Serialization tail. Domain init() methods fire via Promise.all in
 	* MastraCompositeStore.#runInit(), so without our own gate every domain's
@@ -2262,9 +2275,14 @@ var PinnedClientAdapter = class {
 	* the reliance on pg's internal queue.
 	*/
 	#tail = Promise.resolve();
-	constructor($pool, pinnedClient) {
+	#rollbackError;
+	constructor($pool, pinnedClient, onRollbackError) {
 		this.$pool = $pool;
 		this.pinnedClient = pinnedClient;
+		this.onRollbackError = onRollbackError;
+	}
+	get shouldDestroyClient() {
+		return this.#rollbackError !== void 0;
 	}
 	/**
 	* Run `fn` after any previously enqueued work on this pinned client has
@@ -2272,7 +2290,11 @@ var PinnedClientAdapter = class {
 	* resume on the next caller's turn.
 	*/
 	#enqueue(fn) {
-		const next = this.#tail.then(fn, fn);
+		const run = async () => {
+			if (this.shouldDestroyClient) throw this.#rollbackError;
+			return fn();
+		};
+		const next = this.#tail.then(run, run);
 		this.#tail = next.catch(() => void 0);
 		return next;
 	}
@@ -2324,7 +2346,12 @@ var PinnedClientAdapter = class {
 	}
 	tx(callback) {
 		return this.#enqueue(async () => {
-			await this.pinnedClient.query("BEGIN");
+			try {
+				await this.pinnedClient.query("BEGIN");
+			} catch (error) {
+				this.#rollbackError = error;
+				throw error;
+			}
 			const txClient = new TransactionClient(this.pinnedClient);
 			try {
 				const result = await callback(txClient);
@@ -2336,7 +2363,11 @@ var PinnedClientAdapter = class {
 				try {
 					await this.pinnedClient.query("ROLLBACK");
 				} catch (rollbackError) {
-					console.error("Transaction rollback failed:", rollbackError);
+					this.#rollbackError = rollbackError;
+					try {
+						if (this.onRollbackError) await this.onRollbackError(rollbackError);
+						else console.error("Failed to rollback transaction:", rollbackError);
+					} catch {}
 				}
 				throw error;
 			}
@@ -19391,13 +19422,13 @@ const timingFeatures = [
 * condition for rows level with it. Empty values sort last in both directions, so they are past
 * every value and level only with each other.
 */
-function keysetSortBounds(field, direction, sortValue, values) {
+function keysetSortBounds(field, direction, sortValue, values, stringSort = false) {
 	if (sortValue === null) return {
 		past: [],
 		level: `${field} IS NULL`
 	};
 	values.push(sortValue);
-	const parameter = `$${values.length}::${typeof sortValue === "number" ? "float8" : "timestamptz"}`;
+	const parameter = `$${values.length}::${stringSort ? "text COLLATE \"C\"" : typeof sortValue === "number" ? "float8" : "timestamptz"}`;
 	return {
 		past: [`${field} ${direction === "asc" ? ">" : "<"} ${parameter}`, `${field} IS NULL`],
 		level: `${field} = ${parameter}`
@@ -20445,10 +20476,14 @@ function selectedKeys(rows, values) {
 		return `"${field}" ${field === "traceId" || field === "spanId" ? "=" : "IS NOT DISTINCT FROM"} $${values.length}::text`;
 	}).join(" AND ")})`).join(" OR ");
 }
+function sortColumn(plan) {
+	const column = `s."${plan.orderBy.field}"`;
+	return plan.orderBy.field === "model" || plan.orderBy.field === "provider" ? `${column} COLLATE "C"` : column;
+}
 function afterCursor(plan, values) {
 	if (!plan.cursor) return "TRUE";
 	const cursor = plan.cursor;
-	const { past: terms, level } = keysetSortBounds(`s."${plan.orderBy.field}"`, plan.orderBy.direction, cursor.sortValue, values);
+	const { past: terms, level } = keysetSortBounds(sortColumn(plan), plan.orderBy.direction, cursor.sortValue, values, plan.orderBy.field === "model" || plan.orderBy.field === "provider");
 	const equal = [level];
 	for (const key of identityFields) {
 		const ref = `s."${key}" COLLATE "C"`;
@@ -20529,10 +20564,12 @@ function compilePostgresSpanQuery(schema, plan) {
 			"endedAt",
 			"durationMs",
 			"status",
-			"version"
+			"version",
+			"model",
+			"provider"
 		].map((field) => `s."${field}"`).join(", ")}
   FROM selected s WHERE ${after}
-  ORDER BY s."${plan.orderBy.field}" ${plan.orderBy.direction === "asc" ? "ASC" : "DESC"} NULLS LAST, ${identityOrder}
+  ORDER BY ${sortColumn(plan)} ${plan.orderBy.direction === "asc" ? "ASC" : "DESC"} NULLS LAST, ${identityOrder}
   LIMIT $${values.length}`,
 		values
 	};
@@ -20567,6 +20604,7 @@ async function querySpans(client, schema, plan, timeoutMs) {
 		const rows = await tx.any(query.text, query.values);
 		const selected = rows.map(({ version: _version, startedAt, endedAt, timeToFirstChunkMs, timeToFirstTokenMs, outcome, stopReason, ...row }) => ({
 			...row,
+			...(plan.orderBy.field === "model" || plan.orderBy.field === "provider") && { cursorSortValue: row[plan.orderBy.field] },
 			...timeToFirstChunkSupported && { timeToFirstChunkMs },
 			...timeToFirstTokenSupported && { timeToFirstTokenMs },
 			...spanQueryOutcomeFieldsSupported && {
@@ -21962,6 +22000,7 @@ const spanQueryFeatures = typeof _mastra_core_storage.planSpanQuery === "functio
 	"span-query",
 	"span-query-context-fields",
 	..._mastra_core_storage.SPAN_QUERY_ORDER_FIELDS?.includes("durationMs") ? ["span-query-duration-sort"] : [],
+	..._mastra_core_storage.SPAN_QUERY_ORDER_FIELDS?.includes("model") && _mastra_core_storage.SPAN_QUERY_ORDER_FIELDS?.includes("provider") ? ["span-query-model-provider-sort"] : [],
 	..."model" in (_mastra_core_storage.SPAN_QUERY_FIELD_REGISTRY ?? {}) ? ["span-query-model-fields"] : []
 ] : [];
 const spanAggregateFeatures = typeof _mastra_core_storage.planSpanAggregate === "function" ? ["span-aggregate", ..."model" in (_mastra_core_storage.SPAN_AGGREGATE_DIMENSION_REGISTRY ?? {}) ? ["span-aggregate-model-dimensions"] : []] : [];
@@ -28124,18 +28163,29 @@ var PgFactoryStorageOps = class {
 		};
 		if (this.#transactionClient) return run(this.#transactionClient);
 		const client = await this.#queryable.connect();
+		let releaseError;
 		try {
-			await client.query("BEGIN");
+			try {
+				await client.query("BEGIN");
+			} catch (error) {
+				releaseError = error instanceof Error ? error : new Error(String(error));
+				throw error;
+			}
 			try {
 				const result = await run(client);
 				await client.query("COMMIT");
 				return result;
 			} catch (error) {
-				await client.query("ROLLBACK");
+				try {
+					await client.query("ROLLBACK");
+				} catch (rollbackError) {
+					releaseError = rollbackError instanceof Error ? rollbackError : new Error(String(rollbackError));
+					throw new AggregateError([error, rollbackError], "Factory transaction and rollback both failed");
+				}
 				throw error;
 			}
 		} finally {
-			client.release();
+			client.release(releaseError);
 		}
 	}
 };
@@ -28192,7 +28242,12 @@ var PgFactoryStorage = class extends _mastra_core_storage.FactoryStorage {
 			let transactionOpen = false;
 			let releaseError;
 			try {
-				await client.query(options?.isolationLevel === "serializable" ? "BEGIN ISOLATION LEVEL SERIALIZABLE" : "BEGIN");
+				try {
+					await client.query(options?.isolationLevel === "serializable" ? "BEGIN ISOLATION LEVEL SERIALIZABLE" : "BEGIN");
+				} catch (error) {
+					releaseError = error instanceof Error ? error : new Error(String(error));
+					throw error;
+				}
 				transactionOpen = true;
 				const result = await fn(new PgFactoryStorageOps(client, this.#schemas, client));
 				await client.query("COMMIT");
@@ -28394,8 +28449,9 @@ var PostgresStore = class extends _mastra_core_storage.MastraCompositeStore {
 				this.#ownsWritePool = true;
 			}
 			this.#readPool = config.readPool ?? this.#writePool;
-			this.#db = new RoutingDbClient(new PoolAdapter(this.#writePool));
-			this.#readDb = this.#readPool === this.#writePool ? this.#db : new PoolAdapter(this.#readPool);
+			const onRollbackError = (error) => this.logger?.error?.("Transaction rollback failed", { error });
+			this.#db = new RoutingDbClient(new PoolAdapter(this.#writePool, onRollbackError));
+			this.#readDb = this.#readPool === this.#writePool ? this.#db : new PoolAdapter(this.#readPool, onRollbackError);
 			const domainConfig = {
 				client: this.#db,
 				readClient: this.#readDb,
@@ -28464,7 +28520,7 @@ var PostgresStore = class extends _mastra_core_storage.MastraCompositeStore {
 		let pinned;
 		try {
 			pinnedClient = await this.#writePool.connect();
-			pinned = new PinnedClientAdapter(this.#writePool, pinnedClient);
+			pinned = new PinnedClientAdapter(this.#writePool, pinnedClient, (error) => this.logger?.error?.("Transaction rollback failed", { error }));
 			this.#db.pin(pinned);
 			for (let attempt = 1; attempt <= 3; attempt++) {
 				this.#db.setSchemaSnapshot(await loadSchemaSnapshot(pinned, this.schema));
@@ -28489,7 +28545,7 @@ var PostgresStore = class extends _mastra_core_storage.MastraCompositeStore {
 			this.#db.setSchemaSnapshot(null);
 			if (pinnedClient) {
 				this.#db.unpin();
-				pinnedClient.release();
+				pinnedClient.release(pinned?.shouldDestroyClient || void 0);
 			}
 		}
 	}