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

fix(factory.py): handle missing 'content' in cohere assistant messages

Browse source 
Update cohere_messages_pt_v2 function to check for 'content' existence
This commit is contained in:
miraclebakelaser 2024-08-27 19:38:37 +09:00
parent 3ec8a0d25d
commit 97f714d2b0
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/llms/prompt_templates/factory.py
View file

@ -1781,7 +1781,7 @@ def cohere_messages_pt_v2(
assistant_tool_calls: List[ToolCallObject] = [] assistant_tool_calls: List[ToolCallObject] = []
## MERGE CONSECUTIVE ASSISTANT CONTENT ## ## MERGE CONSECUTIVE ASSISTANT CONTENT ##
while msg_i < len(messages) and messages[msg_i]["role"] == "assistant": 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"]: for m in messages[msg_i]["content"]:
if m.get("type", "") == "text": if m.get("type", "") == "text":
assistant_content += m["text"] assistant_content += m["text"]