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

package/dist/index.js

@@ -2696,7 +2696,8 @@ async function loadSchemaSnapshot(client, schemaName) {
 		notNullColumns,
 		indexes,
 		replicaIdentityIndexes,
-		primaryKeyIndexes
+		primaryKeyIndexes,
+		createdTables: /* @__PURE__ */ new Set()
 	};
 }
 // src/storage/db/index.ts
@@ -2856,26 +2857,30 @@ function generateTableSQL({ tableName, schema, schemaName, compositePrimaryKey,
             ` : ""}
           `;
 }
+function indexColumnsSQL(columns, opclass) {
+	return columns.map((col) => {
+		const [colName, ...modifiers] = col.includes(" DESC") || col.includes(" ASC") ? col.split(" ") : [col];
+		if (!colName) throw new Error(`Invalid column specification: ${col}`);
+		return [
+			`"${parseSqlIdentifier(colName, "column name")}"`,
+			opclass,
+			...modifiers
+		].filter(Boolean).join(" ");
+	}).join(", ");
+}
 /**
 * Generates a CREATE INDEX SQL statement from index options.
 * Used by exportSchemas to produce index DDL without a database connection.
 */
 function generateIndexSQL(options, schemaName) {
-	const { name, table, columns, unique = false, where, method = "btree" } = options;
+	const { name, table, columns, unique = false, where, method = "btree", opclass } = options;
 	const fullTableName = getTableName$6({
 		indexName: table,
 		schemaName: getSchemaName$6(schemaName)
 	});
 	const uniqueStr = unique ? "UNIQUE " : "";
 	const methodStr = method !== "btree" ? `USING ${method} ` : "";
-	const columnsStr = columns.map((col) => {
-		if (col.includes(" DESC") || col.includes(" ASC")) {
-			const [colName, ...modifiers] = col.split(" ");
-			if (!colName) throw new Error(`Invalid column specification: ${col}`);
-			return `"${parseSqlIdentifier(colName, "column name")}" ${modifiers.join(" ")}`;
-		}
-		return `"${parseSqlIdentifier(col, "column name")}"`;
-	}).join(", ");
+	const columnsStr = indexColumnsSQL(columns, opclass);
 	const whereStr = where ? ` WHERE ${where}` : "";
 	return `CREATE ${uniqueStr}INDEX IF NOT EXISTS ${`"${parseSqlIdentifier(name, "index name")}"`} ON ${fullTableName} ${methodStr}(${columnsStr})${whereStr};`;
 }
@@ -3386,6 +3391,7 @@ var PgDB = class extends MastraBase {
 			if (this.schemaName) await this.setupSchema();
 			const snapshot = this.schemaSnapshot;
 			if (!snapshot || !this.snapshotShowsTableConverged(snapshot, tableName)) {
+				const isNewTable = snapshot ? !snapshot.tables.has(tableName) : false;
 				const sql = generateTableSQL({
 					tableName,
 					schema,
@@ -3399,6 +3405,7 @@ var PgDB = class extends 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 = parseSqlIdentifier(name, "column name");
@@ -3867,18 +3874,9 @@ MIGRATION REQUIRED: Duplicate spans detected in ${duplicateInfo.tableName}\n====
 			const snapshot = this.schemaSnapshot;
 			if (snapshot?.indexes.has(name)) return;
 			const uniqueStr = unique ? "UNIQUE " : "";
-			const concurrentStr = concurrent ? "CONCURRENTLY " : "";
+			const concurrentStr = concurrent && !this.isTableCreatedInThisInit(table) ? "CONCURRENTLY " : "";
 			const methodStr = method !== "btree" ? `USING ${method} ` : "";
-			const columnsStr = columns.map((col) => {
-				if (col.includes(" DESC") || col.includes(" ASC")) {
-					const [colName, ...modifiers] = col.split(" ");
-					if (!colName) throw new Error(`Invalid column specification: ${col}`);
-					const quotedCol = `"${parseSqlIdentifier(colName, "column name")}" ${modifiers.join(" ")}`;
-					return opclass ? `${quotedCol} ${opclass}` : quotedCol;
-				}
-				const quotedCol = `"${parseSqlIdentifier(col, "column name")}"`;
-				return opclass ? `${quotedCol} ${opclass}` : quotedCol;
-			}).join(", ");
+			const columnsStr = indexColumnsSQL(columns, opclass);
 			const whereStr = where ? ` WHERE ${where}` : "";
 			const tablespaceStr = tablespace ? ` TABLESPACE ${tablespace}` : "";
 			let withStr = "";
@@ -3911,6 +3909,13 @@ MIGRATION REQUIRED: Duplicate spans detected in ${duplicateInfo.tableName}\n====
 		if (snapshot?.indexes.has(indexName)) return;
 		if (await buildIndex(this.client, this.schemaName || "public", indexName, () => this.timedIndexBuild(indexName, sql), this.logger)) snapshot?.indexes.add(indexName);
 	}
+	/**
+	* Whether `tableName` was absent when this init's snapshot was read and this
+	* init created it, so its indexes can be built without CONCURRENTLY.
+	*/
+	isTableCreatedInThisInit(tableName) {
+		return this.schemaSnapshot?.createdTables.has(tableName) ?? false;
+	}
 	async timedIndexBuild(indexName, sql) {
 		const startedAt = Date.now();
 		this.logger.info(`Creating index ${indexName}`);
@@ -18314,6 +18319,11 @@ async function batchCreateFeedback(client, schema, args, partitioning) {
 	const rows = [...new Map(args.feedbacks.map((feedback) => [feedback.feedbackId, feedback])).values()].map(feedbackRecordToRow);
 	await insertRows(client, schema, TABLE_FEEDBACK_EVENTS, rows, partitioning, FEEDBACK_UPSERT_CLAUSE);
 }
+async function getFeedbackById(client, schema, feedbackId) {
+	const row = await client.oneOrNone(`SELECT ${FEEDBACK_SELECT_COLUMNS} FROM ${qualifiedTable(schema, TABLE_FEEDBACK_EVENTS)}
+     WHERE "feedbackId" = $1 ORDER BY "cursorId" DESC LIMIT 1`, [feedbackId]);
+	return row ? rowToFeedbackRecord(row) : null;
+}
 async function updateFeedbackReviewStatus(client, schema, args) {
 	const { feedbackId, reviewStatus } = parseUpdateFeedbackReviewStatusArgs(args);
 	const row = await client.oneOrNone(`UPDATE ${qualifiedTable(schema, TABLE_FEEDBACK_EVENTS)}
@@ -22149,6 +22159,9 @@ var ObservabilityStoragePostgresVNext = class ObservabilityStoragePostgresVNext
 	async getScoreById(scoreId) {
 		return this.#run("GET_SCORE_BY_ID", () => getScoreById(this.#readClient, this.#schema, scoreId), { scoreId });
 	}
+	async getFeedbackById(feedbackId) {
+		return this.#run("GET_FEEDBACK_BY_ID", () => getFeedbackById(this.#readClient, this.#schema, feedbackId), { feedbackId });
+	}
 	async listFeedback(args) {
 		return this.#run("LIST_FEEDBACK", () => listFeedback(this.#readClient, this.#schema, args));
 	}
@@ -25933,15 +25946,16 @@ var WorkflowsPG = class WorkflowsPG extends WorkflowsStorage {
 			this.logger?.warn?.(`Failed to create index ${stateBaseIndexName}:`, error);
 		}
 		if (await this.#db.getColumnType(TABLE_WORKFLOW_SNAPSHOT, "snapshot") !== "jsonb") return;
+		const concurrently = !this.#db.isTableCreatedInThisInit(TABLE_WORKFLOW_SNAPSHOT);
 		const indexName = workflowSnapshotStatusIndexName(this.#schema);
 		try {
-			await this.#db.createIndexFromStatement(indexName, workflowSnapshotStatusIndexSQL(indexName, this.#schema, true));
+			await this.#db.createIndexFromStatement(indexName, workflowSnapshotStatusIndexSQL(indexName, this.#schema, concurrently));
 		} catch (error) {
 			this.logger?.warn?.(`Failed to create index ${indexName}:`, error);
 		}
 		const threadIdIndexName = workflowSnapshotThreadIdIndexName(this.#schema);
 		try {
-			await this.#db.createIndexFromStatement(threadIdIndexName, workflowSnapshotThreadIdIndexSQL(threadIdIndexName, this.#schema, true));
+			await this.#db.createIndexFromStatement(threadIdIndexName, workflowSnapshotThreadIdIndexSQL(threadIdIndexName, this.#schema, concurrently));
 		} catch (error) {
 			this.logger?.warn?.(`Failed to create index ${threadIdIndexName}:`, error);
 		}
@@ -26989,18 +27003,9 @@ var WorkflowsPG = class WorkflowsPG extends WorkflowsStorage {
 			}
 			const whereClause = conditions.length > 0 ? `WHERE ${conditions.join(" AND ")}` : "";
 			const v2WhereClause = v2Conditions.length > 0 ? `WHERE ${v2Conditions.join(" AND ")}` : "";
-			let total = 0;
 			const usePagination = typeof perPage === "number" && typeof page === "number";
-			if (usePagination) {
-				const countResult = await this.#db.client.one(`SELECT (SELECT COUNT(*) FROM ${getTableName({
-					indexName: TABLE_WORKFLOW_SNAPSHOT,
-					schemaName: getSchemaName(this.#schema)
-				})} ${whereClause})
-             + (SELECT COUNT(*) FROM ${this.#runsTable()} ${v2WhereClause}) AS count`, values);
-				total = Number(countResult.count);
-			}
 			const normalizedPerPage = usePagination ? normalizePerPage(perPage, Number.MAX_SAFE_INTEGER) : 0;
-			const offset = usePagination ? page * normalizedPerPage : void 0;
+			const offset = usePagination ? page * normalizedPerPage : 0;
 			let v1Snapshot = "snapshot";
 			if (summary) {
 				const snapshotJson = await this.#db.getColumnType(TABLE_WORKFLOW_SNAPSHOT, "snapshot") === "jsonb" ? "snapshot" : `regexp_replace(snapshot::text, '\\\\u(0000|[Dd][89A-Fa-f][0-9A-Fa-f]{2})', '', 'g')::jsonb`;
@@ -27020,6 +27025,18 @@ var WorkflowsPG = class WorkflowsPG extends WorkflowsStorage {
 			] : values;
 			const result = await this.#db.client.manyOrNone(query, queryValues);
 			const runs = await Promise.all((result || []).map((row) => this.#parseRun(this.#db.client, row)));
+			let total = 0;
+			if (usePagination) {
+				total = offset + runs.length;
+				if (runs.length === normalizedPerPage || runs.length === 0 && offset > 0) {
+					const countResult = await this.#db.client.one(`SELECT (SELECT COUNT(*) FROM ${getTableName({
+						indexName: TABLE_WORKFLOW_SNAPSHOT,
+						schemaName: getSchemaName(this.#schema)
+					})} ${whereClause})
+               + (SELECT COUNT(*) FROM ${this.#runsTable()} ${v2WhereClause}) AS count`, values);
+					total = Number(countResult.count);
+				}
+			}
 			return {
 				runs,
 				total: total || runs.length