@mastra/cloudflare-d1 0.11.1-alpha.0 → 0.11.1-alpha.2

package/dist/index.cjs

@@ -1,6 +1,7 @@
 'use strict';
 
 var agent = require('@mastra/core/agent');
+var error = require('@mastra/core/error');
 var storage = require('@mastra/core/storage');
 var Cloudflare = require('cloudflare');
 var utils = require('@mastra/core/utils');
@@ -256,27 +257,39 @@ var D1Store = class extends storage.MastraStorage {
    * @param config Configuration for D1 access (either REST API or Workers Binding API)
    */
   constructor(config) {
-    super({ name: "D1" });
-    if (config.tablePrefix && !/^[a-zA-Z0-9_]*$/.test(config.tablePrefix)) {
-      throw new Error("Invalid tablePrefix: only letters, numbers, and underscores are allowed.");
-    }
-    this.tablePrefix = config.tablePrefix || "";
-    if ("binding" in config) {
-      if (!config.binding) {
-        throw new Error("D1 binding is required when using Workers Binding API");
+    try {
+      super({ name: "D1" });
+      if (config.tablePrefix && !/^[a-zA-Z0-9_]*$/.test(config.tablePrefix)) {
+        throw new Error("Invalid tablePrefix: only letters, numbers, and underscores are allowed.");
       }
-      this.binding = config.binding;
-      this.logger.info("Using D1 Workers Binding API");
-    } else {
-      if (!config.accountId || !config.databaseId || !config.apiToken) {
-        throw new Error("accountId, databaseId, and apiToken are required when using REST API");
+      this.tablePrefix = config.tablePrefix || "";
+      if ("binding" in config) {
+        if (!config.binding) {
+          throw new Error("D1 binding is required when using Workers Binding API");
+        }
+        this.binding = config.binding;
+        this.logger.info("Using D1 Workers Binding API");
+      } else {
+        if (!config.accountId || !config.databaseId || !config.apiToken) {
+          throw new Error("accountId, databaseId, and apiToken are required when using REST API");
+        }
+        this.accountId = config.accountId;
+        this.databaseId = config.databaseId;
+        this.client = new Cloudflare__default.default({
+          apiToken: config.apiToken
+        });
+        this.logger.info("Using D1 REST API");
       }
-      this.accountId = config.accountId;
-      this.databaseId = config.databaseId;
-      this.client = new Cloudflare__default.default({
-        apiToken: config.apiToken
-      });
-      this.logger.info("Using D1 REST API");
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "CLOUDFLARE_D1_STORAGE_INITIALIZATION_ERROR",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.SYSTEM,
+          text: "Error initializing D1Store"
+        },
+        error$1
+      );
     }
   }
   // Helper method to get the full table name with prefix
@@ -468,16 +481,25 @@ var D1Store = class extends storage.MastraStorage {
     if (tableName === storage.TABLE_WORKFLOW_SNAPSHOT) {
       tableConstraints.push("UNIQUE (workflow_name, run_id)");
     }
-    const query = createSqlBuilder().createTable(fullTableName, columnDefinitions, tableConstraints);
-    const { sql, params } = query.build();
     try {
+      const query = createSqlBuilder().createTable(fullTableName, columnDefinitions, tableConstraints);
+      const { sql, params } = query.build();
       await this.executeQuery({ sql, params });
       this.logger.debug(`Created table ${fullTableName}`);
-    } catch (error) {
+    } catch (error$1) {
       this.logger.error(`Error creating table ${fullTableName}:`, {
-        message: error instanceof Error ? error.message : String(error)
+        message: error$1 instanceof Error ? error$1.message : String(error$1)
       });
-      throw new Error(`Failed to create table ${fullTableName}: ${error}`);
+      throw new error.MastraError(
+        {
+          id: "CLOUDFLARE_D1_STORAGE_CREATE_TABLE_ERROR",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY,
+          text: `Failed to create table ${fullTableName}: ${error$1 instanceof Error ? error$1.message : String(error$1)}`,
+          details: { tableName }
+        },
+        error$1
+      );
     }
   }
   /**
@@ -506,11 +528,17 @@ var D1Store = class extends storage.MastraStorage {
           this.logger.debug(`Added column ${columnName} to table ${fullTableName}`);
         }
       }
-    } catch (error) {
-      this.logger.error(`Error altering table ${fullTableName}:`, {
-        message: error instanceof Error ? error.message : String(error)
-      });
-      throw new Error(`Failed to alter table ${fullTableName}: ${error}`);
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "CLOUDFLARE_D1_STORAGE_ALTER_TABLE_ERROR",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY,
+          text: `Failed to alter table ${fullTableName}: ${error$1 instanceof Error ? error$1.message : String(error$1)}`,
+          details: { tableName }
+        },
+        error$1
+      );
     }
   }
   async clearTable({ tableName }) {
@@ -520,11 +548,17 @@ var D1Store = class extends storage.MastraStorage {
       const { sql, params } = query.build();
       await this.executeQuery({ sql, params });
       this.logger.debug(`Cleared table ${fullTableName}`);
-    } catch (error) {
-      this.logger.error(`Error clearing table ${fullTableName}:`, {
-        message: error instanceof Error ? error.message : String(error)
-      });
-      throw new Error(`Failed to clear table ${fullTableName}: ${error}`);
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "CLOUDFLARE_D1_STORAGE_CLEAR_TABLE_ERROR",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY,
+          text: `Failed to clear table ${fullTableName}: ${error$1 instanceof Error ? error$1.message : String(error$1)}`,
+          details: { tableName }
+        },
+        error$1
+      );
     }
   }
   async processRecord(record) {
@@ -543,10 +577,17 @@ var D1Store = class extends storage.MastraStorage {
     const { sql, params } = query.build();
     try {
       await this.executeQuery({ sql, params });
-    } catch (error) {
-      const message = error instanceof Error ? error.message : String(error);
-      this.logger.error(`Error inserting into ${fullTableName}:`, { message });
-      throw new Error(`Failed to insert into ${fullTableName}: ${error}`);
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "CLOUDFLARE_D1_STORAGE_INSERT_ERROR",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY,
+          text: `Failed to insert into ${fullTableName}: ${error$1 instanceof Error ? error$1.message : String(error$1)}`,
+          details: { tableName }
+        },
+        error$1
+      );
     }
   }
   async load({ tableName, keys }) {
@@ -571,11 +612,17 @@ var D1Store = class extends storage.MastraStorage {
         processedResult[key] = this.deserializeValue(value);
       }
       return processedResult;
-    } catch (error) {
-      this.logger.error(`Error loading from ${fullTableName}:`, {
-        message: error instanceof Error ? error.message : String(error)
-      });
-      return null;
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "CLOUDFLARE_D1_STORAGE_LOAD_ERROR",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY,
+          text: `Failed to load from ${fullTableName}: ${error$1 instanceof Error ? error$1.message : String(error$1)}`,
+          details: { tableName }
+        },
+        error$1
+      );
     }
   }
   async getThreadById({ threadId }) {
@@ -591,10 +638,19 @@ var D1Store = class extends storage.MastraStorage {
         updatedAt: this.ensureDate(thread.updatedAt),
         metadata: typeof thread.metadata === "string" ? JSON.parse(thread.metadata || "{}") : thread.metadata || {}
       };
-    } catch (error) {
-      this.logger.error(`Error processing thread ${threadId}:`, {
-        message: error instanceof Error ? error.message : String(error)
-      });
+    } catch (error$1) {
+      const mastraError = new error.MastraError(
+        {
+          id: "CLOUDFLARE_D1_STORAGE_GET_THREAD_BY_ID_ERROR",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY,
+          text: `Error processing thread ${threadId}: ${error$1 instanceof Error ? error$1.message : String(error$1)}`,
+          details: { threadId }
+        },
+        error$1
+      );
+      this.logger?.error(mastraError.toString());
+      this.logger?.trackException(mastraError);
       return null;
     }
   }
@@ -613,10 +669,19 @@ var D1Store = class extends storage.MastraStorage {
         updatedAt: this.ensureDate(thread.updatedAt),
         metadata: typeof thread.metadata === "string" ? JSON.parse(thread.metadata || "{}") : thread.metadata || {}
       }));
-    } catch (error) {
-      this.logger.error(`Error getting threads by resourceId ${resourceId}:`, {
-        message: error instanceof Error ? error.message : String(error)
-      });
+    } catch (error$1) {
+      const mastraError = new error.MastraError(
+        {
+          id: "CLOUDFLARE_D1_STORAGE_GET_THREADS_BY_RESOURCE_ID_ERROR",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY,
+          text: `Error getting threads by resourceId ${resourceId}: ${error$1 instanceof Error ? error$1.message : String(error$1)}`,
+          details: { resourceId }
+        },
+        error$1
+      );
+      this.logger?.error(mastraError.toString());
+      this.logger?.trackException(mastraError);
       return [];
     }
   }
@@ -629,19 +694,41 @@ var D1Store = class extends storage.MastraStorage {
       updatedAt: this.ensureDate(row.updatedAt),
       metadata: typeof row.metadata === "string" ? JSON.parse(row.metadata || "{}") : row.metadata || {}
     });
-    const countQuery = createSqlBuilder().count().from(fullTableName).where("resourceId = ?", resourceId);
-    const countResult = await this.executeQuery(countQuery.build());
-    const total = Number(countResult?.[0]?.count ?? 0);
-    const selectQuery = createSqlBuilder().select("*").from(fullTableName).where("resourceId = ?", resourceId).orderBy("createdAt", "DESC").limit(perPage).offset(page * perPage);
-    const results = await this.executeQuery(selectQuery.build());
-    const threads = results.map(mapRowToStorageThreadType);
-    return {
-      threads,
-      total,
-      page,
-      perPage,
-      hasMore: page * perPage + threads.length < total
-    };
+    try {
+      const countQuery = createSqlBuilder().count().from(fullTableName).where("resourceId = ?", resourceId);
+      const countResult = await this.executeQuery(countQuery.build());
+      const total = Number(countResult?.[0]?.count ?? 0);
+      const selectQuery = createSqlBuilder().select("*").from(fullTableName).where("resourceId = ?", resourceId).orderBy("createdAt", "DESC").limit(perPage).offset(page * perPage);
+      const results = await this.executeQuery(selectQuery.build());
+      const threads = results.map(mapRowToStorageThreadType);
+      return {
+        threads,
+        total,
+        page,
+        perPage,
+        hasMore: page * perPage + threads.length < total
+      };
+    } catch (error$1) {
+      const mastraError = new error.MastraError(
+        {
+          id: "CLOUDFLARE_D1_STORAGE_GET_THREADS_BY_RESOURCE_ID_PAGINATED_ERROR",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY,
+          text: `Error getting threads by resourceId ${resourceId}: ${error$1 instanceof Error ? error$1.message : String(error$1)}`,
+          details: { resourceId }
+        },
+        error$1
+      );
+      this.logger?.error(mastraError.toString());
+      this.logger?.trackException(mastraError);
+      return {
+        threads: [],
+        total: 0,
+        page,
+        perPage,
+        hasMore: false
+      };
+    }
   }
   async saveThread({ thread }) {
     const fullTableName = this.getTableName(storage.TABLE_THREADS);
@@ -668,10 +755,17 @@ var D1Store = class extends storage.MastraStorage {
     try {
       await this.executeQuery({ sql, params });
       return thread;
-    } catch (error) {
-      const message = error instanceof Error ? error.message : String(error);
-      this.logger.error(`Error saving thread to ${fullTableName}:`, { message });
-      throw error;
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "CLOUDFLARE_D1_STORAGE_SAVE_THREAD_ERROR",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY,
+          text: `Failed to save thread to ${fullTableName}: ${error$1 instanceof Error ? error$1.message : String(error$1)}`,
+          details: { threadId: thread.id }
+        },
+        error$1
+      );
     }
   }
   async updateThread({
@@ -680,19 +774,19 @@ var D1Store = class extends storage.MastraStorage {
     metadata
   }) {
     const thread = await this.getThreadById({ threadId: id });
-    if (!thread) {
-      throw new Error(`Thread ${id} not found`);
-    }
-    const fullTableName = this.getTableName(storage.TABLE_THREADS);
-    const mergedMetadata = {
-      ...typeof thread.metadata === "string" ? JSON.parse(thread.metadata) : thread.metadata,
-      ...metadata
-    };
-    const columns = ["title", "metadata", "updatedAt"];
-    const values = [title, JSON.stringify(mergedMetadata), (/* @__PURE__ */ new Date()).toISOString()];
-    const query = createSqlBuilder().update(fullTableName, columns, values).where("id = ?", id);
-    const { sql, params } = query.build();
     try {
+      if (!thread) {
+        throw new Error(`Thread ${id} not found`);
+      }
+      const fullTableName = this.getTableName(storage.TABLE_THREADS);
+      const mergedMetadata = {
+        ...typeof thread.metadata === "string" ? JSON.parse(thread.metadata) : thread.metadata,
+        ...metadata
+      };
+      const columns = ["title", "metadata", "updatedAt"];
+      const values = [title, JSON.stringify(mergedMetadata), (/* @__PURE__ */ new Date()).toISOString()];
+      const query = createSqlBuilder().update(fullTableName, columns, values).where("id = ?", id);
+      const { sql, params } = query.build();
       await this.executeQuery({ sql, params });
       return {
         ...thread,
@@ -703,10 +797,17 @@ var D1Store = class extends storage.MastraStorage {
         },
         updatedAt: /* @__PURE__ */ new Date()
       };
-    } catch (error) {
-      const message = error instanceof Error ? error.message : String(error);
-      this.logger.error("Error updating thread:", { message });
-      throw error;
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "CLOUDFLARE_D1_STORAGE_UPDATE_THREAD_ERROR",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY,
+          text: `Failed to update thread ${id}: ${error$1 instanceof Error ? error$1.message : String(error$1)}`,
+          details: { threadId: id }
+        },
+        error$1
+      );
     }
   }
   async deleteThread({ threadId }) {
@@ -719,11 +820,17 @@ var D1Store = class extends storage.MastraStorage {
       const deleteMessagesQuery = createSqlBuilder().delete(messagesTableName).where("thread_id = ?", threadId);
       const { sql: messagesSql, params: messagesParams } = deleteMessagesQuery.build();
       await this.executeQuery({ sql: messagesSql, params: messagesParams });
-    } catch (error) {
-      this.logger.error(`Error deleting thread ${threadId}:`, {
-        message: error instanceof Error ? error.message : String(error)
-      });
-      throw new Error(`Failed to delete thread ${threadId}: ${error}`);
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "CLOUDFLARE_D1_STORAGE_DELETE_THREAD_ERROR",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY,
+          text: `Failed to delete thread ${threadId}: ${error$1 instanceof Error ? error$1.message : String(error$1)}`,
+          details: { threadId }
+        },
+        error$1
+      );
     }
   }
   async saveMessages(args) {
@@ -755,7 +862,7 @@ var D1Store = class extends storage.MastraStorage {
         };
       });
       await Promise.all([
-        this.batchInsert({
+        this.batchUpsert({
           tableName: storage.TABLE_MESSAGES,
           records: messagesToInsert
         }),
@@ -769,9 +876,16 @@ var D1Store = class extends storage.MastraStorage {
       const list = new agent.MessageList().add(messages, "memory");
       if (format === `v2`) return list.get.all.v2();
       return list.get.all.v1();
-    } catch (error) {
-      this.logger.error("Error saving messages:", { message: error instanceof Error ? error.message : String(error) });
-      throw error;
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "CLOUDFLARE_D1_STORAGE_SAVE_MESSAGES_ERROR",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY,
+          text: `Failed to save messages: ${error$1 instanceof Error ? error$1.message : String(error$1)}`
+        },
+        error$1
+      );
     }
   }
   async _getIncludedMessages(threadId, selectBy) {
@@ -826,7 +940,7 @@ var D1Store = class extends storage.MastraStorage {
     format
   }) {
     const fullTableName = this.getTableName(storage.TABLE_MESSAGES);
-    const limit = typeof selectBy?.last === "number" ? selectBy.last : 40;
+    const limit = this.resolveMessageLimit({ last: selectBy?.last, defaultLimit: 40 });
     const include = selectBy?.include || [];
     const messages = [];
     try {
@@ -862,12 +976,20 @@ var D1Store = class extends storage.MastraStorage {
       const list = new agent.MessageList().add(processedMessages, "memory");
       if (format === `v2`) return list.get.all.v2();
       return list.get.all.v1();
-    } catch (error) {
-      this.logger.error("Error retrieving messages for thread", {
-        threadId,
-        message: error instanceof Error ? error.message : String(error)
-      });
-      return [];
+    } catch (error$1) {
+      const mastraError = new error.MastraError(
+        {
+          id: "CLOUDFLARE_D1_STORAGE_GET_MESSAGES_ERROR",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY,
+          text: `Failed to retrieve messages for thread ${threadId}: ${error$1 instanceof Error ? error$1.message : String(error$1)}`,
+          details: { threadId }
+        },
+        error$1
+      );
+      this.logger?.error(mastraError.toString());
+      this.logger?.trackException(mastraError);
+      throw mastraError;
     }
   }
   async getMessagesPaginated({
@@ -879,37 +1001,59 @@ var D1Store = class extends storage.MastraStorage {
     const { start: fromDate, end: toDate } = dateRange || {};
     const fullTableName = this.getTableName(storage.TABLE_MESSAGES);
     const messages = [];
-    if (selectBy?.include?.length) {
-      const includeResult = await this._getIncludedMessages(threadId, selectBy);
-      if (Array.isArray(includeResult)) messages.push(...includeResult);
-    }
-    const countQuery = createSqlBuilder().count().from(fullTableName).where("thread_id = ?", threadId);
-    if (fromDate) {
-      countQuery.andWhere("createdAt >= ?", this.serializeDate(fromDate));
-    }
-    if (toDate) {
-      countQuery.andWhere("createdAt <= ?", this.serializeDate(toDate));
-    }
-    const countResult = await this.executeQuery(countQuery.build());
-    const total = Number(countResult[0]?.count ?? 0);
-    const query = createSqlBuilder().select(["id", "content", "role", "type", "createdAt", "thread_id AS threadId"]).from(fullTableName).where("thread_id = ?", threadId);
-    if (fromDate) {
-      query.andWhere("createdAt >= ?", this.serializeDate(fromDate));
-    }
-    if (toDate) {
-      query.andWhere("createdAt <= ?", this.serializeDate(toDate));
+    try {
+      if (selectBy?.include?.length) {
+        const includeResult = await this._getIncludedMessages(threadId, selectBy);
+        if (Array.isArray(includeResult)) messages.push(...includeResult);
+      }
+      const countQuery = createSqlBuilder().count().from(fullTableName).where("thread_id = ?", threadId);
+      if (fromDate) {
+        countQuery.andWhere("createdAt >= ?", this.serializeDate(fromDate));
+      }
+      if (toDate) {
+        countQuery.andWhere("createdAt <= ?", this.serializeDate(toDate));
+      }
+      const countResult = await this.executeQuery(countQuery.build());
+      const total = Number(countResult[0]?.count ?? 0);
+      const query = createSqlBuilder().select(["id", "content", "role", "type", "createdAt", "thread_id AS threadId"]).from(fullTableName).where("thread_id = ?", threadId);
+      if (fromDate) {
+        query.andWhere("createdAt >= ?", this.serializeDate(fromDate));
+      }
+      if (toDate) {
+        query.andWhere("createdAt <= ?", this.serializeDate(toDate));
+      }
+      query.orderBy("createdAt", "DESC").limit(perPage).offset(page * perPage);
+      const results = await this.executeQuery(query.build());
+      const list = new agent.MessageList().add(results, "memory");
+      messages.push(...format === `v2` ? list.get.all.v2() : list.get.all.v1());
+      return {
+        messages,
+        total,
+        page,
+        perPage,
+        hasMore: page * perPage + messages.length < total
+      };
+    } catch (error$1) {
+      const mastraError = new error.MastraError(
+        {
+          id: "CLOUDFLARE_D1_STORAGE_GET_MESSAGES_PAGINATED_ERROR",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY,
+          text: `Failed to retrieve messages for thread ${threadId}: ${error$1 instanceof Error ? error$1.message : String(error$1)}`,
+          details: { threadId }
+        },
+        error$1
+      );
+      this.logger?.error(mastraError.toString());
+      this.logger?.trackException(mastraError);
+      return {
+        messages: [],
+        total: 0,
+        page,
+        perPage,
+        hasMore: false
+      };
     }
-    query.orderBy("createdAt", "DESC").limit(perPage).offset(page * perPage);
-    const results = await this.executeQuery(query.build());
-    const list = new agent.MessageList().add(results, "memory");
-    messages.push(...format === `v2` ? list.get.all.v2() : list.get.all.v1());
-    return {
-      messages,
-      total,
-      page,
-      perPage,
-      hasMore: page * perPage + messages.length < total
-    };
   }
   async persistWorkflowSnapshot({
     workflowName,
@@ -945,24 +1089,43 @@ var D1Store = class extends storage.MastraStorage {
     const { sql, params } = query.build();
     try {
       await this.executeQuery({ sql, params });
-    } catch (error) {
-      this.logger.error("Error persisting workflow snapshot:", {
-        message: error instanceof Error ? error.message : String(error)
-      });
-      throw error;
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "CLOUDFLARE_D1_STORAGE_PERSIST_WORKFLOW_SNAPSHOT_ERROR",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY,
+          text: `Failed to persist workflow snapshot: ${error$1 instanceof Error ? error$1.message : String(error$1)}`,
+          details: { workflowName, runId }
+        },
+        error$1
+      );
     }
   }
   async loadWorkflowSnapshot(params) {
     const { workflowName, runId } = params;
     this.logger.debug("Loading workflow snapshot", { workflowName, runId });
-    const d = await this.load({
-      tableName: storage.TABLE_WORKFLOW_SNAPSHOT,
-      keys: {
-        workflow_name: workflowName,
-        run_id: runId
-      }
-    });
-    return d ? d.snapshot : null;
+    try {
+      const d = await this.load({
+        tableName: storage.TABLE_WORKFLOW_SNAPSHOT,
+        keys: {
+          workflow_name: workflowName,
+          run_id: runId
+        }
+      });
+      return d ? d.snapshot : null;
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "CLOUDFLARE_D1_STORAGE_LOAD_WORKFLOW_SNAPSHOT_ERROR",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY,
+          text: `Failed to load workflow snapshot: ${error$1 instanceof Error ? error$1.message : String(error$1)}`,
+          details: { workflowName, runId }
+        },
+        error$1
+      );
+    }
   }
   /**
    * Insert multiple records in a batch operation
@@ -996,11 +1159,72 @@ var D1Store = class extends storage.MastraStorage {
         );
       }
       this.logger.debug(`Successfully batch inserted ${records.length} records into ${tableName}`);
-    } catch (error) {
-      this.logger.error(`Error batch inserting into ${tableName}:`, {
-        message: error instanceof Error ? error.message : String(error)
-      });
-      throw new Error(`Failed to batch insert into ${tableName}: ${error}`);
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "CLOUDFLARE_D1_STORAGE_BATCH_INSERT_ERROR",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY,
+          text: `Failed to batch insert into ${tableName}: ${error$1 instanceof Error ? error$1.message : String(error$1)}`,
+          details: { tableName }
+        },
+        error$1
+      );
+    }
+  }
+  /**
+   * Upsert multiple records in a batch operation
+   * @param tableName The table to insert into
+   * @param records The records to insert
+   */
+  async batchUpsert({
+    tableName,
+    records
+  }) {
+    if (records.length === 0) return;
+    const fullTableName = this.getTableName(tableName);
+    try {
+      const batchSize = 50;
+      for (let i = 0; i < records.length; i += batchSize) {
+        const batch = records.slice(i, i + batchSize);
+        const recordsToInsert = batch;
+        if (recordsToInsert.length > 0) {
+          const firstRecord = recordsToInsert[0];
+          const columns = Object.keys(firstRecord || {});
+          for (const record of recordsToInsert) {
+            const values = columns.map((col) => {
+              if (!record) return null;
+              const value = typeof col === "string" ? record[col] : null;
+              return this.serializeValue(value);
+            });
+            const recordToUpsert = columns.reduce(
+              (acc, col) => {
+                if (col !== "createdAt") acc[col] = `excluded.${col}`;
+                return acc;
+              },
+              {}
+            );
+            const query = createSqlBuilder().insert(fullTableName, columns, values, ["id"], recordToUpsert);
+            const { sql, params } = query.build();
+            await this.executeQuery({ sql, params });
+          }
+        }
+        this.logger.debug(
+          `Processed batch ${Math.floor(i / batchSize) + 1} of ${Math.ceil(records.length / batchSize)}`
+        );
+      }
+      this.logger.debug(`Successfully batch upserted ${records.length} records into ${tableName}`);
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "CLOUDFLARE_D1_STORAGE_BATCH_UPSERT_ERROR",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY,
+          text: `Failed to batch upsert into ${tableName}: ${error$1 instanceof Error ? error$1.message : String(error$1)}`,
+          details: { tableName }
+        },
+        error$1
+      );
     }
   }
   /**
@@ -1048,8 +1272,22 @@ var D1Store = class extends storage.MastraStorage {
           other: this.deserializeValue(trace.other, "jsonb")
         })
       ) : [];
-    } catch (error) {
-      this.logger.error("Error getting traces:", { message: error instanceof Error ? error.message : String(error) });
+    } catch (error$1) {
+      const mastraError = new error.MastraError(
+        {
+          id: "CLOUDFLARE_D1_STORAGE_GET_TRACES_ERROR",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY,
+          text: `Failed to retrieve traces: ${error$1 instanceof Error ? error$1.message : String(error$1)}`,
+          details: {
+            name: name ?? "",
+            scope: scope ?? ""
+          }
+        },
+        error$1
+      );
+      this.logger?.error(mastraError.toString());
+      this.logger?.trackException(mastraError);
       return [];
     }
   }
@@ -1104,8 +1342,19 @@ var D1Store = class extends storage.MastraStorage {
         perPage,
         hasMore: page * perPage + traces.length < total
       };
-    } catch (error) {
-      this.logger.error("Error getting traces:", { message: error instanceof Error ? error.message : String(error) });
+    } catch (error$1) {
+      const mastraError = new error.MastraError(
+        {
+          id: "CLOUDFLARE_D1_STORAGE_GET_TRACES_ERROR",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY,
+          text: `Failed to retrieve traces: ${error$1 instanceof Error ? error$1.message : String(error$1)}`,
+          details: { name: name ?? "", scope: scope ?? "" }
+        },
+        error$1
+      );
+      this.logger?.error(mastraError.toString());
+      this.logger?.trackException(mastraError);
       return { traces: [], total: 0, page, perPage, hasMore: false };
     }
   }
@@ -1140,10 +1389,19 @@ var D1Store = class extends storage.MastraStorage {
           testInfo
         };
       }) : [];
-    } catch (error) {
-      this.logger.error(`Error getting evals for agent ${agentName}:`, {
-        message: error instanceof Error ? error.message : String(error)
-      });
+    } catch (error$1) {
+      const mastraError = new error.MastraError(
+        {
+          id: "CLOUDFLARE_D1_STORAGE_GET_EVALS_ERROR",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY,
+          text: `Failed to retrieve evals for agent ${agentName}: ${error$1 instanceof Error ? error$1.message : String(error$1)}`,
+          details: { agentName }
+        },
+        error$1
+      );
+      this.logger?.error(mastraError.toString());
+      this.logger?.trackException(mastraError);
       return [];
     }
   }
@@ -1174,52 +1432,65 @@ var D1Store = class extends storage.MastraStorage {
       countQueryBuilder.where(conditions.join(" AND "), ...queryParams);
     }
     const { sql: countSql, params: countParams } = countQueryBuilder.build();
-    const countResult = await this.executeQuery({ sql: countSql, params: countParams, first: true });
-    const total = Number(countResult?.count || 0);
-    const currentOffset = page * perPage;
-    if (total === 0) {
+    try {
+      const countResult = await this.executeQuery({ sql: countSql, params: countParams, first: true });
+      const total = Number(countResult?.count || 0);
+      const currentOffset = page * perPage;
+      if (total === 0) {
+        return {
+          evals: [],
+          total: 0,
+          page,
+          perPage,
+          hasMore: false
+        };
+      }
+      const dataQueryBuilder = createSqlBuilder().select("*").from(fullTableName);
+      if (conditions.length > 0) {
+        dataQueryBuilder.where(conditions.join(" AND "), ...queryParams);
+      }
+      dataQueryBuilder.orderBy("createdAt", "DESC").limit(perPage).offset(currentOffset);
+      const { sql: dataSql, params: dataParams } = dataQueryBuilder.build();
+      const rows = await this.executeQuery({ sql: dataSql, params: dataParams });
+      const evals = (isArrayOfRecords(rows) ? rows : []).map((row) => {
+        const result = this.deserializeValue(row.result);
+        const testInfo = row.test_info ? this.deserializeValue(row.test_info) : void 0;
+        if (!result || typeof result !== "object" || !("score" in result)) {
+          throw new Error(`Invalid MetricResult format: ${JSON.stringify(result)}`);
+        }
+        return {
+          input: row.input,
+          output: row.output,
+          result,
+          agentName: row.agent_name,
+          metricName: row.metric_name,
+          instructions: row.instructions,
+          testInfo,
+          globalRunId: row.global_run_id,
+          runId: row.run_id,
+          createdAt: row.createdAt
+        };
+      });
+      const hasMore = currentOffset + evals.length < total;
       return {
-        evals: [],
-        total: 0,
+        evals,
+        total,
         page,
         perPage,
-        hasMore: false
+        hasMore
       };
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "CLOUDFLARE_D1_STORAGE_GET_EVALS_ERROR",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY,
+          text: `Failed to retrieve evals for agent ${agentName}: ${error$1 instanceof Error ? error$1.message : String(error$1)}`,
+          details: { agentName: agentName ?? "", type: type ?? "" }
+        },
+        error$1
+      );
     }
-    const dataQueryBuilder = createSqlBuilder().select("*").from(fullTableName);
-    if (conditions.length > 0) {
-      dataQueryBuilder.where(conditions.join(" AND "), ...queryParams);
-    }
-    dataQueryBuilder.orderBy("createdAt", "DESC").limit(perPage).offset(currentOffset);
-    const { sql: dataSql, params: dataParams } = dataQueryBuilder.build();
-    const rows = await this.executeQuery({ sql: dataSql, params: dataParams });
-    const evals = (isArrayOfRecords(rows) ? rows : []).map((row) => {
-      const result = this.deserializeValue(row.result);
-      const testInfo = row.test_info ? this.deserializeValue(row.test_info) : void 0;
-      if (!result || typeof result !== "object" || !("score" in result)) {
-        throw new Error(`Invalid MetricResult format: ${JSON.stringify(result)}`);
-      }
-      return {
-        input: row.input,
-        output: row.output,
-        result,
-        agentName: row.agent_name,
-        metricName: row.metric_name,
-        instructions: row.instructions,
-        testInfo,
-        globalRunId: row.global_run_id,
-        runId: row.run_id,
-        createdAt: row.createdAt
-      };
-    });
-    const hasMore = currentOffset + evals.length < total;
-    return {
-      evals,
-      total,
-      page,
-      perPage,
-      hasMore
-    };
   }
   parseWorkflowRun(row) {
     let parsedSnapshot = row.snapshot;
@@ -1288,11 +1559,17 @@ var D1Store = class extends storage.MastraStorage {
       const results = await this.executeQuery({ sql, params });
       const runs = (isArrayOfRecords(results) ? results : []).map((row) => this.parseWorkflowRun(row));
       return { runs, total: total || runs.length };
-    } catch (error) {
-      this.logger.error("Error getting workflow runs:", {
-        message: error instanceof Error ? error.message : String(error)
-      });
-      throw error;
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "CLOUDFLARE_D1_STORAGE_GET_WORKFLOW_RUNS_ERROR",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY,
+          text: `Failed to retrieve workflow runs: ${error$1 instanceof Error ? error$1.message : String(error$1)}`,
+          details: { workflowName: workflowName ?? "", resourceId: resourceId ?? "" }
+        },
+        error$1
+      );
     }
   }
   async getWorkflowRunById({
@@ -1316,11 +1593,17 @@ var D1Store = class extends storage.MastraStorage {
       const result = await this.executeQuery({ sql, params, first: true });
       if (!result) return null;
       return this.parseWorkflowRun(result);
-    } catch (error) {
-      this.logger.error("Error getting workflow run by ID:", {
-        message: error instanceof Error ? error.message : String(error)
-      });
-      throw error;
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "CLOUDFLARE_D1_STORAGE_GET_WORKFLOW_RUN_BY_ID_ERROR",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY,
+          text: `Failed to retrieve workflow run by ID: ${error$1 instanceof Error ? error$1.message : String(error$1)}`,
+          details: { runId, workflowName: workflowName ?? "" }
+        },
+        error$1
+      );
     }
   }
   /**