Commit 941f36f5 authored by myhloli's avatar myhloli
Browse files

fix: ensure title_aided_config is defined before accessing its properties

parent 92d43c7b
...@@ -12,8 +12,9 @@ from mineru.version import __version__ ...@@ -12,8 +12,9 @@ from mineru.version import __version__
heading_level_import_success = False heading_level_import_success = False
llm_aided_config = get_llm_aided_config() llm_aided_config = get_llm_aided_config()
title_aided_config = llm_aided_config.get('title_aided', {}) if llm_aided_config:
if llm_aided_config and title_aided_config.get('enable', False): title_aided_config = llm_aided_config.get('title_aided', {})
if title_aided_config.get('enable', False):
try: try:
from mineru.utils.llm_aided import llm_aided_title from mineru.utils.llm_aided import llm_aided_title
from mineru.backend.pipeline.model_init import AtomModelSingleton from mineru.backend.pipeline.model_init import AtomModelSingleton
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment