@mastra/duckdb 1.19.0-alpha.1 → 1.19.0-alpha.3

package/dist/{observability-YXMSShsG.cjs → observability-DGLhUhwg.cjs}

@@ -1151,11 +1151,11 @@ function getComparisonDateRange$1(comparePeriod, timestamp) {
 	}
 }
 /** Insert a single feedback event. */
-async function createFeedback(db, args) {
+async function createFeedback(db, args, preserveIdentity = false) {
 	const f = args.feedback;
 	const feedbackSource = f.feedbackSource ?? f.source ?? "";
 	const feedbackUserId = f.feedbackUserId !== void 0 ? f.feedbackUserId : f.userId ?? null;
-	await db.execute(`INSERT INTO feedback_events (
+	const sql = `INSERT INTO feedback_events (
       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,
@@ -1203,12 +1203,25 @@ async function createFeedback(db, args) {
 		jsonV(f.metadata),
 		jsonV(f.scope ?? null)
 	].join(", ")})
-     ${FEEDBACK_UPSERT_CLAUSE}`);
+     ${args.onConflict === "ignore" ? "ON CONFLICT DO NOTHING" : FEEDBACK_UPSERT_CLAUSE}`;
+	if (preserveIdentity) {
+		if (!(await db.query(`${sql}
+      WHERE feedback_events.traceId IS NOT DISTINCT FROM excluded.traceId
+        AND feedback_events.timestamp IS NOT DISTINCT FROM excluded.timestamp
+        AND feedback_events.organizationId IS NOT DISTINCT FROM excluded.organizationId
+        AND feedback_events.resourceId IS NOT DISTINCT FROM excluded.resourceId
+      RETURNING feedbackId`)).length) throw new _mastra_core_error.MastraError({
+			id: "OBSERVABILITY_FEEDBACK_IDENTITY_CONFLICT",
+			domain: _mastra_core_error.ErrorDomain.MASTRA_OBSERVABILITY,
+			category: _mastra_core_error.ErrorCategory.USER,
+			text: "Feedback replacement must preserve its trace, timestamp, and tenant"
+		});
+	} else await db.execute(sql);
 }
 /** Insert multiple feedback events in a single statement. */
 async function batchCreateFeedback(db, args) {
 	if (args.feedbacks.length === 0) return;
-	const tuples = [...new Map(args.feedbacks.map((feedback) => [feedback.feedbackId, feedback])).values()].map((f) => {
+	const tuples = [...new Map((args.onConflict === "ignore" ? [...args.feedbacks].reverse() : args.feedbacks).map((feedback) => [feedback.feedbackId, feedback])).values()].map((f) => {
 		const legacyFeedback = f;
 		const feedbackSource = legacyFeedback.feedbackSource ?? legacyFeedback.source ?? "";
 		const feedbackUserId = legacyFeedback.feedbackUserId !== void 0 ? legacyFeedback.feedbackUserId : legacyFeedback.userId ?? null;
@@ -1262,7 +1275,7 @@ async function batchCreateFeedback(db, args) {
       feedbackUserId, sourceId, reviewStatus, feedbackSource, feedbackType, value, valueString, valueNumber, comment, tags, metadata, scope
     )
      VALUES ${tuples.join(",\n       ")}
-     ${FEEDBACK_UPSERT_CLAUSE}`);
+     ${args.onConflict === "ignore" ? "ON CONFLICT DO NOTHING" : FEEDBACK_UPSERT_CLAUSE}`);
 }
 /**
 * Delete feedback events by feedbackId. Optional `organizationId` and
@@ -6372,6 +6385,9 @@ ${args.tables.map((table) => `  - ${table.table}`).join("\n")}\n\nTo fix this, r
 * Uses an append-only event-sourced model with SQL-based reconstruction for spans.
 */
 var ObservabilityStorageDuckDB = class ObservabilityStorageDuckDB extends _mastra_core_storage.ObservabilityStorage {
+	get persistenceLocation() {
+		return "local";
+	}
 	static retentionTables = {
 		spans: {
 			table: "span_events",
@@ -6497,6 +6513,7 @@ var ObservabilityStorageDuckDB = class ObservabilityStorageDuckDB extends _mastr
 			"thread-query",
 			"trace-query-tenant-scope",
 			"feedback",
+			"feedback-replacement",
 			"trace-query-context-ids",
 			...require_features.contextFieldFeatures,
 			...require_features.timingFeatures,
@@ -6525,6 +6542,7 @@ var ObservabilityStorageDuckDB = class ObservabilityStorageDuckDB extends _mastr
 			"thread-query",
 			"trace-query-tenant-scope",
 			"feedback",
+			"feedback-replacement",
 			"trace-query-context-ids",
 			...require_features.contextFieldFeatures,
 			...require_features.timingFeatures,
@@ -6667,6 +6685,12 @@ var ObservabilityStorageDuckDB = class ObservabilityStorageDuckDB extends _mastr
 	async getScorePercentiles(args) {
 		return getScorePercentiles(this.db, args);
 	}
+	async writeFeedbackReplacement(args) {
+		return createFeedback(this.db, {
+			feedback: args.feedback,
+			onConflict: "replace"
+		}, true);
+	}
 	async createFeedback(args) {
 		return createFeedback(this.db, args);
 	}
@@ -6697,4 +6721,4 @@ var ObservabilityStorageDuckDB = class ObservabilityStorageDuckDB extends _mastr
 };
 exports.ObservabilityStorageDuckDB = ObservabilityStorageDuckDB;
 
-//# sourceMappingURL=observability-YXMSShsG.cjs.map
+//# sourceMappingURL=observability-DGLhUhwg.cjs.map