@mastra/duckdb 1.17.0 → 1.18.0-alpha.0

package/dist/{observability-B_ARB4Uv.cjs → observability-DFLDzgiC.cjs}

@@ -1,4 +1,4 @@
-const require_features = require("./features-CBdrOdEO.cjs");
+const require_features = require("./features-hDeKZlYS.cjs");
 let _mastra_core_error = require("@mastra/core/error");
 let _mastra_core_storage = require("@mastra/core/storage");
 _mastra_core_storage = require_features.__toESM(_mastra_core_storage, 1);
@@ -4508,6 +4508,13 @@ const SPAN_DIMENSION_SQL = Object.fromEntries([
 	"model",
 	"provider"
 ].map((field) => [field, `r.${field}`]));
+/** Quantile for each percentile statistic of a time-to-first measure. */
+const STATISTIC_PERCENTILES = {
+	p50: .5,
+	p90: .9,
+	p95: .95,
+	p99: .99
+};
 const PERCENTILES = {
 	"duration.p50": .5,
 	"duration.p90": .9,
@@ -4586,6 +4593,8 @@ function compileDuckDBAggregate(plan, options, source) {
 		values.push(...compiled.values);
 	}
 	factColumns.push(`CAST(${durationMsSql("r.startedAt", "r.endedAt")} AS DOUBLE) AS durationMs`, `r.error IS NOT NULL AS isError`);
+	const timeToFirstFields = new Set(plan.measures.map((measure) => measureRule(measure.name)?.field));
+	for (const field of timeToFirstFields) if (field) factColumns.push(`CAST(r.${field} AS DOUBLE) AS ${field}`);
 	if (plan.interval !== void 0) {
 		const intervalMs = _mastra_core_storage.TRACE_AGGREGATE_INTERVAL_MS[plan.interval];
 		if (intervalMs === void 0) throw new Error(`Unsupported trusted trace-aggregate interval: ${plan.interval}`);
@@ -4621,6 +4630,16 @@ function compileDuckDBAggregate(plan, options, source) {
 	const measureSql = (name, alias = "") => {
 		const column = (name) => `${alias}${name}`;
 		if (name === "count") return "CAST(count(*) AS DOUBLE)";
+		const timeToFirst = measureRule(name);
+		if (timeToFirst?.field) {
+			const value = column(timeToFirst.field);
+			if (timeToFirst.statistic === "count") return `CAST(count(${value}) AS DOUBLE)`;
+			if (timeToFirst.statistic === "avg") return `avg(${value})`;
+			if (timeToFirst.statistic === "min") return `min(${value})`;
+			if (timeToFirst.statistic === "max") return `max(${value})`;
+			const quantile = timeToFirst.statistic && STATISTIC_PERCENTILES[timeToFirst.statistic];
+			if (quantile !== void 0) return `quantile_cont(${value}, ${quantile})`;
+		}
 		if (name === "errorCount") return `CAST(count_if(${column("isError")}) AS DOUBLE)`;
 		if (name === "errorRate") return `CAST(count_if(${column("isError")}) AS DOUBLE) / count(*)`;
 		if (name === "duration.avg") return `avg(${column("durationMs")})`;
@@ -6390,6 +6409,7 @@ var ObservabilityStorageDuckDB = class ObservabilityStorageDuckDB extends _mastr
 			"metric-discovery",
 			"trace-query",
 			"trace-aggregate",
+			...require_features.aggregateTimeToFirstFeatures,
 			...require_features.spanQueryFeatures,
 			...require_features.spanAggregateFeatures,
 			...typeof _mastra_core_storage.planThreadRowsQuery === "function" ? ["thread-query-rows"] : [],
@@ -6416,6 +6436,7 @@ var ObservabilityStorageDuckDB = class ObservabilityStorageDuckDB extends _mastr
 			"delta-polling",
 			"trace-query",
 			"trace-aggregate",
+			...require_features.aggregateTimeToFirstFeatures,
 			...require_features.spanQueryFeatures,
 			...require_features.spanAggregateFeatures,
 			...typeof _mastra_core_storage.planThreadRowsQuery === "function" ? ["thread-query-rows"] : [],
@@ -6592,4 +6613,4 @@ var ObservabilityStorageDuckDB = class ObservabilityStorageDuckDB extends _mastr
 };
 exports.ObservabilityStorageDuckDB = ObservabilityStorageDuckDB;
 
-//# sourceMappingURL=observability-B_ARB4Uv.cjs.map
+//# sourceMappingURL=observability-DFLDzgiC.cjs.map