From 2a02fcbffbd311f5eda0789844324f87d3b839b6 Mon Sep 17 00:00:00 2001
From: Krrish Dholakia <krrishdholakia@gmail.com>
Date: Tue, 5 Dec 2023 12:38:18 -0800
Subject: [PATCH] fix(utils.py): map cohere finish reasons

---
 litellm/utils.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/litellm/utils.py b/litellm/utils.py
index 6518d88529..e8e2b57b3f 100644
--- a/litellm/utils.py
+++ b/litellm/utils.py
@@ -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):