alexkueck commited on
Commit
46dda78
1 Parent(s): 8445624

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +1 -1
utils.py CHANGED
@@ -341,7 +341,7 @@ def document_storage_chroma(splits):
341
  embedding_fn = HuggingFaceEmbeddings(model_name="sentence-transformers/all-mpnet-base-v2", model_kwargs={"device": "cpu"}, encode_kwargs={'normalize_embeddings': False})
342
 
343
  # Vectorstore initialisieren und Dokumente hinzufügen
344
- vectorstore = Chroma.from_documents(documents=splits, embedding=embedding_fn, persist_directory=CHROMA_DIR)
345
  retriever = vectorstore.as_retriever(search_kwargs = {"k": ANZAHL_DOCS})
346
  # Persist the vectorstore to disk
347
  vectorstore.persist()
 
341
  embedding_fn = HuggingFaceEmbeddings(model_name="sentence-transformers/all-mpnet-base-v2", model_kwargs={"device": "cpu"}, encode_kwargs={'normalize_embeddings': False})
342
 
343
  # Vectorstore initialisieren und Dokumente hinzufügen
344
+ vectorstore = Chroma.from_documents(documents=splits, embedding=embedding_fn, persist_directory = PATH_WORK + CHROMA_DIR)
345
  retriever = vectorstore.as_retriever(search_kwargs = {"k": ANZAHL_DOCS})
346
  # Persist the vectorstore to disk
347
  vectorstore.persist()