@mastra/upstash 0.14.0 → 0.14.1

package/dist/index.js

@@ -251,6 +251,8 @@ function processRecord(tableName, record) {
     });
   } else if (tableName === TABLE_EVALS) {
     key = getKey(tableName, { id: record.run_id });
+  } else if (tableName === TABLE_SCORERS) {
+    key = getKey(tableName, { runId: record.runId });
   } else {
     key = getKey(tableName, { id: record.id });
   }
@@ -1199,6 +1201,9 @@ var ScoresUpstash = class extends ScoresStorage {
   }
   async getScoresByScorerId({
     scorerId,
+    entityId,
+    entityType,
+    source,
     pagination = { page: 0, perPage: 20 }
   }) {
     const pattern = `${TABLE_SCORERS}:*`;
@@ -1212,7 +1217,14 @@ var ScoresUpstash = class extends ScoresStorage {
     const pipeline = this.client.pipeline();
     keys.forEach((key) => pipeline.get(key));
     const results = await pipeline.exec();
-    const filtered = results.map((row) => row).filter((row) => !!row && typeof row === "object" && row.scorerId === scorerId);
+    const filtered = results.map((row) => row).filter((row) => {
+      if (!row || typeof row !== "object") return false;
+      if (row.scorerId !== scorerId) return false;
+      if (entityId && row.entityId !== entityId) return false;
+      if (entityType && row.entityType !== entityType) return false;
+      if (source && row.source !== source) return false;
+      return true;
+    });
     const total = filtered.length;
     const { page, perPage } = pagination;
     const start = page * perPage;
@@ -1836,9 +1848,12 @@ var UpstashStore = class extends MastraStorage {
   }
   async getScoresByScorerId({
     scorerId,
-    pagination
+    pagination,
+    entityId,
+    entityType,
+    source
   }) {
-    return this.stores.scores.getScoresByScorerId({ scorerId, pagination });
+    return this.stores.scores.getScoresByScorerId({ scorerId, pagination, entityId, entityType, source });
   }
 };
 var UpstashFilterTranslator = class extends BaseFilterTranslator {