@mastra/dynamodb 0.0.0-pgvector-index-fix-20250905222058 → 0.0.0-playground-studio-cloud-20251031080052

package/dist/index.js

@@ -1,9 +1,10 @@
 import { DynamoDBClient, DescribeTableCommand } from '@aws-sdk/client-dynamodb';
 import { DynamoDBDocumentClient } from '@aws-sdk/lib-dynamodb';
 import { MastraError, ErrorCategory, ErrorDomain } from '@mastra/core/error';
-import { MastraStorage, StoreOperations, TracesStorage, TABLE_TRACES, WorkflowsStorage, MemoryStorage, resolveMessageLimit, ScoresStorage, LegacyEvalsStorage, TABLE_AI_SPANS, TABLE_RESOURCES, TABLE_SCORERS, TABLE_EVALS, TABLE_WORKFLOW_SNAPSHOT, TABLE_MESSAGES, TABLE_THREADS } from '@mastra/core/storage';
+import { MastraStorage, StoreOperations, WorkflowsStorage, MemoryStorage, resolveMessageLimit, ScoresStorage, LegacyEvalsStorage, TABLE_AI_SPANS, TABLE_RESOURCES, TABLE_TRACES, TABLE_SCORERS, TABLE_EVALS, TABLE_WORKFLOW_SNAPSHOT, TABLE_MESSAGES, TABLE_THREADS } from '@mastra/core/storage';
 import { Entity, Service } from 'electrodb';
 import { MessageList } from '@mastra/core/agent';
+import { saveScorePayloadSchema } from '@mastra/core/scores';
 
 // src/storage/index.ts
 
@@ -370,6 +371,10 @@ var scoreEntity = new Entity({
       type: "string",
       required: false
     },
+    spanId: {
+      type: "string",
+      required: false
+    },
     runId: {
       type: "string",
       required: true
@@ -656,6 +661,11 @@ var scoreEntity = new Entity({
       index: "gsi6",
       pk: { field: "gsi6pk", composite: ["entity", "threadId"] },
       sk: { field: "gsi6sk", composite: ["createdAt"] }
+    },
+    bySpan: {
+      index: "gsi7",
+      pk: { field: "gsi7pk", composite: ["entity", "traceId", "spanId"] },
+      sk: { field: "gsi7sk", composite: ["createdAt"] }
     }
   }
 });
