@mastra/spanner 1.2.1 → 1.2.2-alpha.0

package/dist/index.js

@@ -3032,7 +3032,15 @@ var DatasetsSpanner = class _DatasetsSpanner extends DatasetsStorage {
     await this.db.alterTable({
       tableName: TABLE_DATASETS,
       schema: TABLE_SCHEMAS[TABLE_DATASETS],
-      ifNotExists: ["organizationId", "projectId", "candidateKey", "candidateId"]
+      ifNotExists: [
+        "organizationId",
+        "projectId",
+        "candidateKey",
+        "candidateId",
+        "targetType",
+        "targetIds",
+        "scorerIds"
+      ]
     });
     await this.db.alterTable({
       tableName: TABLE_DATASET_ITEMS,
@@ -3079,6 +3087,20 @@ var DatasetsSpanner = class _DatasetsSpanner extends DatasetsStorage {
     await this.db.clearTable({ tableName: TABLE_DATASET_ITEMS });
     await this.db.clearTable({ tableName: TABLE_DATASETS });
   }
+  async experimentTablesExist() {
+    try {
+      const [rows] = await this.database.run({
+        sql: `SELECT COUNT(*) AS c FROM INFORMATION_SCHEMA.TABLES
+              WHERE TABLE_SCHEMA = "" AND TABLE_NAME IN (@a, @b)`,
+        params: { a: TABLE_EXPERIMENTS, b: TABLE_EXPERIMENT_RESULTS },
+        json: true
+      });
+      const row = rows?.[0];
+      return Number(row?.c ?? 0) === 2;
+    } catch {
+      return false;
+    }
+  }
   // ==========================================================================
   // Dataset CRUD
   // ==========================================================================
@@ -3143,7 +3165,27 @@ var DatasetsSpanner = class _DatasetsSpanner extends DatasetsStorage {
   }
   async getDatasetById(args) {
     try {
-      const row = await this.db.load({ tableName: TABLE_DATASETS, keys: { id: args.id } });
+      const hasTenancy = args.filters?.organizationId !== void 0 || args.filters?.projectId !== void 0;
+      if (!hasTenancy) {
+        const row2 = await this.db.load({ tableName: TABLE_DATASETS, keys: { id: args.id } });
+        return row2 ? rowToDataset(row2) : null;
+      }
+      const conditions = [`${quoteIdent("id", "column name")} = @id`];
+      const params = { id: args.id };
+      if (args.filters?.organizationId !== void 0) {
+        conditions.push(`${quoteIdent("organizationId", "column name")} = @organizationId`);
+        params.organizationId = args.filters.organizationId;
+      }
+      if (args.filters?.projectId !== void 0) {
+        conditions.push(`${quoteIdent("projectId", "column name")} = @projectId`);
+        params.projectId = args.filters.projectId;
+      }
+      const [rows] = await this.database.run({
+        sql: `SELECT * FROM ${quoteIdent(TABLE_DATASETS, "table name")} WHERE ${conditions.join(" AND ")} LIMIT 1`,
+        params,
+        json: true
+      });
+      const row = rows[0];
       return row ? rowToDataset(row) : null;
     } catch (error) {
       throw new MastraError(
@@ -3172,7 +3214,7 @@ var DatasetsSpanner = class _DatasetsSpanner extends DatasetsStorage {
       if (args.scorerIds !== void 0) data.scorerIds = args.scorerIds;
       data.updatedAt = /* @__PURE__ */ new Date();
       await this.db.update({ tableName: TABLE_DATASETS, keys: { id: args.id }, data });
-      const updated = await this.getDatasetById({ id: args.id });
+      const updated = await this.getDatasetById({ id: args.id, filters: args.filters });
       if (!updated) {
         throw new MastraError({
           id: createStorageErrorId("SPANNER", "UPDATE_DATASET", "NOT_FOUND"),
@@ -3198,26 +3240,46 @@ var DatasetsSpanner = class _DatasetsSpanner extends DatasetsStorage {
   }
   async deleteDataset(args) {
     try {
-      try {
-        await this.db.runDml({
-          sql: `DELETE FROM ${quoteIdent(TABLE_EXPERIMENT_RESULTS, "table name")}
-                WHERE ${quoteIdent("experimentId", "column name")} IN (
-                  SELECT ${quoteIdent("id", "column name")} FROM ${quoteIdent(TABLE_EXPERIMENTS, "table name")}
-                  WHERE ${quoteIdent("datasetId", "column name")} = @id)`,
-          params: { id: args.id }
-        });
-        await this.db.runDml({
-          sql: `UPDATE ${quoteIdent(TABLE_EXPERIMENTS, "table name")}
-                SET ${quoteIdent("datasetId", "column name")} = NULL,
-                    ${quoteIdent("datasetVersion", "column name")} = NULL
-                WHERE ${quoteIdent("datasetId", "column name")} = @id`,
-          params: { id: args.id }
-        });
-      } catch {
+      const tenancyConditions = [];
+      const tenancyParams = {};
+      if (args.filters?.organizationId !== void 0) {
+        tenancyConditions.push(`${quoteIdent("organizationId", "column name")} = @organizationId`);
+        tenancyParams.organizationId = args.filters.organizationId;
+      }
+      if (args.filters?.projectId !== void 0) {
+        tenancyConditions.push(`${quoteIdent("projectId", "column name")} = @projectId`);
+        tenancyParams.projectId = args.filters.projectId;
       }
+      const scopedWhere = [`${quoteIdent("id", "column name")} = @id`, ...tenancyConditions].join(" AND ");
+      const experimentTablesExist = await this.experimentTablesExist();
       await this.db.runWithAbortRetry(
         () => this.database.runTransactionAsync(async (tx) => {
           try {
+            const [rows] = await tx.run({
+              sql: `SELECT ${quoteIdent("id", "column name")} FROM ${quoteIdent(TABLE_DATASETS, "table name")} WHERE ${scopedWhere}`,
+              params: { id: args.id, ...tenancyParams },
+              json: true
+            });
+            if (!rows || rows.length === 0) {
+              await tx.commit();
+              return;
+            }
+            if (experimentTablesExist) {
+              await tx.runUpdate({
+                sql: `DELETE FROM ${quoteIdent(TABLE_EXPERIMENT_RESULTS, "table name")}
+                      WHERE ${quoteIdent("experimentId", "column name")} IN (
+                        SELECT ${quoteIdent("id", "column name")} FROM ${quoteIdent(TABLE_EXPERIMENTS, "table name")}
+                        WHERE ${quoteIdent("datasetId", "column name")} = @id)`,
+                params: { id: args.id }
+              });
+              await tx.runUpdate({
+                sql: `UPDATE ${quoteIdent(TABLE_EXPERIMENTS, "table name")}
+                      SET ${quoteIdent("datasetId", "column name")} = NULL,
+                          ${quoteIdent("datasetVersion", "column name")} = NULL
+                      WHERE ${quoteIdent("datasetId", "column name")} = @id`,
+                params: { id: args.id }
+              });
+            }
             for (const table of [TABLE_DATASET_VERSIONS, TABLE_DATASET_ITEMS]) {
               await tx.runUpdate({
                 sql: `DELETE FROM ${quoteIdent(table, "table name")} WHERE ${quoteIdent("datasetId", "column name")} = @id`,
@@ -3225,8 +3287,8 @@ var DatasetsSpanner = class _DatasetsSpanner extends DatasetsStorage {
               });
             }
             await tx.runUpdate({
-              sql: `DELETE FROM ${quoteIdent(TABLE_DATASETS, "table name")} WHERE ${quoteIdent("id", "column name")} = @id`,
-              params: { id: args.id }
+              sql: `DELETE FROM ${quoteIdent(TABLE_DATASETS, "table name")} WHERE ${scopedWhere}`,
+              params: { id: args.id, ...tenancyParams }
             });
             await tx.commit();
           } catch (err) {
@@ -3272,10 +3334,26 @@ var DatasetsSpanner = class _DatasetsSpanner extends DatasetsStorage {
         filterConditions.push(`${quoteIdent("candidateId", "column name")} = @candidateId`);
         filterParams.candidateId = args.filters.candidateId;
       }
+      if (args.filters?.targetType !== void 0) {
+        filterConditions.push(`${quoteIdent("targetType", "column name")} = @targetType`);
+        filterParams.targetType = args.filters.targetType;
+      }
+      if (args.filters?.targetIds !== void 0 && args.filters.targetIds.length > 0) {
+        filterConditions.push(
+          `EXISTS (SELECT 1 FROM UNNEST(JSON_QUERY_ARRAY(${quoteIdent("targetIds", "column name")})) AS t WHERE JSON_VALUE(t) IN UNNEST(@targetIds))`
+        );
+        filterParams.targetIds = args.filters.targetIds;
+      }
+      if (args.filters?.name !== void 0 && args.filters.name.length > 0) {
+        filterConditions.push(`LOWER(${quoteIdent("name", "column name")}) LIKE LOWER(@nameSubstring)`);
+        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: { type: "array", child: { type: "string" } } } : {};
       const [countRows] = await this.database.run({
         sql: `SELECT COUNT(*) AS count FROM ${tableName} ${whereClause}`,
         params: filterParams,
+        types: filterTypes,
         json: true
       });
       const total = Number(countRows[0]?.count ?? 0);
@@ -3288,6 +3366,7 @@ var DatasetsSpanner = class _DatasetsSpanner extends DatasetsStorage {
               ORDER BY ${quoteIdent("createdAt", "column name")} DESC, ${quoteIdent("id", "column name")} ASC
               LIMIT @limit OFFSET @offset`,
         params: { ...filterParams, limit, offset },
+        types: filterTypes,
         json: true
       });
       const datasets = rows.map(rowToDataset);
@@ -4212,7 +4291,27 @@ var ExperimentsSpanner = class _ExperimentsSpanner extends ExperimentsStorage {
   }
   async getExperimentById(args) {
     try {
-      const row = await this.db.load({ tableName: TABLE_EXPERIMENTS, keys: { id: args.id } });
+      const hasTenancy = args.filters?.organizationId !== void 0 || args.filters?.projectId !== void 0;
+      if (!hasTenancy) {
+        const row2 = await this.db.load({ tableName: TABLE_EXPERIMENTS, keys: { id: args.id } });
+        return row2 ? rowToExperiment(row2) : null;
+      }
+      const conditions = [`${quoteIdent("id", "column name")} = @id`];
+      const params = { id: args.id };
+      if (args.filters?.organizationId !== void 0) {
+        conditions.push(`${quoteIdent("organizationId", "column name")} = @organizationId`);
+        params.organizationId = args.filters.organizationId;
+      }
+      if (args.filters?.projectId !== void 0) {
+        conditions.push(`${quoteIdent("projectId", "column name")} = @projectId`);
+        params.projectId = args.filters.projectId;
+      }
+      const [rows] = await this.database.run({
+        sql: `SELECT * FROM ${quoteIdent(TABLE_EXPERIMENTS, "table name")} WHERE ${conditions.join(" AND ")} LIMIT 1`,
+        params,
+        json: true
+      });
+      const row = rows[0];
       return row ? rowToExperiment(row) : null;
     } catch (error) {
       throw new MastraError(
@@ -4306,18 +4405,43 @@ var ExperimentsSpanner = class _ExperimentsSpanner extends ExperimentsStorage {
   }
   async deleteExperiment(args) {
     try {
+      const tenancyConditions = [];
+      const tenancyParams = {};
+      if (args.filters?.organizationId !== void 0) {
+        tenancyConditions.push(`${quoteIdent("organizationId", "column name")} = @organizationId`);
+        tenancyParams.organizationId = args.filters.organizationId;
+      }
+      if (args.filters?.projectId !== void 0) {
+        tenancyConditions.push(`${quoteIdent("projectId", "column name")} = @projectId`);
+        tenancyParams.projectId = args.filters.projectId;
+      }
+      const gateWhere = [`${quoteIdent("id", "column name")} = @id`, ...tenancyConditions].join(" AND ");
       await this.db.runWithAbortRetry(
         () => this.database.runTransactionAsync(async (tx) => {
           try {
+            const [gateRows] = await tx.run({
+              sql: `SELECT ${quoteIdent("id", "column name")} FROM ${quoteIdent(TABLE_EXPERIMENTS, "table name")}
+                    WHERE ${gateWhere} LIMIT 1`,
+              params: { id: args.id, ...tenancyParams },
+              json: true
+            });
+            if (!Array.isArray(gateRows) || gateRows.length === 0) {
+              await tx.commit();
+              return;
+            }
+            const cascadeWhere = [
+              `${quoteIdent("experimentId", "column name")} = @id`,
+              // Result rows carry the same organizationId/projectId as their parent,
+              // so applying tenancy here makes the cascade itself tenant-scoped.
+              ...tenancyConditions
+            ].join(" AND ");
             await tx.runUpdate({
-              sql: `DELETE FROM ${quoteIdent(TABLE_EXPERIMENT_RESULTS, "table name")}
-                    WHERE ${quoteIdent("experimentId", "column name")} = @id`,
-              params: { id: args.id }
+              sql: `DELETE FROM ${quoteIdent(TABLE_EXPERIMENT_RESULTS, "table name")} WHERE ${cascadeWhere}`,
+              params: { id: args.id, ...tenancyParams }
             });
             await tx.runUpdate({
-              sql: `DELETE FROM ${quoteIdent(TABLE_EXPERIMENTS, "table name")}
-                    WHERE ${quoteIdent("id", "column name")} = @id`,
-              params: { id: args.id }
+              sql: `DELETE FROM ${quoteIdent(TABLE_EXPERIMENTS, "table name")} WHERE ${gateWhere}`,
+              params: { id: args.id, ...tenancyParams }
             });
             await tx.commit();
           } catch (err) {
@@ -4473,10 +4597,30 @@ var ExperimentsSpanner = class _ExperimentsSpanner extends ExperimentsStorage {
   }
   async getExperimentResultById(args) {
     try {
-      const row = await this.db.load({
-        tableName: TABLE_EXPERIMENT_RESULTS,
-        keys: { id: args.id }
+      const hasTenancy = args.filters?.organizationId !== void 0 || args.filters?.projectId !== void 0;
+      if (!hasTenancy) {
+        const row2 = await this.db.load({
+          tableName: TABLE_EXPERIMENT_RESULTS,
+          keys: { id: args.id }
+        });
+        return row2 ? rowToExperimentResult(row2) : null;
+      }
+      const conditions = [`${quoteIdent("id", "column name")} = @id`];
+      const params = { id: args.id };
+      if (args.filters?.organizationId !== void 0) {
+        conditions.push(`${quoteIdent("organizationId", "column name")} = @organizationId`);
+        params.organizationId = args.filters.organizationId;
+      }
+      if (args.filters?.projectId !== void 0) {
+        conditions.push(`${quoteIdent("projectId", "column name")} = @projectId`);
+        params.projectId = args.filters.projectId;
+      }
+      const [rows] = await this.database.run({
+        sql: `SELECT * FROM ${quoteIdent(TABLE_EXPERIMENT_RESULTS, "table name")} WHERE ${conditions.join(" AND ")} LIMIT 1`,
+        params,
+        json: true
       });
+      const row = rows[0];
       return row ? rowToExperimentResult(row) : null;
     } catch (error) {
       throw new MastraError(
@@ -4559,6 +4703,23 @@ var ExperimentsSpanner = class _ExperimentsSpanner extends ExperimentsStorage {
   }
   async deleteExperimentResults(args) {
     try {
+      if (args.filters?.organizationId !== void 0 || args.filters?.projectId !== void 0) {
+        const conditions = [`${quoteIdent("experimentId", "column name")} = @experimentId`];
+        const params = { experimentId: args.experimentId };
+        if (args.filters?.organizationId !== void 0) {
+          conditions.push(`${quoteIdent("organizationId", "column name")} = @organizationId`);
+          params.organizationId = args.filters.organizationId;
+        }
+        if (args.filters?.projectId !== void 0) {
+          conditions.push(`${quoteIdent("projectId", "column name")} = @projectId`);
+          params.projectId = args.filters.projectId;
+        }
+        await this.db.runDml({
+          sql: `DELETE FROM ${quoteIdent(TABLE_EXPERIMENT_RESULTS, "table name")} WHERE ${conditions.join(" AND ")}`,
+          params
+        });
+        return;
+      }
       await this.db.runDml({
         sql: `DELETE FROM ${quoteIdent(TABLE_EXPERIMENT_RESULTS, "table name")}
               WHERE ${quoteIdent("experimentId", "column name")} = @experimentId`,
@@ -10548,6 +10709,16 @@ function transformScoreRow(row) {
   const normalized = transformFromSpannerRow({ tableName: TABLE_SCORERS, row });
   return transformScoreRow$1(normalized, { convertTimestamps: true });
 }
+function applyTenancyFilters(conditions, params, filters) {
+  if (filters?.organizationId !== void 0) {
+    conditions.push(`${quoteIdent("organizationId", "column name")} = @organizationId`);
+    params.organizationId = filters.organizationId;
+  }
+  if (filters?.projectId !== void 0) {
+    conditions.push(`${quoteIdent("projectId", "column name")} = @projectId`);
+    params.projectId = filters.projectId;
+  }
+}
 var ScoresSpanner = class _ScoresSpanner extends ScoresStorage {
   database;
   db;
@@ -10726,7 +10897,8 @@ var ScoresSpanner = class _ScoresSpanner extends ScoresStorage {
     pagination,
     entityId,
     entityType,
-    source
+    source,
+    filters
   }) {
     try {
       const conditions = [`${quoteIdent("scorerId", "column name")} = @scorerId`];
@@ -10743,6 +10915,7 @@ var ScoresSpanner = class _ScoresSpanner extends ScoresStorage {
         conditions.push(`${quoteIdent("source", "column name")} = @source`);
         params.source = source;
       }
+      applyTenancyFilters(conditions, params, filters);
       return await this.listScoresByConditions(conditions, params, pagination);
     } catch (error) {
       throw new MastraError(
@@ -10758,14 +10931,14 @@ var ScoresSpanner = class _ScoresSpanner extends ScoresStorage {
   }
   async listScoresByRunId({
     runId,
-    pagination
+    pagination,
+    filters
   }) {
     try {
-      return await this.listScoresByConditions(
-        [`${quoteIdent("runId", "column name")} = @runId`],
-        { runId },
-        pagination
-      );
+      const conditions = [`${quoteIdent("runId", "column name")} = @runId`];
+      const params = { runId };
+      applyTenancyFilters(conditions, params, filters);
+      return await this.listScoresByConditions(conditions, params, pagination);
     } catch (error) {
       throw new MastraError(
         {
@@ -10781,17 +10954,17 @@ var ScoresSpanner = class _ScoresSpanner extends ScoresStorage {
   async listScoresByEntityId({
     entityId,
     entityType,
-    pagination
+    pagination,
+    filters
   }) {
     try {
-      return await this.listScoresByConditions(
-        [
-          `${quoteIdent("entityId", "column name")} = @entityId`,
-          `${quoteIdent("entityType", "column name")} = @entityType`
-        ],
-        { entityId, entityType },
-        pagination
-      );
+      const conditions = [
+        `${quoteIdent("entityId", "column name")} = @entityId`,
+        `${quoteIdent("entityType", "column name")} = @entityType`
+      ];
+      const params = { entityId, entityType };
+      applyTenancyFilters(conditions, params, filters);
+      return await this.listScoresByConditions(conditions, params, pagination);
     } catch (error) {
       throw new MastraError(
         {
@@ -10807,14 +10980,17 @@ var ScoresSpanner = class _ScoresSpanner extends ScoresStorage {
   async listScoresBySpan({
     traceId,
     spanId,
-    pagination
+    pagination,
+    filters
   }) {
     try {
-      return await this.listScoresByConditions(
-        [`${quoteIdent("traceId", "column name")} = @traceId`, `${quoteIdent("spanId", "column name")} = @spanId`],
-        { traceId, spanId },
-        pagination
-      );
+      const conditions = [
+        `${quoteIdent("traceId", "column name")} = @traceId`,
+        `${quoteIdent("spanId", "column name")} = @spanId`
+      ];
+      const params = { traceId, spanId };
+      applyTenancyFilters(conditions, params, filters);
+      return await this.listScoresByConditions(conditions, params, pagination);
     } catch (error) {
       throw new MastraError(
         {