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

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):