@mastra/cloudflare-d1 0.0.0-update-stores-peerDeps-20250723031338 → 0.0.0-usechat-duplicate-20251016110554

package/dist/index.js

@@ -1,8 +1,9 @@
 import { MastraError, ErrorCategory, ErrorDomain } from '@mastra/core/error';
-import { MastraStorage, StoreOperations, ScoresStorage, TABLE_SCORERS, LegacyEvalsStorage, TABLE_EVALS, serializeDate, TracesStorage, TABLE_TRACES, WorkflowsStorage, TABLE_WORKFLOW_SNAPSHOT, ensureDate, MemoryStorage, TABLE_RESOURCES, TABLE_THREADS, TABLE_MESSAGES, resolveMessageLimit } from '@mastra/core/storage';
+import { MastraStorage, StoreOperations, ScoresStorage, TABLE_SCORERS, LegacyEvalsStorage, TABLE_EVALS, serializeDate, TracesStorage, TABLE_TRACES, WorkflowsStorage, TABLE_WORKFLOW_SNAPSHOT, ensureDate, MemoryStorage, TABLE_RESOURCES, TABLE_THREADS, TABLE_MESSAGES, resolveMessageLimit, safelyParseJSON } from '@mastra/core/storage';
 import Cloudflare from 'cloudflare';
 import { parseSqlIdentifier } from '@mastra/core/utils';
 import { MessageList } from '@mastra/core/agent';
+import { saveScorePayloadSchema } from '@mastra/core/scores';
 
 // src/storage/index.ts
 var SqlBuilder = class {
@@ -534,7 +535,6 @@ var MemoryStorageD1 = class extends MemoryStorage {
       keys: { id: threadId }
     });
     if (!thread) return null;