@@ -2034,6 +2044,10 @@ var StoreOperationsDynamoDB = class extends StoreOperations {
             if (!item.id) throw new Error(`Missing required key 'id' for entity 'score'`);
             key.id = item.id;
             break;
+          case "resource":
+            if (!item.id) throw new Error(`Missing required key 'id' for entity 'resource'`);
+            key.id = item.id;
+            break;
           default:
             this.logger.warn(`Unknown entity type encountered during clearTable: ${entityName}`);
             throw new Error(`Cannot construct delete key for unknown entity type: ${entityName}`);
@@ -2176,34 +2190,47 @@ var ScoresStorageDynamoDB = class extends ScoresStorage {
     }
   }
   async saveScore(score) {
-    this.logger.debug("Saving score", { scorerId: score.scorerId, runId: score.runId });
+    let validatedScore;
+    try {
+      validatedScore = saveScorePayloadSchema.parse(score);
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: "STORAGE_DYNAMODB_STORE_SAVE_SCORE_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY
+        },
+        error
+      );
+    }
     const now = /* @__PURE__ */ new Date();
     const scoreId = `score-${Date.now()}-${Math.random().toString(36).substr(2, 9)}`;
     const scoreData = {
       entity: "score",
       id: scoreId,
-      scorerId: score.scorerId,
-      traceId: score.traceId || "",
-      runId: score.runId,
-      scorer: typeof score.scorer === "string" ? score.scorer : JSON.stringify(score.scorer),
-      preprocessStepResult: typeof score.preprocessStepResult === "string" ? score.preprocessStepResult : JSON.stringify(score.preprocessStepResult),
-      analyzeStepResult: typeof score.analyzeStepResult === "string" ? score.analyzeStepResult : JSON.stringify(score.analyzeStepResult),
-      score: score.score,
-      reason: score.reason,
-      preprocessPrompt: score.preprocessPrompt,
-      generateScorePrompt: score.generateScorePrompt,
-      analyzePrompt: score.analyzePrompt,
-      reasonPrompt: score.reasonPrompt,
-      input: typeof score.input === "string" ? score.input : JSON.stringify(score.input),
-      output: typeof score.output === "string" ? score.output : JSON.stringify(score.output),
-      additionalContext: typeof score.additionalContext === "string" ? score.additionalContext : JSON.stringify(score.additionalContext),
-      runtimeContext: typeof score.runtimeContext === "string" ? score.runtimeContext : JSON.stringify(score.runtimeContext),
-      entityType: score.entityType,
-      entityData: typeof score.entity === "string" ? score.entity : JSON.stringify(score.entity),
-      entityId: score.entityId,
-      source: score.source,
-      resourceId: score.resourceId || "",
-      threadId: score.threadId || "",
+      scorerId: validatedScore.scorerId,
+      traceId: validatedScore.traceId || "",
+      spanId: validatedScore.spanId || "",
+      runId: validatedScore.runId,
+      scorer: typeof validatedScore.scorer === "string" ? validatedScore.scorer : JSON.stringify(validatedScore.scorer),
+      preprocessStepResult: typeof validatedScore.preprocessStepResult === "string" ? validatedScore.preprocessStepResult : JSON.stringify(validatedScore.preprocessStepResult),
+      analyzeStepResult: typeof validatedScore.analyzeStepResult === "string" ? validatedScore.analyzeStepResult : JSON.stringify(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" ? validatedScore.input : JSON.stringify(validatedScore.input),
+      output: typeof validatedScore.output === "string" ? validatedScore.output : JSON.stringify(validatedScore.output),
+      additionalContext: typeof validatedScore.additionalContext === "string" ? validatedScore.additionalContext : JSON.stringify(validatedScore.additionalContext),
+      runtimeContext: typeof validatedScore.runtimeContext === "string" ? validatedScore.runtimeContext : JSON.stringify(validatedScore.runtimeContext),
+      entityType: validatedScore.entityType,
+      entityData: typeof validatedScore.entity === "string" ? validatedScore.entity : JSON.stringify(validatedScore.entity),
+      entityId: validatedScore.entityId,
+      source: validatedScore.source,
+      resourceId: validatedScore.resourceId || "",
+      threadId: validatedScore.threadId || "",
       createdAt: now.toISOString(),
       updatedAt: now.toISOString()
     };
@@ -2356,234 +2383,38 @@ var ScoresStorageDynamoDB = class extends ScoresStorage {
       );
     }
   }
