@mastra/upstash 0.15.2-alpha.1 → 0.15.3-alpha.0

package/dist/index.js

@@ -2,6 +2,7 @@ import { MastraStorage, StoreOperations, TracesStorage, TABLE_TRACES, ScoresStor
 import { Redis } from '@upstash/redis';
 import { MastraError, ErrorCategory, ErrorDomain } from '@mastra/core/error';
 import { MessageList } from '@mastra/core/agent';
+import { saveScorePayloadSchema } from '@mastra/core/scores';
 import { randomUUID } from 'crypto';
 import { MastraVector } from '@mastra/core/vector';
 import { Index } from '@upstash/vector';
@@ -1291,7 +1292,20 @@ var ScoresUpstash = class extends ScoresStorage {
     };
   }
   async saveScore(score) {
-    const { key, processedRecord } = processRecord(TABLE_SCORERS, score);
+    let validatedScore;
+    try {
+      validatedScore = saveScorePayloadSchema.parse(score);
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: "STORAGE_UPSTASH_STORAGE_SAVE_SCORE_VALIDATION_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY
+        },
+        error
+      );
+    }
+    const { key, processedRecord } = processRecord(TABLE_SCORERS, validatedScore);
     try {
       await this.client.set(key, processedRecord);
       return { score };
@@ -1377,6 +1391,44 @@ var ScoresUpstash = class extends ScoresStorage {
       }
     };
   }
+  async getScoresBySpan({
+    traceId,
+    spanId,
+    pagination = { page: 0, perPage: 20 }
+  }) {
+    const pattern = `${TABLE_SCORERS}:*`;
+    const keys = await this.operations.scanKeys(pattern);
+    if (keys.length === 0) {
+      return {
+        scores: [],
+        pagination: { total: 0, page: pagination.page, perPage: pagination.perPage, hasMore: false }
+      };
+    }
+    const pipeline = this.client.pipeline();
+    keys.forEach((key) => pipeline.get(key));
+    const results = await pipeline.exec();
+    const filtered = results.map((row) => row).filter((row) => {
+      if (!row || typeof row !== "object") return false;
+      if (row.traceId !== traceId) return false;
+      if (row.spanId !== spanId) return false;
+      return true;
+    });
+    const total = filtered.length;
+    const { page, perPage } = pagination;
+    const start = page * perPage;
+    const end = start + perPage;
+    const paged = filtered.slice(start, end);
+    const scores = paged.map((row) => transformScoreRow(row));
+    return {
+      scores,
+      pagination: {
+        total,
+        page,
+        perPage,
+        hasMore: end < total
+      }
+    };
+  }
 };
 var TracesUpstash = class extends TracesStorage {
   client;
@@ -1749,7 +1801,8 @@ var UpstashStore = class extends MastraStorage {
       resourceWorkingMemory: true,
       hasColumn: false,
       createTable: false,
-      deleteMessages: true
+      deleteMessages: true,
+      getScoresBySpan: true
     };
   }
   /**
@@ -1943,6 +1996,13 @@ var UpstashStore = class extends MastraStorage {
   }) {
     return this.stores.scores.getScoresByScorerId({ scorerId, pagination, entityId, entityType, source });
   }
+  async getScoresBySpan({
+    traceId,
+    spanId,
+    pagination
+  }) {
+    return this.stores.scores.getScoresBySpan({ traceId, spanId, pagination });
+  }
 };
 var UpstashFilterTranslator = class extends BaseFilterTranslator {
   getSupportedOperators() {