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") diff --git a/lightrag/kg/networkx_impl.py b/lightrag/kg/networkx_impl.py index 313d9f8d..a6d0edbb 100644 --- a/lightrag/kg/networkx_impl.py +++ b/lightrag/kg/networkx_impl.py @@ -20,13 +20,8 @@ if not pm.is_installed("networkx"): if not pm.is_installed("graspologic"): pm.install("graspologic") -try: - from graspologic import embed - import networkx as nx -except ImportError as e: - raise ImportError( - "`networkx` library is not installed. Please install it via pip: `pip install networkx`." - ) from e +from graspologic import embed +import networkx as nx @final