From 9e823de74e47ec88d2e5a145486869a9bdf6e69c Mon Sep 17 00:00:00 2001 From: yangdx Date: Mon, 7 Jul 2025 04:43:46 +0800 Subject: [PATCH] Exit program on vector index creation failure for MongoDB --- lightrag/kg/mongo_impl.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/lightrag/kg/mongo_impl.py b/lightrag/kg/mongo_impl.py index 527ef5db..fd0e6d15 100644 --- a/lightrag/kg/mongo_impl.py +++ b/lightrag/kg/mongo_impl.py @@ -1419,7 +1419,9 @@ class MongoVectorDBStorage(BaseVectorStorage): logger.info(f"Vector index {self._index_name} created successfully.") except PyMongoError as e: - logger.error(f"Error creating vector index {self._index_name}: {e}") + error_msg = f"Error creating vector index {self._index_name}: {e}" + logger.error(error_msg) + raise SystemExit(f"Failed to create MongoDB vector index. Program cannot continue. {error_msg}") async def upsert(self, data: dict[str, dict[str, Any]]) -> None: logger.debug(f"Inserting {len(data)} to {self.namespace}")