Skip to content

Commit 0d513a6

Browse files
committed
Fix CI: pyright type errors in test and strict-no-cover false positive
1 parent 3179143 commit 0d513a6

File tree

2 files changed

+5
-5
lines changed

2 files changed

+5
-5
lines changed

src/mcp/server/streamable_http.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -849,7 +849,7 @@ async def _validate_protocol_version(self, request: Request, send: Send) -> bool
849849
protocol_version = request.headers.get(MCP_PROTOCOL_VERSION_HEADER)
850850

851851
# If no protocol version provided, assume default version
852-
if protocol_version is None: # pragma: no cover
852+
if protocol_version is None: # pragma: lax no cover
853853
protocol_version = DEFAULT_NEGOTIATED_VERSION
854854

855855
# Check if the protocol version is supported

tests/server/test_streamable_http_manager.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88
import anyio
99
import httpx
1010
import pytest
11-
from starlette.types import Message
11+
from starlette.types import ASGIApp, Message, Receive, Scope, Send
1212

1313
from mcp import Client, types
1414
from mcp.client.streamable_http import streamable_http_client
@@ -563,15 +563,15 @@ class SSECloseTracker:
563563
"upstream prematurely closed connection while reading upstream".
564564
"""
565565

566-
def __init__(self, app: StreamableHTTPASGIApp) -> None:
566+
def __init__(self, app: ASGIApp) -> None:
567567
self.app = app
568568
self.sse_streams_opened = 0
569569
self.sse_streams_closed_cleanly = 0
570570

571-
async def __call__(self, scope: dict[str, Any], receive: Any, send: Any) -> None:
571+
async def __call__(self, scope: Scope, receive: Receive, send: Send) -> None:
572572
is_sse = False
573573

574-
async def tracking_send(message: dict[str, Any]) -> None:
574+
async def tracking_send(message: Message) -> None:
575575
nonlocal is_sse
576576
if message["type"] == "http.response.start":
577577
for name, value in message.get("headers", []):

0 commit comments

Comments
 (0)