gardel/lightrag
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix dead lock problem for FAISS vectory storage

Browse Source
This commit is contained in:
yangdx
2025-04-26 00:48:30 +08:00
parent dff73f078d
commit d8c530364f
1 changed files with 5 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
lightrag/kg/faiss_impl.py
View File

@@ -85,7 +85,7 @@ class FaissVectorDBStorage(BaseVectorStorage):
self._id_to_meta = {}
self._load_faiss_index()
self.storage_updated.value = False
return self._index
return self._index
async def upsert(self, data: dict[str, dict[str, Any]]) -> None:
"""
@@ -362,11 +362,10 @@ class FaissVectorDBStorage(BaseVectorStorage):
logger.warning(
f"Storage for FAISS {self.namespace} was updated by another process, reloading..."
)
async with self._storage_lock:
self._index = faiss.IndexFlatIP(self._dim)
self._id_to_meta = {}
self._load_faiss_index()
self.storage_updated.value = False
self._index = faiss.IndexFlatIP(self._dim)
self._id_to_meta = {}
self._load_faiss_index()
self.storage_updated.value = False
return False # Return error
# Acquire lock and perform persistence