@mastra/clickhouse 1.27.0-alpha.1 → 1.27.0-alpha.2

package/dist/index.js

@@ -8566,7 +8566,7 @@ async function queryThreadRows(client, plan, timeoutMs, strategy = "auto") {
     traceId, spanId, startedAt, endedAt) IN {roots:Array(Tuple(UInt8, String, UInt8, String, String, String, DateTime64(3, 'UTC'), DateTime64(3, 'UTC')))}`;
 	const stored = [];
 	for (const batch of batches([...roots.values()])) stored.push(...await run({
-		query: `SELECT organizationId, traceId, spanId, startedAt, endedAt, spanType,
+		query: `SELECT organizationId, traceId, spanId, startedAt, endedAt,
       ${contextFields.join(", ")}, if(spanType = 'agent_run', timeToFirstChunkMs, NULL) AS timeToFirstChunkMs,
       if(isNull(error), 'success', 'error') AS status, ${PREVIEW_COLUMNS.join(", ")}
       FROM ${TABLE_TRACE_ROOTS} WHERE ${rootWhere}
@@ -8586,31 +8586,10 @@ async function queryThreadRows(client, plan, timeoutMs, strategy = "auto") {
 			if (row) Object.assign(row, resolveStoredPreviews(row, payload));
 		}
 	}
-	const firstRoots = page.map((row) => representative(row, "first")).filter((root) => byRoot.get(rootKey(root))?.spanType === "agent_run");
-	const models = [];
-	for (const batch of batches(firstRoots)) models.push(...await run({
-		query: `SELECT organizationId, traceId,
-      if(JSONType(attributes, 'model') = 'String', JSONExtractString(attributes, 'model'), NULL) AS model,
-      if(JSONType(attributes, 'provider') = 'String', JSONExtractString(attributes, 'provider'), NULL) AS provider
-      FROM ${TABLE_SPAN_EVENTS} WHERE spanType = 'model_generation' AND
-      tuple(isNull(organizationId), ifNull(organizationId, ''), isNull(resourceId), ifNull(resourceId, ''), traceId, parentSpanId)
-        IN {agents:Array(Tuple(UInt8, String, UInt8, String, String, String))}
-      ORDER BY organizationId, traceId, startedAt, spanId LIMIT 1 BY organizationId, traceId`,
-		query_params: { agents: batch.map((root) => new TupleParam([
-			root.organizationId === null ? 1 : 0,
-			root.organizationId ?? "",
-			root.resourceId === null ? 1 : 0,
-			root.resourceId ?? "",
-			root.traceId,
-			root.spanId
-		])) }
-	}));
-	const byTrace = new Map(models.map((row) => [JSON.stringify([row.organizationId, row.traceId]), row]));
 	const threads = page.flatMap((row) => {
 		const first = byRoot.get(rootKey(representative(row, "first")));
 		const last = byRoot.get(rootKey(representative(row, "last")));
 		if (!first || !last) return [];
-		const model = byTrace.get(JSON.stringify([row.organizationId, row.firstTraceId]));
 		return [coreStorage.threadQueryRowSchema.parse({
 			organizationId: row.organizationId,
 			threadId: row.threadId,
@@ -8622,8 +8601,6 @@ async function queryThreadRows(client, plan, timeoutMs, strategy = "auto") {
 			endedAt: iso(row.endedAt),
 			traceCount: Number(row.traceCount),
 			timeToFirstChunkMs: first.timeToFirstChunkMs,
-			model: model?.model ?? null,
-			provider: model?.provider ?? null,
 			inputPreview: resolveStoredPreviews(first).inputPreview,
 			outputPreview: resolveStoredPreviews(last).outputPreview,
 			errorPreview: resolveStoredPreviews(last).errorPreview