Skip to content

Commit 34949a8

Browse files
fix: Update test_http_unicode to use new streamable_http_client API
Replace deprecated streamablehttp_client with streamable_http_client to fix test failures and pyright errors.
1 parent 601ac99 commit 34949a8

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

src/mcp/client/streamable_http.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -560,8 +560,8 @@ def start_get_stream() -> None:
560560
await write_stream.aclose()
561561

562562

563-
@deprecated("Use `streamable_http_client` instead.")
564563
@asynccontextmanager
564+
@deprecated("Use `streamable_http_client` instead.")
565565
async def streamablehttp_client(
566566
url: str,
567567
headers: dict[str, str] | None = None,

tests/client/test_http_unicode.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@
1212
import pytest
1313

1414
from mcp.client.session import ClientSession
15-
from mcp.client.streamable_http import streamablehttp_client
15+
from mcp.client.streamable_http import streamable_http_client
1616
from tests.test_helpers import wait_for_server
1717

1818
# Test constants with various Unicode characters
@@ -178,7 +178,7 @@ async def test_streamable_http_client_unicode_tool_call(running_unicode_server:
178178
base_url = running_unicode_server
179179
endpoint_url = f"{base_url}/mcp"
180180

181-
async with streamablehttp_client(endpoint_url) as (read_stream, write_stream, _get_session_id):
181+
async with streamable_http_client(endpoint_url) as (read_stream, write_stream, _get_session_id):
182182
async with ClientSession(read_stream, write_stream) as session:
183183
await session.initialize()
184184

@@ -210,7 +210,7 @@ async def test_streamable_http_client_unicode_prompts(running_unicode_server: st
210210
base_url = running_unicode_server
211211
endpoint_url = f"{base_url}/mcp"
212212

213-
async with streamablehttp_client(endpoint_url) as (read_stream, write_stream, _get_session_id):
213+
async with streamable_http_client(endpoint_url) as (read_stream, write_stream, _get_session_id):
214214
async with ClientSession(read_stream, write_stream) as session:
215215
await session.initialize()
216216

0 commit comments

Comments
 (0)