@@ -369,9 +369,11 @@ async def _handle_post_request(self, scope: Scope, request: Request, receive: Re
369369 return
370370
371371 # Check if this is an initialization request
372- is_initialization_request = isinstance (message .root , JSONRPCRequest ) and message .root .method == "initialize" # pragma: no cover
372+ is_initialization_request = (
373+ isinstance (message .root , JSONRPCRequest ) and message .root .method == "initialize"
374+ ) # pragma: no cover
373375
374- if is_initialization_request : # pragma: no cover
376+ if is_initialization_request : # pragma: no cover
375377 # Check if the server already has an established session
376378 if self .mcp_session_id :
377379 # Check if request has a session ID
@@ -385,11 +387,11 @@ async def _handle_post_request(self, scope: Scope, request: Request, receive: Re
385387 )
386388 await response (scope , receive , send )
387389 return
388- elif not await self ._validate_request_headers (request , send ): # pragma: no cover
390+ elif not await self ._validate_request_headers (request , send ): # pragma: no cover
389391 return
390392
391393 # For notifications and responses only, return 202 Accepted
392- if not isinstance (message .root , JSONRPCRequest ): # pragma: no cover
394+ if not isinstance (message .root , JSONRPCRequest ): # pragma: no cover
393395 # Create response object and send it
394396 response = self ._create_json_response (
395397 None ,
@@ -405,10 +407,10 @@ async def _handle_post_request(self, scope: Scope, request: Request, receive: Re
405407 return
406408
407409 # Extract the request ID outside the try block for proper scope
408- request_id = str (message .root .id ) # pragma: no cover
410+ request_id = str (message .root .id ) # pragma: no cover
409411 # Register this stream for the request ID
410- self ._request_streams [request_id ] = anyio .create_memory_object_stream [EventMessage ](0 ) # pragma: no cover
411- request_stream_reader = self ._request_streams [request_id ][1 ] # pragma: no cover
412+ self ._request_streams [request_id ] = anyio .create_memory_object_stream [EventMessage ](0 ) # pragma: no cover
413+ request_stream_reader = self ._request_streams [request_id ][1 ] # pragma: no cover
412414
413415 if self .is_json_response_enabled : # pragma: no cover
414416 # Process the message
@@ -453,7 +455,7 @@ async def _handle_post_request(self, scope: Scope, request: Request, receive: Re
453455 await response (scope , receive , send )
454456 finally :
455457 await self ._clean_up_memory_streams (request_id )
456- else : # pragma: no cover
458+ else : # pragma: no cover
457459 # Create SSE stream
458460 sse_stream_writer , sse_stream_reader = anyio .create_memory_object_stream [dict [str , str ]](0 )
459461
@@ -509,7 +511,7 @@ async def sse_writer():
509511 await sse_stream_reader .aclose ()
510512 await self ._clean_up_memory_streams (request_id )
511513
512- except Exception as err : # pragma: no cover
514+ except Exception as err : # pragma: no cover
513515 logger .exception ("Error handling POST request" )
514516 response = self ._create_error_response (
515517 f"Error handling POST request: { err } " ,
0 commit comments