From 6dea61609d5da70223b6103fe464520c4347f799 Mon Sep 17 00:00:00 2001 From: Yuan Tang Date: Fri, 7 Feb 2025 22:35:22 -0500 Subject: [PATCH] test: Make text-based chat completion tests run faster Signed-off-by: Yuan Tang --- tests/client-sdk/inference/test_text_inference.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/client-sdk/inference/test_text_inference.py b/tests/client-sdk/inference/test_text_inference.py index 4b24f1d38..d75686666 100644 --- a/tests/client-sdk/inference/test_text_inference.py +++ b/tests/client-sdk/inference/test_text_inference.py @@ -156,8 +156,8 @@ def test_text_completion_structured_output(llama_stack_client, text_model_id, in @pytest.mark.parametrize( "question,expected", [ - ("What are the names of planets in our solar system?", "Earth"), - ("What are the names of the planets that have rings around them?", "Saturn"), + ("Which planet do humans live?", "Earth"), + ("Which planet has rings around them with a name starting with letter S?", "Saturn"), ], ) def test_text_chat_completion_non_streaming(llama_stack_client, text_model_id, question, expected):