@mastra/pg 1.36.0-alpha.3 → 1.36.0-alpha.5

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.3"
+  version: "1.36.0-alpha.5"
 ---
 
 ## When to use

package/dist/docs/assets/SOURCE_MAP.json

@@ -1,161 +1,161 @@
 {
-  "version": "1.36.0-alpha.3",
+  "version": "1.36.0-alpha.5",
   "package": "@mastra/pg",
   "exports": {
     "AgentsPG": {
       "types": "dist/storage/domains/agents/index.d.ts",
       "implementation": "dist/index.js",
-      "line": 4371
+      "line": 4404
     },
     "BackgroundTasksPG": {
       "types": "dist/storage/domains/background-tasks/index.d.ts",
       "implementation": "dist/index.js",
-      "line": 5302
+      "line": 5335
     },
     "BlobsPG": {
       "types": "dist/storage/domains/blobs/index.d.ts",
       "implementation": "dist/index.js",
-      "line": 5693
+      "line": 5726
     },
     "ChannelsPG": {
       "types": "dist/storage/domains/channels/index.d.ts",
       "implementation": "dist/index.js",
-      "line": 5795
+      "line": 5828
     },
     "DatasetsPG": {
       "types": "dist/storage/domains/datasets/index.d.ts",
       "implementation": "dist/index.js",
-      "line": 6037
+      "line": 6070
     },
     "ExperimentsPG": {
       "types": "dist/storage/domains/experiments/index.d.ts",
       "implementation": "dist/index.js",
-      "line": 7294
+      "line": 7327
     },
     "FavoritesPG": {
       "types": "dist/storage/domains/favorites/index.d.ts",
       "implementation": "dist/index.js",
-      "line": 8266
+      "line": 8299
     },
     "KnowledgePG": {
       "types": "dist/storage/domains/knowledge/index.d.ts",
       "implementation": "dist/index.js",
-      "line": 10144
+      "line": 10177
     },
     "MCPClientsPG": {
       "types": "dist/storage/domains/mcp-clients/index.d.ts",
       "implementation": "dist/index.js",
-      "line": 10301
+      "line": 10334
     },
     "MCPServersPG": {
       "types": "dist/storage/domains/mcp-servers/index.d.ts",
       "implementation": "dist/index.js",
-      "line": 10870
+      "line": 10903
     },
     "MemoryPG": {
       "types": "dist/storage/domains/memory/index.d.ts",
       "implementation": "dist/index.js",
-      "line": 11665
+      "line": 11698
     },
     "NotificationsPG": {
       "types": "dist/storage/domains/notifications/index.d.ts",
       "implementation": "dist/index.js",
-      "line": 14562
+      "line": 14605
     },
     "ObservabilityPG": {
       "types": "dist/storage/domains/observability/index.d.ts",
       "implementation": "dist/index.js",
-      "line": 14984
+      "line": 15027
     },
     "ObservabilityStoragePostgresVNext": {
       "types": "dist/storage/domains/observability/v-next/index.d.ts",
       "implementation": "dist/index.js",
-      "line": 21941
+      "line": 21985
     },
     "PGVECTOR_PROMPT": {
       "types": "dist/vector/prompt.d.ts",
       "implementation": "dist/index.js",
-      "line": 28617
+      "line": 28689
     },
     "PgFactoryStorage": {
       "types": "dist/storage/factory-storage.d.ts",
       "implementation": "dist/index.js",
-      "line": 28103
+      "line": 28169
     },
     "PgVector": {
       "types": "dist/vector/index.d.ts",
       "implementation": "dist/index.js",
-      "line": 593
+      "line": 591
     },
     "PoolAdapter": {
       "types": "dist/storage/client.d.ts",
       "implementation": "dist/index.js",
-      "line": 2069
+      "line": 2067
     },
     "PostgresStore": {
       "types": "dist/storage/index.d.ts",
       "implementation": "dist/index.js",
-      "line": 28318
+      "line": 28389
     },
     "PostgresStoreVNext": {
       "types": "dist/storage/index.d.ts",
       "implementation": "dist/index.js",
-      "line": 28536
+      "line": 28608
     },
     "PromptBlocksPG": {
       "types": "dist/storage/domains/prompt-blocks/index.d.ts",
       "implementation": "dist/index.js",
-      "line": 22391
+      "line": 22435
     },
     "SchedulesPG": {
       "types": "dist/storage/domains/schedules/index.d.ts",
       "implementation": "dist/index.js",
-      "line": 23018
+      "line": 23064
     },
     "ScorerDefinitionsPG": {
       "types": "dist/storage/domains/scorer-definitions/index.d.ts",
       "implementation": "dist/index.js",
-      "line": 23339
+      "line": 23394
     },
     "ScoresPG": {
       "types": "dist/storage/domains/scores/index.d.ts",
       "implementation": "dist/index.js",
-      "line": 23962
+      "line": 24017
     },
     "SkillsPG": {
       "types": "dist/storage/domains/skills/index.d.ts",
       "implementation": "dist/index.js",
-      "line": 24405
+      "line": 24460
     },
     "ThreadStatePG": {
       "types": "dist/storage/domains/thread-state/index.d.ts",
       "implementation": "dist/index.js",
-      "line": 25096
+      "line": 25151
     },
     "ToolProviderConnectionsPG": {
       "types": "dist/storage/domains/tool-provider-connections/index.d.ts",
       "implementation": "dist/index.js",
-      "line": 25270
+      "line": 25325
     },
     "WorkflowDefinitionsPG": {
       "types": "dist/storage/domains/workflow-definitions/index.d.ts",
       "implementation": "dist/index.js",
-      "line": 25534
+      "line": 25589
     },
     "WorkflowsPG": {
       "types": "dist/storage/domains/workflows/index.d.ts",
       "implementation": "dist/index.js",
-      "line": 25917
+      "line": 25972
     },
     "WorkspacesPG": {
       "types": "dist/storage/domains/workspaces/index.d.ts",
       "implementation": "dist/index.js",
-      "line": 27203
+      "line": 27258
     },
     "exportSchemas": {
       "types": "dist/storage/index.d.ts",
       "implementation": "dist/index.js",
-      "line": 28285
+      "line": 28356
     }
   },
   "modules": {}

package/dist/index.cjs

@@ -273,31 +273,29 @@ function buildElemMatchConditions(value, paramIndex) {
 	if (typeof value !== "object" || Array.isArray(value)) throw new Error("$elemMatch requires an object with conditions");
 	const conditions = [];
 	const values = [];
+	function buildFieldCondition(field, operator, operatorValue) {
+		if (operator === "$not") {
+			const nested = Object.entries(operatorValue);
+			if (nested.length === 0) throw new Error("$not requires at least one operator");
+			return `NOT (COALESCE(${nested.map(([nestedOp, nestedValue]) => buildFieldCondition(field, nestedOp, nestedValue)).join(" AND ")}, FALSE))`;
+		}
+		const operatorFn = FILTER_OPERATORS[operator];
+		if (!operatorFn) throw new Error(`Invalid operator: ${operator}`);
+		const scalarElement = field.startsWith("$");
+		const result = operatorFn(scalarElement ? "" : field, paramIndex + values.length, operatorValue);
+		const sql = result.sql.replaceAll("metadata->>", "elem->>").replaceAll("metadata->", "elem->").replaceAll("metadata#>>", "elem#>>").replaceAll("metadata#>", "elem#>").replaceAll("metadata ?", "elem ?");
+		const elementSql = scalarElement ? sql.replaceAll("elem->>''", "elem #>> '{}'").replaceAll("elem->''", "elem") : sql;
+		if (result.needsValue) {
+			const transformedValue = result.transformValue ? result.transformValue() : operatorValue;
+			if (operator === "$elemMatch") values.push(...transformedValue);
+			else values.push(transformedValue);
+		}
+		return elementSql;
+	}
 	Object.entries(value).forEach(([field, val]) => {
-		const nextParamIndex = paramIndex + values.length;
-		let paramOperator;
-		let paramKey;
-		let paramValue;
-		if (field.startsWith("$")) {
-			paramOperator = field;
-			paramKey = "";
-			paramValue = val;
-		} else if (typeof val === "object" && !Array.isArray(val)) {
-			const [op, opValue] = Object.entries(val || {})[0] || [];
-			paramOperator = op;
-			paramKey = field;
-			paramValue = opValue;
-		} else {
-			paramOperator = "$eq";
-			paramKey = field;
-			paramValue = val;
-		}
-		const operatorFn = FILTER_OPERATORS[paramOperator];
-		if (!operatorFn) throw new Error(`Invalid operator: ${paramOperator}`);
-		const result = operatorFn(paramKey, nextParamIndex, paramValue);
-		const sql = result.sql.replaceAll("metadata->>", "elem->>").replaceAll("metadata->", "elem->").replaceAll("metadata#>>", "elem#>>").replaceAll("metadata#>", "elem#>");
-		conditions.push(sql);
-		if (result.needsValue) values.push(paramValue);
+		const operators = field.startsWith("$") ? [[field, val]] : typeof val === "object" && val !== null && !Array.isArray(val) ? Object.entries(val) : [["$eq", val]];
+		if (operators.length === 0) throw new Error(`$elemMatch field ${field} requires at least one operator`);
+		for (const [operator, operatorValue] of operators) conditions.push(buildFieldCondition(field, operator, operatorValue));
 	});
 	return {
 		sql: conditions.join(" AND "),
@@ -2091,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();
@@ -2129,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);
@@ -2142,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);
 		}
 	}
 };
