diff --git a/llama_stack/providers/tests/inference/groq/test_groq_utils.py b/llama_stack/providers/tests/inference/groq/test_groq_utils.py index 4bc633751..53b5c29cb 100644 --- a/llama_stack/providers/tests/inference/groq/test_groq_utils.py +++ b/llama_stack/providers/tests/inference/groq/test_groq_utils.py @@ -186,7 +186,7 @@ class TestConvertNonStreamChatCompletionResponse: converted = convert_chat_completion_response(response) - assert converted.completion_message.stop_reason == StopReason.end_of_message + assert converted.completion_message.stop_reason == StopReason.out_of_tokens def _dummy_chat_completion_response(self): return ChatCompletion(