Преглед на файлове

dev:修正了模块间的引用问题;

ChenJiSheng преди 2 месеца
родител
ревизия
3e90cbca86
променени са 1 файла, в които са добавени 3 реда и са изтрити 0 реда
  1. 3 0
      core/construction_review/component/doc_worker/__init__.py

+ 3 - 0
core/construction_review/component/doc_worker/__init__.py

@@ -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'
 ]