Update pages/jarvis.py

pages/jarvis.py

@@ -31,7 +31,7 @@ def load_doc(list_file_path, chunk_size, chunk_overlap):
     doc_splits = text_splitter.split_documents(pages)
     return doc_splits
 
-# Function to create vector database
+# Create vector database
 def create_db(splits, collection_name):
     embedding = HuggingFaceEmbeddings()
     new_client = chromadb.EphemeralClient()
@@ -40,9 +40,19 @@ def create_db(splits, collection_name):
         embedding=embedding,
         client=new_client,
         collection_name=collection_name,
+        # persist_directory=default_persist_directory
     )
     return vectordb
 
+
+# Load vector database
+def load_db():
+    embedding = HuggingFaceEmbeddings()
+    vectordb = Chroma(
+        # persist_directory=default_persist_directory, 
+        embedding_function=embedding)
+    return vectordb
+
 # Initialize Langchain LLM chain
 def initialize_llmchain(llm_model, temperature, max_tokens, top_k, vector_db):
     if llm_model == "mistralai/Mixtral-8x7B-Instruct-v0.1":