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

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

@@ -1,7 +1,7 @@
-import { a as timeToFirstChunkSupported, i as timeToFirstChunkFeatures, n as spanQueryFeatures, o as DuckDBConnection, s as DuckDBQueryTimeoutError, t as spanAggregateFeatures } from "./features-B8gl38Ue.js";
+import { a as timeToFirstTokenSupported, c as DuckDBQueryTimeoutError, i as timeToFirstChunkSupported, n as spanQueryFeatures, o as timingFeatures, s as DuckDBConnection, t as spanAggregateFeatures } from "./features-dc98I6Rx.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 { BRANCH_SPAN_TYPES, METRIC_DISTINCT_COLUMNS, ObservabilityStorage, computeTraceStatus, createStorageErrorId, executeRetentionPrune, feedbackRecordSchema, getStringAttribute, 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";
@@ -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`,
@@ -2879,6 +2899,10 @@ const TRACE_FIELDS = {
 		sql: "r.timeToFirstChunkMs",
 		parameterType: "scalar"
 	},
+	timeToFirstTokenMs: {
+		sql: "r.timeToFirstTokenMs",
+		parameterType: "scalar"
+	},
 	entityId: {
 		sql: "r.entityId",
 		parameterType: "scalar"
@@ -2945,6 +2969,10 @@ const SPAN_FIELDS = {
 		sql: "s.timeToFirstChunkMs",
 		parameterType: "scalar"
 	},
+	timeToFirstTokenMs: {
+		sql: "s.timeToFirstTokenMs",
+		parameterType: "scalar"
+	},
 	status: {
 		sql: "s.status",
 		parameterType: "scalar"
@@ -3155,6 +3183,7 @@ const TRACE_SELECT = `
   r.startedAt AS startedAt,
   r.endedAt AS endedAt,
   r.timeToFirstChunkMs AS timeToFirstChunkMs,
+  r.timeToFirstTokenMs AS timeToFirstTokenMs,
   r.entityName AS entityName,
   r.entityType AS entityType,
   r.environment AS environment,
