@mastra/dynamodb 0.0.0-remove-unused-model-providers-api-20251030210744 → 0.0.0-safe-stringify-telemetry-20251205024938

package/dist/index.js

@@ -1,7 +1,7 @@
 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, WorkflowsStorage, MemoryStorage, resolveMessageLimit, ScoresStorage, TABLE_AI_SPANS, TABLE_RESOURCES, TABLE_TRACES, TABLE_SCORERS, TABLE_WORKFLOW_SNAPSHOT, TABLE_MESSAGES, TABLE_THREADS } from '@mastra/core/storage';
+import { MastraStorage, StoreOperations, TracesStorage, TABLE_TRACES, WorkflowsStorage, MemoryStorage, resolveMessageLimit, ScoresStorage, SCORERS_SCHEMA, LegacyEvalsStorage, TABLE_AI_SPANS, TABLE_RESOURCES, 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';
@@ -423,6 +423,10 @@ var scoreEntity = new Entity({
         return value;
       }
     },
+    preprocessPrompt: {
+      type: "string",
+      required: false
+    },
     preprocessStepResult: {
       type: "string",
       required: false,
@@ -562,7 +566,7 @@ var scoreEntity = new Entity({
         return value;
       }
     },
-    requestContext: {
+    runtimeContext: {
       type: "string",
       required: false,
       set: (value) => {
@@ -933,6 +937,187 @@ function getElectroDbService(client, tableName) {
     }
   );
 }
+var LegacyEvalsDynamoDB = class extends LegacyEvalsStorage {
+  service;
+  tableName;
+  constructor({ service, tableName }) {
+    super();
+    this.service = service;
+    this.tableName = tableName;
+  }
+  // Eval operations
+  async getEvalsByAgentName(agentName, type) {
+    this.logger.debug("Getting evals for agent", { agentName, type });
+    try {
+      const query = this.service.entities.eval.query.byAgent({ entity: "eval", agent_name: agentName });
+      const results = await query.go({ order: "desc", limit: 100 });
+      if (!results.data.length) {
+        return [];
+      }
+      let filteredData = results.data;
+      if (type) {
+        filteredData = filteredData.filter((evalRecord) => {
+          try {
+            const testInfo = evalRecord.test_info && typeof evalRecord.test_info === "string" ? JSON.parse(evalRecord.test_info) : void 0;
+            if (type === "test" && !testInfo) {
+              return false;
+            }
+            if (type === "live" && testInfo) {
+              return false;
+            }
+          } catch (e) {
+            this.logger.warn("Failed to parse test_info during filtering", { record: evalRecord, error: e });
+          }
+          return true;
+        });
+      }
+      return filteredData.map((evalRecord) => {
+        try {
+          return {
+            input: evalRecord.input,
+            output: evalRecord.output,
+            // Safely parse result and test_info
+            result: evalRecord.result && typeof evalRecord.result === "string" ? JSON.parse(evalRecord.result) : void 0,
+            agentName: evalRecord.agent_name,
+            createdAt: evalRecord.created_at,
+            // Keep as string from DDB?
+            metricName: evalRecord.metric_name,
+            instructions: evalRecord.instructions,
+            runId: evalRecord.run_id,
+            globalRunId: evalRecord.global_run_id,
+            testInfo: evalRecord.test_info && typeof evalRecord.test_info === "string" ? JSON.parse(evalRecord.test_info) : void 0
+          };
+        } catch (parseError) {
+          this.logger.error("Failed to parse eval record", { record: evalRecord, error: parseError });
+          return {
+            agentName: evalRecord.agent_name,
+            createdAt: evalRecord.created_at,
+            runId: evalRecord.run_id,
+            globalRunId: evalRecord.global_run_id
+          };
+        }
+      });
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: "STORAGE_DYNAMODB_STORE_GET_EVALS_BY_AGENT_NAME_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { agentName }
+        },
+        error
+      );
+    }
+  }
+  async getEvals(options = {}) {
+    const { agentName, type, page = 0, perPage = 100, dateRange } = options;
+    this.logger.debug("Getting evals with pagination", { agentName, type, page, perPage, dateRange });
+    try {
+      let query;
+      if (agentName) {
+        query = this.service.entities.eval.query.byAgent({ entity: "eval", agent_name: agentName });
+      } else {
+        query = this.service.entities.eval.query.byEntity({ entity: "eval" });
+      }
+      const results = await query.go({
+        order: "desc",
+        pages: "all"
+        // Get all pages to apply filtering and pagination
+      });
+      if (!results.data.length) {
+        return {
+          evals: [],
+          total: 0,
+          page,
+          perPage,
+          hasMore: false
+        };
+      }
+      let filteredData = results.data;
+      if (type) {
+        filteredData = filteredData.filter((evalRecord) => {
+          try {
+            const testInfo = evalRecord.test_info && typeof evalRecord.test_info === "string" ? JSON.parse(evalRecord.test_info) : void 0;
+            if (type === "test" && !testInfo) {
+              return false;
+            }
+            if (type === "live" && testInfo) {
+              return false;
+            }
+          } catch (e) {
+            this.logger.warn("Failed to parse test_info during filtering", { record: evalRecord, error: e });
+          }
+          return true;
+        });
+      }
+      if (dateRange) {
+        const fromDate = dateRange.start;
+        const toDate = dateRange.end;
+        filteredData = filteredData.filter((evalRecord) => {
+          const recordDate = new Date(evalRecord.created_at);
+          if (fromDate && recordDate < fromDate) {
+            return false;
+          }
+          if (toDate && recordDate > toDate) {
+            return false;
+          }
+          return true;
+        });
+      }
+      const total = filteredData.length;
+      const start = page * perPage;
+      const end = start + perPage;
+      const paginatedData = filteredData.slice(start, end);
+      const evals = paginatedData.map((evalRecord) => {
+        try {
+          return {
+            input: evalRecord.input,
+            output: evalRecord.output,
+            result: evalRecord.result && typeof evalRecord.result === "string" ? JSON.parse(evalRecord.result) : void 0,
+            agentName: evalRecord.agent_name,
+            createdAt: evalRecord.created_at,
+            metricName: evalRecord.metric_name,
+            instructions: evalRecord.instructions,
+            runId: evalRecord.run_id,
+            globalRunId: evalRecord.global_run_id,
+            testInfo: evalRecord.test_info && typeof evalRecord.test_info === "string" ? JSON.parse(evalRecord.test_info) : void 0
+          };
+        } catch (parseError) {
+          this.logger.error("Failed to parse eval record", { record: evalRecord, error: parseError });
+          return {
+            agentName: evalRecord.agent_name,
+            createdAt: evalRecord.created_at,
+            runId: evalRecord.run_id,
+            globalRunId: evalRecord.global_run_id
+          };
+        }
+      });
+      const hasMore = end < total;
+      return {
+        evals,
+        total,
+        page,
+        perPage,
+        hasMore
+      };
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: "STORAGE_DYNAMODB_STORE_GET_EVALS_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: {
+            agentName: agentName || "all",
+            type: type || "all",
+            page,
+            perPage
+          }
+        },
+        error
+      );
+    }
+  }
+};
 var MemoryStorageDynamoDB = class extends MemoryStorage {
   service;
   constructor({ service }) {
@@ -1037,7 +1222,7 @@ var MemoryStorageDynamoDB = class extends MemoryStorage {
         resourceId: thread.resourceId,
         title: threadData.title,
         createdAt: thread.createdAt || now,
-        updatedAt: thread.updatedAt || now,
+        updatedAt: now,
         metadata: thread.metadata
       };
     } catch (error) {
@@ -1189,7 +1374,10 @@ var MemoryStorageDynamoDB = class extends MemoryStorage {
       );
     }
   }
-  async listMessagesById({ messageIds }) {
+  async getMessagesById({
+    messageIds,
+    format
+  }) {
     this.logger.debug("Getting messages by ID", { messageIds });
     if (messageIds.length === 0) return [];
     try {
@@ -1202,6 +1390,7 @@ var MemoryStorageDynamoDB = class extends MemoryStorage {
         (message, index, self) => index === self.findIndex((m) => m.id === message.id)
       );
       const list = new MessageList().add(uniqueMessages, "memory");
+      if (format === `v1`) return list.get.all.v1();
       return list.get.all.v2();
     } catch (error) {
       throw new MastraError(
@@ -1215,150 +1404,6 @@ var MemoryStorageDynamoDB = class extends MemoryStorage {
       );
     }
   }
-  async listMessages(args) {
-    const { threadId, resourceId, include, filter, limit, offset = 0, orderBy } = args;
-    if (!threadId.trim()) {
-      throw new MastraError(
-        {
-          id: "STORAGE_DYNAMODB_LIST_MESSAGES_INVALID_THREAD_ID",
-          domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.THIRD_PARTY,
-          details: { threadId }
-        },
-        new Error("threadId must be a non-empty string")
-      );
-    }
-    try {
-      let perPage = 40;
-      if (limit !== void 0) {
-        if (limit === false) {
-          perPage = Number.MAX_SAFE_INTEGER;
-        } else if (limit === 0) {
-          perPage = 0;
-        } else if (typeof limit === "number" && limit > 0) {
-          perPage = limit;
-        }
-      }
-      const page = perPage === 0 ? 0 : Math.floor(offset / perPage);
-      const sortField = orderBy?.field || "createdAt";
-      const sortDirection = orderBy?.direction || "DESC";
-      this.logger.debug("Getting messages with listMessages", {
-        threadId,
-        resourceId,
-        limit,
-        offset,
-        perPage,
-        page,
-        sortField,
-        sortDirection
-      });
-      const query = this.service.entities.message.query.byThread({ entity: "message", threadId });
-      const results = await query.go();
-      let allThreadMessages = results.data.map((data) => this.parseMessageData(data)).filter((msg) => "content" in msg && typeof msg.content === "object");
-      if (resourceId) {
-        allThreadMessages = allThreadMessages.filter((msg) => msg.resourceId === resourceId);
-      }
-      if (filter?.dateRange) {
-        const dateRange = filter.dateRange;
-        allThreadMessages = allThreadMessages.filter((msg) => {
-          const createdAt = new Date(msg.createdAt).getTime();
-          if (dateRange.start) {
-            const startTime = dateRange.start instanceof Date ? dateRange.start.getTime() : new Date(dateRange.start).getTime();
-            if (createdAt < startTime) return false;
-          }
-          if (dateRange.end) {
-            const endTime = dateRange.end instanceof Date ? dateRange.end.getTime() : new Date(dateRange.end).getTime();
-            if (createdAt > endTime) return false;
-          }
-          return true;
-        });
-      }
-      allThreadMessages.sort((a, b) => {
-        const aValue = sortField === "createdAt" ? new Date(a.createdAt).getTime() : a[sortField];
-        const bValue = sortField === "createdAt" ? new Date(b.createdAt).getTime() : b[sortField];
-        if (aValue === bValue) {
-          return a.id.localeCompare(b.id);
-        }
-        return sortDirection === "ASC" ? aValue - bValue : bValue - aValue;
-      });
-      const total = allThreadMessages.length;
-      const paginatedMessages = allThreadMessages.slice(offset, offset + perPage);
-      const paginatedCount = paginatedMessages.length;
-      if (total === 0 && paginatedCount === 0) {
-        return {
-          messages: [],
-          total: 0,
-          page,
-          perPage,
-          hasMore: false
-        };
-      }
-      const messageIds = new Set(paginatedMessages.map((m) => m.id));
-      let includeMessages = [];
-      if (include && include.length > 0) {
-        const selectBy = { include };
-        includeMessages = await this._getIncludedMessages(threadId, selectBy);
-        for (const includeMsg of includeMessages) {
-          if (!messageIds.has(includeMsg.id)) {
-            paginatedMessages.push(includeMsg);
-            messageIds.add(includeMsg.id);
-          }
-        }
-      }
-      const list = new MessageList().add(paginatedMessages, "memory");
-      let finalMessages = list.get.all.v2();
-      finalMessages = finalMessages.sort((a, b) => {
-        const aValue = sortField === "createdAt" ? new Date(a.createdAt).getTime() : a[sortField];
-        const bValue = sortField === "createdAt" ? new Date(b.createdAt).getTime() : b[sortField];
-        if (aValue === bValue) {
-          return a.id.localeCompare(b.id);
-        }
-        return sortDirection === "ASC" ? aValue - bValue : bValue - aValue;
-      });
-      const returnedThreadMessageIds = new Set(finalMessages.filter((m) => m.threadId === threadId).map((m) => m.id));
-      const allThreadMessagesReturned = returnedThreadMessageIds.size >= total;
-      const hasMore = limit === false ? false : allThreadMessagesReturned ? false : offset + paginatedCount < total;
-      return {
-        messages: finalMessages,
-        total,
-        page,
-        perPage,
-        hasMore
-      };
-    } catch (error) {
-      const mastraError = new MastraError(
-        {
-          id: "STORAGE_DYNAMODB_STORE_LIST_MESSAGES_FAILED",
-          domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.THIRD_PARTY,
-          details: {
-            threadId,
-            resourceId: resourceId ?? ""
-          }
-        },
-        error
-      );
-      this.logger?.error?.(mastraError.toString());
-      this.logger?.trackException?.(mastraError);
-      return {
-        messages: [],
-        total: 0,
-        page: Math.floor(offset / (limit === false ? Number.MAX_SAFE_INTEGER : limit || 40)),
-        perPage: limit === false ? Number.MAX_SAFE_INTEGER : limit || 40,
-        hasMore: false
-      };
-    }
-  }
-  /**
-   * @todo When migrating from getThreadsByResourceIdPaginated to this method,
-   * implement orderBy and sortDirection support for full sorting capabilities
-   */
-  async listThreadsByResourceId(args) {
-    const { resourceId, limit, offset } = args;
-    const page = Math.floor(offset / limit);
-    const perPage = limit;
-    return this.getThreadsByResourceIdPaginated({ resourceId, page, perPage });
-  }
   async saveMessages(args) {
     const { messages, format = "v1" } = args;
     this.logger.debug("Saving messages", { count: messages.length });
@@ -1525,7 +1570,8 @@ var MemoryStorageDynamoDB = class extends MemoryStorage {
       const paginatedMessages = messages.slice(start, end);
       const hasMore = end < total;
       const list = new MessageList({ threadId, resourceId }).add(paginatedMessages, "memory");
-      const finalMessages = format === "v2" ? list.get.all.v2() : list.get.all.v1();
+      let finalMessages = format === "v2" ? list.get.all.v2() : list.get.all.v1();
+      finalMessages = finalMessages.sort((a, b) => new Date(a.createdAt).getTime() - new Date(b.createdAt).getTime());
       return {
         messages: finalMessages,
         total,
@@ -1814,6 +1860,7 @@ var StoreOperationsDynamoDB = class extends StoreOperations {
       [TABLE_THREADS]: "thread",
       [TABLE_MESSAGES]: "message",
       [TABLE_WORKFLOW_SNAPSHOT]: "workflow_snapshot",
+      [TABLE_EVALS]: "eval",
       [TABLE_SCORERS]: "score",
       [TABLE_TRACES]: "trace",
       [TABLE_RESOURCES]: "resource",
@@ -2002,10 +2049,6 @@ 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}`);
@@ -2119,8 +2162,17 @@ var ScoresStorageDynamoDB = class extends ScoresStorage {
   }
   // Helper function to parse score data (handle JSON fields)
   parseScoreData(data) {
+    const result = {};
+    for (const key of Object.keys(SCORERS_SCHEMA)) {
+      if (["traceId", "resourceId", "threadId", "spanId"].includes(key)) {
+        result[key] = data[key] === "" ? null : data[key];
+        continue;
+      }
+      result[key] = data[key];
+    }
+    result.entity = data.entityData ? data.entityData : null;
     return {
-      ...data,
+      ...result,
       // Convert date strings back to Date objects for consistency
       createdAt: data.createdAt ? new Date(data.createdAt) : /* @__PURE__ */ new Date(),
       updatedAt: data.updatedAt ? new Date(data.updatedAt) : /* @__PURE__ */ new Date()
@@ -2182,7 +2234,7 @@ var ScoresStorageDynamoDB = class extends ScoresStorage {
       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),
-      requestContext: typeof validatedScore.requestContext === "string" ? validatedScore.requestContext : JSON.stringify(validatedScore.requestContext),
+      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,
@@ -2379,6 +2431,239 @@ 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 });
+    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
+        };
+      });
+      return {
+        traces,
+        total,
+        page,
+        perPage,
+        hasMore: end < total
+      };
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: "STORAGE_DYNAMODB_STORE_GET_TRACES_PAGINATED_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY
+        },
+        error
+      );
+    }
+  }
+};
 function formatWorkflowRun(snapshotData) {
   return {
     workflowName: snapshotData.workflow_name,
@@ -2400,7 +2685,7 @@ var WorkflowStorageDynamoDB = class extends WorkflowsStorage {
     // runId,
     // stepId,
     // result,
-    // requestContext,
+    // runtimeContext,
   }) {
     throw new Error("Method not implemented.");
   }
@@ -2427,7 +2712,6 @@ var WorkflowStorageDynamoDB = class extends WorkflowsStorage {
         workflow_name: workflowName,
         run_id: runId,
         snapshot: JSON.stringify(snapshot),
-        // Stringify the snapshot object
         createdAt: now,
         updatedAt: now,
         resourceId
@@ -2473,7 +2757,7 @@ var WorkflowStorageDynamoDB = class extends WorkflowsStorage {
       );
     }
   }
-  async listWorkflowRuns(args) {
+  async getWorkflowRuns(args) {
     this.logger.debug("Getting workflow runs", { args });
     try {
       const limit = args?.limit || 10;
@@ -2499,6 +2783,11 @@ var WorkflowStorageDynamoDB = class extends WorkflowsStorage {
         });
         if (pageResults.data && pageResults.data.length > 0) {
           let pageFilteredData = pageResults.data;
+          if (args?.status) {
+            pageFilteredData = pageFilteredData.filter((snapshot) => {
+              return snapshot.snapshot.status === args.status;
+            });
+          }
           if (args?.fromDate || args?.toDate) {
             pageFilteredData = pageFilteredData.filter((snapshot) => {
               const createdAt = new Date(snapshot.createdAt);
@@ -2629,11 +2918,14 @@ 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
@@ -2775,6 +3067,12 @@ var DynamoDBStore = class extends MastraStorage {
   }) {
     return this.stores.memory.getMessages({ threadId, resourceId, selectBy, format });
   }
+  async getMessagesById({
+    messageIds,
+    format
+  }) {
+    return this.stores.memory.getMessagesById({ messageIds, format });
+  }
   async saveMessages(args) {
     return this.stores.memory.saveMessages(args);
   }
@@ -2787,15 +3085,25 @@ 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,
     runId,
     stepId,
     result,
-    requestContext
+    runtimeContext
   }) {
-    return this.stores.workflows.updateWorkflowResults({ workflowName, runId, stepId, result, requestContext });
+    return this.stores.workflows.updateWorkflowResults({ workflowName, runId, stepId, result, runtimeContext });
   }
   async updateWorkflowState({
     workflowName,
@@ -2818,8 +3126,8 @@ var DynamoDBStore = class extends MastraStorage {
   }) {
     return this.stores.workflows.loadWorkflowSnapshot({ workflowName, runId });
   }
-  async listWorkflowRuns(args) {
-    return this.stores.workflows.listWorkflowRuns(args);
+  async getWorkflowRuns(args) {
+    return this.stores.workflows.getWorkflowRuns(args);
   }
   async getWorkflowRunById(args) {
     return this.stores.workflows.getWorkflowRunById(args);
@@ -2837,6 +3145,13 @@ var DynamoDBStore = class extends MastraStorage {
   }) {
     return this.stores.memory.updateResource({ resourceId, workingMemory, metadata });
   }
+  // Eval operations
+  async getEvalsByAgentName(agentName, type) {
+    return this.stores.legacyEvals.getEvalsByAgentName(agentName, type);
+  }
+  async getEvals(options) {
+    return this.stores.legacyEvals.getEvals(options);
+  }
   /**
    * Closes the DynamoDB client connection and cleans up resources.
    * Should be called when the store is no longer needed, e.g., at the end of tests or application shutdown.