@mastra/lance 0.2.4 → 0.2.5

package/dist/index.js

@@ -1359,6 +1359,7 @@ var StoreScoresLance = class extends ScoresStorage {
   }
   async saveScore(score) {
     try {
+      const id = crypto.randomUUID();
       const table = await this.client.openTable(TABLE_SCORERS);
       const schema = await getTableSchema({ tableName: TABLE_SCORERS, client: this.client });
       const allowedFields = new Set(schema.fields.map((f) => f.name));
@@ -1373,7 +1374,7 @@ var StoreScoresLance = class extends ScoresStorage {
           filteredScore[key] = JSON.stringify(filteredScore[key]);
         }
       }
-      console.log("Saving score to LanceStorage:", filteredScore);
+      filteredScore.id = id;
       await table.add([filteredScore], { mode: "append" });
       return { score };
     } catch (error) {
@@ -1412,18 +1413,35 @@ var StoreScoresLance = class extends ScoresStorage {
   }
   async getScoresByScorerId({
     scorerId,
-    pagination
+    pagination,
+    entityId,
+    entityType,
+    source
   }) {
     try {
       const table = await this.client.openTable(TABLE_SCORERS);
       const { page = 0, perPage = 10 } = pagination || {};
       const offset = page * perPage;
-      const query = table.query().where(`\`scorerId\` = '${scorerId}'`).limit(perPage);
+      let query = table.query().where(`\`scorerId\` = '${scorerId}'`);
+      if (source) {
+        query = query.where(`\`source\` = '${source}'`);
+      }
+      if (entityId) {
+        query = query.where(`\`entityId\` = '${entityId}'`);
+      }
+      if (entityType) {
+        query = query.where(`\`entityType\` = '${entityType}'`);
+      }
+      query = query.limit(perPage);
       if (offset > 0) query.offset(offset);
       const records = await query.toArray();
       const schema = await getTableSchema({ tableName: TABLE_SCORERS, client: this.client });
       const scores = processResultWithTypeConversion(records, schema);
-      const allRecords = await table.query().where(`\`scorerId\` = '${scorerId}'`).toArray();
+      let totalQuery = table.query().where(`\`scorerId\` = '${scorerId}'`);
+      if (source) {
+        totalQuery = totalQuery.where(`\`source\` = '${source}'`);
+      }
+      const allRecords = await totalQuery.toArray();
       const total = allRecords.length;
       return {
         pagination: {
@@ -2120,9 +2138,12 @@ var LanceStorage = class _LanceStorage extends MastraStorage {
   }
   async getScoresByScorerId({
     scorerId,
+    source,
+    entityId,
+    entityType,
     pagination
   }) {
-    return this.stores.scores.getScoresByScorerId({ scorerId, pagination });
+    return this.stores.scores.getScoresByScorerId({ scorerId, source, pagination, entityId, entityType });
   }
   async saveScore(_score) {
     return this.stores.scores.saveScore(_score);