@mastra/duckdb 1.3.1 → 1.3.2

package/dist/{observability-MKVTGAKX.js → observability-TS5QIHIC.js}

@@ -2692,7 +2692,7 @@ async function getTraceLight(db, args) {
     spans: rows.map((row) => rowToLightSpanRecord(row))
   };
 }
-async function listTraces(db, args) {
+async function listTraceRows(db, args, reconstructSelect, mapRow, toSpans) {
   const filters = args.filters ?? {};
   const page = Number(args.pagination?.page ?? 0);
   const perPage = Number(args.pagination?.perPage ?? 10);
@@ -2727,16 +2727,16 @@ async function listTraces(db, args) {
         ${prefilterOrderBy}
         LIMIT ? OFFSET ?
       )
-      ${SPAN_RECONSTRUCT_SELECT}
+      ${reconstructSelect}
       WHERE (traceId, spanId) IN (SELECT traceId, spanId FROM page_roots)
       GROUP BY traceId, spanId
       ${buildOrderByClause(orderBy)}
     `;
     const rows2 = await db.query(pageSql, [...prefilterParams, perPage, offset]);
-    const spans2 = rows2.map((row) => rowToSpanRecord(row));
+    const spans2 = rows2.map((row) => mapRow(row));
     return {
       pagination: { total: total2, page, perPage, hasMore: (page + 1) * perPage < total2 },
-      spans: toTraceSpans(spans2)
+      spans: toSpans(spans2)
     };
   }
   const { clause: postAggClause, params: postAggParams } = buildWhereClause(postAgg);
@@ -2752,7 +2752,7 @@ async function listTraces(db, args) {
       ${prefilterWhere}
     ),
     root_spans AS (
-      ${SPAN_RECONSTRUCT_SELECT}
+      ${reconstructSelect}
       WHERE (traceId, spanId) IN (SELECT traceId, spanId FROM candidate_roots)
       GROUP BY traceId, spanId
     )
@@ -2770,12 +2770,30 @@ async function listTraces(db, args) {
     SELECT * FROM root_spans ${postAggWhere} ${orderByClause} ${paginationClause}
   `;
   const rows = await db.query(dataSql, [...prefilterParams, ...postAggParams, ...paginationParams]);
-  const spans = rows.map((row) => rowToSpanRecord(row));
+  const spans = rows.map((row) => mapRow(row));
   return {
     pagination: { total, page, perPage, hasMore: (page + 1) * perPage < total },
-    spans: toTraceSpans(spans)
+    spans: toSpans(spans)
   };
 }
+async function listTraces(db, args) {
+  return listTraceRows(
+    db,
+    args,
+    SPAN_RECONSTRUCT_SELECT,
+    rowToSpanRecord,
+    (spans) => toTraceSpans(spans)
+  );
+}
+async function listTracesLight(db, args) {
+  return listTraceRows(
+    db,
+    args,
+    SPAN_RECONSTRUCT_SELECT_LIGHT,
+    rowToLightSpanRecord,
+    (spans) => spans
+  );
+}
 var BRANCH_SPAN_TYPE_PLACEHOLDERS = BRANCH_SPAN_TYPES.map(() => "?").join(", ");
 async function getSpans(db, args) {
   if (args.spanIds.length === 0) {
@@ -3014,6 +3032,9 @@ var ObservabilityStorageDuckDB = class extends ObservabilityStorage {
   async listTraces(args) {
     return listTraces(this.db, args);
   }
+  async listTracesLight(args) {
+    return listTracesLight(this.db, args);
+  }
   async listBranches(args) {
     return listBranches(this.db, args);
   }
@@ -3119,5 +3140,5 @@ var ObservabilityStorageDuckDB = class extends ObservabilityStorage {
 };
 
 export { ObservabilityStorageDuckDB };
-//# sourceMappingURL=observability-MKVTGAKX.js.map
-//# sourceMappingURL=observability-MKVTGAKX.js.map
+//# sourceMappingURL=observability-TS5QIHIC.js.map
+//# sourceMappingURL=observability-TS5QIHIC.js.map