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

package/.turbo/turbo-build.log

@@ -1,23 +1,23 @@
 
-> @mastra/lance@0.1.3-alpha.1 build /home/runner/work/mastra/mastra/stores/lance
+> @mastra/lance@0.1.3-alpha.2 build /home/runner/work/mastra/mastra/stores/lance
 > tsup src/index.ts --format esm,cjs --experimental-dts --clean --treeshake=smallest --splitting
 
 CLI Building entry: src/index.ts
 CLI Using tsconfig: tsconfig.json
 CLI tsup v8.5.0
 TSC Build start
-TSC ⚡️ Build success in 9945ms
+TSC ⚡️ Build success in 10342ms
 DTS Build start
 CLI Target: es2022
 Analysis will use the bundled TypeScript version 5.8.3
 Writing package typings: /home/runner/work/mastra/mastra/stores/lance/dist/_tsup-dts-rollup.d.ts
 Analysis will use the bundled TypeScript version 5.8.3
 Writing package typings: /home/runner/work/mastra/mastra/stores/lance/dist/_tsup-dts-rollup.d.cts
-DTS ⚡️ Build success in 12711ms
+DTS ⚡️ Build success in 13286ms
 CLI Cleaning output folder
 ESM Build start
 CJS Build start
-ESM dist/index.js 70.32 KB
-ESM ⚡️ Build success in 1611ms
-CJS dist/index.cjs 71.85 KB
-CJS ⚡️ Build success in 1612ms
+CJS dist/index.cjs 72.48 KB
+CJS ⚡️ Build success in 1677ms
+ESM dist/index.js 70.94 KB
+ESM ⚡️ Build success in 1693ms

package/CHANGELOG.md

@@ -1,5 +1,17 @@
 # @mastra/lance
 
+## 0.1.3-alpha.2
+
+### Patch Changes
+
+- 0fb9d64: [MASTRA-4018] Update saveMessages in storage adapters to upsert messages
+- 144eb0b: [MASTRA-3669] Metadata Filter Types
+- Updated dependencies [15e9d26]
+- Updated dependencies [07d6d88]
+- Updated dependencies [5d74aab]
+- Updated dependencies [144eb0b]
+  - @mastra/core@0.10.7-alpha.2
+
 ## 0.1.3-alpha.1
 
 ### Patch Changes

package/dist/_tsup-dts-rollup.d.cts

@@ -1,6 +1,7 @@
 import { ArrayOperator } from '@mastra/core/vector/filter';
 import { BaseFilterTranslator } from '@mastra/core/vector/filter';
 import { BasicOperator } from '@mastra/core/vector/filter';
+import type { BlacklistedRootOperators } from '@mastra/core/vector/filter';
 import type { ConnectionOptions } from '@lancedb/lancedb';
 import type { CreateIndexParams } from '@mastra/core';
 import type { CreateTableOptions } from '@lancedb/lancedb';
@@ -10,12 +11,14 @@ import type { DescribeIndexParams } from '@mastra/core';
 import type { EvalRow } from '@mastra/core/storage';
 import type { IndexStats } from '@mastra/core';
 import { LogicalOperator } from '@mastra/core/vector/filter';
+import type { LogicalOperatorValueMap } from '@mastra/core/vector/filter';
 import type { MastraMessageContentV2 } from '@mastra/core/agent';
 import type { MastraMessageV1 } from '@mastra/core/memory';
 import type { MastraMessageV2 } from '@mastra/core/memory';
 import { MastraStorage } from '@mastra/core/storage';
 import { MastraVector } from '@mastra/core/vector';
 import { NumericOperator } from '@mastra/core/vector/filter';
+import type { OperatorValueMap } from '@mastra/core/vector/filter';
 import type { PaginationInfo } from '@mastra/core/storage';
 import type { QueryResult } from '@mastra/core';
 import type { QueryVectorParams } from '@mastra/core';
@@ -53,13 +56,15 @@ declare interface IVFConfig {
     lists?: number;
 }
 
+declare type LanceBlacklisted = BlacklistedRootOperators | '$like' | '$notLike' | '$contains';
+
 declare interface LanceCreateIndexParams extends CreateIndexParams {
     indexConfig?: LanceIndexConfig;
     tableName?: string;
 }
 
