Rulga commited on
Commit
f631be3
·
1 Parent(s): a38fa3f

Improve knowledge base loading logic and logging in app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -248,7 +248,7 @@ try:
248
  llm, embeddings = init_models()
249
  vector_store = None
250
 
251
- if os.path.exists(VECTOR_STORE_PATH):
252
  try:
253
  vector_store = FAISS.load_local(
254
  VECTOR_STORE_PATH,
@@ -257,8 +257,10 @@ try:
257
  )
258
  logger.info("Successfully loaded existing knowledge base")
259
  except Exception as e:
260
- logger.error(f"Failed to load existing knowledge base: {str(e)}")
261
- logger.error(traceback.format_exc())
 
 
262
 
263
  if vector_store is None:
264
  logger.info("Building new knowledge base...")
 
248
  llm, embeddings = init_models()
249
  vector_store = None
250
 
251
+ if os.path.exists(os.path.join(VECTOR_STORE_PATH, "index.faiss")):
252
  try:
253
  vector_store = FAISS.load_local(
254
  VECTOR_STORE_PATH,
 
257
  )
258
  logger.info("Successfully loaded existing knowledge base")
259
  except Exception as e:
260
+ logger.warning(f"Could not load existing knowledge base, will create new one: {str(e)}")
261
+ vector_store = None
262
+ else:
263
+ logger.info("No existing knowledge base found, will create new one")
264
 
265
  if vector_store is None:
266
  logger.info("Building new knowledge base...")