@mastra/lance 0.0.0-tsconfig-compile-20250703214351 → 0.0.0-workflow-deno-20250616130925

package/src/storage/index.ts

@@ -1,8 +1,6 @@
 import { connect } from '@lancedb/lancedb';
 import type { Connection, ConnectionOptions, SchemaLike, FieldLike } from '@lancedb/lancedb';
-import type { MastraMessageContentV2 } from '@mastra/core/agent';
 import { MessageList } from '@mastra/core/agent';
-import { ErrorCategory, ErrorDomain, MastraError } from '@mastra/core/error';
 import type { MastraMessageV1, MastraMessageV2, StorageThreadType, TraceType } from '@mastra/core/memory';
 import {
   MastraStorage,
@@ -58,30 +56,10 @@ export class LanceStorage extends MastraStorage {
       instance.lanceClient = await connect(uri, options);
       return instance;
     } catch (e: any) {
-      throw new MastraError(
-        {
-          id: 'STORAGE_LANCE_STORAGE_CONNECT_FAILED',
-          domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.THIRD_PARTY,
-          text: `Failed to connect to LanceDB: ${e.message || e}`,
-          details: { uri, optionsProvided: !!options },
-        },
-        e,
-      );
+      throw new Error(`Failed to connect to LanceDB: ${e}`);
     }
   }
 
