src/vectorstore.py

from langchain_chroma import Chroma

from src.embeddings import get_embeddings
from src.config import CHROMA_DIR

def get_vectorstore():
    return Chroma(
        collection_name="rag_collection",
        embedding_function=get_embeddings(),
        persist_directory=str(CHROMA_DIR)
    )