fix: the mcp server disable status

pull/22036/head
Novice 11 months ago
parent ef35e9febb
commit ecc4ad0770

@ -2,6 +2,7 @@ from flask_restful import Resource, reqparse
from pydantic import ValidationError from pydantic import ValidationError
from werkzeug.exceptions import NotFound from werkzeug.exceptions import NotFound
from controllers.console.app.mcp_server import AppMCPServerStatus
from controllers.mcp import api from controllers.mcp import api
from controllers.web.error import ( from controllers.web.error import (
AppUnavailableError, AppUnavailableError,
@ -33,6 +34,8 @@ class MCPAppApi(Resource):
server = db.session.query(AppMCPServer).filter(AppMCPServer.server_code == server_code).first() server = db.session.query(AppMCPServer).filter(AppMCPServer.server_code == server_code).first()
if not server: if not server:
raise NotFound("Server Not Found") raise NotFound("Server Not Found")
if server.status != AppMCPServerStatus.ACTIVE:
raise NotFound("Server is not active")
app = db.session.query(App).filter(App.id == server.app_id).first() app = db.session.query(App).filter(App.id == server.app_id).first()
if not app: if not app:
raise NotFound("App Not Found") raise NotFound("App Not Found")

@ -322,7 +322,7 @@ class StreamableHTTPTransport:
jsonrpc_error = JSONRPCError( jsonrpc_error = JSONRPCError(
jsonrpc="2.0", jsonrpc="2.0",
id=request_id, id=request_id,
error=ErrorData(code=32600, message="Session terminated"), error=ErrorData(code=32600, message="Session terminated by server"),
) )
session_message = SessionMessage(JSONRPCMessage(jsonrpc_error)) session_message = SessionMessage(JSONRPCMessage(jsonrpc_error))
server_to_client_queue.put(session_message) server_to_client_queue.put(session_message)

Loading…
Cancel
Save