From 307184985b05cabee01e7a1929b76e8c0327fc1d Mon Sep 17 00:00:00 2001
From: ArnoChen <arnochen101@gmail.com>
Date: Fri, 14 Feb 2025 03:00:56 +0800
Subject: [PATCH] fix configuration errors of milvus

---
 lightrag/kg/milvus_impl.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/lightrag/kg/milvus_impl.py b/lightrag/kg/milvus_impl.py
index 88a8f475..f4d9d47f 100644
--- a/lightrag/kg/milvus_impl.py
+++ b/lightrag/kg/milvus_impl.py
@@ -32,8 +32,8 @@ class MilvusVectorDBStorage(BaseVectorStorage):
         )
 
     def __post_init__(self):
-        config = self.global_config.get("vector_db_storage_cls_kwargs", {})
-        cosine_threshold = config.get("cosine_better_than_threshold")
+        kwargs = self.global_config.get("vector_db_storage_cls_kwargs", {})
+        cosine_threshold = kwargs.get("cosine_better_than_threshold")
         if cosine_threshold is None:
             raise ValueError(
                 "cosine_better_than_threshold must be specified in vector_db_storage_cls_kwargs"