@mastra/cloudflare 1.0.0-beta.2 → 1.0.0-beta.3

package/dist/index.js

@@ -1,5 +1,5 @@
 import { MastraError, ErrorCategory, ErrorDomain } from '@mastra/core/error';
-import { MastraStorage, TABLE_THREADS, TABLE_MESSAGES, TABLE_WORKFLOW_SNAPSHOT, TABLE_SCORERS, StoreOperations, TABLE_TRACES, WorkflowsStorage, ensureDate, normalizePerPage, MemoryStorage, calculatePagination, serializeDate, TABLE_RESOURCES, ScoresStorage, safelyParseJSON } from '@mastra/core/storage';
+import { MastraStorage, TABLE_THREADS, TABLE_MESSAGES, TABLE_WORKFLOW_SNAPSHOT, TABLE_SCORERS, StoreOperations, TABLE_TRACES, WorkflowsStorage, ensureDate, normalizePerPage, MemoryStorage, calculatePagination, serializeDate, TABLE_RESOURCES, ScoresStorage, transformScoreRow as transformScoreRow$1 } from '@mastra/core/storage';
 import Cloudflare from 'cloudflare';
 import { MessageList } from '@mastra/core/agent';
 import { saveScorePayloadSchema } from '@mastra/core/evals';
@@ -207,6 +207,17 @@ var MemoryStorageCloudflare = class extends MemoryStorage {
       );
     }
   }
+  /**
+   * Searches all threads in the KV store to find a message by its ID.
+   *
+   * **Performance Warning**: This method sequentially scans all threads to locate
+   * the message. For stores with many threads, this can result in significant
+   * latency and API calls. When possible, callers should provide the `threadId`
+   * directly to avoid this full scan.
+   *
+   * @param messageId - The globally unique message ID to search for
+   * @returns The message with its threadId if found, null otherwise
+   */
   async findMessageInAnyThread(messageId) {
     try {
       const prefix = this.operations.namespacePrefix ? `${this.operations.namespacePrefix}:` : "";
@@ -436,10 +447,25 @@ var MemoryStorageCloudflare = class extends MemoryStorage {
   async getFullOrder(orderKey) {
     return this.getRange(orderKey, 0, -1);
   }
-  async getIncludedMessagesWithContext(threadId, include, messageIds) {
+  /**
+   * Retrieves messages specified in the include array along with their surrounding context.
+   *
+   * **Performance Note**: When `threadId` is not provided in an include entry, this method
+   * must call `findMessageInAnyThread` which sequentially scans all threads in the KV store.
+   * For optimal performance, callers should provide `threadId` in include entries when known.
+   *
+   * @param include - Array of message IDs to include, optionally with context windows
+   * @param messageIds - Set to accumulate the message IDs that should be fetched
+   */
+  async getIncludedMessagesWithContext(include, messageIds) {
     await Promise.all(
       include.map(async (item) => {
-        const targetThreadId = item.threadId || threadId;
+        let targetThreadId = item.threadId;
+        if (!targetThreadId) {
+          const foundMessage = await this.findMessageInAnyThread(item.id);
+          if (!foundMessage) return;
+          targetThreadId = foundMessage.threadId;
+        }
         if (!targetThreadId) return;
         const threadMessagesKey = this.getThreadMessagesKey(targetThreadId);
         messageIds.add(item.id);
@@ -472,6 +498,13 @@ var MemoryStorageCloudflare = class extends MemoryStorage {
       this.logger?.debug(`No message order found for thread ${threadId}, skipping latest messages`);
     }
   }
+  /**
+   * Fetches and parses messages from one or more threads.
+   *
+   * **Performance Note**: When neither `include` entries with `threadId` nor `targetThreadId`
+   * are provided, this method falls back to `findMessageInAnyThread` which scans all threads.
+   * For optimal performance, provide `threadId` in include entries or specify `targetThreadId`.
+   */
   async fetchAndParseMessagesFromMultipleThreads(messageIds, include, targetThreadId) {
     const messageIdToThreadId = /* @__PURE__ */ new Map();
     if (include) {
@@ -513,6 +546,14 @@ var MemoryStorageCloudflare = class extends MemoryStorage {
     );
     return messages.filter((msg) => msg !== null);
   }
+  /**
+   * Retrieves messages by their IDs.
+   *
+   * **Performance Warning**: This method calls `findMessageInAnyThread` for each message ID,
+   * which scans all threads in the KV store. For large numbers of messages or threads,
+   * this can result in significant latency. Consider using `listMessages` with specific
+   * thread IDs when the thread context is known.
+   */
   async listMessagesById({ messageIds }) {
     if (messageIds.length === 0) return { messages: [] };
     try {
@@ -546,15 +587,17 @@ var MemoryStorageCloudflare = class extends MemoryStorage {
   }
   async listMessages(args) {
     const { threadId, resourceId, include, filter, perPage: perPageInput, page = 0, orderBy } = args;
-    if (!threadId.trim()) {
+    const threadIds = Array.isArray(threadId) ? threadId : [threadId];
+    const isValidThreadId = (id) => typeof id === "string" && id.trim().length > 0;
+    if (threadIds.length === 0 || threadIds.some((id) => !isValidThreadId(id))) {
       throw new MastraError(
         {
           id: "STORAGE_CLOUDFLARE_LIST_MESSAGES_INVALID_THREAD_ID",
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
-          details: { threadId }
+          details: { threadId: Array.isArray(threadId) ? JSON.stringify(threadId) : String(threadId) }
         },
-        new Error("threadId must be a non-empty string")
+        new Error("threadId must be a non-empty string or array of non-empty strings")
       );
     }
     const perPage = normalizePerPage(perPageInput, 40);
@@ -573,16 +616,18 @@ var MemoryStorageCloudflare = class extends MemoryStorage {
       }
       const { field, direction } = this.parseOrderBy(orderBy, "ASC");
       const threadMessageIds = /* @__PURE__ */ new Set();
-      try {
-        const threadMessagesKey = this.getThreadMessagesKey(threadId);
-        const allIds = await this.getFullOrder(threadMessagesKey);
-        allIds.forEach((id) => threadMessageIds.add(id));
-      } catch {
+      for (const tid of threadIds) {
+        try {
+          const threadMessagesKey = this.getThreadMessagesKey(tid);
+          const allIds = await this.getFullOrder(threadMessagesKey);
+          allIds.forEach((id) => threadMessageIds.add(id));
+        } catch {
+        }
       }
       const threadMessages = await this.fetchAndParseMessagesFromMultipleThreads(
         Array.from(threadMessageIds),
         void 0,
-        threadId
+        threadIds.length === 1 ? threadIds[0] : void 0
       );
       let filteredThreadMessages = threadMessages;
       if (resourceId) {
@@ -627,7 +672,7 @@ var MemoryStorageCloudflare = class extends MemoryStorage {
       let includedMessages = [];
       if (include && include.length > 0) {
         const includedMessageIds = /* @__PURE__ */ new Set();
-        await this.getIncludedMessagesWithContext(threadId, include, includedMessageIds);
+        await this.getIncludedMessagesWithContext(include, includedMessageIds);
         const paginatedIds = new Set(paginatedMessages.map((m) => m.id));
         const idsToFetch = Array.from(includedMessageIds).filter((id) => !paginatedIds.has(id));
         if (idsToFetch.length > 0) {
@@ -673,7 +718,11 @@ var MemoryStorageCloudflare = class extends MemoryStorage {
         type: message.type !== "v2" ? message.type : void 0,
         createdAt: ensureDate(message.createdAt)
       }));
-      const list = new MessageList({ threadId, resourceId }).add(prepared, "memory");
+      const primaryThreadId = Array.isArray(threadId) ? threadId[0] : threadId;
+      const list = new MessageList({ threadId: primaryThreadId, resourceId }).add(
+        prepared,
+        "memory"
+      );
       let finalMessages = list.get.all.db();
       finalMessages = finalMessages.sort((a, b) => {
         const isDateField = field === "createdAt" || field === "updatedAt";
@@ -689,7 +738,10 @@ var MemoryStorageCloudflare = class extends MemoryStorage {
         const cmp = direction === "ASC" ? String(aVal).localeCompare(String(bVal)) : String(bVal).localeCompare(String(aVal));
         return cmp !== 0 ? cmp : a.id.localeCompare(b.id);
       });
-      const returnedThreadMessageIds = new Set(finalMessages.filter((m) => m.threadId === threadId).map((m) => m.id));
+      const threadIdSet = new Set(threadIds);
+      const returnedThreadMessageIds = new Set(
+        finalMessages.filter((m) => m.threadId && threadIdSet.has(m.threadId)).map((m) => m.id)
+      );
       const allThreadMessagesReturned = returnedThreadMessageIds.size >= total;
       const hasMore = perPageInput !== false && !allThreadMessagesReturned && offset + paginatedCount < total;
       return {
@@ -705,9 +757,9 @@ var MemoryStorageCloudflare = class extends MemoryStorage {
           id: "CLOUDFLARE_STORAGE_LIST_MESSAGES_FAILED",
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
-          text: `Failed to list messages for thread ${threadId}: ${error instanceof Error ? error.message : String(error)}`,
+          text: `Failed to list messages for thread ${Array.isArray(threadId) ? threadId.join(",") : threadId}: ${error instanceof Error ? error.message : String(error)}`,
           details: {
-            threadId,
+            threadId: Array.isArray(threadId) ? threadId.join(",") : threadId,
             resourceId: resourceId ?? ""
           }
         },
@@ -1459,17 +1511,7 @@ var StoreOperationsCloudflare = class extends StoreOperations {
   }
 };
 function transformScoreRow(row) {
-  const deserialized = { ...row };
-  deserialized.input = safelyParseJSON(row.input);
-  deserialized.output = safelyParseJSON(row.output);
-  deserialized.scorer = safelyParseJSON(row.scorer);
-  deserialized.preprocessStepResult = safelyParseJSON(row.preprocessStepResult);
-  deserialized.analyzeStepResult = safelyParseJSON(row.analyzeStepResult);
-  deserialized.metadata = safelyParseJSON(row.metadata);
-  deserialized.additionalContext = safelyParseJSON(row.additionalContext);
-  deserialized.requestContext = safelyParseJSON(row.requestContext);
-  deserialized.entity = safelyParseJSON(row.entity);
-  return deserialized;
+  return transformScoreRow$1(row);
 }
 var ScoresStorageCloudflare = class extends ScoresStorage {
   operations;