|
39 | 39 | from mcp.server.transport_security import TransportSecuritySettings |
40 | 40 | from mcp.shared.context import RequestContext |
41 | 41 | from mcp.shared.exceptions import McpError |
42 | | -from mcp.shared.message import ClientMessageMetadata |
| 42 | +from mcp.shared.message import ClientMessageMetadata, SessionMessage |
43 | 43 | from mcp.shared.session import RequestResponder |
44 | 44 | from mcp.types import InitializeResult, TextContent, TextResourceContents, Tool |
45 | 45 | from tests.test_helpers import wait_for_server |
@@ -1768,8 +1768,10 @@ async def test_resume_stream_without_session_id(): |
1768 | 1768 | transport = StreamableHTTPTransport("http://localhost:8000") |
1769 | 1769 | assert transport.session_id is None |
1770 | 1770 |
|
1771 | | - # Create a dummy stream writer |
1772 | | - read_stream_writer, read_stream_reader = anyio.create_memory_object_stream(0) |
| 1771 | + # Create a dummy stream writer with type annotation |
| 1772 | + read_stream_writer, read_stream_reader = anyio.create_memory_object_stream[ |
| 1773 | + SessionMessage | Exception |
| 1774 | + ](0) |
1773 | 1775 |
|
1774 | 1776 | async with httpx.AsyncClient() as client: |
1775 | 1777 | # Should return early without making request |
@@ -1804,7 +1806,9 @@ async def test_resume_stream_with_405_response(basic_server: None, basic_server_ |
1804 | 1806 | transport.session_id = get_session_id() |
1805 | 1807 |
|
1806 | 1808 | # Now try to resume with the session - server might return 405 |
1807 | | - read_stream_writer, read_stream_reader = anyio.create_memory_object_stream(0) |
| 1809 | + read_stream_writer, read_stream_reader = anyio.create_memory_object_stream[ |
| 1810 | + SessionMessage | Exception |
| 1811 | + ](0) |
1808 | 1812 |
|
1809 | 1813 | async with httpx.AsyncClient() as client: |
1810 | 1814 | # This should handle the 405 gracefully |
|
0 commit comments