adai/MemoryBear
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #737 from SuanmoSuanyangTechnology/fix/Timebomb_029

Browse Source 
fix(mcp)
This commit is contained in:
Mark
2026-03-30 17:32:22 +08:00
committed by GitHub
parent ea8db7cd90 9d91453200
commit 8b44b2cd61
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
api/app/core/tools/mcp/client.py
View File

@@ -99,7 +99,7 @@ class SimpleMCPClient:
# 建立 SSE 连接 # 建立 SSE 连接
response = await self._session.get(self.server_url) response = await self._session.get(self.server_url)
if response.status != 200: if response.status not in (200, 202):
error_text = await response.text() error_text = await response.text()
raise MCPConnectionError(f"SSE 连接失败 {response.status}: {error_text}") raise MCPConnectionError(f"SSE 连接失败 {response.status}: {error_text}")
@@ -190,7 +190,9 @@ class SimpleMCPClient:
try: try:
async with self._session.post(self._endpoint_url, json=request) as response: async with self._session.post(self._endpoint_url, json=request) as response:
if response.status != 200: # MCP SSE 协议：POST 请求返回 200 或 202 均为正常
# 202 Accepted 表示请求已接受，结果通过 SSE 流异步返回
if response.status not in (200, 202):
error_text = await response.text() error_text = await response.text()
raise MCPConnectionError(f"请求失败 {response.status}: {error_text}") raise MCPConnectionError(f"请求失败 {response.status}: {error_text}")
@@ -205,7 +207,7 @@ class SimpleMCPClient:
raise MCPConnectionError("endpoint URL 未初始化") raise MCPConnectionError("endpoint URL 未初始化")
async with self._session.post(self._endpoint_url, json=notification) as response: async with self._session.post(self._endpoint_url, json=notification) as response:
if response.status != 200: if response.status not in (200, 202):
logger.warning(f"通知发送失败: {response.status}") logger.warning(f"通知发送失败: {response.status}")
async def _initialize_modelscope_session(self): async def _initialize_modelscope_session(self):