bupa1018 commited on
Commit
86e6eeb
·
1 Parent(s): af955d9

Update vectorstore.py

Browse files
Files changed (1) hide show
  1. vectorstore.py +1 -1
vectorstore.py CHANGED
@@ -1,6 +1,6 @@
1
  from langchain_community.vectorstores import Chroma
2
 
3
- def get_chroma_vectorstore(embeddings, vectorstore_path):
4
  vectorstore = Chroma(persist_directory=vectorstore_path, embedding_function=embedding_model)
5
 
6
  return vectorstore
 
1
  from langchain_community.vectorstores import Chroma
2
 
3
+ def get_chroma_vectorstore(embedding_model, vectorstore_path):
4
  vectorstore = Chroma(persist_directory=vectorstore_path, embedding_function=embedding_model)
5
 
6
  return vectorstore