updating test

ServiceNow · Nov 29, 2024 · 3d19905 · 3d19905
1 parent 05870ed
commit 3d19905
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/tests/llm/test_llm_utils.py b/tests/llm/test_llm_utils.py
@@ -251,8 +251,7 @@ def test_message_merge_only_text():
     ]
     message = llm_utils.BaseMessage(role="system", content=content)
     message.merge()
-    assert len(message["content"]) == 1
-    assert message["content"][0]["text"] == "Hello, world!\nThis is a test."
+    assert message["content"] == "Hello, world!\nThis is a test."
 
 
 def test_message_merge_text_image():