@mastra/cloudflare-d1 0.0.0-roamin-openaivoice-speak-options-passing-20250926163614 → 0.0.0-safe-stringify-telemetry-20251205024938

package/dist/index.js

@@ -3,6 +3,7 @@ import { MastraStorage, StoreOperations, ScoresStorage, TABLE_SCORERS, LegacyEva
 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 {
@@ -1045,17 +1046,16 @@ var MemoryStorageD1 = class extends MemoryStorage {
         }
         return processedMsg;
       });
-      if (selectBy?.last) {
-        processedMessages.sort((a, b) => new Date(a.createdAt).getTime() - new Date(b.createdAt).getTime());
-      }
-      const list = new MessageList().add(processedMessages, "memory");
-      messages.push(...format === `v2` ? list.get.all.v2() : list.get.all.v1());
+      const allMessages = [...messages, ...processedMessages];
+      const list = new MessageList().add(allMessages, "memory");
+      let finalMessages = format === `v2` ? list.get.all.v2() : list.get.all.v1();
+      finalMessages = finalMessages.sort((a, b) => new Date(a.createdAt).getTime() - new Date(b.createdAt).getTime());
       return {
-        messages,
+        messages: finalMessages,
         total,
         page,
         perPage,
-        hasMore: selectBy?.last ? false : page * perPage + messages.length < total
+        hasMore: selectBy?.last ? false : page * perPage + processedMessages.length < total
       };
     } catch (error) {
       const mastraError = new MastraError(
@@ -1568,15 +1568,15 @@ var StoreOperationsD1 = class extends StoreOperations {
 };
 function transformScoreRow(row) {
   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.input = row.input ? safelyParseJSON(row.input) : void 0;
+  deserialized.output = row.output ? safelyParseJSON(row.output) : void 0;
+  deserialized.scorer = row.scorer ? safelyParseJSON(row.scorer) : void 0;
+  deserialized.preprocessStepResult = row.preprocessStepResult ? safelyParseJSON(row.preprocessStepResult) : void 0;
+  deserialized.analyzeStepResult = row.analyzeStepResult ? safelyParseJSON(row.analyzeStepResult) : void 0;
+  deserialized.metadata = row.metadata ? safelyParseJSON(row.metadata) : void 0;
+  deserialized.additionalContext = row.additionalContext ? safelyParseJSON(row.additionalContext) : void 0;
+  deserialized.runtimeContext = row.runtimeContext ? safelyParseJSON(row.runtimeContext) : void 0;
+  deserialized.entity = row.entity ? safelyParseJSON(row.entity) : void 0;
   deserialized.createdAt = row.createdAtZ || row.createdAt;
   deserialized.updatedAt = row.updatedAtZ || row.updatedAt;
   return deserialized;
@@ -1609,12 +1609,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);
@@ -1626,7 +1639,6 @@ var ScoresStorageD1 = class extends ScoresStorage {
         }
       }
       serializedRecord.id = id;
-      serializedRecord.input = JSON.stringify(input);
       serializedRecord.createdAt = (/* @__PURE__ */ new Date()).toISOString();
       serializedRecord.updatedAt = (/* @__PURE__ */ new Date()).toISOString();
       const columns = Object.keys(serializedRecord);
@@ -1802,6 +1814,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;
@@ -2046,13 +2105,18 @@ var WorkflowsStorageD1 = class extends WorkflowsStorage {
     toDate,
     limit,
     offset,
-    resourceId
+    resourceId,
+    status
   } = {}) {
     const fullTableName = this.operations.getTableName(TABLE_WORKFLOW_SNAPSHOT);
     try {
       const builder = createSqlBuilder().select().from(fullTableName);
       const countBuilder = createSqlBuilder().count().from(fullTableName);
       if (workflowName) builder.whereAnd("workflow_name = ?", workflowName);
+      if (status) {
+        builder.whereAnd("json_extract(snapshot, '$.status') = ?", status);
+        countBuilder.whereAnd("json_extract(snapshot, '$.status') = ?", status);
+      }
       if (resourceId) {
         const hasResourceId = await this.operations.hasColumn(fullTableName, "resourceId");
         if (hasResourceId) {
@@ -2233,7 +2297,8 @@ var D1Store = class extends MastraStorage {
       resourceWorkingMemory: true,
       hasColumn: true,
       createTable: true,
-      deleteMessages: false
+      deleteMessages: false,
+      getScoresBySpan: true
     };
   }
   async createTable({
@@ -2345,15 +2410,8 @@ var D1Store = class extends MastraStorage {
   async loadWorkflowSnapshot(params) {
     return this.stores.workflows.loadWorkflowSnapshot(params);
   }
-  async getWorkflowRuns({
-    workflowName,
-    fromDate,
-    toDate,
-    limit,
-    offset,
-    resourceId
-  } = {}) {
-    return this.stores.workflows.getWorkflowRuns({ workflowName, fromDate, toDate, limit, offset, resourceId });
+  async getWorkflowRuns(args = {}) {
+    return this.stores.workflows.getWorkflowRuns(args);
   }
   async getWorkflowRunById({
     runId,
@@ -2435,6 +2493,13 @@ var D1Store = class extends MastraStorage {
   }) {
     return this.stores.scores.getScoresByScorerId({ scorerId, pagination, entityId, entityType, source });
   }
+  async getScoresBySpan({
+    traceId,
+    spanId,
+    pagination
+  }) {
+    return this.stores.scores.getScoresBySpan({ traceId, spanId, pagination });
+  }
   /**
    * Close the database connection
    * No explicit cleanup needed for D1 in either REST or Workers Binding mode