diff --git a/tests/ci_use/XPU_45T/run_45T.py b/tests/ci_use/XPU_45T/run_45T.py index 3341cfc35..5841a1482 100644 --- a/tests/ci_use/XPU_45T/run_45T.py +++ b/tests/ci_use/XPU_45T/run_45T.py @@ -36,7 +36,7 @@ def test_45t(): ) print(response.choices[0].message.content) # print(base_response) - assert "人工智能" in response.choices[0].message.content + assert any(keyword in response.choices[0].message.content for keyword in ["人工智能", "文心一言"]) if __name__ == "__main__": diff --git a/tests/ci_use/XPU_45T/run_45vl.py b/tests/ci_use/XPU_45T/run_45vl.py index 2c0b9acbd..084ab0e51 100644 --- a/tests/ci_use/XPU_45T/run_45vl.py +++ b/tests/ci_use/XPU_45T/run_45vl.py @@ -21,7 +21,6 @@ def test_45vl(): gpu_id = int(os.getenv("GPU_ID", "0")) service_http_port = 8188 + gpu_id * 100 # 服务配置的 client = openai.Client(base_url=f"http://{ip}:{service_http_port}/v1", api_key="EMPTY_API_KEY") - base_response = "北魏时期" # 非流式对话 response = client.chat.completions.create( model="default", @@ -46,7 +45,7 @@ def test_45vl(): ) print(response.choices[0].message.content) # print(base_response) - assert base_response in response.choices[0].message.content + assert any(keyword in response.choices[0].message.content for keyword in ["北魏", "北齐", "释迦牟尼"]) if __name__ == "__main__": diff --git a/tests/ci_use/XPU_45T/run_w4a8.py b/tests/ci_use/XPU_45T/run_w4a8.py index 75857d936..aa9bab160 100644 --- a/tests/ci_use/XPU_45T/run_w4a8.py +++ b/tests/ci_use/XPU_45T/run_w4a8.py @@ -36,7 +36,7 @@ def test_w4a8(): ) print(response.choices[0].message.content) # print(base_response) - assert "人工智能" in response.choices[0].message.content + assert any(keyword in response.choices[0].message.content for keyword in ["人工智能", "文心一言"]) if __name__ == "__main__":