-  private getPrimaryKeys(tableName: TABLE_NAMES): string[] {
-    let primaryId: string[] = ['id'];
-    if (tableName === TABLE_WORKFLOW_SNAPSHOT) {
-      primaryId = ['workflow_name', 'run_id'];
-    } else if (tableName === TABLE_EVALS) {
-      primaryId = ['agent_name', 'metric_name', 'run_id'];
-    }
-
-    return primaryId;
-  }
-
   /**
    * @internal
    * Private constructor to enforce using the create factory method
@@ -97,41 +75,11 @@ export class LanceStorage extends MastraStorage {
     tableName: TABLE_NAMES;
     schema: Record<string, StorageColumn>;
   }): Promise<void> {
-    try {
-      if (!this.lanceClient) {
-        throw new Error('LanceDB client not initialized. Call LanceStorage.create() first.');
-      }
-      if (!tableName) {
-        throw new Error('tableName is required for createTable.');
-      }
-      if (!schema) {
-        throw new Error('schema is required for createTable.');
-      }
-    } catch (error) {
-      throw new MastraError(
-        {
-          id: 'STORAGE_LANCE_STORAGE_CREATE_TABLE_INVALID_ARGS',
-          domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.USER,
-          details: { tableName },
-        },
-        error,
-      );
-    }
-
     try {
       const arrowSchema = this.translateSchema(schema);
       await this.lanceClient.createEmptyTable(tableName, arrowSchema);
     } catch (error: any) {
-      throw new MastraError(
-        {
-          id: 'STORAGE_LANCE_STORAGE_CREATE_TABLE_FAILED',
-          domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.THIRD_PARTY,
-          details: { tableName },
-        },
-        error,
-      );
+      throw new Error(`Failed to create table: ${error}`);
     }
   }
 
@@ -181,42 +129,15 @@ export class LanceStorage extends MastraStorage {
    * @param tableName Name of the table to drop
    */
   async dropTable(tableName: TABLE_NAMES): Promise<void> {
-    try {
-      if (!this.lanceClient) {
-        throw new Error('LanceDB client not initialized. Call LanceStorage.create() first.');
-      }
-      if (!tableName) {
-        throw new Error('tableName is required for dropTable.');
-      }
-    } catch (validationError: any) {
-      throw new MastraError(
-        {
-          id: 'STORAGE_LANCE_STORAGE_DROP_TABLE_INVALID_ARGS',
-          domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.USER,
-          text: validationError.message,
-          details: { tableName },
-        },
-        validationError,
-      );
-    }
-
     try {
       await this.lanceClient.dropTable(tableName);
     } catch (error: any) {
-      if (error.toString().includes('was not found') || error.message?.includes('Table not found')) {
+      // Don't throw if the table doesn't exist
+      if (error.toString().includes('was not found')) {
         this.logger.debug(`Table '${tableName}' does not exist, skipping drop`);
         return;
       }
-      throw new MastraError(
-        {
-          id: 'STORAGE_LANCE_STORAGE_DROP_TABLE_FAILED',
-          domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.THIRD_PARTY,
-          details: { tableName },
-        },
-        error,
-      );
+      throw new Error(`Failed to drop table: ${error}`);
     }
   }
 
@@ -226,26 +147,6 @@ export class LanceStorage extends MastraStorage {
    * @returns Table schema
    */
   async getTableSchema(tableName: TABLE_NAMES): Promise<SchemaLike> {
-    try {
-      if (!this.lanceClient) {
-        throw new Error('LanceDB client not initialized. Call LanceStorage.create() first.');
-      }
-      if (!tableName) {
-        throw new Error('tableName is required for getTableSchema.');
-      }
-    } catch (validationError: any) {
-      throw new MastraError(
-        {
-          id: 'STORAGE_LANCE_STORAGE_GET_TABLE_SCHEMA_INVALID_ARGS',
-          domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.USER,
-          text: validationError.message,
-          details: { tableName },
-        },
-        validationError,
-      );
-    }
-
     try {
       const table = await this.lanceClient.openTable(tableName);
       const rawSchema = await table.schema();
@@ -260,15 +161,7 @@ export class LanceStorage extends MastraStorage {
         },
       };
     } catch (error: any) {
-      throw new MastraError(
-        {
-          id: 'STORAGE_LANCE_STORAGE_GET_TABLE_SCHEMA_FAILED',
-          domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.THIRD_PARTY,
-          details: { tableName },
-        },
-        error,
-      );
+      throw new Error(`Failed to get table schema: ${error}`);
     }
   }
 
@@ -305,114 +198,43 @@ export class LanceStorage extends MastraStorage {
     schema: Record<string, StorageColumn>;
     ifNotExists: string[];
   }): Promise<void> {
-    try {
-      if (!this.lanceClient) {
-        throw new Error('LanceDB client not initialized. Call LanceStorage.create() first.');
-      }
-      if (!tableName) {
-        throw new Error('tableName is required for alterTable.');
-      }
-      if (!schema) {
-        throw new Error('schema is required for alterTable.');
-      }
-      if (!ifNotExists || ifNotExists.length === 0) {
-        this.logger.debug('No columns specified to add in alterTable, skipping.');
-        return;
-      }
-    } catch (validationError: any) {
-      throw new MastraError(
-        {
-          id: 'STORAGE_LANCE_STORAGE_ALTER_TABLE_INVALID_ARGS',
-          domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.USER,
-          text: validationError.message,
-          details: { tableName },
-        },
-        validationError,
-      );
-    }
+    const table = await this.lanceClient.openTable(tableName);
+    const currentSchema = await table.schema();
+    const existingFields = new Set(currentSchema.fields.map((f: any) => f.name));
+
+    const typeMap: Record<string, string> = {
+      text: 'string',
+      integer: 'int',
+      bigint: 'bigint',
+      timestamp: 'timestamp',
+      jsonb: 'string',
+      uuid: 'string',
+    };
 
-    try {
-      const table = await this.lanceClient.openTable(tableName);
-      const currentSchema = await table.schema();
-      const existingFields = new Set(currentSchema.fields.map((f: any) => f.name));
-
-      const typeMap: Record<string, string> = {
-        text: 'string',
-        integer: 'int',
-        bigint: 'bigint',
-        timestamp: 'timestamp',
-        jsonb: 'string',
-        uuid: 'string',
-      };
+    // Find columns to add
+    const columnsToAdd = ifNotExists
+      .filter(col => schema[col] && !existingFields.has(col))
+      .map(col => {
+        const colDef = schema[col];
+        return {
+          name: col,
+          valueSql: colDef?.nullable
+            ? `cast(NULL as ${typeMap[colDef.type ?? 'text']})`
+            : `cast(${this.getDefaultValue(colDef?.type ?? 'text')} as ${typeMap[colDef?.type ?? 'text']})`,
+        };
+      });
 
-      // Find columns to add
-      const columnsToAdd = ifNotExists
-        .filter(col => schema[col] && !existingFields.has(col))
-        .map(col => {
-          const colDef = schema[col];
-          return {
-            name: col,
-            valueSql: colDef?.nullable
-              ? `cast(NULL as ${typeMap[colDef.type ?? 'text']})`
-              : `cast(${this.getDefaultValue(colDef?.type ?? 'text')} as ${typeMap[colDef?.type ?? 'text']})`,
-          };
-        });
-
-      if (columnsToAdd.length > 0) {
-        await table.addColumns(columnsToAdd);
-        this.logger?.info?.(`Added columns [${columnsToAdd.map(c => c.name).join(', ')}] to table ${tableName}`);
-      }
-    } catch (error: any) {
-      throw new MastraError(
-        {
-          id: 'STORAGE_LANCE_STORAGE_ALTER_TABLE_FAILED',
-          domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.THIRD_PARTY,
-          details: { tableName },
-        },
-        error,
-      );
+    if (columnsToAdd.length > 0) {
+      await table.addColumns(columnsToAdd);
+      this.logger?.info?.(`Added columns [${columnsToAdd.map(c => c.name).join(', ')}] to table ${tableName}`);
     }
   }
 
   async clearTable({ tableName }: { tableName: TABLE_NAMES }): Promise<void> {
-    try {
-      if (!this.lanceClient) {
-        throw new Error('LanceDB client not initialized. Call LanceStorage.create() first.');
-      }
-      if (!tableName) {
-        throw new Error('tableName is required for clearTable.');
-      }
-    } catch (validationError: any) {
-      throw new MastraError(
-        {
-          id: 'STORAGE_LANCE_STORAGE_CLEAR_TABLE_INVALID_ARGS',
-          domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.USER,
-          text: validationError.message,
-          details: { tableName },
-        },
-        validationError,
-      );
-    }
-
-    try {
-      const table = await this.lanceClient.openTable(tableName);
+    const table = await this.lanceClient.openTable(tableName);
 
-      // delete function always takes a predicate as an argument, so we use '1=1' to delete all records because it is always true.
-      await table.delete('1=1');
-    } catch (error: any) {
-      throw new MastraError(
-        {
-          id: 'STORAGE_LANCE_STORAGE_CLEAR_TABLE_FAILED',
-          domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.THIRD_PARTY,
-          details: { tableName },
-        },
-        error,
-      );
-    }
+    // delete function always takes a predicate as an argument, so we use '1=1' to delete all records because it is always true.
+    await table.delete('1=1');
   }
 
   /**
@@ -421,34 +243,9 @@ export class LanceStorage extends MastraStorage {
    * @param record The record to insert.
    */
   async insert({ tableName, record }: { tableName: string; record: Record<string, any> }): Promise<void> {
-    try {
-      if (!this.lanceClient) {
-        throw new Error('LanceDB client not initialized. Call LanceStorage.create() first.');
-      }
-      if (!tableName) {
-        throw new Error('tableName is required for insert.');
-      }
-      if (!record || Object.keys(record).length === 0) {
-        throw new Error('record is required and cannot be empty for insert.');
-      }
-    } catch (validationError: any) {
-      throw new MastraError(
-        {
-          id: 'STORAGE_LANCE_STORAGE_INSERT_INVALID_ARGS',
-          domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.USER,
-          text: validationError.message,
-          details: { tableName },
-        },
-        validationError,
-      );
-    }
-
     try {
       const table = await this.lanceClient.openTable(tableName);
 
-      const primaryId = this.getPrimaryKeys(tableName as TABLE_NAMES);
-
       const processedRecord = { ...record };
 
       for (const key in processedRecord) {
@@ -462,17 +259,9 @@ export class LanceStorage extends MastraStorage {
         }
       }
 
-      await table.mergeInsert(primaryId).whenMatchedUpdateAll().whenNotMatchedInsertAll().execute([processedRecord]);
+      await table.add([processedRecord], { mode: 'overwrite' });
     } catch (error: any) {
-      throw new MastraError(
-        {
-          id: 'STORAGE_LANCE_STORAGE_INSERT_FAILED',
-          domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.THIRD_PARTY,
-          details: { tableName },
-        },
-        error,
-      );
+      throw new Error(`Failed to insert record: ${error}`);
     }
   }
 
@@ -482,34 +271,9 @@ export class LanceStorage extends MastraStorage {
    * @param records The records to insert.
    */
   async batchInsert({ tableName, records }: { tableName: string; records: Record<string, any>[] }): Promise<void> {
-    try {
-      if (!this.lanceClient) {
-        throw new Error('LanceDB client not initialized. Call LanceStorage.create() first.');
-      }
-      if (!tableName) {
-        throw new Error('tableName is required for batchInsert.');
-      }
-      if (!records || records.length === 0) {
-        throw new Error('records array is required and cannot be empty for batchInsert.');
-      }
-    } catch (validationError: any) {
-      throw new MastraError(
-        {
-          id: 'STORAGE_LANCE_STORAGE_BATCH_INSERT_INVALID_ARGS',
-          domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.USER,
-          text: validationError.message,
-          details: { tableName },
-        },
-        validationError,
-      );
-    }
-
     try {
       const table = await this.lanceClient.openTable(tableName);
 
-      const primaryId = this.getPrimaryKeys(tableName as TABLE_NAMES);
-
       const processedRecords = records.map(record => {
         const processedRecord = { ...record };
 
@@ -530,17 +294,9 @@ export class LanceStorage extends MastraStorage {
         return processedRecord;
       });
 
-      await table.mergeInsert(primaryId).whenMatchedUpdateAll().whenNotMatchedInsertAll().execute(processedRecords);
+      await table.add(processedRecords, { mode: 'overwrite' });
     } catch (error: any) {
-      throw new MastraError(
-        {
-          id: 'STORAGE_LANCE_STORAGE_BATCH_INSERT_FAILED',
-          domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.THIRD_PARTY,
-          details: { tableName },
-        },
-        error,
-      );
+      throw new Error(`Failed to batch insert records: ${error}`);
     }
   }
 
@@ -552,29 +308,6 @@ export class LanceStorage extends MastraStorage {
    * @returns The loaded record with proper type conversions, or null if not found
    */
   async load({ tableName, keys }: { tableName: TABLE_NAMES; keys: Record<string, any> }): Promise<any> {
-    try {
-      if (!this.lanceClient) {
-        throw new Error('LanceDB client not initialized. Call LanceStorage.create() first.');
-      }
-      if (!tableName) {
-        throw new Error('tableName is required for load.');
-      }
-      if (!keys || Object.keys(keys).length === 0) {
-        throw new Error('keys are required and cannot be empty for load.');
-      }
-    } catch (validationError: any) {
-      throw new MastraError(
-        {
-          id: 'STORAGE_LANCE_STORAGE_LOAD_INVALID_ARGS',
-          domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.USER,
-          text: validationError.message,
-          details: { tableName },
-        },
-        validationError,
-      );
-    }
-
     try {
       const table = await this.lanceClient.openTable(tableName);
       const tableSchema = await this.getTableSchema(tableName);
@@ -617,17 +350,7 @@ export class LanceStorage extends MastraStorage {
       // Process the result with type conversions
       return this.processResultWithTypeConversion(result[0], tableSchema);
     } catch (error: any) {
-      // If it's already a MastraError (e.g. from validateKeyTypes if we change it later), rethrow
-      if (error instanceof MastraError) throw error;
-      throw new MastraError(
-        {
-          id: 'STORAGE_LANCE_STORAGE_LOAD_FAILED',
-          domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.THIRD_PARTY,
-          details: { tableName, keyCount: Object.keys(keys).length, firstKey: Object.keys(keys)[0] ?? '' },
-        },
-        error,
-      );
+      throw new Error(`Failed to load record: ${error}`);
     }
   }
 
@@ -733,14 +456,7 @@ export class LanceStorage extends MastraStorage {
     try {
       return this.load({ tableName: TABLE_THREADS, keys: { id: threadId } });
     } catch (error: any) {
-      throw new MastraError(
-        {
-          id: 'LANCE_STORE_GET_THREAD_BY_ID_FAILED',
-          domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.THIRD_PARTY,
-        },
-        error,
-      );
+      throw new Error(`Failed to get thread by ID: ${error}`);
     }
   }
 
@@ -756,14 +472,7 @@ export class LanceStorage extends MastraStorage {
         await this.getTableSchema(TABLE_THREADS),
       ) as StorageThreadType[];
     } catch (error: any) {
-      throw new MastraError(
-        {
-          id: 'LANCE_STORE_GET_THREADS_BY_RESOURCE_ID_FAILED',
-          domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.THIRD_PARTY,
-        },
-        error,
-      );
+      throw new Error(`Failed to get threads by resource ID: ${error}`);
     }
   }
 
@@ -780,14 +489,7 @@ export class LanceStorage extends MastraStorage {
 
       return thread;
     } catch (error: any) {
-      throw new MastraError(
-        {
-          id: 'LANCE_STORE_SAVE_THREAD_FAILED',
-          domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.THIRD_PARTY,
-        },
-        error,
-      );
+      throw new Error(`Failed to save thread: ${error}`);
     }
   }
 
@@ -803,7 +505,7 @@ export class LanceStorage extends MastraStorage {
     try {
       const record = { id, title, metadata: JSON.stringify(metadata) };
       const table = await this.lanceClient.openTable(TABLE_THREADS);
-      await table.mergeInsert('id').whenMatchedUpdateAll().whenNotMatchedInsertAll().execute([record]);
+      await table.add([record], { mode: 'overwrite' });
 
       const query = table.query().where(`id = '${id}'`);
 
@@ -813,14 +515,7 @@ export class LanceStorage extends MastraStorage {
         await this.getTableSchema(TABLE_THREADS),
       ) as StorageThreadType;
     } catch (error: any) {
-      throw new MastraError(
-        {
-          id: 'LANCE_STORE_UPDATE_THREAD_FAILED',
-          domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.THIRD_PARTY,
-        },
-        error,
-      );
+      throw new Error(`Failed to update thread: ${error}`);
     }
   }
 
@@ -829,14 +524,7 @@ export class LanceStorage extends MastraStorage {
       const table = await this.lanceClient.openTable(TABLE_THREADS);
       await table.delete(`id = '${threadId}'`);
     } catch (error: any) {
-      throw new MastraError(
-        {
-          id: 'LANCE_STORE_DELETE_THREAD_FAILED',
-          domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.THIRD_PARTY,
-        },
-        error,
-      );
+      throw new Error(`Failed to delete thread: ${error}`);
     }
   }
 
@@ -929,7 +617,7 @@ export class LanceStorage extends MastraStorage {
       if (threadConfig) {
         throw new Error('ThreadConfig is not supported by LanceDB storage');
       }
-      const limit = this.resolveMessageLimit({ last: selectBy?.last, defaultLimit: Number.MAX_SAFE_INTEGER });
+
       const table = await this.lanceClient.openTable(TABLE_MESSAGES);
       let query = table.query().where(`\`threadId\` = '${threadId}'`);
 
@@ -964,8 +652,8 @@ export class LanceStorage extends MastraStorage {
       }
 
       // If we're fetching the last N messages, take only the last N after sorting
-      if (limit !== Number.MAX_SAFE_INTEGER) {
-        records = records.slice(-limit);
+      if (selectBy?.last !== undefined && selectBy.last !== false) {
+        records = records.slice(-selectBy.last);
       }
 
       const messages = this.processResultWithTypeConversion(records, await this.getTableSchema(TABLE_MESSAGES));
@@ -986,14 +674,7 @@ export class LanceStorage extends MastraStorage {
       if (format === 'v2') return list.get.all.v2();
       return list.get.all.v1();
     } catch (error: any) {
-      throw new MastraError(
-        {
-          id: 'LANCE_STORE_GET_MESSAGES_FAILED',
-          domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.THIRD_PARTY,
-        },
-        error,
-      );
+      throw new Error(`Failed to get messages: ${error}`);
     }
   }
 
@@ -1020,20 +701,12 @@ export class LanceStorage extends MastraStorage {
       }));
 
       const table = await this.lanceClient.openTable(TABLE_MESSAGES);
