yujunjun/LightRAG
1
0
Fork 0
mirror of https://github.com/HKUDS/LightRAG.git synced 2025-12-27 14:52:34 +00:00
Code Issues Packages Projects Releases Wiki Activity

change init=False to default=None for db fields to prevent no attribute error

Browse Source
This commit is contained in:
ArnoChen 2025-02-19 04:55:59 +08:00
parent f3b030e1a3
commit bfc548edf2
4 changed files with 18 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
lightrag/kg/mongo_impl.py
View File

@ -83,8 +83,8 @@ class ClientManager:
@final
@dataclass
class MongoKVStorage(BaseKVStorage):
db: AsyncIOMotorDatabase = field(init=False)
_data: AsyncIOMotorCollection = field(init=False)
db: AsyncIOMotorDatabase = field(default=None)
_data: AsyncIOMotorCollection = field(default=None)
def __post_init__(self):
self._collection_name = self.namespace
@ -156,8 +156,8 @@ class MongoKVStorage(BaseKVStorage):
@final
@dataclass
class MongoDocStatusStorage(DocStatusStorage):
db: AsyncIOMotorDatabase = field(init=False)
_data: AsyncIOMotorCollection = field(init=False)
db: AsyncIOMotorDatabase = field(default=None)
_data: AsyncIOMotorCollection = field(default=None)
def __post_init__(self):
self._collection_name = self.namespace
@ -236,8 +236,8 @@ class MongoGraphStorage(BaseGraphStorage):
A concrete implementation using MongoDB's $graphLookup to demonstrate multi-hop queries.
"""
db: AsyncIOMotorDatabase = field(init=False)
collection: AsyncIOMotorCollection = field(init=False)
db: AsyncIOMotorDatabase = field(default=None)
collection: AsyncIOMotorCollection = field(default=None)
def __init__(self, namespace, global_config, embedding_func):
super().__init__(
@ -798,8 +798,8 @@ class MongoGraphStorage(BaseGraphStorage):
@final
@dataclass
class MongoVectorDBStorage(BaseVectorStorage):
db: AsyncIOMotorDatabase = field(init=False)
_data: AsyncIOMotorCollection = field(init=False)
db: AsyncIOMotorDatabase = field(default=None)
_data: AsyncIOMotorCollection = field(default=None)
def __post_init__(self):
kwargs = self.global_config.get("vector_db_storage_cls_kwargs", {})

6
lightrag/kg/oracle_impl.py
View File

@ -242,7 +242,7 @@ class ClientManager:
@final
@dataclass
class OracleKVStorage(BaseKVStorage):
db: OracleDB = field(init=False)
db: OracleDB = field(default=None)
meta_fields = None
def __post_init__(self):
@ -394,7 +394,7 @@ class OracleKVStorage(BaseKVStorage):
@final
@dataclass
class OracleVectorDBStorage(BaseVectorStorage):
db: OracleDB = field(init=False)
db: OracleDB = field(default=None)
def __post_init__(self):
config = self.global_config.get("vector_db_storage_cls_kwargs", {})
@ -450,7 +450,7 @@ class OracleVectorDBStorage(BaseVectorStorage):
@final
@dataclass
class OracleGraphStorage(BaseGraphStorage):
db: OracleDB = field(init=False)
db: OracleDB = field(default=None)
def __post_init__(self):
self._max_batch_size = self.global_config.get("embedding_batch_num", 10)

8
lightrag/kg/postgres_impl.py
View File

@ -246,7 +246,7 @@ class ClientManager:
@final
@dataclass
class PGKVStorage(BaseKVStorage):
db: PostgreSQLDB = field(init=False)
db: PostgreSQLDB = field(default=None)
def __post_init__(self):
self._max_batch_size = self.global_config["embedding_batch_num"]
@ -378,7 +378,7 @@ class PGKVStorage(BaseKVStorage):
@final
@dataclass
class PGVectorStorage(BaseVectorStorage):
db: PostgreSQLDB = field(init=False)
db: PostgreSQLDB = field(default=None)
def __post_init__(self):
self._max_batch_size = self.global_config["embedding_batch_num"]
@ -515,7 +515,7 @@ class PGVectorStorage(BaseVectorStorage):
@final
@dataclass
class PGDocStatusStorage(DocStatusStorage):
db: PostgreSQLDB = field(init=False)
db: PostgreSQLDB = field(default=None)
async def initialize(self):
if self.db is None:
@ -665,7 +665,7 @@ class PGGraphQueryException(Exception):
@final
@dataclass
class PGGraphStorage(BaseGraphStorage):
db: PostgreSQLDB = field(init=False)
db: PostgreSQLDB = field(default=None)
@staticmethod
def load_nx_graph(file_name):

6
lightrag/kg/tidb_impl.py
View File

@ -166,7 +166,7 @@ class ClientManager:
@final
@dataclass
class TiDBKVStorage(BaseKVStorage):
db: TiDB = field(init=False)
db: TiDB = field(default=None)
def __post_init__(self):
self._data = {}
@ -279,7 +279,7 @@ class TiDBKVStorage(BaseKVStorage):
@final
@dataclass
class TiDBVectorDBStorage(BaseVectorStorage):
db: TiDB = field(init=False)
db: TiDB = field(default=None)
def __post_init__(self):
self._client_file_name = os.path.join(
@ -422,7 +422,7 @@ class TiDBVectorDBStorage(BaseVectorStorage):
@final
@dataclass
class TiDBGraphStorage(BaseGraphStorage):
db: TiDB = field(init=False)
db: TiDB = field(default=None)
def __post_init__(self):
self._max_batch_size = self.global_config["embedding_batch_num"]