-    console.log("thread", thread);
     try {
       return {
         ...thread,
@@ -803,6 +803,7 @@ var MemoryStorageD1 = class extends MemoryStorage {
     }
   }
   async _getIncludedMessages(threadId, selectBy) {
+    if (!threadId.trim()) throw new Error("threadId must be a non-empty string");
     const include = selectBy?.include;
     if (!include) return null;
     const unionQueries = [];
@@ -861,17 +862,19 @@ var MemoryStorageD1 = class extends MemoryStorage {
   }
   async getMessages({
     threadId,
+    resourceId,
     selectBy,
     format
   }) {
-    const fullTableName = this.operations.getTableName(TABLE_MESSAGES);
-    const limit = resolveMessageLimit({
-      last: selectBy?.last,
-      defaultLimit: 40
-    });
-    const include = selectBy?.include || [];
-    const messages = [];
     try {
+      if (!threadId.trim()) throw new Error("threadId must be a non-empty string");
+      const fullTableName = this.operations.getTableName(TABLE_MESSAGES);
+      const limit = resolveMessageLimit({
+        last: selectBy?.last,
+        defaultLimit: 40
+      });
+      const include = selectBy?.include || [];
+      const messages = [];
       if (include.length) {
         const includeResult = await this._getIncludedMessages(threadId, selectBy);
         if (Array.isArray(includeResult)) messages.push(...includeResult);
@@ -911,7 +914,48 @@ var MemoryStorageD1 = class extends MemoryStorage {
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           text: `Failed to retrieve messages for thread ${threadId}: ${error instanceof Error ? error.message : String(error)}`,
-          details: { threadId }
+          details: { threadId, resourceId: resourceId ?? "" }
+        },
+        error
+      );
+      this.logger?.error(mastraError.toString());
+      this.logger?.trackException(mastraError);
+      throw mastraError;
+    }
+  }
+  async getMessagesById({
+    messageIds,
+    format
+  }) {
+    if (messageIds.length === 0) return [];
+    const fullTableName = this.operations.getTableName(TABLE_MESSAGES);
+    const messages = [];
+    try {
+      const query = createSqlBuilder().select(["id", "content", "role", "type", "createdAt", "thread_id AS threadId", "resourceId"]).from(fullTableName).where(`id in (${messageIds.map(() => "?").join(",")})`, ...messageIds);
+      query.orderBy("createdAt", "DESC");
+      const { sql, params } = query.build();
+      const result = await this.operations.executeQuery({ sql, params });
+      if (Array.isArray(result)) messages.push(...result);
+      const processedMessages = messages.map((message) => {
+        const processedMsg = {};
+        for (const [key, value] of Object.entries(message)) {
+          if (key === `type` && value === `v2`) continue;
+          processedMsg[key] = deserializeValue(value);
+        }
+        return processedMsg;
+      });
+      this.logger.debug(`Retrieved ${messages.length} messages`);
+      const list = new MessageList().add(processedMessages, "memory");
+      if (format === `v1`) return list.get.all.v1();
+      return list.get.all.v2();
+    } catch (error) {
+      const mastraError = new MastraError(
+        {
+          id: "CLOUDFLARE_D1_STORAGE_GET_MESSAGES_BY_ID_ERROR",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          text: `Failed to retrieve messages by ID: ${error instanceof Error ? error.message : String(error)}`,
+          details: { messageIds: JSON.stringify(messageIds) }
         },
         error
       );
@@ -922,6 +966,7 @@ var MemoryStorageD1 = class extends MemoryStorage {
   }
   async getMessagesPaginated({
     threadId,
+    resourceId,
     selectBy,
     format
   }) {
@@ -931,6 +976,7 @@ var MemoryStorageD1 = class extends MemoryStorage {
     const fullTableName = this.operations.getTableName(TABLE_MESSAGES);
     const messages = [];
     try {
+      if (!threadId.trim()) throw new Error("threadId must be a non-empty string");
       if (selectBy?.include?.length) {
         const includeResult = await this._getIncludedMessages(threadId, selectBy);
         if (Array.isArray(includeResult)) messages.push(...includeResult);
@@ -1019,7 +1065,7 @@ var MemoryStorageD1 = class extends MemoryStorage {
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.THIRD_PARTY,
           text: `Failed to retrieve messages for thread ${threadId}: ${error instanceof Error ? error.message : String(error)}`,
-          details: { threadId }
+          details: { threadId, resourceId: resourceId ?? "" }
         },
         error
       );
@@ -1217,13 +1263,12 @@ var StoreOperationsD1 = class extends StoreOperations {
         sql,
         params: formattedParams
       });
-      if (!response.result) {
-        return first ? null : [];
-      }
+      const result = response.result || [];
+      const results = result.flatMap((r) => r.results || []);
       if (first) {
-        return response.result[0] || null;
+        return results[0] || null;
       }
-      return response.result;
+      return results;
     } catch (error) {
       throw new MastraError(
         {
@@ -1438,6 +1483,7 @@ var StoreOperationsD1 = class extends StoreOperations {
           query.andWhere(`${key} = ?`, value);
         }
       }
+      query.orderBy("createdAt", "DESC");
       query.limit(1);
       const { sql, params } = query.build();
       const result = await this.executeQuery({ sql, params, first: true });
@@ -1522,20 +1568,19 @@ var StoreOperationsD1 = class extends StoreOperations {
   }
 };
 function transformScoreRow(row) {
-  let input = void 0;
-  if (row.input) {
-    try {
-      input = JSON.parse(row.input);
-    } catch {
-      input = row.input;
-    }
-  }
-  return {
-    ...row,
-    input,
-    createdAt: row.createdAtZ || row.createdAt,
-    updatedAt: row.updatedAtZ || row.updatedAt
-  };
+  const deserialized = { ...row };
+  deserialized.input = safelyParseJSON(row.input);
+  deserialized.output = safelyParseJSON(row.output);
+  deserialized.scorer = safelyParseJSON(row.scorer);
+  deserialized.preprocessStepResult = safelyParseJSON(row.preprocessStepResult);
+  deserialized.analyzeStepResult = safelyParseJSON(row.analyzeStepResult);
+  deserialized.metadata = safelyParseJSON(row.metadata);
+  deserialized.additionalContext = safelyParseJSON(row.additionalContext);
+  deserialized.runtimeContext = safelyParseJSON(row.runtimeContext);
+  deserialized.entity = safelyParseJSON(row.entity);
+  deserialized.createdAt = row.createdAtZ || row.createdAt;
+  deserialized.updatedAt = row.updatedAtZ || row.updatedAt;
+  return deserialized;
 }
 var ScoresStorageD1 = class extends ScoresStorage {
   operations;
@@ -1565,11 +1610,25 @@ var ScoresStorageD1 = class extends ScoresStorage {
     }
   }
   async saveScore(score) {
+    let parsedScore;
+    try {
+      parsedScore = saveScorePayloadSchema.parse(score);
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: "CLOUDFLARE_D1_STORE_SCORES_SAVE_SCORE_FAILED_INVALID_SCORE_PAYLOAD",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.USER,
+          details: { scoreId: score.id }
+        },
+        error
+      );
+    }
     try {
+      const id = crypto.randomUUID();
       const fullTableName = this.operations.getTableName(TABLE_SCORERS);
-      const { input, ...rest } = score;
       const serializedRecord = {};
-      for (const [key, value] of Object.entries(rest)) {
+      for (const [key, value] of Object.entries(parsedScore)) {
         if (value !== null && value !== void 0) {
           if (typeof value === "object") {
             serializedRecord[key] = JSON.stringify(value);
@@ -1580,7 +1639,7 @@ var ScoresStorageD1 = class extends ScoresStorage {
           serializedRecord[key] = null;
         }
       }
-      serializedRecord.input = JSON.stringify(input);
+      serializedRecord.id = id;
       serializedRecord.createdAt = (/* @__PURE__ */ new Date()).toISOString();
       serializedRecord.updatedAt = (/* @__PURE__ */ new Date()).toISOString();
       const columns = Object.keys(serializedRecord);
@@ -1588,7 +1647,7 @@ var ScoresStorageD1 = class extends ScoresStorage {
       const query = createSqlBuilder().insert(fullTableName, columns, values);
       const { sql, params } = query.build();
       await this.operations.executeQuery({ sql, params });
-      const scoreFromDb = await this.getScoreById({ id: score.id });
+      const scoreFromDb = await this.getScoreById({ id });
       return { score: scoreFromDb };
     } catch (error) {
       throw new MastraError(
@@ -1603,11 +1662,23 @@ var ScoresStorageD1 = class extends ScoresStorage {
   }
   async getScoresByScorerId({
     scorerId,
+    entityId,
+    entityType,
+    source,
     pagination
   }) {
     try {
       const fullTableName = this.operations.getTableName(TABLE_SCORERS);
       const countQuery = createSqlBuilder().count().from(fullTableName).where("scorerId = ?", scorerId);
+      if (entityId) {
+        countQuery.andWhere("entityId = ?", entityId);
+      }
+      if (entityType) {
+        countQuery.andWhere("entityType = ?", entityType);
+      }
+      if (source) {
+        countQuery.andWhere("source = ?", source);
+      }
       const countResult = await this.operations.executeQuery(countQuery.build());
       const total = Array.isArray(countResult) ? Number(countResult?.[0]?.count ?? 0) : Number(countResult?.count ?? 0);
       if (total === 0) {
@@ -1621,7 +1692,17 @@ var ScoresStorageD1 = class extends ScoresStorage {
           scores: []
         };
       }
-      const selectQuery = createSqlBuilder().select("*").from(fullTableName).where("scorerId = ?", scorerId).limit(pagination.perPage).offset(pagination.page * pagination.perPage);
+      const selectQuery = createSqlBuilder().select("*").from(fullTableName).where("scorerId = ?", scorerId);
+      if (entityId) {
+        selectQuery.andWhere("entityId = ?", entityId);
+      }
+      if (entityType) {
+        selectQuery.andWhere("entityType = ?", entityType);
+      }
+      if (source) {
+        selectQuery.andWhere("source = ?", source);
+      }
+      selectQuery.limit(pagination.perPage).offset(pagination.page * pagination.perPage);
       const { sql, params } = selectQuery.build();
       const results = await this.operations.executeQuery({ sql, params });
       const scores = Array.isArray(results) ? results.map(transformScoreRow) : [];
@@ -1734,6 +1815,53 @@ var ScoresStorageD1 = class extends ScoresStorage {
       );
     }
   }
+  async getScoresBySpan({
+    traceId,
+    spanId,
+    pagination
+  }) {
+    try {
+      const fullTableName = this.operations.getTableName(TABLE_SCORERS);
+      const countQuery = createSqlBuilder().count().from(fullTableName).where("traceId = ?", traceId).andWhere("spanId = ?", spanId);
+      const countResult = await this.operations.executeQuery(countQuery.build());
+      const total = Array.isArray(countResult) ? Number(countResult?.[0]?.count ?? 0) : Number(countResult?.count ?? 0);
+      if (total === 0) {
+        return {
+          pagination: {
+            total: 0,
+            page: pagination.page,
+            perPage: pagination.perPage,
+            hasMore: false
+          },
+          scores: []
+        };
+      }
+      const limit = pagination.perPage + 1;
+      const selectQuery = createSqlBuilder().select("*").from(fullTableName).where("traceId = ?", traceId).andWhere("spanId = ?", spanId).orderBy("createdAt", "DESC").limit(limit).offset(pagination.page * pagination.perPage);
+      const { sql, params } = selectQuery.build();
+      const results = await this.operations.executeQuery({ sql, params });
+      const rows = Array.isArray(results) ? results : [];
+      const scores = rows.slice(0, pagination.perPage).map(transformScoreRow);
+      return {
+        pagination: {
+          total,
+          page: pagination.page,
+          perPage: pagination.perPage,
+          hasMore: rows.length > pagination.perPage
+        },
+        scores
+      };
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: "CLOUDFLARE_D1_STORE_SCORES_GET_SCORES_BY_SPAN_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY
+        },
+        error
+      );
+    }
+  }
 };
 function isArrayOfRecords2(value) {
   return value && Array.isArray(value) && value.length > 0;
@@ -1811,7 +1939,7 @@ var TracesStorageD1 = class extends TracesStorage {
       const allDataResult = await this.operations.executeQuery(
         createSqlBuilder().select("*").from(fullTableName).where("1=1").build()
       );
-      console.log("allDataResult", allDataResult);
+      console.info("allDataResult", allDataResult);
       const countResult = await this.operations.executeQuery(countQuery.build());
       const total = Number(countResult?.[0]?.count ?? 0);
       dataQuery.orderBy("startTime", "DESC").limit(perPage).offset(page * perPage);
@@ -1863,9 +1991,26 @@ var WorkflowsStorageD1 = class extends WorkflowsStorage {
     super();
     this.operations = operations;
   }
+  updateWorkflowResults({
+    // workflowName,
+    // runId,
+    // stepId,
+    // result,
+    // runtimeContext,
+  }) {
+    throw new Error("Method not implemented.");
+  }
+  updateWorkflowState({
+    // workflowName,
+    // runId,
+    // opts,
+  }) {
+    throw new Error("Method not implemented.");
+  }
   async persistWorkflowSnapshot({
     workflowName,
     runId,
+    resourceId,
     snapshot
   }) {
     const fullTableName = this.operations.getTableName(TABLE_WORKFLOW_SNAPSHOT);
@@ -1876,11 +2021,13 @@ var WorkflowsStorageD1 = class extends WorkflowsStorage {
     });
     const persisting = currentSnapshot ? {
       ...currentSnapshot,
+      resourceId,
       snapshot: JSON.stringify(snapshot),
       updatedAt: now
     } : {
       workflow_name: workflowName,
       run_id: runId,
+      resourceId,
       snapshot,
       createdAt: now,
       updatedAt: now
@@ -2145,7 +2292,9 @@ var D1Store = class extends MastraStorage {
       selectByIncludeResourceScope: true,
       resourceWorkingMemory: true,
       hasColumn: true,
-      createTable: true
+      createTable: true,
+      deleteMessages: false,
+      getScoresBySpan: true
     };
   }
   async createTable({
@@ -2217,6 +2366,12 @@ var D1Store = class extends MastraStorage {
   }) {
     return this.stores.memory.getMessages({ threadId, selectBy, format });
   }
+  async getMessagesById({
+    messageIds,
+    format
+  }) {
+    return this.stores.memory.getMessagesById({ messageIds, format });
+  }
   async getMessagesPaginated({
     threadId,
     selectBy,
@@ -2224,12 +2379,29 @@ var D1Store = class extends MastraStorage {
   }) {
     return this.stores.memory.getMessagesPaginated({ threadId, selectBy, format });
   }
+  async updateWorkflowResults({
+    workflowName,
+    runId,
+    stepId,
+    result,
+    runtimeContext
+  }) {
+    return this.stores.workflows.updateWorkflowResults({ workflowName, runId, stepId, result, runtimeContext });
+  }
+  async updateWorkflowState({
+    workflowName,
+    runId,
+    opts
+  }) {
+    return this.stores.workflows.updateWorkflowState({ workflowName, runId, opts });
+  }
   async persistWorkflowSnapshot({
     workflowName,
     runId,
+    resourceId,
     snapshot
   }) {
-    return this.stores.workflows.persistWorkflowSnapshot({ workflowName, runId, snapshot });
+    return this.stores.workflows.persistWorkflowSnapshot({ workflowName, runId, resourceId, snapshot });
   }
   async loadWorkflowSnapshot(params) {
     return this.stores.workflows.loadWorkflowSnapshot(params);
@@ -2316,10 +2488,20 @@ var D1Store = class extends MastraStorage {
     });
   }
   async getScoresByScorerId({
-    scorerId: _scorerId,
-    pagination: _pagination
+    scorerId,
+    pagination,
+    entityId,
+    entityType,
+    source
+  }) {
+    return this.stores.scores.getScoresByScorerId({ scorerId, pagination, entityId, entityType, source });
+  }
+  async getScoresBySpan({
+    traceId,
+    spanId,
+    pagination
   }) {
-    return this.stores.scores.getScoresByScorerId({ scorerId: _scorerId, pagination: _pagination });
+    return this.stores.scores.getScoresBySpan({ traceId, spanId, pagination });
   }
   /**
    * Close the database connection
@@ -2331,3 +2513,5 @@ var D1Store = class extends MastraStorage {
 };
 
 export { D1Store };
+//# sourceMappingURL=index.js.map
+//# sourceMappingURL=index.js.map