Fix some of the mypy errors

Author: holtskinner

src/a2a/client/client_factory.py

@@ -118,7 +118,7 @@ def create(
         if consumers:
             all_consumers.extend(consumers)
         return self._registry[transport](
-            card, self._config, all_consumers, interceptors
+            card, self._config, all_consumers, interceptors or []
         )
 
 

src/a2a/client/client_task_manager.py

@@ -75,7 +75,7 @@ async def save_task_event(
         if not self._task_id:
             self._task_id = task_id_from_event
         if not self._context_id:
-            self._context_id = event.contextId
+            self._context_id = event.context_id
 
         logger.debug(
             'Processing save of task event of type %s for task_id: %s',
@@ -88,7 +88,7 @@ async def save_task_event(
             task = Task(
                 status=TaskStatus(state=TaskState.unknown),
                 id=task_id_from_event,
-                contextId=self._context_id if self._context_id else '',
+                context_id=self._context_id if self._context_id else '',
             )
         if isinstance(event, TaskStatusUpdateEvent):
             logger.debug(
@@ -142,7 +142,7 @@ async def _save_task(self, task: Task) -> None:
         if not self._task_id:
             logger.info('New task created with id: %s', task.id)
             self._task_id = task.id
-            self._context_id = task.contextId
+            self._context_id = task.context_id
 
     def update_with_message(self, message: Message, task: Task) -> Task:
         """Updates a task object adding a new message to its history.

src/a2a/client/grpc_client.py

@@ -253,6 +253,8 @@ async def get_card(
         """
         # If we don't have the public card, try to get that first.
         card = self.agent_card
+        if card is None and not self._needs_extended_card:
+            raise ValueError('Agent card is not available.')
 
         if not self._needs_extended_card:
             return card
@@ -278,7 +280,7 @@ def __init__(
     ):
         super().__init__(consumers, middleware)
         if not config.grpc_channel_factory:
-            raise Exception('GRPC client requires channel factory.')
+            raise ValueError('GRPC client requires channel factory.')
         self._card = card
         self._config = config
         # Defer init to first use.
@@ -452,9 +454,16 @@ async def resubscribe(
             Exception: If streaming is not supported by the client or server.
         """
         if not self._config.streaming or not self._card.capabilities.streaming:
-            raise Exception(
+            raise NotImplementedError(
                 'client and/or server do not support resubscription.'
             )
+        if not self._transport_client:
+            raise ValueError('Transport client is not initialized.')
+        if not hasattr(self._transport_client, 'resubscribe'):
+            # This can happen if the proto definitions are out of date or the method is missing
+            raise NotImplementedError(
+                'Resubscribe is not implemented on the gRPC transport client.'
+            )
         async for event in self._transport_client.resubscribe(
             request,
             context=context,

src/a2a/client/jsonrpc_client.py

@@ -533,7 +533,7 @@ async def get_card(
         # Apply interceptors before sending
         payload, modified_kwargs = await self._apply_interceptors(
             'card/getAuthenticated',
-            '',
+            {},
             http_kwargs,
             context,
         )
@@ -600,7 +600,7 @@ async def send_message(
             context: The client call context.
 
         Yields:
-            The final message or task result from the agent.
+            An async iterator of `ClientEvent` or a final `Message` response.
 
         Raises:
             JSONRPCError: If the agent returns a JSON-RPC error in the response.

src/a2a/client/rest_client.py

@@ -117,7 +117,7 @@ async def send_message(
         pb = a2a_pb2.SendMessageRequest(
             request=proto_utils.ToProto.message(request.message),
             configuration=proto_utils.ToProto.send_message_config(
-                request.config
+                request.configuration
             ),
             metadata=(
                 proto_utils.ToProto.metadata(request.metadata)
@@ -387,9 +387,9 @@ async def set_task_callback(
         """
         pb = a2a_pb2.CreateTaskPushNotificationConfigRequest(
             parent=f'tasks/{request.taskId}',
-            config_id=request.pushNotificationConfig.id,
+            config_id=request.push_notification_config.id,
             config=proto_utils.ToProto.push_notification_config(
-                request.pushNotificationConfig
+                request.push_notification_config
             ),
         )
         payload = MessageToDict(pb)