Skip to content

Commit 0aa5a45

Browse files
committed
fix ruff errors
1 parent c880cc1 commit 0aa5a45

File tree

2 files changed

+8
-4
lines changed

2 files changed

+8
-4
lines changed

src/mcp/server/streaming_asgi_transport.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,8 +9,8 @@
99
"""
1010

1111
import typing
12+
from collections.abc import Awaitable, Callable
1213
from typing import Any, cast
13-
from typing import Callable, Awaitable
1414

1515
import anyio
1616
import anyio.abc

tests/shared/test_sse.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -21,8 +21,8 @@
2121
from mcp.client.sse import sse_client
2222
from mcp.server import Server
2323
from mcp.server.sse import SseServerTransport
24-
from mcp.server.transport_security import TransportSecuritySettings
2524
from mcp.server.streaming_asgi_transport import StreamingASGITransport
25+
from mcp.server.transport_security import TransportSecuritySettings
2626
from mcp.shared.exceptions import McpError
2727
from mcp.types import (
2828
EmptyResult,
@@ -418,7 +418,10 @@ def create_test_client(
418418
follow_redirects=True,
419419
)
420420

421-
async with sse_client("http://testserver/sse", headers=custom_headers, httpx_client_factory=create_test_client) as (
421+
async with sse_client("http://testserver/sse",
422+
headers=custom_headers,
423+
httpx_client_factory=create_test_client,
424+
sse_read_timeout=0.5) as (
422425
read_stream,
423426
write_stream,
424427
):
@@ -432,7 +435,8 @@ def create_test_client(
432435

433436
# Parse the JSON response
434437
assert len(tool_result.content) == 1
435-
headers_data = json.loads(tool_result.content[0].text if tool_result.content[0].type == "text" else "{}")
438+
content_item = tool_result.content[0]
439+
headers_data = json.loads(content_item.text if content_item.type == "text" else "{}")
436440

437441
# Verify headers were propagated
438442
assert headers_data.get("authorization") == "Bearer test-token"

0 commit comments

Comments
 (0)