o cÛ·iûã@s8dZddlmZmZddlmZmZGdd„deƒZdS)a: ResponseRouter - Protocol for pluggable response routing. This module defines a protocol for routing JSON-RPC responses to alternative handlers before falling back to the default response stream mechanism. The primary use case is task-augmented requests: when a TaskSession enqueues a request (like elicitation), the response needs to be routed back to the waiting resolver instead of the normal response stream. Design: - Protocol-based for testability and flexibility - Returns bool to indicate if response was handled - Supports both success responses and errors é)ÚAnyÚProtocol)Ú ErrorDataÚ RequestIdc@sDeZdZdZdedeeefdefdd„Z dede defdd „Z d S) ÚResponseRoutera Protocol for routing responses to alternative handlers. Implementations check if they have a pending request for the given ID and deliver the response/error to the appropriate handler. Example: class TaskResultHandler(ResponseRouter): def route_response(self, request_id, response): resolver = self._pending_requests.pop(request_id, None) if resolver: resolver.set_result(response) return True return False Ú request_idÚresponseÚreturncCódS)a Try to route a response to a pending request handler. Args: request_id: The JSON-RPC request ID from the response response: The response result data Returns: True if the response was handled, False otherwise N©)Úselfrrr r úP/home/ubuntu/vllm_env/lib/python3.10/site-packages/mcp/shared/response_router.pyÚroute_response'ó zResponseRouter.route_responseÚerrorcCr )a  Try to route an error to a pending request handler. Args: request_id: The JSON-RPC request ID from the error response error: The error data Returns: True if the error was handled, False otherwise Nr )r rrr r r Ú route_error4rzResponseRouter.route_errorN) Ú__name__Ú __module__Ú __qualname__Ú__doc__rÚdictÚstrrÚboolrrrr r r r rs rN)rÚtypingrrÚ mcp.typesrrrr r r r Ús