From 09ab5c8eabc5893942df036b762c478c3ebea230 Mon Sep 17 00:00:00 2001 From: Aidan Do Date: Sat, 14 Dec 2024 20:44:30 +1100 Subject: [PATCH] Update llama_stack/providers/remote/inference/groq/groq_utils.py --- llama_stack/providers/remote/inference/groq/groq_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/llama_stack/providers/remote/inference/groq/groq_utils.py b/llama_stack/providers/remote/inference/groq/groq_utils.py index ce448ade6..2aabb80ba 100644 --- a/llama_stack/providers/remote/inference/groq/groq_utils.py +++ b/llama_stack/providers/remote/inference/groq/groq_utils.py @@ -115,7 +115,7 @@ def _map_finish_reason_to_stop_reason( if finish_reason == "stop": return StopReason.end_of_turn elif finish_reason == "length": - return StopReason.end_of_message + return StopReason.out_of_tokens elif finish_reason == "tool_calls": raise NotImplementedError("tool_calls is not supported yet") else: