isayahc commited on
Commit
6f22fc5
·
unverified ·
1 Parent(s): 76dab82

using a constant over hardcoded value

Browse files
Files changed (1) hide show
  1. rag_app/get_db_retriever.py +2 -2
rag_app/get_db_retriever.py CHANGED
@@ -10,11 +10,11 @@ from langchain.chains import RetrievalQA
10
  # prompt template
11
  from langchain.prompts import PromptTemplate
12
  from langchain.memory import ConversationBufferMemory
 
13
 
14
 
15
  def get_db_retriever(vector_db:str=None):
16
- model_name = "sentence-transformers/multi-qa-mpnet-base-dot-v1"
17
- embeddings = HuggingFaceHubEmbeddings(repo_id=model_name)
18
 
19
  #db = Chroma(persist_directory="./vectorstore/lc-chroma-multi-mpnet-500", embedding_function=embeddings)
20
  #db.get()
 
10
  # prompt template
11
  from langchain.prompts import PromptTemplate
12
  from langchain.memory import ConversationBufferMemory
13
+ from config import EMBEDDING_MODEL
14
 
15
 
16
  def get_db_retriever(vector_db:str=None):
17
+ embeddings = HuggingFaceHubEmbeddings(repo_id=EMBEDDING_MODEL)
 
18
 
19
  #db = Chroma(persist_directory="./vectorstore/lc-chroma-multi-mpnet-500", embedding_function=embeddings)
20
  #db.get()