Add pyright strict mode on the whole project (#1254)

examples/servers/simple-streamablehttp/mcp_simple_streamablehttp/event_store.py

@@ -10,13 +10,7 @@ from collections import deque
 from dataclasses import dataclass
 from uuid import uuid4
 
-from mcp.server.streamable_http import (
-    EventCallback,
-    EventId,
-    EventMessage,
-    EventStore,
-    StreamId,
-)
+from mcp.server.streamable_http import EventCallback, EventId, EventMessage, EventStore, StreamId
 from mcp.types import JSONRPCMessage
 
 logger = logging.getLogger(__name__)
@@ -54,14 +48,10 @@ class InMemoryEventStore(EventStore):
         # event_id -> EventEntry for quick lookup
         self.event_index: dict[EventId, EventEntry] = {}
 
-    async def store_event(
-        self, stream_id: StreamId, message: JSONRPCMessage
-    ) -> EventId:
+    async def store_event(self, stream_id: StreamId, message: JSONRPCMessage) -> EventId:
         """Stores an event with a generated event ID."""
         event_id = str(uuid4())
-        event_entry = EventEntry(
-            event_id=event_id, stream_id=stream_id, message=message
-        )
+        event_entry = EventEntry(event_id=event_id, stream_id=stream_id, message=message)
 
         # Get or create deque for this stream
         if stream_id not in self.streams:

examples/servers/simple-streamablehttp/mcp_simple_streamablehttp/server.py

@@ -1,6 +1,7 @@
 import contextlib
 import logging
 from collections.abc import AsyncIterator
+from typing import Any
 
 import anyio
 import click
@@ -45,7 +46,7 @@ def main(
     app = Server("mcp-streamable-http-demo")
 
     @app.call_tool()
-    async def call_tool(name: str, arguments: dict) -> list[types.ContentBlock]:
+    async def call_tool(name: str, arguments: dict[str, Any]) -> list[types.ContentBlock]:
         ctx = app.request_context
         interval = arguments.get("interval", 1.0)
         count = arguments.get("count", 5)
@@ -54,10 +55,7 @@ def main(
         # Send the specified number of notifications with the given interval
         for i in range(count):
             # Include more detailed message for resumability demonstration
-            notification_msg = (
-                f"[{i + 1}/{count}] Event from '{caller}' - "
-                f"Use Last-Event-ID to resume if disconnected"
-            )
+            notification_msg = f"[{i + 1}/{count}] Event from '{caller}' - Use Last-Event-ID to resume if disconnected"
             await ctx.session.send_log_message(
                 level="info",
                 data=notification_msg,
@@ -79,10 +77,7 @@ def main(
         return [
             types.TextContent(
                 type="text",
-                text=(
-                    f"Sent {count} notifications with {interval}s interval"
-                    f" for caller: {caller}"
-                ),
+                text=(f"Sent {count} notifications with {interval}s interval for caller: {caller}"),
             )
         ]
 
@@ -91,10 +86,7 @@ def main(
         return [
             types.Tool(
                 name="start-notification-stream",
-                description=(
-                    "Sends a stream of notifications with configurable count"
-                    " and interval"
-                ),
+                description=("Sends a stream of notifications with configurable count and interval"),
                 inputSchema={
                     "type": "object",
                     "required": ["interval", "count", "caller"],
@@ -109,9 +101,7 @@ def main(
                         },
                         "caller": {
                             "type": "string",
-                            "description": (
-                                "Identifier of the caller to include in notifications"
-                            ),
+                            "description": ("Identifier of the caller to include in notifications"),
                         },
                     },
                 },
@@ -136,9 +126,7 @@ def main(
     )
 
     # ASGI handler for streamable HTTP connections
-    async def handle_streamable_http(
-        scope: Scope, receive: Receive, send: Send
-    ) -> None:
+    async def handle_streamable_http(scope: Scope, receive: Receive, send: Send) -> None:
         await session_manager.handle_request(scope, receive, send)
 
     @contextlib.asynccontextmanager

examples/servers/simple-streamablehttp/pyproject.toml

@@ -29,8 +29,8 @@ select = ["E", "F", "I"]
 ignore = []
 
 [tool.ruff]
-line-length = 88
+line-length = 120
 target-version = "py310"
 
 [tool.uv]
-dev-dependencies = ["pyright>=1.1.378", "pytest>=8.3.3", "ruff>=0.6.9"]
+dev-dependencies = ["pyright>=1.1.378", "pytest>=8.3.3", "ruff>=0.6.9"]