ai-workspace-services/litellm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(factory.py): handle content not being set

Browse Source
This commit is contained in:
Krrish Dholakia 2024-07-16 21:51:30 -07:00
parent c2cb9502ae
commit bc9f5eb628
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
litellm/llms/prompt_templates/factory.py
View File

@ -1283,7 +1283,9 @@ def anthropic_messages_pt(
assistant_content: List[AnthropicMessagesAssistantMessageValues] = []
## MERGE CONSECUTIVE ASSISTANT CONTENT ##
while msg_i < len(messages) and messages[msg_i]["role"] == "assistant":
if isinstance(messages[msg_i]["content"], list):
if "content" in messages[msg_i] and isinstance(
messages[msg_i]["content"], list
):
for m in messages[msg_i]["content"]:
# handle text
if m.get("type", "") == "text":
@ -1291,7 +1293,9 @@ def anthropic_messages_pt(
type="text", text=m.get("text")
)
assistant_content.append(anthropic_message)
elif isinstance(messages[msg_i]["content"], str):
elif "content" in messages[msg_i] and isinstance(
messages[msg_i]["content"], str
):
assistant_content.append(
{"type": "text", "text": messages[msg_i]["content"]}
)