From 685287fcf4cabbb2c46340ee00943347120e9ed1 Mon Sep 17 00:00:00 2001 From: Justin Law Date: Mon, 16 Sep 2024 19:56:50 -0400 Subject: [PATCH] use value of finish_reason again --- src/leapfrogai_sdk/llm.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/leapfrogai_sdk/llm.py b/src/leapfrogai_sdk/llm.py index 69959fb694..1d7a69261f 100644 --- a/src/leapfrogai_sdk/llm.py +++ b/src/leapfrogai_sdk/llm.py @@ -73,7 +73,7 @@ def create_chat_completion_response( choices=[choice], usage=usage ) - response.choices[0].finish_reason = finish_reason + response.choices[0].finish_reason = finish_reason.value return response @@ -95,7 +95,7 @@ def create_completion_response( response: CompletionResponse = CompletionResponse(choices=[choice], usage=usage) - response.choices[0].finish_reason = finish_reason + response.choices[0].finish_reason = finish_reason.value return response