@mastra/duckdb 1.7.0 → 1.8.0-alpha.0

package/dist/{observability-yLoG0tl3.cjs → observability-VHURnMMH.cjs}

@@ -74,6 +74,8 @@ CREATE TABLE IF NOT EXISTS span_events (
   entityId VARCHAR,
   entityName VARCHAR,
   entityVersionId VARCHAR,
+  parentEntityVersionId VARCHAR,
+  rootEntityVersionId VARCHAR,
 
   -- Context
   userId VARCHAR,
@@ -339,6 +341,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`,
@@ -2620,13 +2624,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 = {
@@ -2634,9 +2690,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 = `
@@ -2658,12 +2742,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 "),
@@ -2671,26 +2758,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);
@@ -2702,17 +2808,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) {
@@ -2748,7 +2858,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;
@@ -2797,18 +2907,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 (
@@ -2913,6 +3048,8 @@ const COLUMNS_SQL = [
 	"entityId",
 	"entityName",
 	"entityVersionId",
+	"parentEntityVersionId",
+	"rootEntityVersionId",
 	"userId",
 	"organizationId",
 	"resourceId",
@@ -2955,6 +3092,8 @@ const SPAN_RECONSTRUCT_SELECT = `
     ${argMaxNonNull("entityId")},
     ${argMaxNonNull("entityName")},
     ${argMaxNonNull("entityVersionId")},
+    ${argMaxNonNull("parentEntityVersionId")},
+    ${argMaxNonNull("rootEntityVersionId")},
     ${argMaxNonNull("userId")},
     ${argMaxNonNull("organizationId")},
     ${argMaxNonNull("resourceId")},
@@ -3122,6 +3261,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,
@@ -3277,6 +3418,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),
@@ -3319,6 +3462,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,
@@ -3356,6 +3501,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,
@@ -4268,4 +4415,4 @@ var ObservabilityStorageDuckDB = class extends _mastra_core_storage.Observabilit
 //#endregion
 exports.ObservabilityStorageDuckDB = ObservabilityStorageDuckDB;
 
-//# sourceMappingURL=observability-yLoG0tl3.cjs.map
+//# sourceMappingURL=observability-VHURnMMH.cjs.map