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

Merge pull request #5384 from miraclebakelaser/fix/cohere-tool-calls-content-field

Browse source 
fix(factory.py): handle missing 'content' in cohere assistant messages
This commit is contained in:
Krish Dholakia 2024-08-27 07:13:28 -07:00 • committed by GitHub
commit ec2357f1d2
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
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"]