@mastra/pg 1.23.0-alpha.0 → 1.23.0-alpha.1

package/dist/index.js

@@ -13489,6 +13489,7 @@ var ObservabilityPG = class ObservabilityPG extends ObservabilityStorage {
 		}
 	}
 	async batchDeleteTraces(args) {
+		this.assertUnscopedBatchDeleteTraces(args);
 		try {
 			const tableName = getTableName$5({
 				indexName: TABLE_SPANS,
@@ -17671,11 +17672,39 @@ async function getTraceLight(client, schema, args) {
 		spans: rows.map(rowToLightSpanRecord)
 	};
 }
+/**
+* Delete traces by traceId, 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(client, schema, args) {
 	if (args.traceIds.length === 0) return;
-	const span = qualifiedTable(schema, TABLE_SPAN_EVENTS);
+	const params = [...args.traceIds];
 	const placeholders = args.traceIds.map((_, i) => `$${i + 1}`).join(", ");
-	await client.query(`DELETE FROM ${span} WHERE "traceId" IN (${placeholders})`, args.traceIds);
+	let scopeCondition = "";
+	if (args.organizationId !== void 0) {
+		params.push(args.organizationId);
+		scopeCondition += ` AND "organizationId" = $${params.length}`;
+	}
+	if (args.resourceId !== void 0) {
+		params.push(args.resourceId);
+		scopeCondition += ` AND "resourceId" = $${params.length}`;
+	}
+	const tables = [
+		TABLE_SPAN_EVENTS,
+		TABLE_METRIC_EVENTS,
+		TABLE_LOG_EVENTS,
+		TABLE_SCORE_EVENTS,
+		TABLE_FEEDBACK_EVENTS
+	];
+	await client.tx(async (t) => {
+		for (const tableName of tables) {
+			const table = qualifiedTable(schema, tableName);
+			await t.query(`DELETE FROM ${table} WHERE "traceId" IN (${placeholders})${scopeCondition}`, params);
+		}
+	});
 }
 /** Truncate the span_events table. */
 async function dangerouslyClearTracing(client, schema) {