|
10 | 10 |
|
11 | 11 | from a2a.auth.user import UnauthenticatedUser |
12 | 12 | 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 | +) |
15 | 17 | from a2a.server.context import ServerCallContext |
16 | 18 | from a2a.server.request_handlers.jsonrpc_handler import JSONRPCHandler |
17 | 19 | 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 | +) |
32 | 50 | from a2a.utils.errors import MethodNotImplementedError |
33 | 51 |
|
34 | 52 | logger = logging.getLogger(__name__) |
@@ -468,7 +486,7 @@ async def event_generator( |
468 | 486 | headers=headers, |
469 | 487 | ) |
470 | 488 |
|
471 | | - def _modify_rpc_url(self, agent_card: AgentCard, request: Request): |
| 489 | + def _modify_rpc_url(self, agent_card: AgentCard, request: Request) -> None: |
472 | 490 | """Modifies Agent's RPC URL based on the AgentCard request. |
473 | 491 |
|
474 | 492 | Args: |
|
0 commit comments