cleaned code

2025-02-19 13:42:49 +01:00
parent 05147f4799
commit 4adb4418fd
4 changed files with 20 additions and 60 deletions
--- a/lightrag/kg/mongo_impl.py
+++ b/lightrag/kg/mongo_impl.py
@@ -797,8 +797,8 @@ class MongoGraphStorage(BaseGraphStorage):
@final
@dataclass
 class MongoVectorDBStorage(BaseVectorStorage):
-    db: AsyncIOMotorDatabase = field(default=None)
-    _data: AsyncIOMotorCollection = field(default=None)
+    db: AsyncIOMotorDatabase | None = field(default=None)
+    _data: AsyncIOMotorCollection | None = field(default=None)

    def __post_init__(self):
        kwargs = self.global_config.get("vector_db_storage_cls_kwargs", {})