wehub-resource-sync
69 lines
2.0 KiB
Python
69 lines
2.0 KiB
Python
# Copyright 2023 The Qwen team, Alibaba Group. 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 json
|
|
|
|
import pytest
|
|
|
|
from qwen_agent.tools import AmapWeather, CodeInterpreter, ImageGen, Retrieval, Storage
|
|
|
|
|
|
# [NOTE] 不带“市”会出错
|
|
@pytest.mark.parametrize('params', [json.dumps({'location': '北京市'}), {'location': '杭州市'}])
|
|
def test_amap_weather(params):
|
|
tool = AmapWeather()
|
|
tool.call(params)
|
|
|
|
|
|
def test_code_interpreter():
|
|
tool = CodeInterpreter()
|
|
tool.call("print('hello qwen')")
|
|
|
|
|
|
def test_image_gen():
|
|
tool = ImageGen()
|
|
tool.call({'prompt': 'a dog'})
|
|
|
|
|
|
def test_retrieval():
|
|
tool = Retrieval()
|
|
tool.call({
|
|
'query': 'Who are the authors of this paper?',
|
|
'files': ['https://qianwen-res.oss-cn-beijing.aliyuncs.com/QWEN_TECHNICAL_REPORT.pdf']
|
|
})
|
|
|
|
|
|
@pytest.mark.parametrize('operate', ['put'])
|
|
def test_storage_put(operate):
|
|
tool = Storage()
|
|
tool.call({'operate': operate, 'key': '345/456/11', 'value': 'hello'})
|
|
|
|
tool.call({'operate': operate, 'key': '/345/456/12', 'value': 'hello'})
|
|
|
|
|
|
@pytest.mark.parametrize('operate', ['scan'])
|
|
def test_storage_scan(operate):
|
|
tool = Storage()
|
|
tool.call({'operate': operate, 'key': '345/456/'})
|
|
|
|
tool.call({'operate': operate, 'key': '/345/456'})
|
|
|
|
|
|
@pytest.mark.parametrize('operate', ['get', 'delete'])
|
|
def test_storage_get_delete(operate):
|
|
tool = Storage()
|
|
tool.call({'operate': operate, 'key': '345/456/11'})
|
|
|
|
tool.call({'operate': operate, 'key': '/345/456/12'})
|