@mastra/duckdb 1.7.0-alpha.0 → 1.7.0

package/dist/{observability-BAQ2aKaO.cjs → observability-yLoG0tl3.cjs}

@@ -20,7 +20,7 @@ var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__ge
 	enumerable: true
 }) : target, mod));
 //#endregion
-const require_db = require("./db-TBEcMD49.cjs");
+const require_db = require("./db-CJBcanx4.cjs");
 let _mastra_core_error = require("@mastra/core/error");
 let _mastra_core_storage = require("@mastra/core/storage");
 _mastra_core_storage = __toESM(_mastra_core_storage, 1);
@@ -2093,6 +2093,43 @@ async function migrateSignalTables(db, logger) {
 }
 //#endregion
 //#region src/storage/domains/observability/scores.ts
+const SCORE_UPSERT_CLAUSE = `ON CONFLICT (scoreId) DO UPDATE SET ${[
+	"timestamp",
+	"traceId",
+	"spanId",
+	"experimentId",
+	"scoreTraceId",
+	"entityType",
+	"entityId",
+	"entityName",
+	"entityVersionId",
+	"parentEntityVersionId",
+	"parentEntityType",
+	"parentEntityId",
+	"parentEntityName",
+	"rootEntityVersionId",
+	"rootEntityType",
+	"rootEntityId",
+	"rootEntityName",
+	"userId",
+	"organizationId",
+	"resourceId",
+	"runId",
+	"sessionId",
+	"threadId",
+	"requestId",
+	"environment",
+	"executionSource",
+	"serviceName",
+	"scorerId",
+	"scorerVersion",
+	"scoreSource",
+	"score",
+	"reason",
+	"tags",
+	"metadata",
+	"scope"
+].map((column) => `${column} = excluded.${column}`).join(", ")}`;
 const SCORE_GROUP_BY_COLUMNS = /* @__PURE__ */ new Set([
 	"timestamp",
 	"traceId",
@@ -2256,7 +2293,7 @@ function getComparisonDateRange(comparePeriod, timestamp) {
 async function createScore(db, args) {
 	const s = args.score;
 	const scoreSource = s.scoreSource ?? s.source ?? null;
-	await db.execute(`INSERT OR REPLACE INTO score_events (
+	await db.execute(`INSERT INTO score_events (
       scoreId, timestamp, cursorId, traceId, spanId, experimentId, scoreTraceId,
       entityType, entityId, entityName, entityVersionId, parentEntityVersionId, parentEntityType, parentEntityId, parentEntityName, rootEntityVersionId, rootEntityType, rootEntityId, rootEntityName,
       userId, organizationId, resourceId, runId, sessionId, threadId, requestId, environment, executionSource, serviceName,
@@ -2300,7 +2337,8 @@ async function createScore(db, args) {
 		jsonV(s.tags ?? null),
 		jsonV(s.metadata),
 		jsonV(s.scope ?? null)
-	].join(", ")})`);
+	].join(", ")})
+     ${SCORE_UPSERT_CLAUSE}`);
 }
 /** Insert multiple score events in a single statement. */
 async function batchCreateScores(db, args) {
@@ -2348,13 +2386,14 @@ async function batchCreateScores(db, args) {
 			jsonV(legacyScore.scope ?? null)
 		].join(", ")})`;
 	});
-	await db.execute(`INSERT OR REPLACE INTO score_events (
+	await db.execute(`INSERT INTO score_events (
       scoreId, timestamp, cursorId, traceId, spanId, experimentId, scoreTraceId,
       entityType, entityId, entityName, entityVersionId, parentEntityVersionId, parentEntityType, parentEntityId, parentEntityName, rootEntityVersionId, rootEntityType, rootEntityId, rootEntityName,
       userId, organizationId, resourceId, runId, sessionId, threadId, requestId, environment, executionSource, serviceName,
       scorerId, scorerVersion, scoreSource, score, reason, tags, metadata, scope
     )
-     VALUES ${tuples.join(",\n       ")}`);
+     VALUES ${tuples.join(",\n       ")}
+     ${SCORE_UPSERT_CLAUSE}`);
 }
 /** Query score events with filtering, ordering, and pagination. */
 async function listScores(db, args) {
@@ -3353,11 +3392,36 @@ async function batchCreateSpans(db, args) {
 		return events;
 	}));
 }
-/** Delete all span events for the given trace IDs. */
+/**
+* Delete all span events for the given trace IDs, cascading to trace-linked
+* signal events (metrics, logs, scores, feedback). Signal rows with a NULL
+* traceId are never affected. When the optional tenant scope
+* (`organizationId` / `resourceId`) is set, every DELETE additionally
+* requires the row's tenant columns to match.
+*/
 async function batchDeleteTraces(db, args) {
 	if (args.traceIds.length === 0) return;
 	const placeholders = args.traceIds.map(() => "?").join(", ");
-	await db.execute(`DELETE FROM span_events WHERE traceId IN (${placeholders})`, args.traceIds);
+	const params = [...args.traceIds];
+	let scopeCondition = "";
+	if (args.organizationId !== void 0) {
+		scopeCondition += ` AND organizationId = ?`;
+		params.push(args.organizationId);
+	}
+	if (args.resourceId !== void 0) {
+		scopeCondition += ` AND resourceId = ?`;
+		params.push(args.resourceId);
+	}
+	await db.executeTransaction([
+		"span_events",
+		"metric_events",
+		"log_events",
+		"score_events",
+		"feedback_events"
+	].map((table) => ({
+		sql: `DELETE FROM ${table} WHERE traceId IN (${placeholders})${scopeCondition}`,
+		params
+	})));
 }
 /** Reconstruct a single span from its event history. */
 async function getSpan(db, args) {
@@ -4204,4 +4268,4 @@ var ObservabilityStorageDuckDB = class extends _mastra_core_storage.Observabilit
 //#endregion
 exports.ObservabilityStorageDuckDB = ObservabilityStorageDuckDB;
 
-//# sourceMappingURL=observability-BAQ2aKaO.cjs.map
+//# sourceMappingURL=observability-yLoG0tl3.cjs.map