updating test

TLSDC · TLSDC · commit 3d19905f335d · 2024-11-29T17:18:10.000-05:00
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():

Original file line number	Diff line number	Diff line change
`@@ -251,8 +251,7 @@ def test_message_merge_only_text():`
`251`	`251`	`]`
`252`	`252`	`message = llm_utils.BaseMessage(role="system", content=content)`
`253`	`253`	`message.merge()`
`254`		`- assert len(message["content"]) == 1`
`255`		`- assert message["content"][0]["text"] == "Hello, world!\nThis is a test."`
	`254`	`+ assert message["content"] == "Hello, world!\nThis is a test."`
`256`	`255`
`257`	`256`
`258`	`257`	`def test_message_merge_text_image():`