@mastra/cloudflare-d1 0.0.0-pgvector-index-fix-20250905222058 → 0.0.0-playground-studio-cloud-20251031080052

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, safelyParseJSON } from '@mastra/core/storage';
+import { MastraStorage, StoreOperations, ScoresStorage, TABLE_SCORERS, LegacyEvalsStorage, TABLE_EVALS, serializeDate, 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 {
@@ -241,16 +242,6 @@ function isArrayOfRecords(value) {
 }
 function deserializeValue(value, type) {
   if (value === null || value === void 0) return null;
-  if (type === "date" && typeof value === "string") {
-    return new Date(value);
-  }
-  if (type === "jsonb" && typeof value === "string") {
-    try {
-      return JSON.parse(value);
-    } catch {
-      return value;
-    }
-  }
   if (typeof value === "string" && (value.startsWith("{") || value.startsWith("["))) {
     try {
       return JSON.parse(value);
@@ -1609,12 +1600,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 +1630,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,128 +1805,53 @@ var ScoresStorageD1 = class extends ScoresStorage {
       );
     }
   }
-};
-function isArrayOfRecords2(value) {
-  return value && Array.isArray(value) && value.length > 0;
-}
-var TracesStorageD1 = class extends TracesStorage {
-  operations;
-  constructor({ operations }) {
-    super();
-    this.operations = operations;
-  }
-  async getTraces(args) {
-    const paginatedArgs = {
-      name: args.name,
-      scope: args.scope,
-      page: args.page,
-      perPage: args.perPage,
-      attributes: args.attributes,
-      filters: args.filters,
-      dateRange: args.fromDate || args.toDate ? {
-        start: args.fromDate,
-        end: args.toDate
-      } : void 0
-    };
-    try {
-      const result = await this.getTracesPaginated(paginatedArgs);
-      return result.traces;
-    } catch (error) {
-      throw new MastraError(
-        {
-          id: "CLOUDFLARE_D1_STORAGE_GET_TRACES_ERROR",
-          domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.THIRD_PARTY,
-          text: `Failed to retrieve traces: ${error instanceof Error ? error.message : String(error)}`,
-          details: {
-            name: args.name ?? "",
-            scope: args.scope ?? ""
-          }
-        },
-        error
-      );
-    }
-  }
-  async getTracesPaginated(args) {
-    const { name, scope, page = 0, perPage = 100, attributes, dateRange } = args;
-    const fromDate = dateRange?.start;
-    const toDate = dateRange?.end;
-    const fullTableName = this.operations.getTableName(TABLE_TRACES);
+  async getScoresBySpan({
+    traceId,
+    spanId,
+    pagination
+  }) {
     try {
-      const dataQuery = createSqlBuilder().select("*").from(fullTableName).where("1=1");
-      const countQuery = createSqlBuilder().count().from(fullTableName).where("1=1");
-      if (name) {
-        dataQuery.andWhere("name LIKE ?", `%${name}%`);
-        countQuery.andWhere("name LIKE ?", `%${name}%`);
-      }
-      if (scope) {
-        dataQuery.andWhere("scope = ?", scope);
-        countQuery.andWhere("scope = ?", scope);
-      }
-      if (attributes && Object.keys(attributes).length > 0) {
-        for (const [key, value] of Object.entries(attributes)) {
-          dataQuery.jsonLike("attributes", key, value);
-          countQuery.jsonLike("attributes", key, value);
-        }
-      }
-      if (fromDate) {
-        const fromDateStr = fromDate instanceof Date ? fromDate.toISOString() : fromDate;
-        dataQuery.andWhere("createdAt >= ?", fromDateStr);
-        countQuery.andWhere("createdAt >= ?", fromDateStr);
-      }
-      if (toDate) {
-        const toDateStr = toDate instanceof Date ? toDate.toISOString() : toDate;
-        dataQuery.andWhere("createdAt <= ?", toDateStr);
-        countQuery.andWhere("createdAt <= ?", toDateStr);
-      }
-      const allDataResult = await this.operations.executeQuery(
-        createSqlBuilder().select("*").from(fullTableName).where("1=1").build()
-      );
-      console.log("allDataResult", allDataResult);
+      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 = Number(countResult?.[0]?.count ?? 0);
-      dataQuery.orderBy("startTime", "DESC").limit(perPage).offset(page * perPage);
-      const results = await this.operations.executeQuery(dataQuery.build());
-      const traces = isArrayOfRecords2(results) ? results.map(
-        (trace) => ({
-          ...trace,
-          attributes: deserializeValue(trace.attributes, "jsonb"),
-          status: deserializeValue(trace.status, "jsonb"),
-          events: deserializeValue(trace.events, "jsonb"),
-          links: deserializeValue(trace.links, "jsonb"),
-          other: deserializeValue(trace.other, "jsonb")
-        })
-      ) : [];
+      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 {
-        traces,
-        total,
-        page,
-        perPage,
-        hasMore: page * perPage + traces.length < total
+        pagination: {
+          total,
+          page: pagination.page,
+          perPage: pagination.perPage,
+          hasMore: rows.length > pagination.perPage
+        },
+        scores
       };
     } catch (error) {
-      const mastraError = new MastraError(
+      throw new MastraError(
         {
-          id: "CLOUDFLARE_D1_STORAGE_GET_TRACES_PAGINATED_ERROR",
+          id: "CLOUDFLARE_D1_STORE_SCORES_GET_SCORES_BY_SPAN_FAILED",
           domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.THIRD_PARTY,
-          text: `Failed to retrieve traces: ${error instanceof Error ? error.message : String(error)}`,
-          details: { name: name ?? "", scope: scope ?? "" }
+          category: ErrorCategory.THIRD_PARTY
         },
         error
       );
-      this.logger?.error(mastraError.toString());
-      this.logger?.trackException(mastraError);
-      return { traces: [], total: 0, page, perPage, hasMore: false };
     }
   }
-  async batchTraceInsert({ records }) {
-    this.logger.debug("Batch inserting traces", { count: records.length });
-    await this.operations.batchInsert({
-      tableName: TABLE_TRACES,
-      records
-    });
-  }
 };
 var WorkflowsStorageD1 = class extends WorkflowsStorage {
   operations;
@@ -1950,6 +1878,7 @@ var WorkflowsStorageD1 = class extends WorkflowsStorage {
   async persistWorkflowSnapshot({
     workflowName,
     runId,
+    resourceId,
     snapshot
   }) {
     const fullTableName = this.operations.getTableName(TABLE_WORKFLOW_SNAPSHOT);
@@ -1960,11 +1889,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
@@ -2206,9 +2137,6 @@ var D1Store = class extends MastraStorage {
     const legacyEvals = new LegacyEvalsStorageD1({
       operations
     });
-    const traces = new TracesStorageD1({
-      operations
-    });
     const workflows = new WorkflowsStorageD1({
       operations
     });
@@ -2219,7 +2147,6 @@ var D1Store = class extends MastraStorage {
       operations,
       scores,
       legacyEvals,
-      traces,
       workflows,
       memory
     };
@@ -2230,7 +2157,8 @@ var D1Store = class extends MastraStorage {
       resourceWorkingMemory: true,
       hasColumn: true,
       createTable: true,
-      deleteMessages: false
+      deleteMessages: false,
+      getScoresBySpan: true
     };
   }
   async createTable({
@@ -2334,9 +2262,10 @@ var D1Store = class extends MastraStorage {
   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);
@@ -2365,15 +2294,6 @@ var D1Store = class extends MastraStorage {
   async batchInsert({ tableName, records }) {
     return this.stores.operations.batchInsert({ tableName, records });
   }
-  /**
-   * @deprecated use getTracesPaginated instead
-   */
-  async getTraces(args) {
-    return this.stores.traces.getTraces(args);
-  }
-  async getTracesPaginated(args) {
-    return this.stores.traces.getTracesPaginated(args);
-  }
   /**
    * @deprecated use getEvals instead
    */
@@ -2431,6 +2351,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