@mastra/cloudflare 0.0.0-update-scorers-api-20250801170445 → 0.0.0-usechat-duplicate-20251016110554

package/dist/index.cjs

@@ -4,6 +4,7 @@ var error = require('@mastra/core/error');
 var storage = require('@mastra/core/storage');
 var Cloudflare = require('cloudflare');
 var agent = require('@mastra/core/agent');
+var scores = require('@mastra/core/scores');
 
 function _interopDefault (e) { return e && e.__esModule ? e : { default: e }; }
 
@@ -469,9 +470,9 @@ var MemoryStorageCloudflare = class extends storage.MemoryStorage {
       const messageMigrationTasks = [];
       for (const message of validatedMessages) {
         const existingMessage = await this.findMessageInAnyThread(message.id);
-        console.log(`Checking message ${message.id}: existing=${existingMessage?.threadId}, new=${message.threadId}`);
+        console.info(`Checking message ${message.id}: existing=${existingMessage?.threadId}, new=${message.threadId}`);
         if (existingMessage && existingMessage.threadId && existingMessage.threadId !== message.threadId) {
-          console.log(`Migrating message ${message.id} from ${existingMessage.threadId} to ${message.threadId}`);
+          console.info(`Migrating message ${message.id} from ${existingMessage.threadId} to ${message.threadId}`);
           messageMigrationTasks.push(this.migrateMessage(message.id, existingMessage.threadId, message.threadId));
         }
       }
@@ -500,7 +501,7 @@ var MemoryStorageCloudflare = class extends storage.MemoryStorage {
                   ...cleanMessage,
                   createdAt: storage.serializeDate(cleanMessage.createdAt)
                 };
-                console.log(`Saving message ${message.id} with content:`, {
+                console.info(`Saving message ${message.id} with content:`, {
                   content: serializedMessage.content,
                   contentType: typeof serializedMessage.content,
                   isArray: Array.isArray(serializedMessage.content)
@@ -596,13 +597,14 @@ var MemoryStorageCloudflare = class extends storage.MemoryStorage {
     );
   }
   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 {
-      console.log(`No message order found for thread ${threadId}, skipping latest messages`);
+      console.info(`No message order found for thread ${threadId}, skipping latest messages`);
     }
   }
   async fetchAndParseMessagesFromMultipleThreads(messageIds, include, targetThreadId) {
@@ -633,7 +635,7 @@ var MemoryStorageCloudflare = class extends storage.MemoryStorage {
           const data = await this.operations.getKV(storage.TABLE_MESSAGES, key);
           if (!data) return null;
           const parsed = typeof data === "string" ? JSON.parse(data) : data;
-          console.log(`Retrieved message ${id} from thread ${threadId} with content:`, {
+          console.info(`Retrieved message ${id} from thread ${threadId} with content:`, {
             content: parsed.content,
             contentType: typeof parsed.content,
             isArray: Array.isArray(parsed.content)
@@ -654,15 +656,14 @@ var MemoryStorageCloudflare = class extends storage.MemoryStorage {
     selectBy,
     format
   }) {
-    console.log(`getMessages called with format: ${format}, threadId: ${threadId}`);
-    if (!threadId) throw new Error("threadId is required");
+    console.info(`getMessages called with format: ${format}, threadId: ${threadId}`);
     const actualFormat = format || "v1";
-    console.log(`Using format: ${actualFormat}`);
-    if (!threadId) throw new Error("threadId is required");
+    console.info(`Using format: ${actualFormat}`);
     const limit = storage.resolveMessageLimit({ last: selectBy?.last, defaultLimit: 40 });
     const messageIds = /* @__PURE__ */ new Set();
     if (limit === 0 && !selectBy?.include?.length) return [];
     try {
+      if (!threadId.trim()) throw new Error("threadId must be a non-empty string");
       await Promise.all([
         selectBy?.include?.length ? this.getIncludedMessagesWithContext(threadId, selectBy.include, messageIds) : Promise.resolve(),
         limit > 0 ? this.getRecentMessages(threadId, limit, messageIds) : Promise.resolve()
@@ -707,7 +708,7 @@ var MemoryStorageCloudflare = class extends storage.MemoryStorage {
         createdAt: storage.ensureDate(message.createdAt)
       }));
       if (actualFormat === `v1`) {
-        console.log(`Processing ${prepared.length} messages for v1 format - returning directly without MessageList`);
+        console.info(`Processing ${prepared.length} messages for v1 format - returning directly without MessageList`);
         return prepared.map((msg) => ({
           ...msg,
           createdAt: new Date(msg.createdAt)
@@ -723,7 +724,43 @@ var MemoryStorageCloudflare = class extends storage.MemoryStorage {
           category: error.ErrorCategory.THIRD_PARTY,
           text: `Error retrieving messages for thread ${threadId}`,
           details: {
-            threadId
+            threadId,
+            resourceId: resourceId ?? ""
+          }
+        },
+        error$1
+      );
+      this.logger?.trackException(mastraError);
+      this.logger?.error(mastraError.toString());
+      return [];
+    }
+  }
+  async getMessagesById({
+    messageIds,
+    format
+  }) {
+    if (messageIds.length === 0) return [];
+    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: storage.ensureDate(message.createdAt)
+      }));
+      const list = new agent.MessageList().add(prepared, "memory");
+      if (format === `v1`) return list.get.all.v1();
+      return list.get.all.v2();
+    } catch (error$1) {
+      const mastraError = new error.MastraError(
+        {
+          id: "CLOUDFLARE_STORAGE_GET_MESSAGES_BY_ID_FAILED",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY,
+          text: `Error retrieving messages by ID`,
+          details: {
+            messageIds: JSON.stringify(messageIds)
           }
         },
         error$1
@@ -734,9 +771,10 @@ var MemoryStorageCloudflare = class extends storage.MemoryStorage {
     }
   }
   async getMessagesPaginated(args) {
+    const { threadId, resourceId, selectBy, format = "v1" } = args;
+    const { page = 0, perPage = 100 } = selectBy?.pagination || {};
     try {
-      const { threadId, selectBy, format = "v1" } = args;
-      const { page = 0, perPage = 100 } = selectBy?.pagination || {};
+      if (!threadId.trim()) throw new Error("threadId must be a non-empty string");
       const messages = format === "v2" ? await this.getMessages({ threadId, selectBy, format: "v2" }) : await this.getMessages({ threadId, selectBy, format: "v1" });
       let filteredMessages = messages;
       if (selectBy?.pagination?.dateRange) {
@@ -759,15 +797,22 @@ var MemoryStorageCloudflare = class extends storage.MemoryStorage {
         messages: paginatedMessages
       };
     } catch (error$1) {
-      throw new error.MastraError(
+      const mastraError = new error.MastraError(
         {
           id: "CLOUDFLARE_STORAGE_GET_MESSAGES_PAGINATED_FAILED",
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY,
-          text: "Failed to get messages with pagination"
+          text: "Failed to get messages with pagination",
+          details: {
+            threadId,
+            resourceId: resourceId ?? ""
+          }
         },
         error$1
       );
+      this.logger?.trackException?.(mastraError);
+      this.logger?.error?.(mastraError.toString());
+      return { messages: [], total: 0, page, perPage: perPage || 40, hasMore: false };
     }
   }
   async updateMessages(args) {
@@ -1535,18 +1580,17 @@ var StoreOperationsCloudflare = class extends storage.StoreOperations {
   }
 };
 function transformScoreRow(row) {
-  let input = void 0;
-  if (row.input) {
-    try {
-      input = JSON.parse(row.input);
-    } catch {
-      input = row.input;
-    }
-  }
-  return {
-    ...row,
-    input
-  };
+  const deserialized = { ...row };
+  deserialized.input = storage.safelyParseJSON(row.input);
+  deserialized.output = storage.safelyParseJSON(row.output);
+  deserialized.scorer = storage.safelyParseJSON(row.scorer);
+  deserialized.preprocessStepResult = storage.safelyParseJSON(row.preprocessStepResult);
+  deserialized.analyzeStepResult = storage.safelyParseJSON(row.analyzeStepResult);
+  deserialized.metadata = storage.safelyParseJSON(row.metadata);
+  deserialized.additionalContext = storage.safelyParseJSON(row.additionalContext);
+  deserialized.runtimeContext = storage.safelyParseJSON(row.runtimeContext);
+  deserialized.entity = storage.safelyParseJSON(row.entity);
+  return deserialized;
 }
 var ScoresStorageCloudflare = class extends storage.ScoresStorage {
   operations;
@@ -1577,10 +1621,24 @@ var ScoresStorageCloudflare = class extends storage.ScoresStorage {
     }
   }
   async saveScore(score) {
+    let parsedScore;
     try {
-      const { input, ...rest } = score;
+      parsedScore = scores.saveScorePayloadSchema.parse(score);
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "CLOUDFLARE_STORAGE_SAVE_SCORE_FAILED_INVALID_SCORE_PAYLOAD",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.USER,
+          details: { scoreId: score.id }
+        },
+        error$1
+      );
+    }
+    try {
+      const id = crypto.randomUUID();
       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);
@@ -1591,12 +1649,12 @@ var ScoresStorageCloudflare = class extends storage.ScoresStorage {
           serializedRecord[key] = null;
         }
       }
-      serializedRecord.input = JSON.stringify(input);
+      serializedRecord.id = id;
       serializedRecord.createdAt = (/* @__PURE__ */ new Date()).toISOString();
       serializedRecord.updatedAt = (/* @__PURE__ */ new Date()).toISOString();
       await this.operations.putKV({
         tableName: storage.TABLE_SCORERS,
-        key: score.id,
+        key: id,
         value: serializedRecord
       });
       const scoreFromDb = await this.getScoreById({ id: score.id });
@@ -1618,6 +1676,9 @@ var ScoresStorageCloudflare = class extends storage.ScoresStorage {
   }
   async getScoresByScorerId({
     scorerId,
+    entityId,
+    entityType,
+    source,
     pagination
   }) {
     try {
@@ -1625,6 +1686,15 @@ var ScoresStorageCloudflare = class extends storage.ScoresStorage {
       const scores = [];
       for (const { name: key } of keys) {
         const score = await this.operations.getKV(storage.TABLE_SCORERS, key);
+        if (entityId && score.entityId !== entityId) {
+          continue;
+        }
+        if (entityType && score.entityType !== entityType) {
+          continue;
+        }
+        if (source && score.source !== source) {
+          continue;
+        }
         if (score && score.scorerId === scorerId) {
           scores.push(transformScoreRow(score));
         }
@@ -1755,6 +1825,50 @@ var ScoresStorageCloudflare = class extends storage.ScoresStorage {
       return { pagination: { total: 0, page: 0, perPage: 100, hasMore: false }, scores: [] };
     }
   }
+  async getScoresBySpan({
+    traceId,
+    spanId,
+    pagination
+  }) {
+    try {
+      const keys = await this.operations.listKV(storage.TABLE_SCORERS);
+      const scores = [];
+      for (const { name: key } of keys) {
+        const score = await this.operations.getKV(storage.TABLE_SCORERS, key);
+        if (score && score.traceId === traceId && score.spanId === spanId) {
+          scores.push(transformScoreRow(score));
+        }
+      }
+      scores.sort((a, b) => {
+        const dateA = new Date(a.createdAt || 0).getTime();
+        const dateB = new Date(b.createdAt || 0).getTime();
+        return dateB - dateA;
+      });
+      const total = scores.length;
+      const start = pagination.page * pagination.perPage;
+      const end = start + pagination.perPage;
+      const pagedScores = scores.slice(start, end);
+      return {
+        pagination: {
+          total,
+          page: pagination.page,
+          perPage: pagination.perPage,
+          hasMore: end < total
+        },
+        scores: pagedScores
+      };
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "CLOUDFLARE_STORAGE_SCORES_GET_SCORES_BY_SPAN_FAILED",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY,
+          text: `Failed to get scores by span: traceId=${traceId}, spanId=${spanId}`
+        },
+        error$1
+      );
+    }
+  }
 };
 var TracesStorageCloudflare = class extends storage.TracesStorage {
   operations;
@@ -1888,15 +2002,32 @@ var WorkflowsStorageCloudflare = class extends storage.WorkflowsStorage {
       throw new Error("Invalid workflow snapshot parameters");
     }
   }
+  updateWorkflowResults({
+    // workflowName,
+    // runId,
+    // stepId,
+    // result,
+    // runtimeContext,
+  }) {
+    throw new Error("Method not implemented.");
+  }
+  updateWorkflowState({
+    // workflowName,
+    // runId,
+    // opts,
+  }) {
+    throw new Error("Method not implemented.");
+  }
   async persistWorkflowSnapshot(params) {
     try {
-      const { workflowName, runId, snapshot } = params;
+      const { workflowName, runId, resourceId, snapshot } = params;
       await this.operations.putKV({
         tableName: storage.TABLE_WORKFLOW_SNAPSHOT,
         key: this.operations.getKey(storage.TABLE_WORKFLOW_SNAPSHOT, { workflow_name: workflowName, run_id: runId }),
         value: {
           workflow_name: workflowName,
           run_id: runId,
+          resourceId,
           snapshot: typeof snapshot === "string" ? snapshot : JSON.stringify(snapshot),
           createdAt: /* @__PURE__ */ new Date(),
           updatedAt: /* @__PURE__ */ new Date()
@@ -2128,6 +2259,11 @@ var CloudflareStore = class extends storage.MastraStorage {
       throw new Error("apiToken is required for REST API");
     }
   }
+  get supports() {
+    const supports = super.supports;
+    supports.getScoresBySpan = true;
+    return supports;
+  }
   constructor(config) {
     super({ name: "Cloudflare" });
     try {
@@ -2239,6 +2375,28 @@ var CloudflareStore = class extends storage.MastraStorage {
   }) {
     return this.stores.memory.getMessages({ threadId, resourceId, 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 getMessagesById({
+    messageIds,
+    format
+  }) {
+    return this.stores.memory.getMessagesById({ messageIds, format });
+  }
   async persistWorkflowSnapshot(params) {
     return this.stores.workflows.persistWorkflowSnapshot(params);
   }
@@ -2329,9 +2487,19 @@ var CloudflareStore = class extends storage.MastraStorage {
   }
   async getScoresByScorerId({
     scorerId,
+    entityId,
+    entityType,
+    source,
+    pagination
+  }) {
+    return this.stores.scores.getScoresByScorerId({ scorerId, entityId, entityType, source, pagination });
+  }
+  async getScoresBySpan({
+    traceId,
+    spanId,
     pagination
   }) {
-    return this.stores.scores.getScoresByScorerId({ scorerId, pagination });
+    return this.stores.scores.getScoresBySpan({ traceId, spanId, pagination });
   }
   async getResourceById({ resourceId }) {
     return this.stores.memory.getResourceById({ resourceId });