from langchain_huggingface import HuggingFaceEmbeddings
from langchain_community.vectorstores import Chroma

# Setup Vectorstore
def setup_vectorstore(docs, model_name, persist_directory):
    print("Start setup_vectorstore_function")
    embedding_model = HuggingFaceEmbeddings(model_name=model_name)   
    vectorstore = get_chroma_vectorstore(embedding_model, persist_directory)
    vectorstore.add_documents(docs)
    return vectorstore


def get_chroma_vectorstore(embedding_model, vectorstore_path):
    vectorstore = Chroma(persist_directory=vectorstore_path, embedding_function=embedding_model)
    
    return vectorstore