@mastra/duckdb 1.16.0-alpha.0 → 1.16.0-alpha.2

package/dist/{observability-Bv3eObB-.js → observability-BWSYzhM2.js}

@@ -1,11 +1,11 @@
-import { a as timeToFirstChunkSupported, i as timeToFirstChunkFeatures, n as spanQueryFeatures, o as DuckDBConnection, s as DuckDBQueryTimeoutError, t as spanAggregateFeatures } from "./features-uOeZunqR.js";
+import { a as timeToFirstChunkSupported, i as timeToFirstChunkFeatures, n as spanQueryFeatures, o as DuckDBConnection, s as DuckDBQueryTimeoutError, t as spanAggregateFeatures } from "./features-B8gl38Ue.js";
 import { ErrorCategory, ErrorDomain, MastraError } from "@mastra/core/error";
 import * as coreStorage from "@mastra/core/storage";
 import { BRANCH_SPAN_TYPES, METRIC_DISTINCT_COLUMNS, ObservabilityStorage, computeTraceStatus, createStorageErrorId, executeRetentionPrune, feedbackRecordSchema, listBranchesArgsSchema, listFeedbackArgsSchema, listLogsArgsSchema, listMetricsArgsSchema, listScoresArgsSchema, listTracesArgsSchema, resolveRetentionTargets, retentionCutoffMs, toTraceSpans } from "@mastra/core/storage";
 import { coreFeatures } from "@mastra/core/features";
 import { EntityType } from "@mastra/core/observability";
 import { parseFieldKey } from "@mastra/core/utils";
-//#region src/storage/retention.ts
+// src/storage/retention.ts
 function cutoffFor(policy, now = Date.now()) {
 	return new Date(retentionCutoffMs(policy, now));
 }
@@ -30,8 +30,7 @@ function resolveTargets({ policies, descriptor, order }) {
 		order
 	});
 }
-//#endregion
-//#region src/storage/domains/observability/ddl.ts
+// src/storage/domains/observability/ddl.ts
 /**
 * DDL statements for DuckDB observability tables.
 * All tables use append-only patterns with a single `timestamp` column.
@@ -493,8 +492,7 @@ const ALL_MIGRATIONS = [
 	`ALTER TABLE feedback_events ADD COLUMN IF NOT EXISTS reviewStatus VARCHAR DEFAULT 'needs-review'`,
 	`ALTER TABLE feedback_events ALTER COLUMN traceId DROP NOT NULL`
 ];
-//#endregion
-//#region src/storage/domains/observability/discovery.ts
+// src/storage/domains/observability/discovery.ts
 function unionDistinctQueries(selects, orderBy) {
 	return `${selects.join("\nUNION\n")}\nORDER BY ${orderBy}`;
 }
@@ -562,8 +560,7 @@ async function getTags(db, args) {
 		buildSelect("log_events")
 	], "tag"), params)).map((r) => r.tag) };
 }
-//#endregion
-//#region src/storage/domains/observability/filters.ts
+// src/storage/domains/observability/filters.ts
 function buildJsonPath(key) {
 	try {
 		return `$.${parseFieldKey(key)}`;
@@ -713,8 +710,7 @@ function buildPaginationClause(pagination) {
 		params: [pagination.perPage, offset]
 	};
 }
-//#endregion
-//#region src/storage/domains/observability/preview-compat.ts
+// src/storage/domains/observability/preview-compat.ts
 const generator = coreStorage.buildStoredPreview;
 const canGeneratePreviews = typeof generator === "function";
 /** SQL detection only; preview formatting stays in TypeScript. Empty strings are intentional. */
@@ -771,8 +767,7 @@ function buildSpanQueryResponse(...args) {
 	}
 	return response;
 }
-//#endregion
-//#region src/storage/domains/observability/helpers.ts
+// src/storage/domains/observability/helpers.ts
 /** Shorthand for {@link DuckDBConnection.sqlValue}. */
 const v = DuckDBConnection.sqlValue;
 /** Serialize a value to JSON then SQL-escape it, or return 'NULL'. */
