Skip to content

Commit bec6717

Browse files
committed
style: apply pyink formatting and add return type annotations
1 parent ec8a49e commit bec6717

2 files changed

Lines changed: 5 additions & 11 deletions

File tree

src/google/adk/a2a/converters/to_adk_event.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -384,9 +384,7 @@ def convert_a2a_message_to_event(
384384
output_parts, _ = _convert_a2a_parts_to_adk_parts(
385385
a2a_message.parts, part_converter
386386
)
387-
content_role = _a2a_role_to_content_role(
388-
getattr(a2a_message, "role", None)
389-
)
387+
content_role = _a2a_role_to_content_role(getattr(a2a_message, "role", None))
390388
return _create_event(
391389
output_parts,
392390
invocation_context,

tests/unittests/a2a/converters/test_to_adk.py

Lines changed: 4 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)