o *i@s8dZddlmZmZddlmZmZGdddeZdS)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 RequestIdc@sDeZdZdZdedeeefdefddZ 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_idresponsereturncCdS)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 W/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/mcp/shared/response_router.pyroute_response' zResponseRouter.route_responseerrorcCr )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__rdictstrrboolrrrr r r r rs rN)rtypingrr mcp.typesrrrr r r r s