jarif commited on
Commit
7788b02
·
verified ·
1 Parent(s): 5b8cb2c

Update ingest.py

Browse files
Files changed (1) hide show
  1. ingest.py +6 -1
ingest.py CHANGED
@@ -10,7 +10,6 @@ logging.basicConfig(level=logging.INFO)
10
  logger = logging.getLogger(__name__)
11
 
12
  def create_vector_store():
13
- # Directory to store the vector data
14
  persist_directory = "./chroma_db"
15
  os.makedirs(persist_directory, exist_ok=True)
16
  logger.info(f"Directory '{persist_directory}' created or already exists.")
@@ -65,5 +64,11 @@ def create_vector_store():
65
  except Exception as e:
66
  logger.error(f"Failed to create Chroma vector store: {e}")
67
 
 
 
 
 
 
 
68
  if __name__ == "__main__":
69
  create_vector_store()
 
10
  logger = logging.getLogger(__name__)
11
 
12
  def create_vector_store():
 
13
  persist_directory = "./chroma_db"
14
  os.makedirs(persist_directory, exist_ok=True)
15
  logger.info(f"Directory '{persist_directory}' created or already exists.")
 
64
  except Exception as e:
65
  logger.error(f"Failed to create Chroma vector store: {e}")
66
 
67
+ # Verify creation of chroma_db directory
68
+ if os.path.exists(persist_directory):
69
+ logger.info(f"Verified that '{persist_directory}' exists.")
70
+ else:
71
+ logger.error(f"'{persist_directory}' was not created. Please check for errors.")
72
+
73
  if __name__ == "__main__":
74
  create_vector_store()