From aaff1a261692d332b60c8a454034a178370b3f15 Mon Sep 17 00:00:00 2001
From: myhloli <moe@myhloli.com>
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)