ArnoChen commited on
Commit
b688c30
·
1 Parent(s): 973c7b6

fix configuration errors of milvus

Browse files
Files changed (1) hide show
  1. lightrag/kg/milvus_impl.py +2 -2
lightrag/kg/milvus_impl.py CHANGED
@@ -32,8 +32,8 @@ class MilvusVectorDBStorage(BaseVectorStorage):
32
  )
33
 
34
  def __post_init__(self):
35
- config = self.global_config.get("vector_db_storage_cls_kwargs", {})
36
- cosine_threshold = config.get("cosine_better_than_threshold")
37
  if cosine_threshold is None:
38
  raise ValueError(
39
  "cosine_better_than_threshold must be specified in vector_db_storage_cls_kwargs"
 
32
  )
33
 
34
  def __post_init__(self):
35
+ kwargs = self.global_config.get("vector_db_storage_cls_kwargs", {})
36
+ cosine_threshold = kwargs.get("cosine_better_than_threshold")
37
  if cosine_threshold is None:
38
  raise ValueError(
39
  "cosine_better_than_threshold must be specified in vector_db_storage_cls_kwargs"