From 2ecbf6663a70dbee13520fda133f74142a9df017 Mon Sep 17 00:00:00 2001
From: Emir Ayar <emirayar@amazon.lu>
Date: Sat, 27 Apr 2024 12:27:12 +0200
Subject: [PATCH] Add test for completion with text content dictionaries

---
 litellm/tests/test_completion.py | 42 ++++++++++++++++++++++++++++++++
 1 file changed, 42 insertions(+)

diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py
index d2c004a0a..e2c635a5c 100644
--- a/litellm/tests/test_completion.py
+++ b/litellm/tests/test_completion.py
@@ -230,6 +230,48 @@ 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