@@ -420,14 +420,12 @@ def test_convert_a2a_artifact_update_to_event_none(self):
420420 with pytest .raises (ValueError , match = "A2A artifact update cannot be None" ):
421421 convert_a2a_artifact_update_to_event (None )
422422
423- def test_convert_a2a_message_to_event_user_role (self ):
423+ def test_convert_a2a_message_to_event_user_role (self ) -> None :
424424 """Test that A2A user role maps to GenAI content role 'user'."""
425425 a2a_part = Mock (spec = A2APart )
426426 a2a_part .root = Mock (spec = TextPart )
427427 a2a_part .root .metadata = {}
428- message = Message (
429- message_id = "msg-1" , role = A2ARole .user , parts = [a2a_part ]
430- )
428+ message = Message (message_id = "msg-1" , role = A2ARole .user , parts = [a2a_part ])
431429
432430 mock_genai_part = genai_types .Part .from_text (text = "hello from user" )
433431 mock_part_converter = Mock (return_value = [mock_genai_part ])
@@ -441,14 +439,12 @@ def test_convert_a2a_message_to_event_user_role(self):
441439
442440 assert event .content .role == "user"
443441
444- def test_convert_a2a_message_to_event_agent_role (self ):
442+ def test_convert_a2a_message_to_event_agent_role (self ) -> None :
445443 """Test that A2A agent role maps to GenAI content role 'model'."""
446444 a2a_part = Mock (spec = A2APart )
447445 a2a_part .root = Mock (spec = TextPart )
448446 a2a_part .root .metadata = {}
449- message = Message (
450- message_id = "msg-1" , role = A2ARole .agent , parts = [a2a_part ]
451- )
447+ message = Message (message_id = "msg-1" , role = A2ARole .agent , parts = [a2a_part ])
452448
453449 mock_genai_part = genai_types .Part .from_text (text = "hello from agent" )
454450 mock_part_converter = Mock (return_value = [mock_genai_part ])
0 commit comments