catch errors in _handle_post_request

Author: gyang-xai

src/mcp/client/streamable_http.py

@@ -265,41 +265,47 @@ async def _handle_post_request(self, ctx: RequestContext) -> None:
         message = ctx.session_message.message
         is_initialization = self._is_initialization_request(message)
 
-        async with ctx.client.stream(
-            "POST",
-            self.url,
-            json=message.model_dump(by_alias=True, mode="json", exclude_none=True),
-            headers=headers,
-        ) as response:
-            if response.status_code == 202:
-                logger.debug("Received 202 Accepted")
-                return
+        try:
+            async with ctx.client.stream(
+                "POST",
+                self.url,
+                json=message.model_dump(by_alias=True, mode="json", exclude_none=True),
+                headers=headers,
+            ) as response:
+                if response.status_code == 202:
+                    logger.debug("Received 202 Accepted")
+                    return
 
-            if response.status_code == 404:
-                if isinstance(message.root, JSONRPCRequest):
-                    await self._send_session_terminated_error(
-                        ctx.read_stream_writer,
-                        message.root.id,
-                    )
-                return
+                if response.status_code == 404:
+                    if isinstance(message.root, JSONRPCRequest):
+                        await self._send_session_terminated_error(
+                            ctx.read_stream_writer,
+                            message.root.id,
+                        )
+                    return
+
+                response.raise_for_status()
+                if is_initialization:
+                    self._maybe_extract_session_id_from_response(response)
 
-            response.raise_for_status()
+                # Per https://modelcontextprotocol.io/specification/2025-06-18/basic#notifications:
+                # The server MUST NOT send a response to notifications.
+                if isinstance(message.root, JSONRPCRequest):
+                    content_type = response.headers.get(CONTENT_TYPE, "").lower()
+                    if content_type.startswith(JSON):
+                        await self._handle_json_response(response, ctx.read_stream_writer, is_initialization)
+                    elif content_type.startswith(SSE):
+                        await self._handle_sse_response(response, ctx, is_initialization)
+                    else:
+                        await self._handle_unexpected_content_type(
+                            content_type,
+                            ctx.read_stream_writer,
+                        )
+        except Exception as exc:
             if is_initialization:
-                self._maybe_extract_session_id_from_response(response)
-
-            # Per https://modelcontextprotocol.io/specification/2025-06-18/basic#notifications:
-            # The server MUST NOT send a response to notifications.
-            if isinstance(message.root, JSONRPCRequest):
-                content_type = response.headers.get(CONTENT_TYPE, "").lower()
-                if content_type.startswith(JSON):
-                    await self._handle_json_response(response, ctx.read_stream_writer, is_initialization)
-                elif content_type.startswith(SSE):
-                    await self._handle_sse_response(response, ctx, is_initialization)
-                else:
-                    await self._handle_unexpected_content_type(
-                        content_type,
-                        ctx.read_stream_writer,
-                    )
+                raise exc
+            else:
+                await self._send_error_response(ctx, exc)
 
     async def _handle_json_response(
         self,