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

package/dist/index.cjs

@@ -5,6 +5,7 @@ var error = require('@mastra/core/error');
 var storage = require('@mastra/core/storage');
 var agent = require('@mastra/core/agent');
 var apacheArrow = require('apache-arrow');
+var scores = require('@mastra/core/scores');
 var vector = require('@mastra/core/vector');
 var filter = require('@mastra/core/vector/filter');
 
@@ -189,7 +190,6 @@ function processResultWithTypeConversion(rawResult, tableSchema) {
     } else if (fieldTypeStr.includes("float64") && ["createdAt", "updatedAt"].includes(key)) {
       processedResult[key] = new Date(processedResult[key]);
     }
-    console.log(key, "processedResult", processedResult);
   }
   return processedResult;
 }
@@ -381,6 +381,20 @@ var StoreMemoryLance = class extends storage.MemoryStorage {
       );
     }
   }
+  normalizeMessage(message) {
+    const { thread_id, ...rest } = message;
+    return {
+      ...rest,
+      threadId: thread_id,
+      content: typeof message.content === "string" ? (() => {
+        try {
+          return JSON.parse(message.content);
+        } catch {
+          return message.content;
+        }
+      })() : message.content
+    };
+  }
   async getMessages({
     threadId,
     resourceId,
@@ -389,6 +403,7 @@ var StoreMemoryLance = class extends storage.MemoryStorage {
     threadConfig
   }) {
     try {
+      if (!threadId.trim()) throw new Error("threadId must be a non-empty string");
       if (threadConfig) {
         throw new Error("ThreadConfig is not supported by LanceDB storage");
       }
@@ -421,21 +436,7 @@ var StoreMemoryLance = class extends storage.MemoryStorage {
         allRecords,
         await getTableSchema({ tableName: storage.TABLE_MESSAGES, client: this.client })
       );
-      const normalized = messages.map((msg) => {
-        const { thread_id, ...rest } = msg;
-        return {
-          ...rest,
-          threadId: thread_id,
-          content: typeof msg.content === "string" ? (() => {
-            try {
-              return JSON.parse(msg.content);
-            } catch {
-              return msg.content;
-            }
-          })() : msg.content
-        };
-      });
-      const list = new agent.MessageList({ threadId, resourceId }).add(normalized, "memory");
+      const list = new agent.MessageList({ threadId, resourceId }).add(messages.map(this.normalizeMessage), "memory");
       if (format === "v2") return list.get.all.v2();
       return list.get.all.v1();
     } catch (error$1) {
@@ -443,7 +444,41 @@ var StoreMemoryLance = class extends storage.MemoryStorage {
         {
           id: "LANCE_STORE_GET_MESSAGES_FAILED",
           domain: error.ErrorDomain.STORAGE,
-          category: error.ErrorCategory.THIRD_PARTY
+          category: error.ErrorCategory.THIRD_PARTY,
+          details: {
+            threadId,
+            resourceId: resourceId ?? ""
+          }
+        },
+        error$1
+      );
+    }
+  }
+  async getMessagesById({
+    messageIds,
+    format
+  }) {
+    if (messageIds.length === 0) return [];
+    try {
+      const table = await this.client.openTable(storage.TABLE_MESSAGES);
+      const quotedIds = messageIds.map((id) => `'${id}'`).join(", ");
+      const allRecords = await table.query().where(`id IN (${quotedIds})`).toArray();
+      const messages = processResultWithTypeConversion(
+        allRecords,
+        await getTableSchema({ tableName: storage.TABLE_MESSAGES, client: this.client })
+      );
+      const list = new agent.MessageList().add(messages.map(this.normalizeMessage), "memory");
+      if (format === `v1`) return list.get.all.v1();
+      return list.get.all.v2();
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "LANCE_STORE_GET_MESSAGES_BY_ID_FAILED",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY,
+          details: {
+            messageIds: JSON.stringify(messageIds)
+          }
         },
         error$1
       );
@@ -584,13 +619,11 @@ var StoreMemoryLance = class extends storage.MemoryStorage {
     return Array.from(allIndices).sort((a, b) => a - b).map((index) => records[index]);
   }
   async getMessagesPaginated(args) {
+    const { threadId, resourceId, selectBy, format = "v1" } = args;
+    const page = selectBy?.pagination?.page ?? 0;
+    const perPage = selectBy?.pagination?.perPage ?? 10;
     try {
-      const { threadId, resourceId, selectBy, format = "v1" } = args;
-      if (!threadId) {
-        throw new Error("Thread ID is required for getMessagesPaginated");
-      }
-      const page = selectBy?.pagination?.page ?? 0;
-      const perPage = selectBy?.pagination?.perPage ?? 10;
+      if (!threadId.trim()) throw new Error("threadId must be a non-empty string");
       const dateRange = selectBy?.pagination?.dateRange;
       const fromDate = dateRange?.start;
       const toDate = dateRange?.end;
@@ -692,14 +725,21 @@ var StoreMemoryLance = class extends storage.MemoryStorage {
         hasMore: total > (page + 1) * perPage
       };
     } catch (error$1) {
-      throw new error.MastraError(
+      const mastraError = new error.MastraError(
         {
           id: "LANCE_STORE_GET_MESSAGES_PAGINATED_FAILED",
           domain: error.ErrorDomain.STORAGE,
-          category: error.ErrorCategory.THIRD_PARTY
+          category: error.ErrorCategory.THIRD_PARTY,
+          details: {
+            threadId,
+            resourceId: resourceId ?? ""
+          }
         },
         error$1
       );
+      this.logger?.trackException?.(mastraError);
+      this.logger?.error?.(mastraError.toString());
+      return { messages: [], total: 0, page, perPage, hasMore: false };
     }
   }
   /**
@@ -1226,7 +1266,7 @@ var StoreOperationsLance = class extends storage.StoreOperations {
           processedRecord[key] = JSON.stringify(processedRecord[key]);
         }
       }
-      console.log(await table.schema());
+      console.info(await table.schema());
       await table.mergeInsert(primaryId).whenMatchedUpdateAll().whenNotMatchedInsertAll().execute([processedRecord]);
     } catch (error$1) {
       throw new error.MastraError(
@@ -1276,7 +1316,6 @@ var StoreOperationsLance = class extends storage.StoreOperations {
         }
         return processedRecord;
       });
-      console.log(processedRecords);
       await table.mergeInsert(primaryId).whenMatchedUpdateAll().whenNotMatchedInsertAll().execute(processedRecords);
     } catch (error$1) {
       throw new error.MastraError(
@@ -1360,12 +1399,27 @@ var StoreScoresLance = class extends storage.ScoresStorage {
     this.client = client;
   }
   async saveScore(score) {
+    let validatedScore;
+    try {
+      validatedScore = scores.saveScorePayloadSchema.parse(score);
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "LANCE_STORAGE_SAVE_SCORE_FAILED",
+          text: "Failed to save score in LanceStorage",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY
+        },
+        error$1
+      );
+    }
     try {
+      const id = crypto.randomUUID();
       const table = await this.client.openTable(storage.TABLE_SCORERS);
       const schema = await getTableSchema({ tableName: storage.TABLE_SCORERS, client: this.client });
       const allowedFields = new Set(schema.fields.map((f) => f.name));
       const filteredScore = {};
-      Object.keys(score).forEach((key) => {
+      Object.keys(validatedScore).forEach((key) => {
         if (allowedFields.has(key)) {
           filteredScore[key] = score[key];
         }
@@ -1375,7 +1429,7 @@ var StoreScoresLance = class extends storage.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$1) {
@@ -1414,18 +1468,35 @@ var StoreScoresLance = class extends storage.ScoresStorage {
   }
   async getScoresByScorerId({
     scorerId,
-    pagination
+    pagination,
+    entityId,
+    entityType,
+    source
   }) {
     try {
       const table = await this.client.openTable(storage.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: storage.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: {
@@ -1524,6 +1595,44 @@ var StoreScoresLance = class extends storage.ScoresStorage {
       );
     }
   }
+  async getScoresBySpan({
+    traceId,
+    spanId,
+    pagination
+  }) {
+    try {
+      const table = await this.client.openTable(storage.TABLE_SCORERS);
+      const { page = 0, perPage = 10 } = pagination || {};
+      const offset = page * perPage;
+      const query = table.query().where(`\`traceId\` = '${traceId}' AND \`spanId\` = '${spanId}'`).limit(perPage);
+      if (offset > 0) query.offset(offset);
+      const records = await query.toArray();
+      const schema = await getTableSchema({ tableName: storage.TABLE_SCORERS, client: this.client });
+      const scores = processResultWithTypeConversion(records, schema);
+      const allRecords = await table.query().where(`\`traceId\` = '${traceId}' AND \`spanId\` = '${spanId}'`).toArray();
+      const total = allRecords.length;
+      return {
+        pagination: {
+          page,
+          perPage,
+          total,
+          hasMore: offset + scores.length < total
+        },
+        scores
+      };
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "LANCE_STORAGE_GET_SCORES_BY_SPAN_FAILED",
+          text: "Failed to get scores by traceId and spanId in LanceStorage",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY,
+          details: { error: error$1?.message }
+        },
+        error$1
+      );
+    }
+  }
 };
 var StoreTracesLance = class extends storage.TracesStorage {
   client;
@@ -1741,9 +1850,26 @@ var StoreWorkflowsLance = class extends storage.WorkflowsStorage {
     super();
     this.client = client;
   }
+  updateWorkflowResults({
+    // workflowName,
+    // runId,
+    // stepId,
+    // result,
+    // runtimeContext,
+  }) {
+    throw new Error("Method not implemented.");
+  }
+  updateWorkflowState({
+    // workflowName,
+    // runId,
+    // opts,
+  }) {
+    throw new Error("Method not implemented.");
+  }
   async persistWorkflowSnapshot({
     workflowName,
     runId,
+    resourceId,
     snapshot
   }) {
     try {
@@ -1760,6 +1886,7 @@ var StoreWorkflowsLance = class extends storage.WorkflowsStorage {
       const record = {
         workflow_name: workflowName,
         run_id: runId,
+        resourceId,
         snapshot: JSON.stringify(snapshot),
         createdAt,
         updatedAt: now
@@ -1998,7 +2125,8 @@ var LanceStorage = class _LanceStorage extends storage.MastraStorage {
       resourceWorkingMemory: true,
       hasColumn: true,
       createTable: true,
-      deleteMessages: false
+      deleteMessages: false,
+      getScoresBySpan: true
     };
   }
   async getResourceById({ resourceId }) {
@@ -2071,6 +2199,12 @@ var LanceStorage = class _LanceStorage extends storage.MastraStorage {
   }) {
     return this.stores.memory.getMessages({ threadId, resourceId, selectBy, format, threadConfig });
   }
+  async getMessagesById({
+    messageIds,
+    format
+  }) {
+    return this.stores.memory.getMessagesById({ messageIds, format });
+  }
   async saveMessages(args) {
     return this.stores.memory.saveMessages(args);
   }
@@ -2104,12 +2238,29 @@ var LanceStorage = class _LanceStorage extends storage.MastraStorage {
   async getWorkflowRunById(args) {
     return this.stores.workflows.getWorkflowRunById(args);
   }
+  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 persistWorkflowSnapshot({
     workflowName,
     runId,
+    resourceId,
     snapshot
   }) {
-    return this.stores.workflows.persistWorkflowSnapshot({ workflowName, runId, snapshot });
+    return this.stores.workflows.persistWorkflowSnapshot({ workflowName, runId, resourceId, snapshot });
   }
   async loadWorkflowSnapshot({
     workflowName,
@@ -2122,9 +2273,12 @@ var LanceStorage = class _LanceStorage extends storage.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);
@@ -2142,6 +2296,13 @@ var LanceStorage = class _LanceStorage extends storage.MastraStorage {
   }) {
     return this.stores.scores.getScoresByEntityId({ entityId, entityType, pagination });
   }
+  async getScoresBySpan({
+    traceId,
+    spanId,
+    pagination
+  }) {
+    return this.stores.scores.getScoresBySpan({ traceId, spanId, pagination });
+  }
 };
 var LanceFilterTranslator = class extends filter.BaseFilterTranslator {
   translate(filter) {