@@ -173,9 +173,7 @@ async def _handle_stateless_request(
173173 )
174174
175175 # Start server in a new task
176- async def run_stateless_server (
177- * , task_status : TaskStatus [None ] = anyio .TASK_STATUS_IGNORED
178- ):
176+ async def run_stateless_server (* , task_status : TaskStatus [None ] = anyio .TASK_STATUS_IGNORED ):
179177 async with http_transport .connect () as streams :
180178 read_stream , write_stream = streams
181179 task_status .started ()
@@ -218,10 +216,7 @@ async def _handle_stateful_request(
218216 request_mcp_session_id = request .headers .get (MCP_SESSION_ID_HEADER )
219217
220218 # Existing session case
221- if (
222- request_mcp_session_id is not None
223- and request_mcp_session_id in self ._server_instances
224- ): # pragma: no cover
219+ if request_mcp_session_id is not None and request_mcp_session_id in self ._server_instances : # pragma: no cover
225220 transport = self ._server_instances [request_mcp_session_id ]
226221 logger .debug ("Session already exists, handling request directly" )
227222 await transport .handle_request (scope , receive , send )
@@ -245,9 +240,7 @@ async def _handle_stateful_request(
245240 logger .info (f"Created new transport with session ID: { new_session_id } " )
246241
247242 # Define the server runner
248- async def run_server (
249- * , task_status : TaskStatus [None ] = anyio .TASK_STATUS_IGNORED
250- ) -> None :
243+ async def run_server (* , task_status : TaskStatus [None ] = anyio .TASK_STATUS_IGNORED ) -> None :
251244 async with http_transport .connect () as streams :
252245 read_stream , write_stream = streams
253246 task_status .started ()
@@ -267,18 +260,15 @@ async def run_server(
267260 # Only remove from instances if not terminated
268261 if ( # pragma: no branch
269262 http_transport .mcp_session_id
270- and http_transport .mcp_session_id
271- in self ._server_instances
263+ and http_transport .mcp_session_id in self ._server_instances
272264 and not http_transport .is_terminated
273265 ):
274266 logger .info (
275267 "Cleaning up crashed session "
276268 f"{ http_transport .mcp_session_id } from "
277269 "active instances."
278270 )
279- del self ._server_instances [
280- http_transport .mcp_session_id
281- ]
271+ del self ._server_instances [http_transport .mcp_session_id ]
282272
283273 # Assert task group is not None for type checking
284274 assert self ._task_group is not None
0 commit comments