@@ -2254,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
@@ -2264,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
@@ -2274,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;
 	}
@@ -2326,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);
@@ -2338,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;
 			}
@@ -3428,12 +3457,16 @@ var PgDB = class extends _mastra_core_base.MastraBase {
 				}
 				if (snapshot) {
 					snapshot.tables.add(tableName);
-					if (isNewTable) snapshot.createdTables.add(tableName);
-					const created = this.snapshotColumns(snapshot, tableName);
-					for (const [name, def] of Object.entries(schema)) {
-						const parsedName = (0, _mastra_core_utils.parseSqlIdentifier)(name, "column name");
-						created.add(parsedName);
-						if (def.type === "timestamp") created.add(`${parsedName}Z`);
+					if (isNewTable) {
+						snapshot.createdTables.add(tableName);
+						snapshot.columns.set(tableName, new Set(await this.getTableColumns(tableName)));
+					} else {
+						const created = this.snapshotColumns(snapshot, tableName);
+						for (const [name, def] of Object.entries(schema)) {
+							const parsedName = (0, _mastra_core_utils.parseSqlIdentifier)(name, "column name");
+							created.add(parsedName);
+							if (def.type === "timestamp") created.add(`${parsedName}Z`);
+						}
 					}
 				}
 			}
@@ -11822,15 +11855,25 @@ var MemoryPG = class MemoryPG extends _mastra_core_storage.MemoryStorage {
 	* @param schemaPrefix - Prefix for index names (e.g. "my_schema_" or "")
 	*/
 	static getDefaultIndexDefs(schemaPrefix) {
-		return [{
-			name: `${schemaPrefix}mastra_threads_resourceid_createdat_idx`,
-			table: _mastra_core_storage.TABLE_THREADS,
-			columns: ["resourceId", "createdAt DESC"]
-		}, {
-			name: `${schemaPrefix}mastra_messages_thread_id_createdat_idx`,
-			table: _mastra_core_storage.TABLE_MESSAGES,
-			columns: ["thread_id", "createdAt DESC"]
-		}];
+		return [
+			{
+				name: `${schemaPrefix}mastra_threads_resourceid_createdat_idx`,
+				table: _mastra_core_storage.TABLE_THREADS,
+				columns: ["resourceId", "createdAt DESC"]
+			},
+			{
+				name: `${schemaPrefix}mastra_threads_metadata_gin_idx`,
+				table: _mastra_core_storage.TABLE_THREADS,
+				columns: ["metadata"],
+				method: "gin",
+				opclass: "jsonb_path_ops"
+			},
+			{
+				name: `${schemaPrefix}mastra_messages_thread_id_createdat_idx`,
+				table: _mastra_core_storage.TABLE_MESSAGES,
+				columns: ["thread_id", "createdAt DESC"]
+			}
+		];
 	}
 	/**
 	* Returns all DDL statements for this domain: tables (threads, messages, resources, OM), indexes.
@@ -21949,6 +21992,7 @@ async function dangerouslyClearTracing(client, schema) {
 const spanQueryFeatures = typeof _mastra_core_storage.planSpanQuery === "function" ? [
 	"span-query",
 	"span-query-context-fields",
+	..._mastra_core_storage.SPAN_QUERY_ORDER_FIELDS?.includes("durationMs") ? ["span-query-duration-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"] : []] : [];
@@ -23020,6 +23064,8 @@ function rowToSchedule(row) {
 	if (metadata !== void 0) schedule.metadata = metadata;
 	if (row.owner_type != null) schedule.ownerType = String(row.owner_type);
 	if (row.owner_id != null) schedule.ownerId = String(row.owner_id);
+	if (row.run_at != null) schedule.runAt = Number(row.run_at);
+	if (row.end_at != null) schedule.endAt = Number(row.end_at);
 	return schedule;
 }
 function rowToTrigger(row) {
@@ -23079,6 +23125,11 @@ var SchedulesPG = class SchedulesPG extends _mastra_core_storage.SchedulesStorag
 			tableName: _mastra_core_storage.TABLE_SCHEDULES,
 			schema: _mastra_core_storage.TABLE_SCHEMAS[_mastra_core_storage.TABLE_SCHEDULES]
 		});
+		await this.#db.alterTable({
+			tableName: _mastra_core_storage.TABLE_SCHEDULES,
+			schema: _mastra_core_storage.TABLE_SCHEMAS[_mastra_core_storage.TABLE_SCHEDULES],
+			ifNotExists: ["run_at", "end_at"]
+		});
 		await this.#db.createTable({
 			tableName: _mastra_core_storage.TABLE_SCHEDULE_TRIGGERS,
 			schema: _mastra_core_storage.TABLE_SCHEMAS[_mastra_core_storage.TABLE_SCHEDULE_TRIGGERS]
@@ -23206,7 +23257,9 @@ var SchedulesPG = class SchedulesPG extends _mastra_core_storage.SchedulesStorag
 				updated_at: schedule.updatedAt,
 				metadata: schedule.metadata ?? null,
 				owner_type: schedule.ownerType ?? null,
-				owner_id: schedule.ownerId ?? null
+				owner_id: schedule.ownerId ?? null,
+				run_at: schedule.runAt ?? null,
+				end_at: schedule.endAt ?? null
 			}
 		});
 		return schedule;
@@ -23276,6 +23329,8 @@ var SchedulesPG = class SchedulesPG extends _mastra_core_storage.SchedulesStorag
 		if ("metadata" in patch) push("metadata = ?::jsonb", patch.metadata != null ? toPgJson(patch.metadata) : null);
 		if ("ownerType" in patch) push("owner_type = ?", patch.ownerType ?? null);
 		if ("ownerId" in patch) push("owner_id = ?", patch.ownerId ?? null);
+		if ("runAt" in patch) push("run_at = ?", patch.runAt ?? null);
+		if ("endAt" in patch) push("end_at = ?", patch.endAt ?? null);
 		push("updated_at = ?", Date.now());
 		if (setClauses.length === 1) {
 			const existing = await this.#getSchedule(this.#client, id);
@@ -28100,18 +28155,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);
 		}
 	}
 };
@@ -28168,7 +28234,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");
@@ -28370,8 +28441,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,
@@ -28440,7 +28512,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));
@@ -28465,7 +28537,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);
 			}
 		}
 	}