@mastra/lance 1.0.0-beta.4 → 1.0.0-beta.6

package/dist/index.js

@@ -1243,35 +1243,43 @@ var StoreScoresLance = class extends ScoresStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: createStorageErrorId("LANCE", "SAVE_SCORE", "INVALID_PAYLOAD"),
+          id: createStorageErrorId("LANCE", "SAVE_SCORE", "VALIDATION_FAILED"),
           text: "Failed to save score in LanceStorage",
           domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.THIRD_PARTY
+          category: ErrorCategory.USER,
+          details: {
+            scorer: score.scorer?.id ?? "unknown",
+            entityId: score.entityId ?? "unknown",
+            entityType: score.entityType ?? "unknown",
+            traceId: score.traceId ?? "",
+            spanId: score.spanId ?? ""
+          }
         },
         error
       );
     }
+    const id = crypto.randomUUID();
+    const now = /* @__PURE__ */ new Date();
     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));
       const filteredScore = {};
-      Object.keys(validatedScore).forEach((key) => {
+      for (const key of Object.keys(validatedScore)) {
         if (allowedFields.has(key)) {
-          filteredScore[key] = score[key];
+          filteredScore[key] = validatedScore[key];
         }
-      });
+      }
       for (const key in filteredScore) {
         if (filteredScore[key] !== null && typeof filteredScore[key] === "object" && !(filteredScore[key] instanceof Date)) {
           filteredScore[key] = JSON.stringify(filteredScore[key]);
         }
       }
-      filteredScore.createdAt = /* @__PURE__ */ new Date();
-      filteredScore.updatedAt = /* @__PURE__ */ new Date();
       filteredScore.id = id;
+      filteredScore.createdAt = now;
+      filteredScore.updatedAt = now;
       await table.add([filteredScore], { mode: "append" });
-      return { score };
+      return { score: { ...validatedScore, id, createdAt: now, updatedAt: now } };
     } catch (error) {
       throw new MastraError(
         {
@@ -1635,6 +1643,23 @@ var StoreWorkflowsLance = class extends WorkflowsStorage {
       );
     }
   }
+  async deleteWorkflowRunById({ runId, workflowName }) {
+    try {
+      const table = await this.client.openTable(TABLE_WORKFLOW_SNAPSHOT);
+      const whereClause = `run_id = '${runId.replace(/'/g, "''")}' AND workflow_name = '${workflowName.replace(/'/g, "''")}'`;
+      await table.delete(whereClause);
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: createStorageErrorId("LANCE", "DELETE_WORKFLOW_RUN_BY_ID", "FAILED"),
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { runId, workflowName }
+        },
+        error
+      );
+    }
+  }
   async listWorkflowRuns(args) {
     try {
       const table = await this.client.openTable(TABLE_WORKFLOW_SNAPSHOT);
@@ -1907,6 +1932,9 @@ var LanceStorage = class _LanceStorage extends MastraStorage {
   async getWorkflowRunById(args) {
     return this.stores.workflows.getWorkflowRunById(args);
   }
+  async deleteWorkflowRunById({ runId, workflowName }) {
+    return this.stores.workflows.deleteWorkflowRunById({ runId, workflowName });
+  }
   async updateWorkflowResults({
     workflowName,
     runId,
@@ -1949,8 +1977,8 @@ var LanceStorage = class _LanceStorage extends MastraStorage {
   }) {
     return this.stores.scores.listScoresByScorerId({ scorerId, source, pagination, entityId, entityType });
   }
-  async saveScore(_score) {
-    return this.stores.scores.saveScore(_score);
+  async saveScore(score) {
+    return this.stores.scores.saveScore(score);
   }
   async listScoresByRunId({
     runId,