Update app.py

app.py

@@ -282,7 +282,7 @@ def split_into_chunks(texts, references, chunk_size, chunk_overlap):
     return chunks
 
 # Setup Vectorstore
-def setup_vectorstore(chunks, model_name):
+def setup_vectorstore(chunks, model_name, persist_directory):
     print("Start setup_vectorstore_function")
     embedding_model = HuggingFaceEmbeddings(model_name=model_name)
     
@@ -386,7 +386,7 @@ def initialize():
     #print(f"Total number of code_chunks: {len(code_chunks)}")
     print(f"Total number of doc_chunks: {len(doc_chunks)}")
 
-    docstore = setup_vectorstore(doc_chunks, EMBEDDING_MODEL_NAME)
+    docstore = setup_vectorstore(doc_chunks, EMBEDDING_MODEL_NAME, PERSIST_DOC_DIRECTORY)
     #codestore = setup_vectorstore(code_chunks, EMBEDDING_MODEL_NAME, PERSIST_CODE_DIRECTORY)
     
     #llm = setup_llm(LLM_MODEL_NAME, LLM_TEMPERATURE, GROQ_API_KEY)