Update llama_stack/providers/tests/inference/groq/test_groq_utils.py

meta-llama · Dec 14, 2024 · 78912e6 · 78912e6
1 parent 09ab5c8
commit 78912e6
Showing 1 changed file with 1 addition and 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
@@ -186,7 +186,7 @@ def test_maps_length_to_end_of_message(self):
 
         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(