@mastra/duckdb 1.8.0-alpha.0 → 1.8.0-alpha.2

package/dist/{observability-CgtrnaKz.js → observability-COgD6BOK.js}

@@ -745,6 +745,45 @@ function parseUpdateFeedbackReviewStatusArgs(args) {
 }
 //#endregion
 //#region src/storage/domains/observability/feedback.ts
+const FEEDBACK_UPSERT_CLAUSE = `ON CONFLICT (feedbackId) DO UPDATE SET ${[
+	"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",
+	"reviewStatus",
+	"feedbackSource",
+	"feedbackType",
+	"value",
+	"comment",
+	"tags",
+	"metadata",
+	"scope"
+].map((column) => `${column} = excluded.${column}`).join(", ")}`;
 const FEEDBACK_GROUP_BY_COLUMNS = /* @__PURE__ */ new Set([
 	"timestamp",
 	"traceId",
@@ -961,12 +1000,12 @@ async function createFeedback(db, args) {
 		jsonV(f.metadata),
 		jsonV(f.scope ?? null)
 	].join(", ")})
-     ON CONFLICT DO NOTHING`);
+     ${FEEDBACK_UPSERT_CLAUSE}`);
 }
 /** Insert multiple feedback events in a single statement. */
 async function batchCreateFeedback(db, args) {
 	if (args.feedbacks.length === 0) return;
-	const tuples = args.feedbacks.map((f) => {
+	const tuples = [...new Map(args.feedbacks.map((feedback) => [feedback.feedbackId, feedback])).values()].map((f) => {
 		const legacyFeedback = f;
 		const feedbackSource = legacyFeedback.feedbackSource ?? legacyFeedback.source ?? "";
 		const feedbackUserId = legacyFeedback.feedbackUserId ?? legacyFeedback.userId ?? null;
@@ -1018,7 +1057,26 @@ async function batchCreateFeedback(db, args) {
       feedbackUserId, sourceId, reviewStatus, feedbackSource, feedbackType, value, comment, tags, metadata, scope
     )
      VALUES ${tuples.join(",\n       ")}
-     ON CONFLICT DO NOTHING`);
+     ${FEEDBACK_UPSERT_CLAUSE}`);
+}
+/**
+* Delete feedback events by feedbackId. Optional `organizationId` and
+* `resourceId` values are ANDed into the predicate to restrict deletion to
+* records with matching scope fields.
+*/
+async function deleteFeedback(db, args) {
+	if (args.feedbackIds.length === 0) return;
+	const conditions = [`feedbackId IN (${args.feedbackIds.map(() => "?").join(", ")})`];
+	const params = [...args.feedbackIds];
+	if (args.organizationId !== void 0) {
+		conditions.push("organizationId = ?");
+		params.push(args.organizationId);
+	}
+	if (args.resourceId !== void 0) {
+		conditions.push("resourceId = ?");
+		params.push(args.resourceId);
+	}
+	await db.execute(`DELETE FROM feedback_events WHERE ${conditions.join(" AND ")}`, params);
 }
 /** Update the review workflow status of a single feedback event. */
 async function updateFeedbackReviewStatus(db, args) {
@@ -2377,6 +2435,25 @@ async function batchCreateScores(db, args) {
      VALUES ${tuples.join(",\n       ")}
      ${SCORE_UPSERT_CLAUSE}`);
 }
+/**
+* Delete score events by scoreId. Optional `organizationId` and `resourceId`
+* values are ANDed into the predicate to restrict deletion to records with
+* matching scope fields.
+*/
+async function deleteScores(db, args) {
+	if (args.scoreIds.length === 0) return;
+	const conditions = [`scoreId IN (${args.scoreIds.map(() => "?").join(", ")})`];
+	const params = [...args.scoreIds];
+	if (args.organizationId !== void 0) {
+		conditions.push("organizationId = ?");
+		params.push(args.organizationId);
+	}
+	if (args.resourceId !== void 0) {
+		conditions.push("resourceId = ?");
+		params.push(args.resourceId);
+	}
+	await db.execute(`DELETE FROM score_events WHERE ${conditions.join(" AND ")}`, params);
+}
 /** Query score events with filtering, ordering, and pagination. */
 async function listScores(db, args) {
 	const { mode, filters, pagination, orderBy, after, limit } = listScoresArgsSchema.parse(args);
@@ -2701,6 +2778,44 @@ const SCORE_FIELDS = {
 		parameterType: "scalar"
 	}
 };
+const FEEDBACK_FIELDS = {
+	feedbackType: {
+		sql: "s.feedbackType",
+		parameterType: "scalar"
+	},
+	feedbackSource: {
+		sql: "s.feedbackSource",
+		parameterType: "scalar"
+	},
+	feedbackUserId: {
+		sql: "s.feedbackUserId",
+		parameterType: "scalar"
+	},
+	sourceId: {
+		sql: "s.sourceId",
+		parameterType: "scalar"
+	},
+	entityVersionId: {
+		sql: "s.entityVersionId",
+		parameterType: "scalar"
+	},
+	parentEntityVersionId: {
+		sql: "s.parentEntityVersionId",
+		parameterType: "scalar"
+	},
+	rootEntityVersionId: {
+		sql: "s.rootEntityVersionId",
+		parameterType: "scalar"
+	},
+	timestamp: {
+		sql: "s.timestamp",
+		parameterType: "timestamp"
+	},
+	comment: {
+		sql: "s.comment",
+		parameterType: "scalar"
+	}
+};
 const TRACE_SELECT = `
   r.traceId AS traceId,
   r.spanId AS rootSpanId,
@@ -2803,12 +2918,40 @@ function compileScalarPredicate(predicate, registry, allowMetadata = false) {
 		]
 	};
 }
+function compileFeedbackScalarPredicate(predicate) {
+	if (predicate.type === "boolean") {
+		const values = [];
+		return {
+			sql: predicate.args.map((arg) => {
+				const compiled = compileFeedbackScalarPredicate(arg);
+				values.push(...compiled.values);
+				return `(${compiled.sql})`;
+			}).join(predicate.operator === "and" ? " AND " : " OR "),
+			values
+		};
+	}
+	if (predicate.type === "not") {
+		const compiled = compileFeedbackScalarPredicate(predicate.arg);
+		return {
+			sql: `NOT (${compiled.sql})`,
+			values: compiled.values
+		};
+	}
+	if (predicate.field !== "value") return compileScalarPredicate(predicate, FEEDBACK_FIELDS);
+	if (predicate.type === "presence") return {
+		sql: `s.value IS ${predicate.operator === "exists" ? "NOT " : ""}NULL`,
+		values: []
+	};
+	return compileScalarPredicate(predicate, { value: {
+		sql: typeof (predicate.type === "membership" ? predicate.values[0] : predicate.value) === "number" ? "TRY_CAST(s.value AS DOUBLE)" : "s.value",
+		parameterType: "scalar"
+	} });
+}
 function compilePredicate(predicate) {
 	if (predicate.type === "relation") {
-		const registry = predicate.collection === "spans" ? SPAN_FIELDS : SCORE_FIELDS;
-		const compiled = compileScalarPredicate(predicate.predicate, registry);
+		const compiled = predicate.collection === "feedback" ? compileFeedbackScalarPredicate(predicate.predicate) : compileScalarPredicate(predicate.predicate, predicate.collection === "spans" ? SPAN_FIELDS : SCORE_FIELDS);
 		const existence = `EXISTS (
-      SELECT 1 FROM ${predicate.collection === "spans" ? "current_spans" : "current_scores"} s
+      SELECT 1 FROM ${predicate.collection === "spans" ? "current_spans" : predicate.collection === "scores" ? "current_scores" : "current_feedback"} s
       WHERE s.traceId IS NOT NULL
         AND s.traceId = r.traceId
         AND (${compiled.sql})
@@ -2928,6 +3071,11 @@ function compileDuckDBTraceQuery(plan) {
       SELECT s.*
       FROM score_events s
       INNER JOIN root_scope roots ON roots.traceId = s.traceId
+    )`);
+	if (relatedCollections.has("feedback")) ctes.push(`current_feedback AS (
+      SELECT f.*
+      FROM feedback_events f
+      INNER JOIN root_scope roots ON roots.traceId = f.traceId
     )`);
 	ctes.push(`candidates AS (
     SELECT ${TRACE_SELECT}
@@ -4347,6 +4495,9 @@ var ObservabilityStorageDuckDB = class extends ObservabilityStorage {
 	async batchCreateScores(args) {
 		return batchCreateScores(this.db, args);
 	}
+	async deleteScores(args) {
+		return deleteScores(this.db, args);
+	}
 	async listScores(args) {
 		return listScores(this.db, args);
 	}
@@ -4371,6 +4522,9 @@ var ObservabilityStorageDuckDB = class extends ObservabilityStorage {
 	async batchCreateFeedback(args) {
 		return batchCreateFeedback(this.db, args);
 	}
+	async deleteFeedback(args) {
+		return deleteFeedback(this.db, args);
+	}
 	async listFeedback(args) {
 		return listFeedback(this.db, args);
 	}
@@ -4393,4 +4547,4 @@ var ObservabilityStorageDuckDB = class extends ObservabilityStorage {
 //#endregion
 export { ObservabilityStorageDuckDB };
 
-//# sourceMappingURL=observability-CgtrnaKz.js.map
+//# sourceMappingURL=observability-COgD6BOK.js.map