phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

(fix) completion - always map finish_reason

Browse source
This commit is contained in:
ishaan-jaff 2023-11-20 17:24:03 -08:00
parent 53743f2124
commit 980a3502de
1 changed files with 1 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

5
litellm/utils.py
View file

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