@mastra/lance 1.2.1 → 1.2.2

package/dist/index.js

@@ -313,14 +313,14 @@ var LanceDB = class extends MastraBase {
 			}, error);
 		}
 	}
-	async insert({ tableName, record }) {
+	async update({ tableName, record }) {
 		try {
 			if (!this.client) 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.");
+			if (!tableName) throw new Error("tableName is required for update.");
+			if (!record || Object.keys(record).length === 0) throw new Error("record is required and cannot be empty for update.");
 		} catch (validationError) {
 			throw new MastraError({
-				id: createStorageErrorId("LANCE", "INSERT", "INVALID_ARGS"),
+				id: createStorageErrorId("LANCE", "UPDATE", "INVALID_ARGS"),
 				domain: ErrorDomain.STORAGE,
 				category: ErrorCategory.USER,
 				text: validationError.message,
@@ -337,10 +337,10 @@ var LanceDB = class extends MastraBase {
 			}
 			const filteredRecord = await this.filterRecordToKnownColumns(tableName, processedRecord);
 			if (Object.keys(filteredRecord).length === 0) return;
-			await table.mergeInsert(primaryId).whenMatchedUpdateAll().whenNotMatchedInsertAll().execute([filteredRecord]);
+			await table.mergeInsert(primaryId).whenMatchedUpdateAll().execute([filteredRecord]);
 		} catch (error) {
 			throw new MastraError({
-				id: createStorageErrorId("LANCE", "INSERT", "FAILED"),
+				id: createStorageErrorId("LANCE", "UPDATE", "FAILED"),
 				domain: ErrorDomain.STORAGE,
 				category: ErrorCategory.THIRD_PARTY,
 				details: { tableName }
@@ -688,7 +688,7 @@ var StoreMemoryLance = class extends MemoryStorage {
 						createdAt: new Date(thread.createdAt).getTime(),
 						updatedAt: now
 					};
-					await threadsTable.mergeInsert("id").whenMatchedUpdateAll().whenNotMatchedInsertAll().execute([record]);
+					await threadsTable.mergeInsert("id").whenMatchedUpdateAll().execute([record]);
 				}
 			}
 		} catch (error) {
@@ -759,7 +759,7 @@ var StoreMemoryLance = class extends MemoryStorage {
 				metadata: JSON.stringify(mergedMetadata),
 				updatedAt: (/* @__PURE__ */ new Date()).getTime()
 			};
-			await (await this.client.openTable(TABLE_THREADS)).mergeInsert("id").whenMatchedUpdateAll().whenNotMatchedInsertAll().execute([record]);
+			await (await this.client.openTable(TABLE_THREADS)).mergeInsert("id").whenMatchedUpdateAll().execute([record]);
 			const updatedThread = await this.getThreadById({ threadId: id });
 			if (!updatedThread) throw new Error(`Failed to retrieve updated thread ${id}`);
 			return updatedThread;
@@ -944,13 +944,13 @@ var StoreMemoryLance = class extends MemoryStorage {
 		try {
 			const { messages } = args;
 			if (messages.length === 0) return { messages: [] };
-			const threadId = messages[0]?.threadId;
-			if (!threadId) throw new Error("Thread ID is required");
+			const threadIds = /* @__PURE__ */ new Set();
 			for (const message of messages) {
 				if (!message.id) throw new Error("Message ID is required");
 				if (!message.threadId) throw new Error("Thread ID is required for all messages");
 				if (message.resourceId === null || message.resourceId === void 0) throw new Error("Resource ID cannot be null or undefined");
 				if (!message.content) throw new Error("Message content is required");
+				threadIds.add(message.threadId);
 			}
 			const transformedMessages = messages.map((message) => {
 				const { threadId, type, ...rest } = message;
@@ -961,13 +961,13 @@ var StoreMemoryLance = class extends MemoryStorage {
 					content: JSON.stringify(message.content)
 				};
 			});
-			await (await this.client.openTable(TABLE_MESSAGES)).mergeInsert("id").whenMatchedUpdateAll().whenNotMatchedInsertAll().execute(transformedMessages);
 			const threadsTable = await this.client.openTable(TABLE_THREADS);
-			const updateRecord = {
-				id: threadId,
-				updatedAt: (/* @__PURE__ */ new Date()).getTime()
-			};
-			await threadsTable.mergeInsert("id").whenMatchedUpdateAll().whenNotMatchedInsertAll().execute([updateRecord]);
+			const currentTime = (/* @__PURE__ */ new Date()).getTime();
+			for (const id of threadIds) if ((await threadsTable.mergeInsert("id").whenMatchedUpdateAll().execute([{
+				id,
+				updatedAt: currentTime
+			}])).numUpdatedRows === 0) throw new Error(`Cannot save messages because parent thread ${id} does not exist`);
+			await (await this.client.openTable(TABLE_MESSAGES)).mergeInsert("id").whenMatchedUpdateAll().whenNotMatchedInsertAll().execute(transformedMessages);
 			return { messages: new MessageList().add(messages, "memory").get.all.db() };
 		} catch (error) {
 			throw new MastraError({
@@ -1201,7 +1201,7 @@ var StoreMemoryLance = class extends MemoryStorage {
 					if ("parts" in updates.content && updates.content.parts !== void 0) newContent.parts = updates.content.parts;
 					updatePayload.content = JSON.stringify(newContent);
 				}
-				await this.#db.insert({
+				await this.#db.update({
 					tableName: TABLE_MESSAGES,
 					record: {
 						id,
@@ -1214,7 +1214,7 @@ var StoreMemoryLance = class extends MemoryStorage {
 				});
 				if (updatedMessage) updatedMessages.push(this.parseMessageData(updatedMessage));
 			}
-			for (const threadId of affectedThreadIds) await this.#db.insert({
+			for (const threadId of affectedThreadIds) await this.#db.update({
 				tableName: TABLE_THREADS,
 				record: {
 					id: threadId,
@@ -1331,7 +1331,10 @@ var StoreMemoryLance = class extends MemoryStorage {
 				metadata: updatedResource.metadata ? JSON.stringify(updatedResource.metadata) : "",
 				updatedAt: updatedResource.updatedAt.getTime()
 			};
-			await (await this.client.openTable(TABLE_RESOURCES)).mergeInsert("id").whenMatchedUpdateAll().whenNotMatchedInsertAll().execute([record]);
+			if ((await (await this.client.openTable(TABLE_RESOURCES)).mergeInsert("id").whenMatchedUpdateAll().execute([record])).numUpdatedRows === 0) {
+				if (attempt < maxRetries - 1) continue;
+				throw new Error(`Resource ${resourceId} disappeared while it was being updated`);
+			}
 			return updatedResource;
 		} catch (error) {
 			if (error.message?.includes("Commit conflict") && attempt < maxRetries - 1) {