Create utils/retriever.py

utils/retriever.py

@@ -0,0 +1,10 @@
+# utils/retriever.py
+
+import os
+from langchain.vectorstores import Chroma
+from langchain.embeddings import HuggingFaceEmbeddings
+
+def load_vectorstore(folder_path="data/vectorstore"):
+    embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2")
+    vectordb = Chroma(persist_directory=folder_path, embedding_function=embeddings)
+    return vectordb