@@ -46,11 +46,12 @@ def test_serializer_uses_default_handler_for_messages():
4646 human_msg = HumanMessage (content = "What is the weather?" , id = "msg-1" )
4747
4848 # This should NOT raise TypeError
49- serialized_bytes = dumps_helper (serializer , human_msg )
49+ typed_data = dumps_helper (serializer , human_msg )
50+ type_str , serialized_bytes = typed_data
5051 assert isinstance (serialized_bytes , bytes )
5152
5253 # Deserialize and verify
53- deserialized = loads_helper (serializer , serialized_bytes )
54+ deserialized = loads_helper (serializer , typed_data )
5455 assert isinstance (deserialized , HumanMessage )
5556 assert deserialized .content == "What is the weather?"
5657 assert deserialized .id == "msg-1"
@@ -72,11 +73,12 @@ def test_serializer_handles_all_message_types():
7273
7374 for msg in messages :
7475 # Serialize
75- serialized = dumps_helper (serializer , msg )
76+ typed_data = dumps_helper (serializer , msg )
77+ type_str , serialized = typed_data
7678 assert isinstance (serialized , bytes )
7779
7880 # Deserialize
79- deserialized = loads_helper (serializer , serialized )
81+ deserialized = loads_helper (serializer , typed_data )
8082
8183 # Verify type is preserved
8284 assert type (deserialized ) == type (msg )
@@ -98,11 +100,12 @@ def test_serializer_handles_message_lists():
98100 ]
99101
100102 # Serialize the list
101- serialized = dumps_helper (serializer , messages )
103+ typed_data = dumps_helper (serializer , messages )
104+ type_str , serialized = typed_data
102105 assert isinstance (serialized , bytes )
103106
104107 # Deserialize
105- deserialized = loads_helper (serializer , serialized )
108+ deserialized = loads_helper (serializer , typed_data )
106109
107110 # Verify structure
108111 assert isinstance (deserialized , list )
@@ -131,11 +134,12 @@ def test_serializer_handles_nested_structures_with_messages():
131134 }
132135
133136 # Serialize
134- serialized = dumps_helper (serializer , state )
137+ typed_data = dumps_helper (serializer , state )
138+ type_str , serialized = typed_data
135139 assert isinstance (serialized , bytes )
136140
137141 # Deserialize
138- deserialized = loads_helper (serializer , serialized )
142+ deserialized = loads_helper (serializer , typed_data )
139143
140144 # Verify structure
141145 assert "messages" in deserialized
0 commit comments