@mastra/cloudflare 1.5.0 → 1.5.1

package/dist/{chunk-63YHZAI7.cjs → chunk-UQ4IOLSC.cjs}

@@ -1781,6 +1781,14 @@ var MemoryStorageDO = class extends storage.MemoryStorage {
     }
   }
 };
+function applyTenancyFilters(query, filters) {
+  if (filters?.organizationId !== void 0) {
+    query.andWhere("organizationId = ?", filters.organizationId);
+  }
+  if (filters?.projectId !== void 0) {
+    query.andWhere("projectId = ?", filters.projectId);
+  }
+}
 function transformScoreRow(row) {
   return storage.transformScoreRow(row, {
     preferredTimestampFields: {
@@ -1891,7 +1899,8 @@ var ScoresStorageDO = class extends storage.ScoresStorage {
     entityId,
     entityType,
     source,
-    pagination
+    pagination,
+    filters
   }) {
     try {
       const { page, perPage: perPageInput } = pagination;
@@ -1908,6 +1917,7 @@ var ScoresStorageDO = class extends storage.ScoresStorage {
       if (source) {
         countQuery.andWhere("source = ?", source);
       }
+      applyTenancyFilters(countQuery, filters);
       const countResult = await this.#db.executeQuery(countQuery.build());
       const total = Array.isArray(countResult) ? Number(countResult?.[0]?.count ?? 0) : Number(countResult?.count ?? 0);
       if (total === 0) {
@@ -1933,6 +1943,7 @@ var ScoresStorageDO = class extends storage.ScoresStorage {
       if (source) {
         selectQuery.andWhere("source = ?", source);
       }
+      applyTenancyFilters(selectQuery, filters);
       selectQuery.limit(limitValue).offset(start);
       const { sql, params } = selectQuery.build();
       const results = await this.#db.executeQuery({ sql, params });
@@ -1959,7 +1970,8 @@ var ScoresStorageDO = class extends storage.ScoresStorage {
   }
   async listScoresByRunId({
     runId,
-    pagination
+    pagination,
+    filters
   }) {
     try {
       const { page, perPage: perPageInput } = pagination;
@@ -1967,6 +1979,7 @@ var ScoresStorageDO = class extends storage.ScoresStorage {
       const { offset: start, perPage: perPageForResponse } = storage.calculatePagination(page, perPageInput, perPage);
       const fullTableName = this.#db.getTableName(storage.TABLE_SCORERS);
       const countQuery = createSqlBuilder().count().from(fullTableName).where("runId = ?", runId);
+      applyTenancyFilters(countQuery, filters);
       const countResult = await this.#db.executeQuery(countQuery.build());
       const total = Array.isArray(countResult) ? Number(countResult?.[0]?.count ?? 0) : Number(countResult?.count ?? 0);
       if (total === 0) {
@@ -1982,7 +1995,9 @@ var ScoresStorageDO = class extends storage.ScoresStorage {
       }
       const end = perPageInput === false ? total : start + perPage;
       const limitValue = perPageInput === false ? total : perPage;
-      const selectQuery = createSqlBuilder().select("*").from(fullTableName).where("runId = ?", runId).limit(limitValue).offset(start);
+      const selectQuery = createSqlBuilder().select("*").from(fullTableName).where("runId = ?", runId);
+      applyTenancyFilters(selectQuery, filters);
+      selectQuery.limit(limitValue).offset(start);
       const { sql, params } = selectQuery.build();
       const results = await this.#db.executeQuery({ sql, params });
       const scores = Array.isArray(results) ? results.map((r) => transformScoreRow(r)) : [];
@@ -2009,7 +2024,8 @@ var ScoresStorageDO = class extends storage.ScoresStorage {
   async listScoresByEntityId({
     entityId,
     entityType,
-    pagination
+    pagination,
+    filters
   }) {
     try {
       const { page, perPage: perPageInput } = pagination;
@@ -2017,6 +2033,7 @@ var ScoresStorageDO = class extends storage.ScoresStorage {
       const { offset: start, perPage: perPageForResponse } = storage.calculatePagination(page, perPageInput, perPage);
       const fullTableName = this.#db.getTableName(storage.TABLE_SCORERS);
       const countQuery = createSqlBuilder().count().from(fullTableName).where("entityId = ?", entityId).andWhere("entityType = ?", entityType);
+      applyTenancyFilters(countQuery, filters);
       const countResult = await this.#db.executeQuery(countQuery.build());
       const total = Array.isArray(countResult) ? Number(countResult?.[0]?.count ?? 0) : Number(countResult?.count ?? 0);
       if (total === 0) {
@@ -2032,7 +2049,9 @@ var ScoresStorageDO = class extends storage.ScoresStorage {
       }
       const end = perPageInput === false ? total : start + perPage;
       const limitValue = perPageInput === false ? total : perPage;
-      const selectQuery = createSqlBuilder().select("*").from(fullTableName).where("entityId = ?", entityId).andWhere("entityType = ?", entityType).limit(limitValue).offset(start);
+      const selectQuery = createSqlBuilder().select("*").from(fullTableName).where("entityId = ?", entityId).andWhere("entityType = ?", entityType);
+      applyTenancyFilters(selectQuery, filters);
+      selectQuery.limit(limitValue).offset(start);
       const { sql, params } = selectQuery.build();
       const results = await this.#db.executeQuery({ sql, params });
       const scores = Array.isArray(results) ? results.map((r) => transformScoreRow(r)) : [];
@@ -2059,7 +2078,8 @@ var ScoresStorageDO = class extends storage.ScoresStorage {
   async listScoresBySpan({
     traceId,
     spanId,
-    pagination
+    pagination,
+    filters
   }) {
     try {
       const { page, perPage: perPageInput } = pagination;
@@ -2067,6 +2087,7 @@ var ScoresStorageDO = class extends storage.ScoresStorage {
       const { offset: start, perPage: perPageForResponse } = storage.calculatePagination(page, perPageInput, perPage);
       const fullTableName = this.#db.getTableName(storage.TABLE_SCORERS);
       const countQuery = createSqlBuilder().count().from(fullTableName).where("traceId = ?", traceId).andWhere("spanId = ?", spanId);
+      applyTenancyFilters(countQuery, filters);
       const countResult = await this.#db.executeQuery(countQuery.build());
       const total = Array.isArray(countResult) ? Number(countResult?.[0]?.count ?? 0) : Number(countResult?.count ?? 0);
       if (total === 0) {
@@ -2082,7 +2103,9 @@ var ScoresStorageDO = class extends storage.ScoresStorage {
       }
       const end = perPageInput === false ? total : start + perPage;
       const limitValue = perPageInput === false ? total : perPage;
-      const selectQuery = createSqlBuilder().select("*").from(fullTableName).where("traceId = ?", traceId).andWhere("spanId = ?", spanId).orderBy("createdAt", "DESC").limit(limitValue).offset(start);
+      const selectQuery = createSqlBuilder().select("*").from(fullTableName).where("traceId = ?", traceId).andWhere("spanId = ?", spanId);
+      applyTenancyFilters(selectQuery, filters);
+      selectQuery.orderBy("createdAt", "DESC").limit(limitValue).offset(start);
       const { sql, params } = selectQuery.build();
       const results = await this.#db.executeQuery({ sql, params });
       const scores = Array.isArray(results) ? results.map((r) => transformScoreRow(r)) : [];
@@ -2423,5 +2446,5 @@ exports.DOStore = DOStore;
 exports.MemoryStorageDO = MemoryStorageDO;
 exports.ScoresStorageDO = ScoresStorageDO;
 exports.WorkflowsStorageDO = WorkflowsStorageDO;
-//# sourceMappingURL=chunk-63YHZAI7.cjs.map
-//# sourceMappingURL=chunk-63YHZAI7.cjs.map
+//# sourceMappingURL=chunk-UQ4IOLSC.cjs.map
+//# sourceMappingURL=chunk-UQ4IOLSC.cjs.map