fix(llm_aided): update prompt
@@ -145,7 +145,11 @@ def doc_analyze(
table_enable=None,
) -> InferenceResult:
- end_page_id = end_page_id if end_page_id is not None else len(dataset) - 1
+ end_page_id = (
+ end_page_id
+ if end_page_id is not None and end_page_id >= 0
+ else len(dataset) - 1
+ )
model_manager = ModelSingleton()
custom_model = model_manager.get_model(
@@ -120,7 +120,7 @@ def llm_aided_title(pdf_info_dict, title_aided_config):
- 在完成初步分级后,仔细检查分级结果的合理性
- 根据上下文关系和逻辑顺序,对不合理的分级进行微调
- 确保最终的分级结果符合文档的实际结构和逻辑
- - 字典中包含OCR错误识别的标题,你可以通过将其层级标记为 0 来排除它们
+ - 字典中可能包含被误当成标题的正文,你可以通过将其层级标记为 0 来排除它们
IMPORTANT:
请直接返回优化过的由标题层级组成的字典,格式为{{标题id:标题层级}},如下: