@mastra/duckdb 1.17.0 → 1.18.0-alpha.1

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

@@ -1,4 +1,4 @@
-const require_features = require("./features-CBdrOdEO.cjs");
+const require_features = require("./features-CP_ka777.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);
@@ -2883,6 +2883,18 @@ const TRACE_FIELDS = {
 		sql: "r.organizationId",
 		parameterType: "scalar"
 	},
+	requestId: {
+		sql: "r.requestId",
+		parameterType: "scalar"
+	},
+	serviceName: {
+		sql: "r.serviceName",
+		parameterType: "scalar"
+	},
+	experimentId: {
+		sql: "r.experimentId",
+		parameterType: "scalar"
+	},
 	startedAt: {
 		sql: "r.startedAt",
 		parameterType: "timestamp"
@@ -3036,6 +3048,18 @@ const SPAN_FIELDS = {
 	organizationId: {
 		sql: "s.organizationId",
 		parameterType: "scalar"
+	},
+	requestId: {
+		sql: "s.requestId",
+		parameterType: "scalar"
+	},
+	serviceName: {
+		sql: "s.serviceName",
+		parameterType: "scalar"
+	},
+	experimentId: {
+		sql: "s.experimentId",
+		parameterType: "scalar"
 	}
 };
 const SPAN_QUERY_FIELDS = {
@@ -3060,18 +3084,6 @@ const SPAN_QUERY_FIELDS = {
 		sql: "s.resourceId",
 		parameterType: "scalar"
 	},
-	requestId: {
-		sql: "s.requestId",
-		parameterType: "scalar"
-	},
-	serviceName: {
-		sql: "s.serviceName",
-		parameterType: "scalar"
-	},
-	experimentId: {
-		sql: "s.experimentId",
-		parameterType: "scalar"
-	},
 	environment: {
 		sql: "s.environment",
 		parameterType: "scalar"
@@ -3437,6 +3449,9 @@ const ROOT_PUSHDOWN_FIELDS = /* @__PURE__ */ new Set([
 	"sessionId",
 	"userId",
 	"organizationId",
+	"requestId",
+	"serviceName",
+	"experimentId",
 	"endedAt",
 	"timeToFirstChunkMs",
 	"timeToFirstTokenMs",
@@ -3587,7 +3602,10 @@ function compileDuckDBTraceScope(relatedCollections, scope, timeRange, rootPredi
         runId,
         sessionId,
         userId,
-        organizationId
+        organizationId,
+        requestId,
+        serviceName,
+        experimentId
       FROM current_span_rows
       WHERE currentRank = 1
     )`);
@@ -4508,6 +4526,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 +4611,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 +4648,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 +6427,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"] : [],
@@ -6399,6 +6437,7 @@ var ObservabilityStorageDuckDB = class ObservabilityStorageDuckDB extends _mastr
 			"trace-query-tenant-scope",
 			"feedback",
 			"trace-query-context-ids",
+			...require_features.contextFieldFeatures,
 			...require_features.timingFeatures,
 			...require_features.outcomeFeatures,
 			"trace-query-root-span-fields",
@@ -6416,6 +6455,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"] : [],
@@ -6425,6 +6465,7 @@ var ObservabilityStorageDuckDB = class ObservabilityStorageDuckDB extends _mastr
 			"trace-query-tenant-scope",
 			"feedback",
 			"trace-query-context-ids",
+			...require_features.contextFieldFeatures,
 			...require_features.timingFeatures,
 			...require_features.outcomeFeatures,
 			"trace-query-root-span-fields",
@@ -6592,4 +6633,4 @@ var ObservabilityStorageDuckDB = class ObservabilityStorageDuckDB extends _mastr
 };
 exports.ObservabilityStorageDuckDB = ObservabilityStorageDuckDB;
 
-//# sourceMappingURL=observability-B_ARB4Uv.cjs.map
+//# sourceMappingURL=observability-De8nSw6B.cjs.map