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

package/dist/index.js

@@ -18314,6 +18314,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 +22154,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));
 	}
@@ -26989,18 +26997,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 +27019,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