From 3372af7c3d6cf7e181403eba07b9739d09b26110 Mon Sep 17 00:00:00 2001
From: yangdx <yangdx@znipower.com>
Date: Wed, 12 Feb 2025 22:54:22 +0800
Subject: [PATCH] refactor: remove injected db field from PGDocStatusStorage,
 it must be injected after object is created

---
 lightrag/kg/postgres_impl.py | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/lightrag/kg/postgres_impl.py b/lightrag/kg/postgres_impl.py
index 221202ab..eaa1dd92 100644
--- a/lightrag/kg/postgres_impl.py
+++ b/lightrag/kg/postgres_impl.py
@@ -418,10 +418,8 @@ class PGVectorStorage(BaseVectorStorage):
 
 @dataclass
 class PGDocStatusStorage(DocStatusStorage):
-    """PostgreSQL implementation of document status storage"""
-
     # db instance must be injected before use
-    db: PostgreSQLDB
+    # db: PostgreSQLDB
 
     async def filter_keys(self, data: set[str]) -> set[str]:
         """Return keys that don't exist in storage"""