diff --git a/api/db/services/task_service.py b/api/db/services/task_service.py index 20887cab0f..10ff8c7c29 100644 --- a/api/db/services/task_service.py +++ b/api/db/services/task_service.py @@ -390,14 +390,15 @@ def queue_tasks(doc: dict, bucket: str, name: str, priority: int): if doc["type"] == FileType.PDF.value: file_bin = settings.STORAGE_IMPL.get(bucket, name) + do_layout = doc["parser_config"].get("layout_recognize", "DeepDOC") pages = PdfParser.total_page_number(doc["name"], file_bin) if pages is None: pages = 0 page_size = doc["parser_config"].get("task_page_size") or 12 if doc["parser_id"] == "paper": page_size = doc["parser_config"].get("task_page_size") or 22 - if doc["parser_id"] in ["one", "knowledge_graph"] or doc["parser_config"].get("toc_extraction", False): - page_size = doc["parser_config"].get("task_page_size") or 30 + if doc["parser_id"] in ["one", "knowledge_graph"] or do_layout != "DeepDOC" or doc["parser_config"].get("toc_extraction", False): + page_size = MAXIMUM_TASK_PAGE_NUMBER page_ranges = doc["parser_config"].get("pages") or [(1, MAXIMUM_PAGE_NUMBER)] for s, e in page_ranges: s -= 1