return store when calling faiss_store write

tshong · Jul 28, 2023 · b1dda7c · b1dda7c
1 parent 8cc8b80
commit b1dda7c
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/metagpt/document_store/faiss_store.py b/metagpt/document_store/faiss_store.py
@@ -67,6 +67,7 @@ def write(self):
 
         self.store = self._write(docs, metadatas)
         self.persist()
+        return self.store
 
     def add(self, texts: list[str], *args, **kwargs) -> list[str]:
         """FIXME: 目前add之后没有更新store"""