多场景对话架构一期0525

eosphoros-ai · May 25, 2023 · 7a17234 · 7a17234
1 parent 57519b9
commit 7a17234
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git 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
@@ -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