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

fix(utils.py): map cohere finish reasons

Browse source
This commit is contained in:
Krrish Dholakia 2023-12-05 12:38:18 -08:00
parent 55b34f969c
commit 2a02fcbffb
1 changed files with 9 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

9
litellm/utils.py
View file

@ -123,6 +123,15 @@ def map_finish_reason(finish_reason: str): # openai supports 5 stop sequences -
# anthropic mapping
if finish_reason == "stop_sequence":
return "stop"
# cohere mapping - https://docs.cohere.com/reference/generate
elif finish_reason == "COMPLETE":
return "stop"
elif finish_reason == "MAX_TOKENS":
return "length"
elif finish_reason == "ERROR_TOXIC":
return "content_filter"
elif finish_reason == "ERROR": # openai currently doesn't support an 'error' finish reason
return "stop"
return finish_reason
class FunctionCall(OpenAIObject):