@mastra/pg 1.35.0-alpha.0 → 1.35.0-alpha.1

package/dist/index.js

@@ -20503,6 +20503,18 @@ const PERCENTILES = {
 	"duration.p95": .95,
 	"duration.p99": .99
 };
+/** Quantile for each percentile statistic of a time-to-first measure. */
+const STATISTIC_PERCENTILES = {
+	p50: .5,
+	p90: .9,
+	p95: .95,
+	p99: .99
+};
+/**
+* Store features this compiler supports beyond `trace-aggregate` / `span-aggregate`. Core before
+* the time-to-first measures rejects them, so they are only declared when core plans them.
+*/
+const aggregateTimeToFirstFeatures = "timeToFirstToken.p50" in coreStorage.TRACE_AGGREGATE_MEASURE_REGISTRY ? ["aggregate-time-to-first-measures"] : [];
 const COMPARISON_SQL = {
 	eq: "=",
 	ne: "<>",
@@ -20570,6 +20582,8 @@ function compilePostgresAggregate(schema, plan, source) {
 		factColumns.push(`(${distinctSql(measure.field)})::text AS ${column}`);
 	}
 	factColumns.push(`(${durationMsSql("r.\"startedAt\"", "r.\"endedAt\"")})::float8 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(`r."${field}"::float8 AS "${field}"`);
 	if (plan.interval !== void 0) {
 		const intervalMs = coreStorage.TRACE_AGGREGATE_INTERVAL_MS[plan.interval];
 		if (intervalMs === void 0) throw new Error(`Unsupported trusted trace-aggregate interval: ${plan.interval}`);
@@ -20601,6 +20615,15 @@ function compilePostgresAggregate(schema, plan, source) {
 			const aggregate = rule.statistic === "avg" ? "AVG" : "SUM";
 			return `(CASE WHEN ${mixedUnitsSql(alias)} THEN NULL ELSE ${aggregate}(${column("traceCost")}) END)::float8`;
 		}
+		if (rule?.field) {
+			const value = column(rule.field);
+			if (rule.statistic === "count") return `COUNT(${value})::float8`;
+			if (rule.statistic === "avg") return `AVG(${value})`;
+			if (rule.statistic === "min") return `MIN(${value})`;
+			if (rule.statistic === "max") return `MAX(${value})`;
+			const quantile = rule.statistic && STATISTIC_PERCENTILES[rule.statistic];
+			if (quantile !== void 0) return `percentile_cont(${quantile}) WITHIN GROUP (ORDER BY ${value})`;
+		}
 		if (name === "errorCount") return `(COUNT(*) FILTER (WHERE ${column("isError")}))::float8`;
 		if (name === "errorRate") return `(COUNT(*) FILTER (WHERE ${column("isError")}))::float8 / COUNT(*)`;
 		if (name === "duration.avg") return `AVG(${column("durationMs")})`;
@@ -21755,6 +21778,7 @@ var ObservabilityStoragePostgresVNext = class ObservabilityStoragePostgresVNext
 			"metric-discovery",
 			"trace-query",
 			"trace-aggregate",
+			...aggregateTimeToFirstFeatures,
 			...spanQueryFeatures,
 			...spanAggregateFeatures,
 			...typeof coreStorage.planThreadRowsQuery === "function" ? ["thread-query-rows"] : [],
@@ -21781,6 +21805,7 @@ var ObservabilityStoragePostgresVNext = class ObservabilityStoragePostgresVNext
 			"delta-polling",
 			"trace-query",
 			"trace-aggregate",
+			...aggregateTimeToFirstFeatures,
 			...spanQueryFeatures,
 			...spanAggregateFeatures,
 			...typeof coreStorage.planThreadRowsQuery === "function" ? ["thread-query-rows"] : [],