gmakstutis/LightRAG
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Merge pull request #1465 from danielaskdd/main

Browse source 
Hot Fix: solved dead lock problem for FAISS vector storage
This commit is contained in:
Daniel.y 2025-04-26 00:53:44 +08:00 • committed by GitHub
commit 2d5401d475
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 5 additions and 6 deletions
Show stats 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._id_to_meta = {}
self._load_faiss_index() self._load_faiss_index()
self.storage_updated.value = False self.storage_updated.value = False
return self._index return self._index
async def upsert(self, data: dict[str, dict[str, Any]]) -> None: async def upsert(self, data: dict[str, dict[str, Any]]) -> None:
""" """
@ -362,11 +362,10 @@ class FaissVectorDBStorage(BaseVectorStorage):
logger.warning( logger.warning(
f"Storage for FAISS {self.namespace} was updated by another process, reloading..." f"Storage for FAISS {self.namespace} was updated by another process, reloading..."
) )
async with self._storage_lock: self._index = faiss.IndexFlatIP(self._dim)
self._index = faiss.IndexFlatIP(self._dim) self._id_to_meta = {}
self._id_to_meta = {} self._load_faiss_index()
self._load_faiss_index() self.storage_updated.value = False
self.storage_updated.value = False
return False # Return error return False # Return error
# Acquire lock and perform persistence # Acquire lock and perform persistence