-export declare class LanceFilterTranslator extends BaseFilterTranslator {
-    translate(filter: VectorFilter): string;
+export declare class LanceFilterTranslator extends BaseFilterTranslator<LanceVectorFilter, string> {
+    translate(filter: LanceVectorFilter): string;
     private processFilter;
     private processLogicalOperator;
     private processNestedObject;
@@ -94,7 +99,13 @@ declare interface LanceIndexConfig extends IndexConfig {
     numSubVectors?: number;
 }
 
-declare interface LanceQueryVectorParams extends QueryVectorParams {
+declare type LanceOperatorValueMap = OperatorValueMap & {
+    $like: string;
+    $notLike: string;
+    $contains: string;
+};
+
+declare interface LanceQueryVectorParams extends QueryVectorParams<LanceVectorFilter> {
     tableName: string;
     columns?: string[];
     includeAllColumns?: boolean;
@@ -125,6 +136,7 @@ declare class LanceStorage extends MastraStorage {
      * ```
      */
     static create(name: string, uri: string, options?: ConnectionOptions): Promise<LanceStorage>;
+    private getPrimaryKeys;
     /**
      * @internal
      * Private constructor to enforce using the create factory method
@@ -327,7 +339,9 @@ declare interface LanceUpsertVectorParams extends UpsertVectorParams {
     tableName: string;
 }
 
-declare class LanceVectorStore extends MastraVector {
+export declare type LanceVectorFilter = VectorFilter<keyof LanceOperatorValueMap, LanceOperatorValueMap, LogicalOperatorValueMap, LanceBlacklisted>;
+
+declare class LanceVectorStore extends MastraVector<LanceVectorFilter> {
     private lanceClient;
     /**
      * Creates a new instance of LanceVectorStore

package/dist/_tsup-dts-rollup.d.ts

@@ -1,6 +1,7 @@
 import { ArrayOperator } from '@mastra/core/vector/filter';
 import { BaseFilterTranslator } from '@mastra/core/vector/filter';
 import { BasicOperator } from '@mastra/core/vector/filter';
+import type { BlacklistedRootOperators } from '@mastra/core/vector/filter';
 import type { ConnectionOptions } from '@lancedb/lancedb';
 import type { CreateIndexParams } from '@mastra/core';
 import type { CreateTableOptions } from '@lancedb/lancedb';
@@ -10,12 +11,14 @@ import type { DescribeIndexParams } from '@mastra/core';
 import type { EvalRow } from '@mastra/core/storage';
 import type { IndexStats } from '@mastra/core';
 import { LogicalOperator } from '@mastra/core/vector/filter';
+import type { LogicalOperatorValueMap } from '@mastra/core/vector/filter';
 import type { MastraMessageContentV2 } from '@mastra/core/agent';
 import type { MastraMessageV1 } from '@mastra/core/memory';
 import type { MastraMessageV2 } from '@mastra/core/memory';
 import { MastraStorage } from '@mastra/core/storage';
 import { MastraVector } from '@mastra/core/vector';
 import { NumericOperator } from '@mastra/core/vector/filter';
+import type { OperatorValueMap } from '@mastra/core/vector/filter';
 import type { PaginationInfo } from '@mastra/core/storage';
 import type { QueryResult } from '@mastra/core';
 import type { QueryVectorParams } from '@mastra/core';
@@ -53,13 +56,15 @@ declare interface IVFConfig {
     lists?: number;
 }
 
+declare type LanceBlacklisted = BlacklistedRootOperators | '$like' | '$notLike' | '$contains';
+
 declare interface LanceCreateIndexParams extends CreateIndexParams {
     indexConfig?: LanceIndexConfig;
     tableName?: string;
 }
 
-export declare class LanceFilterTranslator extends BaseFilterTranslator {
-    translate(filter: VectorFilter): string;
+export declare class LanceFilterTranslator extends BaseFilterTranslator<LanceVectorFilter, string> {
+    translate(filter: LanceVectorFilter): string;
     private processFilter;
     private processLogicalOperator;
     private processNestedObject;
@@ -94,7 +99,13 @@ declare interface LanceIndexConfig extends IndexConfig {
     numSubVectors?: number;
 }
 
-declare interface LanceQueryVectorParams extends QueryVectorParams {
+declare type LanceOperatorValueMap = OperatorValueMap & {
+    $like: string;
+    $notLike: string;
+    $contains: string;
+};
+
+declare interface LanceQueryVectorParams extends QueryVectorParams<LanceVectorFilter> {
     tableName: string;
     columns?: string[];
     includeAllColumns?: boolean;
@@ -125,6 +136,7 @@ declare class LanceStorage extends MastraStorage {
      * ```
      */
     static create(name: string, uri: string, options?: ConnectionOptions): Promise<LanceStorage>;
+    private getPrimaryKeys;
     /**
      * @internal
      * Private constructor to enforce using the create factory method
@@ -327,7 +339,9 @@ declare interface LanceUpsertVectorParams extends UpsertVectorParams {
     tableName: string;
 }
 
-declare class LanceVectorStore extends MastraVector {
+export declare type LanceVectorFilter = VectorFilter<keyof LanceOperatorValueMap, LanceOperatorValueMap, LogicalOperatorValueMap, LanceBlacklisted>;
+
+declare class LanceVectorStore extends MastraVector<LanceVectorFilter> {
     private lanceClient;
     /**
      * Creates a new instance of LanceVectorStore

package/dist/index.cjs

@@ -51,6 +51,15 @@ var LanceStorage = class _LanceStorage extends storage.MastraStorage {
       );
     }
   }
+  getPrimaryKeys(tableName) {
+    let primaryId = ["id"];
+    if (tableName === storage.TABLE_WORKFLOW_SNAPSHOT) {
+      primaryId = ["workflow_name", "run_id"];
+    } else if (tableName === storage.TABLE_EVALS) {
+      primaryId = ["agent_name", "metric_name", "run_id"];
+    }
+    return primaryId;
+  }
   /**
    * @internal
    * Private constructor to enforce using the create factory method
@@ -377,6 +386,7 @@ var LanceStorage = class _LanceStorage extends storage.MastraStorage {
     }
     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)) {
@@ -384,7 +394,7 @@ var LanceStorage = class _LanceStorage extends storage.MastraStorage {
           processedRecord[key] = JSON.stringify(processedRecord[key]);
         }
       }
-      await table.add([processedRecord], { mode: "overwrite" });
+      await table.mergeInsert(primaryId).whenMatchedUpdateAll().whenNotMatchedInsertAll().execute([processedRecord]);
     } catch (error$1) {
       throw new error.MastraError(
         {
@@ -427,6 +437,7 @@ var LanceStorage = class _LanceStorage extends storage.MastraStorage {
     }
     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) {
@@ -437,7 +448,7 @@ var LanceStorage = class _LanceStorage extends storage.MastraStorage {
         }
         return processedRecord;
       });
-      await table.add(processedRecords, { mode: "overwrite" });
+      await table.mergeInsert(primaryId).whenMatchedUpdateAll().whenNotMatchedInsertAll().execute(processedRecords);
     } catch (error$1) {
       throw new error.MastraError(
         {
@@ -651,7 +662,7 @@ var LanceStorage = class _LanceStorage extends storage.MastraStorage {
     try {
       const record = { id, title, metadata: JSON.stringify(metadata) };
       const table = await this.lanceClient.openTable(storage.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(
@@ -805,7 +816,7 @@ var LanceStorage = class _LanceStorage extends storage.MastraStorage {
         content: JSON.stringify(message.content)
       }));
       const table = await this.lanceClient.openTable(storage.TABLE_MESSAGES);
-      await table.add(transformedMessages, { mode: "overwrite" });
+      await table.mergeInsert("id").whenMatchedUpdateAll().whenNotMatchedInsertAll().execute(transformedMessages);
       const list = new agent.MessageList().add(messages, "memory");
       if (format === `v2`) return list.get.all.v2();
       return list.get.all.v1();
@@ -1068,10 +1079,8 @@ var LanceStorage = class _LanceStorage extends storage.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;
       }
@@ -1082,7 +1091,7 @@ var LanceStorage = class _LanceStorage extends storage.MastraStorage {
         createdAt,
         updatedAt: now
       };
-      await table.add([record], { mode });
+      await table.mergeInsert(["workflow_name", "run_id"]).whenMatchedUpdateAll().whenNotMatchedInsertAll().execute([record]);
     } catch (error$1) {
       throw new error.MastraError(
         {

package/dist/index.js

@@ -49,6 +49,15 @@ var LanceStorage = class _LanceStorage extends MastraStorage {
       );
     }
   }
+  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
    * Private constructor to enforce using the create factory method
@@ -375,6 +384,7 @@ var LanceStorage = class _LanceStorage extends MastraStorage {
     }
     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)) {
@@ -382,7 +392,7 @@ 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 MastraError(
         {
@@ -425,6 +435,7 @@ var LanceStorage = class _LanceStorage extends MastraStorage {
     }
     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) {
@@ -435,7 +446,7 @@ 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 MastraError(
         {
@@ -649,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(
@@ -803,7 +814,7 @@ 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();
@@ -1066,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;
       }
@@ -1080,7 +1089,7 @@ 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 MastraError(
         {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mastra/lance",
-  "version": "0.1.3-alpha.1",
+  "version": "0.1.3-alpha.2",
   "description": "Lance provider for Mastra - includes both vector and db storage capabilities",
   "type": "module",
   "main": "dist/index.js",
@@ -25,13 +25,13 @@
   "devDependencies": {
     "@microsoft/api-extractor": "^7.52.8",
     "@types/node": "^20.19.0",
-    "eslint": "^9.28.0",
+    "eslint": "^9.29.0",
     "tsup": "^8.5.0",
     "typescript": "^5.8.3",
     "vitest": "^3.2.3",
     "@internal/lint": "0.0.13",
-    "@internal/storage-test-utils": "0.0.9",
-    "@mastra/core": "^0.10.7-alpha.1"
+    "@mastra/core": "^0.10.7-alpha.2",
+    "@internal/storage-test-utils": "0.0.9"
   },
   "peerDependencies": {
     "@mastra/core": ">=0.10.4-0 <0.11.0"

package/src/storage/index.test.ts

@@ -54,6 +54,7 @@ function generateMessageRecords(count: number, threadId?: string): MastraMessage
     toolCallIds: [],
     toolCallArgs: [],
     toolNames: [],
+    type: 'v2',
   }));
 }
 
@@ -563,7 +564,6 @@ describe('LanceStorage tests', async () => {
         expect(message.threadId).toEqual(threadId);
         expect(message.id.toString()).toEqual(messages[index].id);
         expect(message.content).toEqual(messages[index].content);
-        expect(new Date(message.createdAt)).toEqual(new Date(messages[index].createdAt));
         expect(message.role).toEqual(messages[index].role);
         expect(message.resourceId).toEqual(messages[index].resourceId);
         expect(message.type).toEqual(messages[index].type);
@@ -708,6 +708,76 @@ describe('LanceStorage tests', async () => {
         expect(currentDate).toBeLessThanOrEqual(nextDate);
       }
     });
+
+    it('should upsert messages: duplicate id+threadId results in update, not duplicate row', async () => {
+      const thread = 'thread-1';
+      const baseMessage = generateMessageRecords(1, thread)[0];
+
+      // Insert the message for the first time
+      await storage.saveMessages({ messages: [baseMessage], format: 'v2' });
+
+      // Insert again with the same id and threadId but different content
+      const updatedMessage: MastraMessageV2 = {
+        ...generateMessageRecords(1, thread)[0],
+        id: baseMessage.id,
+        content: { format: 2, parts: [{ type: 'text', text: 'Updated' }] },
+      };
+
+      await storage.saveMessages({ messages: [updatedMessage], format: 'v2' });
+
+      // Retrieve messages for the thread
+      const retrievedMessages = await storage.getMessages({ threadId: thread, format: 'v2' });
+      // Only one message should exist for that id+threadId
+      expect(retrievedMessages.filter(m => m.id.toString() === baseMessage.id)).toHaveLength(1);
+
+      // The content should be the updated one
+      expect(retrievedMessages.find(m => m.id.toString() === baseMessage.id)?.content.parts[0].text).toBe('Updated');
+    });
+
+    it('should upsert messages: duplicate id and different threadid', async () => {
+      const thread1 = 'thread-1';
+      const thread2 = 'thread-2';
+      const thread3 = 'thread-3';
+
+      const message = generateMessageRecords(1, thread1)[0];
+
+      // Insert message into thread1
+      await storage.saveMessages({ messages: [message], format: 'v2' });
+
+      // Attempt to insert a message with the same id but different threadId
+      const conflictingMessage: MastraMessageV2 = {
+        ...generateMessageRecords(1, thread2)[0],
+        id: message.id,
+        content: { format: 2, parts: [{ type: 'text', text: 'Thread2 Content' }] },
+      };
+
+      const differentMessage: MastraMessageV2 = {
+        ...generateMessageRecords(1, thread3)[0],
+        id: '2',
+        content: { format: 2, parts: [{ type: 'text', text: 'Another Message Content' }] },
+      };
+
+      // Save should move the message to the new thread
+      await storage.saveMessages({ messages: [conflictingMessage], format: 'v2' });
+
+      await storage.saveMessages({ messages: [differentMessage], format: 'v2' });
+
+      // Retrieve messages for both threads
+      const thread1Messages = await storage.getMessages({ threadId: thread1, format: 'v2' });
+      const thread2Messages = await storage.getMessages({ threadId: thread2, format: 'v2' });
+      const thread3Messages = await storage.getMessages({ threadId: thread3, format: 'v2' });
+
+      // Thread 1 should NOT have the message with that id
+      expect(thread1Messages.find(m => m.id.toString() === message.id)).toBeUndefined();
+
+      // Thread 2 should have the message with that id
+      expect(thread2Messages.find(m => m.id.toString() === message.id)?.content.parts[0].text).toBe('Thread2 Content');
+
+      // Thread 2 should have the other message
+      expect(thread3Messages.find(m => m.id.toString() === differentMessage.id)?.content.parts[0].text).toBe(
+        'Another Message Content',
+      );
+    });
   });
 
   describe('Trace operations', () => {

package/src/storage/index.ts

@@ -71,6 +71,17 @@ export class LanceStorage extends MastraStorage {
     }
   }
 
+  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
@@ -436,6 +447,8 @@ export class LanceStorage extends MastraStorage {
     try {
       const table = await this.lanceClient.openTable(tableName);
 
+      const primaryId = this.getPrimaryKeys(tableName as TABLE_NAMES);
+
       const processedRecord = { ...record };
 
       for (const key in processedRecord) {
@@ -449,7 +462,7 @@ export class LanceStorage extends MastraStorage {
         }
       }
 
-      await table.add([processedRecord], { mode: 'overwrite' });
+      await table.mergeInsert(primaryId).whenMatchedUpdateAll().whenNotMatchedInsertAll().execute([processedRecord]);
     } catch (error: any) {
       throw new MastraError(
         {
@@ -495,6 +508,8 @@ export class LanceStorage extends MastraStorage {
     try {
       const table = await this.lanceClient.openTable(tableName);
 
+      const primaryId = this.getPrimaryKeys(tableName as TABLE_NAMES);
+
       const processedRecords = records.map(record => {
         const processedRecord = { ...record };
 
@@ -515,7 +530,7 @@ export class LanceStorage extends MastraStorage {
         return processedRecord;
       });
 
-      await table.add(processedRecords, { mode: 'overwrite' });
+      await table.mergeInsert(primaryId).whenMatchedUpdateAll().whenNotMatchedInsertAll().execute(processedRecords);
     } catch (error: any) {
       throw new MastraError(
         {
@@ -788,7 +803,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.add([record], { mode: 'overwrite' });
+      await table.mergeInsert('id').whenMatchedUpdateAll().whenNotMatchedInsertAll().execute([record]);
 
       const query = table.query().where(`id = '${id}'`);
 
@@ -1005,7 +1020,8 @@ export class LanceStorage extends MastraStorage {
       }));
 
       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();
@@ -1322,11 +1338,9 @@ 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;
       }
@@ -1339,7 +1353,11 @@ export class LanceStorage extends MastraStorage {
         updatedAt: now,
       };
 
-      await table.add([record], { mode });
+      await table
+        .mergeInsert(['workflow_name', 'run_id'])
+        .whenMatchedUpdateAll()
+        .whenNotMatchedInsertAll()
+        .execute([record]);
     } catch (error: any) {
       throw new MastraError(
         {

package/src/vector/filter.test.ts

@@ -13,8 +13,8 @@ describe('LanceFilterTranslator', () => {
   describe('basic operations', () => {
     it('handles empty filters', () => {
       expect(translator.translate({})).toEqual('');
-      expect(translator.translate(null as any)).toEqual('');
-      expect(translator.translate(undefined as any)).toEqual('');
+      expect(translator.translate(null)).toEqual('');
+      expect(translator.translate(undefined)).toEqual('');
     });
 
     it('translates equality operation', () => {
@@ -244,7 +244,7 @@ describe('LanceFilterTranslator', () => {
     });
 
     it('throws error for invalid operators at top level', () => {
-      const invalidFilters = [{ $gt: 100 }, { $in: ['value1', 'value2'] }, { $like: '%pattern%' }];
+      const invalidFilters: any = [{ $gt: 100 }, { $in: ['value1', 'value2'] }, { $like: '%pattern%' }];
 
       invalidFilters.forEach(filter => {
         expect(() => translator.translate(filter)).toThrow(/Invalid top-level operator/);

package/src/vector/filter.ts

@@ -1,8 +1,28 @@
-import type { VectorFilter } from '@mastra/core/vector/filter';
 import { BaseFilterTranslator } from '@mastra/core/vector/filter';
-
-export class LanceFilterTranslator extends BaseFilterTranslator {
-  translate(filter: VectorFilter): string {
+import type {
+  VectorFilter,
+  OperatorValueMap,
+  LogicalOperatorValueMap,
+  BlacklistedRootOperators,
+} from '@mastra/core/vector/filter';
+
+type LanceOperatorValueMap = OperatorValueMap & {
+  $like: string;
+  $notLike: string;
+  $contains: string;
+};
+
+type LanceBlacklisted = BlacklistedRootOperators | '$like' | '$notLike' | '$contains';
+
+export type LanceVectorFilter = VectorFilter<
+  keyof LanceOperatorValueMap,
+  LanceOperatorValueMap,
+  LogicalOperatorValueMap,
+  LanceBlacklisted
+>;
+
+export class LanceFilterTranslator extends BaseFilterTranslator<LanceVectorFilter, string> {
+  translate(filter: LanceVectorFilter): string {
     if (!filter || Object.keys(filter).length === 0) {
       return '';
     }

package/src/vector/index.ts

@@ -15,7 +15,7 @@ import type {
 import { ErrorCategory, ErrorDomain, MastraError } from '@mastra/core/error';
 
 import { MastraVector } from '@mastra/core/vector';
-import type { VectorFilter } from '@mastra/core/vector/filter';
+import type { LanceVectorFilter } from './filter';
 import { LanceFilterTranslator } from './filter';
 import type { IndexConfig } from './types';
 
@@ -33,13 +33,13 @@ interface LanceUpsertVectorParams extends UpsertVectorParams {
   tableName: string;
 }
 
-interface LanceQueryVectorParams extends QueryVectorParams {
+interface LanceQueryVectorParams extends QueryVectorParams<LanceVectorFilter> {
   tableName: string;
   columns?: string[];
   includeAllColumns?: boolean;
 }
 
-export class LanceVectorStore extends MastraVector {
+export class LanceVectorStore extends MastraVector<LanceVectorFilter> {
   private lanceClient!: Connection;
 
   /**
@@ -204,7 +204,7 @@ export class LanceVectorStore extends MastraVector {
     }
   }
 
-  private filterTranslator(filter: VectorFilter): string {
+  private filterTranslator(filter: LanceVectorFilter): string {
     // Add metadata_ prefix to filter keys if they don't already have it
     const processFilterKeys = (filterObj: Record<string, any>): Record<string, any> => {
       const result: Record<string, any> = {};