@mastra/duckdb 1.6.4-alpha.1 → 1.6.4

package/dist/{observability-CRajwPel.cjs → observability-CYOty2wB.cjs}

@@ -279,6 +279,9 @@ CREATE TABLE IF NOT EXISTS feedback_events (
   feedbackUserId VARCHAR,
   sourceId VARCHAR,
 
+  -- Review workflow
+  reviewStatus VARCHAR NOT NULL DEFAULT 'needs-review',
+
   -- Feedback-specific scalars
   source VARCHAR,
   feedbackSource VARCHAR NOT NULL,
@@ -417,6 +420,7 @@ const ALL_MIGRATIONS = [
 	`ALTER TABLE feedback_events ADD COLUMN IF NOT EXISTS scope JSON`,
 	`ALTER TABLE feedback_events ADD COLUMN IF NOT EXISTS source VARCHAR`,
 	`ALTER TABLE feedback_events ADD COLUMN IF NOT EXISTS feedbackSource VARCHAR`,
+	`ALTER TABLE feedback_events ADD COLUMN IF NOT EXISTS reviewStatus VARCHAR DEFAULT 'needs-review'`,
 	`ALTER TABLE feedback_events ALTER COLUMN traceId DROP NOT NULL`
 ];
 //#endregion
@@ -710,6 +714,31 @@ function extendWhereClause(baseClause, extraConditions) {
 	return `${baseClause} AND ${conditions.join(" AND ")}`;
 }
 //#endregion
+//#region src/storage/domains/observability/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 _mastra_core_error.MastraError({
+		id: "OBSERVABILITY_UPDATE_FEEDBACK_REVIEW_STATUS_INVALID_ARGS",
+		domain: _mastra_core_error.ErrorDomain.STORAGE,
+		category: _mastra_core_error.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/feedback.ts
 const FEEDBACK_GROUP_BY_COLUMNS = /* @__PURE__ */ new Set([
 	"timestamp",
@@ -811,7 +840,7 @@ function rowToFeedbackRecord(row) {
 	let value = rawValue;
 	const numValue = Number(rawValue);
 	if (!isNaN(numValue)) value = numValue;
-	return {
+	return _mastra_core_storage.feedbackRecordSchema.parse({
 		feedbackId: row.feedbackId,
 		timestamp: toDate(row.timestamp),
 		traceId: row.traceId ?? null,
@@ -841,6 +870,7 @@ function rowToFeedbackRecord(row) {
 		serviceName: row.serviceName ?? null,
 		feedbackUserId: row.feedbackUserId ?? null,
 		sourceId: row.sourceId ?? null,
+		reviewStatus: coerceFeedbackReviewStatus(row.reviewStatus),
 		source: row.feedbackSource,
 		feedbackSource: row.feedbackSource,
 		feedbackType: row.feedbackType,
@@ -849,7 +879,7 @@ function rowToFeedbackRecord(row) {
 		tags: parseJsonArray(row.tags),
 		metadata: parseJson(row.metadata),
 		scope: parseJson(row.scope)
-	};
+	});
 }
 function getComparisonDateRange$1(comparePeriod, timestamp) {
 	if (!timestamp.start || !timestamp.end) return null;
@@ -884,7 +914,7 @@ async function createFeedback(db, args) {
       feedbackId, timestamp, cursorId, traceId, spanId, experimentId,
       entityType, entityId, entityName, entityVersionId, parentEntityVersionId, parentEntityType, parentEntityId, parentEntityName, rootEntityVersionId, rootEntityType, rootEntityId, rootEntityName,
       userId, organizationId, resourceId, runId, sessionId, threadId, requestId, environment, executionSource, serviceName,
-      feedbackUserId, sourceId, feedbackSource, feedbackType, value, comment, tags, metadata, scope
+      feedbackUserId, sourceId, reviewStatus, feedbackSource, feedbackType, value, comment, tags, metadata, scope
     )
      VALUES (${[
 		v(f.feedbackId),
@@ -917,6 +947,7 @@ async function createFeedback(db, args) {
 		v(f.serviceName ?? null),
 		v(feedbackUserId),
 		v(f.sourceId ?? null),
+		v(f.reviewStatus ?? "needs-review"),
 		v(feedbackSource),
 		v(f.feedbackType),
 		v(String(f.value)),
@@ -965,6 +996,7 @@ async function batchCreateFeedback(db, args) {
 			v(legacyFeedback.serviceName ?? null),
 			v(feedbackUserId),
 			v(legacyFeedback.sourceId ?? null),
+			v(legacyFeedback.reviewStatus ?? "needs-review"),
 			v(feedbackSource),
 			v(legacyFeedback.feedbackType),
 			v(String(legacyFeedback.value)),
@@ -978,11 +1010,25 @@ async function batchCreateFeedback(db, args) {
       feedbackId, timestamp, cursorId, traceId, spanId, experimentId,
       entityType, entityId, entityName, entityVersionId, parentEntityVersionId, parentEntityType, parentEntityId, parentEntityName, rootEntityVersionId, rootEntityType, rootEntityId, rootEntityName,
       userId, organizationId, resourceId, runId, sessionId, threadId, requestId, environment, executionSource, serviceName,
-      feedbackUserId, sourceId, feedbackSource, feedbackType, value, comment, tags, metadata, scope
+      feedbackUserId, sourceId, reviewStatus, feedbackSource, feedbackType, value, comment, tags, metadata, scope
     )
      VALUES ${tuples.join(",\n       ")}
      ON CONFLICT DO NOTHING`);
 }
+/** Update the review workflow status of a single feedback event. */
+async function updateFeedbackReviewStatus(db, args) {
+	const { feedbackId, reviewStatus } = parseUpdateFeedbackReviewStatusArgs(args);
+	await db.execute(`UPDATE feedback_events SET reviewStatus = ? WHERE feedbackId = ?`, [reviewStatus, feedbackId]);
+	const row = (await db.query(`SELECT * FROM feedback_events WHERE feedbackId = ? ORDER BY timestamp DESC LIMIT 1`, [feedbackId]))[0];
+	if (!row) throw new _mastra_core_error.MastraError({
+		id: "OBSERVABILITY_UPDATE_FEEDBACK_REVIEW_STATUS_NOT_FOUND",
+		domain: _mastra_core_error.ErrorDomain.MASTRA_OBSERVABILITY,
+		category: _mastra_core_error.ErrorCategory.USER,
+		text: "Feedback record not found",
+		details: { feedbackId }
+	});
+	return rowToFeedbackRecord(row);
+}
 /** Query feedback events with filtering, ordering, and pagination. */
 async function listFeedback(db, args) {
 	const { mode, filters, pagination, orderBy, after, limit } = _mastra_core_storage.listFeedbackArgsSchema.parse(args);
@@ -3793,6 +3839,9 @@ var ObservabilityStorageDuckDB = class extends _mastra_core_storage.Observabilit
 	async listFeedback(args) {
 		return listFeedback(this.db, args);
 	}
+	async updateFeedbackReviewStatus(args) {
+		return updateFeedbackReviewStatus(this.db, args);
+	}
 	async getFeedbackAggregate(args) {
 		return getFeedbackAggregate(this.db, args);
 	}
@@ -3809,4 +3858,4 @@ var ObservabilityStorageDuckDB = class extends _mastra_core_storage.Observabilit
 //#endregion
 exports.ObservabilityStorageDuckDB = ObservabilityStorageDuckDB;
 
-//# sourceMappingURL=observability-CRajwPel.cjs.map
+//# sourceMappingURL=observability-CYOty2wB.cjs.map