From a2418d1373c409a77f6c9002c11d2f45abaebc43 Mon Sep 17 00:00:00 2001
From: Alex <a@tushynski.me>
Date: Mon, 29 Jan 2024 20:51:28 +0000
Subject: [PATCH] Add sentence-transformers library to requirements.txt and
 comment out model_name in base.py

---
 application/requirements.txt    | 1 +
 application/vectorstore/base.py | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/application/requirements.txt b/application/requirements.txt
index b970d982..0bb34365 100644
--- a/application/requirements.txt
+++ b/application/requirements.txt
@@ -24,6 +24,7 @@ python-dotenv==1.0.1
 redis==5.0.1
 Requests==2.31.0
 retry==0.9.2
+sentence-transformers
 tiktoken==0.5.2
 torch==2.1.2
 tqdm==4.66.1
diff --git a/application/vectorstore/base.py b/application/vectorstore/base.py
index 293d8016..b7146907 100644
--- a/application/vectorstore/base.py
+++ b/application/vectorstore/base.py
@@ -46,7 +46,7 @@ class BaseVectorStore(ABC):
             )
         elif embeddings_name == "huggingface_sentence-transformers/all-mpnet-base-v2":
             embedding_instance = embeddings_factory[embeddings_name](
-                model_name="./model/all-mpnet-base-v2",
+                #model_name="./model/all-mpnet-base-v2",
                 model_kwargs={"device": "cpu"},
             )
         else: