fix(factory.py): handle assistant null content

2024-08-14 15:44:17 -07:00 · 2024-08-14 15:44:17 -07:00 · 6333b04be3
commit 6333b04be3
parent 75ed22d391
1 changed files with 3 additions and 1 deletions
--- a/litellm/llms/prompt_templates/factory.py
+++ b/litellm/llms/prompt_templates/factory.py
@ -1705,7 +1705,9 @@ def cohere_messages_pt_v2(
                for m in messages[msg_i]["content"]:
                    if m.get("type", "") == "text":
                        assistant_content += m["text"]
-            else:
+            elif messages[msg_i].get("content") is not None and isinstance(
+                messages[msg_i]["content"], str
+            ):
                assistant_content += messages[msg_i]["content"]
            if messages[msg_i].get(
                "tool_calls", []