Exit program on vector index creation failure for MongoDB
Browse files
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 |
-
|
|
|
|
|
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}")
|