Skip to content

Commit dcc1e9f

Browse files
committed
Address Gemini code assist comments
1 parent 31daf62 commit dcc1e9f

File tree

1 file changed

+4
-8
lines changed

1 file changed

+4
-8
lines changed

tests/client/test_grpc_client.py

Lines changed: 4 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -362,10 +362,8 @@ async def test_set_task_callback_with_invalid_task(
362362
f'invalid-path-to-tasks/{sample_task_push_notification_config.task_id}/'
363363
f'pushNotificationConfigs/{sample_task_push_notification_config.push_notification_config.id}'
364364
),
365-
push_notification_config=a2a_pb2.PushNotificationConfig(
366-
id=sample_task_push_notification_config.push_notification_config.id,
367-
url=sample_task_push_notification_config.push_notification_config.url,
368-
token=sample_task_push_notification_config.push_notification_config.token,
365+
push_notification_config=proto_utils.ToProto.push_notification_config(
366+
sample_task_push_notification_config.push_notification_config
369367
),
370368
)
371369

@@ -421,10 +419,8 @@ async def test_get_task_callback_with_invalid_task(
421419
f'invalid-path-to-tasks/{sample_task_push_notification_config.task_id}/'
422420
f'pushNotificationConfigs/{sample_task_push_notification_config.push_notification_config.id}'
423421
),
424-
push_notification_config=a2a_pb2.PushNotificationConfig(
425-
id=sample_task_push_notification_config.push_notification_config.id,
426-
url=sample_task_push_notification_config.push_notification_config.url,
427-
token=sample_task_push_notification_config.push_notification_config.token,
422+
push_notification_config=proto_utils.ToProto.push_notification_config(
423+
sample_task_push_notification_config.push_notification_config
428424
),
429425
)
430426
params = GetTaskPushNotificationConfigParams(

0 commit comments

Comments
 (0)