@@ -3391,6 +3420,7 @@ const ROOT_PUSHDOWN_FIELDS = /* @__PURE__ */ new Set([
 	"organizationId",
 	"endedAt",
 	"timeToFirstChunkMs",
+	"timeToFirstTokenMs",
 	"entityId",
 	"entityName",
 	"entityType",
@@ -3522,6 +3552,7 @@ function compileDuckDBTraceScope(relatedCollections, scope, timeRange, rootPredi
         endedAt,
         ${durationMsSql("startedAt", "endedAt")} AS durationMs,
         timeToFirstChunkMs,
+        timeToFirstTokenMs,
         CASE WHEN error IS NOT NULL THEN 'error' ELSE 'success' END AS status,
         error,
         entityType,
@@ -3862,7 +3893,8 @@ function traceFromRow(row) {
 		entityType: row.entityType == null ? null : String(row.entityType),
 		environment: row.environment == null ? null : String(row.environment),
 		status: row.status,
-		...timeToFirstChunkSupported && { timeToFirstChunkMs: row.timeToFirstChunkMs == null ? null : Number(row.timeToFirstChunkMs) }
+		...timeToFirstChunkSupported && { timeToFirstChunkMs: row.timeToFirstChunkMs == null ? null : Number(row.timeToFirstChunkMs) },
+		...timeToFirstTokenSupported && { timeToFirstTokenMs: row.timeToFirstTokenMs == null ? null : Number(row.timeToFirstTokenMs) }
 	};
 }
 async function queryTraces(db, plan) {
@@ -3983,7 +4015,8 @@ const fields = [
 	"experimentId",
 	"environment",
 	"executionSource",
-	"timeToFirstChunkMs"
+	"timeToFirstChunkMs",
+	"timeToFirstTokenMs"
 ];
 function keys(rows, values, alias = "") {
 	return rows.map((row) => `(${identityFields.map((field) => {
@@ -4110,11 +4143,12 @@ async function querySpans(db, plan) {
 	};
 	const compiled = compileDuckDBSpanQuery(plan);
 	const rows = await query(compiled.sql, compiled.values);
-	const selected = rows.map(({ cursorId: _cursorId, startedAt, endedAt, timeToFirstChunkMs, ...row }) => ({
+	const selected = rows.map(({ cursorId: _cursorId, startedAt, endedAt, timeToFirstChunkMs, timeToFirstTokenMs, ...row }) => ({
 		...row,
 		startedAt: startedAt.toISOString(),
 		endedAt: endedAt.toISOString(),
-		...timeToFirstChunkSupported && { timeToFirstChunkMs: timeToFirstChunkMs == null ? null : Number(timeToFirstChunkMs) }
+		...timeToFirstChunkSupported && { timeToFirstChunkMs: timeToFirstChunkMs == null ? null : Number(timeToFirstChunkMs) },
+		...timeToFirstTokenSupported && { timeToFirstTokenMs: timeToFirstTokenMs == null ? null : Number(timeToFirstTokenMs) }
 	}));
 	const page = rows.slice(0, plan.limit);
 	if (!page.length) return buildSpanQueryResponse(plan, [], []);
@@ -4258,8 +4292,12 @@ function compileDuckDBThreadRowsQuery(plan, strategy = "full") {
 	}
 	values.push(plan.limit + 1);
 	const filtered = predicateFields(plan.where);
-	const firstContext = [...contextFields, "timeToFirstChunkMs"].filter((field) => field === "resourceId" || filtered.has(field));
-	const contextExpression = (field) => field === "executionSource" ? "source AS executionSource" : field === "timeToFirstChunkMs" ? "CASE WHEN spanType = 'agent_run' THEN timeToFirstChunkMs END AS timeToFirstChunkMs" : field;
+	const firstContext = [
+		...contextFields,
+		"timeToFirstChunkMs",
+		"timeToFirstTokenMs"
+	].filter((field) => field === "resourceId" || filtered.has(field));
+	const contextExpression = (field) => field === "executionSource" ? "source AS executionSource" : field === "timeToFirstChunkMs" || field === "timeToFirstTokenMs" ? `CASE WHEN spanType = 'agent_run' THEN ${field} END AS ${field}` : field;
 	const lastFields = [
 		"traceId",
 		"spanId",
@@ -4345,6 +4383,7 @@ async function queryThreadRows(db, plan, strategy = "auto") {
 	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 spanType = 'agent_run' THEN timeToFirstTokenMs END AS timeToFirstTokenMs,
     CASE WHEN error IS NULL THEN 'success' ELSE 'error' END AS status,
     ${[
 		"input",
@@ -4373,6 +4412,7 @@ async function queryThreadRows(db, plan, strategy = "auto") {
 			endedAt: row.endedAt.toISOString(),
 			traceCount: Number(row.traceCount),
 			timeToFirstChunkMs: first.timeToFirstChunkMs,
+			...timeToFirstTokenSupported && { timeToFirstTokenMs: first.timeToFirstTokenMs },
 			inputPreview: readPreview(first.inputPreview, first.input, true),
 			outputPreview: readPreview(last.outputPreview, last.output, true),
 			errorPreview: readPreview(last.errorPreview, last.error, true)
@@ -4782,6 +4822,15 @@ const COLUMNS_SQL = [
 	"isEvent",
 	"endedAt",
 	"timeToFirstChunkMs",
+	"externalParentSpanId",
+	"resumedFromSpanId",
+	"model",
+	"provider",
+	"outcome",
+	"stopReason",
+	"timeToFirstTokenMs",
+	"errorCount",
+	"stopReasonContext",
 	"experimentId",
 	"entityType",
 	"entityId",
@@ -4897,7 +4946,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. */
@@ -5093,6 +5151,7 @@ function rowToLightSpanRecordWithPreview(row) {
 	};
 }
 function rowToSpanRecord(row) {
+	const attributes = parseJson(row.attributes);
 	return {
 		traceId: row.traceId,
 		spanId: row.spanId,
@@ -5102,6 +5161,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 ?? getStringAttribute(attributes, "model"),
+		provider: row.provider ?? 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,
@@ -5125,7 +5193,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),
@@ -5288,6 +5356,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),
@@ -5333,6 +5410,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);
@@ -5346,7 +5441,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,
@@ -5401,7 +5496,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,
@@ -6265,7 +6360,7 @@ var ObservabilityStorageDuckDB = class ObservabilityStorageDuckDB extends Observ
 			"trace-query-tenant-scope",
 			"feedback",
 			"trace-query-context-ids",
-			...timeToFirstChunkFeatures,
+			...timingFeatures,
 			"trace-query-root-span-fields",
 			"trace-query-root-entity-id"
 		];
@@ -6290,7 +6385,7 @@ var ObservabilityStorageDuckDB = class ObservabilityStorageDuckDB extends Observ
 			"trace-query-tenant-scope",
 			"feedback",
 			"trace-query-context-ids",
-			...timeToFirstChunkFeatures,
+			...timingFeatures,
 			"trace-query-root-span-fields",
 			"trace-query-root-entity-id"
 		];
@@ -6456,4 +6551,4 @@ var ObservabilityStorageDuckDB = class ObservabilityStorageDuckDB extends Observ
 };
 export { ObservabilityStorageDuckDB };
 
-//# sourceMappingURL=observability-BWSYzhM2.js.map
+//# sourceMappingURL=observability-DlRTQlrI.js.map