Fix lint errors in jsonrpc_client

Author: holtskinner

src/a2a/client/jsonrpc_client.py

@@ -33,6 +33,7 @@
     GetTaskResponse,
     JSONRPCErrorResponse,
     Message,
+    MessageSendConfiguration,
     MessageSendParams,
     SendMessageRequest,
     SendMessageResponse,
@@ -573,6 +574,14 @@ def __init__(
     def get_http_args(
         self, context: ClientCallContext
     ) -> dict[str, Any] | None:
+        """Extract HTTP-specific keyword arguments from the client call context.
+
+        Args:
+            context: The client call context.
+
+        Returns:
+            A dictionary of HTTP arguments, or None.
+        """
         return context.state.get('http_kwargs', None) if context else None
 
     async def send_message(
@@ -581,6 +590,21 @@ async def send_message(
         *,
         context: ClientCallContext | None = None,
     ) -> AsyncIterator[Task | Message]:
+        """Send a message to the agent and consumes the response(s).
+
+        This method handles both blocking (non-streaming) and streaming responses
+        based on the client configuration and agent capabilities.
+
+        Args:
+            request: The message to send.
+            context: The client call context.
+
+        Yields:
+            The final message or task result from the agent.
+
+        Raises:
+            JSONRPCError: If the agent returns a JSON-RPC error in the response.
+        """
         config = MessageSendConfiguration(
             accepted_output_modes=self._config.accepted_output_modes,
             blocking=not self._config.polling,
@@ -642,6 +666,15 @@ async def get_task(
         *,
         context: ClientCallContext | None = None,
     ) -> Task:
+        """Retrieve a task from the agent.
+
+        Args:
+            request: Parameters to identify the task.
+            context: The client call context.
+
+        Returns:
+            The requested task.
+        """
         response = await self._transport_client.get_task(
             GetTaskRequest(
                 params=request,
@@ -658,6 +691,15 @@ async def cancel_task(
         *,
         context: ClientCallContext | None = None,
     ) -> Task:
+        """Cancel an ongoing task on the agent.
+
+        Args:
+            request: Parameters to identify the task to cancel.
+            context: The client call context.
+
+        Returns:
+            The task after the cancellation request.
+        """
         response = await self._transport_client.cancel_task(
             CancelTaskRequest(
                 params=request,
@@ -674,6 +716,15 @@ async def set_task_callback(
         *,
         context: ClientCallContext | None = None,
     ) -> TaskPushNotificationConfig:
+        """Set a push notification callback for a task.
+
+        Args:
+            request: The push notification configuration to set.
+            context: The client call context.
+
+        Returns:
+            The configured task push notification configuration.
+        """
         response = await self._transport_client.set_task_callback(
             SetTaskPushNotificationConfigRequest(
                 params=request,
@@ -690,6 +741,15 @@ async def get_task_callback(
         *,
         context: ClientCallContext | None = None,
     ) -> TaskPushNotificationConfig:
+        """Retrieve the push notification callback configuration for a task.
+
+        Args:
+            request: Parameters to identify the task and configuration.
+            context: The client call context.
+
+        Returns:
+            The requested task push notification configuration.
+        """
         response = await self._transport_client.get_task_callback(
             GetTaskPushNotificationConfigRequest(
                 params=request,
@@ -706,6 +766,20 @@ async def resubscribe(
         *,
         context: ClientCallContext | None = None,
     ) -> AsyncIterator[Task | Message]:
+        """Resubscribe to a task's event stream.
+
+        This is only available if both the client and server support streaming.
+
+        Args:
+            request: Parameters to identify the task to resubscribe to.
+            context: The client call context.
+
+        Yields:
+            Task events from the agent.
+
+        Raises:
+            Exception: If streaming is not supported.
+        """
         if not self._config.streaming or not self._card.capabilities.streaming:
             raise Exception(
                 'client and/or server do not support resubscription.'
@@ -726,17 +800,28 @@ async def get_card(
         *,
         context: ClientCallContext | None = None,
     ) -> AgentCard:
+        """Retrieve the agent's card.
+
+        This may involve fetching the public card first if not already available,
+        and then fetching the authenticated extended card if supported and required.
+
+        Args:
+            context: The client call context.
+
+        Returns:
+            The agent's card.
+        """
         return await self._transport_client.get_card(
             http_kwargs=self.get_http_args(context),
             context=context,
         )
 
 
-def NewJsonRpcClient(
+def NewJsonRpcClient(  # noqa: N802
     card: AgentCard,
     config: ClientConfig,
     consumers: list[Consumer],
     middleware: list[ClientCallInterceptor],
 ) -> Client:
-    """Generator for the `JsonRpcClient` implementation."""
+    """Factory function for the `JsonRpcClient` implementation."""
     return JsonRpcClient(card, config, consumers, middleware)