opensource/MinerU
3
0
Fork 0
mirror of https://github.com/opendatalab/MinerU.git synced 2026-09-21 12:42:22 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor(magic_pdf): remove unnecessary logging statements

Browse Source 
- Comment out logging statements for title list, title completion, and length comparison
- Improve code readability and reduce clutter by removing unused debug information
This commit is contained in:
myhloli
2024-12-25 18:06:09 +08:00
parent 0a468eca6e
commit 192047a1e8
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
magic_pdf/post_proc/llm_aided.py
+3 -3
View File
@@ -84,7 +84,7 @@ def llm_aided_title(pdf_info_dict, title_aided_config):
title_text = merge_para_with_text(block)
title_dict[f"{i}"] = title_text
i += 1
logger.info(f"Title list: {title_dict}")
# logger.info(f"Title list: {title_dict}")
title_optimize_prompt = f"""输入的内容是一篇文档中所有标题组成的字典，请根据以下指南优化标题的结果，使结果符合正常文档的层次结构：
@@ -118,9 +118,9 @@ Corrected title list:
json_completion = json.loads(completion.choices[0].message.content)
logger.info(f"Title completion: {json_completion}")
# logger.info(f"Title completion: {json_completion}")
logger.info(f"len(json_completion): {len(json_completion)}, len(title_dict): {len(title_dict)}")
# logger.info(f"len(json_completion): {len(json_completion)}, len(title_dict): {len(title_dict)}")
if len(json_completion) == len(title_dict):
try:
for i, origin_title_block in enumerate(origin_title_list):