@mastra/cloudflare-d1 1.0.0-beta.2 → 1.0.0-beta.4

package/dist/index.js

@@ -1,5 +1,5 @@
 import { MastraError, ErrorCategory, ErrorDomain } from '@mastra/core/error';
-import { MastraStorage, StoreOperations, ScoresStorage, TABLE_SCORERS, normalizePerPage, calculatePagination, WorkflowsStorage, TABLE_WORKFLOW_SNAPSHOT, ensureDate, MemoryStorage, TABLE_RESOURCES, TABLE_THREADS, TABLE_MESSAGES, serializeDate, transformScoreRow as transformScoreRow$1 } from '@mastra/core/storage';
+import { MastraStorage, createStorageErrorId, StoreOperations, ScoresStorage, TABLE_SCORERS, normalizePerPage, calculatePagination, WorkflowsStorage, TABLE_WORKFLOW_SNAPSHOT, ensureDate, MemoryStorage, TABLE_RESOURCES, TABLE_THREADS, TABLE_MESSAGES, serializeDate, transformScoreRow as transformScoreRow$1 } from '@mastra/core/storage';
 import Cloudflare from 'cloudflare';
 import { MessageList } from '@mastra/core/agent';
 import { parseSqlIdentifier } from '@mastra/core/utils';
