test_oral_query_generation.py•4.93 kB
# Copyright (c) 2023 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
TEST_TEXT = ('文档标题：在OPPO Reno5上使用视频超级防抖\n'
'文档摘要：OPPO Reno5上的视频超级防抖，视频超级防抖3.0，多代视频防抖算法积累，这一代依旧超级防抖超级稳。 开启视频超级'
'防抖 开启路径：打开「相机 > 视频 > 点击屏幕上方的“超级防抖”标识」 后置视频同时支持超级防抖和超级防抖Pro功能，开启超级'
'防抖后手机屏幕将出现超级防抖Pro开关，点击即可开启或关闭。 除此之外，前置视频同样加持防抖算法，边走边拍也能稳定聚焦脸部'
'，实时视频分享您的生活。')
class TestOralQueryGenerationComponent(unittest.TestCase):
def setUp(self):
"""
设置环境变量。
Args:
无参数，默认值为空。
Returns:
无返回值，方法中执行了环境变量的赋值操作。
"""
self.model_name = "ERNIE-3.5-8K"
secret_key = os.getenv('SECRET_KEY', None)
self.query_generation = appbuilder.OralQueryGeneration(model=self.model_name, secret_key=secret_key)
def test_run_with_default_params(self):
"""测试 run 方法使用默认参数
"""
text = TEST_TEXT
msg = appbuilder.Message(text)
answer = self.query_generation(msg)
# print(answer)
self.assertIsNotNone(answer)
print(f'\n[result]\n{answer.content}\n')
def test_run_with_question_output_and_json_output(self):
"""测试 run 方法，输出query类型为问题，输出格式为json
"""
text = TEST_TEXT
msg = appbuilder.Message(text)
answer = self.query_generation(msg, query_type='问题', output_format='json')
# print(answer)
self.assertIsNotNone(answer)
print(f'\n[result]\n{answer.content}\n')
def test_run_with_phrase_output_and_str_output(self):
"""测试 run 方法，输出query类型为短语，输出格式为str
"""
text = TEST_TEXT
msg = appbuilder.Message(text)
answer = self.query_generation(msg, query_type='短语', output_format='str')
# print(answer)
self.assertIsNotNone(answer)
print(f'\n[result]\n{answer.content}\n')
def test_run_with_stream_and_temperature(self):
"""测试不同的 stream 和 temperature 参数值
"""
text = TEST_TEXT
msg = appbuilder.Message(text)
answer = self.query_generation(msg, stream=False, temperature=0.5)
# print(answer)
self.assertIsNotNone(answer)
print(f'\n[result]\n{answer.content}\n')
def test_tool_eval_with_default_params(self):
"""测试 tool_eval 方法使用默认参数
"""
text = TEST_TEXT
answer = self.query_generation.tool_eval(name='', stream=False, text=text)
# print(answer)
self.assertIsNotNone(answer)
print(f'\n[result]\n{answer}\n')
def test_tool_eval_with_model_configs(self):
"""测试 tool_eval 方法使用不同temperature和top_p参数值。
"""
text = TEST_TEXT
model_configs = {'temperature': 0.5, 'top_p': 0.5}
answer = self.query_generation.tool_eval(name='', stream=True, text=text, model_configs=model_configs)
# print(answer)
print(f'\n[result]\n')
for ans in answer:
print(ans)
def test_tool_eval_with_default_params(self):
"""测试 tool_eval 方法使用默认参数
"""
text = TEST_TEXT
answer = self.query_generation.tool_eval(name='', stream=False, text=text)
# print(answer)
self.assertIsNotNone(answer)
print(f'\n[result]\n{answer}\n')
def test_tool_eval_with_model_configs(self):
"""测试 tool_eval 方法使用不同temperature和top_p参数值。
"""
text = TEST_TEXT
model_configs = {'temperature': 0.5, 'top_p': 0.5}
answer = self.query_generation.tool_eval(name='', stream=True, text=text, model_configs=model_configs)
# print(answer)
print(f'\n[result]\n')
for ans in answer:
print(ans)
if __name__ == '__main__':
unittest.main()