From e42175241e369c9d3b7b6acfa6870df3060c0233 Mon Sep 17 00:00:00 2001 From: Bowen Liang Date: Fri, 23 Feb 2024 12:30:39 +0800 Subject: [PATCH] fix: tolerate exceptions in cleaning up index when vector db service unavailable (#2533) --- api/tasks/remove_document_from_index_task.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/api/tasks/remove_document_from_index_task.py b/api/tasks/remove_document_from_index_task.py index 514712991c..cff8dddc53 100644 --- a/api/tasks/remove_document_from_index_task.py +++ b/api/tasks/remove_document_from_index_task.py @@ -42,7 +42,10 @@ def remove_document_from_index_task(document_id: str): segments = db.session.query(DocumentSegment).filter(DocumentSegment.document_id == document.id).all() index_node_ids = [segment.index_node_id for segment in segments] if index_node_ids: - index_processor.clean(dataset, index_node_ids) + try: + index_processor.clean(dataset, index_node_ids) + except Exception: + logging.exception(f"clean dataset {dataset.id} from index failed") end_at = time.perf_counter() logging.info(