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

package/dist/{observability-CdS3iFtO.cjs → observability-D276d4Zq.cjs}

@@ -709,6 +709,23 @@ function jsonV(val) {
 	if (val === null || val === void 0) return "NULL";
 	return require_db.DuckDBConnection.sqlValue(JSON.stringify(val));
 }
+/**
+* Trim, drop blank, and dedupe tags before they are written, matching the
+* PostgreSQL and ClickHouse stores so tag predicates and discovery agree.
+*/
+function normalizeTags(tags) {
+	if (!Array.isArray(tags)) return [];
+	const seen = /* @__PURE__ */ new Set();
+	const result = [];
+	for (const t of tags) {
+		if (typeof t !== "string") continue;
+		const trimmed = t.trim();
+		if (trimmed === "" || seen.has(trimmed)) continue;
+		seen.add(trimmed);
+		result.push(trimmed);
+	}
+	return result;
+}
 /** Coerce a value to a Date. Throws if value is nullish. */
 function toDate(val) {
 	if (val === null || val === void 0) throw new Error("Expected date value but received null/undefined");
@@ -2705,6 +2722,9 @@ async function getScorePercentiles(db, args) {
 //#endregion
 //#region src/storage/domains/observability/trace-query.ts
 const TRACE_STATUS_SQL = `CASE WHEN r.error IS NOT NULL THEN 'error' ELSE 'success' END`;
+function durationMsSql(startedAt, endedAt) {
+	return `date_diff('millisecond', ${startedAt}, ${endedAt})`;
+}
 const TRACE_FIELDS = {
 	traceId: {
 		sql: "r.traceId",
@@ -2726,6 +2746,10 @@ const TRACE_FIELDS = {
 		sql: "r.endedAt",
 		parameterType: "timestamp"
 	},
+	durationMs: {
+		sql: durationMsSql("r.startedAt", "r.endedAt"),
+		parameterType: "scalar"
+	},
 	entityName: {
 		sql: "r.entityName",
 		parameterType: "scalar"
@@ -2741,6 +2765,10 @@ const TRACE_FIELDS = {
 	status: {
 		sql: TRACE_STATUS_SQL,
 		parameterType: "scalar"
+	},
+	tags: {
+		sql: "r.tags",
+		parameterType: "scalar"
 	}
 };
 const SPAN_FIELDS = {
@@ -2943,6 +2971,25 @@ function compileScalarPredicate(predicate, registry, allowMetadata = false) {
 		sql: `${field.sql} IS ${predicate.operator === "exists" ? "NOT " : ""}NULL`,
 		values: fieldValues
 	};
+	if (predicate.type === "collection") {
+		const members = `coalesce(TRY_CAST(${field.sql} AS VARCHAR[]), []::VARCHAR[])`;
+		if (!("value" in predicate)) return {
+			sql: `len(${members}) ${predicate.operator === "empty" ? "=" : ">"} 0`,
+			values: fieldValues
+		};
+		if (predicate.operator === "includes") return {
+			sql: `list_contains(${members}, ?)`,
+			values: [...fieldValues, predicate.value]
+		};
+		return {
+			sql: `len(${members}) > 0 AND NOT list_contains(${members}, ?)`,
+			values: [
+				...fieldValues,
+				...fieldValues,
+				predicate.value
+			]
+		};
+	}
 	if (predicate.type === "membership") {
 		const list = predicate.values.map(() => parameterSql(field.parameterType)).join(", ");
 		if (predicate.operator === "in") return {
@@ -3007,7 +3054,7 @@ function compileFeedbackScalarPredicate(predicate) {
 			values: compiled.values
 		};
 	}
-	if (predicate.field !== "value") return compileScalarPredicate(predicate, FEEDBACK_FIELDS);
+	if (predicate.field !== "value" || predicate.type === "collection") return compileScalarPredicate(predicate, FEEDBACK_FIELDS);
 	if (predicate.type === "presence") return {
 		sql: `s.value IS ${predicate.operator === "exists" ? "NOT " : ""}NULL`,
 		values: []
@@ -3161,7 +3208,7 @@ function compileDuckDBTraceScope(relatedCollections, scope) {
         END AS provider,
         startedAt,
         endedAt,
-        date_diff('millisecond', startedAt, endedAt) AS durationMs,
+        ${durationMsSql("startedAt", "endedAt")} AS durationMs,
         CASE WHEN error IS NOT NULL THEN 'error' ELSE 'success' END AS status,
         error,
         entityType,
@@ -3393,7 +3440,11 @@ function compileDuckDBTraceQueryValues(plan) {
 		...scopeValues
 	];
 	let fieldSql;
-	if (plan.predicateScope === "trace" && plan.path.startsWith("metadata.")) {
+	let source = discoverySource(plan.predicateScope);
+	if (plan.predicateScope === "trace" && plan.path === "tags") {
+		fieldSql = "unnest(list_distinct(TRY_CAST(r.tags AS VARCHAR[])))";
+		source = `${source} WHERE r.tags IS NOT NULL`;
+	} else if (plan.predicateScope === "trace" && plan.path.startsWith("metadata.")) {
 		const jsonPath = `$.${JSON.stringify(plan.path.slice(9))}`;
 		fieldSql = `NULLIF(trim(CASE WHEN json_type(r.metadata, ?) = 'VARCHAR' THEN json_extract_string(r.metadata, ?) END), '')`;
 		values.push(jsonPath, jsonPath);
@@ -3403,7 +3454,7 @@ function compileDuckDBTraceQueryValues(plan) {
 	const search = plan.search ? "AND strpos(lower(CAST(value AS VARCHAR)), lower(?)) > 0" : "";
 	return {
 		sql: `WITH ${ctes.join(",\n  ")}, extracted AS (
-  SELECT ${fieldSql} AS value FROM ${discoverySource(plan.predicateScope)}
+  SELECT ${fieldSql} AS value FROM ${source}
 )
 SELECT CAST(value AS VARCHAR) AS value, count(*) AS count
 FROM extracted
@@ -4001,7 +4052,7 @@ function createStartSpanRow(s) {
 		serviceName: s.serviceName ?? null,
 		attributes: s.attributes ?? null,
 		metadata: s.metadata ?? null,
-		tags: s.tags ?? null,
+		tags: s.tags == null ? null : normalizeTags(s.tags),
 		scope: s.scope ?? null,
 		links: null,
 		input: s.input ?? null,
@@ -4040,7 +4091,7 @@ function createEndSpanRow(s) {
 		serviceName: s.serviceName ?? null,
 		attributes: s.attributes ?? null,
 		metadata: s.metadata ?? null,
-		tags: s.tags ?? null,
+		tags: s.tags == null ? null : normalizeTags(s.tags),
 		scope: s.scope ?? null,
 		links: s.links ?? null,
 		input: s.input ?? null,
@@ -4842,6 +4893,7 @@ var ObservabilityStorageDuckDB = class ObservabilityStorageDuckDB extends _mastr
 			"metrics",
 			"logs",
 			"trace-query",
+			"trace-query-root-duration",
 			"trace-query-discovery",
 			"thread-query",
 			"trace-query-tenant-scope"
@@ -4851,6 +4903,7 @@ var ObservabilityStorageDuckDB = class ObservabilityStorageDuckDB extends _mastr
 			"logs",
 			"delta-polling",
 			"trace-query",
+			"trace-query-root-duration",
 			"trace-query-discovery",
 			"thread-query",
 			"trace-query-tenant-scope"
@@ -5008,4 +5061,4 @@ var ObservabilityStorageDuckDB = class ObservabilityStorageDuckDB extends _mastr
 //#endregion
 exports.ObservabilityStorageDuckDB = ObservabilityStorageDuckDB;
 
-//# sourceMappingURL=observability-CdS3iFtO.cjs.map
+//# sourceMappingURL=observability-D276d4Zq.cjs.map