@mastra/duckdb 1.10.0-alpha.1 → 1.10.0-alpha.3

package/dist/docs/SKILL.md

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

package/dist/docs/assets/SOURCE_MAP.json

@@ -1,5 +1,5 @@
 {
-  "version": "1.10.0-alpha.1",
+  "version": "1.10.0-alpha.3",
   "package": "@mastra/duckdb",
   "exports": {},
   "modules": {}

package/dist/index.cjs

@@ -532,7 +532,7 @@ var ObservabilityStorageDuckDB = class extends _mastra_core_storage.Observabilit
 	async loadDelegate() {
 		if (this.delegate) return this.delegate;
 		if (this.unavailableError) return null;
-		if (!this.loadPromise) this.loadPromise = Promise.resolve().then(() => require("./observability-CL_Bv53d.cjs")).then(({ ObservabilityStorageDuckDB }) => {
+		if (!this.loadPromise) this.loadPromise = Promise.resolve().then(() => require("./observability-POEhuHwk.cjs")).then(({ ObservabilityStorageDuckDB }) => {
 			const delegate = new ObservabilityStorageDuckDB({ db: this.db });
 			this.delegate = delegate;
 			return delegate;

package/dist/index.js

@@ -531,7 +531,7 @@ var ObservabilityStorageDuckDB = class extends ObservabilityStorage {
 	async loadDelegate() {
 		if (this.delegate) return this.delegate;
 		if (this.unavailableError) return null;
-		if (!this.loadPromise) this.loadPromise = import("./observability-W6La6xQK.js").then(({ ObservabilityStorageDuckDB }) => {
+		if (!this.loadPromise) this.loadPromise = import("./observability-D3311DPa.js").then(({ ObservabilityStorageDuckDB }) => {
 			const delegate = new ObservabilityStorageDuckDB({ db: this.db });
 			this.delegate = delegate;
 			return delegate;

package/dist/{observability-W6La6xQK.js → observability-D3311DPa.js}

@@ -2420,7 +2420,13 @@ async function createScore(db, args) {
 /** Insert multiple score events in a single statement. */
 async function batchCreateScores(db, args) {
 	if (args.scores.length === 0) return;
-	const tuples = args.scores.map((s) => {
+	const currentScores = /* @__PURE__ */ new Map();
+	for (const score of args.scores) {
+		const key = score.scoreId ?? Symbol();
+		currentScores.delete(key);
+		currentScores.set(key, score);
+	}
+	const tuples = [...currentScores.values()].map((s) => {
 		const legacyScore = s;
 		const scoreSource = legacyScore.scoreSource ?? legacyScore.source ?? null;
 		return `(${[
@@ -3160,7 +3166,7 @@ function compileDuckDBTraceQuery(plan) {
 	}
 	const ctes = compileDuckDBTraceScope(collectRelatedCollections(plan.where));
 	ctes.push(`candidates AS (
-    SELECT ${TRACE_SELECT}
+    SELECT ${TRACE_SELECT}${plan.paginationMode === "delta" ? ", r.cursorId AS deltaWatermark" : ""}
     FROM root_scope r
     WHERE ${conditions.slice(3).join("\n      AND ") || "TRUE"}
   )`);
@@ -3180,8 +3186,49 @@ LIMIT ?`,
 			values
 		};
 	}
+	if (plan.paginationMode === "delta") {
+		const watermark = coreStorage.getTraceQueryDeltaWatermark(plan, "duckdb");
+		if (watermark !== void 0 && (!/^\d+$/.test(watermark) || BigInt(watermark) > 9223372036854775807n)) throw new coreStorage.TraceQueryCursorError("TRACE_QUERY_CURSOR_MALFORMED");
+		values.push(watermark ?? "0", plan.limit + 1);
+		return {
+			sql: `${candidates},
+  delta_head AS (SELECT coalesce(max(cursorId), 0) AS streamHead FROM root_events),
+  delta_rows AS (
+    SELECT * FROM candidates
+    WHERE deltaWatermark > CAST(? AS BIGINT) ${watermark === void 0 ? "AND FALSE" : ""}
+    ORDER BY deltaWatermark ASC, traceId ASC
+    LIMIT ?
+  )
+SELECT delta_rows.*, delta_head.streamHead
+FROM delta_head
+LEFT JOIN delta_rows ON TRUE
+ORDER BY delta_rows.deltaWatermark ASC NULLS LAST, delta_rows.traceId ASC`,
+			values
+		};
+	}
 	const orderField = plan.orderBy.field;
 	const direction = plan.orderBy.direction === "asc" ? "ASC" : "DESC";
+	if (plan.paginationMode === "page") {
+		values.push(plan.perPage, plan.page * plan.perPage);
+		return {
+			sql: `${candidates},
+  page_rows AS (
+    SELECT *, row_number() OVER (ORDER BY ${orderField} ${direction}, traceId ASC) AS __row_position
+    FROM candidates
+    ORDER BY ${orderField} ${direction}, traceId ASC
+    LIMIT ? OFFSET ?
+  ),
+  page_total AS (
+    SELECT COUNT(*) AS total
+    FROM candidates
+  )
+SELECT page_rows.*, page_total.total${deltaPollingFeatureEnabled() ? ", (SELECT coalesce(max(cursorId), 0) FROM root_events) AS streamHead" : ""}
+FROM page_total
+LEFT JOIN page_rows ON TRUE
+ORDER BY page_rows.__row_position ASC NULLS LAST`,
+			values
+		};
+	}
 	let pageCondition = "";
 	if (plan.cursor) {
 		pageCondition = `WHERE (${orderField} ${plan.orderBy.direction === "asc" ? ">" : "<"} CAST(? AS TIMESTAMP) OR (${orderField} = CAST(? AS TIMESTAMP) AND traceId > ?))`;
@@ -3344,9 +3391,44 @@ function asIsoTimestamp(value) {
 	return value instanceof Date ? value.toISOString() : new Date(value).toISOString();
 }
 async function queryTraces(db, plan) {
+	if (plan.paginationMode === "delta") assertDeltaPollingEnabled();
+	if (plan.paginationMode === "page") {
+		const query = compileDuckDBTraceQuery(plan);
+		const rows = await db.query(query.sql, query.values);
+		const total = Number(rows[0]?.total ?? 0);
+		const traces = rows.filter((row) => row.traceId != null).map((row) => ({
+			traceId: String(row.traceId),
+			rootSpanId: String(row.rootSpanId),
+			name: row.name,
+			entityId: row.entityId ?? null,
+			parentSpanId: row.parentSpanId ?? null,
+			createdAt: asIsoTimestamp(row.startedAt),
+			metadata: parseJson(row.metadata) ?? null,
+			inputPreview: coreStorage.buildInputPreview(row.input) ?? null,
+			threadId: row.threadId == null ? null : String(row.threadId),
+			resourceId: row.resourceId == null ? null : String(row.resourceId),
+			startedAt: asIsoTimestamp(row.startedAt),
+			endedAt: asIsoTimestamp(row.endedAt),
+			entityName: row.entityName == null ? null : String(row.entityName),
+			entityType: row.entityType == null ? null : String(row.entityType),
+			environment: row.environment == null ? null : String(row.environment),
+			status: row.status
+		}));
+		return coreStorage.traceQueryResponseSchema.parse({
+			traces,
+			...deltaPollingFeatureEnabled() && typeof coreStorage.encodeTraceQueryDeltaCursor === "function" ? { deltaCursor: coreStorage.encodeTraceQueryDeltaCursor(plan, "duckdb", String(rows[0]?.streamHead ?? 0)) } : {},
+			pagination: {
+				total,
+				page: plan.page,
+				perPage: plan.perPage,
+				hasMore: (plan.page + 1) * plan.perPage < total
+			}
+		});
+	}
 	const query = compileDuckDBTraceQuery(plan);
 	const rows = await db.query(query.sql, query.values);
-	const visibleRows = rows.slice(0, plan.limit);
+	const matchingRows = plan.paginationMode === "delta" ? rows.filter((row) => row.traceId != null) : rows;
+	const visibleRows = matchingRows.slice(0, plan.limit);
 	if (plan.result === "groups") {
 		const groups = visibleRows.map((row) => ({ threadId: String(row.threadId) }));
 		const last = groups.at(-1);
@@ -3376,6 +3458,19 @@ async function queryTraces(db, plan) {
 		environment: row.environment == null ? null : String(row.environment),
 		status: row.status
 	}));
+	if (plan.paginationMode === "delta") {
+		const previous = coreStorage.getTraceQueryDeltaWatermark(plan, "duckdb") ?? "0";
+		const head = String(rows[0]?.streamHead ?? 0);
+		const watermark = visibleRows.length ? String(visibleRows.at(-1).deltaWatermark) : BigInt(head) > BigInt(previous) ? head : previous;
+		return coreStorage.traceQueryResponseSchema.parse({
+			traces,
+			delta: {
+				limit: plan.limit,
+				hasMore: matchingRows.length > plan.limit
+			},
+			deltaCursor: coreStorage.encodeTraceQueryDeltaCursor(plan, "duckdb", watermark)
+		});
+	}
 	const last = traces.at(-1);
 	return coreStorage.traceQueryResponseSchema.parse({
 		traces,
@@ -4850,4 +4945,4 @@ var ObservabilityStorageDuckDB = class ObservabilityStorageDuckDB extends Observ
 //#endregion
 export { ObservabilityStorageDuckDB };
 
-//# sourceMappingURL=observability-W6La6xQK.js.map
+//# sourceMappingURL=observability-D3311DPa.js.map