fix json response

Author: notsobad

src/mcp/client/streamable_http.py

@@ -11,6 +11,7 @@
 from contextlib import asynccontextmanager
 from dataclasses import dataclass
 from datetime import timedelta
+import json
 
 import anyio
 import httpx
@@ -299,14 +300,22 @@ async def _handle_json_response(
         """Handle JSON response from the server."""
         try:
             content = await response.aread()
-            message = JSONRPCMessage.model_validate_json(content)
 
-            # Extract protocol version from initialization response
-            if is_initialization:
-                self._maybe_extract_protocol_version_from_message(message)
+            # Parse JSON first to determine structure
+            data = json.loads(content)
+
+            if isinstance(data, list):
+                messages = [JSONRPCMessage.model_validate(item) for item in data]
+            else:
+                message = JSONRPCMessage.model_validate(data)
+                messages = [message]
+
+            for message in messages:
+                if is_initialization:
+                    self._maybe_extract_protocol_version_from_message(message)
 
-            session_message = SessionMessage(message)
-            await read_stream_writer.send(session_message)
+                session_message = SessionMessage(message)
+                await read_stream_writer.send(session_message)
         except Exception as exc:
             logger.error(f"Error parsing JSON response: {exc}")
             await read_stream_writer.send(exc)