Update src/rag_pipeline.py

src/rag_pipeline.py

@@ -4,6 +4,8 @@ from langchain_cohere import CohereEmbeddings
 from langchain_community.document_loaders import DirectoryLoader, UnstructuredPDFLoader
 from langchain_text_splitters import RecursiveCharacterTextSplitter
 
+import os
+
 base_dir = Path(__file__).resolve().parent.parent
 chroma_db_path = str(base_dir / "chroma_db_folder/chroma_db_cohere")
 docs_dir = str(base_dir / "rag_docs/")