From 4dde73f8979f9f4aa47b59123d558ec1d7c0e991 Mon Sep 17 00:00:00 2001 From: Jin Hai Date: Fri, 10 Jan 2025 16:39:13 +0800 Subject: [PATCH] Error message: Infinity not support table parsing method (#4439) ### What problem does this PR solve? Specific error message. ### Type of change - [x] Refactoring Signed-off-by: jinhai --- rag/svr/task_executor.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/rag/svr/task_executor.py b/rag/svr/task_executor.py index a81ce55d5..cd3db9ad2 100644 --- a/rag/svr/task_executor.py +++ b/rag/svr/task_executor.py @@ -448,6 +448,13 @@ def do_handle_task(task): # prepare the progress callback function progress_callback = partial(set_progress, task_id, task_from_page, task_to_page) + # FIXME: workaround, Infinity doesn't support table parsing method, this check is to notify user + lower_case_doc_engine = settings.DOC_ENGINE.lower() + if lower_case_doc_engine == 'infinity' and task['parser_id'].lower() == 'table': + error_message = "Table parsing method is not supported by Infinity, please use other parsing methods or use Elasticsearch as the document engine." + progress_callback(-1, msg=error_message) + raise Exception(error_message) + try: task_canceled = TaskService.do_cancel(task_id) except DoesNotExist: