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

fix(anthropic.py): fix translation

Browse source
This commit is contained in:
Krrish Dholakia 2024-08-08 17:32:55 -07:00
parent 1f43368110
commit 9839a7858c
2 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

6
litellm/llms/anthropic.py
View file

@ -223,6 +223,9 @@ class AnthropicConfig:
for m in messages: for m in messages:
user_message: Optional[ChatCompletionUserMessage] = None user_message: Optional[ChatCompletionUserMessage] = None
tool_message_list: List[ChatCompletionToolMessage] = [] tool_message_list: List[ChatCompletionToolMessage] = []
new_user_content_list: List[
Union[ChatCompletionTextObject, ChatCompletionImageObject]
] = []
## USER MESSAGE ## ## USER MESSAGE ##
if m["role"] == "user": if m["role"] == "user":
## translate user message ## translate user message
@ -231,9 +234,6 @@ class AnthropicConfig:
role="user", content=m["content"] role="user", content=m["content"]
) )
elif isinstance(m["content"], list): elif isinstance(m["content"], list):
new_user_content_list: List[
Union[ChatCompletionTextObject, ChatCompletionImageObject]
] = []
for content in m["content"]: for content in m["content"]:
if content["type"] == "text": if content["type"] == "text":
text_obj = ChatCompletionTextObject( text_obj = ChatCompletionTextObject(

2
litellm/tests/test_anthropic_completion.py
View file

@ -275,4 +275,4 @@ def test_anthropic_tool_calling_translation():
print(translated_params["messages"]) print(translated_params["messages"])
assert len(translated_params["messages"]) > 0 assert len(translated_params["messages"]) > 0
assert translated_params["messages"][1]["role"] == "user" assert translated_params["messages"][0]["role"] == "user"