(fix) completion - always map finish_reason

This commit is contained in:
ishaan-jaff 2023-11-20 17:24:03 -08:00
parent 53743f2124
commit 980a3502de

View file

@ -185,10 +185,7 @@ class Delta(OpenAIObject):
class Choices(OpenAIObject):
def __init__(self, finish_reason=None, index=0, message=None, **params):
super(Choices, self).__init__(**params)
if finish_reason:
self.finish_reason = map_finish_reason(finish_reason)
else:
self.finish_reason = "stop"
self.finish_reason = map_finish_reason(finish_reason) # set finish_reason for all responses
self.index = index
if message is None:
self.message = Message(content=None)