Update rag.py

rag.py

@@ -11,8 +11,9 @@ from langchain_community.embeddings import HuggingFaceEmbeddings
 class RAG:
     NO_ANSWER_MESSAGE: str = "Ho sento, no he pogut respondre la teva pregunta."
 
-    vectorstore = "index-intfloat_multilingual-e5-small-500-100-CA-ES" # mixed
+    #vectorstore = "index-intfloat_multilingual-e5-small-500-100-CA-ES" # mixed
     #vectorstore = "vectorestore" # CA only
+    vectorstore = "index-BAAI_bge-m3-1500-200-recursive_splitter-CA_ES"
 
     def __init__(self, hf_token, embeddings_model, model_name):
 
@@ -22,7 +23,7 @@ class RAG:
         
         # load vectore store
         embeddings = HuggingFaceEmbeddings(model_name=embeddings_model, model_kwargs={'device': 'cpu'})
-        self.vectore_store = FAISS.load_local("index-intfloat_multilingual-e5-small-500-100-CA-ES", embeddings, allow_dangerous_deserialization=True)#, allow_dangerous_deserialization=True)
+        self.vectore_store = FAISS.load_local(vectorstore, embeddings, allow_dangerous_deserialization=True)#, allow_dangerous_deserialization=True)
 
         logging.info("RAG loaded!")