@mastra/dynamodb 0.0.0-vnext-20251104230439 → 0.0.0-vnext-20251119160359

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, normalizePerPage, MemoryStorage, resolveMessageLimit, calculatePagination, ScoresStorage, TABLE_AI_SPANS, TABLE_RESOURCES, TABLE_TRACES, TABLE_SCORERS, TABLE_WORKFLOW_SNAPSHOT, TABLE_MESSAGES, TABLE_THREADS } from '@mastra/core/storage';
+import { MastraStorage, StoreOperations, WorkflowsStorage, normalizePerPage, MemoryStorage, calculatePagination, ScoresStorage, TABLE_SPANS, TABLE_RESOURCES, TABLE_TRACES, TABLE_SCORERS, 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/evals';
@@ -1071,7 +1071,7 @@ var MemoryStorageDynamoDB = class extends MemoryStorage {
   async deleteThread({ threadId }) {
     this.logger.debug("Deleting thread", { threadId });
     try {
-      const { messages } = await this.getMessages({ threadId });
+      const { messages } = await this.listMessages({ threadId, perPage: false });
       if (messages.length > 0) {
         const batchSize = 25;
         for (let i = 0; i < messages.length; i += batchSize) {
@@ -1100,70 +1100,6 @@ var MemoryStorageDynamoDB = class extends MemoryStorage {
       );
     }
   }
-  async getMessages({
-    threadId,
-    resourceId,
-    selectBy
-  }) {
-    this.logger.debug("Getting messages", { threadId, selectBy });
-    try {
-      if (!threadId.trim()) throw new Error("threadId must be a non-empty string");
-      const messages = [];
-      const limit = resolveMessageLimit({ last: selectBy?.last, defaultLimit: Number.MAX_SAFE_INTEGER });
-      if (selectBy?.include?.length) {
-        const includeMessages = await this._getIncludedMessages(threadId, selectBy);
-        if (includeMessages) {
-          messages.push(...includeMessages);
-        }
-      }
-      if (limit !== 0) {
-        const query = this.service.entities.message.query.byThread({ entity: "message", threadId });
-        let results;
-        if (limit !== Number.MAX_SAFE_INTEGER && limit > 0) {
-          results = await query.go({ limit, order: "desc" });
-          results.data = results.data.reverse();
-        } else {
-          results = await query.go();
-        }
-        let allThreadMessages = results.data.map((data) => this.parseMessageData(data)).filter((msg) => "content" in msg);
-        allThreadMessages.sort((a, b) => {
-          const timeA = a.createdAt.getTime();
-          const timeB = b.createdAt.getTime();
-          if (timeA === timeB) {
-            return a.id.localeCompare(b.id);
-          }
-          return timeA - timeB;
-        });
-        messages.push(...allThreadMessages);
-      }
-      messages.sort((a, b) => {
-        const timeA = a.createdAt.getTime();
-        const timeB = b.createdAt.getTime();
-        if (timeA === timeB) {
-          return a.id.localeCompare(b.id);
-        }
-        return timeA - timeB;
-      });
-      const uniqueMessages = messages.filter(
-        (message, index, self) => index === self.findIndex((m) => m.id === message.id)
-      );
-      const list = new MessageList({ threadId, resourceId }).add(
-        uniqueMessages,
-        "memory"
-      );
-      return { messages: list.get.all.db() };
-    } catch (error) {
-      throw new MastraError(
-        {
-          id: "STORAGE_DYNAMODB_STORE_GET_MESSAGES_FAILED",
-          domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.THIRD_PARTY,
-          details: { threadId, resourceId: resourceId ?? "" }
-        },
-        error
-      );
-    }
-  }
   async listMessagesById({ messageIds }) {
     this.logger.debug("Getting messages by ID", { messageIds });
     if (messageIds.length === 0) return { messages: [] };
@@ -1217,7 +1153,7 @@ var MemoryStorageDynamoDB = class extends MemoryStorage {
           new Error("page must be >= 0")
         );
       }
-      const { field, direction } = this.parseOrderBy(orderBy);
+      const { field, direction } = this.parseOrderBy(orderBy, "ASC");
       this.logger.debug("Getting messages with listMessages", {
         threadId,
         resourceId,
@@ -1718,7 +1654,7 @@ var StoreOperationsDynamoDB = class extends StoreOperations {
       [TABLE_SCORERS]: "score",
       [TABLE_TRACES]: "trace",
       [TABLE_RESOURCES]: "resource",
-      [TABLE_AI_SPANS]: "ai_span"
+      [TABLE_SPANS]: "ai_span"
     };
     return mapping[tableName] || null;
   }
@@ -2332,7 +2268,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
@@ -2417,6 +2352,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);
@@ -2524,7 +2464,7 @@ var DynamoDBStore = class extends MastraStorage {
   hasInitialized = null;
   stores;
   constructor({ name, config }) {
-    super({ name });
+    super({ id: config.id, name });
     try {
       if (!config.tableName || typeof config.tableName !== "string" || config.tableName.trim() === "") {
         throw new Error("DynamoDBStore: config.tableName must be provided and cannot be empty.");
@@ -2682,10 +2622,6 @@ var DynamoDBStore = class extends MastraStorage {
   async deleteThread({ threadId }) {
     return this.stores.memory.deleteThread({ threadId });
   }
-  // Message operations
-  async getMessages(args) {
-    return this.stores.memory.getMessages(args);
-  }
   async listMessagesById(args) {
     return this.stores.memory.listMessagesById(args);
   }