From 97f714d2b02d1615a75424fdede20199b7ebe4a1 Mon Sep 17 00:00:00 2001 From: miraclebakelaser <65143272+miraclebakelaser@users.noreply.github.com> Date: Tue, 27 Aug 2024 19:38:37 +0900 Subject: [PATCH] fix(factory.py): handle missing 'content' in cohere assistant messages Update cohere_messages_pt_v2 function to check for 'content' existence --- litellm/llms/prompt_templates/factory.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/litellm/llms/prompt_templates/factory.py b/litellm/llms/prompt_templates/factory.py index 0786f1ec3..8dbab29ad 100644 --- a/litellm/llms/prompt_templates/factory.py +++ b/litellm/llms/prompt_templates/factory.py @@ -1781,7 +1781,7 @@ def cohere_messages_pt_v2( assistant_tool_calls: List[ToolCallObject] = [] ## MERGE CONSECUTIVE ASSISTANT CONTENT ## while msg_i < len(messages) and messages[msg_i]["role"] == "assistant": - if isinstance(messages[msg_i]["content"], list): + if messages[msg_i].get("content", None) is not None and isinstance(messages[msg_i]["content"], list): for m in messages[msg_i]["content"]: if m.get("type", "") == "text": assistant_content += m["text"]