From aaff1a261692d332b60c8a454034a178370b3f15 Mon Sep 17 00:00:00 2001 From: myhloli Date: Fri, 10 Jan 2025 18:40:16 +0800 Subject: [PATCH] fix(llm_aided): add enable flag check for LLM aided optimizations - Add enable flag check for formula, text, and title optimizations --- magic_pdf/pdf_parse_union_core_v2.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/magic_pdf/pdf_parse_union_core_v2.py b/magic_pdf/pdf_parse_union_core_v2.py index 6dcf6e59..5408e6f6 100644 --- a/magic_pdf/pdf_parse_union_core_v2.py +++ b/magic_pdf/pdf_parse_union_core_v2.py @@ -860,15 +860,18 @@ def pdf_parse_union( """公式优化""" formula_aided_config = llm_aided_config.get('formula_aided', None) if formula_aided_config is not None: - llm_aided_formula(pdf_info_dict, formula_aided_config) + if formula_aided_config.get('enable', False): + llm_aided_formula(pdf_info_dict, formula_aided_config) """文本优化""" text_aided_config = llm_aided_config.get('text_aided', None) if text_aided_config is not None: - llm_aided_text(pdf_info_dict, text_aided_config) + if text_aided_config.get('enable', False): + llm_aided_text(pdf_info_dict, text_aided_config) """标题优化""" title_aided_config = llm_aided_config.get('title_aided', None) if title_aided_config is not None: - llm_aided_title(pdf_info_dict, title_aided_config) + if title_aided_config.get('enable', False): + llm_aided_title(pdf_info_dict, title_aided_config) """dict转list""" pdf_info_list = dict_to_list(pdf_info_dict)