@mastra/cloudflare-d1 0.0.0-zod-v4-compat-part-2-20250822105954 → 0.0.0-zod-v4-stuff-20250825154219

package/dist/index.cjs

@@ -926,6 +926,47 @@ var MemoryStorageD1 = class extends storage.MemoryStorage {
       throw mastraError;
     }
   }
+  async getMessagesById({
+    messageIds,
+    format
+  }) {
+    if (messageIds.length === 0) return [];
+    const fullTableName = this.operations.getTableName(storage.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 agent.MessageList().add(processedMessages, "memory");
+      if (format === `v1`) return list.get.all.v1();
+      return list.get.all.v2();
+    } catch (error$1) {
+      const mastraError = new error.MastraError(
+        {
+          id: "CLOUDFLARE_D1_STORAGE_GET_MESSAGES_BY_ID_ERROR",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY,
+          text: `Failed to retrieve messages by ID: ${error$1 instanceof Error ? error$1.message : String(error$1)}`,
+          details: { messageIds: JSON.stringify(messageIds) }
+        },
+        error$1
+      );
+      this.logger?.error(mastraError.toString());
+      this.logger?.trackException(mastraError);
+      throw mastraError;
+    }
+  }
   async getMessagesPaginated({
     threadId,
     selectBy,
@@ -1443,6 +1484,7 @@ var StoreOperationsD1 = class extends storage.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 });
@@ -1527,20 +1569,19 @@ var StoreOperationsD1 = 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,
-    createdAt: row.createdAtZ || row.createdAt,
-    updatedAt: row.updatedAtZ || row.updatedAt
-  };
+  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);
+  deserialized.createdAt = row.createdAtZ || row.createdAt;
+  deserialized.updatedAt = row.updatedAtZ || row.updatedAt;
+  return deserialized;
 }
 var ScoresStorageD1 = class extends storage.ScoresStorage {
   operations;
@@ -1571,6 +1612,7 @@ var ScoresStorageD1 = class extends storage.ScoresStorage {
   }
   async saveScore(score) {
     try {
+      const id = crypto.randomUUID();
       const fullTableName = this.operations.getTableName(storage.TABLE_SCORERS);
       const { input, ...rest } = score;
       const serializedRecord = {};
@@ -1585,6 +1627,7 @@ var ScoresStorageD1 = class extends storage.ScoresStorage {
           serializedRecord[key] = null;
         }
       }
+      serializedRecord.id = id;
       serializedRecord.input = JSON.stringify(input);
       serializedRecord.createdAt = (/* @__PURE__ */ new Date()).toISOString();
       serializedRecord.updatedAt = (/* @__PURE__ */ new Date()).toISOString();
@@ -1593,7 +1636,7 @@ var ScoresStorageD1 = class extends storage.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$1) {
       throw new error.MastraError(
@@ -1608,11 +1651,23 @@ var ScoresStorageD1 = class extends storage.ScoresStorage {
   }
   async getScoresByScorerId({
     scorerId,
+    entityId,
+    entityType,
+    source,
     pagination
   }) {
     try {
       const fullTableName = this.operations.getTableName(storage.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) {
@@ -1626,7 +1681,17 @@ var ScoresStorageD1 = class extends storage.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) : [];
@@ -2223,6 +2288,12 @@ var D1Store = class extends storage.MastraStorage {
   }) {
     return this.stores.memory.getMessages({ threadId, selectBy, format });
   }
+  async getMessagesById({
+    messageIds,
+    format
+  }) {
+    return this.stores.memory.getMessagesById({ messageIds, format });
+  }
   async getMessagesPaginated({
     threadId,
     selectBy,
@@ -2322,10 +2393,13 @@ var D1Store = class extends storage.MastraStorage {
     });
   }
   async getScoresByScorerId({
-    scorerId: _scorerId,
-    pagination: _pagination
+    scorerId,
+    pagination,
+    entityId,
+    entityType,
+    source
   }) {
-    return this.stores.scores.getScoresByScorerId({ scorerId: _scorerId, pagination: _pagination });
+    return this.stores.scores.getScoresByScorerId({ scorerId, pagination, entityId, entityType, source });
   }
   /**
    * Close the database connection