diff --git a/python/core/components/v2/table_ocr/component.py b/python/core/components/v2/table_ocr/component.py index 915cdf92..020b698c 100644 --- a/python/core/components/v2/table_ocr/component.py +++ b/python/core/components/v2/table_ocr/component.py @@ -17,6 +17,8 @@ import base64 import json +from typing import Optional + from appbuilder.core import utils from appbuilder.core.component import Component from appbuilder.core.components.table_ocr.model import * @@ -189,7 +191,7 @@ def get_table_markdown(self, tables_result): @components_run_stream_trace def tool_eval(self, - file_names: List[str], + file_names: Optional[List[str]] = [], **kwargs): """ 处理并评估传入的文件列表,并返回表格数据的Markdown格式表示。 @@ -206,8 +208,10 @@ def tool_eval(self, """ result = {} - traceid = kwargs.get("_sys_traceid") + traceid = kwargs.get("_sys_traceid", "") file_urls = kwargs.get("_sys_file_urls", {}) + if not file_names: + file_names = kwargs.get("_sys_file_names", []) for file_name in file_names: if utils.is_url(file_name): file_url = file_name diff --git a/python/tests/test_dialog_summary.py b/python/tests/test_dialog_summary.py index 7c7aac45..90235721 100644 --- a/python/tests/test_dialog_summary.py +++ b/python/tests/test_dialog_summary.py @@ -48,9 +48,9 @@ def test_run_with_model_names(self): """测试不同的 stream 和 temperature 参数值""" chats = appbuilder.get_model_list(api_type_filter=["chat"]) - self.assertTrue("EB-turbo-AppBuilder专用版" in chats) + self.assertTrue("ERNIE-3.5-8K" in chats) - appbuilder.DialogSummary(model="EB-turbo-AppBuilder专用版") + appbuilder.DialogSummary(model="ERNIE-3.5-8K") with self.assertRaises(Exception): appbuilder.DialogSummary(model="") diff --git a/python/tests/test_v2_table_ocr.py b/python/tests/test_v2_table_ocr.py index b4b70c8e..94a1cb62 100644 --- a/python/tests/test_v2_table_ocr.py +++ b/python/tests/test_v2_table_ocr.py @@ -30,11 +30,20 @@ def test_run(self): print(out) - def test_tool_eval(self): + def test_tool_eval_01(self): result = self.com.tool_eval([self.image_url]) for res in result: assert isinstance(res, ComponentOutput) print(res.role, res.content) + def test_tool_eval_02(self): + _sys_file_urls = { + "test-name": self.image_url + } + result = self.com.tool_eval(_sys_file_names = ["test-name"], _sys_file_urls = _sys_file_urls) + for res in result: + assert isinstance(res, ComponentOutput) + print(res.role, res.content) + if __name__ == '__main__': unittest.main() \ No newline at end of file