diff --git a/litellm/llms/prompt_templates/factory.py b/litellm/llms/prompt_templates/factory.py index cb1b2eb73d..c51dc89be5 100644 --- a/litellm/llms/prompt_templates/factory.py +++ b/litellm/llms/prompt_templates/factory.py @@ -748,15 +748,9 @@ def anthropic_messages_pt_xml(messages: list): assistant_content = [] ## MERGE CONSECUTIVE ASSISTANT CONTENT ## while msg_i < len(messages) and messages[msg_i]["role"] == "assistant": - # Handle assistant messages as string, none, or list of text-content dictionaries. - if isinstance(messages[msg_i].get("content"), list): - assistant_text = '' - for content in messages[msg_i]["content"]: - if content.get("type") == "text": - assistant_text += content["text"] - else: - assistant_text = messages[msg_i].get("content") or "" - + assistant_text = ( + messages[msg_i].get("content") or "" + ) # either string or none if messages[msg_i].get( "tool_calls", [] ): # support assistant tool invoke convertion diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py index b4e06f596d..5317108412 100644 --- a/litellm/tests/test_completion.py +++ b/litellm/tests/test_completion.py @@ -230,48 +230,6 @@ def test_completion_claude_3_function_call(): except Exception as e: pytest.fail(f"Error occurred: {e}") -def test_completion_claude_3_with_text_content_dictionaries(): - litellm.set_verbose = True - messages = [ - { - "role": "user", - "content": [ - { - "type": "text", - "text": "Hello" - } - ] - }, - { - "role": "assistant", - "content": [ - { - "type": "text", - "text": "Hello! How can I assist you today?" - } - ] - }, - { - "role": "user", - "content": [ - { - "type": "text", - "text": "Hello again!" - } - ] - } - ] - - try: - # test without max tokens - response = completion( - model="anthropic/claude-3-opus-20240229", - messages=messages, - ) - # Add any assertions, here to check response args - print(response) - except Exception as e: - pytest.fail(f"Error occurred: {e}") def test_parse_xml_params(): from litellm.llms.prompt_templates.factory import parse_xml_params