@mastra/duckdb 1.11.2-alpha.0 → 1.12.0-alpha.2

package/dist/{observability-D4A1jrAF.js → observability-DR9FesfB.js}

@@ -2716,6 +2716,22 @@ const TRACE_FIELDS = {
 		sql: "r.resourceId",
 		parameterType: "scalar"
 	},
+	runId: {
+		sql: "r.runId",
+		parameterType: "scalar"
+	},
+	sessionId: {
+		sql: "r.sessionId",
+		parameterType: "scalar"
+	},
+	userId: {
+		sql: "r.userId",
+		parameterType: "scalar"
+	},
+	organizationId: {
+		sql: "r.organizationId",
+		parameterType: "scalar"
+	},
 	startedAt: {
 		sql: "r.startedAt",
 		parameterType: "timestamp"
@@ -2809,6 +2825,22 @@ const SPAN_FIELDS = {
 	rootEntityVersionId: {
 		sql: "s.rootEntityVersionId",
 		parameterType: "scalar"
+	},
+	runId: {
+		sql: "s.runId",
+		parameterType: "scalar"
+	},
+	sessionId: {
+		sql: "s.sessionId",
+		parameterType: "scalar"
+	},
+	userId: {
+		sql: "s.userId",
+		parameterType: "scalar"
+	},
+	organizationId: {
+		sql: "s.organizationId",
+		parameterType: "scalar"
 	}
 };
 const SCORE_FIELDS = {
@@ -3194,7 +3226,11 @@ function compileDuckDBTraceScope(relatedCollections, scope) {
         entityName,
         entityVersionId,
         parentEntityVersionId,
-        rootEntityVersionId
+        rootEntityVersionId,
+        runId,
+        sessionId,
+        userId,
+        organizationId
       FROM current_span_rows
       WHERE currentRank = 1
     )`);
@@ -4009,6 +4045,7 @@ async function insertSpanEvents(db, rows) {
 	await db.execute(`INSERT INTO span_events (${COLUMNS_SQL}) VALUES ${tuples}`);
 }
 function createStartSpanRow(s) {
+	const payloadOnEndRow = !!s.endedAt && !s.isEvent;
 	return {
 		eventType: "start",
 		timestamp: s.startedAt,
@@ -4036,15 +4073,15 @@ function createStartSpanRow(s) {
 		environment: s.environment ?? null,
 		source: s.source ?? null,
 		serviceName: s.serviceName ?? null,
-		attributes: s.attributes ?? null,
-		metadata: s.metadata ?? null,
+		attributes: payloadOnEndRow ? null : s.attributes ?? null,
+		metadata: payloadOnEndRow ? null : s.metadata ?? null,
 		tags: s.tags == null ? null : normalizeTags(s.tags),
 		scope: s.scope ?? null,
 		links: null,
-		input: s.input ?? null,
+		input: payloadOnEndRow ? null : s.input ?? null,
 		output: null,
 		error: null,
-		requestContext: s.requestContext ?? null
+		requestContext: payloadOnEndRow ? null : s.requestContext ?? null
 	};
 }
 function createEndSpanRow(s) {
@@ -4911,7 +4948,8 @@ var ObservabilityStorageDuckDB = class ObservabilityStorageDuckDB extends Observ
 			"trace-query-discovery",
 			"thread-query",
 			"trace-query-tenant-scope",
-			"feedback"
+			"feedback",
+			"trace-query-context-ids"
 		];
 		return [
 			"metrics",
@@ -4928,7 +4966,8 @@ var ObservabilityStorageDuckDB = class ObservabilityStorageDuckDB extends Observ
 			"trace-query-discovery",
 			"thread-query",
 			"trace-query-tenant-scope",
-			"feedback"
+			"feedback",
+			"trace-query-context-ids"
 		];
 	}
 	async createSpan(args) {
@@ -5083,4 +5122,4 @@ var ObservabilityStorageDuckDB = class ObservabilityStorageDuckDB extends Observ
 //#endregion
 export { ObservabilityStorageDuckDB };
 
-//# sourceMappingURL=observability-D4A1jrAF.js.map
+//# sourceMappingURL=observability-DR9FesfB.js.map