@mastra/duckdb 1.6.0 → 1.6.1

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -1,6 +1,6 @@
1
1
  import { t as DuckDBConnection } from "./db-Dl0fY488.js";
2
2
  import { ErrorCategory, ErrorDomain, MastraError } from "@mastra/core/error";
3
- import { BRANCH_SPAN_TYPES, METRIC_DISTINCT_COLUMNS, ObservabilityStorage, createStorageErrorId, listBranchesArgsSchema, listFeedbackArgsSchema, listLogsArgsSchema, listMetricsArgsSchema, listScoresArgsSchema, listTracesArgsSchema, toTraceSpans } from "@mastra/core/storage";
3
+ import { BRANCH_SPAN_TYPES, METRIC_DISTINCT_COLUMNS, ObservabilityStorage, buildInputPreview, computeTraceStatus, createStorageErrorId, listBranchesArgsSchema, listFeedbackArgsSchema, listLogsArgsSchema, listMetricsArgsSchema, listScoresArgsSchema, listTracesArgsSchema, toTraceSpans } from "@mastra/core/storage";
4
4
  import { coreFeatures } from "@mastra/core/features";
5
5
  import { EntityType } from "@mastra/core/observability";
6
6
  import { parseFieldKey } from "@mastra/core/utils";
@@ -2538,6 +2538,28 @@ const SPAN_RECONSTRUCT_SELECT_LIGHT = `
2538
2538
  FROM span_events
2539
2539
  `;
2540
2540
  /**
2541
+ * Lightweight list variant — also reconstructs `metadata` for the list's
2542
+ * configurable columns and `input` so the row mapper can derive `inputPreview`
2543
+ * without shipping the blob to the caller.
2544
+ */
2545
+ const SPAN_RECONSTRUCT_SELECT_LIGHT_LIST = `
2546
+ SELECT
2547
+ traceId, spanId,
2548
+ ${argMaxNonNull("name")},
2549
+ ${argMaxNonNull("spanType")},
2550
+ ${argMaxNonNull("parentSpanId")},
2551
+ ${argMaxNonNull("isEvent")},
2552
+ coalesce(min(timestamp) FILTER (WHERE eventType = 'start'), min(timestamp)) as startedAt,
2553
+ ${argMaxNonNull("endedAt")},
2554
+ ${argMaxNonNull("entityType")},
2555
+ ${argMaxNonNull("entityId")},
2556
+ ${argMaxNonNull("entityName")},
2557
+ ${argMaxNonNull("error")},
2558
+ ${argMaxNonNull("metadata")},
2559
+ ${argMaxNonNull("input")}
2560
+ FROM span_events
2561
+ `;
2562
+ /**
2541
2563
  * Which reconstructed columns each post-aggregation filter key can reference.
2542
2564
  * `status` is derived from endedAt + error (see buildWhereClause).
2543
2565
  */
@@ -2621,6 +2643,15 @@ function rowToLightSpanRecord(row) {
2621
2643
  updatedAt: toDateOrNull(row.endedAt)
2622
2644
  };
2623
2645
  }
2646
+ function rowToLightSpanRecordWithPreview(row) {
2647
+ const record = rowToLightSpanRecord(row);
2648
+ return {
2649
+ ...record,
2650
+ status: computeTraceStatus(record),
2651
+ metadata: parseJson(row.metadata),
2652
+ inputPreview: buildInputPreview(row.input)
2653
+ };
2654
+ }
2624
2655
  function rowToSpanRecord(row) {
2625
2656
  return {
2626
2657
  traceId: row.traceId,
@@ -3152,7 +3183,18 @@ async function listTraces(db, args) {
3152
3183
  };
3153
3184
  }
3154
3185
  async function listTracesLight(db, args) {
3155
- return listTraceRows(db, args, SPAN_RECONSTRUCT_SELECT_LIGHT, rowToLightSpanRecord, (spans) => spans);
3186
+ const { filters, pagination, orderBy } = listTracesArgsSchema.parse(args);
3187
+ const currentDeltaCursor = deltaPollingFeatureEnabled() ? await getTraceDeltaCursor(db, filters) : void 0;
3188
+ const { pagination: resultPagination, spans } = await listTraceRows(db, {
3189
+ filters,
3190
+ pagination,
3191
+ orderBy
3192
+ }, SPAN_RECONSTRUCT_SELECT_LIGHT_LIST, rowToLightSpanRecordWithPreview, (spans) => spans);
3193
+ return {
3194
+ pagination: resultPagination,
3195
+ spans,
3196
+ ...deltaPollingFeatureEnabled() ? { deltaCursor: currentDeltaCursor } : {}
3197
+ };
3156
3198
  }
3157
3199
  const BRANCH_SPAN_TYPE_PLACEHOLDERS = BRANCH_SPAN_TYPES.map(() => "?").join(", ");
3158
3200
  /**
@@ -3628,6 +3670,7 @@ var ObservabilityStorageDuckDB = class extends ObservabilityStorage {
3628
3670
  return listTraces(this.db, args);
3629
3671
  }
3630
3672
  async listTracesLight(args) {
3673
+ if (args.mode === "delta") return super.listTracesLight(args);
3631
3674
  return listTracesLight(this.db, args);
3632
3675
  }
3633
3676
  async listBranches(args) {
@@ -3730,4 +3773,4 @@ var ObservabilityStorageDuckDB = class extends ObservabilityStorage {
3730
3773
  //#endregion
3731
3774
  export { ObservabilityStorageDuckDB };
3732
3775
 
3733
- //# sourceMappingURL=observability-5ro3rotr.js.map
3776
+ //# sourceMappingURL=observability-BtPeWunq.js.map