@mastra/duckdb 1.2.0 → 1.3.0-alpha.0

package/dist/{observability-AILZGFQT.cjs → observability-YJW7WSXX.cjs}

@@ -1188,7 +1188,16 @@ async function listLogs(db, args) {
     logs
   };
 }
-function getAggregationSql2(aggregation, measure = "value") {
+function resolveDistinctColumnSql(distinctColumn) {
+  if (!distinctColumn) {
+    throw new Error(`count_distinct aggregation requires a 'distinctColumn' argument`);
+  }
+  if (!storage.METRIC_DISTINCT_COLUMNS.includes(distinctColumn)) {
+    throw new Error(`Invalid distinctColumn: ${distinctColumn}`);
+  }
+  return utils.parseFieldKey(distinctColumn);
+}
+function getAggregationSql2(aggregation, measure = "value", distinctColumn) {
   switch (aggregation) {
     case "sum":
       return `SUM(${measure})`;
@@ -1200,6 +1209,9 @@ function getAggregationSql2(aggregation, measure = "value") {
       return `MAX(${measure})`;
     case "count":
       return `CAST(COUNT(${measure}) AS DOUBLE)`;
+    case "count_distinct": {
+      return `CAST(approx_count_distinct(${resolveDistinctColumnSql(distinctColumn)}) AS DOUBLE)`;
+    }
     case "last":
       return `arg_max(${measure}, timestamp)`;
     default:
@@ -1445,7 +1457,7 @@ async function listMetrics(db, args) {
   };
 }
 async function getMetricAggregate(db, args) {
-  const aggSql = getAggregationSql2(args.aggregation);
+  const aggSql = getAggregationSql2(args.aggregation, "value", args.distinctColumn);
   const { clause: nameClause, params: nameParams } = buildMetricNameFilter(args.name);
   const { clause: filterClause, params: filterParams } = buildWhereClause(
     args.filters
@@ -1529,7 +1541,7 @@ async function getMetricAggregate(db, args) {
   return { value, estimatedCost: costSummary.estimatedCost, costUnit: costSummary.costUnit };
 }
 async function getMetricBreakdown(db, args) {
-  const aggSql = getAggregationSql2(args.aggregation);
+  const aggSql = getAggregationSql2(args.aggregation, "value", args.distinctColumn);
   const { clause: nameClause, params: nameParams } = buildMetricNameFilter(args.name);
   const { clause: filterClause, params: filterParams } = buildWhereClause(
     args.filters
@@ -1543,8 +1555,11 @@ async function getMetricBreakdown(db, args) {
   const resolvedGroupBy = resolveGroupBy(args.groupBy);
   const selectGroupBy = resolvedGroupBy.map((entry) => entry.selectSql).join(", ");
   const groupByCols = resolvedGroupBy.map((entry) => entry.groupSql).join(", ");
-  const sql = `SELECT ${selectGroupBy}, ${aggSql} AS value, ${getCostSummarySelect()} FROM metric_events ${whereClause} GROUP BY ${groupByCols} ORDER BY value DESC`;
-  const rows = await db.query(sql, allParams);
+  const orderDirection = args.orderDirection === "ASC" ? "ASC" : "DESC";
+  const limitClause = typeof args.limit === "number" ? `LIMIT ?` : "";
+  const limitParams = typeof args.limit === "number" ? [args.limit] : [];
+  const sql = `SELECT ${selectGroupBy}, ${aggSql} AS value, ${getCostSummarySelect()} FROM metric_events ${whereClause} GROUP BY ${groupByCols} ORDER BY value ${orderDirection} ${limitClause}`;
+  const rows = await db.query(sql, [...allParams, ...limitParams]);
   const groups = rows.map((row) => {
     const dimensions = {};
     for (const entry of resolvedGroupBy) {
@@ -1562,7 +1577,7 @@ async function getMetricBreakdown(db, args) {
   return { groups };
 }
 async function getMetricTimeSeries(db, args) {
-  const aggSql = getAggregationSql2(args.aggregation);
+  const aggSql = getAggregationSql2(args.aggregation, "value", args.distinctColumn);
   const intervalSql = getIntervalSql2(args.interval);
   const { clause: nameClause, params: nameParams } = buildMetricNameFilter(args.name);
   const { clause: filterClause, params: filterParams } = buildWhereClause(
@@ -2146,6 +2161,12 @@ async function listScores(db, args) {
     scores: rows.map((row) => rowToScoreRecord(row))
   };
 }
+async function getScoreById(db, scoreId) {
+  const rows = await db.query(`SELECT * FROM score_events WHERE scoreId = ? LIMIT 1`, [
+    scoreId
+  ]);
+  return rows[0] ? rowToScoreRecord(rows[0]) : null;
+}
 async function getScoreAggregate(db, args) {
   const aggSql = getAggregationSql3(args.aggregation);
   const { clause, params } = buildScoreWhereClause(args);
@@ -2639,6 +2660,82 @@ async function listTraces(db, args) {
     spans: storage.toTraceSpans(spans)
   };
 }
+var BRANCH_SPAN_TYPE_PLACEHOLDERS = storage.BRANCH_SPAN_TYPES.map(() => "?").join(", ");
+async function getSpans(db, args) {
+  if (args.spanIds.length === 0) {
+    return { traceId: args.traceId, spans: [] };
+  }
+  const placeholders = args.spanIds.map(() => "?").join(", ");
+  const rows = await db.query(
+    `${SPAN_RECONSTRUCT_SELECT}
+     WHERE traceId = ? AND spanId IN (${placeholders})
+     GROUP BY traceId, spanId`,
+    [args.traceId, ...args.spanIds]
+  );
+  return {
+    traceId: args.traceId,
+    spans: rows.map((row) => rowToSpanRecord(row))
+  };
+}
+async function listBranches(db, args) {
+  const filters = args.filters ?? {};
+  const page = Number(args.pagination?.page ?? 0);
+  const perPage = Number(args.pagination?.perPage ?? 10);
+  const orderBy = { field: args.orderBy?.field ?? "startedAt", direction: args.orderBy?.direction ?? "DESC" };
+  const { spanType, ...passthroughFilters } = filters;
+  const { clause: filterClause, params: filterParams } = buildWhereClause(passthroughFilters);
+  const orderByClause = buildOrderByClause(orderBy);
+  const { clause: paginationClause, params: paginationParams } = buildPaginationClause({ page, perPage });
+  let prefilterClause;
+  let prefilterParams;
+  if (typeof spanType === "string") {
+    if (!storage.BRANCH_SPAN_TYPES.includes(spanType)) {
+      return {
+        pagination: { total: 0, page, perPage, hasMore: false },
+        branches: []
+      };
+    }
+    prefilterClause = `WHERE spanType = ?`;
+    prefilterParams = [spanType];
+  } else {
+    prefilterClause = `WHERE spanType IN (${BRANCH_SPAN_TYPE_PLACEHOLDERS})`;
+    prefilterParams = [...storage.BRANCH_SPAN_TYPES];
+  }
+  const cteSql = `
+    WITH branch_anchors AS (
+      ${SPAN_RECONSTRUCT_SELECT}
+      ${prefilterClause}
+      GROUP BY traceId, spanId
+    )
+  `;
+  const countSql = `
+    ${cteSql}
+    SELECT COUNT(*) as total FROM branch_anchors ${filterClause}
+  `;
+  const countResult = await db.query(countSql, [...prefilterParams, ...filterParams]);
+  const total = Number(countResult[0]?.total ?? 0);
+  if (total === 0) {
+    return {
+      pagination: { total: 0, page, perPage, hasMore: false },
+      branches: []
+    };
+  }
+  const dataSql = `
+    ${cteSql}
+    SELECT * FROM branch_anchors ${filterClause} ${orderByClause} ${paginationClause}
+  `;
+  const rows = await db.query(dataSql, [...prefilterParams, ...filterParams, ...paginationParams]);
+  const spans = rows.map((row) => rowToSpanRecord(row));
+  return {
+    pagination: {
+      total,
+      page,
+      perPage,
+      hasMore: (page + 1) * perPage < total
+    },
+    branches: storage.toTraceSpans(spans)
+  };
+}
 
 // src/storage/domains/observability/index.ts
 function buildSignalMigrationRequiredMessage(args) {
@@ -2741,6 +2838,9 @@ var ObservabilityStorageDuckDB = class extends storage.ObservabilityStorage {
   async getSpan(args) {
     return getSpan(this.db, args);
   }
+  async getSpans(args) {
+    return getSpans(this.db, args);
+  }
   async getRootSpan(args) {
     return getRootSpan(this.db, args);
   }
@@ -2753,6 +2853,9 @@ var ObservabilityStorageDuckDB = class extends storage.ObservabilityStorage {
   async listTraces(args) {
     return listTraces(this.db, args);
   }
+  async listBranches(args) {
+    return listBranches(this.db, args);
+  }
   // Logs
   async batchCreateLogs(args) {
     return batchCreateLogs(this.db, args);
@@ -2815,6 +2918,9 @@ var ObservabilityStorageDuckDB = class extends storage.ObservabilityStorage {
   async listScores(args) {
     return listScores(this.db, args);
   }
+  async getScoreById(scoreId) {
+    return getScoreById(this.db, scoreId);
+  }
   async getScoreAggregate(args) {
     return getScoreAggregate(this.db, args);
   }
@@ -2852,5 +2958,5 @@ var ObservabilityStorageDuckDB = class extends storage.ObservabilityStorage {
 };
 
 exports.ObservabilityStorageDuckDB = ObservabilityStorageDuckDB;
-//# sourceMappingURL=observability-AILZGFQT.cjs.map
-//# sourceMappingURL=observability-AILZGFQT.cjs.map
+//# sourceMappingURL=observability-YJW7WSXX.cjs.map
+//# sourceMappingURL=observability-YJW7WSXX.cjs.map