@dudousxd/nestjs-agent-rag 0.14.1 → 0.14.2

package/dist/index.d.cts

@@ -407,8 +407,9 @@ declare function describeSharedSource(legs: readonly unknown[]): Omit<RetrievalD
  * (`removeWhere` refusing an empty filter, an empty-array filter deleting nothing) are behavioural,
  * not incidental to Redis.
  */
-declare class MemoryVectorStore implements VectorStore {
+declare class MemoryVectorStore implements AtomicReplaceVectorStore {
     private readonly records;
+    private readonly replacing;
     /**
      * 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
@@ -416,6 +417,8 @@ declare class MemoryVectorStore implements VectorStore {
      */
     describeRetrieval(): RetrievalDescriptor;
     upsert(records: VectorRecord[]): Promise<void>;
+    /** Replaces take turns per document. A `remove` during one isn't queued, and may be undone by it. */
+    replaceDocument(documentId: string, records: VectorRecord[]): Promise<void>;
     remove(documentId: string): Promise<void>;
     /**
      * Merge `patch` into every chunk of `documentId`, leaving `text` and `embedding` untouched. The

package/dist/index.d.ts

@@ -407,8 +407,9 @@ declare function describeSharedSource(legs: readonly unknown[]): Omit<RetrievalD
  * (`removeWhere` refusing an empty filter, an empty-array filter deleting nothing) are behavioural,
  * not incidental to Redis.
  */
-declare class MemoryVectorStore implements VectorStore {
+declare class MemoryVectorStore implements AtomicReplaceVectorStore {
     private readonly records;
+    private readonly replacing;
     /**
      * 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
@@ -416,6 +417,8 @@ declare class MemoryVectorStore implements VectorStore {
      */
     describeRetrieval(): RetrievalDescriptor;
     upsert(records: VectorRecord[]): Promise<void>;
+    /** Replaces take turns per document. A `remove` during one isn't queued, and may be undone by it. */
+    replaceDocument(documentId: string, records: VectorRecord[]): Promise<void>;
     remove(documentId: string): Promise<void>;
     /**
      * Merge `patch` into every chunk of `documentId`, leaving `text` and `embedding` untouched. The

package/dist/index.js

@@ -191,6 +191,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
@@ -206,6 +207,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) {
@@ -1320,7 +1338,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);
@@ -1328,7 +1346,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(() => {
@@ -1388,13 +1406,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`