@mastra/duckdb 1.16.0 → 1.17.0-alpha.1

package/dist/{observability-B-Kt8vrw.js → observability-CN4Rwgv9.js}

@@ -1,4 +1,4 @@
-import { a as timeToFirstChunkSupported, i as timeToFirstChunkFeatures, n as spanQueryFeatures, o as DuckDBConnection, s as DuckDBQueryTimeoutError, t as spanAggregateFeatures } from "./features-B8gl38Ue.js";
+import { c as timingFeatures, d as DuckDBQueryTimeoutError, i as spanQueryOutcomeFieldsSupported, l as traceQueryOutcomeFieldsSupported, n as spanAggregateFeatures, o as timeToFirstChunkSupported, r as spanQueryFeatures, s as timeToFirstTokenSupported, t as outcomeFeatures, u as DuckDBConnection } from "./features-D9RrctvA.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, getStringAttribute, listBranchesArgsSchema, listFeedbackArgsSchema, listLogsArgsSchema, listMetricsArgsSchema, listScoresArgsSchema, listTracesArgsSchema, resolveRetentionTargets, retentionCutoffMs, toTraceSpans } from "@mastra/core/storage";
@@ -2899,6 +2899,10 @@ const TRACE_FIELDS = {
 		sql: "r.timeToFirstChunkMs",
 		parameterType: "scalar"
 	},
