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

Fix __post_init__ usage in Mongo and Qdrant storage implementations

Browse Source 
* Remove manual `__post_init__` in `__init__`
* Add `super().__post_init__` in vector DBs
* Ensure base validation runs correctly
* Cleanup Mongo and Qdrant init logic
This commit is contained in:
yangdx
2025-12-20 12:53:19 +08:00
parent 0ac35bfee4
commit e596512ba3
2 changed files with 10 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
lightrag/kg/mongo_impl.py
View File

@@ -89,7 +89,7 @@ class MongoKVStorage(BaseKVStorage):
global_config=global_config,
embedding_func=embedding_func,
)
self.__post_init__()
# __post_init__() is automatically called by dataclass
def __post_init__(self):
# Check for MONGODB_WORKSPACE environment variable first (higher priority)
@@ -317,7 +317,7 @@ class MongoDocStatusStorage(DocStatusStorage):
global_config=global_config,
embedding_func=embedding_func,
)
self.__post_init__()
# __post_init__() is automatically called by dataclass
def __post_init__(self):
# Check for MONGODB_WORKSPACE environment variable first (higher priority)
@@ -2052,9 +2052,12 @@ class MongoVectorDBStorage(BaseVectorStorage):
embedding_func=embedding_func,
meta_fields=meta_fields or set(),
)
self.__post_init__()
# __post_init__() is automatically called by dataclass
def __post_init__(self):
# Call parent class __post_init__ to validate embedding_func
super().__post_init__()
# Check for MONGODB_WORKSPACE environment variable first (higher priority)
# This allows administrators to force a specific workspace for all MongoDB storage instances
mongodb_workspace = os.environ.get("MONGODB_WORKSPACE")

5
lightrag/kg/qdrant_impl.py
View File

@@ -121,7 +121,7 @@ class QdrantVectorDBStorage(BaseVectorStorage):
embedding_func=embedding_func,
meta_fields=meta_fields or set(),
)
self.__post_init__()
# __post_init__() is automatically called by dataclass
@staticmethod
def setup_collection(
@@ -410,6 +410,9 @@ class QdrantVectorDBStorage(BaseVectorStorage):
)
def __post_init__(self):
# Call parent class __post_init__ to validate embedding_func
super().__post_init__()
# Check for QDRANT_WORKSPACE environment variable first (higher priority)
# This allows administrators to force a specific workspace for all Qdrant storage instances
qdrant_workspace = os.environ.get("QDRANT_WORKSPACE")