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

Fix cosine threshold parameter setting error

Browse Source
This commit is contained in:
yangdx
2025-01-29 21:09:11 +08:00
parent 4302c65a04
commit d0052456d4
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
lightrag/kg/nano_vector_db_impl.py
View File

@@ -83,8 +83,11 @@ class NanoVectorDBStorage(BaseVectorStorage):
self._client = NanoVectorDB(
self.embedding_func.embedding_dim, storage_file=self._client_file_name
)
self.cosine_better_than_threshold = self.global_config.get(
"cosine_better_than_threshold", self.cosine_better_than_threshold
# get cosine_better_than_threshold from LightRAG
vector_db_kwargs = self.global_config.get("vector_db_storage_cls_kwargs", {})
self.cosine_better_than_threshold = vector_db_kwargs.get(
"cosine_better_than_threshold",
self.global_config.get("cosine_better_than_threshold", self.cosine_better_than_threshold)
)
async def upsert(self, data: dict[str, dict]):