mirror of
https://github.com/OpenHands/OpenHands.git
synced 2025-12-26 13:52:43 +08:00
add test
This commit is contained in:
parent
67de279ef9
commit
0bec9cd30a
@ -83,3 +83,34 @@ def test_message_with_only_text_content_and_vision_disabled():
|
||||
|
||||
assert serialized_message == expected_serialized_message
|
||||
assert message.contains_image is False
|
||||
|
||||
|
||||
def test_message_with_mixed_content_and_vision_disabled():
|
||||
# Create a message with both text and image content
|
||||
text_content1 = TextContent(text='This is a text message')
|
||||
image_content1 = ImageContent(
|
||||
image_urls=['http://example.com/image1.png', 'http://example.com/image2.png']
|
||||
)
|
||||
text_content2 = TextContent(text='This is another text message')
|
||||
image_content2 = ImageContent(
|
||||
image_urls=['http://example.com/image3.png', 'http://example.com/image4.png']
|
||||
)
|
||||
|
||||
# Initialize Message with vision disabled
|
||||
message = Message(
|
||||
role='user',
|
||||
content=[text_content1, image_content1, text_content2, image_content2],
|
||||
vision_enabled=False,
|
||||
)
|
||||
serialized_message = message.serialize_model()
|
||||
|
||||
# Expected serialization ignores images and concatenates text
|
||||
expected_serialized_message = {
|
||||
'role': 'user',
|
||||
'content': 'This is a text message\nThis is another text message',
|
||||
}
|
||||
|
||||
# Assert serialized message matches expectation
|
||||
assert serialized_message == expected_serialized_message
|
||||
# Assert that images exist in the original message
|
||||
assert message.contains_image
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user