)`iFdZddlmZmZddlmZmZGddeZdS)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 n/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/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