comments fix

Author: akash329d

src/mcp/server/message_queue/base.py

@@ -16,7 +16,7 @@
 class MessageWrapper(BaseModel):
     message_id: str
     payload: str
-    
+
     def get_json_rpc_message(self) -> types.JSONRPCMessage | ValidationError:
         """Parse the payload into a JSONRPCMessage or return ValidationError."""
         try:
@@ -48,9 +48,12 @@ async def publish_message(
         ...
 
     async def publish_message_sync(
-        self, session_id: UUID, message: types.JSONRPCMessage | str, timeout: float = 30.0
+        self,
+        session_id: UUID,
+        message: types.JSONRPCMessage | str,
+        timeout: float = 120.0,
     ) -> bool:
-        """Publish a message for the specified session and wait for consumption confirmation.
+        """Publish a message for the specified session and wait for confirmation.
 
         This method blocks until the message has been fully consumed by the subscriber,
         or until the timeout is reached.
@@ -121,12 +124,15 @@ async def publish_message(
 
         logger.debug(f"Message dispatched to session {session_id}")
         return True
-        
+
     async def publish_message_sync(
-        self, session_id: UUID, message: types.JSONRPCMessage | str, timeout: float = 30.0
+        self,
+        session_id: UUID,
+        message: types.JSONRPCMessage | str,
+        timeout: float = 30.0,
     ) -> bool:
         """Publish a message for the specified session and wait for consumption.
-        
+
         For InMemoryMessageDispatch, this is the same as publish_message since
         the callback is executed synchronously.
         """

src/mcp/server/message_queue/redis.py

@@ -130,10 +130,8 @@ async def _handle_ack_message(self, channel: str, data: str) -> None:
         if not channel.startswith(ack_prefix):
             return
 
-        # Validate channel format exactly matches our expected format
         session_hex = channel[len(ack_prefix) :]
         try:
-            # Validate this is a valid UUID hex and channel has correct format
             session_id = UUID(hex=session_hex)
             expected_channel = self._ack_channel(session_id)
             if channel != expected_channel:

src/mcp/server/sse.py

@@ -174,7 +174,6 @@ async def handle_post_message(
             logger.debug(f"Validated client message: {message}")
         except ValidationError as err:
             logger.error(f"Failed to parse message: {err}")
-            # Still publish the invalid message, but using synchronized version
             response = Response("Could not parse message", status_code=400)
             await response(scope, receive, send)
             # Pass raw JSON string; receiver will recreate identical ValidationError
@@ -184,7 +183,7 @@ async def handle_post_message(
 
         logger.debug(f"Publishing message for session {session_id}: {message}")
 
-        # Use sync publish to block until the message is processed
+        # Use sync publish, block POST response until the message is processed
         result = await self._message_dispatch.publish_message_sync(session_id, message)
 
         if result: