gzdaniel commited on
Commit
a60fe01
·
1 Parent(s): 7cd4bea

Exit program on vector index creation failure for MongoDB

Browse files
Files changed (1) hide show
  1. lightrag/kg/mongo_impl.py +3 -1
lightrag/kg/mongo_impl.py CHANGED
@@ -1419,7 +1419,9 @@ class MongoVectorDBStorage(BaseVectorStorage):
1419
  logger.info(f"Vector index {self._index_name} created successfully.")
1420
 
1421
  except PyMongoError as e:
1422
- logger.error(f"Error creating vector index {self._index_name}: {e}")
 
 
1423
 
1424
  async def upsert(self, data: dict[str, dict[str, Any]]) -> None:
1425
  logger.debug(f"Inserting {len(data)} to {self.namespace}")
 
1419
  logger.info(f"Vector index {self._index_name} created successfully.")
1420
 
1421
  except PyMongoError as e:
1422
+ error_msg = f"Error creating vector index {self._index_name}: {e}"
1423
+ logger.error(error_msg)
1424
+ raise SystemExit(f"Failed to create MongoDB vector index. Program cannot continue. {error_msg}")
1425
 
1426
  async def upsert(self, data: dict[str, dict[str, Any]]) -> None:
1427
  logger.debug(f"Inserting {len(data)} to {self.namespace}")