tiennm99/DocsGPT
1
0 Fork 0
mirror of https://github.com/tiennm99/DocsGPT.git synced 2026-10-04 16:13:23 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: clean error

Browse Source
This commit is contained in:
Alex committed 2026-04-03 18:08:38 +01:00
1 parent 2652f8a5b0
commit 3fc81ac5d8
2 files changed
+32 -10
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

application/api/user/tools/mcp.py
+28 -6
View File
@@ -123,18 +123,39 @@ class TestMCPServerConfig(Resource):
result = mcp_tool.test_connection()
if result.get("requires_oauth"):
return make_response(jsonify(result), 200)
safe_result = {
k: v
for k, v in result.items()
if k in ("success", "requires_oauth", "auth_url")
}
return make_response(jsonify(safe_result), 200)
if not result.get("success") and "message" in result:
if not result.get("success"):
current_app.logger.error(
f"MCP connection test failed: {result.get('message')}"
)
result["message"] = "Connection test failed"
return make_response(
jsonify(
{
"success": False,
"message": "Connection test failed",
"tools_count": 0,
}
),
200,
)
return make_response(jsonify(result), 200)
safe_result = {
"success": True,
"message": result.get("message", "Connection successful"),
"tools_count": result.get("tools_count", 0),
"tools": result.get("tools", []),
}
return make_response(jsonify(safe_result), 200)
except ValueError as e:
current_app.logger.warning(f"Invalid MCP server test request: {e}")
return make_response(
jsonify({"success": False, "error": str(e)}),
jsonify({"success": False, "error": "Invalid MCP server configuration"}),
400,
)
except Exception as e:
@@ -305,8 +326,9 @@ class MCPServerSave(Resource):
}
return make_response(jsonify(response_data), 200)
except ValueError as e:
current_app.logger.warning(f"Invalid MCP server save request: {e}")
return make_response(
jsonify({"success": False, "error": str(e)}),
jsonify({"success": False, "error": "Invalid MCP server configuration"}),
400,
)
except Exception as e:
application/api/user/tools/routes.py
+4 -4
View File
@@ -244,9 +244,9 @@ class CreateTool(Resource):
if server_url:
try:
validate_url(server_url)
except SSRFError as exc:
except SSRFError:
return make_response(
jsonify({"success": False, "message": f"Invalid server URL: {exc}"}),
jsonify({"success": False, "message": "Invalid server URL"}),
400,
)
tool_instance = tool_manager.tools.get(data["name"])
@@ -439,9 +439,9 @@ class UpdateToolConfig(Resource):
if server_url:
try:
validate_url(server_url)
except SSRFError as exc:
except SSRFError:
return make_response(
jsonify({"success": False, "message": f"Invalid server URL: {exc}"}),
jsonify({"success": False, "message": "Invalid server URL"}),
400,
)
tool_instance = tool_manager.tools.get(tool_name)