+	timeToFirstTokenMs: {
+		sql: "r.timeToFirstTokenMs",
+		parameterType: "scalar"
+	},
 	entityId: {
 		sql: "r.entityId",
 		parameterType: "scalar"
@@ -2927,6 +2931,14 @@ const TRACE_FIELDS = {
 		sql: TRACE_STATUS_SQL,
 		parameterType: "scalar"
 	},
+	outcome: {
+		sql: "r.outcome",
+		parameterType: "scalar"
+	},
+	stopReason: {
+		sql: "r.stopReason",
+		parameterType: "scalar"
+	},
 	tags: {
 		sql: "r.tags",
 		parameterType: "scalar"
@@ -2965,10 +2977,22 @@ const SPAN_FIELDS = {
 		sql: "s.timeToFirstChunkMs",
 		parameterType: "scalar"
 	},
+	timeToFirstTokenMs: {
+		sql: "s.timeToFirstTokenMs",
+		parameterType: "scalar"
+	},
 	status: {
 		sql: "s.status",
 		parameterType: "scalar"
 	},
+	outcome: {
+		sql: "s.outcome",
+		parameterType: "scalar"
+	},
+	stopReason: {
+		sql: "s.stopReason",
+		parameterType: "scalar"
+	},
 	error: {
 		sql: "s.error",
 		parameterType: "scalar"
@@ -3175,9 +3199,13 @@ 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,
+  r.outcome AS outcome,
+  r.stopReason AS stopReason,
+  r.stopReasonContext AS stopReasonContext,
   ${TRACE_STATUS_SQL} AS status`;
 function fieldDefinition(registry, field) {
 	const definition = registry[field];
@@ -3411,6 +3439,7 @@ const ROOT_PUSHDOWN_FIELDS = /* @__PURE__ */ new Set([
 	"organizationId",
 	"endedAt",
 	"timeToFirstChunkMs",
+	"timeToFirstTokenMs",
 	"entityId",
 	"entityName",
 	"entityType",
@@ -3532,17 +3561,22 @@ function compileDuckDBTraceScope(relatedCollections, scope, timeRange, rootPredi
         traceId,
         name,
         spanType,
-        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,
+        COALESCE(
+          model,
+          CASE WHEN json_type(attributes, '$.model') = 'VARCHAR' THEN json_extract_string(attributes, '$.model') END
+        ) AS model,
+        COALESCE(
+          provider,
+          CASE WHEN json_type(attributes, '$.provider') = 'VARCHAR' THEN json_extract_string(attributes, '$.provider') END
+        ) AS provider,
         startedAt,
         endedAt,
         ${durationMsSql("startedAt", "endedAt")} AS durationMs,
         timeToFirstChunkMs,
+        timeToFirstTokenMs,
         CASE WHEN error IS NOT NULL THEN 'error' ELSE 'success' END AS status,
+        outcome,
+        stopReason,
         error,
         entityType,
         entityId,
@@ -3882,7 +3916,13 @@ 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) },
+		...traceQueryOutcomeFieldsSupported && {
+			outcome: row.outcome == null ? null : String(row.outcome),
+			stopReason: row.stopReason == null ? null : String(row.stopReason),
+			stopReasonContext: parseJson(row.stopReasonContext) ?? null
+		}
 	};
 }
 async function queryTraces(db, plan) {
@@ -4003,7 +4043,10 @@ const fields = [
 	"experimentId",
 	"environment",
 	"executionSource",
-	"timeToFirstChunkMs"
+	"timeToFirstChunkMs",
+	"timeToFirstTokenMs",
+	"outcome",
+	"stopReason"
 ];
 function keys(rows, values, alias = "") {
 	return rows.map((row) => `(${identityFields.map((field) => {
@@ -4061,8 +4104,6 @@ function compileDuckDBSpanSelection(plan) {
 	values.push(plan.timeRange.from, plan.timeRange.to, ...predicate.values);
 	const join = identityFields.map((field) => `e.${field} IS NOT DISTINCT FROM c.${field}`).join(" AND ");
 	const partition = identityFields.map((field) => `e.${field}`).join(", ");
-	const legacyModelFields = coreStorage.SPAN_QUERY_FIELD_REGISTRY ? "" : `CASE WHEN json_type(e.attributes, '$.model') = 'VARCHAR' THEN json_extract_string(e.attributes, '$.model') END AS model,
-       CASE WHEN json_type(e.attributes, '$.provider') = 'VARCHAR' THEN json_extract_string(e.attributes, '$.provider') END AS provider,`;
 	return {
 		ctes: [
 			`candidates AS MATERIALIZED (
@@ -4074,7 +4115,7 @@ function compileDuckDBSpanSelection(plan) {
 			`span_versions AS (
     SELECT ${fields.map(columnSql).join(", ")},
       min(e.timestamp) FILTER (WHERE e.eventType = 'start') OVER (PARTITION BY ${partition}) AS startedAt, e.endedAt, e.cursorId,
-      ${legacyModelFields}
+      e.model, e.provider,
       date_diff('millisecond', min(e.timestamp) FILTER (WHERE e.eventType = 'start') OVER (PARTITION BY ${partition}), e.endedAt) AS durationMs,
       CASE WHEN e.error IS NULL THEN 'success' ELSE 'error' END AS status,
       CASE WHEN e.error IS NULL THEN NULL ELSE 'error' END AS error,
@@ -4130,11 +4171,16 @@ 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, outcome, stopReason, ...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) },
+		...spanQueryOutcomeFieldsSupported && {
+			outcome: outcome ?? null,
+			stopReason: stopReason ?? null
+		}
 	}));
 	const page = rows.slice(0, plan.limit);
 	if (!page.length) return buildSpanQueryResponse(plan, [], []);
@@ -4144,9 +4190,9 @@ async function querySpans(db, plan) {
 		payloadValues.push(row.cursorId, row.endedAt);
 		return `(${key} AND cursorId IS NOT DISTINCT FROM ? AND endedAt = ?)`;
 	}).join(" OR ");
-	const previewRows = await query(`SELECT DISTINCT ${identities},
-    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,
+	const previewRows = await query(`SELECT DISTINCT ${identities}, CAST(stopReasonContext AS VARCHAR) AS stopReasonContext,
+    COALESCE(model, CASE WHEN json_type(attributes, '$.model') = 'VARCHAR' THEN json_extract_string(attributes, '$.model') END) AS model,
+    COALESCE(provider, CASE WHEN json_type(attributes, '$.provider') = 'VARCHAR' THEN json_extract_string(attributes, '$.provider') END) AS provider,
     ${previewPayloads.map(([preview, payload]) => `${preview}, CASE WHEN ${missingPreviewSql(preview)} THEN CAST(${payload} AS VARCHAR) END AS ${preview}Payload`).join(",\n    ")}
     FROM span_events WHERE ${payloadWhere}`, payloadValues);
 	const payloads = previewRows.map((row) => {
@@ -4169,7 +4215,8 @@ async function querySpans(db, plan) {
 	const hydrated = selected.map((row) => ({
 		...row,
 		model: metadata.get(coreStorage.spanQueryIdentityKey(row))?.model ?? null,
-		provider: metadata.get(coreStorage.spanQueryIdentityKey(row))?.provider ?? null
+		provider: metadata.get(coreStorage.spanQueryIdentityKey(row))?.provider ?? null,
+		...spanQueryOutcomeFieldsSupported && { stopReasonContext: parseJson(metadata.get(coreStorage.spanQueryIdentityKey(row))?.stopReasonContext) ?? null }
 	}));
 	let metrics = [];
 	if (!coreStorage.SPAN_QUERY_FIELD_REGISTRY) {
@@ -4278,8 +4325,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",
@@ -4365,6 +4416,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",
@@ -4393,6 +4445,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)
@@ -4418,6 +4471,8 @@ const DIMENSION_SQL = {
 	entityId: "r.entityId",
 	environment: "r.environment",
 	status: TRACE_STATUS_SQL,
+	outcome: "r.outcome",
+	stopReason: "r.stopReason",
 	serviceName: "r.serviceName",
 	executionSource: "r.source",
 	threadId: "r.threadId",
@@ -4440,6 +4495,8 @@ const SPAN_DIMENSION_SQL = Object.fromEntries([
 	"rootEntityName",
 	"environment",
 	"status",
+	"outcome",
+	"stopReason",
 	"serviceName",
 	"executionSource",
 	"threadId",
@@ -4447,7 +4504,9 @@ const SPAN_DIMENSION_SQL = Object.fromEntries([
 	"userId",
 	"sessionId",
 	"organizationId",
-	"experimentId"
+	"experimentId",
+	"model",
+	"provider"
 ].map((field) => [field, `r.${field}`]));
 const PERCENTILES = {
 	"duration.p50": .5,
@@ -6340,7 +6399,8 @@ var ObservabilityStorageDuckDB = class ObservabilityStorageDuckDB extends Observ
 			"trace-query-tenant-scope",
 			"feedback",
 			"trace-query-context-ids",
-			...timeToFirstChunkFeatures,
+			...timingFeatures,
+			...outcomeFeatures,
 			"trace-query-root-span-fields",
 			"trace-query-root-entity-id"
 		];
@@ -6365,7 +6425,8 @@ var ObservabilityStorageDuckDB = class ObservabilityStorageDuckDB extends Observ
 			"trace-query-tenant-scope",
 			"feedback",
 			"trace-query-context-ids",
-			...timeToFirstChunkFeatures,
+			...timingFeatures,
+			...outcomeFeatures,
 			"trace-query-root-span-fields",
 			"trace-query-root-entity-id"
 		];
@@ -6531,4 +6592,4 @@ var ObservabilityStorageDuckDB = class ObservabilityStorageDuckDB extends Observ
 };
 export { ObservabilityStorageDuckDB };
 
-//# sourceMappingURL=observability-B-Kt8vrw.js.map
+//# sourceMappingURL=observability-CN4Rwgv9.js.map