test_appbuilder_client.py•2.72 kB
import unittest
import appbuilder
import requests
import tempfile
import os
from tests.pytest_utils import Utils
@unittest.skipUnless(os.getenv("TEST_CASE", "UNKNOWN") == "CPU_SERIAL","")
class TestAgentRuntime(unittest.TestCase):
def setUp(self):
"""
设置环境变量。
Args:
无参数，默认值为空。
Returns:
无返回值，方法中执行了环境变量的赋值操作。
"""
self.app_id = "aa8af334-df27-4855-b3d1-0d249c61fc08"
def test_agent_builder_client(self):
agent_builder = appbuilder.AppBuilderClient(self.app_id)
def test_agent_builder_run(self):
# 如果app_id为空，则跳过单测执行, 避免单测因配置无效而失败
"""
如果app_id为空，则跳过单测执行, 避免单测因配置无效而失败
Args:
self (unittest.TestCase): unittest的TestCase对象
Raises:
None: 如果app_id不为空，则不会引发任何异常
unittest.SkipTest (optional): 如果app_id为空，则跳过单测执行
"""
import tests.pytest_utils as pu
if len(self.app_id) == 0:
self.skipTest("self.app_id is empty")
builder = appbuilder.AppBuilderClient(self.app_id)
conversation_id = builder.create_conversation()
msg = builder.run(conversation_id, "你可以做什么？")
print(msg)
file_path = Utils.get_data_file("qa_doc_parser_extract_table_from_doc.png")
respid = builder.upload_local_file(conversation_id, file_path)
print(respid)
def test_upload_local_file_raise(self):
builder = appbuilder.AppBuilderClient(self.app_id)
with self.assertRaises(ValueError):
builder.upload_local_file(conversation_id='', local_file_path='')
with self.assertRaises(ValueError):
builder.run(conversation_id='', query='')
conversation_id = builder.create_conversation()
with self.assertRaises(FileNotFoundError):
builder.upload_local_file(conversation_id=conversation_id, local_file_path='not_exist')
def test_upload_file_url(self):
file_url = "https://bj.bcebos.com/v1/appbuilder/animal_recognize_test.png?authorization=bce-auth-v1%2FALTAKGa8m4qCUasgoljdEDAzLm%2F2024-01-24T12%3A19%3A16Z%2F-1%2Fhost%2F411bad53034fa8f9c6edbe5c4909d76ecf6fad6862cf937c03f8c5260d51c6ae"
builder = appbuilder.AppBuilderClient(self.app_id)
conversation_id = builder.create_conversation()
builder.upload_file(conversation_id=conversation_id, file_url=file_url)
if __name__ == '__main__':
unittest.main()