Upload embed_documents.py

retriever/embed_documents.py

@@ -29,7 +29,7 @@ from langchain_huggingface import HuggingFaceEmbeddings
 from config import ConfigConstants  
 
 
-def embed_documents(documents: List[Dict], embedding_path: str = "/persistent/embeddings/embeddings.faiss", metadata_path: str = "/persistent/embeddings/metadata.json") -> FAISS:
+def embed_documents(documents: List[Dict], embedding_path: str = ConfigConstants.DATA_SET_PATH + "embeddings/embeddings.faiss", metadata_path: str = ConfigConstants.DATA_SET_PATH + "embeddings/metadata.json") -> FAISS:
     logging.info(f"Total documents got :{len(documents)}")
     os.makedirs(os.path.dirname(embedding_path), exist_ok=True)
     os.makedirs(os.path.dirname(metadata_path), exist_ok=True)