-      await table.mergeInsert('id').whenMatchedUpdateAll().whenNotMatchedInsertAll().execute(transformedMessages);
-
+      await table.add(transformedMessages, { mode: 'overwrite' });
       const list = new MessageList().add(messages, 'memory');
       if (format === `v2`) return list.get.all.v2();
       return list.get.all.v1();
     } catch (error: any) {
-      throw new MastraError(
-        {
-          id: 'LANCE_STORE_SAVE_MESSAGES_FAILED',
-          domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.THIRD_PARTY,
-        },
-        error,
-      );
+      throw new Error(`Failed to save messages: ${error}`);
     }
   }
 
@@ -1052,14 +725,7 @@ export class LanceStorage extends MastraStorage {
 
       return trace;
     } catch (error: any) {
-      throw new MastraError(
-        {
-          id: 'LANCE_STORE_SAVE_TRACE_FAILED',
-          domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.THIRD_PARTY,
-        },
-        error,
-      );
+      throw new Error(`Failed to save trace: ${error}`);
     }
   }
 
@@ -1070,14 +736,7 @@ export class LanceStorage extends MastraStorage {
       const records = await query.toArray();
       return this.processResultWithTypeConversion(records[0], await this.getTableSchema(TABLE_TRACES)) as TraceType;
     } catch (error: any) {
-      throw new MastraError(
-        {
-          id: 'LANCE_STORE_GET_TRACE_BY_ID_FAILED',
-          domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.THIRD_PARTY,
-        },
-        error,
-      );
+      throw new Error(`Failed to get trace by ID: ${error}`);
     }
   }
 
