From cefa0bb80c469f8dcdafc7c542b999df2b8c700a Mon Sep 17 00:00:00 2001 From: ulleo Date: Mon, 22 Sep 2025 18:34:53 +0800 Subject: [PATCH] feat: MCP mcp/mcp_question support param to disable streaming --- backend/common/core/response_middleware.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/backend/common/core/response_middleware.py b/backend/common/core/response_middleware.py index 69c49e54..c60a959d 100644 --- a/backend/common/core/response_middleware.py +++ b/backend/common/core/response_middleware.py @@ -17,16 +17,17 @@ class ResponseMiddleware(BaseHTTPMiddleware): response = await call_next(request) direct_paths = [ - f"{settings.API_V1_STR}/openapi.json", f"{settings.API_V1_STR}/mcp/mcp_question", f"{settings.API_V1_STR}/mcp/mcp_assistant" ] route = request.scope.get("route") # 获取定义的路径模式,例如 '/items/{item_id}' - path_pattern = route.path_format + path_pattern = '' if not route else route.path_format - if isinstance(response, JSONResponse) or path_pattern in direct_paths: + if (isinstance(response, JSONResponse) + or request.url.path == f"{settings.API_V1_STR}/openapi.json" + or path_pattern in direct_paths): return response if response.status_code != 200: return response