@mastra/lance 0.1.3-alpha.0 → 0.1.3-alpha.2

package/dist/index.js

@@ -1,5 +1,6 @@
 import { connect, Index } from '@lancedb/lancedb';
 import { MessageList } from '@mastra/core/agent';
+import { MastraError, ErrorCategory, ErrorDomain } from '@mastra/core/error';
 import { MastraStorage, TABLE_THREADS, TABLE_MESSAGES, TABLE_TRACES, TABLE_EVALS, TABLE_WORKFLOW_SNAPSHOT } from '@mastra/core/storage';
 import { Utf8, Float64, Binary, Float32, Int32, Field, Schema } from 'apache-arrow';
 import { MastraVector } from '@mastra/core/vector';
@@ -36,8 +37,26 @@ var LanceStorage = class _LanceStorage extends MastraStorage {
       instance.lanceClient = await connect(uri, options);
       return instance;
     } catch (e) {
-      throw new Error(`Failed to connect to LanceDB: ${e}`);
+      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
+      );
+    }
+  }
+  getPrimaryKeys(tableName) {
+    let primaryId = ["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
@@ -50,11 +69,40 @@ var LanceStorage = class _LanceStorage extends MastraStorage {
     tableName,
     schema
   }) {
+    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) {
-      throw new Error(`Failed to create table: ${error}`);
+      throw new MastraError(
+        {
+          id: "STORAGE_LANCE_STORAGE_CREATE_TABLE_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { tableName }
+        },
+        error
+      );
     }
   }
   translateSchema(schema) {
@@ -97,14 +145,41 @@ var LanceStorage = class _LanceStorage extends MastraStorage {
    * @param tableName Name of the table to drop
    */
   async dropTable(tableName) {
+    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) {
+      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) {
-      if (error.toString().includes("was not found")) {
+      if (error.toString().includes("was not found") || error.message?.includes("Table not found")) {
         this.logger.debug(`Table '${tableName}' does not exist, skipping drop`);
         return;
       }
-      throw new Error(`Failed to drop table: ${error}`);
+      throw new MastraError(
+        {
+          id: "STORAGE_LANCE_STORAGE_DROP_TABLE_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { tableName }
+        },
+        error
+      );
     }
   }
   /**
@@ -113,6 +188,25 @@ var LanceStorage = class _LanceStorage extends MastraStorage {
    * @returns Table schema
    */
   async getTableSchema(tableName) {
+    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) {
+      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();
@@ -125,7 +219,15 @@ var LanceStorage = class _LanceStorage extends MastraStorage {
         }
       };
     } catch (error) {
-      throw new Error(`Failed to get table schema: ${error}`);
+      throw new MastraError(
+        {
+          id: "STORAGE_LANCE_STORAGE_GET_TABLE_SCHEMA_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { tableName }
+        },
+        error
+      );
     }
   }
   getDefaultValue(type) {
@@ -156,32 +258,101 @@ var LanceStorage = class _LanceStorage extends MastraStorage {
     schema,
     ifNotExists
   }) {
-    const table = await this.lanceClient.openTable(tableName);
-    const currentSchema = await table.schema();
-    const existingFields = new Set(currentSchema.fields.map((f) => f.name));
-    const typeMap = {
-      text: "string",
-      integer: "int",
-      bigint: "bigint",
-      timestamp: "timestamp",
-      jsonb: "string",
-      uuid: "string"
-    };
-    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"]})`
+    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) {
+      throw new MastraError(
+        {
+          id: "STORAGE_LANCE_STORAGE_ALTER_TABLE_INVALID_ARGS",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.USER,
+          text: validationError.message,
+          details: { tableName }
+        },
+        validationError
+      );
+    }
+    try {
+      const table = await this.lanceClient.openTable(tableName);
+      const currentSchema = await table.schema();
+      const existingFields = new Set(currentSchema.fields.map((f) => f.name));
+      const typeMap = {
+        text: "string",
+        integer: "int",
+        bigint: "bigint",
+        timestamp: "timestamp",
+        jsonb: "string",
+        uuid: "string"
       };
-    });
-    if (columnsToAdd.length > 0) {
-      await table.addColumns(columnsToAdd);
-      this.logger?.info?.(`Added columns [${columnsToAdd.map((c) => c.name).join(", ")}] to table ${tableName}`);
+      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) {
+      throw new MastraError(
+        {
+          id: "STORAGE_LANCE_STORAGE_ALTER_TABLE_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { tableName }
+        },
+        error
+      );
     }
   }
   async clearTable({ tableName }) {
-    const table = await this.lanceClient.openTable(tableName);
-    await table.delete("1=1");
+    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) {
+      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);
+      await table.delete("1=1");
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: "STORAGE_LANCE_STORAGE_CLEAR_TABLE_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { tableName }
+        },
+        error
+      );
+    }
   }
   /**
    * Insert a single record into a table. This function overwrites the existing record if it exists. Use this function for inserting records into tables with custom schemas.
@@ -189,8 +360,31 @@ var LanceStorage = class _LanceStorage extends MastraStorage {
    * @param record The record to insert.
    */
   async insert({ tableName, record }) {
+    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) {
+      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);
       const processedRecord = { ...record };
       for (const key in processedRecord) {
         if (processedRecord[key] !== null && typeof processedRecord[key] === "object" && !(processedRecord[key] instanceof Date)) {
@@ -198,9 +392,17 @@ var LanceStorage = class _LanceStorage extends MastraStorage {
           processedRecord[key] = JSON.stringify(processedRecord[key]);
         }
       }
-      await table.add([processedRecord], { mode: "overwrite" });
+      await table.mergeInsert(primaryId).whenMatchedUpdateAll().whenNotMatchedInsertAll().execute([processedRecord]);
     } catch (error) {
-      throw new Error(`Failed to insert record: ${error}`);
+      throw new MastraError(
+        {
+          id: "STORAGE_LANCE_STORAGE_INSERT_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { tableName }
+        },
+        error
+      );
     }
   }
   /**
@@ -209,8 +411,31 @@ var LanceStorage = class _LanceStorage extends MastraStorage {
    * @param records The records to insert.
    */
   async batchInsert({ tableName, records }) {
+    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) {
+      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);
       const processedRecords = records.map((record) => {
         const processedRecord = { ...record };
         for (const key in processedRecord) {
@@ -221,9 +446,17 @@ var LanceStorage = class _LanceStorage extends MastraStorage {
         }
         return processedRecord;
       });
-      await table.add(processedRecords, { mode: "overwrite" });
+      await table.mergeInsert(primaryId).whenMatchedUpdateAll().whenNotMatchedInsertAll().execute(processedRecords);
     } catch (error) {
-      throw new Error(`Failed to batch insert records: ${error}`);
+      throw new MastraError(
+        {
+          id: "STORAGE_LANCE_STORAGE_BATCH_INSERT_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { tableName }
+        },
+        error
+      );
     }
   }
   /**
@@ -234,6 +467,28 @@ var LanceStorage = class _LanceStorage extends MastraStorage {
    * @returns The loaded record with proper type conversions, or null if not found
    */
   async load({ tableName, keys }) {
+    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) {
+      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);
@@ -261,7 +516,16 @@ var LanceStorage = class _LanceStorage extends MastraStorage {
       }
       return this.processResultWithTypeConversion(result[0], tableSchema);
     } catch (error) {
-      throw new Error(`Failed to load record: ${error}`);
+      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
+      );
     }
   }
   /**
@@ -336,7 +600,14 @@ var LanceStorage = class _LanceStorage extends MastraStorage {
     try {
       return this.load({ tableName: TABLE_THREADS, keys: { id: threadId } });
     } catch (error) {
-      throw new Error(`Failed to get thread by ID: ${error}`);
+      throw new MastraError(
+        {
+          id: "LANCE_STORE_GET_THREAD_BY_ID_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY
+        },
+        error
+      );
     }
   }
   async getThreadsByResourceId({ resourceId }) {
@@ -349,7 +620,14 @@ var LanceStorage = class _LanceStorage extends MastraStorage {
         await this.getTableSchema(TABLE_THREADS)
       );
     } catch (error) {
-      throw new Error(`Failed to get threads by resource ID: ${error}`);
+      throw new MastraError(
+        {
+          id: "LANCE_STORE_GET_THREADS_BY_RESOURCE_ID_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY
+        },
+        error
+      );
     }
   }
   /**
@@ -364,7 +642,14 @@ var LanceStorage = class _LanceStorage extends MastraStorage {
       await table.add([record], { mode: "append" });
       return thread;
     } catch (error) {
-      throw new Error(`Failed to save thread: ${error}`);
+      throw new MastraError(
+        {
+          id: "LANCE_STORE_SAVE_THREAD_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY
+        },
+        error
+      );
     }
   }
   async updateThread({
@@ -375,7 +660,7 @@ var LanceStorage = class _LanceStorage extends MastraStorage {
     try {
       const record = { id, title, metadata: JSON.stringify(metadata) };
       const table = await this.lanceClient.openTable(TABLE_THREADS);
-      await table.add([record], { mode: "overwrite" });
+      await table.mergeInsert("id").whenMatchedUpdateAll().whenNotMatchedInsertAll().execute([record]);
       const query = table.query().where(`id = '${id}'`);
       const records = await query.toArray();
       return this.processResultWithTypeConversion(
@@ -383,7 +668,14 @@ var LanceStorage = class _LanceStorage extends MastraStorage {
         await this.getTableSchema(TABLE_THREADS)
       );
     } catch (error) {
-      throw new Error(`Failed to update thread: ${error}`);
+      throw new MastraError(
+        {
+          id: "LANCE_STORE_UPDATE_THREAD_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY
+        },
+        error
+      );
     }
   }
   async deleteThread({ threadId }) {
@@ -391,7 +683,14 @@ var LanceStorage = class _LanceStorage extends MastraStorage {
       const table = await this.lanceClient.openTable(TABLE_THREADS);
       await table.delete(`id = '${threadId}'`);
     } catch (error) {
-      throw new Error(`Failed to delete thread: ${error}`);
+      throw new MastraError(
+        {
+          id: "LANCE_STORE_DELETE_THREAD_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY
+        },
+        error
+      );
     }
   }
   /**
@@ -453,6 +752,7 @@ var LanceStorage = 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}'`);
       if (selectBy) {
@@ -471,8 +771,8 @@ var LanceStorage = class _LanceStorage extends MastraStorage {
       if (selectBy?.include && selectBy.include.length > 0) {
         records = this.processMessagesWithContext(records, selectBy.include);
       }
-      if (selectBy?.last !== void 0 && selectBy.last !== false) {
-        records = records.slice(-selectBy.last);
+      if (limit !== Number.MAX_SAFE_INTEGER) {
+        records = records.slice(-limit);
       }
       const messages = this.processResultWithTypeConversion(records, await this.getTableSchema(TABLE_MESSAGES));
       const normalized = messages.map((msg) => ({
@@ -489,7 +789,14 @@ var LanceStorage = class _LanceStorage extends MastraStorage {
       if (format === "v2") return list.get.all.v2();
       return list.get.all.v1();
     } catch (error) {
-      throw new Error(`Failed to get messages: ${error}`);
+      throw new MastraError(
+        {
+          id: "LANCE_STORE_GET_MESSAGES_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY
+        },
+        error
+      );
     }
   }
   async saveMessages(args) {
@@ -507,12 +814,19 @@ var LanceStorage = class _LanceStorage extends MastraStorage {
         content: JSON.stringify(message.content)
       }));
       const table = await this.lanceClient.openTable(TABLE_MESSAGES);
-      await table.add(transformedMessages, { mode: "overwrite" });
+      await table.mergeInsert("id").whenMatchedUpdateAll().whenNotMatchedInsertAll().execute(transformedMessages);
       const list = new MessageList().add(messages, "memory");
       if (format === `v2`) return list.get.all.v2();
       return list.get.all.v1();
     } catch (error) {
-      throw new Error(`Failed to save messages: ${error}`);
+      throw new MastraError(
+        {
+          id: "LANCE_STORE_SAVE_MESSAGES_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY
+        },
+        error
+      );
     }
   }
   async saveTrace({ trace }) {
@@ -529,7 +843,14 @@ var LanceStorage = class _LanceStorage extends MastraStorage {
       await table.add([record], { mode: "append" });
       return trace;
     } catch (error) {
-      throw new Error(`Failed to save trace: ${error}`);
+      throw new MastraError(
+        {
+          id: "LANCE_STORE_SAVE_TRACE_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY
+        },
+        error
+      );
     }
   }
   async getTraceById({ traceId }) {
@@ -539,7 +860,14 @@ var LanceStorage = class _LanceStorage extends MastraStorage {
       const records = await query.toArray();
       return this.processResultWithTypeConversion(records[0], await this.getTableSchema(TABLE_TRACES));
     } catch (error) {
-      throw new Error(`Failed to get trace by ID: ${error}`);
+      throw new MastraError(
+        {
+          id: "LANCE_STORE_GET_TRACE_BY_ID_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY
+        },
+        error
+      );
     }
   }
   async getTraces({
@@ -581,7 +909,15 @@ var LanceStorage = class _LanceStorage extends MastraStorage {
         };
       });
     } catch (error) {
-      throw new Error(`Failed to get traces: ${error}`);
+      throw new MastraError(
+        {
+          id: "LANCE_STORE_GET_TRACES_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { name: name ?? "", scope: scope ?? "" }
+        },
+        error
+      );
     }
   }
   async saveEvals({ evals }) {
@@ -602,7 +938,14 @@ var LanceStorage = class _LanceStorage extends MastraStorage {
       await table.add(transformedEvals, { mode: "append" });
       return evals;
     } catch (error) {
-      throw new Error(`Failed to save evals: ${error}`);
+      throw new MastraError(
+        {
+          id: "LANCE_STORE_SAVE_EVALS_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY
+        },
+        error
+      );
     }
   }
   async getEvalsByAgentName(agentName, type) {
@@ -629,7 +972,15 @@ var LanceStorage = class _LanceStorage extends MastraStorage {
         };
       });
     } catch (error) {
-      throw new Error(`Failed to get evals by agent name: ${error}`);
+      throw new MastraError(
+        {
+          id: "LANCE_STORE_GET_EVALS_BY_AGENT_NAME_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { agentName }
+        },
+        error
+      );
     }
   }
   parseWorkflowRun(row) {
@@ -675,7 +1026,15 @@ var LanceStorage = class _LanceStorage extends MastraStorage {
         total: records.length
       };
     } catch (error) {
-      throw new Error(`Failed to get workflow runs: ${error}`);
+      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
+      );
     }
   }
   /**
@@ -696,7 +1055,15 @@ var LanceStorage = class _LanceStorage extends MastraStorage {
       const record = records[0];
       return this.parseWorkflowRun(record);
     } catch (error) {
-      throw new Error(`Failed to get workflow run by id: ${error}`);
+      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
+      );
     }
   }
   async persistWorkflowSnapshot({
@@ -710,10 +1077,8 @@ var LanceStorage = class _LanceStorage extends MastraStorage {
       const records = await query.toArray();
       let createdAt;
       const now = Date.now();
-      let mode = "append";
       if (records.length > 0) {
         createdAt = records[0].createdAt ?? now;
-        mode = "overwrite";
       } else {
         createdAt = now;
       }
@@ -724,9 +1089,17 @@ var LanceStorage = class _LanceStorage extends MastraStorage {
         createdAt,
         updatedAt: now
       };
-      await table.add([record], { mode });
+      await table.mergeInsert(["workflow_name", "run_id"]).whenMatchedUpdateAll().whenNotMatchedInsertAll().execute([record]);
     } catch (error) {
-      throw new Error(`Failed to persist workflow snapshot: ${error}`);
+      throw new MastraError(
+        {
+          id: "LANCE_STORE_PERSIST_WORKFLOW_SNAPSHOT_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { workflowName, runId }
+        },
+        error
+      );
     }
   }
   async loadWorkflowSnapshot({
@@ -739,17 +1112,46 @@ var LanceStorage = class _LanceStorage extends MastraStorage {
       const records = await query.toArray();
       return records.length > 0 ? JSON.parse(records[0].snapshot) : null;
     } catch (error) {
-      throw new Error(`Failed to load workflow snapshot: ${error}`);
+      throw new MastraError(
+        {
+          id: "LANCE_STORE_LOAD_WORKFLOW_SNAPSHOT_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { workflowName, runId }
+        },
+        error
+      );
     }
   }
   async getTracesPaginated(_args) {
-    throw new Error("Method not implemented.");
+    throw new MastraError(
+      {
+        id: "LANCE_STORE_GET_TRACES_PAGINATED_FAILED",
+        domain: ErrorDomain.STORAGE,
+        category: ErrorCategory.THIRD_PARTY
+      },
+      "Method not implemented."
+    );
   }
   async getThreadsByResourceIdPaginated(_args) {
-    throw new Error("Method not implemented.");
+    throw new MastraError(
+      {
+        id: "LANCE_STORE_GET_THREADS_BY_RESOURCE_ID_PAGINATED_FAILED",
+        domain: ErrorDomain.STORAGE,
+        category: ErrorCategory.THIRD_PARTY
+      },
+      "Method not implemented."
+    );
   }
   async getMessagesPaginated(_args) {
-    throw new Error("Method not implemented.");
+    throw new MastraError(
+      {
+        id: "LANCE_STORE_GET_MESSAGES_PAGINATED_FAILED",
+        domain: ErrorDomain.STORAGE,
+        category: ErrorCategory.THIRD_PARTY
+      },
+      "Method not implemented."
+    );
   }
   async updateMessages(_args) {
     this.logger.error("updateMessages is not yet implemented in LanceStore");
@@ -1107,7 +1509,15 @@ var LanceVectorStore = class _LanceVectorStore extends MastraVector {
       instance.lanceClient = await connect(uri, options);
       return instance;
     } catch (e) {
-      throw new Error(`Failed to connect to LanceDB: ${e}`);
+      throw new MastraError(
+        {
+          id: "STORAGE_LANCE_VECTOR_CONNECT_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { uri }
+        },
+        e
+      );
     }
   }
   /**
@@ -1131,14 +1541,27 @@ var LanceVectorStore = class _LanceVectorStore extends MastraVector {
     columns = [],
     includeAllColumns = false
   }) {
-    if (!this.lanceClient) {
-      throw new Error("LanceDB client not initialized. Use LanceVectorStore.create() to create an instance");
-    }
-    if (!tableName) {
-      throw new Error("tableName is required");
-    }
-    if (!queryVector) {
-      throw new Error("queryVector is required");
+    try {
+      if (!this.lanceClient) {
+        throw new Error("LanceDB client not initialized. Use LanceVectorStore.create() to create an instance");
+      }
+      if (!tableName) {
+        throw new Error("tableName is required");
+      }
+      if (!queryVector) {
+        throw new Error("queryVector is required");
+      }
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: "STORAGE_LANCE_VECTOR_QUERY_FAILED_INVALID_ARGS",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.USER,
+          text: "LanceDB client not initialized. Use LanceVectorStore.create() to create an instance",
+          details: { tableName }
+        },
+        error
+      );
     }
     try {
       const table = await this.lanceClient.openTable(tableName);
@@ -1177,7 +1600,15 @@ var LanceVectorStore = class _LanceVectorStore extends MastraVector {
         };
       });
     } catch (error) {
-      throw new Error(`Failed to query vectors: ${error.message}`);
+      throw new MastraError(
+        {
+          id: "STORAGE_LANCE_VECTOR_QUERY_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { tableName, includeVector, columnsCount: columns?.length, includeAllColumns }
+        },
+        error
+      );
     }
   }
   filterTranslator(filter) {
@@ -1210,14 +1641,27 @@ var LanceVectorStore = class _LanceVectorStore extends MastraVector {
     return translator.translate(prefixedFilter);
   }
   async upsert({ tableName, vectors, metadata = [], ids = [] }) {
-    if (!this.lanceClient) {
-      throw new Error("LanceDB client not initialized. Use LanceVectorStore.create() to create an instance");
-    }
-    if (!tableName) {
-      throw new Error("tableName is required");
-    }
-    if (!vectors || !Array.isArray(vectors) || vectors.length === 0) {
-      throw new Error("vectors array is required and must not be empty");
+    try {
+      if (!this.lanceClient) {
+        throw new Error("LanceDB client not initialized. Use LanceVectorStore.create() to create an instance");
+      }
+      if (!tableName) {
+        throw new Error("tableName is required");
+      }
+      if (!vectors || !Array.isArray(vectors) || vectors.length === 0) {
+        throw new Error("vectors array is required and must not be empty");
+      }
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: "STORAGE_LANCE_VECTOR_UPSERT_FAILED_INVALID_ARGS",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.USER,
+          text: "LanceDB client not initialized. Use LanceVectorStore.create() to create an instance",
+          details: { tableName }
+        },
+        error
+      );
     }
     try {
       const tables = await this.lanceClient.tableNames();
@@ -1244,7 +1688,15 @@ var LanceVectorStore = class _LanceVectorStore extends MastraVector {
       await table.add(data, { mode: "overwrite" });
       return vectorIds;
     } catch (error) {
-      throw new Error(`Failed to upsert vectors: ${error.message}`);
+      throw new MastraError(
+        {
+          id: "STORAGE_LANCE_VECTOR_UPSERT_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { tableName, vectorCount: vectors.length, metadataCount: metadata.length, idsCount: ids.length }
+        },
+        error
+      );
     }
   }
   /**
@@ -1264,29 +1716,78 @@ var LanceVectorStore = class _LanceVectorStore extends MastraVector {
   }
   async createTable(tableName, data, options) {
     if (!this.lanceClient) {
-      throw new Error("LanceDB client not initialized. Use LanceVectorStore.create() to create an instance");
+      throw new MastraError({
+        id: "STORAGE_LANCE_VECTOR_CREATE_TABLE_FAILED_INVALID_ARGS",
+        domain: ErrorDomain.STORAGE,
+        category: ErrorCategory.USER,
+        text: "LanceDB client not initialized. Use LanceVectorStore.create() to create an instance",
+        details: { tableName }
+      });
+    }
+    if (Array.isArray(data)) {
+      data = data.map((record) => this.flattenObject(record));
     }
     try {
-      if (Array.isArray(data)) {
-        data = data.map((record) => this.flattenObject(record));
-      }
       return await this.lanceClient.createTable(tableName, data, options);
     } catch (error) {
-      throw new Error(`Failed to create table: ${error.message}`);
+      throw new MastraError(
+        {
+          id: "STORAGE_LANCE_VECTOR_CREATE_TABLE_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { tableName }
+        },
+        error
+      );
     }
   }
   async listTables() {
     if (!this.lanceClient) {
-      throw new Error("LanceDB client not initialized. Use LanceVectorStore.create() to create an instance");
+      throw new MastraError({
+        id: "STORAGE_LANCE_VECTOR_LIST_TABLES_FAILED_INVALID_ARGS",
+        domain: ErrorDomain.STORAGE,
+        category: ErrorCategory.USER,
+        text: "LanceDB client not initialized. Use LanceVectorStore.create() to create an instance",
+        details: { methodName: "listTables" }
+      });
+    }
+    try {
+      return await this.lanceClient.tableNames();
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: "STORAGE_LANCE_VECTOR_LIST_TABLES_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY
+        },
+        error
+      );
     }
-    return await this.lanceClient.tableNames();
   }
   async getTableSchema(tableName) {
     if (!this.lanceClient) {
-      throw new Error("LanceDB client not initialized. Use LanceVectorStore.create() to create an instance");
+      throw new MastraError({
+        id: "STORAGE_LANCE_VECTOR_GET_TABLE_SCHEMA_FAILED_INVALID_ARGS",
+        domain: ErrorDomain.STORAGE,
+        category: ErrorCategory.USER,
+        text: "LanceDB client not initialized. Use LanceVectorStore.create() to create an instance",
+        details: { tableName }
+      });
+    }
+    try {
+      const table = await this.lanceClient.openTable(tableName);
+      return await table.schema();
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: "STORAGE_LANCE_VECTOR_GET_TABLE_SCHEMA_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { tableName }
+        },
+        error
+      );
     }
-    const table = await this.lanceClient.openTable(tableName);
-    return await table.schema();
   }
   /**
    * indexName is actually a column name in a table in lanceDB
@@ -1298,10 +1799,10 @@ var LanceVectorStore = class _LanceVectorStore extends MastraVector {
     metric = "cosine",
     indexConfig = {}
   }) {
-    if (!this.lanceClient) {
-      throw new Error("LanceDB client not initialized. Use LanceVectorStore.create() to create an instance");
-    }
     try {
+      if (!this.lanceClient) {
+        throw new Error("LanceDB client not initialized. Use LanceVectorStore.create() to create an instance");
+      }
       if (!tableName) {
         throw new Error("tableName is required");
       }
@@ -1311,6 +1812,18 @@ var LanceVectorStore = class _LanceVectorStore extends MastraVector {
       if (typeof dimension !== "number" || dimension <= 0) {
         throw new Error("dimension must be a positive number");
       }
+    } catch (err) {
+      throw new MastraError(
+        {
+          id: "STORAGE_LANCE_VECTOR_CREATE_INDEX_FAILED_INVALID_ARGS",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.USER,
+          details: { tableName: tableName || "", indexName, dimension, metric }
+        },
+        err
+      );
+    }
+    try {
       const tables = await this.lanceClient.tableNames();
       if (!tables.includes(tableName)) {
         throw new Error(
@@ -1345,12 +1858,26 @@ var LanceVectorStore = class _LanceVectorStore extends MastraVector {
         });
       }
     } catch (error) {
-      throw new Error(`Failed to create index: ${error.message}`);
+      throw new MastraError(
+        {
+          id: "STORAGE_LANCE_VECTOR_CREATE_INDEX_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { tableName: tableName || "", indexName, dimension }
+        },
+        error
+      );
     }
   }
   async listIndexes() {
     if (!this.lanceClient) {
-      throw new Error("LanceDB client not initialized. Use LanceVectorStore.create() to create an instance");
+      throw new MastraError({
+        id: "STORAGE_LANCE_VECTOR_LIST_INDEXES_FAILED_INVALID_ARGS",
+        domain: ErrorDomain.STORAGE,
+        category: ErrorCategory.USER,
+        text: "LanceDB client not initialized. Use LanceVectorStore.create() to create an instance",
+        details: { methodName: "listIndexes" }
+      });
     }
     try {
       const tables = await this.lanceClient.tableNames();
@@ -1362,15 +1889,34 @@ var LanceVectorStore = class _LanceVectorStore extends MastraVector {
       }
       return allIndices;
     } catch (error) {
-      throw new Error(`Failed to list indexes: ${error.message}`);
+      throw new MastraError(
+        {
+          id: "STORAGE_LANCE_VECTOR_LIST_INDEXES_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY
+        },
+        error
+      );
     }
   }
   async describeIndex({ indexName }) {
-    if (!this.lanceClient) {
-      throw new Error("LanceDB client not initialized. Use LanceVectorStore.create() to create an instance");
-    }
-    if (!indexName) {
-      throw new Error("indexName is required");
+    try {
+      if (!this.lanceClient) {
+        throw new Error("LanceDB client not initialized. Use LanceVectorStore.create() to create an instance");
+      }
+      if (!indexName) {
+        throw new Error("indexName is required");
+      }
+    } catch (err) {
+      throw new MastraError(
+        {
+          id: "STORAGE_LANCE_VECTOR_DESCRIBE_INDEX_FAILED_INVALID_ARGS",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.USER,
+          details: { indexName }
+        },
+        err
+      );
     }
     try {
       const tables = await this.lanceClient.tableNames();
@@ -1397,15 +1943,35 @@ var LanceVectorStore = class _LanceVectorStore extends MastraVector {
       }
       throw new Error(`IndexName: ${indexName} not found`);
     } catch (error) {
-      throw new Error(`Failed to describe index: ${error.message}`);
+      throw new MastraError(
+        {
+          id: "STORAGE_LANCE_VECTOR_DESCRIBE_INDEX_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { indexName }
+        },
+        error
+      );
     }
   }
   async deleteIndex({ indexName }) {
-    if (!this.lanceClient) {
-      throw new Error("LanceDB client not initialized. Use LanceVectorStore.create() to create an instance");
-    }
-    if (!indexName) {
-      throw new Error("indexName is required");
+    try {
+      if (!this.lanceClient) {
+        throw new Error("LanceDB client not initialized. Use LanceVectorStore.create() to create an instance");
+      }
+      if (!indexName) {
+        throw new Error("indexName is required");
+      }
+    } catch (err) {
+      throw new MastraError(
+        {
+          id: "STORAGE_LANCE_VECTOR_DELETE_INDEX_FAILED_INVALID_ARGS",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.USER,
+          details: { indexName }
+        },
+        err
+      );
     }
     try {
       const tables = await this.lanceClient.tableNames();
@@ -1420,7 +1986,15 @@ var LanceVectorStore = class _LanceVectorStore extends MastraVector {
       }
       throw new Error(`Index ${indexName} not found`);
     } catch (error) {
-      throw new Error(`Failed to delete index: ${error.message}`);
+      throw new MastraError(
+        {
+          id: "STORAGE_LANCE_VECTOR_DELETE_INDEX_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { indexName }
+        },
+        error
+      );
     }
   }
   /**
@@ -1428,33 +2002,73 @@ var LanceVectorStore = class _LanceVectorStore extends MastraVector {
    */
   async deleteAllTables() {
     if (!this.lanceClient) {
-      throw new Error("LanceDB client not initialized. Use LanceVectorStore.create() to create an instance");
+      throw new MastraError({
+        id: "STORAGE_LANCE_VECTOR_DELETE_ALL_TABLES_FAILED_INVALID_ARGS",
+        domain: ErrorDomain.STORAGE,
+        category: ErrorCategory.USER,
+        details: { methodName: "deleteAllTables" },
+        text: "LanceDB client not initialized. Use LanceVectorStore.create() to create an instance"
+      });
     }
     try {
       await this.lanceClient.dropAllTables();
     } catch (error) {
-      throw new Error(`Failed to delete tables: ${error.message}`);
+      throw new MastraError(
+        {
+          id: "STORAGE_LANCE_VECTOR_DELETE_ALL_TABLES_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { methodName: "deleteAllTables" }
+        },
+        error
+      );
     }
   }
   async deleteTable(tableName) {
     if (!this.lanceClient) {
-      throw new Error("LanceDB client not initialized. Use LanceVectorStore.create() to create an instance");
+      throw new MastraError({
+        id: "STORAGE_LANCE_VECTOR_DELETE_TABLE_FAILED_INVALID_ARGS",
+        domain: ErrorDomain.STORAGE,
+        category: ErrorCategory.USER,
+        details: { tableName },
+        text: "LanceDB client not initialized. Use LanceVectorStore.create() to create an instance"
+      });
     }
     try {
       await this.lanceClient.dropTable(tableName);
     } catch (error) {
-      throw new Error(`Failed to delete tables: ${error.message}`);
+      throw new MastraError(
+        {
+          id: "STORAGE_LANCE_VECTOR_DELETE_TABLE_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { tableName }
+        },
+        error
+      );
     }
   }
   async updateVector({ indexName, id, update }) {
-    if (!this.lanceClient) {
-      throw new Error("LanceDB client not initialized. Use LanceVectorStore.create() to create an instance");
-    }
-    if (!indexName) {
-      throw new Error("indexName is required");
-    }
-    if (!id) {
-      throw new Error("id is required");
+    try {
+      if (!this.lanceClient) {
+        throw new Error("LanceDB client not initialized. Use LanceVectorStore.create() to create an instance");
+      }
+      if (!indexName) {
+        throw new Error("indexName is required");
+      }
+      if (!id) {
+        throw new Error("id is required");
+      }
+    } catch (err) {
+      throw new MastraError(
+        {
+          id: "STORAGE_LANCE_VECTOR_UPDATE_VECTOR_FAILED_INVALID_ARGS",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.USER,
+          details: { indexName, id }
+        },
+        err
+      );
     }
     try {
       const tables = await this.lanceClient.tableNames();
@@ -1508,18 +2122,38 @@ var LanceVectorStore = class _LanceVectorStore extends MastraVector {
       }
       throw new Error(`No table found with column/index '${indexName}'`);
     } catch (error) {
-      throw new Error(`Failed to update index: ${error.message}`);
+      throw new MastraError(
+        {
+          id: "STORAGE_LANCE_VECTOR_UPDATE_VECTOR_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { indexName, id, hasVector: !!update.vector, hasMetadata: !!update.metadata }
+        },
+        error
+      );
     }
   }
   async deleteVector({ indexName, id }) {
-    if (!this.lanceClient) {
-      throw new Error("LanceDB client not initialized. Use LanceVectorStore.create() to create an instance");
-    }
-    if (!indexName) {
-      throw new Error("indexName is required");
-    }
-    if (!id) {
-      throw new Error("id is required");
+    try {
+      if (!this.lanceClient) {
+        throw new Error("LanceDB client not initialized. Use LanceVectorStore.create() to create an instance");
+      }
+      if (!indexName) {
+        throw new Error("indexName is required");
+      }
+      if (!id) {
+        throw new Error("id is required");
+      }
+    } catch (err) {
+      throw new MastraError(
+        {
+          id: "STORAGE_LANCE_VECTOR_DELETE_VECTOR_FAILED_INVALID_ARGS",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.USER,
+          details: { indexName, id }
+        },
+        err
+      );
     }
     try {
       const tables = await this.lanceClient.tableNames();
@@ -1541,7 +2175,15 @@ var LanceVectorStore = class _LanceVectorStore extends MastraVector {
       }
       throw new Error(`No table found with column/index '${indexName}'`);
     } catch (error) {
-      throw new Error(`Failed to delete index: ${error.message}`);
+      throw new MastraError(
+        {
+          id: "STORAGE_LANCE_VECTOR_DELETE_VECTOR_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { indexName, id }
+        },
+        error
+      );
     }
   }
   /**