فهرست منبع

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 .toc.toc_extractor import TOCExtractor
     from .chunking.text_splitter import TextSplitter
     from .chunking.text_splitter import TextSplitter
     from .classification.hierarchy_classifier import HierarchyClassifier
     from .classification.hierarchy_classifier import HierarchyClassifier
+    from .classification.llm_classifier import LLMClassifier
     from .output.result_saver import ResultSaver
     from .output.result_saver import ResultSaver
 except ImportError:
 except ImportError:
     from core import DocumentClassifier
     from core import DocumentClassifier
     from toc.toc_extractor import TOCExtractor
     from toc.toc_extractor import TOCExtractor
     from chunking.text_splitter import TextSplitter
     from chunking.text_splitter import TextSplitter
     from classification.hierarchy_classifier import HierarchyClassifier
     from classification.hierarchy_classifier import HierarchyClassifier
+    from classification.llm_classifier import LLMClassifier
     from output.result_saver import ResultSaver
     from output.result_saver import ResultSaver
 
 
 __all__ = [
 __all__ = [
@@ -44,6 +46,7 @@ __all__ = [
     'TOCExtractor',
     'TOCExtractor',
     'TextSplitter',
     'TextSplitter',
     'HierarchyClassifier',
     'HierarchyClassifier',
+    'LLMClassifier',
     'ResultSaver'
     'ResultSaver'
 ]
 ]