add response is not success throw exception

Author: junjie.miao

utils/mcp_client.py

@@ -110,10 +110,11 @@ def send_message(self, data: dict):
             json=data,
             headers={'Content-Type': 'application/json', 'trace-id': data["id"] if "id" in data else ""},
             timeout=self.timeout,
-            follow_redirects=True
+            follow_redirects=True,
         )
         response.raise_for_status()
-        logging.debug(f"{self.name} - Client message sent successfully: {response.status_code}")
+        if not response.is_success:
+            raise ValueError(f"{self.name} - MCP Server response: {response.status_code} {response.reason_phrase}")
         if "id" in data:
             message_id = data["id"]
             while True:
@@ -233,8 +234,10 @@ def send_message(self, data: dict):
             json=data,
             headers=headers,
             timeout=self.timeout,
-            follow_redirects=True
+            follow_redirects=True,
         )
+        if not response.is_success:
+            raise ValueError(f"{self.name} - MCP Server response: {response.status_code} {response.reason_phrase}")
         self.session_id = response.headers.get("mcp-session-id", None)
         content_type = response.headers.get("content-type", "None")
         if content_type == "text/event-stream":