Simplify condition check

Author: kthota-g

src/a2a/server/request_handlers/default_request_handler.py

@@ -33,9 +33,7 @@
     InvalidParamsError,
     ListTaskPushNotificationConfigParams,
     Message,
-    MessageSendConfiguration,
     MessageSendParams,
-    PushNotificationConfig,
     Task,
     TaskIdParams,
     TaskNotFoundError,
@@ -217,13 +215,11 @@ async def _setup_message_execution(
         # dictating the task ID at this layer is useful for tracking running
         # agents.
 
-        if self.should_add_push_info(params):
-            assert self._push_config_store is not None
-            assert isinstance(params.configuration, MessageSendConfiguration)
-            assert isinstance(
-                params.configuration.pushNotificationConfig,
-                PushNotificationConfig,
-            )
+        if (
+            self._push_config_store
+            and params.configuration
+            and params.configuration.pushNotificationConfig
+        ):
             await self._push_config_store.set_info(
                 task_id, params.configuration.pushNotificationConfig
             )
@@ -499,11 +495,3 @@ async def on_delete_task_push_notification_config(
         await self._push_config_store.delete_info(
             params.id, params.pushNotificationConfigId
         )
-
-    def should_add_push_info(self, params: MessageSendParams) -> bool:
-        """Determines if push notification info should be set for a task."""
-        return bool(
-            self._push_config_store
-            and params.configuration
-            and params.configuration.pushNotificationConfig
-        )