use from_string method

defenseunicorns · Sep 16, 2024 · 41870db · 41870db
1 parent 1ef883b
commit 41870db
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/leapfrogai_sdk/llm.py b/src/leapfrogai_sdk/llm.py
@@ -87,7 +87,7 @@ def create_chat_completion_response(
             choices=[choice], usage=usage
         )
 
-        response.choices[0].finish_reason = str(finish_reason)
+        response.choices[0].finish_reason = finish_reason.from_string(finish_reason)
 
         return response
 
@@ -109,7 +109,7 @@ def create_completion_response(
 
         response: CompletionResponse = CompletionResponse(choices=[choice], usage=usage)
 
-        response.choices[0].finish_reason = str(finish_reason)
+        response.choices[0].finish_reason = finish_reason.from_string(finish_reason)
 
         return response