@@ -864,8 +859,7 @@ function previewFromPayload(value) {
 function apiPreview(value) {
 	return readPreview(value);
 }
-//#endregion
-//#region src/storage/domains/observability/polling.ts
+// src/storage/domains/observability/polling.ts
 const OBSERVABILITY_DELTA_POLLING_FEATURE = "observability-delta-polling";
 function deltaPollingFeatureEnabled() {
 	return coreFeatures.has(OBSERVABILITY_DELTA_POLLING_FEATURE);
@@ -897,8 +891,7 @@ function extendWhereClause(baseClause, extraConditions) {
 	if (!baseClause) return `WHERE ${conditions.join(" AND ")}`;
 	return `${baseClause} AND ${conditions.join(" AND ")}`;
 }
-//#endregion
-//#region src/storage/domains/observability/review-status.ts
+// src/storage/domains/observability/review-status.ts
 const FEEDBACK_REVIEW_STATUSES = ["needs-review", "reviewed"];
 function isFeedbackReviewStatus(value) {
 	return FEEDBACK_REVIEW_STATUSES.some((status) => status === value);
@@ -922,8 +915,7 @@ function parseUpdateFeedbackReviewStatusArgs(args) {
 		reviewStatus: args.reviewStatus
 	};
 }
-//#endregion
-//#region src/storage/domains/observability/feedback.ts
+// src/storage/domains/observability/feedback.ts
 const FEEDBACK_UPSERT_CLAUSE = `ON CONFLICT (feedbackId) DO UPDATE SET ${[
 	"timestamp",
 	"cursorId",
@@ -1452,8 +1444,7 @@ async function getFeedbackPercentiles(db, args) {
 	}
 	return { series };
 }
-//#endregion
-//#region src/storage/domains/observability/logs.ts
+// src/storage/domains/observability/logs.ts
 const COLUMNS_SQL$1 = [
 	"logId",
 	"timestamp",
@@ -1636,8 +1627,7 @@ async function getDeltaCursor$2(db, filterClause, filterParams) {
 async function getStreamHeadCursor$2(db) {
 	return encodeDeltaCursor((await db.query(`SELECT max(cursorId) AS cursorId FROM log_events`))[0]?.cursorId);
 }
-//#endregion
-//#region src/storage/domains/observability/metrics.ts
+// src/storage/domains/observability/metrics.ts
 const METRIC_FILTER_FIELD_MAPPINGS = {
 	traceIds: "traceId",
 	experimentIds: "experimentId"
@@ -2170,8 +2160,7 @@ async function getMetricLabelValues(db, args) {
 	if (args.limit) params.push(args.limit);
 	return { values: (await db.query(`SELECT DISTINCT json_extract_string(labels, ?) AS val FROM metric_events WHERE ${conditions.join(" AND ")} ORDER BY val ${limitClause}`, [labelPath, ...params])).map((r) => r.val) };
 }
-//#endregion
-//#region src/storage/domains/observability/migration.ts
+// src/storage/domains/observability/migration.ts
 const CURSOR_ID_TABLES = [
 	"span_events",
 	"metric_events",
@@ -2320,8 +2309,7 @@ async function migrateSignalTables(db, logger) {
 		}
 	}
 }
-//#endregion
-//#region src/storage/domains/observability/scores.ts
+// src/storage/domains/observability/scores.ts
 const SCORE_UPSERT_CLAUSE = `ON CONFLICT (scoreId) DO UPDATE SET ${[
 	"timestamp",
 	"traceId",
@@ -2832,8 +2820,7 @@ async function getScorePercentiles(db, args) {
 	}
 	return { series };
 }
-//#endregion
-//#region src/storage/domains/observability/trace-query.ts
+// src/storage/domains/observability/trace-query.ts
 const TRACE_STATUS_SQL = `CASE WHEN r.error IS NOT NULL THEN 'error' ELSE 'success' END`;
 function durationMsSql(startedAt, endedAt) {
 	return `date_diff('millisecond', ${startedAt}, ${endedAt})`;
@@ -3960,8 +3947,7 @@ function compileThreadQueryPredicate(predicate) {
 		parameterType: rule.valueKind === "timestamp" ? "timestamp" : "scalar"
 	}])));
 }
-//#endregion
-//#region src/storage/domains/observability/span-query.ts
+// src/storage/domains/observability/span-query.ts
 const identityFields = [
 	"organizationId",
 	"resourceId",
@@ -4178,8 +4164,7 @@ async function querySpans(db, plan) {
 	}
 	return buildSpanQueryResponse(plan, hydrated, payloads, metrics);
 }
