@mastra/pg 1.22.3-alpha.3 → 1.22.3

package/dist/index.js

@@ -14049,6 +14049,11 @@ const FEEDBACK_EVENT_COLUMNS = [
 		type: "text",
 		nullable: true
 	},
+	{
+		name: "reviewStatus",
+		type: "text",
+		defaultSql: "'needs-review'"
+	},
 	...COMMON_CONTEXT_COLUMNS,
 	{
 		name: "tags",
@@ -14489,6 +14494,10 @@ function additiveColumns(schema) {
 		table: TABLE_SPAN_EVENTS,
 		column: "isPending",
 		ddl: `ALTER TABLE ${qualifiedTable(schema, TABLE_SPAN_EVENTS)} ADD COLUMN IF NOT EXISTS "isPending" boolean NOT NULL DEFAULT false`
+	}, {
+		table: TABLE_FEEDBACK_EVENTS,
+		column: "reviewStatus",
+		ddl: `ALTER TABLE ${qualifiedTable(schema, TABLE_FEEDBACK_EVENTS)} ADD COLUMN IF NOT EXISTS "reviewStatus" text NOT NULL DEFAULT 'needs-review'`
 	}];
 }
 /** Existence probe for an additive column. Params: schema, table, column. */
@@ -14811,6 +14820,31 @@ function whereOrEmpty(acc) {
 	return acc.conditions.length ? `WHERE ${acc.conditions.join(" AND ")}` : "";
 }
 //#endregion
+//#region 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);
+}
+/** Normalize a stored value to a review status, defaulting legacy/unknown values to `needs-review`. */
+function coerceFeedbackReviewStatus(value) {
+	return isFeedbackReviewStatus(value) ? value : "needs-review";
+}
+function parseUpdateFeedbackReviewStatusArgs(args) {
+	const invalid = (text) => new MastraError({
+		id: "OBSERVABILITY_UPDATE_FEEDBACK_REVIEW_STATUS_INVALID_ARGS",
+		domain: ErrorDomain.STORAGE,
+		category: ErrorCategory.USER,
+		text
+	});
+	if (typeof args !== "object" || args === null) throw invalid("args must be an object");
+	if (typeof args.feedbackId !== "string" || args.feedbackId.length === 0) throw invalid("feedbackId is required");
+	if (!isFeedbackReviewStatus(args.reviewStatus)) throw invalid(`reviewStatus must be one of: ${FEEDBACK_REVIEW_STATUSES.join(", ")}`);
+	return {
+		feedbackId: args.feedbackId,
+		reviewStatus: args.reviewStatus
+	};
+}
+//#endregion
 //#region src/storage/domains/observability/v-next/helpers.ts
 const PROMOTED_KEYS = /* @__PURE__ */ new Set([
 	"experimentId",
@@ -15175,6 +15209,7 @@ function feedbackRecordToRow(feedback) {
 		spanId: feedback.spanId ?? null,
 		feedbackUserId: feedback.feedbackUserId ?? null,
 		sourceId: feedback.sourceId ?? null,
+		reviewStatus: feedback.reviewStatus ?? "needs-review",
 		feedbackSource,
 		feedbackType: feedback.feedbackType,
 		valueString: typeof feedback.value === "string" ? feedback.value : null,
@@ -15196,6 +15231,7 @@ function rowToFeedbackRecord(row) {
 		spanId: nullableString(row.spanId),
 		feedbackUserId: nullableString(row.feedbackUserId),
 		sourceId: nullableString(row.sourceId),
+		reviewStatus: coerceFeedbackReviewStatus(row.reviewStatus),
 		feedbackSource,
 		feedbackType: row.feedbackType,
 		value: hasNumber ? Number(row.valueNumber) : nullableString(row.valueString) ?? "",
@@ -15715,6 +15751,10 @@ function applyFeedbackFilters(acc, filters) {
 		acc.conditions.push(`"feedbackUserId" = $${acc.next++}`);
 		acc.params.push(filters.feedbackUserId);
 	}
+	if (filters?.reviewStatus) {
+		acc.conditions.push(`"reviewStatus" = $${acc.next++}`);
+		acc.params.push(filters.reviewStatus);
+	}
 }
 /**
 * OLAP queries take an explicit feedbackType / feedbackSource pair as
@@ -15738,6 +15778,21 @@ async function batchCreateFeedback(client, schema, args) {
 	const insert = buildInsert(schema, TABLE_FEEDBACK_EVENTS, args.feedbacks.map(feedbackRecordToRow));
 	if (insert) await client.query(insert.text, insert.values);
 }
+async function updateFeedbackReviewStatus(client, schema, args) {
+	const { feedbackId, reviewStatus } = parseUpdateFeedbackReviewStatusArgs(args);
+	const row = await client.oneOrNone(`UPDATE ${qualifiedTable(schema, TABLE_FEEDBACK_EVENTS)}
+     SET "reviewStatus" = $2
+     WHERE "feedbackId" = $1
+     RETURNING ${FEEDBACK_SELECT_COLUMNS}`, [feedbackId, reviewStatus]);
+	if (!row) throw new MastraError({
+		id: "OBSERVABILITY_UPDATE_FEEDBACK_REVIEW_STATUS_NOT_FOUND",
+		domain: ErrorDomain.MASTRA_OBSERVABILITY,
+		category: ErrorCategory.USER,
+		text: "Feedback record not found",
+		details: { feedbackId }
+	});
+	return rowToFeedbackRecord(row);
+}
 async function listFeedback(client, schema, args) {
 	const { mode, filters, pagination, orderBy, after, limit } = listFeedbackArgsSchema.parse(args);
 	const table = qualifiedTable(schema, TABLE_FEEDBACK_EVENTS);
@@ -17601,6 +17656,9 @@ var ObservabilityStoragePostgresVNext = class ObservabilityStoragePostgresVNext
 	async listFeedback(args) {
 		return this.#run("LIST_FEEDBACK", () => listFeedback(this.#client, this.#schema, args));
 	}
+	async updateFeedbackReviewStatus(args) {
+		return this.#run("UPDATE_FEEDBACK_REVIEW_STATUS", () => updateFeedbackReviewStatus(this.#client, this.#schema, args), { feedbackId: args.feedbackId });
+	}
 	async getMetricAggregate(args) {
 		return this.#run("GET_METRIC_AGGREGATE", () => getMetricAggregate(this.#client, this.#schema, args));
 	}