@mastra/dynamodb 1.1.1 → 1.1.2-alpha.0

package/dist/index.js

@@ -797,6 +797,26 @@ var scoreEntity = new Entity({
       type: "string",
       required: false
     },
+    organizationId: {
+      type: "string",
+      required: false
+    },
+    projectId: {
+      type: "string",
+      required: false
+    },
+    batchId: {
+      type: "string",
+      required: false
+    },
+    datasetId: {
+      type: "string",
+      required: false
+    },
+    datasetItemId: {
+      type: "string",
+      required: false
+    },
     threadId: {
       type: "string",
       required: false
@@ -2260,6 +2280,11 @@ var MemoryStorageDynamoDB = class extends MemoryStorage {
     }
   }
 };
+function matchesTenancy(score, filters) {
+  if (filters?.organizationId !== void 0 && score.organizationId !== filters.organizationId) return false;
+  if (filters?.projectId !== void 0 && score.projectId !== filters.projectId) return false;
+  return true;
+}
 var ScoresStorageDynamoDB = class extends ScoresStorage {
   service;
   ttlConfig;
@@ -2398,7 +2423,8 @@ var ScoresStorageDynamoDB = class extends ScoresStorage {
     pagination,
     entityId,
     entityType,
-    source
+    source,
+    filters
   }) {
     try {
       const query = this.service.entities.score.query.byScorer({ entity: "score", scorerId });
@@ -2413,6 +2439,7 @@ var ScoresStorageDynamoDB = class extends ScoresStorage {
       if (source) {
         allScores = allScores.filter((score) => score.source === source);
       }
+      allScores = allScores.filter((score) => matchesTenancy(score, filters));
       allScores.sort((a, b) => b.createdAt.getTime() - a.createdAt.getTime());
       const { page, perPage: perPageInput } = pagination;
       const perPage = normalizePerPage(perPageInput, Number.MAX_SAFE_INTEGER);
@@ -2450,13 +2477,14 @@ var ScoresStorageDynamoDB = class extends ScoresStorage {
   }
   async listScoresByRunId({
     runId,
-    pagination
+    pagination,
+    filters
   }) {
     this.logger.debug("Getting scores by run ID", { runId, pagination });
     try {
       const query = this.service.entities.score.query.byRun({ entity: "score", runId });
       const results = await query.go();
-      const allScores = results.data.map((data) => this.parseScoreData(data));
+      const allScores = results.data.map((data) => this.parseScoreData(data)).filter((score) => matchesTenancy(score, filters));
       allScores.sort((a, b) => b.createdAt.getTime() - a.createdAt.getTime());
       const { page, perPage: perPageInput } = pagination;
       const perPage = normalizePerPage(perPageInput, Number.MAX_SAFE_INTEGER);
@@ -2488,7 +2516,8 @@ var ScoresStorageDynamoDB = class extends ScoresStorage {
   async listScoresByEntityId({
     entityId,
     entityType,
-    pagination
+    pagination,
+    filters
   }) {
     this.logger.debug("Getting scores by entity ID", { entityId, entityType, pagination });
     try {
@@ -2496,6 +2525,7 @@ var ScoresStorageDynamoDB = class extends ScoresStorage {
       const results = await query.go();
       let allScores = results.data.map((data) => this.parseScoreData(data));
       allScores = allScores.filter((score) => score.entityType === entityType);
+      allScores = allScores.filter((score) => matchesTenancy(score, filters));
       allScores.sort((a, b) => b.createdAt.getTime() - a.createdAt.getTime());
       const { page, perPage: perPageInput } = pagination;
       const perPage = normalizePerPage(perPageInput, Number.MAX_SAFE_INTEGER);
@@ -2527,13 +2557,14 @@ var ScoresStorageDynamoDB = class extends ScoresStorage {
   async listScoresBySpan({
     traceId,
     spanId,
-    pagination
+    pagination,
+    filters
   }) {
     this.logger.debug("Getting scores by span", { traceId, spanId, pagination });
     try {
       const query = this.service.entities.score.query.bySpan({ entity: "score", traceId, spanId });
       const results = await query.go();
-      const allScores = results.data.map((data) => this.parseScoreData(data));
+      const allScores = results.data.map((data) => this.parseScoreData(data)).filter((score) => matchesTenancy(score, filters));
       allScores.sort((a, b) => b.createdAt.getTime() - a.createdAt.getTime());
       const { page, perPage: perPageInput } = pagination;
       const perPage = normalizePerPage(perPageInput, Number.MAX_SAFE_INTEGER);