gardel/lightrag
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor: remove injected db field from PGDocStatusStorage, it must be injected after object is created

Browse Source
This commit is contained in:
yangdx
2025-02-12 22:54:22 +08:00
parent 7b79427097
commit 3372af7c3d
1 changed files with 1 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
lightrag/kg/postgres_impl.py
View File

@@ -418,10 +418,8 @@ class PGVectorStorage(BaseVectorStorage):
@dataclass @dataclass
class PGDocStatusStorage(DocStatusStorage): class PGDocStatusStorage(DocStatusStorage):
"""PostgreSQL implementation of document status storage"""
# db instance must be injected before use # db instance must be injected before use
db: PostgreSQLDB # db: PostgreSQLDB
async def filter_keys(self, data: set[str]) -> set[str]: async def filter_keys(self, data: set[str]) -> set[str]:
"""Return keys that don't exist in storage""" """Return keys that don't exist in storage"""