@mastra/duckdb 1.16.0-alpha.1 → 1.16.0

package/dist/{observability-Dlt85fvH.cjs → observability-BEdkwaeY.cjs}

@@ -106,6 +106,17 @@ CREATE TABLE IF NOT EXISTS span_events (
   endedAt TIMESTAMP,
   timeToFirstChunkMs DOUBLE,
 
+  -- Native span fields, stored as supplied (NULL when the producer set none)
+  externalParentSpanId VARCHAR,
+  resumedFromSpanId VARCHAR,
+  model VARCHAR,
+  provider VARCHAR,
+  outcome VARCHAR,
+  stopReason VARCHAR,
+  timeToFirstTokenMs DOUBLE,
+  errorCount INTEGER,
+  stopReasonContext JSON,
+
   -- JSON fields
   attributes JSON,
   metadata JSON,
@@ -377,6 +388,15 @@ const ALL_MIGRATIONS = [
 	`ALTER TABLE span_events ADD COLUMN IF NOT EXISTS hasOutputPayload BOOLEAN`,
 	`ALTER TABLE span_events ADD COLUMN IF NOT EXISTS hasErrorPayload BOOLEAN`,
 	`ALTER TABLE span_events ADD COLUMN IF NOT EXISTS timeToFirstChunkMs DOUBLE`,
+	`ALTER TABLE span_events ADD COLUMN IF NOT EXISTS externalParentSpanId VARCHAR`,
+	`ALTER TABLE span_events ADD COLUMN IF NOT EXISTS resumedFromSpanId VARCHAR`,
+	`ALTER TABLE span_events ADD COLUMN IF NOT EXISTS model VARCHAR`,
+	`ALTER TABLE span_events ADD COLUMN IF NOT EXISTS provider VARCHAR`,
+	`ALTER TABLE span_events ADD COLUMN IF NOT EXISTS outcome VARCHAR`,
+	`ALTER TABLE span_events ADD COLUMN IF NOT EXISTS stopReason VARCHAR`,
+	`ALTER TABLE span_events ADD COLUMN IF NOT EXISTS timeToFirstTokenMs DOUBLE`,
+	`ALTER TABLE span_events ADD COLUMN IF NOT EXISTS errorCount INTEGER`,
+	`ALTER TABLE span_events ADD COLUMN IF NOT EXISTS stopReasonContext JSON`,
 	`ALTER TABLE span_events ADD COLUMN IF NOT EXISTS parentEntityType VARCHAR`,
 	`ALTER TABLE span_events ADD COLUMN IF NOT EXISTS parentEntityId VARCHAR`,
 	`ALTER TABLE span_events ADD COLUMN IF NOT EXISTS parentEntityName VARCHAR`,
@@ -4342,7 +4362,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,
@@ -4358,34 +4378,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 [_mastra_core_storage.threadQueryRowSchema.parse({
 			organizationId: row.organizationId,
 			threadId: row.threadId,
@@ -4397,8 +4393,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)
@@ -4808,6 +4802,15 @@ const COLUMNS_SQL = [
 	"isEvent",
 	"endedAt",
 	"timeToFirstChunkMs",
+	"externalParentSpanId",
+	"resumedFromSpanId",
+	"model",
+	"provider",
+	"outcome",
+	"stopReason",
+	"timeToFirstTokenMs",
+	"errorCount",
+	"stopReasonContext",
 	"experimentId",
 	"entityType",
 	"entityId",
@@ -4923,7 +4926,16 @@ const SPAN_RECONSTRUCT_SELECT = `
     ${argMaxNonNull("input")},
     ${argMaxNonNull("output")},
     ${argMaxNonNull("error")},
-    ${argMaxNonNull("requestContext")}
+    ${argMaxNonNull("requestContext")},
+    ${argMaxNonNull("externalParentSpanId")},
+    ${argMaxNonNull("resumedFromSpanId")},
+    ${argMaxNonNull("model")},
+    ${argMaxNonNull("provider")},
+    ${argMaxNonNull("outcome")},
+    ${argMaxNonNull("stopReason")},
+    ${argMaxNonNull("timeToFirstTokenMs")},
+    ${argMaxNonNull("errorCount")},
+    ${argMaxNonNull("stopReasonContext")}
   FROM span_events
 `;
 /** Lightweight variant — only timeline-relevant columns. */
@@ -5119,6 +5131,7 @@ function rowToLightSpanRecordWithPreview(row) {
 	};
 }
 function rowToSpanRecord(row) {
+	const attributes = parseJson(row.attributes);
 	return {
 		traceId: row.traceId,
 		spanId: row.spanId,
@@ -5128,6 +5141,15 @@ function rowToSpanRecord(row) {
 		isEvent: row.isEvent,
 		startedAt: toDate(row.startedAt),
 		endedAt: rowEndedAt(row),
+		externalParentSpanId: row.externalParentSpanId ?? null,
+		resumedFromSpanId: row.resumedFromSpanId ?? null,
+		model: row.model ?? (0, _mastra_core_storage.getStringAttribute)(attributes, "model"),
+		provider: row.provider ?? (0, _mastra_core_storage.getStringAttribute)(attributes, "provider"),
+		outcome: row.outcome ?? null,
+		stopReason: row.stopReason ?? null,
+		timeToFirstTokenMs: row.timeToFirstTokenMs ?? null,
+		errorCount: row.errorCount ?? null,
+		stopReasonContext: parseJson(row.stopReasonContext),
 		experimentId: row.experimentId ?? null,
 		entityType: row.entityType ?? null,
 		entityId: row.entityId ?? null,
@@ -5151,7 +5173,7 @@ function rowToSpanRecord(row) {
 		environment: row.environment ?? null,
 		source: row.source ?? null,
 		serviceName: row.serviceName ?? null,
-		attributes: parseJson(row.attributes),
+		attributes,
 		metadata: parseJson(row.metadata),
 		tags: parseJsonArray(row.tags),
 		scope: parseJson(row.scope),
@@ -5314,6 +5336,15 @@ function toValuesTuple(row) {
 		v(row.isEvent),
 		v(row.endedAt),
 		v(row.timeToFirstChunkMs),
+		v(row.externalParentSpanId),
+		v(row.resumedFromSpanId),
+		v(row.model),
+		v(row.provider),
+		v(row.outcome),
+		v(row.stopReason),
+		v(row.timeToFirstTokenMs),
+		v(row.errorCount),
+		jsonV(row.stopReasonContext),
 		v(row.experimentId),
 		v(row.entityType),
 		v(row.entityId),
@@ -5359,6 +5390,24 @@ async function insertSpanEvents(db, rows) {
 	const tuples = rows.map((row) => `(${toValuesTuple(row)})`).join(",\n");
 	await db.execute(`INSERT INTO span_events (${COLUMNS_SQL}) VALUES ${tuples}`);
 }
+/**
+* Columns copied as supplied onto both event rows (NULL when unset). Previews
+* stay out: the start row derives them from the payload it keeps.
+*/
+function copiedSpanColumns(s) {
+	return {
+		timeToFirstChunkMs: s.timeToFirstChunkMs ?? null,
+		externalParentSpanId: s.externalParentSpanId ?? null,
+		resumedFromSpanId: s.resumedFromSpanId ?? null,
+		model: s.model ?? null,
+		provider: s.provider ?? null,
+		outcome: s.outcome ?? null,
+		stopReason: s.stopReason ?? null,
+		timeToFirstTokenMs: s.timeToFirstTokenMs ?? null,
+		errorCount: s.errorCount ?? null,
+		stopReasonContext: s.stopReasonContext ?? null
+	};
+}
 function createStartSpanRow(s) {
 	const payloadOnEndRow = !!s.endedAt && !s.isEvent;
 	const input = previewOf(!payloadOnEndRow, s.input, s.inputPreview);
@@ -5372,7 +5421,7 @@ function createStartSpanRow(s) {
 		spanType: s.spanType,
 		isEvent: s.isEvent,
 		endedAt: s.isEvent ? s.startedAt : null,
-		timeToFirstChunkMs: s.timeToFirstChunkMs ?? null,
+		...copiedSpanColumns(s),
 		experimentId: s.experimentId ?? null,
 		entityType: s.entityType ?? null,
 		entityId: s.entityId ?? null,
@@ -5427,7 +5476,7 @@ function createEndSpanRow(s) {
 		spanType: s.spanType,
 		isEvent: s.isEvent,
 		endedAt: s.endedAt ?? null,
-		timeToFirstChunkMs: s.timeToFirstChunkMs ?? null,
+		...copiedSpanColumns(s),
 		experimentId: s.experimentId ?? null,
 		entityType: s.entityType ?? null,
 		entityId: s.entityId ?? null,
@@ -6482,4 +6531,4 @@ var ObservabilityStorageDuckDB = class ObservabilityStorageDuckDB extends _mastr
 };
 exports.ObservabilityStorageDuckDB = ObservabilityStorageDuckDB;
 
-//# sourceMappingURL=observability-Dlt85fvH.cjs.map
+//# sourceMappingURL=observability-BEdkwaeY.cjs.map