j:FdZddlmZmZddlmZmZGddeZdS)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 )AnyProtocol) ErrorData RequestIdcJeZdZdZdedeeefdefdZ dede defdZ dS) 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_idresponsereturncdS)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)selfr r s HC:\Projects\DSCons\.venv\Lib\site-packages\mcp\shared\response_router.pyroute_responsezResponseRouter.route_response'  errorcdS)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 )rr rs r route_errorzResponseRouter.route_error4rrN) __name__ __module__ __qualname____doc__rdictstrrboolrrrr rrrrsy    d38n  QU       i   d       rrN)rtypingrr mcp.typesrrrr rrrsu ! ********) ) ) ) ) X) ) ) ) ) r