@mastra/clickhouse 1.7.1 → 1.7.2

package/dist/index.js

@@ -3685,6 +3685,26 @@ function rowToSpanRecord(row) {
     updatedAt: null
   };
 }
+function rowToLightSpanRecord(row) {
+  const startedAt = toDate(row.startedAt);
+  const endedAt = row.isEvent ? startedAt : toDateOrNull(row.endedAt);
+  return {
+    traceId: row.traceId,
+    spanId: row.spanId,
+    parentSpanId: nullableString(row.parentSpanId),
+    name: row.name,
+    spanType: row.spanType,
+    isEvent: Boolean(row.isEvent),
+    startedAt,
+    endedAt,
+    entityType: nullableEntityType(row.entityType),
+    entityId: nullableString(row.entityId),
+    entityName: nullableString(row.entityName),
+    error: parseJson(row.error) ?? void 0,
+    createdAt: startedAt,
+    updatedAt: null
+  };
+}
 function spanRecordToRow(span) {
   const endedAt = span.isEvent ? span.startedAt : span.endedAt ?? span.startedAt;
   const metadata = span.metadata ?? null;
@@ -5587,7 +5607,7 @@ async function getRootSpan(client, args) {
   if (!rows || rows.length === 0) return null;
   return { span: rowToSpanRecord(rows[0]) };
 }
-async function listTraces(client, args) {
+async function listTraceRows(client, args, selectClause, mapRows) {
   const { filters, pagination, orderBy } = listTracesArgsSchema.parse(args);
   const page = pagination?.page ?? 0;
   const perPage = pagination?.perPage ?? 10;
@@ -5635,8 +5655,8 @@ async function listTraces(client, args) {
   }
   const dataResult = await client.query({
     query: `
-      SELECT * FROM (
-        SELECT *
+      SELECT ${selectClause} FROM (
+        SELECT ${selectClause}
         FROM ${TABLE_TRACE_ROOTS} r
         ${whereClause}
         ORDER BY dedupeKey
@@ -5655,7 +5675,6 @@ async function listTraces(client, args) {
     clickhouse_settings: CH_SETTINGS
   });
   const rows = await dataResult.json();
-  const spans = rows.map(rowToSpanRecord);
   return {
     pagination: {
       total,
@@ -5663,9 +5682,29 @@ async function listTraces(client, args) {
       perPage,
       hasMore: (page + 1) * perPage < total
     },
-    spans: toTraceSpans(spans)
+    spans: mapRows(rows)
   };
 }
+async function listTraces(client, args) {
+  return listTraceRows(client, args, "*", (rows) => toTraceSpans(rows.map(rowToSpanRecord)));
+}
+var LIGHT_TRACE_ROOT_COLUMNS = [
+  "traceId",
+  "spanId",
+  "parentSpanId",
+  "name",
+  "spanType",
+  "isEvent",
+  "startedAt",
+  "endedAt",
+  "entityType",
+  "entityId",
+  "entityName",
+  "error"
+].join(", ");
+async function listTracesLight(client, args) {
+  return listTraceRows(client, args, LIGHT_TRACE_ROOT_COLUMNS, (rows) => rows.map(rowToLightSpanRecord));
+}
 var BRANCH_SPAN_TYPE_SQL_LIST = BRANCH_SPAN_TYPES.map((t) => `'${t}'`).join(", ");
 async function createSpan(client, args) {
   const row = spanRecordToRow(args.span);
@@ -6286,6 +6325,21 @@ var ObservabilityStorageClickhouseVNext = class extends ObservabilityStorage {
       );
     }
   }
+  async listTracesLight(args) {
+    try {
+      return await listTracesLight(this.#client, args);
+    } catch (error) {
+      if (error instanceof MastraError) throw error;
+      throw new MastraError(
+        {
+          id: createStorageErrorId("CLICKHOUSE", "LIST_TRACES_LIGHT", "FAILED"),
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY
+        },
+        error
+      );
+    }
+  }
   async listBranches(args) {
     try {
       return await listBranches(this.#client, args);