-//#endregion
-//#region src/storage/domains/observability/thread-query.ts
+// src/storage/domains/observability/thread-query.ts
 const contextFields = [
 	"resourceId",
 	"userId",
@@ -4357,7 +4342,7 @@ async function queryThreadRows(db, plan, strategy = "auto") {
 			return `${field} ${field === "organizationId" || field === "resourceId" ? "IS NOT DISTINCT FROM" : "="} ?`;
 		}).join(" AND ")})`;
 	}));
-	const payloads = await run(`SELECT organizationId, traceId, cursorId, spanType,
+	const payloads = await run(`SELECT organizationId, traceId, cursorId,
     ${contextFields.map((field) => field === "executionSource" ? "source AS executionSource" : field).join(", ")},
     CASE WHEN spanType = 'agent_run' THEN timeToFirstChunkMs END AS timeToFirstChunkMs,
     CASE WHEN error IS NULL THEN 'success' ELSE 'error' END AS status,
@@ -4373,34 +4358,10 @@ async function queryThreadRows(db, plan, strategy = "auto") {
 		String(version)
 	]);
 	const byRoot = new Map(payloads.map((row) => [key(row.organizationId, row.traceId, row.cursorId), row]));
-	const agents = page.filter((row) => byRoot.get(key(row.organizationId, row.firstTraceId, row.firstCursorId))?.spanType === "agent_run");
-	const models = /* @__PURE__ */ new Map();
-	if (agents.length) {
-		const modelValues = [];
-		const rows = await run(`WITH generations AS (
-        SELECT organizationId, traceId, spanId, endedAt, cursorId, attributes,
-          min(timestamp) FILTER (WHERE eventType = 'start') OVER (PARTITION BY organizationId, traceId, spanId) AS startedAt
-        FROM span_events WHERE spanType = 'model_generation' AND (${agents.map((row) => `(${Object.entries({
-			organizationId: row.organizationId,
-			resourceId: row.resourceId,
-			traceId: row.firstTraceId,
-			parentSpanId: row.firstSpanId
-		}).map(([field, value]) => {
-			modelValues.push(value);
-			return `${field} ${field === "organizationId" || field === "resourceId" ? "IS NOT DISTINCT FROM" : "="} ?`;
-		}).join(" AND ")})`).join(" OR ")})
-      ) SELECT organizationId, traceId,
-        CASE WHEN json_type(attributes, '$.model') = 'VARCHAR' THEN json_extract_string(attributes, '$.model') END AS model,
-        CASE WHEN json_type(attributes, '$.provider') = 'VARCHAR' THEN json_extract_string(attributes, '$.provider') END AS provider
-        FROM generations WHERE endedAt IS NOT NULL
-        QUALIFY row_number() OVER (PARTITION BY organizationId, traceId ORDER BY startedAt, spanId, cursorId DESC) = 1`, modelValues);
-		for (const row of rows) models.set(JSON.stringify([row.organizationId, row.traceId]), row);
-	}
 	const threads = page.flatMap((row) => {
 		const first = byRoot.get(key(row.organizationId, row.firstTraceId, row.firstCursorId));
 		const last = byRoot.get(key(row.organizationId, row.lastTraceId, row.lastCursorId));
 		if (!first || !last) return [];
-		const model = models.get(JSON.stringify([row.organizationId, row.firstTraceId]));
 		return [coreStorage.threadQueryRowSchema.parse({
 			organizationId: row.organizationId,
 			threadId: row.threadId,
@@ -4412,8 +4373,6 @@ async function queryThreadRows(db, plan, strategy = "auto") {
 			endedAt: row.endedAt.toISOString(),
 			traceCount: Number(row.traceCount),
 			timeToFirstChunkMs: first.timeToFirstChunkMs,
-			model: model?.model ?? null,
-			provider: model?.provider ?? null,
 			inputPreview: readPreview(first.inputPreview, first.input, true),
 			outputPreview: readPreview(last.outputPreview, last.output, true),
 			errorPreview: readPreview(last.errorPreview, last.error, true)
@@ -4429,8 +4388,7 @@ async function queryThreadRows(db, plan, strategy = "auto") {
 		}) : null }
 	};
 }
-//#endregion
-//#region src/storage/domains/observability/trace-aggregate.ts
+// src/storage/domains/observability/trace-aggregate.ts
 /** Trace-root column expressions (alias `r`) for each groupable dimension; same values `where` sees. */
 const DIMENSION_SQL = {
 	entityType: "r.entityType",
@@ -4811,8 +4769,7 @@ async function runAggregate(db, plan, compiled) {
 		truncated
 	};
 }
-//#endregion
-//#region src/storage/domains/observability/tracing.ts
+// src/storage/domains/observability/tracing.ts
 const COLUMNS_SQL = [
 	"eventType",
 	"timestamp",
@@ -6166,8 +6123,7 @@ async function getBranchStreamHeadCursor(db, userSpanType) {
 	if (userSpanType) return encodeDeltaCursor((await db.query(`SELECT max(cursorId) AS cursorId FROM span_events WHERE eventType = 'start' AND spanType = ? AND cursorId IS NOT NULL`, [userSpanType]))[0]?.cursorId);
 	return encodeDeltaCursor((await db.query(`SELECT max(cursorId) AS cursorId FROM span_events WHERE eventType = 'start' AND spanType IN (${BRANCH_SPAN_TYPE_PLACEHOLDERS}) AND cursorId IS NOT NULL`, [...BRANCH_SPAN_TYPES]))[0]?.cursorId);
 }
-//#endregion
-//#region src/storage/domains/observability/index.ts
+// src/storage/domains/observability/index.ts
 function buildSignalMigrationRequiredMessage(args) {
 	return `
 ===========================================================================
@@ -6498,7 +6454,6 @@ var ObservabilityStorageDuckDB = class ObservabilityStorageDuckDB extends Observ
 		return getFeedbackPercentiles(this.db, args);
 	}
 };
-//#endregion
 export { ObservabilityStorageDuckDB };
 
-//# sourceMappingURL=observability-Bv3eObB-.js.map
+//# sourceMappingURL=observability-BWSYzhM2.js.map