gardel/lightrag
1
0
Fork 0
Code Issues Pull Requests Actions 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
Download Patch File Download Diff File Expand all files Collapse all files

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"]