Update Component Check (#536)

* update commit * update for test_all_components * update check * update fix ut * update * update
baidubce · Oct 15, 2024 · 0ff39b5 · 0ff39b5
1 parent 321a105
commit 0ff39b5
Show file tree

Hide file tree

Showing 11 changed files with 510 additions and 768 deletions.
diff --git a/.gitignore b/.gitignore
@@ -51,6 +51,7 @@ coverage.xml
 .pytest_cache/
 # Unit test / production
 asr_test.pcm
+components_error_info.txt
 
 # Translations
 *.mo

diff --git a/appbuilder/__init__.py b/appbuilder/__init__.py
@@ -114,7 +114,56 @@ def get_default_header():
 from .core.components.image_understand.component import ImageUnderstand
 from .core.components.mix_card_ocr.component import MixCardOCR
 
-from .tests.component_test import AppbuilderTestToolEval, AutomaticTestToolEval
+__COMPONENTS__ = [
+    "RagWithBaiduSearchPro",
+    "RAGWithBaiduSearch",
+    "Excel2Figure",
+    "MRC",
+    "OralQueryGeneration",
+    "QAPairMining",
+    "SimilarQuestion",
+    "StyleWriting",
+    "StyleRewrite",
+    "TagExtraction",
+    "Nl2pandasComponent",
+    "QueryRewrite",
+    "DialogSummary",
+    "HallucinationDetection",
+    "Playground",
+    "ASR",
+    "GeneralOCR",
+    "ObjectRecognition",
+    "Text2Image",
+    "LandmarkRecognition",
+    "TTS",
+    "ExtractTableFromDoc",
+    "DocParser",
+    "DocSplitter",
+    "BESRetriever",
+    "BESVectorStoreIndex",
+    "BaiduVDBVectorStoreIndex",
+    "BaiduVDBRetriever",
+    "TableParams",
+    "Reranker",
+    "PPTGenerationFromInstruction",
+    "PPTGenerationFromPaper",
+    "PPTGenerationFromFile",
+    "DishRecognition",
+    "Translation",
+    "AnimalRecognition",
+    "DocCropEnhance",
+    "QRcodeOCR",
+    "TableOCR",
+    "DocFormatConverter",
+    "Embedding",
+    "Matching",
+    "NL2Sql",
+    "SelectTable",
+    "PlantRecognition",
+    "HandwriteOCR",
+    "ImageUnderstand",
+    "MixCardOCR",
+] # NOQA
 
 from appbuilder.core.message import Message
 from appbuilder.core.agent import AgentRuntime
@@ -149,76 +198,20 @@ def get_default_header():
 
 __all__ = [
     'logger',
-
     'BadRequestException',
     'ForbiddenException',
     'NotFoundException',
     'PreconditionFailedException',
     'InternalServerErrorException',
     'HTTPConnectionException',
     'AppBuilderServerException',
-
-    'StyleWriting',
-    'MRC',
-    'Playground',
-    'OralQueryGeneration',
-    'QAPairMining',
-    'SimilarQuestion',
-    'IsComplexQuery',
-    'QueryDecomposition',
-    'TagExtraction',
-    'StyleRewrite',
-    'QueryRewrite',
-    'DialogSummary',
-    'ASR',
-    'GeneralOCR',
-    'ObjectRecognition',
-    'Text2Image',
-    'LandmarkRecognition',
-    'TTS',
-    "ExtractTableFromDoc",
-    "DocParser",
-    "ParserConfig",
-    "DocSplitter",
-    "BESRetriever",
-    "BESVectorStoreIndex",
-    "BaiduVDBVectorStoreIndex",
-    "BaiduVDBRetriever",
-    "TableParams",
-    "Reranker",
-    "HallucinationDetection",
-
-    'DishRecognition',
-    'Translation',
-    'Message',
-    'AnimalRecognition',
-    'DocCropEnhance',
-    'QRcodeOCR',
-    'TableOCR',
-
-    'Embedding',
-
-    'Matching',
-
-    "PlantRecognition",
-    "HandwriteOCR",
-    "ImageUnderstand",
-    "MixCardOCR",
-
-    'PPTGenerationFromInstruction',
-    'PPTGenerationFromPaper',
-    'PPTGenerationFromFile',
-
     'AppbuilderTestToolEval',
     'AutomaticTestToolEval',
-
     "get_model_list",
-
     "AppBuilderClient",
     "AgentBuilder",
     "get_app_list",
     "get_all_apps",
-
     "KnowledgeBase",
     "CustomProcessRule",
     "DocumentSource",
@@ -227,12 +220,10 @@ def get_default_header():
     "DocumentSeparator",
     "DocumentPattern",
     "DocumentProcessOption",
-
     "assistant",
     "StreamRunContext",
     "AssistantEventHandler",
     "AssistantStreamManager",
-
     "AppBuilderTracer",
     "AppbuilderInstrumentor"
-]
+] + __COMPONENTS__