@dudousxd/nestjs-agent-rag 0.14.1 → 0.14.2

package/dist/index.cjs

@@ -279,6 +279,7 @@ var MemoryVectorStore = class {
     __name(this, "MemoryVectorStore");
   }
   records = /* @__PURE__ */ new Map();
+  replacing = /* @__PURE__ */ new Map();
   /**
   * Telemetry self-description. No `collection`: there is no namespace to name — the corpus IS this
   * instance, and two `MemoryVectorStore`s in one process are two corpora with nothing to tell them
@@ -294,6 +295,23 @@ var MemoryVectorStore = class {
       this.records.set(record.id, record);
     }
   }
+  /** Replaces take turns per document. A `remove` during one isn't queued, and may be undone by it. */
+  async replaceDocument(documentId, records) {
+    const previous = this.replacing.get(documentId) ?? Promise.resolve();
+    const run = previous.catch(() => {
+    }).then(async () => {
+      await this.remove(documentId);
+      await this.upsert(records);
+    });
+    this.replacing.set(documentId, run);
+    try {
+      await run;
+    } finally {
+      if (this.replacing.get(documentId) === run) {
+        this.replacing.delete(documentId);
+      }
+    }
+  }
   async remove(documentId) {
     for (const id of this.records.keys()) {
       if (documentIdOf(id) === documentId) {
@@ -1408,7 +1426,7 @@ var RedisVectorStore = class {
       String(REPLACE_LOCK_TTL_MS)
     ]) === null) {
       if (Date.now() >= deadline) {
-        throw new Error(`RedisVectorStore: timed out after ${REPLACE_LOCK_WAIT_MS}ms waiting to replace document "${documentId}"`);
+        throw new Error(`RedisVectorStore: timed out after ${REPLACE_LOCK_WAIT_MS}ms waiting to change document "${documentId}"`);
       }
       await new Promise((resolve) => setTimeout(resolve, delay));
       delay = Math.min(delay * 2, 250);
@@ -1416,7 +1434,7 @@ var RedisVectorStore = class {
     let lost = false;
     const assertHeld = /* @__PURE__ */ __name(() => {
       if (lost) {
-        throw new Error(`RedisVectorStore: lost the lock while replacing document "${documentId}"; call again to repair it`);
+        throw new Error(`RedisVectorStore: lost the lock while changing document "${documentId}"; call again to repair it`);
       }
     }, "assertHeld");
     const renewal = setInterval(() => {
@@ -1476,13 +1494,15 @@ var RedisVectorStore = class {
   * than N scans). Neither depends on a field that older chunks lack, so neither has a migration.
   */
   async remove(documentId) {
-    const keys = await this.chunkKeys(documentId);
-    if (keys.length > 0) {
-      await this.client.sendCommand([
-        "DEL",
-        ...keys
-      ]);
-    }
+    await this.withDocumentLock(documentId, async () => {
+      const keys = await this.chunkKeys(documentId);
+      if (keys.length > 0) {
+        await this.client.sendCommand([
+          "DEL",
+          ...keys
+        ]);
+      }
+    });
   }
   /**
   * Rewrite the metadata of every chunk of `documentId` in place, leaving `text` and `embedding`