Update utils.py

utils.py

@@ -214,8 +214,9 @@ def document_loading_splitting():
 #Chroma DB die splits ablegen - vektorisiert...
 def document_storage_chroma(splits):
     #HF embeddings--------------------------------------
-    Chroma.from_documents(documents = splits, embedding = HuggingFaceEmbeddings(model_name="sentence-transformers/all-mpnet-base-v2", model_kwargs={"device": "cpu"}, encode_kwargs={'normalize_embeddings': False}),  persist_directory = PATH_WORK + CHROMA_DIR)  
-    
+    vectorstore = Chroma.from_documents(documents = splits, embedding = HuggingFaceEmbeddings(model_name="sentence-transformers/all-mpnet-base-v2", model_kwargs={"device": "cpu"}, encode_kwargs={'normalize_embeddings': False}),  persist_directory = PATH_WORK + CHROMA_DIR)  
+    retriever = vectorstore.as_retriever(search_kwargs = {"k": ANZAHL_DOCS})
+    return vectorstore, retriever
 
 ############################################
 #dokumente in chroma db vektorisiert ablegen können - die Db vorbereiten daüfur