|
|
@@ -31,12 +31,14 @@ try:
|
|
|
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
|
|
|
|
|
|
__all__ = [
|
|
|
@@ -44,6 +46,7 @@ __all__ = [
|
|
|
'TOCExtractor',
|
|
|
'TextSplitter',
|
|
|
'HierarchyClassifier',
|
|
|
+ 'LLMClassifier',
|
|
|
'ResultSaver'
|
|
|
]
|
|
|
|