File tree Expand file tree Collapse file tree 1 file changed +5
-2
lines changed
src/a2a/server/request_handlers Expand file tree Collapse file tree 1 file changed +5
-2
lines changed Original file line number Diff line number Diff line change 1+ # ruff: noqa: N802
12import contextlib
23import logging
34
@@ -55,7 +56,7 @@ def __init__(
5556 self ,
5657 agent_card : AgentCard ,
5758 request_handler : RequestHandler ,
58- context_builder : CallContextBuilder = DefaultCallContextBuilder (),
59+ context_builder : CallContextBuilder | None = None
5960 ):
6061 """Initializes the GrpcHandler.
6162
@@ -66,7 +67,7 @@ def __init__(
6667 """
6768 self .agent_card = agent_card
6869 self .request_handler = request_handler
69- self .context_builder = context_builder
70+ self .context_builder = context_builder or DefaultCallContextBuilder ()
7071
7172 async def SendMessage (
7273 self ,
@@ -295,11 +296,13 @@ async def GetAgentCard(
295296 request : a2a_pb2 .GetAgentCardRequest ,
296297 context : grpc .aio .ServicerContext ,
297298 ) -> a2a_pb2 .AgentCard :
299+ """Get the agent card for the agent served."""
298300 return proto_utils .ToProto .agent_card (self .agent_card )
299301
300302 async def abort_context (
301303 self , error : ServerError , context : grpc .ServicerContext
302304 ) -> None :
305+ """Sets the grpc errors appropriately in the context."""
303306 match error .error :
304307 case types .JSONParseError ():
305308 await context .abort (
You can’t perform that action at this time.
0 commit comments