-};
-var TracesStorageDynamoDB = class extends TracesStorage {
-  service;
-  operations;
-  constructor({ service, operations }) {
-    super();
-    this.service = service;
-    this.operations = operations;
-  }
-  // Trace operations
-  async getTraces(args) {
-    const { name, scope, page, perPage } = args;
-    this.logger.debug("Getting traces", { name, scope, page, perPage });
-    try {
-      let query;
-      if (name) {
-        query = this.service.entities.trace.query.byName({ entity: "trace", name });
-      } else if (scope) {
-        query = this.service.entities.trace.query.byScope({ entity: "trace", scope });
-      } else {
-        this.logger.warn("Performing a scan operation on traces - consider using a more specific query");
-        query = this.service.entities.trace.scan;
-      }
-      let items = [];
-      let cursor = null;
-      let pagesFetched = 0;
-      const startPage = page > 0 ? page : 1;
-      do {
-        const results = await query.go({ cursor, limit: perPage });
-        pagesFetched++;
-        if (pagesFetched === startPage) {
-          items = results.data;
-          break;
-        }
-        cursor = results.cursor;
-        if (!cursor && results.data.length > 0 && pagesFetched < startPage) {
-          break;
-        }
-      } while (cursor && pagesFetched < startPage);
-      return items;
-    } catch (error) {
-      throw new MastraError(
-        {
-          id: "STORAGE_DYNAMODB_STORE_GET_TRACES_FAILED",
-          domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.THIRD_PARTY
-        },
-        error
-      );
-    }
-  }
-  async batchTraceInsert({ records }) {
-    this.logger.debug("Batch inserting traces", { count: records.length });
-    if (!records.length) {
-      return;
-    }
-    try {
-      const recordsToSave = records.map((rec) => ({ entity: "trace", ...rec }));
-      await this.operations.batchInsert({
-        tableName: TABLE_TRACES,
-        records: recordsToSave
-        // Pass records with 'entity' included
-      });
-    } catch (error) {
-      throw new MastraError(
-        {
-          id: "STORAGE_DYNAMODB_STORE_BATCH_TRACE_INSERT_FAILED",
-          domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.THIRD_PARTY,
-          details: { count: records.length }
-        },
-        error
-      );
-    }
-  }
-  async getTracesPaginated(args) {
-    const { name, scope, page = 0, perPage = 100, attributes, filters, dateRange } = args;
-    this.logger.debug("Getting traces with pagination", { name, scope, page, perPage, attributes, filters, dateRange });
+  async getScoresBySpan({
+    traceId,
+    spanId,
+    pagination
+  }) {
+    this.logger.debug("Getting scores by span", { traceId, spanId, pagination });
     try {
-      let query;
-      if (name) {
-        query = this.service.entities.trace.query.byName({ entity: "trace", name });
-      } else if (scope) {
-        query = this.service.entities.trace.query.byScope({ entity: "trace", scope });
-      } else {
-        this.logger.warn("Performing a scan operation on traces - consider using a more specific query");
-        query = this.service.entities.trace.scan;
-      }
-      const results = await query.go({
-        order: "desc",
-        pages: "all"
-        // Get all pages to apply filtering and pagination
-      });
-      if (!results.data.length) {
-        return {
-          traces: [],
-          total: 0,
-          page,
-          perPage,
-          hasMore: false
-        };
-      }
-      let filteredData = results.data;
-      if (attributes) {
-        filteredData = filteredData.filter((item) => {
-          try {
-            let itemAttributes = {};
-            if (item.attributes) {
-              if (typeof item.attributes === "string") {
-                if (item.attributes === "[object Object]") {
-                  itemAttributes = {};
-                } else {
-                  try {
-                    itemAttributes = JSON.parse(item.attributes);
-                  } catch {
-                    itemAttributes = {};
-                  }
-                }
-              } else if (typeof item.attributes === "object") {
-                itemAttributes = item.attributes;
-              }
-            }
-            return Object.entries(attributes).every(([key, value]) => itemAttributes[key] === value);
-          } catch (e) {
-            this.logger.warn("Failed to parse attributes during filtering", { item, error: e });
-            return false;
-          }
-        });
-      }
-      if (dateRange?.start) {
-        filteredData = filteredData.filter((item) => {
-          const itemDate = new Date(item.createdAt);
-          return itemDate >= dateRange.start;
-        });
-      }
-      if (dateRange?.end) {
-        filteredData = filteredData.filter((item) => {
-          const itemDate = new Date(item.createdAt);
-          return itemDate <= dateRange.end;
-        });
-      }
-      const total = filteredData.length;
-      const start = page * perPage;
-      const end = start + perPage;
-      const paginatedData = filteredData.slice(start, end);
-      const traces = paginatedData.map((item) => {
-        let attributes2;
-        if (item.attributes) {
-          if (typeof item.attributes === "string") {
-            if (item.attributes === "[object Object]") {
-              attributes2 = void 0;
-            } else {
-              try {
-                attributes2 = JSON.parse(item.attributes);
-              } catch {
-                attributes2 = void 0;
-              }
-            }
-          } else if (typeof item.attributes === "object") {
-            attributes2 = item.attributes;
-          }
-        }
-        let status;
-        if (item.status) {
-          if (typeof item.status === "string") {
-            try {
-              status = JSON.parse(item.status);
-            } catch {
-              status = void 0;
-            }
-          } else if (typeof item.status === "object") {
-            status = item.status;
-          }
-        }
-        let events;
-        if (item.events) {
-          if (typeof item.events === "string") {
-            try {
-              events = JSON.parse(item.events);
-            } catch {
-              events = void 0;
-            }
-          } else if (Array.isArray(item.events)) {
-            events = item.events;
-          }
-        }
-        let links;
-        if (item.links) {
-          if (typeof item.links === "string") {
-            try {
-              links = JSON.parse(item.links);
-            } catch {
-              links = void 0;
-            }
-          } else if (Array.isArray(item.links)) {
-            links = item.links;
-          }
-        }
-        return {
-          id: item.id,
-          parentSpanId: item.parentSpanId,
-          name: item.name,
-          traceId: item.traceId,
-          scope: item.scope,
-          kind: item.kind,
-          attributes: attributes2,
-          status,
-          events,
-          links,
-          other: item.other,
-          startTime: item.startTime,
-          endTime: item.endTime,
-          createdAt: item.createdAt
-        };
-      });
+      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));
+      allScores.sort((a, b) => b.createdAt.getTime() - a.createdAt.getTime());
+      const startIndex = pagination.page * pagination.perPage;
+      const endIndex = startIndex + pagination.perPage;
+      const paginatedScores = allScores.slice(startIndex, endIndex);
+      const total = allScores.length;
+      const hasMore = endIndex < total;
       return {
-        traces,
-        total,
-        page,
-        perPage,
-        hasMore: end < total
+        scores: paginatedScores,
+        pagination: {
+          total,
+          page: pagination.page,
+          perPage: pagination.perPage,
+          hasMore
+        }
       };
     } catch (error) {
       throw new MastraError(
         {
-          id: "STORAGE_DYNAMODB_STORE_GET_TRACES_PAGINATED_FAILED",
+          id: "STORAGE_DYNAMODB_STORE_GET_SCORES_BY_SPAN_FAILED",
           domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.THIRD_PARTY
+          category: ErrorCategory.THIRD_PARTY,
+          details: { traceId, spanId, page: pagination.page, perPage: pagination.perPage }
         },
         error
       );
@@ -2626,11 +2457,11 @@ var WorkflowStorageDynamoDB = class extends WorkflowsStorage {
   async persistWorkflowSnapshot({
     workflowName,
     runId,
+    resourceId,
     snapshot
   }) {
     this.logger.debug("Persisting workflow snapshot", { workflowName, runId });
     try {
-      const resourceId = "resourceId" in snapshot ? snapshot.resourceId : void 0;
       const now = (/* @__PURE__ */ new Date()).toISOString();
       const data = {
         entity: "workflow_snapshot",
@@ -2756,8 +2587,6 @@ var WorkflowStorageDynamoDB = class extends WorkflowsStorage {
   async getWorkflowRunById(args) {
     const { runId, workflowName } = args;
     this.logger.debug("Getting workflow run by ID", { runId, workflowName });
-    console.log("workflowName", workflowName);
-    console.log("runId", runId);
     try {
       if (workflowName) {
         this.logger.debug("WorkflowName provided, using direct GET operation.");
@@ -2767,7 +2596,6 @@ var WorkflowStorageDynamoDB = class extends WorkflowsStorage {
           workflow_name: workflowName,
           run_id: runId
         }).go();
-        console.log("result", result2);
         if (!result2.data) {
           return null;
         }
@@ -2843,14 +2671,12 @@ var DynamoDBStore = class extends MastraStorage {
         tableName: this.tableName,
         client: this.client
       });
-      const traces = new TracesStorageDynamoDB({ service: this.service, operations });
       const workflows = new WorkflowStorageDynamoDB({ service: this.service });
       const memory = new MemoryStorageDynamoDB({ service: this.service });
       const scores = new ScoresStorageDynamoDB({ service: this.service });
       this.stores = {
         operations,
         legacyEvals: new LegacyEvalsDynamoDB({ service: this.service, tableName: this.tableName }),
-        traces,
         workflows,
         memory,
         scores
@@ -2872,7 +2698,8 @@ var DynamoDBStore = class extends MastraStorage {
       resourceWorkingMemory: true,
       hasColumn: false,
       createTable: false,
-      deleteMessages: false
+      deleteMessages: false,
+      getScoresBySpan: true
     };
   }
   /**
@@ -3009,16 +2836,6 @@ var DynamoDBStore = class extends MastraStorage {
   async updateMessages(_args) {
     return this.stores.memory.updateMessages(_args);
   }
-  // Trace operations
-  async getTraces(args) {
-    return this.stores.traces.getTraces(args);
-  }
-  async batchTraceInsert({ records }) {
-    return this.stores.traces.batchTraceInsert({ records });
-  }
-  async getTracesPaginated(_args) {
-    return this.stores.traces.getTracesPaginated(_args);
-  }
   // Workflow operations
   async updateWorkflowResults({
     workflowName,
@@ -3039,9 +2856,10 @@ var DynamoDBStore = class extends MastraStorage {
   async persistWorkflowSnapshot({
     workflowName,
     runId,
+    resourceId,
     snapshot
   }) {
-    return this.stores.workflows.persistWorkflowSnapshot({ workflowName, runId, snapshot });
+    return this.stores.workflows.persistWorkflowSnapshot({ workflowName, runId, resourceId, snapshot });
   }
   async loadWorkflowSnapshot({
     workflowName,
@@ -3130,6 +2948,13 @@ var DynamoDBStore = class extends MastraStorage {
   }) {
     return this.stores.scores.getScoresByScorerId({ scorerId, source, entityId, entityType, pagination });
   }
+  async getScoresBySpan({
+    traceId,
+    spanId,
+    pagination
+  }) {
+    return this.stores.scores.getScoresBySpan({ traceId, spanId, pagination });
+  }
 };
 
 export { DynamoDBStore };