@mastra/duckdb 1.7.0 → 1.8.0-alpha.1

package/dist/{observability-Bt92lKRG.js → observability-2dvzwZpP.js}

@@ -52,6 +52,8 @@ CREATE TABLE IF NOT EXISTS span_events (
   entityId VARCHAR,
   entityName VARCHAR,
   entityVersionId VARCHAR,
+  parentEntityVersionId VARCHAR,
+  rootEntityVersionId VARCHAR,
 
   -- Context
   userId VARCHAR,
@@ -317,6 +319,8 @@ const ALL_MIGRATIONS = [
 	`CREATE SEQUENCE IF NOT EXISTS feedback_events_cursor_id_seq START 1`,
 	`ALTER TABLE span_events ADD COLUMN IF NOT EXISTS cursorId BIGINT`,
 	`ALTER TABLE span_events ADD COLUMN IF NOT EXISTS entityVersionId VARCHAR`,
+	`ALTER TABLE span_events ADD COLUMN IF NOT EXISTS parentEntityVersionId VARCHAR`,
+	`ALTER TABLE span_events ADD COLUMN IF NOT EXISTS rootEntityVersionId VARCHAR`,
 	`ALTER TABLE metric_events ADD COLUMN IF NOT EXISTS cursorId BIGINT`,
 	`ALTER TABLE metric_events ADD COLUMN IF NOT EXISTS entityVersionId VARCHAR`,
 	`ALTER TABLE metric_events ADD COLUMN IF NOT EXISTS parentEntityVersionId VARCHAR`,
@@ -1016,6 +1020,25 @@ async function batchCreateFeedback(db, args) {
      VALUES ${tuples.join(",\n       ")}
      ON CONFLICT DO NOTHING`);
 }
+/**
+* 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) {
 	const { feedbackId, reviewStatus } = parseUpdateFeedbackReviewStatusArgs(args);
@@ -2373,6 +2396,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);
@@ -2598,13 +2640,65 @@ const TRACE_FIELDS = {
 	}
 };
 const SPAN_FIELDS = {
+	name: {
+		sql: "s.name",
+		parameterType: "scalar"
+	},
 	spanType: {
 		sql: "s.spanType",
 		parameterType: "scalar"
 	},
+	model: {
+		sql: "s.model",
+		parameterType: "scalar"
+	},
+	provider: {
+		sql: "s.provider",
+		parameterType: "scalar"
+	},
+	startedAt: {
+		sql: "s.startedAt",
+		parameterType: "timestamp"
+	},
+	endedAt: {
+		sql: "s.endedAt",
+		parameterType: "timestamp"
+	},
+	durationMs: {
+		sql: "s.durationMs",
+		parameterType: "scalar"
+	},
+	status: {
+		sql: "s.status",
+		parameterType: "scalar"
+	},
 	error: {
 		sql: "s.error",
 		parameterType: "scalar"
+	},
+	entityType: {
+		sql: "s.entityType",
+		parameterType: "scalar"
+	},
+	entityId: {
+		sql: "s.entityId",
+		parameterType: "scalar"
+	},
+	entityName: {
+		sql: "s.entityName",
+		parameterType: "scalar"
+	},
+	entityVersionId: {
+		sql: "s.entityVersionId",
+		parameterType: "scalar"
+	},
+	parentEntityVersionId: {
+		sql: "s.parentEntityVersionId",
+		parameterType: "scalar"
+	},
+	rootEntityVersionId: {
+		sql: "s.rootEntityVersionId",
+		parameterType: "scalar"
 	}
 };
 const SCORE_FIELDS = {
@@ -2612,9 +2706,37 @@ const SCORE_FIELDS = {
 		sql: "s.scorerId",
 		parameterType: "scalar"
 	},
+	scorerVersion: {
+		sql: "s.scorerVersion",
+		parameterType: "scalar"
+	},
+	scoreSource: {
+		sql: "s.scoreSource",
+		parameterType: "scalar"
+	},
 	score: {
 		sql: "s.score",
 		parameterType: "scalar"
+	},
+	timestamp: {
+		sql: "s.timestamp",
+		parameterType: "timestamp"
+	},
+	spanId: {
+		sql: "s.spanId",
+		parameterType: "scalar"
+	},
+	entityVersionId: {
+		sql: "s.entityVersionId",
+		parameterType: "scalar"
+	},
+	parentEntityVersionId: {
+		sql: "s.parentEntityVersionId",
+		parameterType: "scalar"
+	},
+	rootEntityVersionId: {
+		sql: "s.rootEntityVersionId",
+		parameterType: "scalar"
 	}
 };
 const TRACE_SELECT = `
@@ -2636,12 +2758,15 @@ function fieldDefinition(registry, field) {
 function parameterSql(type) {
 	return type === "timestamp" ? "CAST(? AS TIMESTAMP)" : "?";
 }
-function compileScalarPredicate(predicate, registry) {
+function isMetadataField(field) {
+	return field.startsWith("metadata.");
+}
+function compileScalarPredicate(predicate, registry, allowMetadata = false) {
 	if (predicate.type === "boolean") {
 		const values = [];
 		return {
 			sql: predicate.args.map((arg) => {
-				const compiled = compileScalarPredicate(arg, registry);
+				const compiled = compileScalarPredicate(arg, registry, allowMetadata);
 				values.push(...compiled.values);
 				return `(${compiled.sql})`;
 			}).join(predicate.operator === "and" ? " AND " : " OR "),
@@ -2649,26 +2774,45 @@ function compileScalarPredicate(predicate, registry) {
 		};
 	}
 	if (predicate.type === "not") {
-		const compiled = compileScalarPredicate(predicate.arg, registry);
+		const compiled = compileScalarPredicate(predicate.arg, registry, allowMetadata);
 		return {
 			sql: `NOT (${compiled.sql})`,
 			values: compiled.values
 		};
 	}
-	const field = fieldDefinition(registry, predicate.field);
+	let field;
+	let fieldValues = [];
+	if (isMetadataField(predicate.field)) {
+		if (!allowMetadata) throw new Error(`Unsupported trusted trace-query field: ${predicate.field}`);
+		const key = predicate.field.slice(9);
+		const path = `$.${JSON.stringify(key)}`;
+		field = {
+			sql: `NULLIF(trim(CASE WHEN json_type(r.metadata, ?) = 'VARCHAR' THEN json_extract_string(r.metadata, ?) END), '')`,
+			parameterType: "scalar"
+		};
+		fieldValues = [path, path];
+	} else field = fieldDefinition(registry, predicate.field);
 	if (predicate.type === "presence") return {
 		sql: `${field.sql} IS ${predicate.operator === "exists" ? "NOT " : ""}NULL`,
-		values: []
+		values: fieldValues
 	};
 	if (predicate.type === "membership") {
 		const list = predicate.values.map(() => parameterSql(field.parameterType)).join(", ");
 		if (predicate.operator === "in") return {
 			sql: `${field.sql} IS NOT NULL AND ${field.sql} IN (${list})`,
-			values: predicate.values
+			values: [
+				...fieldValues,
+				...fieldValues,
+				...predicate.values
+			]
 		};
 		return {
 			sql: `${field.sql} IS NULL OR ${field.sql} NOT IN (${list})`,
-			values: predicate.values
+			values: [
+				...fieldValues,
+				...fieldValues,
+				...predicate.values
+			]
 		};
 	}
 	const parameter = parameterSql(field.parameterType);
@@ -2680,17 +2824,21 @@ function compileScalarPredicate(predicate, registry) {
 	};
 	if (predicate.operator === "eq") return {
 		sql: `${field.sql} IS NOT DISTINCT FROM ${parameter}`,
-		values: [predicate.value]
+		values: [...fieldValues, predicate.value]
 	};
 	if (predicate.operator === "ne") return {
 		sql: `${field.sql} IS DISTINCT FROM ${parameter}`,
-		values: [predicate.value]
+		values: [...fieldValues, predicate.value]
 	};
 	const operator = operators[predicate.operator];
 	if (operator === void 0) throw new Error(`Unsupported trusted trace-query operator: ${predicate.operator}`);
 	return {
 		sql: `${field.sql} IS NOT NULL AND ${field.sql} ${operator} ${parameter}`,
-		values: [predicate.value]
+		values: [
+			...fieldValues,
+			...fieldValues,
+			predicate.value
+		]
 	};
 }
 function compilePredicate(predicate) {
@@ -2726,7 +2874,7 @@ function compilePredicate(predicate) {
 			values: compiled.values
 		};
 	}
-	return compileScalarPredicate(predicate, TRACE_FIELDS);
+	return compileScalarPredicate(predicate, TRACE_FIELDS, true);
 }
 function collectRelatedCollections(predicate, collections = /* @__PURE__ */ new Set()) {
 	if (!predicate) return collections;
@@ -2775,18 +2923,43 @@ function compileDuckDBTraceQuery(plan) {
         AND r.startedAt < CAST(? AS TIMESTAMP)
     )`
 	];
-	if (relatedCollections.has("spans")) ctes.push(`current_spans AS (
-      SELECT * EXCLUDE (currentRank)
-      FROM (
-        SELECT
-          e.*,
-          row_number() OVER (
-            PARTITION BY e.traceId, e.spanId
-            ORDER BY CASE WHEN e.endedAt IS NULL THEN 1 ELSE 0 END ASC, e.cursorId DESC
-          ) AS currentRank
-        FROM span_events e
-        INNER JOIN root_scope roots ON roots.traceId = e.traceId
-      )
+	if (relatedCollections.has("spans")) ctes.push(`current_span_rows AS (
+      SELECT
+        e.*,
+        coalesce(
+          min(e.timestamp) FILTER (WHERE e.eventType = 'start') OVER (PARTITION BY e.traceId, e.spanId),
+          min(e.timestamp) OVER (PARTITION BY e.traceId, e.spanId)
+        ) AS startedAt,
+        row_number() OVER (
+          PARTITION BY e.traceId, e.spanId
+          ORDER BY CASE WHEN e.endedAt IS NULL THEN 1 ELSE 0 END ASC, e.cursorId DESC
+        ) AS currentRank
+      FROM span_events e
+      INNER JOIN root_scope roots ON roots.traceId = e.traceId
+    ),
+    current_spans AS (
+      SELECT
+        traceId,
+        name,
+        spanType,
+        CASE
+          WHEN json_type(attributes, '$.model') = 'VARCHAR' THEN json_extract_string(attributes, '$.model')
+        END AS model,
+        CASE
+          WHEN json_type(attributes, '$.provider') = 'VARCHAR' THEN json_extract_string(attributes, '$.provider')
+        END AS provider,
+        startedAt,
+        endedAt,
+        date_diff('millisecond', startedAt, endedAt) AS durationMs,
+        CASE WHEN error IS NOT NULL THEN 'error' ELSE 'success' END AS status,
+        error,
+        entityType,
+        entityId,
+        entityName,
+        entityVersionId,
+        parentEntityVersionId,
+        rootEntityVersionId
+      FROM current_span_rows
       WHERE currentRank = 1
     )`);
 	if (relatedCollections.has("scores")) ctes.push(`current_scores AS (
@@ -2891,6 +3064,8 @@ const COLUMNS_SQL = [
 	"entityId",
 	"entityName",
 	"entityVersionId",
+	"parentEntityVersionId",
+	"rootEntityVersionId",
 	"userId",
 	"organizationId",
 	"resourceId",
@@ -2933,6 +3108,8 @@ const SPAN_RECONSTRUCT_SELECT = `
     ${argMaxNonNull("entityId")},
     ${argMaxNonNull("entityName")},
     ${argMaxNonNull("entityVersionId")},
+    ${argMaxNonNull("parentEntityVersionId")},
+    ${argMaxNonNull("rootEntityVersionId")},
     ${argMaxNonNull("userId")},
     ${argMaxNonNull("organizationId")},
     ${argMaxNonNull("resourceId")},
@@ -3100,6 +3277,8 @@ function rowToSpanRecord(row) {
 		entityId: row.entityId ?? null,
 		entityName: row.entityName ?? null,
 		entityVersionId: row.entityVersionId ?? null,
+		parentEntityVersionId: row.parentEntityVersionId ?? null,
+		rootEntityVersionId: row.rootEntityVersionId ?? null,
 		userId: row.userId ?? null,
 		organizationId: row.organizationId ?? null,
 		resourceId: row.resourceId ?? null,
@@ -3255,6 +3434,8 @@ function toValuesTuple(row) {
 		v(row.entityId),
 		v(row.entityName),
 		v(row.entityVersionId),
+		v(row.parentEntityVersionId),
+		v(row.rootEntityVersionId),
 		v(row.userId),
 		v(row.organizationId),
 		v(row.resourceId),
@@ -3297,6 +3478,8 @@ function createStartSpanRow(s) {
 		entityId: s.entityId ?? null,
 		entityName: s.entityName ?? null,
 		entityVersionId: s.entityVersionId ?? null,
+		parentEntityVersionId: s.parentEntityVersionId ?? null,
+		rootEntityVersionId: s.rootEntityVersionId ?? null,
 		userId: s.userId ?? null,
 		organizationId: s.organizationId ?? null,
 		resourceId: s.resourceId ?? null,
@@ -3334,6 +3517,8 @@ function createEndSpanRow(s) {
 		entityId: s.entityId ?? null,
 		entityName: s.entityName ?? null,
 		entityVersionId: s.entityVersionId ?? null,
+		parentEntityVersionId: s.parentEntityVersionId ?? null,
+		rootEntityVersionId: s.rootEntityVersionId ?? null,
 		userId: s.userId ?? null,
 		organizationId: s.organizationId ?? null,
 		resourceId: s.resourceId ?? null,
@@ -4200,6 +4385,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);
 	}
@@ -4224,6 +4412,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);
 	}
@@ -4246,4 +4437,4 @@ var ObservabilityStorageDuckDB = class extends ObservabilityStorage {
 //#endregion
 export { ObservabilityStorageDuckDB };
 
-//# sourceMappingURL=observability-Bt92lKRG.js.map
+//# sourceMappingURL=observability-2dvzwZpP.js.map