diff --git a/api/db/services/task_service.py b/api/db/services/task_service.py index 1fdfed350..401489a69 100644 --- a/api/db/services/task_service.py +++ b/api/db/services/task_service.py @@ -338,9 +338,9 @@ def queue_tasks(doc: dict, bucket: str, name: str, priority: int): file_bin = STORAGE_IMPL.get(bucket, name) do_layout = doc["parser_config"].get("layout_recognize", "DeepDOC") pages = PdfParser.total_page_number(doc["name"], file_bin) - page_size = doc["parser_config"].get("task_page_size", 12) + 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", 22) + page_size = doc["parser_config"].get("task_page_size") or 22 if doc["parser_id"] in ["one", "knowledge_graph"] or do_layout != "DeepDOC": page_size = 10 ** 9 page_ranges = doc["parser_config"].get("pages") or [(1, 10 ** 5)]