@mastra/duckdb 1.9.0 → 1.10.0-alpha.0

package/dist/{observability-BB4mjnZR.cjs → observability-DT-R9Xkn.cjs}

@@ -3235,6 +3235,93 @@ LIMIT ?`,
 		values
 	};
 }
+function discoveryRegistry(scope) {
+	if (scope === "trace") return TRACE_FIELDS;
+	if (scope === "spans") return SPAN_FIELDS;
+	if (scope === "scores") return SCORE_FIELDS;
+	return FEEDBACK_FIELDS;
+}
+function discoverySource(scope) {
+	if (scope === "trace") return "root_scope r";
+	if (scope === "spans") return "current_spans s";
+	if (scope === "scores") return "current_scores s";
+	return "current_feedback s";
+}
+function discoveryCollections(scope) {
+	return scope === "trace" ? /* @__PURE__ */ new Set() : /* @__PURE__ */ new Set([scope]);
+}
+function compileDuckDBTraceQueryObservedFields(plan) {
+	const values = [plan.timeRange.from, plan.timeRange.to];
+	if (plan.search) values.push(plan.search);
+	values.push(plan.limit + 1);
+	const search = plan.search ? `AND strpos(lower('metadata.' || entry.key), lower(?)) > 0` : "";
+	return {
+		sql: `WITH ${compileDuckDBTraceScope(/* @__PURE__ */ new Set()).join(",\n  ")}
+SELECT 'metadata.' || entry.key AS path, count(*) AS occurrences
+FROM root_scope r, LATERAL json_each(r.metadata) entry
+WHERE entry.type = 'VARCHAR'
+  AND trim(json_extract_string(entry.value, '$')) <> ''
+  AND entry.key <> ''
+  AND strpos(entry.key, '.') = 0
+  AND octet_length(encode('metadata.' || entry.key)) <= ${_mastra_core_storage.TRACE_QUERY_MAX_PATH_BYTES}
+  AND octet_length(encode(json_extract_string(entry.value, '$'))) <= ${_mastra_core_storage.TRACE_QUERY_MAX_STRING_BYTES}
+  ${search}
+GROUP BY entry.key
+ORDER BY occurrences DESC, path ASC
+LIMIT ?`,
+		values
+	};
+}
+function compileDuckDBTraceQueryValues(plan) {
+	const values = [plan.timeRange.from, plan.timeRange.to];
+	const ctes = compileDuckDBTraceScope(discoveryCollections(plan.predicateScope));
+	let fieldSql;
+	if (plan.predicateScope === "trace" && plan.path.startsWith("metadata.")) {
+		const jsonPath = `$.${JSON.stringify(plan.path.slice(9))}`;
+		fieldSql = `NULLIF(trim(CASE WHEN json_type(r.metadata, ?) = 'VARCHAR' THEN json_extract_string(r.metadata, ?) END), '')`;
+		values.push(jsonPath, jsonPath);
+	} else fieldSql = fieldDefinition(discoveryRegistry(plan.predicateScope), plan.path).sql;
+	if (plan.search) values.push(plan.search);
+	values.push(plan.limit + 1);
+	const search = plan.search ? "AND strpos(lower(CAST(value AS VARCHAR)), lower(?)) > 0" : "";
+	return {
+		sql: `WITH ${ctes.join(",\n  ")}, extracted AS (
+  SELECT ${fieldSql} AS value FROM ${discoverySource(plan.predicateScope)}
+)
+SELECT CAST(value AS VARCHAR) AS value, count(*) AS count
+FROM extracted
+WHERE value IS NOT NULL
+  AND octet_length(encode(CAST(value AS VARCHAR))) <= ${_mastra_core_storage.TRACE_QUERY_MAX_STRING_BYTES}
+  ${search}
+GROUP BY value
+ORDER BY count DESC, value ASC
+LIMIT ?`,
+		values
+	};
+}
+async function getTraceQueryObservedFields(db, plan) {
+	if (plan.predicateScope !== "trace") return {
+		observedFields: [],
+		observedFieldsTruncated: false
+	};
+	const query = compileDuckDBTraceQueryObservedFields(plan);
+	const rows = await db.query(query.sql, query.values);
+	return {
+		observedFields: rows.slice(0, plan.limit).map((row) => _mastra_core_storage.createTraceQueryObservedFieldDescriptor(String(row.path), Number(row.occurrences))),
+		observedFieldsTruncated: rows.length > plan.limit
+	};
+}
+async function getTraceQueryValues(db, plan) {
+	const query = compileDuckDBTraceQueryValues(plan);
+	const rows = await db.query(query.sql, query.values);
+	return _mastra_core_storage.getTraceQueryValuesResponseSchema.parse({
+		values: rows.slice(0, plan.limit).map((row) => ({
+			value: String(row.value),
+			count: Number(row.count)
+		})),
+		valuesTruncated: rows.length > plan.limit
+	});
+}
 function asIsoTimestamp(value) {
 	return value instanceof Date ? value.toISOString() : new Date(value).toISOString();
 }
@@ -4528,6 +4615,7 @@ var ObservabilityStorageDuckDB = class extends _mastra_core_storage.Observabilit
 			"metrics",
 			"logs",
 			"trace-query",
+			"trace-query-discovery",
 			"thread-query"
 		];
 		return [
@@ -4535,6 +4623,7 @@ var ObservabilityStorageDuckDB = class extends _mastra_core_storage.Observabilit
 			"logs",
 			"delta-polling",
 			"trace-query",
+			"trace-query-discovery",
 			"thread-query"
 		];
 	}
@@ -4568,6 +4657,12 @@ var ObservabilityStorageDuckDB = class extends _mastra_core_storage.Observabilit
 	async queryTraces(plan) {
 		return queryTraces(this.db, plan);
 	}
+	async getTraceQueryObservedFields(plan) {
+		return getTraceQueryObservedFields(this.db, plan);
+	}
+	async getTraceQueryValues(plan) {
+		return getTraceQueryValues(this.db, plan);
+	}
 	async queryThreads(plan) {
 		return queryThreads(this.db, plan);
 	}
@@ -4684,4 +4779,4 @@ var ObservabilityStorageDuckDB = class extends _mastra_core_storage.Observabilit
 //#endregion
 exports.ObservabilityStorageDuckDB = ObservabilityStorageDuckDB;
 
-//# sourceMappingURL=observability-BB4mjnZR.cjs.map
+//# sourceMappingURL=observability-DT-R9Xkn.cjs.map