From 37c4b5c1587419e4178cec80ae83e50b8df48693 Mon Sep 17 00:00:00 2001 From: seehi <6580@pm.me> Date: Wed, 28 Aug 2024 10:30:51 +0800 Subject: [PATCH] update comment --- metagpt/rag/retrievers/bm25_retriever.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/metagpt/rag/retrievers/bm25_retriever.py b/metagpt/rag/retrievers/bm25_retriever.py index 74cba5124..4891fad50 100644 --- a/metagpt/rag/retrievers/bm25_retriever.py +++ b/metagpt/rag/retrievers/bm25_retriever.py @@ -37,6 +37,7 @@ class DynamicBM25Retriever(BM25Retriever): def add_nodes(self, nodes: list[BaseNode], **kwargs) -> None: """Support add nodes.""" + self._nodes.extend(nodes) self._corpus = [self._tokenizer(node.get_content()) for node in self._nodes] self.bm25 = BM25Okapi(self._corpus) @@ -46,6 +47,7 @@ class DynamicBM25Retriever(BM25Retriever): def persist(self, persist_dir: str, **kwargs) -> None: """Support persist.""" + if self._index: self._index.storage_context.persist(persist_dir) @@ -56,6 +58,7 @@ class DynamicBM25Retriever(BM25Retriever): def clear(self, **kwargs) -> None: """Support deleting all nodes.""" + self._delete_json_files(kwargs.get("persist_dir")) self._nodes = []