From 214e3e8ad5c4d479d73afc3aee72ecdbfd3b0bf3 Mon Sep 17 00:00:00 2001 From: Yannick Stephan Date: Thu, 20 Feb 2025 14:12:19 +0100 Subject: [PATCH] fixed last update --- examples/test_faiss.py | 2 +- lightrag/__init__.py | 2 +- lightrag/kg/networkx_impl.py | 4 ++-- lightrag/lightrag.py | 5 ++--- 4 files changed, 6 insertions(+), 7 deletions(-) diff --git a/examples/test_faiss.py b/examples/test_faiss.py index ab0ef9f7..c3ac6f47 100644 --- a/examples/test_faiss.py +++ b/examples/test_faiss.py @@ -70,7 +70,7 @@ def main(): ), vector_storage="FaissVectorDBStorage", vector_db_storage_cls_kwargs={ - "cosine_better_than_threshold": 0.3 # Your desired threshold + "cosine_better_than_threshold": 0.2 # Your desired threshold }, ) diff --git a/lightrag/__init__.py b/lightrag/__init__.py index 025fb73b..99f4052f 100644 --- a/lightrag/__init__.py +++ b/lightrag/__init__.py @@ -1,5 +1,5 @@ from .lightrag import LightRAG as LightRAG, QueryParam as QueryParam -__version__ = "1.1.7" +__version__ = "1.1.10" __author__ = "Zirui Guo" __url__ = "https://github.com/HKUDS/LightRAG" diff --git a/lightrag/kg/networkx_impl.py b/lightrag/kg/networkx_impl.py index 614715c4..853bd369 100644 --- a/lightrag/kg/networkx_impl.py +++ b/lightrag/kg/networkx_impl.py @@ -16,12 +16,12 @@ import pipmaster as pm if not pm.is_installed("networkx"): pm.install("networkx") + if not pm.is_installed("graspologic"): pm.install("graspologic") -from graspologic import embed import networkx as nx - +from graspologic import embed @final @dataclass diff --git a/lightrag/lightrag.py b/lightrag/lightrag.py index 990c1bcf..38a6e835 100644 --- a/lightrag/lightrag.py +++ b/lightrag/lightrag.py @@ -738,9 +738,8 @@ class LightRAG: if new_kg is None: logger.info("No new entities or relationships extracted.") else: - async with self._entity_lock: - logger.info("New entities or relationships extracted.") - self.chunk_entity_relation_graph = new_kg + logger.info("New entities or relationships extracted.") + self.chunk_entity_relation_graph = new_kg except Exception as e: logger.error("Failed to extract entities and relationships")