diff --git a/lightrag/kg/neo4j_impl.py b/lightrag/kg/neo4j_impl.py index 82631cf8..0ddc611d 100644 --- a/lightrag/kg/neo4j_impl.py +++ b/lightrag/kg/neo4j_impl.py @@ -23,18 +23,13 @@ import pipmaster as pm if not pm.is_installed("neo4j"): pm.install("neo4j") -try: - from neo4j import ( - AsyncGraphDatabase, - exceptions as neo4jExceptions, - AsyncDriver, - AsyncManagedTransaction, - GraphDatabase, - ) -except ImportError as e: - raise ImportError( - "`neo4j` library is not installed. Please install it via pip: `pip install neo4j`." - ) from e +from neo4j import ( + AsyncGraphDatabase, + exceptions as neo4jExceptions, + AsyncDriver, + AsyncManagedTransaction, + GraphDatabase, +) config = configparser.ConfigParser() config.read("config.ini", "utf-8")