@mastra/pg 1.32.0-alpha.3 → 1.32.0-alpha.4

package/dist/docs/SKILL.md

@@ -3,7 +3,7 @@ name: mastra-pg
 description: Documentation for @mastra/pg. Use when working with @mastra/pg APIs, configuration, or implementation.
 metadata:
   package: "@mastra/pg"
-  version: "1.32.0-alpha.3"
+  version: "1.32.0-alpha.4"
 ---
 
 ## When to use

package/dist/docs/assets/SOURCE_MAP.json

@@ -1,5 +1,5 @@
 {
-  "version": "1.32.0-alpha.3",
+  "version": "1.32.0-alpha.4",
   "package": "@mastra/pg",
   "exports": {},
   "modules": {}

package/dist/index.cjs

@@ -19211,7 +19211,26 @@ async function getScorePercentiles(client, schema, args) {
 const TRACE_STATUS_SQL = `CASE WHEN r."error" IS NOT NULL THEN 'error' ELSE 'success' END`;
 /** Core versions without the field reject it in strict query results, so it is only returned when core knows it. */
 const timeToFirstChunkSupported = "timeToFirstChunkMs" in _mastra_core_storage.TRACE_QUERY_FIELD_REGISTRY.trace;
-const timeToFirstChunkFeatures = timeToFirstChunkSupported ? ["trace-query-time-to-first-chunk"] : [];
+/** Core before the sort rejects `orderBy` on the field, so the sort is only declared when core plans it. */
+const timeToFirstChunkSortSupported = _mastra_core_storage.TRACE_QUERY_ORDER_FIELDS?.includes("timeToFirstChunkMs") === true;
+const timeToFirstChunkFeatures = [...timeToFirstChunkSupported ? ["trace-query-time-to-first-chunk"] : [], ...timeToFirstChunkSortSupported ? ["trace-query-time-to-first-chunk-sort"] : []];
+/**
+* Keyset bounds for a cursor's sort value: terms for rows strictly past it in `direction`, and the
+* condition for rows level with it. Empty values sort last in both directions, so they are past
+* every value and level only with each other.
+*/
+function keysetSortBounds(field, direction, sortValue, values) {
+	if (sortValue === null) return {
+		past: [],
+		level: `${field} IS NULL`
+	};
+	values.push(sortValue);
+	const parameter = `$${values.length}::${typeof sortValue === "number" ? "float8" : "timestamptz"}`;
+	return {
+		past: [`${field} ${direction === "asc" ? ">" : "<"} ${parameter}`, `${field} IS NULL`],
+		level: `${field} = ${parameter}`
+	};
+}
 function durationMsSql(startedAt, endedAt) {
 	return `EXTRACT(EPOCH FROM (${endedAt} - ${startedAt}))::numeric * 1000`;
 }
@@ -19803,8 +19822,8 @@ LIMIT $${values.length}`,
 			values
 		};
 	}
-	const orderField = plan.orderBy.field === "startedAt" ? "\"startedAt\"" : "\"endedAt\"";
-	const direction = plan.orderBy.direction === "asc" ? "ASC" : "DESC";
+	const orderField = `"${plan.orderBy.field}"`;
+	const direction = `${plan.orderBy.direction === "asc" ? "ASC" : "DESC"} NULLS LAST`;
 	if (plan.paginationMode === "page") {
 		values.push(plan.perPage, plan.page * plan.perPage);
 		return {
@@ -19818,10 +19837,9 @@ LIMIT $${values.length - 1} OFFSET $${values.length}`,
 	}
 	let pageCondition = "";
 	if (plan.cursor) {
-		const comparison = plan.orderBy.direction === "asc" ? ">" : "<";
-		const sortParameter = `$${values.length + 1}`;
-		pageCondition = `WHERE (${orderField} ${comparison} ${sortParameter} OR (${orderField} = ${sortParameter} AND "traceId" > ${`$${values.length + 2}`}))`;
-		values.push(plan.cursor.sortValue, plan.cursor.traceId);
+		const { past, level } = keysetSortBounds(orderField, plan.orderBy.direction, plan.cursor.sortValue, values);
+		values.push(plan.cursor.traceId);
+		pageCondition = `WHERE (${[...past, `(${level} AND "traceId" > $${values.length})`].join(" OR ")})`;
 	}
 	values.push(plan.limit + 1);
 	return {
@@ -20154,7 +20172,7 @@ async function queryTraces(client, schema, plan, timeoutMs) {
 		traces,
 		page: { next: rows.length > plan.limit && last ? _mastra_core_storage.encodeTraceQueryCursor(plan, {
 			result: "traces",
-			sortValue: last[plan.orderBy.field],
+			sortValue: last[plan.orderBy.field] ?? null,
 			traceId: last.traceId
 		}) : null }
 	});
@@ -20226,11 +20244,8 @@ function selectedKeys(rows, values) {
 function afterCursor(plan, values) {
 	if (!plan.cursor) return "TRUE";
 	const cursor = plan.cursor;
-	values.push(cursor.sortValue);
-	const timestamp = `$${values.length}::timestamptz`;
-	const field = `s."${plan.orderBy.field}"`;
-	const terms = [`${field} ${plan.orderBy.direction === "asc" ? ">" : "<"} ${timestamp}`];
-	const equal = [`${field} = ${timestamp}`];
+	const { past: terms, level } = keysetSortBounds(`s."${plan.orderBy.field}"`, plan.orderBy.direction, cursor.sortValue, values);
+	const equal = [level];
 	for (const key of identityFields) {
 		const ref = `s."${key}" COLLATE "C"`;
 		const value = cursor[key];
@@ -20296,7 +20311,7 @@ function compilePostgresSpanQuery(schema, plan) {
 			"version"
 		].map((field) => `s."${field}"`).join(", ")}
   FROM candidates s WHERE s."startedAt" >= $1 AND s."startedAt" < $2 AND (${predicate.sql}) AND ${after}
-  ORDER BY s."${plan.orderBy.field}" ${plan.orderBy.direction === "asc" ? "ASC" : "DESC"}, ${identityOrder}
+  ORDER BY s."${plan.orderBy.field}" ${plan.orderBy.direction === "asc" ? "ASC" : "DESC"} NULLS LAST, ${identityOrder}
   LIMIT $${values.length}`,
 		values
 	};