Update types

Author: holtskinner

src/a2a/client/grpc_client.py

@@ -159,8 +159,8 @@ async def set_task_callback(
         Returns:
             A `TaskPushNotificationConfig` object containing the config.
         """
-        config = await self.stub.CreateTaskPushNotification(
-            a2a_pb2.CreateTaskPushNotificationRequest(
+        config = await self.stub.CreateTaskPushNotificationConfig(
+            a2a_pb2.CreateTaskPushNotificationConfigRequest(
                 parent='',
                 config_id='',
                 config=proto_utils.ToProto.task_push_notification_config(
@@ -182,8 +182,8 @@ async def get_task_callback(
         Returns:
             A `TaskPushNotificationConfig` object containing the configuration.
         """
-        config = await self.stub.GetTaskPushNotification(
-            a2a_pb2.GetTaskPushNotificationRequest(
+        config = await self.stub.GetTaskPushNotificationConfig(
+            a2a_pb2.GetTaskPushNotificationConfigRequest(
                 name=f'tasks/{request.id}/pushNotification/undefined',
             )
         )

src/a2a/server/request_handlers/grpc_handler.py

@@ -55,7 +55,7 @@ def __init__(
         self,
         agent_card: AgentCard,
         request_handler: RequestHandler,
-        context_builder: CallContextBuilder | None = None
+        context_builder: CallContextBuilder | None = None,
     ):
         """Initializes the GrpcHandler.
 
@@ -198,12 +198,12 @@ async def TaskSubscription(
         except ServerError as e:
             await self.abort_context(e, context)
 
-    async def GetTaskPushNotification(
+    async def GetTaskPushNotificationConfig(
         self,
-        request: a2a_pb2.GetTaskPushNotificationRequest,
+        request: a2a_pb2.GetTaskPushNotificationConfigRequest,
         context: grpc.aio.ServicerContext,
     ) -> a2a_pb2.TaskPushNotificationConfig:
-        """Handles the 'GetTaskPushNotification' gRPC method.
+        """Handles the 'GetTaskPushNotificationConfig' gRPC method.
 
         Args:
             request: The incoming `GetTaskPushNotificationConfigRequest` object.
@@ -229,17 +229,17 @@ async def GetTaskPushNotification(
         lambda self: self.agent_card.capabilities.pushNotifications,
         'Push notifications are not supported by the agent',
     )
-    async def CreateTaskPushNotification(
+    async def CreateTaskPushNotificationConfig(
         self,
-        request: a2a_pb2.CreateTaskPushNotificationRequest,
+        request: a2a_pb2.CreateTaskPushNotificationConfigRequest,
         context: grpc.aio.ServicerContext,
     ) -> a2a_pb2.TaskPushNotificationConfig:
-        """Handles the 'CreateTaskPushNotification' gRPC method.
+        """Handles the 'CreateTaskPushNotificationConfig' gRPC method.
 
         Requires the agent to support push notifications.
 
         Args:
-            request: The incoming `CreateTaskPushNotificationRequest` object.
+            request: The incoming `CreateTaskPushNotificationConfigRequest` object.
             context: Context provided by the server.
 
         Returns:

src/a2a/utils/proto_utils.py

@@ -589,12 +589,12 @@ def task_id_params(
         request: (
             a2a_pb2.CancelTaskRequest
             | a2a_pb2.TaskSubscriptionRequest
-            | a2a_pb2.GetTaskPushNotificationRequest
+            | a2a_pb2.GetTaskPushNotificationConfigRequest
         ),
     ) -> types.TaskIdParams:
         # This is currently incomplete until the core sdk supports multiple
         # configs for a single task.
-        if isinstance(request, a2a_pb2.GetTaskPushNotificationRequest):
+        if isinstance(request, a2a_pb2.GetTaskPushNotificationConfigRequest):
             m = re.match(_TASK_PUSH_CONFIG_NAME_MATCH, request.name)
             if not m:
                 raise ServerError(
@@ -615,7 +615,7 @@ def task_id_params(
     @classmethod
     def task_push_notification_config(
         cls,
-        request: a2a_pb2.CreateTaskPushNotificationRequest,
+        request: a2a_pb2.CreateTaskPushNotificationConfigRequest,
     ) -> types.TaskPushNotificationConfig:
         m = re.match(_TASK_NAME_MATCH, request.parent)
         if not m: