diff --git a/pilot/vector_store/connector.py b/pilot/vector_store/connector.py index a1fce8360..ca56986c8 100644 --- a/pilot/vector_store/connector.py +++ b/pilot/vector_store/connector.py @@ -1,8 +1,9 @@ from pilot.vector_store.chroma_store import ChromaStore from pilot.vector_store.milvus_store import MilvusStore +from pilot.vector_store.weaviate_store import WeaviateStore -connector = {"Chroma": ChromaStore, "Milvus": MilvusStore} +connector = {"Chroma": ChromaStore, "Milvus": MilvusStore, "Weaviate": WeaviateStore} class VectorStoreConnector: