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

package/dist/docs/SKILL.md

@@ -3,7 +3,7 @@ name: mastra-clickhouse
 description: Documentation for @mastra/clickhouse. Use when working with @mastra/clickhouse APIs, configuration, or implementation.
 metadata:
   package: "@mastra/clickhouse"
-  version: "1.27.0-alpha.1"
+  version: "1.27.0-alpha.2"
 ---
 
 ## When to use

package/dist/docs/assets/SOURCE_MAP.json

@@ -1,5 +1,5 @@
 {
-  "version": "1.27.0-alpha.1",
+  "version": "1.27.0-alpha.2",
   "package": "@mastra/clickhouse",
   "exports": {},
   "modules": {}

package/dist/index.cjs

@@ -8588,7 +8588,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}
@@ -8608,31 +8608,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 _clickhouse_client.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 [_mastra_core_storage.threadQueryRowSchema.parse({
 			organizationId: row.organizationId,
 			threadId: row.threadId,
@@ -8644,8 +8623,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