Fix mypy errors

Author: holtskinner

.mypy.ini

@@ -1,5 +1,6 @@
 [mypy]
-disable_error_code = import-not-found,annotation-unchecked
+exclude = src/a2a/grpc/
+disable_error_code = import-not-found,annotation-unchecked,import-untyped
 
 [mypy-examples.*]
 follow_imports = skip

src/a2a/server/request_handlers/request_handler.py

@@ -4,6 +4,7 @@
 from a2a.server.context import ServerCallContext
 from a2a.server.events.event_queue import Event
 from a2a.types import (
+    GetTaskPushNotificationConfigParams,
     Message,
     MessageSendParams,
     Task,
@@ -122,7 +123,7 @@ async def on_set_task_push_notification_config(
     @abstractmethod
     async def on_get_task_push_notification_config(
         self,
-        params: TaskIdParams,
+        params: TaskIdParams | GetTaskPushNotificationConfigParams,
         context: ServerCallContext | None = None,
     ) -> TaskPushNotificationConfig:
         """Handles the 'tasks/pushNotificationConfig/get' method.