From 6333b04be3a622f237e0ffe187ea34bc835bd69a Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Wed, 14 Aug 2024 15:44:17 -0700 Subject: [PATCH] fix(factory.py): handle assistant null content --- litellm/llms/prompt_templates/factory.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/litellm/llms/prompt_templates/factory.py b/litellm/llms/prompt_templates/factory.py index f39273c1a2..4e552b3b07 100644 --- 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", []