From 78912e663b207e8e2793986b2d97909ab87b38cc Mon Sep 17 00:00:00 2001 From: Aidan Do Date: Sat, 14 Dec 2024 20:48:38 +1100 Subject: [PATCH] Update llama_stack/providers/tests/inference/groq/test_groq_utils.py --- llama_stack/providers/tests/inference/groq/test_groq_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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(