@@ -1136,15 +795,7 @@ export class LanceStorage extends MastraStorage {
         };
       }) as TraceType[];
     } catch (error: any) {
-      throw new MastraError(
-        {
-          id: 'LANCE_STORE_GET_TRACES_FAILED',
-          domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.THIRD_PARTY,
-          details: { name: name ?? '', scope: scope ?? '' },
-        },
-        error,
-      );
+      throw new Error(`Failed to get traces: ${error}`);
     }
   }
 
@@ -1167,14 +818,7 @@ export class LanceStorage extends MastraStorage {
       await table.add(transformedEvals, { mode: 'append' });
       return evals;
     } catch (error: any) {
-      throw new MastraError(
-        {
-          id: 'LANCE_STORE_SAVE_EVALS_FAILED',
-          domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.THIRD_PARTY,
-        },
-        error,
-      );
+      throw new Error(`Failed to save evals: ${error}`);
     }
   }
 
@@ -1202,15 +846,7 @@ export class LanceStorage extends MastraStorage {
         };
       }) as EvalRow[];
     } catch (error: any) {
-      throw new MastraError(
-        {
-          id: 'LANCE_STORE_GET_EVALS_BY_AGENT_NAME_FAILED',
-          domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.THIRD_PARTY,
-          details: { agentName },
-        },
-        error,
-      );
+      throw new Error(`Failed to get evals by agent name: ${error}`);
     }
   }
 
