diff --git a/tests/ci_use/XPU_45T/run_45T.py b/tests/ci_use/XPU_45T/run_45T.py index 4a13becf8..d7e0cca03 100644 --- a/tests/ci_use/XPU_45T/run_45T.py +++ b/tests/ci_use/XPU_45T/run_45T.py @@ -36,7 +36,9 @@ def test_45t(): ) print(response.choices[0].message.content) # print(base_response) - assert any(keyword in response.choices[0].message.content for keyword in ["人工智能", "文心一言"]) + assert any( + keyword in response.choices[0].message.content for keyword in ["人工智能", "文心一言", "百度", "智能助手"] + ) if __name__ == "__main__": diff --git a/tests/ci_use/XPU_45T/run_ep_online.py b/tests/ci_use/XPU_45T/run_ep_online.py index d6f74b862..c0d71e4c8 100644 --- a/tests/ci_use/XPU_45T/run_ep_online.py +++ b/tests/ci_use/XPU_45T/run_ep_online.py @@ -35,7 +35,9 @@ def test_ep(): print(response.choices[0].message.content) # print(base_response) - assert any(keyword in response.choices[0].message.content for keyword in ["人工智能", "文心一言"]) + 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 b5eea23e1..c1e605215 100644 --- a/tests/ci_use/XPU_45T/run_w4a8.py +++ b/tests/ci_use/XPU_45T/run_w4a8.py @@ -36,7 +36,10 @@ def test_w4a8(): ) print(response.choices[0].message.content) # print(base_response) - assert any(keyword in response.choices[0].message.content for keyword in ["人工智能", "文心一言", "小度"]) + assert any( + keyword in response.choices[0].message.content + for keyword in ["人工智能", "文心一言", "小度", "百度", "智能助手"] + ) if __name__ == "__main__":