|
@@ -26,27 +26,34 @@
|
|
|
__version__ = "2.0.0"
|
|
__version__ = "2.0.0"
|
|
|
__author__ = "Your Name"
|
|
__author__ = "Your Name"
|
|
|
|
|
|
|
|
-try:
|
|
|
|
|
- from .core import DocumentClassifier
|
|
|
|
|
- from .toc.toc_extractor import TOCExtractor
|
|
|
|
|
- from .chunking.text_splitter import TextSplitter
|
|
|
|
|
- from .classification.hierarchy_classifier import HierarchyClassifier
|
|
|
|
|
- from .classification.llm_classifier import LLMClassifier
|
|
|
|
|
- from .output.result_saver import ResultSaver
|
|
|
|
|
-except ImportError:
|
|
|
|
|
- from core import DocumentClassifier
|
|
|
|
|
- from toc.toc_extractor import TOCExtractor
|
|
|
|
|
- from chunking.text_splitter import TextSplitter
|
|
|
|
|
- from classification.hierarchy_classifier import HierarchyClassifier
|
|
|
|
|
- from classification.llm_classifier import LLMClassifier
|
|
|
|
|
- from output.result_saver import ResultSaver
|
|
|
|
|
|
|
+from .core import DocumentClassifier
|
|
|
|
|
+from .toc.toc_extractor import TOCExtractor
|
|
|
|
|
+from .chunking.text_splitter import TextSplitter
|
|
|
|
|
+from .classification.hierarchy_classifier import HierarchyClassifier
|
|
|
|
|
+from .classification.rule_based_classifier import RuleBasedClassifier
|
|
|
|
|
+from .output.result_saver import ResultSaver
|
|
|
|
|
+
|
|
|
|
|
+
|
|
|
|
|
+class LLMClassifier:
|
|
|
|
|
+ """
|
|
|
|
|
+ 占位LLM分类器,避免未实现类的导入错误。
|
|
|
|
|
+ 当前仅提供接口占位,后续可替换为真实的LLM服务实现。
|
|
|
|
|
+ """
|
|
|
|
|
+
|
|
|
|
|
+ def __init__(self, model_url: str):
|
|
|
|
|
+ self.model_url = model_url
|
|
|
|
|
+
|
|
|
|
|
+ def classify(self, toc_items, target_level=None):
|
|
|
|
|
+ # 返回None以触发上层的回退逻辑
|
|
|
|
|
+ return None
|
|
|
|
|
|
|
|
__all__ = [
|
|
__all__ = [
|
|
|
'DocumentClassifier',
|
|
'DocumentClassifier',
|
|
|
'TOCExtractor',
|
|
'TOCExtractor',
|
|
|
'TextSplitter',
|
|
'TextSplitter',
|
|
|
'HierarchyClassifier',
|
|
'HierarchyClassifier',
|
|
|
- 'LLMClassifier',
|
|
|
|
|
- 'ResultSaver'
|
|
|
|
|
|
|
+ 'RuleBasedClassifier',
|
|
|
|
|
+ 'ResultSaver',
|
|
|
|
|
+ 'LLMClassifier'
|
|
|
]
|
|
]
|
|
|
|
|
|