Update vectorstore.py

vectorstore.py

@@ -1,5 +1,14 @@
 from langchain_community.vectorstores import Chroma
 
+# Setup Vectorstore
+def setup_vectorstore(chunks, 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(chunks)
+    return vectorstore
+
+
 def get_chroma_vectorstore(embedding_model, vectorstore_path):
     vectorstore = Chroma(persist_directory=vectorstore_path, embedding_function=embedding_model)