Additional ruff/mypy fixes

Author: pstephengoogle

src/a2a/server/request_handlers/grpc_handler.py

@@ -40,6 +40,7 @@ class DefaultCallContextBuilder(CallContextBuilder):
     """A default implementation of CallContextBuilder."""
 
     def build(self, context: grpc.ServicerContext) -> ServerCallContext:
+        """Builds the ServerCallContext."""
         user = UnauthenticatedUser()
         state = {}
         with contextlib.suppress(Exception):
@@ -48,9 +49,7 @@ def build(self, context: grpc.ServicerContext) -> ServerCallContext:
 
 
 class GrpcHandler(a2a_grpc.A2AServiceServicer):
-    """Maps incoming gRPC requests to the appropriate request handler method
-    and formats responses.
-    """
+    """Maps incoming gRPC requests to the appropriate request handler method."""
 
     def __init__(
         self,
@@ -64,6 +63,8 @@ def __init__(
             agent_card: The AgentCard describing the agent's capabilities.
             request_handler: The underlying `RequestHandler` instance to
                              delegate requests to.
+            context_builder: The CallContextBuilder object. If none the
+                             DefaultCallContextBuilder is used.
         """
         self.agent_card = agent_card
         self.request_handler = request_handler

src/a2a/utils/proto_utils.py

@@ -89,9 +89,9 @@ def task(cls, task: types.Task) -> a2a_pb2.Task:
                 else None
             ),
             history=(
-                [ToProto.message(h) for h in task.history]
+                [ToProto.message(h) for h in task.history]  # type: ignore[misc]
                 if task.history
-                else None  # type: ignore[misc]
+                else None
             ),
         )
 
@@ -103,7 +103,7 @@ def task_status(cls, status: types.TaskStatus) -> a2a_pb2.TaskStatus:
         )
 
     @classmethod
-    def task_state(cls, state: types.TaskState) -> a2a_pb2.TaskState:
+    def task_state(cls, state: types.TaskState) -> a2a_pb2.TaskState: # ruff: noqa: PLR0911
         match state:
             case types.TaskState.submitted:
                 return a2a_pb2.TaskState.TASK_STATE_SUBMITTED
@@ -488,7 +488,7 @@ def task_status(cls, status: a2a_pb2.TaskStatus) -> types.TaskStatus:
         )
 
     @classmethod
-    def task_state(cls, state: a2a_pb2.TaskState) -> types.TaskState:
+    def task_state(cls, state: a2a_pb2.TaskState) -> types.TaskState: # ruff: noqa: PLR0911
         match state:
             case a2a_pb2.TaskState.TASK_STATE_SUBMITTED:
                 return types.TaskState.submitted