21 lines
586 B
Python
21 lines
586 B
Python
|
import sys
|
||
|
|
from pathlib import Path
|
||
|
|
root_path = Path(__file__).parent.parent.parent
|
||
|
|
sys.path.append(str(root_path))
|
||
|
|
|
||
|
|
from server.model_workers.qianfan import request_qianfan_api, MODEL_VERSIONS
|
||
|
|
from pprint import pprint
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.parametrize("version", MODEL_VERSIONS.keys())
|
||
|
|
def test_qianfan(version):
|
||
|
|
messages = [{"role": "user", "content": "你好"}]
|
||
|
|
print("\n" + version + "\n")
|
||
|
|
i = 1
|
||
|
|
for x in request_qianfan_api(messages, version=version):
|
||
|
|
pprint(x)
|
||
|
|
assert isinstance(x, dict)
|
||
|
|
assert "error_code" not in x
|
||
|
|
i += 1
|