@@ -1273,15 +909,7 @@ export class LanceStorage extends MastraStorage {
         total: records.length,
       };
     } catch (error: any) {
-      throw new MastraError(
-        {
-          id: 'LANCE_STORE_GET_WORKFLOW_RUNS_FAILED',
-          domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.THIRD_PARTY,
-          details: { namespace: args?.namespace ?? '', workflowName: args?.workflowName ?? '' },
-        },
-        error,
-      );
+      throw new Error(`Failed to get workflow runs: ${error}`);
     }
   }
 
@@ -1309,15 +937,7 @@ export class LanceStorage extends MastraStorage {
       const record = records[0];
       return this.parseWorkflowRun(record);
     } catch (error: any) {
-      throw new MastraError(
-        {
-          id: 'LANCE_STORE_GET_WORKFLOW_RUN_BY_ID_FAILED',
-          domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.THIRD_PARTY,
-          details: { runId: args.runId, workflowName: args.workflowName ?? '' },
-        },
-        error,
-      );
+      throw new Error(`Failed to get workflow run by id: ${error}`);
     }
   }
 
@@ -1338,9 +958,11 @@ export class LanceStorage extends MastraStorage {
       const records = await query.toArray();
       let createdAt: number;
       const now = Date.now();
+      let mode: 'append' | 'overwrite' = 'append';
 
       if (records.length > 0) {
         createdAt = records[0].createdAt ?? now;
+        mode = 'overwrite';
       } else {
         createdAt = now;
       }
@@ -1353,21 +975,9 @@ export class LanceStorage extends MastraStorage {
         updatedAt: now,
       };
 
-      await table
-        .mergeInsert(['workflow_name', 'run_id'])
-        .whenMatchedUpdateAll()
-        .whenNotMatchedInsertAll()
-        .execute([record]);
+      await table.add([record], { mode });
     } catch (error: any) {
-      throw new MastraError(
-        {
-          id: 'LANCE_STORE_PERSIST_WORKFLOW_SNAPSHOT_FAILED',
-          domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.THIRD_PARTY,
-          details: { workflowName, runId },
-        },
-        error,
-      );
+      throw new Error(`Failed to persist workflow snapshot: ${error}`);
     }
   }
   async loadWorkflowSnapshot({
@@ -1383,27 +993,12 @@ export class LanceStorage extends MastraStorage {
       const records = await query.toArray();
       return records.length > 0 ? JSON.parse(records[0].snapshot) : null;
     } catch (error: any) {
-      throw new MastraError(
-        {
-          id: 'LANCE_STORE_LOAD_WORKFLOW_SNAPSHOT_FAILED',
-          domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.THIRD_PARTY,
-          details: { workflowName, runId },
-        },
-        error,
-      );
+      throw new Error(`Failed to load workflow snapshot: ${error}`);
     }
   }
 
   async getTracesPaginated(_args: StorageGetTracesArg): Promise<PaginationInfo & { traces: Trace[] }> {
-    throw new MastraError(
-      {
-        id: 'LANCE_STORE_GET_TRACES_PAGINATED_FAILED',
-        domain: ErrorDomain.STORAGE,
-        category: ErrorCategory.THIRD_PARTY,
-      },
-      'Method not implemented.',
-    );
+    throw new Error('Method not implemented.');
   }
 
   async getThreadsByResourceIdPaginated(_args: {
@@ -1411,37 +1006,12 @@ export class LanceStorage extends MastraStorage {
     page?: number;
     perPage?: number;
   }): Promise<PaginationInfo & { threads: StorageThreadType[] }> {
-    throw new MastraError(
-      {
-        id: 'LANCE_STORE_GET_THREADS_BY_RESOURCE_ID_PAGINATED_FAILED',
-        domain: ErrorDomain.STORAGE,
-        category: ErrorCategory.THIRD_PARTY,
-      },
-      'Method not implemented.',
-    );
+    throw new Error('Method not implemented.');
   }
 
   async getMessagesPaginated(
     _args: StorageGetMessagesArg,
   ): Promise<PaginationInfo & { messages: MastraMessageV1[] | MastraMessageV2[] }> {
-    throw new MastraError(
-      {
-        id: 'LANCE_STORE_GET_MESSAGES_PAGINATED_FAILED',
-        domain: ErrorDomain.STORAGE,
-        category: ErrorCategory.THIRD_PARTY,
-      },
-      'Method not implemented.',
-    );
-  }
-
-  async updateMessages(_args: {
-    messages: Partial<Omit<MastraMessageV2, 'createdAt'>> &
-      {
-        id: string;
-        content?: { metadata?: MastraMessageContentV2['metadata']; content?: MastraMessageContentV2['content'] };
-      }[];
-  }): Promise<MastraMessageV2[]> {
-    this.logger.error('updateMessages is not yet implemented in LanceStore');
-    throw new Error('Method not implemented');
+    throw new Error('Method not implemented.');
   }
 }