baidu-ai-search

# Copyright (c) 2024 Baidu, Inc. All Rights Reserved. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. # You may obtain a copy of the License at # # http://www.apache.org/licenses/LICENSE-2.0 # # Unless required by applicable law or agreed to in writing, software # distributed under the License is distributed on an "AS IS" BASIS, # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. import os import unittest import appbuilder from appbuilder.core.components.v2 import TableOCR from appbuilder.core.component import ComponentOutput @unittest.skipUnless(os.getenv("TEST_CASE", "UNKNOWN") == "CPU_PARALLEL", "") class TestTableOCR(unittest.TestCase): def setUp(self): self.com = TableOCR() self.image_url = "https://bj.bcebos.com/v1/appbuilder/table_ocr_test.png?"\ "authorization=bce-auth-v1%2FALTAKGa8m4qCUasgoljdEDAzLm%2F2024-01-24T12%3A37%3A09Z%2F-1%2Fhost%2Fab528a5a9120d328dc6d18c6"\ "064079145ff4698856f477b820147768fc2187d3" def test_run(self): out = self.com.run(appbuilder.Message(content={"url": self.image_url})) print(out) def test_tool_eval_01(self): result = self.com.tool_eval(file_urls = [self.image_url]) def test_tool_eval_01(self): result = self.com.tool_eval(file_urls = [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.png": self.image_url } result = self.com.tool_eval(file_names = ["test-name.png"], _sys_file_urls = _sys_file_urls) for res in result: assert isinstance(res, ComponentOutput) print(res.role, res.content) if __name__ == '__main__': unittest.main()