David Soria Parra 827e494df4 feat: add request cancellation and in-flight request tracking ...

This commit adds support for request cancellation and tracking of
in-flight requests in the MCP protocol implementation. The key
architectural changes are:

1. Request Lifecycle Management:
   - Added _in_flight dictionary to BaseSession to track active requests
   - Requests are tracked from receipt until completion/cancellation
   - Added proper cleanup via on_complete callback

2. Cancellation Support:
   - Added CancelledNotification handling in _receive_loop
   - Implemented cancel() method in RequestResponder
   - Uses anyio.CancelScope for robust cancellation
   - Sends error response on cancellation

3. Request Context:
   - Added request_ctx ContextVar for request context
   - Ensures proper cleanup after request handling
   - Maintains request state throughout lifecycle

4. Error Handling:
   - Improved error propagation for cancelled requests
   - Added proper cleanup of cancelled requests
   - Maintains consistency of in-flight tracking

This change enables clients to cancel long-running requests and
servers to properly clean up resources when requests are cancelled.

Github-Issue:#88

2025-02-03 20:50:05 +00:00

..

test_memory.py

rename mcp_python to mcp

2024-11-11 12:40:32 +00:00

test_session.py

feat: add request cancellation and in-flight request tracking

2025-02-03 20:50:05 +00:00

test_sse.py

fix: add test for #176

2025-01-28 11:19:13 +00:00