Spaces:
Sleeping
Sleeping
File size: 630 Bytes
5307492 dc76e41 2227dd0 ae6cea7 2227dd0 d67642e 2227dd0 86e6eeb dc76e41 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
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 |