@@ -275,7 +275,7 @@ var MemoryStorageD1 = class extends MemoryStorage {
     } catch (error) {
       const mastraError = new MastraError(
         {
-          id: "CLOUDFLARE_D1_STORAGE_GET_RESOURCE_BY_ID_ERROR",
+          id: createStorageErrorId("CLOUDFLARE_D1", "GET_RESOURCE_BY_ID", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           text: `Error processing resource ${resourceId}: ${error instanceof Error ? error.message : String(error)}`,
@@ -314,7 +314,7 @@ var MemoryStorageD1 = class extends MemoryStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "CLOUDFLARE_D1_STORAGE_SAVE_RESOURCE_ERROR",
+          id: createStorageErrorId("CLOUDFLARE_D1", "SAVE_RESOURCE", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           text: `Failed to save resource to ${fullTableName}: ${error instanceof Error ? error.message : String(error)}`,
@@ -361,7 +361,7 @@ var MemoryStorageD1 = class extends MemoryStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "CLOUDFLARE_D1_STORAGE_UPDATE_RESOURCE_ERROR",
+          id: createStorageErrorId("CLOUDFLARE_D1", "UPDATE_RESOURCE", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           text: `Failed to update resource ${resourceId}: ${error instanceof Error ? error.message : String(error)}`,
@@ -387,7 +387,7 @@ var MemoryStorageD1 = class extends MemoryStorage {
     } catch (error) {
       const mastraError = new MastraError(
         {
-          id: "CLOUDFLARE_D1_STORAGE_GET_THREAD_BY_ID_ERROR",
+          id: createStorageErrorId("CLOUDFLARE_D1", "GET_THREAD_BY_ID", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           text: `Error processing thread ${threadId}: ${error instanceof Error ? error.message : String(error)}`,
@@ -406,7 +406,7 @@ var MemoryStorageD1 = class extends MemoryStorage {
     if (page < 0) {
       throw new MastraError(
         {
-          id: "STORAGE_CLOUDFLARE_D1_LIST_THREADS_BY_RESOURCE_ID_INVALID_PAGE",
+          id: createStorageErrorId("CLOUDFLARE_D1", "LIST_THREADS_BY_RESOURCE_ID", "INVALID_PAGE"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.USER,
           details: { page }
@@ -441,7 +441,7 @@ var MemoryStorageD1 = class extends MemoryStorage {
     } catch (error) {
       const mastraError = new MastraError(
         {
-          id: "CLOUDFLARE_D1_STORAGE_LIST_THREADS_BY_RESOURCE_ID_ERROR",
+          id: createStorageErrorId("CLOUDFLARE_D1", "LIST_THREADS_BY_RESOURCE_ID", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           text: `Error getting threads by resourceId ${resourceId}: ${error instanceof Error ? error.message : String(error)}`,
@@ -488,7 +488,7 @@ var MemoryStorageD1 = class extends MemoryStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "CLOUDFLARE_D1_STORAGE_SAVE_THREAD_ERROR",
+          id: createStorageErrorId("CLOUDFLARE_D1", "SAVE_THREAD", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           text: `Failed to save thread to ${fullTableName}: ${error instanceof Error ? error.message : String(error)}`,
@@ -531,7 +531,7 @@ var MemoryStorageD1 = class extends MemoryStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "CLOUDFLARE_D1_STORAGE_UPDATE_THREAD_ERROR",
+          id: createStorageErrorId("CLOUDFLARE_D1", "UPDATE_THREAD", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           text: `Failed to update thread ${id}: ${error instanceof Error ? error.message : String(error)}`,
@@ -554,7 +554,7 @@ var MemoryStorageD1 = class extends MemoryStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "CLOUDFLARE_D1_STORAGE_DELETE_THREAD_ERROR",
+          id: createStorageErrorId("CLOUDFLARE_D1", "DELETE_THREAD", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           text: `Failed to delete thread ${threadId}: ${error instanceof Error ? error.message : String(error)}`,
@@ -618,7 +618,7 @@ var MemoryStorageD1 = class extends MemoryStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "CLOUDFLARE_D1_STORAGE_SAVE_MESSAGES_ERROR",
+          id: createStorageErrorId("CLOUDFLARE_D1", "SAVE_MESSAGES", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           text: `Failed to save messages: ${error instanceof Error ? error.message : String(error)}`
@@ -710,7 +710,7 @@ var MemoryStorageD1 = class extends MemoryStorage {
     } catch (error) {
       const mastraError = new MastraError(
         {
-          id: "CLOUDFLARE_D1_STORAGE_LIST_MESSAGES_BY_ID_ERROR",
+          id: createStorageErrorId("CLOUDFLARE_D1", "LIST_MESSAGES_BY_ID", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           text: `Failed to retrieve messages by ID: ${error instanceof Error ? error.message : String(error)}`,
@@ -729,7 +729,7 @@ var MemoryStorageD1 = class extends MemoryStorage {
     if (threadIds.length === 0 || threadIds.some((id) => !id.trim())) {
       throw new MastraError(
         {
-          id: "STORAGE_CLOUDFLARE_D1_LIST_MESSAGES_INVALID_THREAD_ID",
+          id: createStorageErrorId("CLOUDFLARE_D1", "LIST_MESSAGES", "INVALID_THREAD_ID"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { threadId: Array.isArray(threadId) ? threadId.join(",") : threadId }
@@ -740,7 +740,7 @@ var MemoryStorageD1 = class extends MemoryStorage {
     if (page < 0) {
       throw new MastraError(
         {
-          id: "STORAGE_CLOUDFLARE_D1_LIST_MESSAGES_INVALID_PAGE",
+          id: createStorageErrorId("CLOUDFLARE_D1", "LIST_MESSAGES", "INVALID_PAGE"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.USER,
           details: { page }
@@ -857,7 +857,7 @@ var MemoryStorageD1 = class extends MemoryStorage {
     } catch (error) {
       const mastraError = new MastraError(
         {
-          id: "CLOUDFLARE_D1_STORAGE_LIST_MESSAGES_ERROR",
+          id: createStorageErrorId("CLOUDFLARE_D1", "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)}`,
@@ -970,7 +970,7 @@ var MemoryStorageD1 = class extends MemoryStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "CLOUDFLARE_D1_STORAGE_UPDATE_MESSAGES_FAILED",
+          id: createStorageErrorId("CLOUDFLARE_D1", "UPDATE_MESSAGES", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { count: messages.length }
@@ -1039,7 +1039,7 @@ var StoreOperationsD1 = class extends StoreOperations {
     } catch (error) {
       throw new MastraError(
         {
-          id: "CLOUDFLARE_D1_STORE_OPERATIONS_WORKERS_BINDING_QUERY_FAILED",
+          id: createStorageErrorId("CLOUDFLARE_D1", "WORKERS_BINDING_QUERY", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { sql }
@@ -1071,7 +1071,7 @@ var StoreOperationsD1 = class extends StoreOperations {
     } catch (error) {
       throw new MastraError(
         {
-          id: "CLOUDFLARE_D1_STORE_OPERATIONS_REST_QUERY_FAILED",
+          id: createStorageErrorId("CLOUDFLARE_D1", "REST_QUERY", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { sql }
@@ -1152,7 +1152,7 @@ var StoreOperationsD1 = class extends StoreOperations {
     } catch (error) {
       throw new MastraError(
         {
-          id: "CLOUDFLARE_D1_STORE_OPERATIONS_CREATE_TABLE_FAILED",
+          id: createStorageErrorId("CLOUDFLARE_D1", "CREATE_TABLE", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { tableName }
@@ -1171,7 +1171,7 @@ var StoreOperationsD1 = class extends StoreOperations {
     } catch (error) {
       throw new MastraError(
         {
-          id: "CLOUDFLARE_D1_STORE_OPERATIONS_CLEAR_TABLE_FAILED",
+          id: createStorageErrorId("CLOUDFLARE_D1", "CLEAR_TABLE", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { tableName }
@@ -1189,7 +1189,7 @@ var StoreOperationsD1 = class extends StoreOperations {
     } catch (error) {
       throw new MastraError(
         {
-          id: "CLOUDFLARE_D1_STORE_OPERATIONS_DROP_TABLE_FAILED",
+          id: createStorageErrorId("CLOUDFLARE_D1", "DROP_TABLE", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { tableName }
@@ -1215,7 +1215,7 @@ var StoreOperationsD1 = class extends StoreOperations {
     } catch (error) {
       throw new MastraError(
         {
-          id: "CLOUDFLARE_D1_STORE_OPERATIONS_ALTER_TABLE_FAILED",
+          id: createStorageErrorId("CLOUDFLARE_D1", "ALTER_TABLE", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { tableName: args.tableName }
@@ -1236,7 +1236,7 @@ var StoreOperationsD1 = class extends StoreOperations {
     } catch (error) {
       throw new MastraError(
         {
-          id: "CLOUDFLARE_D1_STORE_OPERATIONS_INSERT_FAILED",
+          id: createStorageErrorId("CLOUDFLARE_D1", "INSERT", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { tableName }
@@ -1260,7 +1260,7 @@ var StoreOperationsD1 = class extends StoreOperations {
     } catch (error) {
       throw new MastraError(
         {
-          id: "CLOUDFLARE_D1_STORE_OPERATIONS_BATCH_INSERT_FAILED",
+          id: createStorageErrorId("CLOUDFLARE_D1", "BATCH_INSERT", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { tableName }
@@ -1297,7 +1297,7 @@ var StoreOperationsD1 = class extends StoreOperations {
     } catch (error) {
       throw new MastraError(
         {
-          id: "CLOUDFLARE_D1_STORE_OPERATIONS_LOAD_FAILED",
+          id: createStorageErrorId("CLOUDFLARE_D1", "LOAD", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           details: { tableName }
@@ -1355,7 +1355,7 @@ var StoreOperationsD1 = class extends StoreOperations {
     } catch (error) {
       throw new MastraError(
         {
-          id: "CLOUDFLARE_D1_STORAGE_BATCH_UPSERT_ERROR",
+          id: createStorageErrorId("CLOUDFLARE_D1", "BATCH_UPSERT", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           text: `Failed to batch upsert into ${tableName}: ${error instanceof Error ? error.message : String(error)}`,
@@ -1393,7 +1393,7 @@ var ScoresStorageD1 = class extends ScoresStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "CLOUDFLARE_D1_STORE_SCORES_GET_SCORE_BY_ID_FAILED",
+          id: createStorageErrorId("CLOUDFLARE_D1", "GET_SCORE_BY_ID", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY
         },
@@ -1408,16 +1408,22 @@ var ScoresStorageD1 = class extends ScoresStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "CLOUDFLARE_D1_STORE_SCORES_SAVE_SCORE_FAILED_INVALID_SCORE_PAYLOAD",
+          id: createStorageErrorId("CLOUDFLARE_D1", "SAVE_SCORE", "VALIDATION_FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.USER,
-          details: { scoreId: score.id }
+          details: {
+            scorer: score.scorer?.id ?? "unknown",
+            entityId: score.entityId ?? "unknown",
+            entityType: score.entityType ?? "unknown",
+            traceId: score.traceId ?? "",
+            spanId: score.spanId ?? ""
+          }
         },
         error
       );
     }
+    const id = crypto.randomUUID();
     try {
-      const id = crypto.randomUUID();
       const fullTableName = this.operations.getTableName(TABLE_SCORERS);
       const serializedRecord = {};
       for (const [key, value] of Object.entries(parsedScore)) {
@@ -1431,22 +1437,23 @@ var ScoresStorageD1 = class extends ScoresStorage {
           serializedRecord[key] = null;
         }
       }
+      const now = /* @__PURE__ */ new Date();
       serializedRecord.id = id;
-      serializedRecord.createdAt = (/* @__PURE__ */ new Date()).toISOString();
-      serializedRecord.updatedAt = (/* @__PURE__ */ new Date()).toISOString();
+      serializedRecord.createdAt = now.toISOString();
+      serializedRecord.updatedAt = now.toISOString();
       const columns = Object.keys(serializedRecord);
       const values = Object.values(serializedRecord);
       const query = createSqlBuilder().insert(fullTableName, columns, values);
       const { sql, params } = query.build();
       await this.operations.executeQuery({ sql, params });
-      const scoreFromDb = await this.getScoreById({ id });
-      return { score: scoreFromDb };
+      return { score: { ...parsedScore, id, createdAt: now, updatedAt: now } };
     } catch (error) {
       throw new MastraError(
         {
-          id: "CLOUDFLARE_D1_STORE_SCORES_SAVE_SCORE_FAILED",
+          id: createStorageErrorId("CLOUDFLARE_D1", "SAVE_SCORE", "FAILED"),
           domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.THIRD_PARTY
+          category: ErrorCategory.THIRD_PARTY,
+          details: { id }
         },
         error
       );
@@ -1515,7 +1522,7 @@ var ScoresStorageD1 = class extends ScoresStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "CLOUDFLARE_D1_STORE_SCORES_GET_SCORES_BY_SCORER_ID_FAILED",
+          id: createStorageErrorId("CLOUDFLARE_D1", "GET_SCORES_BY_SCORER_ID", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY
         },
@@ -1564,7 +1571,7 @@ var ScoresStorageD1 = class extends ScoresStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "CLOUDFLARE_D1_STORE_SCORES_GET_SCORES_BY_RUN_ID_FAILED",
+          id: createStorageErrorId("CLOUDFLARE_D1", "GET_SCORES_BY_RUN_ID", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY
         },
@@ -1614,7 +1621,7 @@ var ScoresStorageD1 = class extends ScoresStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "CLOUDFLARE_D1_STORE_SCORES_GET_SCORES_BY_ENTITY_ID_FAILED",
+          id: createStorageErrorId("CLOUDFLARE_D1", "GET_SCORES_BY_ENTITY_ID", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY
         },
@@ -1664,7 +1671,7 @@ var ScoresStorageD1 = class extends ScoresStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "CLOUDFLARE_D1_STORE_SCORES_GET_SCORES_BY_SPAN_FAILED",
+          id: createStorageErrorId("CLOUDFLARE_D1", "GET_SCORES_BY_SPAN", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY
         },
@@ -1735,7 +1742,7 @@ var WorkflowsStorageD1 = class extends WorkflowsStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "CLOUDFLARE_D1_STORAGE_PERSIST_WORKFLOW_SNAPSHOT_ERROR",
+          id: createStorageErrorId("CLOUDFLARE_D1", "PERSIST_WORKFLOW_SNAPSHOT", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           text: `Failed to persist workflow snapshot: ${error instanceof Error ? error.message : String(error)}`,
@@ -1760,7 +1767,7 @@ var WorkflowsStorageD1 = class extends WorkflowsStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "CLOUDFLARE_D1_STORAGE_LOAD_WORKFLOW_SNAPSHOT_ERROR",
+          id: createStorageErrorId("CLOUDFLARE_D1", "LOAD_WORKFLOW_SNAPSHOT", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           text: `Failed to load workflow snapshot: ${error instanceof Error ? error.message : String(error)}`,
@@ -1846,7 +1853,7 @@ var WorkflowsStorageD1 = class extends WorkflowsStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "CLOUDFLARE_D1_STORAGE_LIST_WORKFLOW_RUNS_ERROR",
+          id: createStorageErrorId("CLOUDFLARE_D1", "LIST_WORKFLOW_RUNS", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           text: `Failed to retrieve workflow runs: ${error instanceof Error ? error.message : String(error)}`,
@@ -1883,7 +1890,7 @@ var WorkflowsStorageD1 = class extends WorkflowsStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "CLOUDFLARE_D1_STORAGE_GET_WORKFLOW_RUN_BY_ID_ERROR",
+          id: createStorageErrorId("CLOUDFLARE_D1", "GET_WORKFLOW_RUN_BY_ID", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           text: `Failed to retrieve workflow run by ID: ${error instanceof Error ? error.message : String(error)}`,
@@ -1908,7 +1915,7 @@ var D1Store = class extends MastraStorage {
    */
   constructor(config) {
     try {
-      super({ id: config.id, name: "D1" });
+      super({ id: config.id, name: "D1", disableInit: config.disableInit });
       if (config.tablePrefix && !/^[a-zA-Z0-9_]*$/.test(config.tablePrefix)) {
         throw new Error("Invalid tablePrefix: only letters, numbers, and underscores are allowed.");
       }
@@ -1946,7 +1953,7 @@ var D1Store = class extends MastraStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: "CLOUDFLARE_D1_STORAGE_INITIALIZATION_ERROR",
+          id: createStorageErrorId("CLOUDFLARE_D1", "INITIALIZATION", "FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.SYSTEM,
           text: "Error initializing D1Store"
@@ -2101,8 +2108,8 @@ var D1Store = class extends MastraStorage {
   async getScoreById({ id: _id }) {
     return this.stores.scores.getScoreById({ id: _id });
   }
-  async saveScore(_score) {
-    return this.stores.scores.saveScore(_score);
+  async saveScore(score) {
+    return this.stores.scores.saveScore(score);
   }
   async listScoresByRunId({
     runId: _runId,