@mastra/cloudflare 1.0.0-beta.6 → 1.0.0-beta.8

package/dist/index.js

@@ -1,1512 +1,1581 @@
 import { MastraError, ErrorCategory, ErrorDomain } from '@mastra/core/error';
-import { MastraStorage, TABLE_THREADS, TABLE_MESSAGES, TABLE_WORKFLOW_SNAPSHOT, TABLE_SCORERS, createStorageErrorId, StoreOperations, TABLE_TRACES, WorkflowsStorage, ensureDate, normalizePerPage, MemoryStorage, calculatePagination, serializeDate, TABLE_RESOURCES, ScoresStorage, transformScoreRow as transformScoreRow$1 } from '@mastra/core/storage';
+import { MastraStorage, TABLE_THREADS, TABLE_MESSAGES, TABLE_WORKFLOW_SNAPSHOT, TABLE_SCORERS, createStorageErrorId, WorkflowsStorage, ensureDate, normalizePerPage, MemoryStorage, TABLE_RESOURCES, calculatePagination, serializeDate, ScoresStorage, TABLE_TRACES, transformScoreRow as transformScoreRow$1 } from '@mastra/core/storage';
 import Cloudflare from 'cloudflare';
 import { MessageList } from '@mastra/core/agent';
+import { MastraBase } from '@mastra/core/base';
 import { saveScorePayloadSchema } from '@mastra/core/evals';
 
 // src/storage/index.ts
