From a0844bca2837861b2bdd9670b6ab5970d8d2334d Mon Sep 17 00:00:00 2001 From: Yannick Stephan Date: Sun, 16 Feb 2025 14:45:45 +0100 Subject: [PATCH] cleaned import --- lightrag/kg/age_impl.py | 6 +++--- lightrag/kg/chroma_impl.py | 2 +- lightrag/kg/faiss_impl.py | 2 +- lightrag/kg/gremlin_impl.py | 2 +- lightrag/kg/mongo_impl.py | 3 +-- lightrag/kg/nano_vector_db_impl.py | 2 +- lightrag/kg/neo4j_impl.py | 3 +-- lightrag/kg/networkx_impl.py | 2 +- lightrag/kg/oracle_impl.py | 2 +- lightrag/kg/postgres_impl.py | 2 +- lightrag/kg/qdrant_impl.py | 6 +++--- lightrag/kg/tidb_impl.py | 2 +- 12 files changed, 16 insertions(+), 18 deletions(-) diff --git a/lightrag/kg/age_impl.py b/lightrag/kg/age_impl.py index 24f70de9..f9499376 100644 --- a/lightrag/kg/age_impl.py +++ b/lightrag/kg/age_impl.py @@ -31,10 +31,10 @@ try: import psycopg from psycopg.rows import namedtuple_row from psycopg_pool import AsyncConnectionPool, PoolTimeout -except ImportError as e: +except ImportError: raise ImportError( - "psycopg-pool, psycopg[binary,pool], asyncpg library is not installed. Please install it to proceed." - ) from e + "`psycopg-pool, psycopg[binary,pool], asyncpg` library is not installed. Please install it via pip: `pip install psycopg-pool psycopg[binary,pool] asyncpg`." + ) class AGEQueryException(Exception): diff --git a/lightrag/kg/chroma_impl.py b/lightrag/kg/chroma_impl.py index f2d2293f..ecac2b62 100644 --- a/lightrag/kg/chroma_impl.py +++ b/lightrag/kg/chroma_impl.py @@ -11,7 +11,7 @@ try: from chromadb.config import Settings except ImportError as e: raise ImportError( - "chromadb library is not installed. Please install it to proceed." + "`chromadb` library is not installed. Please install it via pip: `pip install chromadb`." ) from e diff --git a/lightrag/kg/faiss_impl.py b/lightrag/kg/faiss_impl.py index e2c06afe..0f455f1c 100644 --- a/lightrag/kg/faiss_impl.py +++ b/lightrag/kg/faiss_impl.py @@ -20,7 +20,7 @@ try: import faiss except ImportError as e: raise ImportError( - "faiss library is not installed. Please install it to proceed." + "`faiss` library is not installed. Please install it via pip: `pip install faiss`." ) from e diff --git a/lightrag/kg/gremlin_impl.py b/lightrag/kg/gremlin_impl.py index 4038be23..d95bb00b 100644 --- a/lightrag/kg/gremlin_impl.py +++ b/lightrag/kg/gremlin_impl.py @@ -26,7 +26,7 @@ try: from gremlin_python.driver.protocol import GremlinServerError except ImportError as e: raise ImportError( - "gremlin library is not installed. Please install it to proceed." + "`gremlin` library is not installed. Please install it via pip: `pip install gremlin`." ) from e diff --git a/lightrag/kg/mongo_impl.py b/lightrag/kg/mongo_impl.py index f44332bf..330fa474 100644 --- a/lightrag/kg/mongo_impl.py +++ b/lightrag/kg/mongo_impl.py @@ -26,10 +26,9 @@ try: from pymongo.errors import PyMongoError except ImportError as e: raise ImportError( - "motor, pymongo library is not installed. Please install it to proceed." + "`motor, pymongo` library is not installed. Please install it via pip: `pip install motor pymongo`." ) from e - config = configparser.ConfigParser() config.read("config.ini", "utf-8") diff --git a/lightrag/kg/nano_vector_db_impl.py b/lightrag/kg/nano_vector_db_impl.py index 4ab98fe6..fbd6a06a 100644 --- a/lightrag/kg/nano_vector_db_impl.py +++ b/lightrag/kg/nano_vector_db_impl.py @@ -20,7 +20,7 @@ try: from nano_vectordb import NanoVectorDB except ImportError as e: raise ImportError( - "nano-vectordb library is not installed. Please install it to proceed." + "`nano-vectordb` library is not installed. Please install it via pip: `pip install nano-vectordb`." ) from e diff --git a/lightrag/kg/neo4j_impl.py b/lightrag/kg/neo4j_impl.py index 8d078af0..83edd299 100644 --- a/lightrag/kg/neo4j_impl.py +++ b/lightrag/kg/neo4j_impl.py @@ -29,10 +29,9 @@ try: ) except ImportError as e: raise ImportError( - "neo4j library is not installed. Please install it to proceed." + "`neo4j` library is not installed. Please install it via pip: `pip install neo4j`." ) from e - config = configparser.ConfigParser() config.read("config.ini", "utf-8") diff --git a/lightrag/kg/networkx_impl.py b/lightrag/kg/networkx_impl.py index f98a8bbb..04bb3bd7 100644 --- a/lightrag/kg/networkx_impl.py +++ b/lightrag/kg/networkx_impl.py @@ -19,7 +19,7 @@ try: import networkx as nx except ImportError as e: raise ImportError( - "networkx library is not installed. Please install it to proceed." + "`networkx` library is not installed. Please install it via pip: `pip install networkx`." ) from e diff --git a/lightrag/kg/oracle_impl.py b/lightrag/kg/oracle_impl.py index aec4ada4..d2d10141 100644 --- a/lightrag/kg/oracle_impl.py +++ b/lightrag/kg/oracle_impl.py @@ -23,7 +23,7 @@ try: except ImportError as e: raise ImportError( - "oracledb library is not installed. Please install it to proceed." + "`oracledb` library is not installed. Please install it via pip: `pip install oracledb`." ) from e diff --git a/lightrag/kg/postgres_impl.py b/lightrag/kg/postgres_impl.py index c63547ce..2b7996d6 100644 --- a/lightrag/kg/postgres_impl.py +++ b/lightrag/kg/postgres_impl.py @@ -40,7 +40,7 @@ try: except ImportError as e: raise ImportError( - "asyncpg, tqdm_async library is not installed. Please install it to proceed." + "`asyncpg` library is not installed. Please install it via pip: `pip install asyncpg`." ) from e diff --git a/lightrag/kg/qdrant_impl.py b/lightrag/kg/qdrant_impl.py index 1d4a0ca1..124d48d9 100644 --- a/lightrag/kg/qdrant_impl.py +++ b/lightrag/kg/qdrant_impl.py @@ -17,10 +17,10 @@ config.read("config.ini", "utf-8") try: from qdrant_client import QdrantClient, models -except ImportError as e: +except ImportError: raise ImportError( - "qdrant_client library is not installed. Please install it to proceed." - ) from e + "`qdrant_client` library is not installed. Please install it via pip: `pip install qdrant-client`." + ) def compute_mdhash_id_for_qdrant( diff --git a/lightrag/kg/tidb_impl.py b/lightrag/kg/tidb_impl.py index 69a6da2a..003316d3 100644 --- a/lightrag/kg/tidb_impl.py +++ b/lightrag/kg/tidb_impl.py @@ -18,7 +18,7 @@ try: except ImportError as e: raise ImportError( - "pymysql, sqlalchemy library is not installed. Please install it to proceed." + "`pymysql, sqlalchemy` library is not installed. Please install it via pip: `pip install pymysql sqlalchemy`." ) from e