Skip to content

Commit b3ee26d

Browse files
committed
linting
1 parent c8ea787 commit b3ee26d

File tree

2 files changed

+44
-22
lines changed

2 files changed

+44
-22
lines changed

src/a2a/server/apps/jsonrpc/jsonrpc_app.py

Lines changed: 35 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -10,25 +10,43 @@
1010

1111
from a2a.auth.user import UnauthenticatedUser
1212
from a2a.auth.user import User as A2AUser
13-
from a2a.extensions.common import (HTTP_EXTENSION_HEADER,
14-
get_requested_extensions)
13+
from a2a.extensions.common import (
14+
HTTP_EXTENSION_HEADER,
15+
get_requested_extensions
16+
)
1517
from a2a.server.context import ServerCallContext
1618
from a2a.server.request_handlers.jsonrpc_handler import JSONRPCHandler
1719
from a2a.server.request_handlers.request_handler import RequestHandler
18-
from a2a.types import (A2AError, A2ARequest, AgentCard, CancelTaskRequest,
19-
DeleteTaskPushNotificationConfigRequest,
20-
GetAuthenticatedExtendedCardRequest,
21-
GetTaskPushNotificationConfigRequest, GetTaskRequest,
22-
InternalError, InvalidRequestError, JSONParseError,
23-
JSONRPCError, JSONRPCErrorResponse, JSONRPCRequest,
24-
JSONRPCResponse, ListTaskPushNotificationConfigRequest,
25-
SendMessageRequest, SendStreamingMessageRequest,
26-
SendStreamingMessageResponse,
27-
SetTaskPushNotificationConfigRequest,
28-
TaskResubscriptionRequest, UnsupportedOperationError)
29-
from a2a.utils.constants import (AGENT_CARD_WELL_KNOWN_PATH, DEFAULT_RPC_URL,
30-
EXTENDED_AGENT_CARD_PATH,
31-
PREV_AGENT_CARD_WELL_KNOWN_PATH)
20+
from a2a.types import (
21+
A2AError,
22+
A2ARequest,
23+
AgentCard,
24+
CancelTaskRequest,
25+
DeleteTaskPushNotificationConfigRequest,
26+
GetAuthenticatedExtendedCardRequest,
27+
GetTaskPushNotificationConfigRequest,
28+
GetTaskRequest,
29+
InternalError,
30+
InvalidRequestError,
31+
JSONParseError,
32+
JSONRPCError,
33+
JSONRPCErrorResponse,
34+
JSONRPCRequest,
35+
JSONRPCResponse,
36+
ListTaskPushNotificationConfigRequest,
37+
SendMessageRequest,
38+
SendStreamingMessageRequest,
39+
SendStreamingMessageResponse,
40+
SetTaskPushNotificationConfigRequest,
41+
TaskResubscriptionRequest,
42+
UnsupportedOperationError
43+
)
44+
from a2a.utils.constants import (
45+
AGENT_CARD_WELL_KNOWN_PATH,
46+
DEFAULT_RPC_URL,
47+
EXTENDED_AGENT_CARD_PATH,
48+
PREV_AGENT_CARD_WELL_KNOWN_PATH
49+
)
3250
from a2a.utils.errors import MethodNotImplementedError
3351

3452
logger = logging.getLogger(__name__)
@@ -468,7 +486,7 @@ async def event_generator(
468486
headers=headers,
469487
)
470488

471-
def _modify_rpc_url(self, agent_card: AgentCard, request: Request):
489+
def _modify_rpc_url(self, agent_card: AgentCard, request: Request) -> None:
472490
"""Modifies Agent's RPC URL based on the AgentCard request.
473491
474492
Args:

src/a2a/server/apps/rest/rest_adapter.py

Lines changed: 9 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -27,14 +27,18 @@
2727

2828
_package_starlette_installed = False
2929

30-
from a2a.server.apps.jsonrpc import (CallContextBuilder,
31-
DefaultCallContextBuilder)
30+
from a2a.server.apps.jsonrpc import (
31+
CallContextBuilder,
32+
DefaultCallContextBuilder
33+
)
3234
from a2a.server.context import ServerCallContext
3335
from a2a.server.request_handlers.request_handler import RequestHandler
3436
from a2a.server.request_handlers.rest_handler import RESTHandler
3537
from a2a.types import AgentCard, AuthenticatedExtendedCardNotConfiguredError
36-
from a2a.utils.error_handlers import (rest_error_handler,
37-
rest_stream_error_handler)
38+
from a2a.utils.error_handlers import (
39+
rest_error_handler,
40+
rest_stream_error_handler
41+
)
3842
from a2a.utils.errors import ServerError
3943

4044
logger = logging.getLogger(__name__)
@@ -185,7 +189,7 @@ async def handle_authenticated_agent_card(
185189

186190
return card_to_serve.model_dump(mode='json', exclude_none=True)
187191

188-
def _modify_rpc_url(self, agent_card: AgentCard, request: Request):
192+
def _modify_rpc_url(self, agent_card: AgentCard, request: Request) -> None:
189193
"""Modifies Agent's RPC URL based on the AgentCard request.
190194
191195
Args:

0 commit comments

Comments
 (0)