Merge pull request #1571 from myhloli/dev

feat(llm_aided): add reasonability check and fine-tuning guidelines
opendatalab · Jan 17, 2025 · 48a4337 · 48a4337
2 parents b894b78 + d986e39
commit 48a4337
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/magic_pdf/post_proc/llm_aided.py b/magic_pdf/post_proc/llm_aided.py
@@ -115,6 +115,11 @@ def llm_aided_title(pdf_info_dict, title_aided_config):
     - 标题层级最多为4级，不要添加过多的层级
     - 优化后的标题只保留代表该标题的层级的整数，不要保留其他信息
     
+5. 合理性检查与微调：
+    - 在完成初步分级后，仔细检查分级结果的合理性
+    - 根据上下文关系和逻辑顺序，对不合理的分级进行微调
+    - 确保最终的分级结果符合文档的实际结构和逻辑
+    
 IMPORTANT: 
 请直接返回优化过的由标题层级组成的json，格式如下：
 {{"0":1,"1":2,"2":2,"3":3}}