@mastra/cloudflare 1.5.0 → 1.5.1

package/dist/{chunk-ADP2QAK4.js → chunk-OMTTR37Z.js}

@@ -1779,6 +1779,14 @@ var MemoryStorageDO = class extends 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 transformScoreRow$1(row, {
     preferredTimestampFields: {
@@ -1889,7 +1897,8 @@ var ScoresStorageDO = class extends ScoresStorage {
     entityId,
     entityType,
     source,
-    pagination
+    pagination,
+    filters
   }) {
     try {
       const { page, perPage: perPageInput } = pagination;
@@ -1906,6 +1915,7 @@ var ScoresStorageDO = class extends 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) {
@@ -1931,6 +1941,7 @@ var ScoresStorageDO = class extends 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 });
@@ -1957,7 +1968,8 @@ var ScoresStorageDO = class extends ScoresStorage {
   }
   async listScoresByRunId({
     runId,
-    pagination
+    pagination,
+    filters
   }) {
     try {
       const { page, perPage: perPageInput } = pagination;
@@ -1965,6 +1977,7 @@ var ScoresStorageDO = class extends ScoresStorage {
       const { offset: start, perPage: perPageForResponse } = calculatePagination(page, perPageInput, perPage);
       const fullTableName = this.#db.getTableName(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) {
@@ -1980,7 +1993,9 @@ var ScoresStorageDO = class extends 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)) : [];
@@ -2007,7 +2022,8 @@ var ScoresStorageDO = class extends ScoresStorage {
   async listScoresByEntityId({
     entityId,
     entityType,
-    pagination
+    pagination,
+    filters
   }) {
     try {
       const { page, perPage: perPageInput } = pagination;
@@ -2015,6 +2031,7 @@ var ScoresStorageDO = class extends ScoresStorage {
       const { offset: start, perPage: perPageForResponse } = calculatePagination(page, perPageInput, perPage);
       const fullTableName = this.#db.getTableName(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) {
@@ -2030,7 +2047,9 @@ var ScoresStorageDO = class extends 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)) : [];
@@ -2057,7 +2076,8 @@ var ScoresStorageDO = class extends ScoresStorage {
   async listScoresBySpan({
     traceId,
     spanId,
-    pagination
+    pagination,
+    filters
   }) {
     try {
       const { page, perPage: perPageInput } = pagination;
@@ -2065,6 +2085,7 @@ var ScoresStorageDO = class extends ScoresStorage {
       const { offset: start, perPage: perPageForResponse } = calculatePagination(page, perPageInput, perPage);
       const fullTableName = this.#db.getTableName(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) {
@@ -2080,7 +2101,9 @@ var ScoresStorageDO = class extends 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)) : [];
@@ -2415,5 +2438,5 @@ var CloudflareDOStorage = class extends MastraCompositeStore {
 var DOStore = CloudflareDOStorage;
 
 export { BackgroundTasksStorageDO, CloudflareDOStorage, DODB, DOStore, MemoryStorageDO, ScoresStorageDO, WorkflowsStorageDO };
-//# sourceMappingURL=chunk-ADP2QAK4.js.map
-//# sourceMappingURL=chunk-ADP2QAK4.js.map
+//# sourceMappingURL=chunk-OMTTR37Z.js.map
+//# sourceMappingURL=chunk-OMTTR37Z.js.map