@mastra/duckdb 1.19.0-alpha.4 → 1.19.0-alpha.5

package/dist/{observability-cJFzLpu4.cjs → observability-BT6At9hS.cjs}

@@ -1,4 +1,4 @@
-const require_features = require("./features-xTAPLAPX.cjs");
+const require_features = require("./features-TxRZAzYn.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);
@@ -4097,15 +4097,19 @@ function keys(rows, values, alias = "") {
 		return `${alias}${field} IS NOT DISTINCT FROM ?`;
 	}).join(" AND ")})`).join(" OR ");
 }
+function sortColumn(plan) {
+	const column = `s.${plan.orderBy.field}`;
+	return plan.orderBy.field === "model" || plan.orderBy.field === "provider" ? `${column} COLLATE "C"` : column;
+}
 function afterCursor(plan) {
 	if (!plan.cursor) return {
 		sql: "TRUE",
 		values: []
 	};
 	const cursor = plan.cursor;
-	const sort = `s.${plan.orderBy.field}`;
+	const sort = sortColumn(plan);
 	const empty = cursor.sortValue === null;
-	const value = `CAST(? AS ${typeof cursor.sortValue === "number" ? "DOUBLE" : "TIMESTAMP"})`;
+	const value = plan.orderBy.field === "model" || plan.orderBy.field === "provider" ? "CAST(? AS VARCHAR) COLLATE \"C\"" : `CAST(? AS ${typeof cursor.sortValue === "number" ? "DOUBLE" : "TIMESTAMP"})`;
 	const comparisons = empty ? [] : [`${sort} ${plan.orderBy.direction === "asc" ? ">" : "<"} ${value}`, `${sort} IS NULL`];
 	const values = empty ? [] : [cursor.sortValue];
 	for (let i = 0; i < identityFields.length; i++) {
@@ -4186,10 +4190,12 @@ function compileDuckDBSpanQuery(plan) {
 			"endedAt",
 			"durationMs",
 			"status",
-			"cursorId"
+			"cursorId",
+			"model",
+			"provider"
 		].map((field) => `s.${field}`).join(", ")}
   FROM selected s WHERE ${after.sql}
-  ORDER BY s.${plan.orderBy.field} ${plan.orderBy.direction === "asc" ? "ASC" : "DESC"} NULLS LAST, ${identityFields.map((field) => `s.${field} ASC NULLS LAST`).join(", ")}
+  ORDER BY ${sortColumn(plan)} ${plan.orderBy.direction === "asc" ? "ASC" : "DESC"} NULLS LAST, ${identityFields.map((field) => `s.${field} ASC NULLS LAST`).join(", ")}
   LIMIT ?`,
 		values
 	};
@@ -4227,6 +4233,7 @@ async function querySpans(db, plan) {
 	const rows = await query(compiled.sql, compiled.values);
 	const selected = rows.map(({ cursorId: _cursorId, startedAt, endedAt, timeToFirstChunkMs, timeToFirstTokenMs, outcome, stopReason, ...row }) => ({
 		...row,
+		...(plan.orderBy.field === "model" || plan.orderBy.field === "provider") && { cursorSortValue: row[plan.orderBy.field] },
 		startedAt: startedAt.toISOString(),
 		endedAt: endedAt.toISOString(),
 		...require_features.timeToFirstChunkSupported && { timeToFirstChunkMs: timeToFirstChunkMs == null ? null : Number(timeToFirstChunkMs) },
@@ -6721,4 +6728,4 @@ var ObservabilityStorageDuckDB = class ObservabilityStorageDuckDB extends _mastr
 };
 exports.ObservabilityStorageDuckDB = ObservabilityStorageDuckDB;
 
-//# sourceMappingURL=observability-cJFzLpu4.cjs.map
+//# sourceMappingURL=observability-BT6At9hS.cjs.map