Add MPNet model and update vector store for Hugging Face embeddings

2025-12-03 18:43:14 +00:00 · 2024-01-09 11:39:32 +00:00
parent 001c450abb
commit b3461b7134
2 changed files with 11 additions and 0 deletions
--- a/application/vectorstore/base.py
+++ b/application/vectorstore/base.py
@@ -44,6 +44,11 @@ class BaseVectorStore(ABC):
            embedding_instance = embeddings_factory[embeddings_name](
                cohere_api_key=embeddings_key
            )
+        elif embeddings_name == "huggingface_sentence-transformers/all-mpnet-base-v2":
+            embedding_instance = embeddings_factory[embeddings_name](
+                model_name="./model/all-mpnet-base-v2",
+                model_kwargs={"device": "cpu"},
+            )
        else:
            embedding_instance = embeddings_factory[embeddings_name]()