20 lines
595 B
Python
20 lines
595 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", list(MODEL_VERSIONS.keys())[:2])
|
|
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 |