Refactor MongoDBVectorStore and add delete_index method

2025-11-29 08:33:20 +00:00 · 2024-01-06 17:59:01 +00:00
parent 293b7b09a9
commit 4ff834de76
2 changed files with 52 additions and 16 deletions
--- a/application/vectorstore/vector_creator.py
+++ b/application/vectorstore/vector_creator.py
@@ -1,11 +1,13 @@
 from application.vectorstore.faiss import FaissStore
 from application.vectorstore.elasticsearch import ElasticsearchStore
+from application.vectorstore.mongodb import MongoDBVectorStore


 class VectorCreator:
    vectorstores = {
        'faiss': FaissStore,
-        'elasticsearch':ElasticsearchStore
+        'elasticsearch':ElasticsearchStore,
+        'mongodb': MongoDBVectorStore,
    }

    @classmethod