Merge pull request #873 from ParisNeo/main

Removed all useless try except around imports as pipmaster already handles that

lightrag/kg/age_impl.py

@@ -34,14 +34,9 @@ if not pm.is_installed("psycopg-pool"):
 if not pm.is_installed("asyncpg"):
     pm.install("asyncpg")
 
-try:
 import psycopg
 from psycopg.rows import namedtuple_row
 from psycopg_pool import AsyncConnectionPool, PoolTimeout
-except ImportError:
-    raise ImportError(
-        "`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):

lightrag/kg/chroma_impl.py

@@ -10,13 +10,8 @@ import pipmaster as pm
 if not pm.is_installed("chromadb"):
     pm.install("chromadb")
 
-try:
 from chromadb import HttpClient, PersistentClient
 from chromadb.config import Settings
-except ImportError as e:
-    raise ImportError(
-        "`chromadb` library is not installed. Please install it via pip: `pip install chromadb`."
-    ) from e
 
 
 @final

lightrag/kg/faiss_impl.py

@@ -20,12 +20,7 @@ from lightrag.base import (
 if not pm.is_installed("faiss"):
     pm.install("faiss")
 
-try:
 import faiss
-except ImportError as e:
-    raise ImportError(
-        "`faiss` library is not installed. Please install it via pip: `pip install faiss`."
-    ) from e
 
 
 @final

lightrag/kg/gremlin_impl.py

@@ -2,6 +2,7 @@ import asyncio
 import inspect
 import json
 import os
+import pipmaster as pm
 from dataclasses import dataclass
 from typing import Any, Dict, List, final
 
@@ -20,14 +21,12 @@ from lightrag.utils import logger
 
 from ..base import BaseGraphStorage
 
-try:
+if not pm.is_installed("gremlinpython"):
+    pm.install("gremlinpython")
+
 from gremlin_python.driver import client, serializer
 from gremlin_python.driver.aiohttp.transport import AiohttpTransport
 from gremlin_python.driver.protocol import GremlinServerError
-except ImportError as e:
-    raise ImportError(
-        "`gremlin` library is not installed. Please install it via pip: `pip install gremlin`."
-    ) from e
 
 
 @final

lightrag/kg/milvus_impl.py

@@ -14,13 +14,8 @@ if not pm.is_installed("configparser"):
 if not pm.is_installed("pymilvus"):
     pm.install("pymilvus")
 
-try:
 import configparser
 from pymilvus import MilvusClient
-except ImportError as e:
-    raise ImportError(
-        "`pymilvus` library is not installed. Please install it via pip: `pip install pymilvus`."
-    ) from e
 
 config = configparser.ConfigParser()
 config.read("config.ini", "utf-8")

lightrag/kg/mongo_impl.py

@@ -25,7 +25,6 @@ if not pm.is_installed("pymongo"):
 if not pm.is_installed("motor"):
     pm.install("motor")
 
-try:
 from motor.motor_asyncio import (
     AsyncIOMotorClient,
     AsyncIOMotorDatabase,
@@ -33,10 +32,6 @@ try:
 )
 from pymongo.operations import SearchIndexModel
 from pymongo.errors import PyMongoError
-except ImportError as e:
-    raise ImportError(
-        "`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")

lightrag/kg/nano_vector_db_impl.py

@@ -18,12 +18,7 @@ from lightrag.base import (
 if not pm.is_installed("nano-vectordb"):
     pm.install("nano-vectordb")
 
-try:
 from nano_vectordb import NanoVectorDB
-except ImportError as e:
-    raise ImportError(
-        "`nano-vectordb` library is not installed. Please install it via pip: `pip install nano-vectordb`."
-    ) from e
 
 
 @final

lightrag/kg/oracle_impl.py

@@ -26,15 +26,9 @@ if not pm.is_installed("graspologic"):
 if not pm.is_installed("oracledb"):
     pm.install("oracledb")
 
-try:
 from graspologic import embed
 import oracledb
 
-except ImportError as e:
-    raise ImportError(
-        "`oracledb` library is not installed. Please install it via pip: `pip install oracledb`."
-    ) from e
-
 
 class OracleDB:
     def __init__(self, config, **kwargs):

lightrag/kg/postgres_impl.py

@@ -38,15 +38,9 @@ import pipmaster as pm
 if not pm.is_installed("asyncpg"):
     pm.install("asyncpg")
 
-try:
 import asyncpg
 from asyncpg import Pool
 
-except ImportError as e:
-    raise ImportError(
-        "`asyncpg` library is not installed. Please install it via pip: `pip install asyncpg`."
-    ) from e
-
 
 class PostgreSQLDB:
     def __init__(self, config: dict[str, Any], **kwargs: Any):

lightrag/kg/qdrant_impl.py

@@ -15,17 +15,11 @@ config.read("config.ini", "utf-8")
 
 import pipmaster as pm
 
-if not pm.is_installed("qdrant_client"):
+if not pm.is_installed("qdrant-client"):
-    pm.install("qdrant_client")
+    pm.install("qdrant-client")
 
-try:
 from qdrant_client import QdrantClient, models
 
-except ImportError:
-    raise ImportError(
-        "`qdrant_client` library is not installed. Please install it via pip: `pip install qdrant-client`."
-    )
-
 
 def compute_mdhash_id_for_qdrant(
     content: str, prefix: str = "", style: str = "simple"

lightrag/kg/tidb_impl.py

@@ -20,14 +20,8 @@ if not pm.is_installed("pymysql"):
 if not pm.is_installed("sqlalchemy"):
     pm.install("sqlalchemy")
 
-try:
 from sqlalchemy import create_engine, text
 
-except ImportError as e:
-    raise ImportError(
-        "`pymysql, sqlalchemy` library is not installed. Please install it via pip: `pip install pymysql sqlalchemy`."
-    ) from e
-
 
 class TiDB:
     def __init__(self, config, **kwargs):