@mastra/mongodb 0.14.2 → 0.14.3-alpha.0

package/CHANGELOG.md

@@ -1,5 +1,16 @@
 # @mastra/mongodb
 
+## 0.14.3-alpha.0
+
+### Patch Changes
+
+- Mongodb support to fetch scores by trace and spanId ([#8154](https://github.com/mastra-ai/mastra/pull/8154))
+
+- Update peer deps ([#8154](https://github.com/mastra-ai/mastra/pull/8154))
+
+- Updated dependencies [[`504438b`](https://github.com/mastra-ai/mastra/commit/504438b961bde211071186bba63a842c4e3db879), [`a7243e2`](https://github.com/mastra-ai/mastra/commit/a7243e2e58762667a6e3921e755e89d6bb0a3282), [`7fceb0a`](https://github.com/mastra-ai/mastra/commit/7fceb0a327d678e812f90f5387c5bc4f38bd039e), [`df64f9e`](https://github.com/mastra-ai/mastra/commit/df64f9ef814916fff9baedd861c988084e7c41de), [`809eea0`](https://github.com/mastra-ai/mastra/commit/809eea092fa80c3f69b9eaf078d843b57fd2a88e), [`683e5a1`](https://github.com/mastra-ai/mastra/commit/683e5a1466e48b686825b2c11f84680f296138e4), [`3679378`](https://github.com/mastra-ai/mastra/commit/3679378673350aa314741dc826f837b1984149bc), [`7775bc2`](https://github.com/mastra-ai/mastra/commit/7775bc20bb1ad1ab24797fb420e4f96c65b0d8ec), [`db1891a`](https://github.com/mastra-ai/mastra/commit/db1891a4707443720b7cd8a260dc7e1d49b3609c), [`e8f379d`](https://github.com/mastra-ai/mastra/commit/e8f379d390efa264c4e0874f9ac0cf8839b07777), [`652066b`](https://github.com/mastra-ai/mastra/commit/652066bd1efc6bb6813ba950ed1d7573e8b7d9d4), [`ea8d386`](https://github.com/mastra-ai/mastra/commit/ea8d386cd8c5593664515fd5770c06bf2aa980ef), [`c2a4919`](https://github.com/mastra-ai/mastra/commit/c2a4919ba6797d8bdb1509e02287496eef69303e), [`0130986`](https://github.com/mastra-ai/mastra/commit/0130986fc62d0edcc626dd593282661dbb9af141)]:
+  - @mastra/core@0.19.0-alpha.1
+
 ## 0.14.2
 
 ### Patch Changes

package/dist/index.cjs

@@ -7,6 +7,7 @@ var uuid = require('uuid');
 var filter = require('@mastra/core/vector/filter');
 var storage = require('@mastra/core/storage');
 var agent = require('@mastra/core/agent');
+var scores = require('@mastra/core/scores');
 
 // src/vector/index.ts
 var MongoDBFilterTranslator = class extends filter.BaseFilterTranslator {
@@ -1609,21 +1610,31 @@ function transformScoreRow(row) {
       console.warn("Failed to parse runtimeContext:", e);
     }
   }
+  let metadataValue = null;
+  if (row.metadata) {
+    try {
+      metadataValue = typeof row.metadata === "string" ? storage.safelyParseJSON(row.metadata) : row.metadata;
+    } catch (e) {
+      console.warn("Failed to parse metadata:", e);
+    }
+  }
   return {
     id: row.id,
     entityId: row.entityId,
     entityType: row.entityType,
     scorerId: row.scorerId,
     traceId: row.traceId,
+    spanId: row.spanId,
     runId: row.runId,
     scorer: scorerValue,
     preprocessStepResult: preprocessStepResultValue,
+    preprocessPrompt: row.preprocessPrompt,
     analyzeStepResult: analyzeStepResultValue,
+    generateScorePrompt: row.generateScorePrompt,
     score: row.score,
-    reason: row.reason,
-    extractPrompt: row.extractPrompt,
     analyzePrompt: row.analyzePrompt,
     reasonPrompt: row.reasonPrompt,
+    metadata: metadataValue,
     input: inputValue,
     output: outputValue,
     additionalContext: row.additionalContext,
@@ -1663,34 +1674,47 @@ var ScoresStorageMongoDB = class extends storage.ScoresStorage {
     }
   }
   async saveScore(score) {
+    let validatedScore;
+    try {
+      validatedScore = scores.saveScorePayloadSchema.parse(score);
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "STORAGE_MONGODB_STORE_SAVE_SCORE_VALIDATION_FAILED",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY
+        },
+        error$1
+      );
+    }
     try {
       const now = /* @__PURE__ */ new Date();
       const scoreId = `score-${Date.now()}-${Math.random().toString(36).substr(2, 9)}`;
       const scoreData = {
         id: scoreId,
-        entityId: score.entityId,
-        entityType: score.entityType,
-        scorerId: score.scorerId,
-        traceId: score.traceId || "",
-        runId: score.runId,
-        scorer: typeof score.scorer === "string" ? storage.safelyParseJSON(score.scorer) : score.scorer,
-        preprocessStepResult: typeof score.preprocessStepResult === "string" ? storage.safelyParseJSON(score.preprocessStepResult) : score.preprocessStepResult,
-        analyzeStepResult: typeof score.analyzeStepResult === "string" ? storage.safelyParseJSON(score.analyzeStepResult) : score.analyzeStepResult,
-        score: score.score,
-        reason: score.reason,
-        preprocessPrompt: score.preprocessPrompt,
-        generateScorePrompt: score.generateScorePrompt,
-        generateReasonPrompt: score.generateReasonPrompt,
-        analyzePrompt: score.analyzePrompt,
-        reasonPrompt: score.reasonPrompt,
-        input: typeof score.input === "string" ? storage.safelyParseJSON(score.input) : score.input,
-        output: typeof score.output === "string" ? storage.safelyParseJSON(score.output) : score.output,
-        additionalContext: score.additionalContext,
-        runtimeContext: typeof score.runtimeContext === "string" ? storage.safelyParseJSON(score.runtimeContext) : score.runtimeContext,
-        entity: typeof score.entity === "string" ? storage.safelyParseJSON(score.entity) : score.entity,
-        source: score.source,
-        resourceId: score.resourceId || "",
-        threadId: score.threadId || "",
+        entityId: validatedScore.entityId,
+        entityType: validatedScore.entityType,
+        scorerId: validatedScore.scorerId,
+        traceId: validatedScore.traceId || "",
+        spanId: validatedScore.spanId || "",
+        runId: validatedScore.runId,
+        scorer: typeof validatedScore.scorer === "string" ? storage.safelyParseJSON(validatedScore.scorer) : validatedScore.scorer,
+        preprocessStepResult: typeof validatedScore.preprocessStepResult === "string" ? storage.safelyParseJSON(validatedScore.preprocessStepResult) : validatedScore.preprocessStepResult,
+        analyzeStepResult: typeof validatedScore.analyzeStepResult === "string" ? storage.safelyParseJSON(validatedScore.analyzeStepResult) : validatedScore.analyzeStepResult,
+        score: validatedScore.score,
+        reason: validatedScore.reason,
+        preprocessPrompt: validatedScore.preprocessPrompt,
+        generateScorePrompt: validatedScore.generateScorePrompt,
+        generateReasonPrompt: validatedScore.generateReasonPrompt,
+        analyzePrompt: validatedScore.analyzePrompt,
+        input: typeof validatedScore.input === "string" ? storage.safelyParseJSON(validatedScore.input) : validatedScore.input,
+        output: typeof validatedScore.output === "string" ? storage.safelyParseJSON(validatedScore.output) : validatedScore.output,
+        additionalContext: validatedScore.additionalContext,
+        runtimeContext: typeof validatedScore.runtimeContext === "string" ? storage.safelyParseJSON(validatedScore.runtimeContext) : validatedScore.runtimeContext,
+        entity: typeof validatedScore.entity === "string" ? storage.safelyParseJSON(validatedScore.entity) : validatedScore.entity,
+        source: validatedScore.source,
+        resourceId: validatedScore.resourceId || "",
+        threadId: validatedScore.threadId || "",
         createdAt: now,
         updatedAt: now
       };
@@ -1858,6 +1882,51 @@ var ScoresStorageMongoDB = class extends storage.ScoresStorage {
       );
     }
   }
+  async getScoresBySpan({
+    traceId,
+    spanId,
+    pagination
+  }) {
+    try {
+      const query = { traceId, spanId };
+      const collection = await this.operations.getCollection(storage.TABLE_SCORERS);
+      const total = await collection.countDocuments(query);
+      const currentOffset = pagination.page * pagination.perPage;
+      if (total === 0) {
+        return {
+          scores: [],
+          pagination: {
+            total: 0,
+            page: pagination.page,
+            perPage: pagination.perPage,
+            hasMore: false
+          }
+        };
+      }
+      const documents = await collection.find(query).sort({ createdAt: "desc" }).skip(currentOffset).limit(pagination.perPage).toArray();
+      const scores = documents.map((row) => transformScoreRow(row));
+      const hasMore = currentOffset + scores.length < total;
+      return {
+        scores,
+        pagination: {
+          total,
+          page: pagination.page,
+          perPage: pagination.perPage,
+          hasMore
+        }
+      };
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "STORAGE_MONGODB_STORE_GET_SCORES_BY_SPAN_FAILED",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY,
+          details: { traceId, spanId, page: pagination.page, perPage: pagination.perPage }
+        },
+        error$1
+      );
+    }
+  }
 };
 var TracesStorageMongoDB = class extends storage.TracesStorage {
   operations;
@@ -2196,7 +2265,8 @@ var MongoDBStore = class extends storage.MastraStorage {
       resourceWorkingMemory: true,
       hasColumn: false,
       createTable: false,
-      deleteMessages: false
+      deleteMessages: false,
+      getScoresBySpan: true
     };
   }
   constructor(config) {
@@ -2394,6 +2464,13 @@ var MongoDBStore = class extends storage.MastraStorage {
   }) {
     return this.stores.scores.getScoresByScorerId({ scorerId, pagination, entityId, entityType, source });
   }
+  async getScoresBySpan({
+    traceId,
+    spanId,
+    pagination
+  }) {
+    return this.stores.scores.getScoresBySpan({ traceId, spanId, pagination });
+  }
   /**
    * RESOURCES
    */