diff --git a/pilot/vector_store/connector.py b/pilot/vector_store/connector.py
index 3ff473f1e..06fad00f2 100644
--- a/pilot/vector_store/connector.py
+++ b/pilot/vector_store/connector.py
@@ -1,7 +1,7 @@
 from pilot.vector_store.chroma_store import ChromaStore
-# from pilot.vector_store.milvus_store import MilvusStore
+from pilot.vector_store.milvus_store import MilvusStore
 
-connector = {"Chroma": ChromaStore, "Milvus": None}
+connector = {"Chroma": ChromaStore, "Milvus": MilvusStore}
 
 
 class VectorStoreConnector:
diff --git a/requirements.txt b/requirements.txt
index 9595854bb..19d8ca34e 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -54,7 +54,7 @@ gTTS==2.3.1
 langchain
 nltk
 python-dotenv==1.0.0
-# pymilvus==2.2.1
+pymilvus==2.2.1
 vcrpy
 chromadb
 markdown2