use value of finish_reason again

defenseunicorns · Sep 16, 2024 · 685287f · 685287f
1 parent 7c39a31
commit 685287f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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