-var MemoryStorageCloudflare = class extends MemoryStorage {
-  operations;
-  constructor({ operations }) {
-    super();
-    this.operations = operations;
+function resolveCloudflareConfig(config) {
+  if ("client" in config) {
+    return {
+      client: config.client,
+      accountId: config.accountId,
+      namespacePrefix: config.namespacePrefix
+    };
   }
-  ensureMetadata(metadata) {
-    if (!metadata) return void 0;
-    return typeof metadata === "string" ? JSON.parse(metadata) : metadata;
+  if ("bindings" in config) {
+    return {
+      bindings: config.bindings,
+      namespacePrefix: config.keyPrefix
+    };
+  }
+  return {
+    client: new Cloudflare({ apiToken: config.apiToken }),
+    accountId: config.accountId,
+    namespacePrefix: config.namespacePrefix
+  };
+}
+var CloudflareKVDB = class extends MastraBase {
+  bindings;
+  client;
+  accountId;
+  namespacePrefix;
+  constructor(config) {
+    super({
+      component: "STORAGE",
+      name: "CLOUDFLARE_KV_DB"
+    });
+    this.bindings = config.bindings;
+    this.namespacePrefix = config.namespacePrefix || "";
+    this.client = config.client;
+    this.accountId = config.accountId;
+  }
+  getBinding(tableName) {
+    if (!this.bindings) {
+      throw new Error(`Cannot use Workers API binding for ${tableName}: Store initialized with REST API configuration`);
+    }
+    const binding = this.bindings[tableName];
+    if (!binding) throw new Error(`No binding found for namespace ${tableName}`);
+    return binding;
+  }
+  getKey(tableName, record) {
+    const prefix = this.namespacePrefix ? `${this.namespacePrefix}:` : "";
+    switch (tableName) {
+      case TABLE_THREADS:
+        if (!record.id) throw new Error("Thread ID is required");
+        return `${prefix}${tableName}:${record.id}`;
+      case TABLE_MESSAGES:
+        if (!record.threadId || !record.id) throw new Error("Thread ID and Message ID are required");
+        return `${prefix}${tableName}:${record.threadId}:${record.id}`;
+      case TABLE_WORKFLOW_SNAPSHOT:
+        if (!record.workflow_name || !record.run_id) {
+          throw new Error("Workflow name, and run ID are required");
+        }
+        let key = `${prefix}${tableName}:${record.workflow_name}:${record.run_id}`;
+        if (record.resourceId) {
+          key = `${key}:${record.resourceId}`;
+        }
+        return key;
+      case TABLE_TRACES:
+        if (!record.id) throw new Error("Trace ID is required");
+        return `${prefix}${tableName}:${record.id}`;
+      case TABLE_SCORERS:
+        if (!record.id) throw new Error("Score ID is required");
+        return `${prefix}${tableName}:${record.id}`;
+      default:
+        throw new Error(`Unsupported table: ${tableName}`);
+    }
+  }
+  getSchemaKey(tableName) {
+    const prefix = this.namespacePrefix ? `${this.namespacePrefix}:` : "";
+    return `${prefix}schema:${tableName}`;
   }
   /**
-   * Summarizes message content without exposing raw data (for logging).
-   * Returns type, length, and keys only to prevent PII leakage.
+   * Helper to safely parse data from KV storage
    */
-  summarizeMessageContent(content) {
-    if (!content) return { type: "undefined" };
-    if (typeof content === "string") return { type: "string", length: content.length };
-    if (Array.isArray(content)) return { type: "array", length: content.length };
-    if (typeof content === "object") return { type: "object", keys: Object.keys(content) };
-    return { type: typeof content };
-  }
-  async getThreadById({ threadId }) {
-    const thread = await this.operations.load({ tableName: TABLE_THREADS, keys: { id: threadId } });
-    if (!thread) return null;
+  safeParse(text) {
+    if (!text) return null;
     try {
-      return {
-        ...thread,
-        createdAt: ensureDate(thread.createdAt),
-        updatedAt: ensureDate(thread.updatedAt),
-        metadata: this.ensureMetadata(thread.metadata)
-      };
-    } catch (error) {
-      const mastraError = new MastraError(
-        {
-          id: createStorageErrorId("CLOUDFLARE", "GET_THREAD_BY_ID", "FAILED"),
-          domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.THIRD_PARTY,
-          details: {
-            threadId
+      const data = JSON.parse(text);
+      if (data && typeof data === "object" && "value" in data) {
+        if (typeof data.value === "string") {
+          try {
+            return JSON.parse(data.value);
+          } catch {
+            return data.value;
           }
-        },
-        error
-      );
-      this.logger?.trackException(mastraError);
-      this.logger?.error(mastraError.toString());
+        }
+        return null;
+      }
+      return data;
+    } catch (error) {
+      const message = error instanceof Error ? error.message : String(error);
+      this.logger.error("Failed to parse text:", { message, text });
       return null;
     }
   }
-  async listThreadsByResourceId(args) {
-    try {
-      const { resourceId, page = 0, perPage: perPageInput, orderBy } = args;
-      const perPage = normalizePerPage(perPageInput, 100);
-      if (page < 0) {
-        throw new MastraError(
-          {
-            id: createStorageErrorId("CLOUDFLARE", "LIST_THREADS_BY_RESOURCE_ID", "INVALID_PAGE"),
-            domain: ErrorDomain.STORAGE,
-            category: ErrorCategory.USER,
-            details: { page }
-          },
-          new Error("page must be >= 0")
-        );
-      }
-      const { offset, perPage: perPageForResponse } = calculatePagination(page, perPageInput, perPage);
-      const { field, direction } = this.parseOrderBy(orderBy);
-      const prefix = this.operations.namespacePrefix ? `${this.operations.namespacePrefix}:` : "";
-      const keyObjs = await this.operations.listKV(TABLE_THREADS, { prefix: `${prefix}${TABLE_THREADS}` });
-      const threads = [];
-      for (const { name: key } of keyObjs) {
-        const data = await this.operations.getKV(TABLE_THREADS, key);
-        if (!data) continue;
-        if (data.resourceId !== resourceId) continue;
-        threads.push(data);
-      }
-      threads.sort((a, b) => {
-        const aTime = new Date(a[field] || 0).getTime();
-        const bTime = new Date(b[field] || 0).getTime();
-        return direction === "ASC" ? aTime - bTime : bTime - aTime;
-      });
-      const end = perPageInput === false ? threads.length : offset + perPage;
-      const paginatedThreads = threads.slice(offset, end);
+  async createNamespaceById(title) {
+    if (this.bindings) {
       return {
-        page,
-        perPage: perPageForResponse,
-        total: threads.length,
-        hasMore: perPageInput === false ? false : offset + perPage < threads.length,
-        threads: paginatedThreads
+        id: title,
+        title,
+        supports_url_encoding: true
       };
-    } catch (error) {
-      throw new MastraError(
-        {
-          id: createStorageErrorId("CLOUDFLARE", "LIST_THREADS_BY_RESOURCE_ID", "FAILED"),
-          domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.THIRD_PARTY,
-          text: "Failed to get threads by resource ID with pagination"
-        },
-        error
-      );
     }
+    return await this.client.kv.namespaces.create({
+      account_id: this.accountId,
+      title
+    });
   }
-  async saveThread({ thread }) {
+  async createNamespace(namespaceName) {
     try {
-      await this.operations.insert({ tableName: TABLE_THREADS, record: thread });
-      return thread;
+      const response = await this.createNamespaceById(namespaceName);
+      return response.id;
     } catch (error) {
-      throw new MastraError(
-        {
-          id: createStorageErrorId("CLOUDFLARE", "SAVE_THREAD", "FAILED"),
-          domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.THIRD_PARTY,
-          details: {
-            threadId: thread.id
-          }
-        },
-        error
-      );
+      if (error.message && error.message.includes("already exists")) {
+        const namespaces = await this.listNamespaces();
+        const namespace = namespaces.result.find((ns) => ns.title === namespaceName);
+        if (namespace) return namespace.id;
+      }
+      this.logger.error("Error creating namespace:", error);
+      throw new Error(`Failed to create namespace ${namespaceName}: ${error.message}`);
     }
   }
-  async updateThread({
-    id,
-    title,
-    metadata
-  }) {
-    try {
-      const thread = await this.getThreadById({ threadId: id });
-      if (!thread) {
-        throw new Error(`Thread ${id} not found`);
-      }
-      const updatedThread = {
-        ...thread,
-        title,
-        metadata: this.ensureMetadata({
-          ...thread.metadata ?? {},
-          ...metadata
-        }),
-        updatedAt: /* @__PURE__ */ new Date()
+  async listNamespaces() {
+    if (this.bindings) {
+      return {
+        result: Object.keys(this.bindings).map((name) => ({
+          id: name,
+          title: name,
+          supports_url_encoding: true
+        }))
       };
-      await this.operations.insert({ tableName: TABLE_THREADS, record: updatedThread });
-      return updatedThread;
-    } catch (error) {
-      throw new MastraError(
-        {
-          id: createStorageErrorId("CLOUDFLARE", "UPDATE_THREAD", "FAILED"),
-          domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.THIRD_PARTY,
-          details: {
-            threadId: id,
-            title
-          }
-        },
-        error
-      );
     }
-  }
-  getMessageKey(threadId, messageId) {
-    try {
-      return this.operations.getKey(TABLE_MESSAGES, { threadId, id: messageId });
-    } catch (error) {
-      const message = error instanceof Error ? error.message : String(error);
-      this.logger?.error(`Error getting message key for thread ${threadId} and message ${messageId}:`, { message });
-      throw error;
+    let allNamespaces = [];
+    let currentPage = 1;
+    const perPage = 50;
+    let morePagesExist = true;
+    while (morePagesExist) {
+      const response = await this.client.kv.namespaces.list({
+        account_id: this.accountId,
+        page: currentPage,
+        per_page: perPage
+      });
+      if (response.result) {
+        allNamespaces = allNamespaces.concat(response.result);
+      }
+      morePagesExist = response.result ? response.result.length === perPage : false;
+      if (morePagesExist) {
+        currentPage++;
+      }
     }
+    return { result: allNamespaces };
   }
-  getThreadMessagesKey(threadId) {
+  async getNamespaceIdByName(namespaceName) {
     try {
-      return this.operations.getKey(TABLE_MESSAGES, { threadId, id: "messages" });
+      const response = await this.listNamespaces();
+      const namespace = response.result.find((ns) => ns.title === namespaceName);
+      return namespace ? namespace.id : null;
     } catch (error) {
-      const message = error instanceof Error ? error.message : String(error);
-      this.logger?.error(`Error getting thread messages key for thread ${threadId}:`, { message });
-      throw error;
+      this.logger.error(`Failed to get namespace ID for ${namespaceName}:`, error);
+      return null;
     }
   }
-  async deleteThread({ threadId }) {
+  async getOrCreateNamespaceId(namespaceName) {
+    let namespaceId = await this.getNamespaceIdByName(namespaceName);
+    if (!namespaceId) {
+      namespaceId = await this.createNamespace(namespaceName);
+    }
+    return namespaceId;
+  }
+  async getNamespaceId(tableName) {
+    const prefix = this.namespacePrefix ? `${this.namespacePrefix}_` : "";
     try {
-      const thread = await this.getThreadById({ threadId });
-      if (!thread) {
-        throw new Error(`Thread ${threadId} not found`);
-      }
-      const messageKeys = await this.operations.listKV(TABLE_MESSAGES);
-      const threadMessageKeys = messageKeys.filter((key) => key.name.includes(`${TABLE_MESSAGES}:${threadId}:`));
-      await Promise.all([
-        // Delete message order
-        this.operations.deleteKV(TABLE_MESSAGES, this.getThreadMessagesKey(threadId)),
-        // Delete all messages
-        ...threadMessageKeys.map((key) => this.operations.deleteKV(TABLE_MESSAGES, key.name)),
-        // Delete thread
-        this.operations.deleteKV(TABLE_THREADS, this.operations.getKey(TABLE_THREADS, { id: threadId }))
-      ]);
+      return await this.getOrCreateNamespaceId(`${prefix}${tableName}`);
     } catch (error) {
-      throw new MastraError(
-        {
-          id: createStorageErrorId("CLOUDFLARE", "DELETE_THREAD", "FAILED"),
-          domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.THIRD_PARTY,
-          details: {
-            threadId
-          }
-        },
-        error
-      );
+      this.logger.error("Error fetching namespace ID:", error);
+      throw new Error(`Failed to fetch namespace ID for table ${tableName}: ${error.message}`);
     }
   }
-  /**
-   * 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) {
+  async getNamespaceValue(tableName, key) {
     try {
-      const prefix = this.operations.namespacePrefix ? `${this.operations.namespacePrefix}:` : "";
-      const threadKeys = await this.operations.listKV(TABLE_THREADS, { prefix: `${prefix}${TABLE_THREADS}` });
-      for (const { name: threadKey } of threadKeys) {
-        const threadId = threadKey.split(":").pop();
-        if (!threadId || threadId === "messages") continue;
-        const messageKey = this.getMessageKey(threadId, messageId);
-        const message = await this.operations.getKV(TABLE_MESSAGES, messageKey);
-        if (message) {
-          return { ...message, threadId };
-        }
+      if (this.bindings) {
+        const binding = this.getBinding(tableName);
+        const result = await binding.getWithMetadata(key, "text");
+        if (!result) return null;
+        return JSON.stringify(result);
+      } else {
+        const namespaceId = await this.getNamespaceId(tableName);
+        const response = await this.client.kv.namespaces.values.get(namespaceId, key, {
+          account_id: this.accountId
+        });
+        return await response.text();
       }
-      return null;
     } catch (error) {
-      this.logger?.error(`Error finding message ${messageId} in any thread:`, error);
+      if (error.message && error.message.includes("key not found")) {
+        return null;
+      }
+      const message = error instanceof Error ? error.message : String(error);
+      this.logger.error(`Failed to get value for ${tableName} ${key}:`, { message });
+      throw error;
+    }
+  }
+  async getKV(tableName, key) {
+    try {
+      const text = await this.getNamespaceValue(tableName, key);
+      return this.safeParse(text);
+    } catch (error) {
+      this.logger.error(`Failed to get KV value for ${tableName}:${key}:`, error);
+      throw new Error(`Failed to get KV value: ${error.message}`);
+    }
+  }
+  async getTableSchema(tableName) {
+    try {
+      const schemaKey = this.getSchemaKey(tableName);
+      return await this.getKV(tableName, schemaKey);
+    } catch (error) {
+      const message = error instanceof Error ? error.message : String(error);
+      this.logger.error(`Failed to get schema for ${tableName}:`, { message });
       return null;
     }
   }
-  /**
-   * Queue for serializing sorted order updates.
-   * Updates the sorted order for a given key. This operation is eventually consistent.
-   */
-  updateQueue = /* @__PURE__ */ new Map();
-  async updateSorting(threadMessages) {
-    return threadMessages.map((msg) => ({
-      message: msg,
-      // Use _index if available, otherwise timestamp, matching Upstash
-      score: msg._index !== void 0 ? msg._index : msg.createdAt.getTime()
-    })).sort((a, b) => a.score - b.score).map((item) => ({
-      id: item.message.id,
-      score: item.score
-    }));
+  validateColumnValue(value, column) {
+    if (value === void 0 || value === null) {
+      return column.nullable ?? false;
+    }
+    switch (column.type) {
+      case "text":
+      case "uuid":
+        return typeof value === "string";
+      case "integer":
+      case "bigint":
+        return typeof value === "number";
+      case "timestamp":
+        return value instanceof Date || typeof value === "string" && !isNaN(Date.parse(value));
+      case "jsonb":
+        if (typeof value !== "object") return false;
+        try {
+          JSON.stringify(value);
+          return true;
+        } catch {
+          return false;
+        }
+      default:
+        return false;
+    }
   }
-  /**
-   * Updates the sorted order for a given key. This operation is eventually consistent.
-   * Note: Operations on the same orderKey are serialized using a queue to prevent
-   * concurrent updates from conflicting with each other.
-   */
-  async updateSortedMessages(orderKey, newEntries) {
-    const currentPromise = this.updateQueue.get(orderKey) || Promise.resolve();
-    const nextPromise = currentPromise.then(async () => {
-      try {
-        const currentOrder = await this.getSortedMessages(orderKey);
-        const orderMap = new Map(currentOrder.map((entry) => [entry.id, entry]));
-        for (const entry of newEntries) {
-          orderMap.set(entry.id, entry);
+  async validateAgainstSchema(record, schema) {
+    try {
+      if (!schema || typeof schema !== "object" || schema.value === null) {
+        throw new Error("Invalid schema format");
+      }
+      for (const [columnName, column] of Object.entries(schema)) {
+        const value = record[columnName];
+        if (column.primaryKey && (value === void 0 || value === null)) {
+          throw new Error(`Missing primary key value for column ${columnName}`);
         }
-        const updatedOrder = Array.from(orderMap.values()).sort((a, b) => a.score - b.score);
-        await this.operations.putKV({
-          tableName: TABLE_MESSAGES,
-          key: orderKey,
-          value: JSON.stringify(updatedOrder)
-        });
-      } catch (error) {
-        const message = error instanceof Error ? error.message : String(error);
-        this.logger?.error(`Error updating sorted order for key ${orderKey}:`, { message });
-        throw error;
-      } finally {
-        if (this.updateQueue.get(orderKey) === nextPromise) {
-          this.updateQueue.delete(orderKey);
+        if (!this.validateColumnValue(value, column)) {
+          const valueType = value === null ? "null" : typeof value;
+          throw new Error(`Invalid value for column ${columnName}: expected ${column.type}, got ${valueType}`);
         }
       }
-    });
-    this.updateQueue.set(orderKey, nextPromise);
-    return nextPromise;
-  }
-  async getSortedMessages(orderKey) {
-    const raw = await this.operations.getKV(TABLE_MESSAGES, orderKey);
-    if (!raw) return [];
-    try {
-      const arr = JSON.parse(typeof raw === "string" ? raw : JSON.stringify(raw));
-      return Array.isArray(arr) ? arr : [];
-    } catch (e) {
-      this.logger?.error(`Error parsing order data for key ${orderKey}:`, { e });
-      return [];
+    } catch (error) {
+      const message = error instanceof Error ? error.message : String(error);
+      this.logger.error(`Error validating record against schema:`, { message, record, schema });
+      throw error;
     }
   }
-  async migrateMessage(messageId, fromThreadId, toThreadId) {
+  async validateRecord(record, tableName) {
     try {
-      const oldMessageKey = this.getMessageKey(fromThreadId, messageId);
-      const message = await this.operations.getKV(TABLE_MESSAGES, oldMessageKey);
-      if (!message) return;
-      const updatedMessage = {
-        ...message,
-        threadId: toThreadId
-      };
-      const newMessageKey = this.getMessageKey(toThreadId, messageId);
-      await this.operations.putKV({ tableName: TABLE_MESSAGES, key: newMessageKey, value: updatedMessage });
-      const oldOrderKey = this.getThreadMessagesKey(fromThreadId);
-      const oldEntries = await this.getSortedMessages(oldOrderKey);
-      const filteredEntries = oldEntries.filter((entry) => entry.id !== messageId);
-      await this.updateSortedMessages(oldOrderKey, filteredEntries);
-      const newOrderKey = this.getThreadMessagesKey(toThreadId);
-      const newEntries = await this.getSortedMessages(newOrderKey);
-      const newEntry = { id: messageId, score: Date.now() };
-      newEntries.push(newEntry);
-      await this.updateSortedMessages(newOrderKey, newEntries);
-      await this.operations.deleteKV(TABLE_MESSAGES, oldMessageKey);
+      if (!record || typeof record !== "object") {
+        throw new Error("Record must be an object");
+      }
+      const recordTyped = record;
+      const schema = await this.getTableSchema(tableName);
+      if (schema) {
+        await this.validateAgainstSchema(recordTyped, schema);
+        return;
+      }
+      switch (tableName) {
+        case TABLE_THREADS:
+          if (!("id" in recordTyped) || !("resourceId" in recordTyped) || !("title" in recordTyped)) {
+            throw new Error("Thread record missing required fields");
+          }
+          break;
+        case TABLE_MESSAGES:
+          if (!("id" in recordTyped) || !("threadId" in recordTyped) || !("content" in recordTyped) || !("role" in recordTyped)) {
+            throw new Error("Message record missing required fields");
+          }
+          break;
+        case TABLE_WORKFLOW_SNAPSHOT:
+          if (!("workflow_name" in recordTyped) || !("run_id" in recordTyped)) {
+            throw new Error("Workflow record missing required fields");
+          }
+          break;
+        case TABLE_TRACES:
+          if (!("id" in recordTyped)) {
+            throw new Error("Trace record missing required fields");
+          }
+          break;
+        case TABLE_SCORERS:
+          if (!("id" in recordTyped) || !("scorerId" in recordTyped)) {
+            throw new Error("Score record missing required fields");
+          }
+          break;
+        default:
+          throw new Error(`Unknown table type: ${tableName}`);
+      }
     } catch (error) {
-      this.logger?.error(`Error migrating message ${messageId} from ${fromThreadId} to ${toThreadId}:`, error);
+      const message = error instanceof Error ? error.message : String(error);
+      this.logger.error(`Failed to validate record for ${tableName}:`, { message, record });
       throw error;
     }
   }
-  async saveMessages(args) {
-    const { messages } = args;
-    if (!Array.isArray(messages) || messages.length === 0) return { messages: [] };
+  async insert({ tableName, record }) {
     try {
-      const validatedMessages = messages.map((message, index) => {
-        const errors = [];
-        if (!message.id) errors.push("id is required");
-        if (!message.threadId) errors.push("threadId is required");
-        if (!message.content) errors.push("content is required");
-        if (!message.role) errors.push("role is required");
-        if (!message.createdAt) errors.push("createdAt is required");
-        if (message.resourceId === null || message.resourceId === void 0) errors.push("resourceId is required");
-        if (errors.length > 0) {
-          throw new Error(`Invalid message at index ${index}: ${errors.join(", ")}`);
-        }
-        return {
-          ...message,
-          createdAt: ensureDate(message.createdAt),
-          type: message.type || "v2",
-          _index: index
-        };
-      }).filter((m) => !!m);
-      const messageMigrationTasks = [];
-      for (const message of validatedMessages) {
-        const existingMessage = await this.findMessageInAnyThread(message.id);
-        this.logger?.debug(
-          `Checking message ${message.id}: existing=${existingMessage?.threadId}, new=${message.threadId}`
-        );
-        if (existingMessage && existingMessage.threadId && existingMessage.threadId !== message.threadId) {
-          this.logger?.debug(`Migrating message ${message.id} from ${existingMessage.threadId} to ${message.threadId}`);
-          messageMigrationTasks.push(this.migrateMessage(message.id, existingMessage.threadId, message.threadId));
-        }
-      }
-      await Promise.all(messageMigrationTasks);
-      const messagesByThread = validatedMessages.reduce((acc, message) => {
-        if (message.threadId && !acc.has(message.threadId)) {
-          acc.set(message.threadId, []);
-        }
-        if (message.threadId) {
-          acc.get(message.threadId).push(message);
-        }
-        return acc;
-      }, /* @__PURE__ */ new Map());
-      await Promise.all(
-        Array.from(messagesByThread.entries()).map(async ([threadId, threadMessages]) => {
-          try {
-            const thread = await this.getThreadById({ threadId });
-            if (!thread) {
-              throw new Error(`Thread ${threadId} not found`);
-            }
-            await Promise.all(
-              threadMessages.map(async (message) => {
-                const key = this.getMessageKey(threadId, message.id);
-                const { _index, ...cleanMessage } = message;
-                const serializedMessage = {
-                  ...cleanMessage,
-                  createdAt: serializeDate(cleanMessage.createdAt)
-                };
-                this.logger?.debug(`Saving message ${message.id}`, {
-                  contentSummary: this.summarizeMessageContent(serializedMessage.content)
-                });
-                await this.operations.putKV({ tableName: TABLE_MESSAGES, key, value: serializedMessage });
-              })
-            );
-            const orderKey = this.getThreadMessagesKey(threadId);
-            const entries = await this.updateSorting(threadMessages);
-            await this.updateSortedMessages(orderKey, entries);
-            const updatedThread = {
-              ...thread,
-              updatedAt: /* @__PURE__ */ new Date()
-            };
-            await this.operations.putKV({
-              tableName: TABLE_THREADS,
-              key: this.operations.getKey(TABLE_THREADS, { id: threadId }),
-              value: updatedThread
-            });
-          } catch (error) {
-            throw new MastraError(
-              {
-                id: createStorageErrorId("CLOUDFLARE", "SAVE_MESSAGES", "FAILED"),
-                domain: ErrorDomain.STORAGE,
-                category: ErrorCategory.THIRD_PARTY,
-                details: {
-                  threadId
-                }
-              },
-              error
-            );
-          }
-        })
-      );
-      const prepared = validatedMessages.map(
-        ({ _index, ...message }) => ({ ...message, type: message.type !== "v2" ? message.type : void 0 })
-      );
-      const list = new MessageList().add(prepared, "memory");
-      return { messages: list.get.all.db() };
+      const key = this.getKey(tableName, record);
+      const processedRecord = { ...record };
+      await this.validateRecord(processedRecord, tableName);
+      await this.putKV({ tableName, key, value: processedRecord });
     } catch (error) {
       throw new MastraError(
         {
-          id: createStorageErrorId("CLOUDFLARE", "SAVE_MESSAGES", "FAILED"),
+          id: createStorageErrorId("CLOUDFLARE", "INSERT", "FAILED"),
           domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.THIRD_PARTY
+          category: ErrorCategory.THIRD_PARTY,
+          details: {
+            tableName
+          }
         },
         error
       );
     }
   }
-  async getRank(orderKey, id) {
-    const order = await this.getSortedMessages(orderKey);
-    const index = order.findIndex((item) => item.id === id);
-    return index >= 0 ? index : null;
-  }
-  async getRange(orderKey, start, end) {
-    const order = await this.getSortedMessages(orderKey);
-    const actualStart = start < 0 ? Math.max(0, order.length + start) : start;
-    const actualEnd = end < 0 ? order.length + end : Math.min(end, order.length - 1);
-    const sliced = order.slice(actualStart, actualEnd + 1);
-    return sliced.map((item) => item.id);
-  }
-  async getLastN(orderKey, n) {
-    return this.getRange(orderKey, -n, -1);
-  }
-  async getFullOrder(orderKey) {
-    return this.getRange(orderKey, 0, -1);
-  }
-  /**
-   * 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) => {
-        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);
-        if (!item.withPreviousMessages && !item.withNextMessages) return;
-        const rank = await this.getRank(threadMessagesKey, item.id);
-        if (rank === null) return;
-        if (item.withPreviousMessages) {
-          const prevIds = await this.getRange(
-            threadMessagesKey,
-            Math.max(0, rank - item.withPreviousMessages),
-            rank - 1
-          );
-          prevIds.forEach((id) => messageIds.add(id));
-        }
-        if (item.withNextMessages) {
-          const nextIds = await this.getRange(threadMessagesKey, rank + 1, rank + item.withNextMessages);
-          nextIds.forEach((id) => messageIds.add(id));
-        }
-      })
-    );
-  }
-  async getRecentMessages(threadId, limit, messageIds) {
-    if (!threadId.trim()) throw new Error("threadId must be a non-empty string");
-    if (limit <= 0) return;
-    try {
-      const threadMessagesKey = this.getThreadMessagesKey(threadId);
-      const latestIds = await this.getLastN(threadMessagesKey, limit);
-      latestIds.forEach((id) => messageIds.add(id));
-    } catch {
-      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) {
-      for (const item of include) {
-        if (item.threadId) {
-          messageIdToThreadId.set(item.id, item.threadId);
-        }
-      }
-    }
-    const messages = await Promise.all(
-      messageIds.map(async (id) => {
-        try {
-          let threadId = messageIdToThreadId.get(id);
-          if (!threadId) {
-            if (targetThreadId) {
-              threadId = targetThreadId;
-            } else {
-              const foundMessage = await this.findMessageInAnyThread(id);
-              if (foundMessage) {
-                threadId = foundMessage.threadId;
-              }
-            }
-          }
-          if (!threadId) return null;
-          const key = this.getMessageKey(threadId, id);
-          const data = await this.operations.getKV(TABLE_MESSAGES, key);
-          if (!data) return null;
-          const parsed = typeof data === "string" ? JSON.parse(data) : data;
-          this.logger?.debug(`Retrieved message ${id} from thread ${threadId}`, {
-            contentSummary: this.summarizeMessageContent(parsed.content)
-          });
-          return parsed;
-        } catch (error) {
-          const message = error instanceof Error ? error.message : String(error);
-          this.logger?.error(`Error retrieving message ${id}:`, { message });
-          return null;
-        }
-      })
-    );
-    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: [] };
+  async load({ tableName, keys }) {
     try {
-      const messages = (await Promise.all(messageIds.map((id) => this.findMessageInAnyThread(id)))).filter(
-        (result) => !!result
-      );
-      const prepared = messages.map(({ _index, ...message }) => ({
-        ...message,
-        ...message.type !== `v2` && { type: message.type },
-        createdAt: ensureDate(message.createdAt)
-      }));
-      const list = new MessageList().add(prepared, "memory");
-      return { messages: list.get.all.db() };
+      const key = this.getKey(tableName, keys);
+      const data = await this.getKV(tableName, key);
+      if (!data) return null;
+      return data;
     } catch (error) {
       const mastraError = new MastraError(
         {
-          id: createStorageErrorId("CLOUDFLARE", "LIST_MESSAGES_BY_ID", "FAILED"),
+          id: createStorageErrorId("CLOUDFLARE", "LOAD", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
-          text: `Error retrieving messages by ID`,
           details: {
-            messageIds: JSON.stringify(messageIds)
+            tableName
           }
         },
         error
       );
       this.logger?.trackException(mastraError);
       this.logger?.error(mastraError.toString());
-      return { messages: [] };
+      return null;
     }
   }
-  async listMessages(args) {
-    const { threadId, resourceId, include, filter, perPage: perPageInput, page = 0, orderBy } = args;
-    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))) {
+  async batchInsert(input) {
+    if (!input.records || input.records.length === 0) return;
+    try {
+      await Promise.all(
+        input.records.map(async (record) => {
+          const key = this.getKey(input.tableName, record);
+          await this.putKV({ tableName: input.tableName, key, value: record });
+        })
+      );
+    } catch (error) {
       throw new MastraError(
         {
-          id: createStorageErrorId("CLOUDFLARE", "LIST_MESSAGES", "INVALID_THREAD_ID"),
+          id: createStorageErrorId("CLOUDFLARE", "BATCH_INSERT", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
-          details: { threadId: Array.isArray(threadId) ? JSON.stringify(threadId) : String(threadId) }
-        },
-        new Error("threadId must be a non-empty string or array of non-empty strings")
+          text: `Error in batch insert for table ${input.tableName}`,
+          details: {
+            tableName: input.tableName
+          }
+        },
+        error
       );
     }
-    const perPage = normalizePerPage(perPageInput, 40);
-    const { offset, perPage: perPageForResponse } = calculatePagination(page, perPageInput, perPage);
+  }
+  safeSerialize(data) {
+    return typeof data === "string" ? data : JSON.stringify(data);
+  }
+  async putNamespaceValue({
+    tableName,
+    key,
+    value,
+    metadata
+  }) {
     try {
-      if (page < 0) {
-        throw new MastraError(
-          {
-            id: createStorageErrorId("CLOUDFLARE", "LIST_MESSAGES", "INVALID_PAGE"),
-            domain: ErrorDomain.STORAGE,
-            category: ErrorCategory.USER,
-            details: { page }
-          },
-          new Error("page must be >= 0")
-        );
-      }
-      const { field, direction } = this.parseOrderBy(orderBy, "ASC");
-      const threadMessageIds = /* @__PURE__ */ new Set();
-      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,
-        threadIds.length === 1 ? threadIds[0] : void 0
-      );
-      let filteredThreadMessages = threadMessages;
-      if (resourceId) {
-        filteredThreadMessages = filteredThreadMessages.filter((msg) => msg.resourceId === resourceId);
-      }
-      const dateRange = filter?.dateRange;
-      if (dateRange) {
-        filteredThreadMessages = filteredThreadMessages.filter((msg) => {
-          const messageDate = new Date(msg.createdAt);
-          if (dateRange.start && messageDate < new Date(dateRange.start)) return false;
-          if (dateRange.end && messageDate > new Date(dateRange.end)) return false;
-          return true;
-        });
-      }
-      const total = filteredThreadMessages.length;
-      if (perPage === 0 && (!include || include.length === 0)) {
-        return {
-          messages: [],
-          total,
-          page,
-          perPage: perPageForResponse,
-          hasMore: offset < total
-        };
-      }
-      filteredThreadMessages.sort((a, b) => {
-        const timeA = new Date(a.createdAt).getTime();
-        const timeB = new Date(b.createdAt).getTime();
-        const timeDiff = direction === "ASC" ? timeA - timeB : timeB - timeA;
-        if (timeDiff === 0) {
-          return a.id.localeCompare(b.id);
-        }
-        return timeDiff;
-      });
-      let paginatedMessages;
-      if (perPage === 0) {
-        paginatedMessages = [];
-      } else if (perPage === Number.MAX_SAFE_INTEGER) {
-        paginatedMessages = filteredThreadMessages;
+      const serializedValue = this.safeSerialize(value);
+      const serializedMetadata = metadata ? this.safeSerialize(metadata) : "";
+      if (this.bindings) {
+        const binding = this.getBinding(tableName);
+        await binding.put(key, serializedValue, { metadata: serializedMetadata });
       } else {
-        paginatedMessages = filteredThreadMessages.slice(offset, offset + perPage);
-      }
-      let includedMessages = [];
-      if (include && include.length > 0) {
-        const includedMessageIds = /* @__PURE__ */ new Set();
-        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) {
-          includedMessages = await this.fetchAndParseMessagesFromMultipleThreads(idsToFetch, include, void 0);
-        }
-      }
-      const seenIds = /* @__PURE__ */ new Set();
-      const allMessages = [];
-      for (const msg of paginatedMessages) {
-        if (!seenIds.has(msg.id)) {
-          allMessages.push(msg);
-          seenIds.add(msg.id);
-        }
-      }
-      for (const msg of includedMessages) {
-        if (!seenIds.has(msg.id)) {
-          allMessages.push(msg);
-          seenIds.add(msg.id);
-        }
-      }
-      allMessages.sort((a, b) => {
-        const timeA = new Date(a.createdAt).getTime();
-        const timeB = new Date(b.createdAt).getTime();
-        const timeDiff = direction === "ASC" ? timeA - timeB : timeB - timeA;
-        if (timeDiff === 0) {
-          return a.id.localeCompare(b.id);
-        }
-        return timeDiff;
-      });
-      let filteredMessages = allMessages;
-      const paginatedCount = paginatedMessages.length;
-      if (total === 0 && filteredMessages.length === 0 && (!include || include.length === 0)) {
-        return {
-          messages: [],
-          total: 0,
-          page,
-          perPage: perPageForResponse,
-          hasMore: false
-        };
+        const namespaceId = await this.getNamespaceId(tableName);
+        await this.client.kv.namespaces.values.update(namespaceId, key, {
+          account_id: this.accountId,
+          value: serializedValue,
+          metadata: serializedMetadata
+        });
       }
-      const prepared = filteredMessages.map(({ _index, ...message }) => ({
-        ...message,
-        type: message.type !== "v2" ? message.type : void 0,
-        createdAt: ensureDate(message.createdAt)
-      }));
-      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";
-        const aVal = isDateField ? new Date(a[field]).getTime() : a[field];
-        const bVal = isDateField ? new Date(b[field]).getTime() : b[field];
-        if (aVal == null && bVal == null) return a.id.localeCompare(b.id);
-        if (aVal == null) return 1;
-        if (bVal == null) return -1;
-        if (typeof aVal === "number" && typeof bVal === "number") {
-          const cmp2 = direction === "ASC" ? aVal - bVal : bVal - aVal;
-          return cmp2 !== 0 ? cmp2 : a.id.localeCompare(b.id);
-        }
-        const cmp = direction === "ASC" ? String(aVal).localeCompare(String(bVal)) : String(bVal).localeCompare(String(aVal));
-        return cmp !== 0 ? cmp : a.id.localeCompare(b.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 {
-        messages: finalMessages,
-        total,
-        page,
-        perPage: perPageForResponse,
-        hasMore
+    } catch (error) {
+      const message = error instanceof Error ? error.message : String(error);
+      this.logger.error(`Failed to put value for ${tableName} ${key}:`, { message });
+      throw error;
+    }
+  }
+  async putKV({
+    tableName,
+    key,
+    value,
+    metadata
+  }) {
+    try {
+      await this.putNamespaceValue({ tableName, key, value, metadata });
+    } catch (error) {
+      this.logger.error(`Failed to put KV value for ${tableName}:${key}:`, error);
+      throw new Error(`Failed to put KV value: ${error.message}`);
+    }
+  }
+  async createTable({
+    tableName,
+    schema
+  }) {
+    try {
+      const schemaKey = this.getSchemaKey(tableName);
+      const metadata = {
+        type: "table_schema",
+        tableName,
+        createdAt: (/* @__PURE__ */ new Date()).toISOString()
       };
+      await this.putKV({ tableName, key: schemaKey, value: schema, metadata });
     } catch (error) {
-      const mastraError = new MastraError(
+      throw new MastraError(
         {
-          id: createStorageErrorId("CLOUDFLARE", "LIST_MESSAGES", "FAILED"),
+          id: createStorageErrorId("CLOUDFLARE", "CREATE_TABLE", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
-          text: `Failed to list messages for thread ${Array.isArray(threadId) ? threadId.join(",") : threadId}: ${error instanceof Error ? error.message : String(error)}`,
           details: {
-            threadId: Array.isArray(threadId) ? threadId.join(",") : threadId,
-            resourceId: resourceId ?? ""
+            tableName
           }
         },
         error
       );
-      this.logger?.error?.(mastraError.toString());
-      this.logger?.trackException?.(mastraError);
-      return {
-        messages: [],
-        total: 0,
-        page,
-        perPage: perPageForResponse,
-        hasMore: false
-      };
     }
   }
-  async updateMessages(args) {
+  async clearTable({ tableName }) {
     try {
-      const { messages } = args;
-      const updatedMessages = [];
-      for (const messageUpdate of messages) {
-        const { id, content, ...otherFields } = messageUpdate;
-        const prefix = this.operations.namespacePrefix ? `${this.operations.namespacePrefix}:` : "";
-        const keyObjs = await this.operations.listKV(TABLE_MESSAGES, { prefix: `${prefix}${TABLE_MESSAGES}` });
-        let existingMessage = null;
-        let messageKey = "";
-        for (const { name: key } of keyObjs) {
-          const data = await this.operations.getKV(TABLE_MESSAGES, key);
-          if (data && data.id === id) {
-            existingMessage = data;
-            messageKey = key;
-            break;
+      const keys = await this.listKV(tableName);
+      if (keys.length > 0) {
+        await Promise.all(keys.map((keyObj) => this.deleteKV(tableName, keyObj.name)));
+      }
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: createStorageErrorId("CLOUDFLARE", "CLEAR_TABLE", "FAILED"),
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: {
+            tableName
           }
-        }
-        if (!existingMessage) {
-          continue;
-        }
-        const updatedMessage = {
-          ...existingMessage,
-          ...otherFields,
-          id
-        };
-        if (content) {
-          if (content.metadata !== void 0) {
-            updatedMessage.content = {
-              ...updatedMessage.content,
-              metadata: {
-                ...updatedMessage.content?.metadata,
-                ...content.metadata
-              }
-            };
-          }
-          if (content.content !== void 0) {
-            updatedMessage.content = {
-              ...updatedMessage.content,
-              content: content.content
-            };
-          }
-        }
-        if ("threadId" in messageUpdate && messageUpdate.threadId && messageUpdate.threadId !== existingMessage.threadId) {
-          await this.operations.deleteKV(TABLE_MESSAGES, messageKey);
-          updatedMessage.threadId = messageUpdate.threadId;
-          const newMessageKey = this.getMessageKey(messageUpdate.threadId, id);
-          await this.operations.putKV({
-            tableName: TABLE_MESSAGES,
-            key: newMessageKey,
-            value: updatedMessage
-          });
-          if (existingMessage.threadId) {
-            const sourceOrderKey = this.getThreadMessagesKey(existingMessage.threadId);
-            const sourceEntries = await this.getSortedMessages(sourceOrderKey);
-            const filteredEntries = sourceEntries.filter((entry) => entry.id !== id);
-            await this.updateSortedMessages(sourceOrderKey, filteredEntries);
-          }
-          const destOrderKey = this.getThreadMessagesKey(messageUpdate.threadId);
-          const destEntries = await this.getSortedMessages(destOrderKey);
-          const newEntry = { id, score: Date.now() };
-          destEntries.push(newEntry);
-          await this.updateSortedMessages(destOrderKey, destEntries);
-        } else {
-          await this.operations.putKV({
-            tableName: TABLE_MESSAGES,
-            key: messageKey,
-            value: updatedMessage
-          });
-        }
-        const threadsToUpdate = /* @__PURE__ */ new Set();
-        if (updatedMessage.threadId) {
-          threadsToUpdate.add(updatedMessage.threadId);
-        }
-        if ("threadId" in messageUpdate && messageUpdate.threadId && messageUpdate.threadId !== existingMessage.threadId) {
-          if (existingMessage.threadId) {
-            threadsToUpdate.add(existingMessage.threadId);
-          }
-          threadsToUpdate.add(messageUpdate.threadId);
-        }
-        for (const threadId of threadsToUpdate) {
-          const thread = await this.getThreadById({ threadId });
-          if (thread) {
-            const updatedThread = {
-              ...thread,
-              updatedAt: /* @__PURE__ */ new Date()
-            };
-            await this.operations.putKV({
-              tableName: TABLE_THREADS,
-              key: this.operations.getKey(TABLE_THREADS, { id: threadId }),
-              value: updatedThread
-            });
-          }
-        }
-        updatedMessages.push(updatedMessage);
-      }
-      return updatedMessages;
-    } catch (error) {
-      throw new MastraError(
-        {
-          id: createStorageErrorId("CLOUDFLARE", "UPDATE_MESSAGES", "FAILED"),
-          domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.THIRD_PARTY,
-          text: "Failed to update messages"
         },
         error
       );
     }
   }
-  async getResourceById({ resourceId }) {
+  async dropTable({ tableName }) {
     try {
-      const data = await this.operations.getKV(TABLE_RESOURCES, resourceId);
-      if (!data) return null;
-      const resource = typeof data === "string" ? JSON.parse(data) : data;
-      return {
-        ...resource,
-        createdAt: ensureDate(resource.createdAt),
-        updatedAt: ensureDate(resource.updatedAt),
-        metadata: this.ensureMetadata(resource.metadata)
-      };
+      const keys = await this.listKV(tableName);
+      if (keys.length > 0) {
+        await Promise.all(keys.map((keyObj) => this.deleteKV(tableName, keyObj.name)));
+      }
     } catch (error) {
-      const mastraError = new MastraError(
+      throw new MastraError(
         {
-          id: createStorageErrorId("CLOUDFLARE", "GET_RESOURCE_BY_ID", "FAILED"),
+          id: createStorageErrorId("CLOUDFLARE", "DROP_TABLE", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: {
-            resourceId
+            tableName
           }
         },
         error
       );
-      this.logger?.trackException(mastraError);
-      this.logger?.error(mastraError.toString());
-      return null;
     }
   }
-  async saveResource({ resource }) {
+  async listNamespaceKeys(tableName, options) {
     try {
-      const resourceToSave = {
-        ...resource,
-        metadata: resource.metadata ? JSON.stringify(resource.metadata) : null
-      };
-      await this.operations.putKV({
-        tableName: TABLE_RESOURCES,
-        key: resource.id,
-        value: resourceToSave
-      });
-      return resource;
+      if (this.bindings) {
+        const binding = this.getBinding(tableName);
+        const response = await binding.list({
+          limit: options?.limit || 1e3,
+          prefix: options?.prefix
+        });
+        return response.keys;
+      } else {
+        const namespaceId = await this.getNamespaceId(tableName);
+        const response = await this.client.kv.namespaces.keys.list(namespaceId, {
+          account_id: this.accountId,
+          limit: options?.limit || 1e3,
+          prefix: options?.prefix
+        });
+        return response.result;
+      }
     } catch (error) {
       throw new MastraError(
         {
-          id: createStorageErrorId("CLOUDFLARE", "SAVE_RESOURCE", "FAILED"),
+          id: createStorageErrorId("CLOUDFLARE", "LIST_NAMESPACE_KEYS", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: {
-            resourceId: resource.id
+            tableName
           }
         },
         error
       );
     }
   }
-  async updateResource({
-    resourceId,
-    workingMemory,
-    metadata
-  }) {
-    const existingResource = await this.getResourceById({ resourceId });
-    if (!existingResource) {
-      const newResource = {
-        id: resourceId,
-        workingMemory,
-        metadata: metadata || {},
-        createdAt: /* @__PURE__ */ new Date(),
-        updatedAt: /* @__PURE__ */ new Date()
-      };
-      return this.saveResource({ resource: newResource });
+  async deleteNamespaceValue(tableName, key) {
+    if (this.bindings) {
+      const binding = this.getBinding(tableName);
+      await binding.delete(key);
+    } else {
+      const namespaceId = await this.getNamespaceId(tableName);
+      await this.client.kv.namespaces.values.delete(namespaceId, key, {
+        account_id: this.accountId
+      });
+    }
+  }
+  async deleteKV(tableName, key) {
+    try {
+      await this.deleteNamespaceValue(tableName, key);
+    } catch (error) {
+      this.logger.error(`Failed to delete KV value for ${tableName}:${key}:`, error);
+      throw new Error(`Failed to delete KV value: ${error.message}`);
+    }
+  }
+  async listKV(tableName, options) {
+    try {
+      return await this.listNamespaceKeys(tableName, options);
+    } catch (error) {
+      this.logger.error(`Failed to list KV for ${tableName}:`, error);
+      throw new Error(`Failed to list KV: ${error.message}`);
     }
-    const updatedAt = /* @__PURE__ */ new Date();
-    const updatedResource = {
-      ...existingResource,
-      workingMemory: workingMemory !== void 0 ? workingMemory : existingResource.workingMemory,
-      metadata: {
-        ...existingResource.metadata,
-        ...metadata
-      },
-      updatedAt
-    };
-    return this.saveResource({ resource: updatedResource });
   }
 };
-var StoreOperationsCloudflare = class extends StoreOperations {
-  bindings;
-  client;
-  accountId;
-  namespacePrefix;
-  constructor({
-    namespacePrefix,
-    bindings,
-    client,
-    accountId
-  }) {
+
+// src/storage/domains/memory/index.ts
+var MemoryStorageCloudflare = class extends MemoryStorage {
+  #db;
+  constructor(config) {
     super();
-    this.bindings = bindings;
-    this.namespacePrefix = namespacePrefix;
-    this.client = client;
-    this.accountId = accountId;
+    this.#db = new CloudflareKVDB(resolveCloudflareConfig(config));
   }
-  async hasColumn() {
-    return true;
+  async init() {
   }
-  async alterTable(_args) {
+  async dangerouslyClearAll() {
+    await this.#db.clearTable({ tableName: TABLE_MESSAGES });
+    await this.#db.clearTable({ tableName: TABLE_THREADS });
+    await this.#db.clearTable({ tableName: TABLE_RESOURCES });
   }
-  async clearTable({ tableName }) {
+  ensureMetadata(metadata) {
+    if (!metadata) return void 0;
+    return typeof metadata === "string" ? JSON.parse(metadata) : metadata;
+  }
+  /**
+   * Summarizes message content without exposing raw data (for logging).
+   * Returns type, length, and keys only to prevent PII leakage.
+   */
+  summarizeMessageContent(content) {
+    if (!content) return { type: "undefined" };
+    if (typeof content === "string") return { type: "string", length: content.length };
+    if (Array.isArray(content)) return { type: "array", length: content.length };
+    if (typeof content === "object") return { type: "object", keys: Object.keys(content) };
+    return { type: typeof content };
+  }
+  async getThreadById({ threadId }) {
+    const thread = await this.#db.load({ tableName: TABLE_THREADS, keys: { id: threadId } });
+    if (!thread) return null;
     try {
-      const keys = await this.listKV(tableName);
-      if (keys.length > 0) {
-        await Promise.all(keys.map((keyObj) => this.deleteKV(tableName, keyObj.name)));
-      }
+      return {
+        ...thread,
+        createdAt: ensureDate(thread.createdAt),
+        updatedAt: ensureDate(thread.updatedAt),
+        metadata: this.ensureMetadata(thread.metadata)
+      };
     } catch (error) {
-      throw new MastraError(
+      const mastraError = new MastraError(
         {
-          id: createStorageErrorId("CLOUDFLARE", "CLEAR_TABLE", "FAILED"),
+          id: createStorageErrorId("CLOUDFLARE", "GET_THREAD_BY_ID", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: {
-            tableName
+            threadId
           }
         },
         error
       );
+      this.logger?.trackException(mastraError);
+      this.logger?.error(mastraError.toString());
+      return null;
     }
   }
-  async dropTable({ tableName }) {
+  async listThreadsByResourceId(args) {
     try {
-      const keys = await this.listKV(tableName);
-      if (keys.length > 0) {
-        await Promise.all(keys.map((keyObj) => this.deleteKV(tableName, keyObj.name)));
+      const { resourceId, page = 0, perPage: perPageInput, orderBy } = args;
+      const perPage = normalizePerPage(perPageInput, 100);
+      if (page < 0) {
+        throw new MastraError(
+          {
+            id: createStorageErrorId("CLOUDFLARE", "LIST_THREADS_BY_RESOURCE_ID", "INVALID_PAGE"),
+            domain: ErrorDomain.STORAGE,
+            category: ErrorCategory.USER,
+            details: { page }
+          },
+          new Error("page must be >= 0")
+        );
       }
-    } catch (error) {
-      throw new MastraError(
-        {
-          id: createStorageErrorId("CLOUDFLARE", "DROP_TABLE", "FAILED"),
-          domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.THIRD_PARTY,
-          details: {
-            tableName
-          }
-        },
-        error
-      );
-    }
-  }
-  getBinding(tableName) {
-    if (!this.bindings) {
-      throw new Error(`Cannot use Workers API binding for ${tableName}: Store initialized with REST API configuration`);
-    }
-    const binding = this.bindings[tableName];
-    if (!binding) throw new Error(`No binding found for namespace ${tableName}`);
-    return binding;
-  }
-  getKey(tableName, record) {
-    const prefix = this.namespacePrefix ? `${this.namespacePrefix}:` : "";
-    switch (tableName) {
-      case TABLE_THREADS:
-        if (!record.id) throw new Error("Thread ID is required");
-        return `${prefix}${tableName}:${record.id}`;
-      case TABLE_MESSAGES:
-        if (!record.threadId || !record.id) throw new Error("Thread ID and Message ID are required");
-        return `${prefix}${tableName}:${record.threadId}:${record.id}`;
-      case TABLE_WORKFLOW_SNAPSHOT:
-        if (!record.workflow_name || !record.run_id) {
-          throw new Error("Workflow name, and run ID are required");
-        }
-        let key = `${prefix}${tableName}:${record.workflow_name}:${record.run_id}`;
-        if (record.resourceId) {
-          key = `${key}:${record.resourceId}`;
-        }
-        return key;
-      case TABLE_TRACES:
-        if (!record.id) throw new Error("Trace ID is required");
-        return `${prefix}${tableName}:${record.id}`;
-      case TABLE_SCORERS:
-        if (!record.id) throw new Error("Score ID is required");
-        return `${prefix}${tableName}:${record.id}`;
-      default:
-        throw new Error(`Unsupported table: ${tableName}`);
-    }
-  }
-  getSchemaKey(tableName) {
-    const prefix = this.namespacePrefix ? `${this.namespacePrefix}:` : "";
-    return `${prefix}schema:${tableName}`;
-  }
-  /**
-   * Helper to safely parse data from KV storage
-   */
-  safeParse(text) {
-    if (!text) return null;
-    try {
-      const data = JSON.parse(text);
-      if (data && typeof data === "object" && "value" in data) {
-        if (typeof data.value === "string") {
-          try {
-            return JSON.parse(data.value);
-          } catch {
-            return data.value;
-          }
-        }
-        return null;
+      const { offset, perPage: perPageForResponse } = calculatePagination(page, perPageInput, perPage);
+      const { field, direction } = this.parseOrderBy(orderBy);
+      const prefix = this.#db.namespacePrefix ? `${this.#db.namespacePrefix}:` : "";
+      const keyObjs = await this.#db.listKV(TABLE_THREADS, { prefix: `${prefix}${TABLE_THREADS}` });
+      const threads = [];
+      for (const { name: key } of keyObjs) {
+        const data = await this.#db.getKV(TABLE_THREADS, key);
+        if (!data) continue;
+        if (data.resourceId !== resourceId) continue;
+        threads.push(data);
       }
-      return data;
-    } catch (error) {
-      const message = error instanceof Error ? error.message : String(error);
-      this.logger.error("Failed to parse text:", { message, text });
-      return null;
-    }
-  }
-  async createNamespaceById(title) {
-    if (this.bindings) {
+      threads.sort((a, b) => {
+        const aTime = new Date(a[field] || 0).getTime();
+        const bTime = new Date(b[field] || 0).getTime();
+        return direction === "ASC" ? aTime - bTime : bTime - aTime;
+      });
+      const end = perPageInput === false ? threads.length : offset + perPage;
+      const paginatedThreads = threads.slice(offset, end);
       return {
-        id: title,
-        // Use title as ID since that's what we need
-        title,
-        supports_url_encoding: true
+        page,
+        perPage: perPageForResponse,
+        total: threads.length,
+        hasMore: perPageInput === false ? false : offset + perPage < threads.length,
+        threads: paginatedThreads
       };
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: createStorageErrorId("CLOUDFLARE", "LIST_THREADS_BY_RESOURCE_ID", "FAILED"),
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          text: "Failed to get threads by resource ID with pagination"
+        },
+        error
+      );
     }
-    return await this.client.kv.namespaces.create({
-      account_id: this.accountId,
-      title
-    });
   }
-  async createNamespace(namespaceName) {
+  async saveThread({ thread }) {
     try {
-      const response = await this.createNamespaceById(namespaceName);
-      return response.id;
+      await this.#db.insert({ tableName: TABLE_THREADS, record: thread });
+      return thread;
     } catch (error) {
-      if (error.message && error.message.includes("already exists")) {
-        const namespaces = await this.listNamespaces();
-        const namespace = namespaces.result.find((ns) => ns.title === namespaceName);
-        if (namespace) return namespace.id;
-      }
-      this.logger.error("Error creating namespace:", error);
-      throw new Error(`Failed to create namespace ${namespaceName}: ${error.message}`);
-    }
-  }
-  async listNamespaces() {
-    if (this.bindings) {
-      return {
-        result: Object.keys(this.bindings).map((name) => ({
-          id: name,
-          title: name,
-          supports_url_encoding: true
-        }))
-      };
-    }
-    let allNamespaces = [];
-    let currentPage = 1;
-    const perPage = 50;
-    let morePagesExist = true;
-    while (morePagesExist) {
-      const response = await this.client.kv.namespaces.list({
-        account_id: this.accountId,
-        page: currentPage,
-        per_page: perPage
-      });
-      if (response.result) {
-        allNamespaces = allNamespaces.concat(response.result);
-      }
-      morePagesExist = response.result ? response.result.length === perPage : false;
-      if (morePagesExist) {
-        currentPage++;
-      }
+      throw new MastraError(
+        {
+          id: createStorageErrorId("CLOUDFLARE", "SAVE_THREAD", "FAILED"),
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: {
+            threadId: thread.id
+          }
+        },
+        error
+      );
     }
-    return { result: allNamespaces };
   }
-  async getNamespaceIdByName(namespaceName) {
+  async updateThread({
+    id,
+    title,
+    metadata
+  }) {
     try {
-      const response = await this.listNamespaces();
-      const namespace = response.result.find((ns) => ns.title === namespaceName);
-      return namespace ? namespace.id : null;
+      const thread = await this.getThreadById({ threadId: id });
+      if (!thread) {
+        throw new Error(`Thread ${id} not found`);
+      }
+      const updatedThread = {
+        ...thread,
+        title,
+        metadata: this.ensureMetadata({
+          ...thread.metadata ?? {},
+          ...metadata
+        }),
+        updatedAt: /* @__PURE__ */ new Date()
+      };
+      await this.#db.insert({ tableName: TABLE_THREADS, record: updatedThread });
+      return updatedThread;
     } catch (error) {
-      this.logger.error(`Failed to get namespace ID for ${namespaceName}:`, error);
-      return null;
-    }
-  }
-  async getOrCreateNamespaceId(namespaceName) {
-    let namespaceId = await this.getNamespaceIdByName(namespaceName);
-    if (!namespaceId) {
-      namespaceId = await this.createNamespace(namespaceName);
+      throw new MastraError(
+        {
+          id: createStorageErrorId("CLOUDFLARE", "UPDATE_THREAD", "FAILED"),
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: {
+            threadId: id,
+            title
+          }
+        },
+        error
+      );
     }
-    return namespaceId;
   }
-  async getNamespaceId(tableName) {
-    const prefix = this.namespacePrefix ? `${this.namespacePrefix}_` : "";
+  getMessageKey(threadId, messageId) {
     try {
-      return await this.getOrCreateNamespaceId(`${prefix}${tableName}`);
+      return this.#db.getKey(TABLE_MESSAGES, { threadId, id: messageId });
     } catch (error) {
-      this.logger.error("Error fetching namespace ID:", error);
-      throw new Error(`Failed to fetch namespace ID for table ${tableName}: ${error.message}`);
+      const message = error instanceof Error ? error.message : String(error);
+      this.logger?.error(`Error getting message key for thread ${threadId} and message ${messageId}:`, { message });
+      throw error;
     }
   }
-  async getNamespaceValue(tableName, key) {
+  getThreadMessagesKey(threadId) {
     try {
-      if (this.bindings) {
-        const binding = this.getBinding(tableName);
-        const result = await binding.getWithMetadata(key, "text");
-        if (!result) return null;
-        return JSON.stringify(result);
-      } else {
-        const namespaceId = await this.getNamespaceId(tableName);
-        const response = await this.client.kv.namespaces.values.get(namespaceId, key, {
-          account_id: this.accountId
-        });
-        return await response.text();
-      }
+      return this.#db.getKey(TABLE_MESSAGES, { threadId, id: "messages" });
     } catch (error) {
-      if (error.message && error.message.includes("key not found")) {
-        return null;
-      }
       const message = error instanceof Error ? error.message : String(error);
-      this.logger.error(`Failed to get value for ${tableName} ${key}:`, { message });
+      this.logger?.error(`Error getting thread messages key for thread ${threadId}:`, { message });
       throw error;
     }
   }
-  async getKV(tableName, key) {
+  async deleteThread({ threadId }) {
     try {
-      const text = await this.getNamespaceValue(tableName, key);
-      return this.safeParse(text);
+      const thread = await this.getThreadById({ threadId });
+      if (!thread) {
+        throw new Error(`Thread ${threadId} not found`);
+      }
+      const messageKeys = await this.#db.listKV(TABLE_MESSAGES);
+      const threadMessageKeys = messageKeys.filter((key) => key.name.includes(`${TABLE_MESSAGES}:${threadId}:`));
+      await Promise.all([
+        // Delete message order
+        this.#db.deleteKV(TABLE_MESSAGES, this.getThreadMessagesKey(threadId)),
+        // Delete all messages
+        ...threadMessageKeys.map((key) => this.#db.deleteKV(TABLE_MESSAGES, key.name)),
+        // Delete thread
+        this.#db.deleteKV(TABLE_THREADS, this.#db.getKey(TABLE_THREADS, { id: threadId }))
+      ]);
     } catch (error) {
-      this.logger.error(`Failed to get KV value for ${tableName}:${key}:`, error);
-      throw new Error(`Failed to get KV value: ${error.message}`);
+      throw new MastraError(
+        {
+          id: createStorageErrorId("CLOUDFLARE", "DELETE_THREAD", "FAILED"),
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: {
+            threadId
+          }
+        },
+        error
+      );
     }
   }
-  async getTableSchema(tableName) {
-    try {
-      const schemaKey = this.getSchemaKey(tableName);
-      return await this.getKV(tableName, schemaKey);
-    } catch (error) {
-      const message = error instanceof Error ? error.message : String(error);
-      this.logger.error(`Failed to get schema for ${tableName}:`, { message });
+  /**
+   * 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.#db.namespacePrefix ? `${this.#db.namespacePrefix}:` : "";
+      const threadKeys = await this.#db.listKV(TABLE_THREADS, { prefix: `${prefix}${TABLE_THREADS}` });
+      for (const { name: threadKey } of threadKeys) {
+        const threadId = threadKey.split(":").pop();
+        if (!threadId || threadId === "messages") continue;
+        const messageKey = this.getMessageKey(threadId, messageId);
+        const message = await this.#db.getKV(TABLE_MESSAGES, messageKey);
+        if (message) {
+          return { ...message, threadId };
+        }
+      }
       return null;
+    } catch (error) {
+      this.logger?.error(`Error finding message ${messageId} in any thread:`, error);
+      return null;
+    }
+  }
+  /**
+   * Queue for serializing sorted order updates.
+   * Updates the sorted order for a given key. This operation is eventually consistent.
+   */
+  updateQueue = /* @__PURE__ */ new Map();
+  async updateSorting(threadMessages) {
+    return threadMessages.map((msg) => ({
+      message: msg,
+      // Use _index if available, otherwise timestamp, matching Upstash
+      score: msg._index !== void 0 ? msg._index : msg.createdAt.getTime()
+    })).sort((a, b) => a.score - b.score).map((item) => ({
+      id: item.message.id,
+      score: item.score
+    }));
+  }
+  /**
+   * Updates the sorted order for a given key. This operation is eventually consistent.
+   * Note: Operations on the same orderKey are serialized using a queue to prevent
+   * concurrent updates from conflicting with each other.
+   */
+  async updateSortedMessages(orderKey, newEntries) {
+    const currentPromise = this.updateQueue.get(orderKey) || Promise.resolve();
+    const nextPromise = currentPromise.then(async () => {
+      try {
+        const currentOrder = await this.getSortedMessages(orderKey);
+        const orderMap = new Map(currentOrder.map((entry) => [entry.id, entry]));
+        for (const entry of newEntries) {
+          orderMap.set(entry.id, entry);
+        }
+        const updatedOrder = Array.from(orderMap.values()).sort((a, b) => a.score - b.score);
+        await this.#db.putKV({
+          tableName: TABLE_MESSAGES,
+          key: orderKey,
+          value: JSON.stringify(updatedOrder)
+        });
+      } catch (error) {
+        const message = error instanceof Error ? error.message : String(error);
+        this.logger?.error(`Error updating sorted order for key ${orderKey}:`, { message });
+        throw error;
+      } finally {
+        if (this.updateQueue.get(orderKey) === nextPromise) {
+          this.updateQueue.delete(orderKey);
+        }
+      }
+    });
+    this.updateQueue.set(orderKey, nextPromise);
+    return nextPromise;
+  }
+  async getSortedMessages(orderKey) {
+    const raw = await this.#db.getKV(TABLE_MESSAGES, orderKey);
+    if (!raw) return [];
+    try {
+      const arr = JSON.parse(typeof raw === "string" ? raw : JSON.stringify(raw));
+      return Array.isArray(arr) ? arr : [];
+    } catch (e) {
+      this.logger?.error(`Error parsing order data for key ${orderKey}:`, { e });
+      return [];
+    }
+  }
+  async migrateMessage(messageId, fromThreadId, toThreadId) {
+    try {
+      const oldMessageKey = this.getMessageKey(fromThreadId, messageId);
+      const message = await this.#db.getKV(TABLE_MESSAGES, oldMessageKey);
+      if (!message) return;
+      const updatedMessage = {
+        ...message,
+        threadId: toThreadId
+      };
+      const newMessageKey = this.getMessageKey(toThreadId, messageId);
+      await this.#db.putKV({ tableName: TABLE_MESSAGES, key: newMessageKey, value: updatedMessage });
+      const oldOrderKey = this.getThreadMessagesKey(fromThreadId);
+      const oldEntries = await this.getSortedMessages(oldOrderKey);
+      const filteredEntries = oldEntries.filter((entry) => entry.id !== messageId);
+      await this.updateSortedMessages(oldOrderKey, filteredEntries);
+      const newOrderKey = this.getThreadMessagesKey(toThreadId);
+      const newEntries = await this.getSortedMessages(newOrderKey);
+      const newEntry = { id: messageId, score: Date.now() };
+      newEntries.push(newEntry);
+      await this.updateSortedMessages(newOrderKey, newEntries);
+      await this.#db.deleteKV(TABLE_MESSAGES, oldMessageKey);
+    } catch (error) {
+      this.logger?.error(`Error migrating message ${messageId} from ${fromThreadId} to ${toThreadId}:`, error);
+      throw error;
+    }
+  }
+  async saveMessages(args) {
+    const { messages } = args;
+    if (!Array.isArray(messages) || messages.length === 0) return { messages: [] };
+    try {
+      const validatedMessages = messages.map((message, index) => {
+        const errors = [];
+        if (!message.id) errors.push("id is required");
+        if (!message.threadId) errors.push("threadId is required");
+        if (!message.content) errors.push("content is required");
+        if (!message.role) errors.push("role is required");
+        if (!message.createdAt) errors.push("createdAt is required");
+        if (message.resourceId === null || message.resourceId === void 0) errors.push("resourceId is required");
+        if (errors.length > 0) {
+          throw new Error(`Invalid message at index ${index}: ${errors.join(", ")}`);
+        }
+        return {
+          ...message,
+          createdAt: ensureDate(message.createdAt),
+          type: message.type || "v2",
+          _index: index
+        };
+      }).filter((m) => !!m);
+      const messageMigrationTasks = [];
+      for (const message of validatedMessages) {
+        const existingMessage = await this.findMessageInAnyThread(message.id);
+        this.logger?.debug(
+          `Checking message ${message.id}: existing=${existingMessage?.threadId}, new=${message.threadId}`
+        );
+        if (existingMessage && existingMessage.threadId && existingMessage.threadId !== message.threadId) {
+          this.logger?.debug(`Migrating message ${message.id} from ${existingMessage.threadId} to ${message.threadId}`);
+          messageMigrationTasks.push(this.migrateMessage(message.id, existingMessage.threadId, message.threadId));
+        }
+      }
+      await Promise.all(messageMigrationTasks);
+      const messagesByThread = validatedMessages.reduce((acc, message) => {
+        if (message.threadId && !acc.has(message.threadId)) {
+          acc.set(message.threadId, []);
+        }
+        if (message.threadId) {
+          acc.get(message.threadId).push(message);
+        }
+        return acc;
+      }, /* @__PURE__ */ new Map());
+      await Promise.all(
+        Array.from(messagesByThread.entries()).map(async ([threadId, threadMessages]) => {
+          try {
+            const thread = await this.getThreadById({ threadId });
+            if (!thread) {
+              throw new Error(`Thread ${threadId} not found`);
+            }
+            await Promise.all(
+              threadMessages.map(async (message) => {
+                const key = this.getMessageKey(threadId, message.id);
+                const { _index, ...cleanMessage } = message;
+                const serializedMessage = {
+                  ...cleanMessage,
+                  createdAt: serializeDate(cleanMessage.createdAt)
+                };
+                this.logger?.debug(`Saving message ${message.id}`, {
+                  contentSummary: this.summarizeMessageContent(serializedMessage.content)
+                });
+                await this.#db.putKV({ tableName: TABLE_MESSAGES, key, value: serializedMessage });
+              })
+            );
+            const orderKey = this.getThreadMessagesKey(threadId);
+            const entries = await this.updateSorting(threadMessages);
+            await this.updateSortedMessages(orderKey, entries);
+            const updatedThread = {
+              ...thread,
+              updatedAt: /* @__PURE__ */ new Date()
+            };
+            await this.#db.putKV({
+              tableName: TABLE_THREADS,
+              key: this.#db.getKey(TABLE_THREADS, { id: threadId }),
+              value: updatedThread
+            });
+          } catch (error) {
+            throw new MastraError(
+              {
+                id: createStorageErrorId("CLOUDFLARE", "SAVE_MESSAGES", "FAILED"),
+                domain: ErrorDomain.STORAGE,
+                category: ErrorCategory.THIRD_PARTY,
+                details: {
+                  threadId
+                }
+              },
+              error
+            );
+          }
+        })
+      );
+      const prepared = validatedMessages.map(
+        ({ _index, ...message }) => ({ ...message, type: message.type !== "v2" ? message.type : void 0 })
+      );
+      const list = new MessageList().add(prepared, "memory");
+      return { messages: list.get.all.db() };
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: createStorageErrorId("CLOUDFLARE", "SAVE_MESSAGES", "FAILED"),
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY
+        },
+        error
+      );
+    }
+  }
+  async getRank(orderKey, id) {
+    const order = await this.getSortedMessages(orderKey);
+    const index = order.findIndex((item) => item.id === id);
+    return index >= 0 ? index : null;
+  }
+  async getRange(orderKey, start, end) {
+    const order = await this.getSortedMessages(orderKey);
+    const actualStart = start < 0 ? Math.max(0, order.length + start) : start;
+    const actualEnd = end < 0 ? order.length + end : Math.min(end, order.length - 1);
+    const sliced = order.slice(actualStart, actualEnd + 1);
+    return sliced.map((item) => item.id);
+  }
+  async getLastN(orderKey, n) {
+    return this.getRange(orderKey, -n, -1);
+  }
+  async getFullOrder(orderKey) {
+    return this.getRange(orderKey, 0, -1);
+  }
+  /**
+   * 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) => {
+        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);
+        if (!item.withPreviousMessages && !item.withNextMessages) return;
+        const rank = await this.getRank(threadMessagesKey, item.id);
+        if (rank === null) return;
+        if (item.withPreviousMessages) {
+          const prevIds = await this.getRange(
+            threadMessagesKey,
+            Math.max(0, rank - item.withPreviousMessages),
+            rank - 1
+          );
+          prevIds.forEach((id) => messageIds.add(id));
+        }
+        if (item.withNextMessages) {
+          const nextIds = await this.getRange(threadMessagesKey, rank + 1, rank + item.withNextMessages);
+          nextIds.forEach((id) => messageIds.add(id));
+        }
+      })
+    );
+  }
+  async getRecentMessages(threadId, limit, messageIds) {
+    if (!threadId.trim()) throw new Error("threadId must be a non-empty string");
+    if (limit <= 0) return;
+    try {
+      const threadMessagesKey = this.getThreadMessagesKey(threadId);
+      const latestIds = await this.getLastN(threadMessagesKey, limit);
+      latestIds.forEach((id) => messageIds.add(id));
+    } catch {
+      this.logger?.debug(`No message order found for thread ${threadId}, skipping latest messages`);
     }
   }
-  validateColumnValue(value, column) {
-    if (value === void 0 || value === null) {
-      return column.nullable ?? false;
+  /**
+   * 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) {
+      for (const item of include) {
+        if (item.threadId) {
+          messageIdToThreadId.set(item.id, item.threadId);
+        }
+      }
     }
-    switch (column.type) {
-      case "text":
-      case "uuid":
-        return typeof value === "string";
-      case "integer":
-      case "bigint":
-        return typeof value === "number";
-      case "timestamp":
-        return value instanceof Date || typeof value === "string" && !isNaN(Date.parse(value));
-      case "jsonb":
-        if (typeof value !== "object") return false;
+    const messages = await Promise.all(
+      messageIds.map(async (id) => {
         try {
-          JSON.stringify(value);
-          return true;
-        } catch {
-          return false;
+          let threadId = messageIdToThreadId.get(id);
+          if (!threadId) {
+            if (targetThreadId) {
+              threadId = targetThreadId;
+            } else {
+              const foundMessage = await this.findMessageInAnyThread(id);
+              if (foundMessage) {
+                threadId = foundMessage.threadId;
+              }
+            }
+          }
+          if (!threadId) return null;
+          const key = this.getMessageKey(threadId, id);
+          const data = await this.#db.getKV(TABLE_MESSAGES, key);
+          if (!data) return null;
+          const parsed = typeof data === "string" ? JSON.parse(data) : data;
+          this.logger?.debug(`Retrieved message ${id} from thread ${threadId}`, {
+            contentSummary: this.summarizeMessageContent(parsed.content)
+          });
+          return parsed;
+        } catch (error) {
+          const message = error instanceof Error ? error.message : String(error);
+          this.logger?.error(`Error retrieving message ${id}:`, { message });
+          return null;
         }
-      default:
-        return false;
+      })
+    );
+    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 {
+      const messages = (await Promise.all(messageIds.map((id) => this.findMessageInAnyThread(id)))).filter(
+        (result) => !!result
+      );
+      const prepared = messages.map(({ _index, ...message }) => ({
+        ...message,
+        ...message.type !== `v2` && { type: message.type },
+        createdAt: ensureDate(message.createdAt)
+      }));
+      const list = new MessageList().add(prepared, "memory");
+      return { messages: list.get.all.db() };
+    } catch (error) {
+      const mastraError = new MastraError(
+        {
+          id: createStorageErrorId("CLOUDFLARE", "LIST_MESSAGES_BY_ID", "FAILED"),
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          text: `Error retrieving messages by ID`,
+          details: {
+            messageIds: JSON.stringify(messageIds)
+          }
+        },
+        error
+      );
+      this.logger?.trackException(mastraError);
+      this.logger?.error(mastraError.toString());
+      return { messages: [] };
     }
   }
-  async validateAgainstSchema(record, schema) {
+  async listMessages(args) {
+    const { threadId, resourceId, include, filter, perPage: perPageInput, page = 0, orderBy } = args;
+    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: createStorageErrorId("CLOUDFLARE", "LIST_MESSAGES", "INVALID_THREAD_ID"),
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { threadId: Array.isArray(threadId) ? JSON.stringify(threadId) : String(threadId) }
+        },
+        new Error("threadId must be a non-empty string or array of non-empty strings")
+      );
+    }
+    const perPage = normalizePerPage(perPageInput, 40);
+    const { offset, perPage: perPageForResponse } = calculatePagination(page, perPageInput, perPage);
     try {
-      if (!schema || typeof schema !== "object" || schema.value === null) {
-        throw new Error("Invalid schema format");
+      if (page < 0) {
+        throw new MastraError(
+          {
+            id: createStorageErrorId("CLOUDFLARE", "LIST_MESSAGES", "INVALID_PAGE"),
+            domain: ErrorDomain.STORAGE,
+            category: ErrorCategory.USER,
+            details: { page }
+          },
+          new Error("page must be >= 0")
+        );
       }
-      for (const [columnName, column] of Object.entries(schema)) {
-        const value = record[columnName];
-        if (column.primaryKey && (value === void 0 || value === null)) {
-          throw new Error(`Missing primary key value for column ${columnName}`);
+      const { field, direction } = this.parseOrderBy(orderBy, "ASC");
+      const threadMessageIds = /* @__PURE__ */ new Set();
+      for (const tid of threadIds) {
+        try {
+          const threadMessagesKey = this.getThreadMessagesKey(tid);
+          const allIds = await this.getFullOrder(threadMessagesKey);
+          allIds.forEach((id) => threadMessageIds.add(id));
+        } catch {
         }
-        if (!this.validateColumnValue(value, column)) {
-          const valueType = value === null ? "null" : typeof value;
-          throw new Error(`Invalid value for column ${columnName}: expected ${column.type}, got ${valueType}`);
+      }
+      const threadMessages = await this.fetchAndParseMessagesFromMultipleThreads(
+        Array.from(threadMessageIds),
+        void 0,
+        threadIds.length === 1 ? threadIds[0] : void 0
+      );
+      let filteredThreadMessages = threadMessages;
+      if (resourceId) {
+        filteredThreadMessages = filteredThreadMessages.filter((msg) => msg.resourceId === resourceId);
+      }
+      const dateRange = filter?.dateRange;
+      if (dateRange) {
+        filteredThreadMessages = filteredThreadMessages.filter((msg) => {
+          const messageDate = new Date(msg.createdAt);
+          if (dateRange.start && messageDate < new Date(dateRange.start)) return false;
+          if (dateRange.end && messageDate > new Date(dateRange.end)) return false;
+          return true;
+        });
+      }
+      const total = filteredThreadMessages.length;
+      if (perPage === 0 && (!include || include.length === 0)) {
+        return {
+          messages: [],
+          total,
+          page,
+          perPage: perPageForResponse,
+          hasMore: offset < total
+        };
+      }
+      filteredThreadMessages.sort((a, b) => {
+        const timeA = new Date(a.createdAt).getTime();
+        const timeB = new Date(b.createdAt).getTime();
+        const timeDiff = direction === "ASC" ? timeA - timeB : timeB - timeA;
+        if (timeDiff === 0) {
+          return a.id.localeCompare(b.id);
         }
+        return timeDiff;
+      });
+      let paginatedMessages;
+      if (perPage === 0) {
+        paginatedMessages = [];
+      } else if (perPage === Number.MAX_SAFE_INTEGER) {
+        paginatedMessages = filteredThreadMessages;
+      } else {
+        paginatedMessages = filteredThreadMessages.slice(offset, offset + perPage);
       }
-    } catch (error) {
-      const message = error instanceof Error ? error.message : String(error);
-      this.logger.error(`Error validating record against schema:`, { message, record, schema });
-      throw error;
-    }
-  }
-  async validateRecord(record, tableName) {
-    try {
-      if (!record || typeof record !== "object") {
-        throw new Error("Record must be an object");
+      let includedMessages = [];
+      if (include && include.length > 0) {
+        const includedMessageIds = /* @__PURE__ */ new Set();
+        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) {
+          includedMessages = await this.fetchAndParseMessagesFromMultipleThreads(idsToFetch, include, void 0);
+        }
       }
-      const recordTyped = record;
-      const schema = await this.getTableSchema(tableName);
-      if (schema) {
-        await this.validateAgainstSchema(recordTyped, schema);
-        return;
+      const seenIds = /* @__PURE__ */ new Set();
+      const allMessages = [];
+      for (const msg of paginatedMessages) {
+        if (!seenIds.has(msg.id)) {
+          allMessages.push(msg);
+          seenIds.add(msg.id);
+        }
       }
-      switch (tableName) {
-        case TABLE_THREADS:
-          if (!("id" in recordTyped) || !("resourceId" in recordTyped) || !("title" in recordTyped)) {
-            throw new Error("Thread record missing required fields");
-          }
-          break;
-        case TABLE_MESSAGES:
-          if (!("id" in recordTyped) || !("threadId" in recordTyped) || !("content" in recordTyped) || !("role" in recordTyped)) {
-            throw new Error("Message record missing required fields");
-          }
-          break;
-        case TABLE_WORKFLOW_SNAPSHOT:
-          if (!("workflow_name" in recordTyped) || !("run_id" in recordTyped)) {
-            throw new Error("Workflow record missing required fields");
-          }
-          break;
-        case TABLE_TRACES:
-          if (!("id" in recordTyped)) {
-            throw new Error("Trace record missing required fields");
-          }
-          break;
-        case TABLE_SCORERS:
-          if (!("id" in recordTyped) || !("scorerId" in recordTyped)) {
-            throw new Error("Score record missing required fields");
-          }
-          break;
-        default:
-          throw new Error(`Unknown table type: ${tableName}`);
+      for (const msg of includedMessages) {
+        if (!seenIds.has(msg.id)) {
+          allMessages.push(msg);
+          seenIds.add(msg.id);
+        }
       }
-    } catch (error) {
-      const message = error instanceof Error ? error.message : String(error);
-      this.logger.error(`Failed to validate record for ${tableName}:`, { message, record });
-      throw error;
-    }
-  }
-  async insert({ tableName, record }) {
-    try {
-      const key = this.getKey(tableName, record);
-      const processedRecord = { ...record };
-      await this.validateRecord(processedRecord, tableName);
-      await this.putKV({ tableName, key, value: processedRecord });
-    } catch (error) {
-      throw new MastraError(
-        {
-          id: createStorageErrorId("CLOUDFLARE", "INSERT", "FAILED"),
-          domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.THIRD_PARTY,
-          details: {
-            tableName
-          }
-        },
-        error
+      allMessages.sort((a, b) => {
+        const timeA = new Date(a.createdAt).getTime();
+        const timeB = new Date(b.createdAt).getTime();
+        const timeDiff = direction === "ASC" ? timeA - timeB : timeB - timeA;
+        if (timeDiff === 0) {
+          return a.id.localeCompare(b.id);
+        }
+        return timeDiff;
+      });
+      let filteredMessages = allMessages;
+      const paginatedCount = paginatedMessages.length;
+      if (total === 0 && filteredMessages.length === 0 && (!include || include.length === 0)) {
+        return {
+          messages: [],
+          total: 0,
+          page,
+          perPage: perPageForResponse,
+          hasMore: false
+        };
+      }
+      const prepared = filteredMessages.map(({ _index, ...message }) => ({
+        ...message,
+        type: message.type !== "v2" ? message.type : void 0,
+        createdAt: ensureDate(message.createdAt)
+      }));
+      const primaryThreadId = Array.isArray(threadId) ? threadId[0] : threadId;
+      const list = new MessageList({ threadId: primaryThreadId, resourceId }).add(
+        prepared,
+        "memory"
       );
-    }
-  }
-  async load({ tableName, keys }) {
-    try {
-      const key = this.getKey(tableName, keys);
-      const data = await this.getKV(tableName, key);
-      if (!data) return null;
-      return data;
+      let finalMessages = list.get.all.db();
+      finalMessages = finalMessages.sort((a, b) => {
+        const isDateField = field === "createdAt" || field === "updatedAt";
+        const aVal = isDateField ? new Date(a[field]).getTime() : a[field];
+        const bVal = isDateField ? new Date(b[field]).getTime() : b[field];
+        if (aVal == null && bVal == null) return a.id.localeCompare(b.id);
+        if (aVal == null) return 1;
+        if (bVal == null) return -1;
+        if (typeof aVal === "number" && typeof bVal === "number") {
+          const cmp2 = direction === "ASC" ? aVal - bVal : bVal - aVal;
+          return cmp2 !== 0 ? cmp2 : a.id.localeCompare(b.id);
+        }
+        const cmp = direction === "ASC" ? String(aVal).localeCompare(String(bVal)) : String(bVal).localeCompare(String(aVal));
+        return cmp !== 0 ? cmp : a.id.localeCompare(b.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 {
+        messages: finalMessages,
+        total,
+        page,
+        perPage: perPageForResponse,
+        hasMore
+      };
     } catch (error) {
       const mastraError = new MastraError(
         {
-          id: createStorageErrorId("CLOUDFLARE", "LOAD", "FAILED"),
+          id: createStorageErrorId("CLOUDFLARE", "LIST_MESSAGES", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
+          text: `Failed to list messages for thread ${Array.isArray(threadId) ? threadId.join(",") : threadId}: ${error instanceof Error ? error.message : String(error)}`,
           details: {
-            tableName
+            threadId: Array.isArray(threadId) ? threadId.join(",") : threadId,
+            resourceId: resourceId ?? ""
           }
         },
         error
       );
-      this.logger?.trackException(mastraError);
-      this.logger?.error(mastraError.toString());
-      return null;
+      this.logger?.error?.(mastraError.toString());
+      this.logger?.trackException?.(mastraError);
+      return {
+        messages: [],
+        total: 0,
+        page,
+        perPage: perPageForResponse,
+        hasMore: false
+      };
     }
   }
-  async batchInsert(input) {
-    if (!input.records || input.records.length === 0) return;
+  async updateMessages(args) {
     try {
-      await Promise.all(
-        input.records.map(async (record) => {
-          const key = this.getKey(input.tableName, record);
-          await this.putKV({ tableName: input.tableName, key, value: record });
-        })
-      );
+      const { messages } = args;
+      const updatedMessages = [];
+      for (const messageUpdate of messages) {
+        const { id, content, ...otherFields } = messageUpdate;
+        const prefix = this.#db.namespacePrefix ? `${this.#db.namespacePrefix}:` : "";
+        const keyObjs = await this.#db.listKV(TABLE_MESSAGES, { prefix: `${prefix}${TABLE_MESSAGES}` });
+        let existingMessage = null;
+        let messageKey = "";
+        for (const { name: key } of keyObjs) {
+          const data = await this.#db.getKV(TABLE_MESSAGES, key);
+          if (data && data.id === id) {
+            existingMessage = data;
+            messageKey = key;
+            break;
+          }
+        }
+        if (!existingMessage) {
+          continue;
+        }
+        const updatedMessage = {
+          ...existingMessage,
+          ...otherFields,
+          id
+        };
+        if (content) {
+          if (content.metadata !== void 0) {
+            updatedMessage.content = {
+              ...updatedMessage.content,
+              metadata: {
+                ...updatedMessage.content?.metadata,
+                ...content.metadata
+              }
+            };
+          }
+          if (content.content !== void 0) {
+            updatedMessage.content = {
+              ...updatedMessage.content,
+              content: content.content
+            };
+          }
+        }
+        if ("threadId" in messageUpdate && messageUpdate.threadId && messageUpdate.threadId !== existingMessage.threadId) {
+          await this.#db.deleteKV(TABLE_MESSAGES, messageKey);
+          updatedMessage.threadId = messageUpdate.threadId;
+          const newMessageKey = this.getMessageKey(messageUpdate.threadId, id);
+          await this.#db.putKV({
+            tableName: TABLE_MESSAGES,
+            key: newMessageKey,
+            value: updatedMessage
+          });
+          if (existingMessage.threadId) {
+            const sourceOrderKey = this.getThreadMessagesKey(existingMessage.threadId);
+            const sourceEntries = await this.getSortedMessages(sourceOrderKey);
+            const filteredEntries = sourceEntries.filter((entry) => entry.id !== id);
+            await this.updateSortedMessages(sourceOrderKey, filteredEntries);
+          }
+          const destOrderKey = this.getThreadMessagesKey(messageUpdate.threadId);
+          const destEntries = await this.getSortedMessages(destOrderKey);
+          const newEntry = { id, score: Date.now() };
+          destEntries.push(newEntry);
+          await this.updateSortedMessages(destOrderKey, destEntries);
+        } else {
+          await this.#db.putKV({
+            tableName: TABLE_MESSAGES,
+            key: messageKey,
+            value: updatedMessage
+          });
+        }
+        const threadsToUpdate = /* @__PURE__ */ new Set();
+        if (updatedMessage.threadId) {
+          threadsToUpdate.add(updatedMessage.threadId);
+        }
+        if ("threadId" in messageUpdate && messageUpdate.threadId && messageUpdate.threadId !== existingMessage.threadId) {
+          if (existingMessage.threadId) {
+            threadsToUpdate.add(existingMessage.threadId);
+          }
+          threadsToUpdate.add(messageUpdate.threadId);
+        }
+        for (const threadId of threadsToUpdate) {
+          const thread = await this.getThreadById({ threadId });
+          if (thread) {
+            const updatedThread = {
+              ...thread,
+              updatedAt: /* @__PURE__ */ new Date()
+            };
+            await this.#db.putKV({
+              tableName: TABLE_THREADS,
+              key: this.#db.getKey(TABLE_THREADS, { id: threadId }),
+              value: updatedThread
+            });
+          }
+        }
+        updatedMessages.push(updatedMessage);
+      }
+      return updatedMessages;
     } catch (error) {
       throw new MastraError(
         {
-          id: createStorageErrorId("CLOUDFLARE", "BATCH_INSERT", "FAILED"),
+          id: createStorageErrorId("CLOUDFLARE", "UPDATE_MESSAGES", "FAILED"),
           domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.THIRD_PARTY,
-          text: `Error in batch insert for table ${input.tableName}`,
-          details: {
-            tableName: input.tableName
-          }
-        },
-        error
-      );
-    }
-  }
-  /**
-   * Helper to safely serialize data for KV storage
-   */
-  safeSerialize(data) {
-    return typeof data === "string" ? data : JSON.stringify(data);
-  }
-  async putNamespaceValue({
-    tableName,
-    key,
-    value,
-    metadata
-  }) {
-    try {
-      const serializedValue = this.safeSerialize(value);
-      const serializedMetadata = metadata ? this.safeSerialize(metadata) : "";
-      if (this.bindings) {
-        const binding = this.getBinding(tableName);
-        await binding.put(key, serializedValue, { metadata: serializedMetadata });
-      } else {
-        const namespaceId = await this.getNamespaceId(tableName);
-        await this.client.kv.namespaces.values.update(namespaceId, key, {
-          account_id: this.accountId,
-          value: serializedValue,
-          metadata: serializedMetadata
-        });
-      }
-    } catch (error) {
-      const message = error instanceof Error ? error.message : String(error);
-      this.logger.error(`Failed to put value for ${tableName} ${key}:`, { message });
-      throw error;
-    }
-  }
-  async putKV({
-    tableName,
-    key,
-    value,
-    metadata
-  }) {
-    try {
-      await this.putNamespaceValue({ tableName, key, value, metadata });
-    } catch (error) {
-      this.logger.error(`Failed to put KV value for ${tableName}:${key}:`, error);
-      throw new Error(`Failed to put KV value: ${error.message}`);
+          category: ErrorCategory.THIRD_PARTY,
+          text: "Failed to update messages"
+        },
+        error
+      );
     }
   }
-  async createTable({
-    tableName,
-    schema
-  }) {
+  async getResourceById({ resourceId }) {
     try {
-      const schemaKey = this.getSchemaKey(tableName);
-      const metadata = {
-        type: "table_schema",
-        tableName,
-        createdAt: (/* @__PURE__ */ new Date()).toISOString()
+      const data = await this.#db.getKV(TABLE_RESOURCES, resourceId);
+      if (!data) return null;
+      const resource = typeof data === "string" ? JSON.parse(data) : data;
+      return {
+        ...resource,
+        createdAt: ensureDate(resource.createdAt),
+        updatedAt: ensureDate(resource.updatedAt),
+        metadata: this.ensureMetadata(resource.metadata)
       };
-      await this.putKV({ tableName, key: schemaKey, value: schema, metadata });
     } catch (error) {
-      throw new MastraError(
+      const mastraError = new MastraError(
         {
-          id: createStorageErrorId("CLOUDFLARE", "CREATE_TABLE", "FAILED"),
+          id: createStorageErrorId("CLOUDFLARE", "GET_RESOURCE_BY_ID", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: {
-            tableName
+            resourceId
           }
         },
         error
       );
+      this.logger?.trackException(mastraError);
+      this.logger?.error(mastraError.toString());
+      return null;
     }
   }
-  async listNamespaceKeys(tableName, options) {
+  async saveResource({ resource }) {
     try {
-      if (this.bindings) {
-        const binding = this.getBinding(tableName);
-        const response = await binding.list({
-          limit: options?.limit || 1e3,
-          prefix: options?.prefix
-        });
-        return response.keys;
-      } else {
-        const namespaceId = await this.getNamespaceId(tableName);
-        const response = await this.client.kv.namespaces.keys.list(namespaceId, {
-          account_id: this.accountId,
-          limit: options?.limit || 1e3,
-          prefix: options?.prefix
-        });
-        return response.result;
-      }
+      const resourceToSave = {
+        ...resource,
+        metadata: resource.metadata ? JSON.stringify(resource.metadata) : null
+      };
+      await this.#db.putKV({
+        tableName: TABLE_RESOURCES,
+        key: resource.id,
+        value: resourceToSave
+      });
+      return resource;
     } catch (error) {
       throw new MastraError(
         {
-          id: createStorageErrorId("CLOUDFLARE", "LIST_NAMESPACE_KEYS", "FAILED"),
+          id: createStorageErrorId("CLOUDFLARE", "SAVE_RESOURCE", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: {
-            tableName
+            resourceId: resource.id
           }
         },
         error
       );
     }
   }
-  async deleteNamespaceValue(tableName, key) {
-    if (this.bindings) {
-      const binding = this.getBinding(tableName);
-      await binding.delete(key);
-    } else {
-      const namespaceId = await this.getNamespaceId(tableName);
-      await this.client.kv.namespaces.values.delete(namespaceId, key, {
-        account_id: this.accountId
-      });
-    }
-  }
-  async deleteKV(tableName, key) {
-    try {
-      await this.deleteNamespaceValue(tableName, key);
-    } catch (error) {
-      this.logger.error(`Failed to delete KV value for ${tableName}:${key}:`, error);
-      throw new Error(`Failed to delete KV value: ${error.message}`);
+  async updateResource({
+    resourceId,
+    workingMemory,
+    metadata
+  }) {
+    const existingResource = await this.getResourceById({ resourceId });
+    if (!existingResource) {
+      const newResource = {
+        id: resourceId,
+        workingMemory,
+        metadata: metadata || {},
+        createdAt: /* @__PURE__ */ new Date(),
+        updatedAt: /* @__PURE__ */ new Date()
+      };
+      return this.saveResource({ resource: newResource });
     }
+    const updatedAt = /* @__PURE__ */ new Date();
+    const updatedResource = {
+      ...existingResource,
+      workingMemory: workingMemory !== void 0 ? workingMemory : existingResource.workingMemory,
+      metadata: {
+        ...existingResource.metadata,
+        ...metadata
+      },
+      updatedAt
+    };
+    return this.saveResource({ resource: updatedResource });
   }
-  async listKV(tableName, options) {
+  async deleteMessages(messageIds) {
+    if (messageIds.length === 0) return;
     try {
-      return await this.listNamespaceKeys(tableName, options);
+      const threadIds = /* @__PURE__ */ new Set();
+      for (const messageId of messageIds) {
+        const message = await this.findMessageInAnyThread(messageId);
+        if (message?.threadId) {
+          threadIds.add(message.threadId);
+          const messageKey = this.getMessageKey(message.threadId, messageId);
+          await this.#db.deleteKV(TABLE_MESSAGES, messageKey);
+          const orderKey = this.getThreadMessagesKey(message.threadId);
+          const entries = await this.getSortedMessages(orderKey);
+          const filteredEntries = entries.filter((entry) => entry.id !== messageId);
+          if (filteredEntries.length > 0) {
+            await this.#db.putKV({
+              tableName: TABLE_MESSAGES,
+              key: orderKey,
+              value: JSON.stringify(filteredEntries)
+            });
+          } else {
+            await this.#db.deleteKV(TABLE_MESSAGES, orderKey);
+          }
+        }
+      }
+      for (const threadId of threadIds) {
+        const thread = await this.getThreadById({ threadId });
+        if (thread) {
+          const updatedThread = {
+            ...thread,
+            updatedAt: /* @__PURE__ */ new Date()
+          };
+          await this.#db.putKV({
+            tableName: TABLE_THREADS,
+            key: this.#db.getKey(TABLE_THREADS, { id: threadId }),
+            value: updatedThread
+          });
+        }
+      }
     } catch (error) {
-      this.logger.error(`Failed to list KV for ${tableName}:`, error);
-      throw new Error(`Failed to list KV: ${error.message}`);
+      throw new MastraError(
+        {
+          id: createStorageErrorId("CLOUDFLARE", "DELETE_MESSAGES", "FAILED"),
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { messageIds: JSON.stringify(messageIds) }
+        },
+        error
+      );
     }
   }
 };
@@ -1514,14 +1583,19 @@ function transformScoreRow(row) {
   return transformScoreRow$1(row);
 }
 var ScoresStorageCloudflare = class extends ScoresStorage {
-  operations;
-  constructor({ operations }) {
+  #db;
+  constructor(config) {
     super();
-    this.operations = operations;
+    this.#db = new CloudflareKVDB(resolveCloudflareConfig(config));
+  }
+  async init() {
+  }
+  async dangerouslyClearAll() {
+    await this.#db.clearTable({ tableName: TABLE_SCORERS });
   }
   async getScoreById({ id }) {
     try {
-      const score = await this.operations.getKV(TABLE_SCORERS, id);
+      const score = await this.#db.getKV(TABLE_SCORERS, id);
       if (!score) {
         return null;
       }
@@ -1536,8 +1610,8 @@ var ScoresStorageCloudflare = class extends ScoresStorage {
         },
         error
       );
-      this.logger.trackException(mastraError);
-      this.logger.error(mastraError.toString());
+      this.logger?.trackException(mastraError);
+      this.logger?.error(mastraError.toString());
       return null;
     }
   }
@@ -1552,7 +1626,7 @@ var ScoresStorageCloudflare = class extends ScoresStorage {
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.USER,
           details: {
-            scorer: score.scorer?.id ?? "unknown",
+            scorer: typeof score.scorer?.id === "string" ? score.scorer.id : String(score.scorer?.id ?? "unknown"),
             entityId: score.entityId ?? "unknown",
             entityType: score.entityType ?? "unknown",
             traceId: score.traceId ?? "",
@@ -1580,7 +1654,7 @@ var ScoresStorageCloudflare = class extends ScoresStorage {
       serializedRecord.id = id;
       serializedRecord.createdAt = now.toISOString();
       serializedRecord.updatedAt = now.toISOString();
-      await this.operations.putKV({
+      await this.#db.putKV({
         tableName: TABLE_SCORERS,
         key: id,
         value: serializedRecord
@@ -1596,8 +1670,8 @@ var ScoresStorageCloudflare = class extends ScoresStorage {
         },
         error
       );
-      this.logger.trackException(mastraError);
-      this.logger.error(mastraError.toString());
+      this.logger?.trackException(mastraError);
+      this.logger?.error(mastraError.toString());
       throw mastraError;
     }
   }
@@ -1609,10 +1683,10 @@ var ScoresStorageCloudflare = class extends ScoresStorage {
     pagination
   }) {
     try {
-      const keys = await this.operations.listKV(TABLE_SCORERS);
+      const keys = await this.#db.listKV(TABLE_SCORERS);
       const scores = [];
       for (const { name: key } of keys) {
-        const score = await this.operations.getKV(TABLE_SCORERS, key);
+        const score = await this.#db.getKV(TABLE_SCORERS, key);
         if (entityId && score.entityId !== entityId) {
           continue;
         }
@@ -1666,10 +1740,10 @@ var ScoresStorageCloudflare = class extends ScoresStorage {
     pagination
   }) {
     try {
-      const keys = await this.operations.listKV(TABLE_SCORERS);
+      const keys = await this.#db.listKV(TABLE_SCORERS);
       const scores = [];
       for (const { name: key } of keys) {
-        const score = await this.operations.getKV(TABLE_SCORERS, key);
+        const score = await this.#db.getKV(TABLE_SCORERS, key);
         if (score && score.runId === runId) {
           scores.push(transformScoreRow(score));
         }
@@ -1704,8 +1778,8 @@ var ScoresStorageCloudflare = class extends ScoresStorage {
         },
         error
       );
-      this.logger.trackException(mastraError);
-      this.logger.error(mastraError.toString());
+      this.logger?.trackException(mastraError);
+      this.logger?.error(mastraError.toString());
       return { pagination: { total: 0, page: 0, perPage: 100, hasMore: false }, scores: [] };
     }
   }
@@ -1715,10 +1789,10 @@ var ScoresStorageCloudflare = class extends ScoresStorage {
     pagination
   }) {
     try {
-      const keys = await this.operations.listKV(TABLE_SCORERS);
+      const keys = await this.#db.listKV(TABLE_SCORERS);
       const scores = [];
       for (const { name: key } of keys) {
-        const score = await this.operations.getKV(TABLE_SCORERS, key);
+        const score = await this.#db.getKV(TABLE_SCORERS, key);
         if (score && score.entityId === entityId && score.entityType === entityType) {
           scores.push(transformScoreRow(score));
         }
@@ -1753,8 +1827,8 @@ var ScoresStorageCloudflare = class extends ScoresStorage {
         },
         error
       );
-      this.logger.trackException(mastraError);
-      this.logger.error(mastraError.toString());
+      this.logger?.trackException(mastraError);
+      this.logger?.error(mastraError.toString());
       return { pagination: { total: 0, page: 0, perPage: 100, hasMore: false }, scores: [] };
     }
   }
@@ -1764,10 +1838,10 @@ var ScoresStorageCloudflare = class extends ScoresStorage {
     pagination
   }) {
     try {
-      const keys = await this.operations.listKV(TABLE_SCORERS);
+      const keys = await this.#db.listKV(TABLE_SCORERS);
       const scores = [];
       for (const { name: key } of keys) {
-        const score = await this.operations.getKV(TABLE_SCORERS, key);
+        const score = await this.#db.getKV(TABLE_SCORERS, key);
         if (score && score.traceId === traceId && score.spanId === spanId) {
           scores.push(transformScoreRow(score));
         }
@@ -1809,10 +1883,15 @@ var ScoresStorageCloudflare = class extends ScoresStorage {
   }
 };
 var WorkflowsStorageCloudflare = class extends WorkflowsStorage {
-  operations;
-  constructor({ operations }) {
+  #db;
+  constructor(config) {
     super();
-    this.operations = operations;
+    this.#db = new CloudflareKVDB(resolveCloudflareConfig(config));
+  }
+  async init() {
+  }
+  async dangerouslyClearAll() {
+    await this.#db.clearTable({ tableName: TABLE_WORKFLOW_SNAPSHOT });
   }
   validateWorkflowParams(params) {
     const { workflowName, runId } = params;
@@ -1838,17 +1917,20 @@ var WorkflowsStorageCloudflare = class extends WorkflowsStorage {
   }
   async persistWorkflowSnapshot(params) {
     try {
-      const { workflowName, runId, resourceId, snapshot } = params;
-      await this.operations.putKV({
+      const { workflowName, runId, resourceId, snapshot, createdAt, updatedAt } = params;
+      const now = /* @__PURE__ */ new Date();
+      const existingKey = this.#db.getKey(TABLE_WORKFLOW_SNAPSHOT, { workflow_name: workflowName, run_id: runId });
+      const existing = await this.#db.getKV(TABLE_WORKFLOW_SNAPSHOT, existingKey);
+      await this.#db.putKV({
         tableName: TABLE_WORKFLOW_SNAPSHOT,
-        key: this.operations.getKey(TABLE_WORKFLOW_SNAPSHOT, { workflow_name: workflowName, run_id: runId }),
+        key: existingKey,
         value: {
           workflow_name: workflowName,
           run_id: runId,
           resourceId,
           snapshot: JSON.stringify(snapshot),
-          createdAt: /* @__PURE__ */ new Date(),
-          updatedAt: /* @__PURE__ */ new Date()
+          createdAt: existing?.createdAt ?? createdAt ?? now,
+          updatedAt: updatedAt ?? now
         }
       });
     } catch (error) {
@@ -1871,8 +1953,8 @@ var WorkflowsStorageCloudflare = class extends WorkflowsStorage {
     try {
       this.validateWorkflowParams(params);
       const { workflowName, runId } = params;
-      const key = this.operations.getKey(TABLE_WORKFLOW_SNAPSHOT, { workflow_name: workflowName, run_id: runId });
-      const data = await this.operations.getKV(TABLE_WORKFLOW_SNAPSHOT, key);
+      const key = this.#db.getKey(TABLE_WORKFLOW_SNAPSHOT, { workflow_name: workflowName, run_id: runId });
+      const data = await this.#db.getKV(TABLE_WORKFLOW_SNAPSHOT, key);
       if (!data) return null;
       const snapshotData = typeof data.snapshot === "string" ? JSON.parse(data.snapshot) : data.snapshot;
       return snapshotData;
@@ -1918,7 +2000,7 @@ var WorkflowsStorageCloudflare = class extends WorkflowsStorage {
     runId,
     resourceId
   }) {
-    const prefix = this.operations.namespacePrefix ? `${this.operations.namespacePrefix}:` : "";
+    const prefix = this.#db.namespacePrefix ? `${this.#db.namespacePrefix}:` : "";
     let key = `${prefix}${TABLE_WORKFLOW_SNAPSHOT}`;
     if (workflowName) key += `:${workflowName}`;
     if (runId) key += `:${runId}`;
@@ -1949,7 +2031,7 @@ var WorkflowsStorageCloudflare = class extends WorkflowsStorage {
       const normalizedPerPage = normalizePerPage(perPage, 20);
       const offset = page * normalizedPerPage;
       const prefix = this.buildWorkflowSnapshotPrefix({ workflowName });
-      const keyObjs = await this.operations.listKV(TABLE_WORKFLOW_SNAPSHOT, { prefix });
+      const keyObjs = await this.#db.listKV(TABLE_WORKFLOW_SNAPSHOT, { prefix });
       const runs = [];
       for (const { name: key } of keyObjs) {
         const parts = key.split(":");
@@ -1958,21 +2040,20 @@ var WorkflowsStorageCloudflare = class extends WorkflowsStorage {
         const wfName = parts[idx + 1];
         const keyResourceId = parts.length > idx + 3 ? parts[idx + 3] : void 0;
         if (workflowName && wfName !== workflowName) continue;
-        if (resourceId && keyResourceId !== resourceId) continue;
-        const data = await this.operations.getKV(TABLE_WORKFLOW_SNAPSHOT, key);
+        const data = await this.#db.getKV(TABLE_WORKFLOW_SNAPSHOT, key);
         if (!data) continue;
         try {
-          if (resourceId && !keyResourceId) continue;
+          const effectiveResourceId = keyResourceId || data.resourceId;
+          if (resourceId && effectiveResourceId !== resourceId) continue;
           const snapshotData = typeof data.snapshot === "string" ? JSON.parse(data.snapshot) : data.snapshot;
           if (status && snapshotData.status !== status) continue;
           const createdAt = ensureDate(data.createdAt);
           if (fromDate && createdAt && createdAt < fromDate) continue;
           if (toDate && createdAt && createdAt > toDate) continue;
-          const resourceIdToUse = keyResourceId || data.resourceId;
           const run = this.parseWorkflowRun({
             ...data,
             workflow_name: wfName,
-            resourceId: resourceIdToUse,
+            resourceId: effectiveResourceId,
             snapshot: snapshotData
           });
           runs.push(run);
@@ -2013,7 +2094,7 @@ var WorkflowsStorageCloudflare = class extends WorkflowsStorage {
         throw new Error("runId, workflowName, are required");
       }
       const prefix = this.buildWorkflowSnapshotPrefix({ workflowName, runId });
-      const keyObjs = await this.operations.listKV(TABLE_WORKFLOW_SNAPSHOT, { prefix });
+      const keyObjs = await this.#db.listKV(TABLE_WORKFLOW_SNAPSHOT, { prefix });
       if (!keyObjs.length) return null;
       const exactKey = keyObjs.find((k) => {
         const parts = k.name.split(":");
@@ -2024,7 +2105,7 @@ var WorkflowsStorageCloudflare = class extends WorkflowsStorage {
         return wfName === workflowName && rId === runId;
       });
       if (!exactKey) return null;
-      const data = await this.operations.getKV(TABLE_WORKFLOW_SNAPSHOT, exactKey.name);
+      const data = await this.#db.getKV(TABLE_WORKFLOW_SNAPSHOT, exactKey.name);
       if (!data) return null;
       const snapshotData = typeof data.snapshot === "string" ? JSON.parse(data.snapshot) : data.snapshot;
       return this.parseWorkflowRun({ ...data, snapshot: snapshotData });
@@ -2051,8 +2132,8 @@ var WorkflowsStorageCloudflare = class extends WorkflowsStorage {
       if (!runId || !workflowName) {
         throw new Error("runId and workflowName are required");
       }
-      const key = this.operations.getKey(TABLE_WORKFLOW_SNAPSHOT, { workflow_name: workflowName, run_id: runId });
-      await this.operations.deleteKV(TABLE_WORKFLOW_SNAPSHOT, key);
+      const key = this.#db.getKey(TABLE_WORKFLOW_SNAPSHOT, { workflow_name: workflowName, run_id: runId });
+      await this.#db.deleteKV(TABLE_WORKFLOW_SNAPSHOT, key);
     } catch (error) {
       throw new MastraError(
         {
@@ -2108,20 +2189,36 @@ var CloudflareStore = class extends MastraStorage {
     }
   }
   get supports() {
-    const supports = super.supports;
-    supports.listScoresBySpan = true;
-    supports.resourceWorkingMemory = true;
-    supports.selectByIncludeResourceScope = true;
-    return supports;
+    return {
+      selectByIncludeResourceScope: true,
+      resourceWorkingMemory: true,
+      hasColumn: false,
+      createTable: false,
+      deleteMessages: true,
+      observability: false,
+      indexManagement: false,
+      listScoresBySpan: true,
+      agents: false
+    };
   }
   constructor(config) {
     super({ id: config.id, name: "Cloudflare", disableInit: config.disableInit });
     try {
+      let workflows;
+      let memory;
+      let scores;
       if (isWorkersConfig(config)) {
         this.validateWorkersConfig(config);
         this.bindings = config.bindings;
         this.namespacePrefix = config.keyPrefix?.trim() || "";
         this.logger.info("Using Cloudflare KV Workers Binding API");
+        const domainConfig = {
+          bindings: this.bindings,
+          keyPrefix: this.namespacePrefix
+        };
+        workflows = new WorkflowsStorageCloudflare(domainConfig);
+        memory = new MemoryStorageCloudflare(domainConfig);
+        scores = new ScoresStorageCloudflare(domainConfig);
       } else {
         this.validateRestConfig(config);
         this.accountId = config.accountId.trim();
@@ -2130,24 +2227,16 @@ var CloudflareStore = class extends MastraStorage {
           apiToken: config.apiToken.trim()
         });
         this.logger.info("Using Cloudflare KV REST API");
+        const domainConfig = {
+          client: this.client,
+          accountId: this.accountId,
+          namespacePrefix: this.namespacePrefix
+        };
+        workflows = new WorkflowsStorageCloudflare(domainConfig);
+        memory = new MemoryStorageCloudflare(domainConfig);
+        scores = new ScoresStorageCloudflare(domainConfig);
       }
-      const operations = new StoreOperationsCloudflare({
-        accountId: this.accountId,
-        client: this.client,
-        namespacePrefix: this.namespacePrefix,
-        bindings: this.bindings
-      });
-      const workflows = new WorkflowsStorageCloudflare({
-        operations
-      });
-      const memory = new MemoryStorageCloudflare({
-        operations
-      });
-      const scores = new ScoresStorageCloudflare({
-        operations
-      });
       this.stores = {
-        operations,
         workflows,
         memory,
         scores
@@ -2163,156 +2252,6 @@ var CloudflareStore = class extends MastraStorage {
       );
     }
   }
-  async createTable({
-    tableName,
-    schema
-  }) {
-    return this.stores.operations.createTable({ tableName, schema });
-  }
-  async alterTable(_args) {
-    return this.stores.operations.alterTable(_args);
-  }
-  async clearTable({ tableName }) {
-    return this.stores.operations.clearTable({ tableName });
-  }
-  async dropTable({ tableName }) {
-    return this.stores.operations.dropTable({ tableName });
-  }
-  async insert({
-    tableName,
-    record
-  }) {
-    return this.stores.operations.insert({ tableName, record });
-  }
-  async load({ tableName, keys }) {
-    return this.stores.operations.load({ tableName, keys });
-  }
-  async getThreadById({ threadId }) {
-    return this.stores.memory.getThreadById({ threadId });
-  }
-  async saveThread({ thread }) {
-    return this.stores.memory.saveThread({ thread });
-  }
-  async updateThread({
-    id,
-    title,
-    metadata
-  }) {
-    return this.stores.memory.updateThread({ id, title, metadata });
-  }
-  async deleteThread({ threadId }) {
-    return this.stores.memory.deleteThread({ threadId });
-  }
-  async saveMessages(args) {
-    return this.stores.memory.saveMessages(args);
-  }
-  async updateWorkflowResults({
-    workflowName,
-    runId,
-    stepId,
-    result,
-    requestContext
-  }) {
-    return this.stores.workflows.updateWorkflowResults({ workflowName, runId, stepId, result, requestContext });
-  }
-  async updateWorkflowState({
-    workflowName,
-    runId,
-    opts
-  }) {
-    return this.stores.workflows.updateWorkflowState({ workflowName, runId, opts });
-  }
-  async listMessagesById({ messageIds }) {
-    return this.stores.memory.listMessagesById({ messageIds });
-  }
-  async persistWorkflowSnapshot(params) {
-    return this.stores.workflows.persistWorkflowSnapshot(params);
-  }
-  async loadWorkflowSnapshot(params) {
-    return this.stores.workflows.loadWorkflowSnapshot(params);
-  }
-  async batchInsert(input) {
-    return this.stores.operations.batchInsert(input);
-  }
-  async listWorkflowRuns({
-    workflowName,
-    perPage = 20,
-    page = 0,
-    resourceId,
-    fromDate,
-    toDate,
-    status
-  } = {}) {
-    return this.stores.workflows.listWorkflowRuns({
-      workflowName,
-      perPage,
-      page,
-      resourceId,
-      fromDate,
-      toDate,
-      status
-    });
-  }
-  async getWorkflowRunById({
-    runId,
-    workflowName
-  }) {
-    return this.stores.workflows.getWorkflowRunById({ runId, workflowName });
-  }
-  async deleteWorkflowRunById({ runId, workflowName }) {
-    return this.stores.workflows.deleteWorkflowRunById({ runId, workflowName });
-  }
-  async updateMessages(args) {
-    return this.stores.memory.updateMessages(args);
-  }
-  async getScoreById({ id }) {
-    return this.stores.scores.getScoreById({ id });
-  }
-  async saveScore(score) {
-    return this.stores.scores.saveScore(score);
-  }
-  async listScoresByRunId({
-    runId,
-    pagination
-  }) {
-    return this.stores.scores.listScoresByRunId({ runId, pagination });
-  }
-  async listScoresByEntityId({
-    entityId,
-    entityType,
-    pagination
-  }) {
-    return this.stores.scores.listScoresByEntityId({ entityId, entityType, pagination });
-  }
-  async listScoresByScorerId({
-    scorerId,
-    entityId,
-    entityType,
-    source,
-    pagination
-  }) {
-    return this.stores.scores.listScoresByScorerId({ scorerId, entityId, entityType, source, pagination });
-  }
-  async listScoresBySpan({
-    traceId,
-    spanId,
-    pagination
-  }) {
-    return this.stores.scores.listScoresBySpan({ traceId, spanId, pagination });
-  }
-  async getResourceById({ resourceId }) {
-    return this.stores.memory.getResourceById({ resourceId });
-  }
-  async saveResource({ resource }) {
-    return this.stores.memory.saveResource({ resource });
-  }
-  async updateResource({
-    resourceId,
-    workingMemory,
-    metadata
-  }) {
-    return this.stores.memory.updateResource({ resourceId, workingMemory, metadata });
-  }
   async close() {
   }
 };