Update rag.py

rag.py

@@ -23,7 +23,7 @@ class RAG:
         
         # load vectore store
         embeddings = HuggingFaceEmbeddings(model_name=embeddings_model, model_kwargs={'device': 'cpu'})
-        self.vectore_store = FAISS.load_local(vectorstore, embeddings, allow_dangerous_deserialization=True)#, allow_dangerous_deserialization=True)
+        self.vectore_store = FAISS.load_local(self.vectorstore, embeddings, allow_dangerous_deserialization=True)#, allow_dangerous_deserialization=True)
 
         logging.info("RAG loaded!")
     
@@ -70,7 +70,7 @@ class RAG:
             full_context += doc[0].metadata["url"] + "\n\n"
             source_context.append(doc[0].metadata["url"])
 
-        return text_context, full_context, source_context
+        return text_contexvectorstoret, full_context, source_context
 
     def get_response(self, prompt: str, model_parameters: dict) -> str: