Skip to content

Commit ed7f6e4

Browse files
committed
address an ai review comment
1 parent dad36e8 commit ed7f6e4

1 file changed

Lines changed: 1 addition & 8 deletions

File tree

src/a2a/contrib/tasks/vertex_task_converter.py

Lines changed: 1 addition & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -264,16 +264,9 @@ def to_sdk_message(
264264
ptype = part_types[i]
265265
parts.append(to_sdk_part(part, part_metadata=meta, part_type=ptype))
266266

267-
role = None
268-
if stored_msg.role:
269-
try:
270-
role = Role(stored_msg.role)
271-
except ValueError:
272-
role = None
273-
274267
return Message(
275268
message_id=stored_msg.message_id,
276-
role=role, # type: ignore
269+
role=Role(stored_msg.role),
277270
extensions=unpacked_meta.get('extensions'),
278271
reference_task_ids=unpacked_meta.get('reference_tasks'),
279272
metadata=unpacked_meta.get('original_metadata'),

0 commit comments

Comments
 (0)