@mastra/duckdb 1.17.0-alpha.0 → 1.17.0

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

@@ -1,4 +1,4 @@
-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 { 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";
@@ -2931,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"
@@ -2977,6 +2985,14 @@ const SPAN_FIELDS = {
 		sql: "s.status",
 		parameterType: "scalar"
 	},
+	outcome: {
+		sql: "s.outcome",
+		parameterType: "scalar"
+	},
+	stopReason: {
+		sql: "s.stopReason",
+		parameterType: "scalar"
+	},
 	error: {
 		sql: "s.error",
 		parameterType: "scalar"
@@ -3187,6 +3203,9 @@ const TRACE_SELECT = `
   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];
@@ -3542,18 +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,
@@ -3894,7 +3917,12 @@ function traceFromRow(row) {
 		environment: row.environment == null ? null : String(row.environment),
 		status: row.status,
 		...timeToFirstChunkSupported && { timeToFirstChunkMs: row.timeToFirstChunkMs == null ? null : Number(row.timeToFirstChunkMs) },
-		...timeToFirstTokenSupported && { timeToFirstTokenMs: row.timeToFirstTokenMs == null ? null : Number(row.timeToFirstTokenMs) }
+		...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) {
@@ -4016,7 +4044,9 @@ const fields = [
 	"environment",
 	"executionSource",
 	"timeToFirstChunkMs",
-	"timeToFirstTokenMs"
+	"timeToFirstTokenMs",
+	"outcome",
+	"stopReason"
 ];
 function keys(rows, values, alias = "") {
 	return rows.map((row) => `(${identityFields.map((field) => {
@@ -4074,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 (
@@ -4087,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,
@@ -4143,12 +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, timeToFirstTokenMs, ...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) },
-		...timeToFirstTokenSupported && { timeToFirstTokenMs: timeToFirstTokenMs == null ? null : Number(timeToFirstTokenMs) }
+		...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, [], []);
@@ -4158,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) => {
@@ -4183,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) {
@@ -4438,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",
@@ -4460,6 +4495,8 @@ const SPAN_DIMENSION_SQL = Object.fromEntries([
 	"rootEntityName",
 	"environment",
 	"status",
+	"outcome",
+	"stopReason",
 	"serviceName",
 	"executionSource",
 	"threadId",
@@ -4467,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,
@@ -6361,6 +6400,7 @@ var ObservabilityStorageDuckDB = class ObservabilityStorageDuckDB extends Observ
 			"feedback",
 			"trace-query-context-ids",
 			...timingFeatures,
+			...outcomeFeatures,
 			"trace-query-root-span-fields",
 			"trace-query-root-entity-id"
 		];
@@ -6386,6 +6426,7 @@ var ObservabilityStorageDuckDB = class ObservabilityStorageDuckDB extends Observ
 			"feedback",
 			"trace-query-context-ids",
 			...timingFeatures,
+			...outcomeFeatures,
 			"trace-query-root-span-fields",
 			"trace-query-root-entity-id"
 		];
@@ -6551,4 +6592,4 @@ var ObservabilityStorageDuckDB = class ObservabilityStorageDuckDB extends Observ
 };
 export { ObservabilityStorageDuckDB };
 
-//# sourceMappingURL=observability-DlRTQlrI.js.map
+//# sourceMappingURL=observability-CN4Rwgv9.js.map