Merge pull request #20 from almutareb/fix_fetch_s3_vs

app_gui.py

@@ -1,7 +1,10 @@
 # Import Gradio for UI, along with other necessary libraries
 import gradio as gr
+from rag_app.loading_data.load_S3_vector_stores import get_chroma_vs
 from rag_app.agents.react_agent import agent_executor
 
+get_chroma_vs()
+
 
 if __name__ == "__main__":
 

rag_app/loading_data/load_S3_vector_stores.py

@@ -32,7 +32,7 @@ embeddings = SentenceTransformerEmbeddings(model_name=model_name)
 
 ## FAISS
 def get_faiss_vs():
-    if os.listdir(FAISS_INDEX_PATH) == 0:
+    if not os.path.exists(FAISS_INDEX_PATH):
         # Initialize an S3 client with unsigned configuration for public access
         s3 = boto3.client('s3', config=Config(signature_version=UNSIGNED))
 
@@ -56,7 +56,7 @@ def get_faiss_vs():
 
 ## Chroma DB
 def get_chroma_vs():
-    if os.listdir(CHROMA_DIRECTORY) == 0:
+    if not os.path.exists(CHROMA_DIRECTORY):
         # Initialize an S3 client with unsigned configuration for public access
         s3 = boto3.client('s3', config=Config(signature_version=UNSIGNED))
 

rag_app/structured_tools/structured_tools.py

@@ -23,7 +23,7 @@ import os
 
 persist_directory = os.getenv('VECTOR_DATABASE_LOCATION')
 embedding_model = os.getenv("EMBEDDING_MODEL")
-if os.listdir(persist_directory) == 0:
+if not os.path.exists(persist_directory):
     get_chroma_vs()
 
 @tool