@mastra/spanner 1.9.6-alpha.0 → 1.10.0-alpha.1

package/dist/index.js

@@ -3273,6 +3273,10 @@ var DatasetsSpanner = class DatasetsSpanner extends DatasetsStorage {
 			const tableName = quoteIdent(TABLE_DATASETS, "table name");
 			const filterConditions = [];
 			const filterParams = {};
+			if (args.filters?.datasetIds !== void 0) {
+				filterConditions.push(`${quoteIdent("id", "column name")} IN UNNEST(@datasetIds)`);
+				filterParams.datasetIds = args.filters.datasetIds;
+			}
 			if (args.filters?.organizationId !== void 0) {
 				filterConditions.push(`${quoteIdent("organizationId", "column name")} = @organizationId`);
 				filterParams.organizationId = args.filters.organizationId;
@@ -3302,10 +3306,15 @@ var DatasetsSpanner = class DatasetsSpanner extends DatasetsStorage {
 				filterParams.nameSubstring = `%${args.filters.name}%`;
 			}
 			const whereClause = filterConditions.length > 0 ? `WHERE ${filterConditions.join(" AND ")}` : "";
-			const filterTypes = args.filters?.targetIds !== void 0 && args.filters.targetIds.length > 0 ? { targetIds: {
+			const filterTypes = {};
+			if (args.filters?.targetIds !== void 0 && args.filters.targetIds.length > 0) filterTypes.targetIds = {
 				type: "array",
 				child: { type: "string" }
-			} } : {};
+			};
+			if (args.filters?.datasetIds !== void 0) filterTypes.datasetIds = {
+				type: "array",
+				child: { type: "string" }
+			};
 			const [countRows] = await this.database.run({
 				sql: `SELECT COUNT(*) AS count FROM ${tableName} ${whereClause}`,
 				params: filterParams,
@@ -8482,18 +8491,15 @@ function buildWhereClause(filters, startIndex = 0) {
 			conditions.push("1 = 0");
 			return;
 		}
-		const placeholders = [];
-		for (const v of values) {
-			const param = `f${i++}`;
-			placeholders.push(`@${param}`);
-			params[param] = v;
-		}
-		conditions.push(`${quoteIdent(col, "column name")} IN (${placeholders.join(", ")})`);
+		const param = `f${i++}`;
+		params[param] = [...values];
+		conditions.push(`${quoteIdent(col, "column name")} IN UNNEST(@${param})`);
 	};
 	const ts = filters.timestamp;
 	if (ts?.start) bindScalar("timestamp", ts.start, ts.startExclusive ? ">" : ">=");
 	if (ts?.end) bindScalar("timestamp", ts.end, ts.endExclusive ? "<" : "<=");
 	if (Array.isArray(filters.traceIds)) bindIn("traceId", filters.traceIds);
+	if (Array.isArray(filters.experimentIds)) bindIn("experimentId", filters.experimentIds);
 	for (const k of [
 		"traceId",
 		"spanId",