diff --git a/template/v2/dirs/etc/sagemaker-ui/sagemaker-mcp/amzn_aws_data_processing_remote_mcp_local_proxy-0.1.0-py3-none-any.whl b/template/v2/dirs/etc/sagemaker-ui/sagemaker-mcp/amzn_aws_data_processing_remote_mcp_local_proxy-0.1.0-py3-none-any.whl new file mode 100644 index 000000000..4654d2718 Binary files /dev/null and b/template/v2/dirs/etc/sagemaker-ui/sagemaker-mcp/amzn_aws_data_processing_remote_mcp_local_proxy-0.1.0-py3-none-any.whl differ diff --git a/template/v2/dirs/etc/sagemaker-ui/sagemaker-mcp/default.json b/template/v2/dirs/etc/sagemaker-ui/sagemaker-mcp/default.json index 07516d45f..bd1f67dde 100644 --- a/template/v2/dirs/etc/sagemaker-ui/sagemaker-mcp/default.json +++ b/template/v2/dirs/etc/sagemaker-ui/sagemaker-mcp/default.json @@ -9,6 +9,16 @@ "/etc/sagemaker-ui/sagemaker-mcp/smus-mcp.py" ], "env": {} + }, + "aws-dp-remote-mcp": { + "command": "aws-data-processing-mcp-proxy", + "args": [], + "env": { + "AWS_PROFILE_NAME": "remote_mcp_profile", + "SELECTED_SERVERS": "*" + }, + "disabled": false, + "timeout": 60000 } }, "tools": [ @@ -18,7 +28,8 @@ "listDirectory", "fileSearch", "executeBash", - "@smus-local-mcp" + "@smus-local-mcp", + "@aws-dp-remote-mcp" ], "allowedTools": [ "fsRead", diff --git a/template/v2/dirs/etc/sagemaker-ui/sagemaker-mcp/mcp.json b/template/v2/dirs/etc/sagemaker-ui/sagemaker-mcp/mcp.json index 587d172ca..77e1eccff 100644 --- a/template/v2/dirs/etc/sagemaker-ui/sagemaker-mcp/mcp.json +++ b/template/v2/dirs/etc/sagemaker-ui/sagemaker-mcp/mcp.json @@ -3,6 +3,16 @@ "smus-local-mcp": { "command": "python", "args": ["/etc/sagemaker-ui/sagemaker-mcp/smus-mcp.py"] + }, + "aws-dp-remote-mcp": { + "command": "aws-data-processing-mcp-proxy", + "args": [], + "env": { + "AWS_PROFILE_NAME": "remote_mcp_profile", + "SELECTED_SERVERS": "*" + }, + "disabled": false, + "timeout": 60000 } } } \ No newline at end of file diff --git a/template/v2/dirs/etc/sagemaker-ui/sagemaker_ui_post_startup.sh b/template/v2/dirs/etc/sagemaker-ui/sagemaker_ui_post_startup.sh index 06f768550..bb6f8a5f1 100644 --- a/template/v2/dirs/etc/sagemaker-ui/sagemaker_ui_post_startup.sh +++ b/template/v2/dirs/etc/sagemaker-ui/sagemaker_ui_post_startup.sh @@ -404,6 +404,11 @@ else echo "Warning: Source configuration file not found at $agents_source_file" fi +# Install AWS Data Processing MCP proxy wheel file +echo "Installing AWS Data Processing MCP proxy..." +pip install /etc/sagemaker-ui/sagemaker-mcp/amzn_aws_data_processing_remote_mcp_local_proxy-0.1.0-py3-none-any.whl +echo "Successfully installed AWS Data Processing MCP proxy" + # Generate sagemaker pysdk intelligent default config nohup python /etc/sagemaker/sm_pysdk_default_config.py & # Only run the following commands if SAGEMAKER_APP_TYPE_LOWERCASE is jupyterlab and domain is not in express mode diff --git a/template/v3/dirs/etc/sagemaker-ui/sagemaker-mcp/amzn_aws_data_processing_remote_mcp_local_proxy-0.1.0-py3-none-any.whl b/template/v3/dirs/etc/sagemaker-ui/sagemaker-mcp/amzn_aws_data_processing_remote_mcp_local_proxy-0.1.0-py3-none-any.whl new file mode 100644 index 000000000..4654d2718 Binary files /dev/null and b/template/v3/dirs/etc/sagemaker-ui/sagemaker-mcp/amzn_aws_data_processing_remote_mcp_local_proxy-0.1.0-py3-none-any.whl differ diff --git a/template/v3/dirs/etc/sagemaker-ui/sagemaker-mcp/default.json b/template/v3/dirs/etc/sagemaker-ui/sagemaker-mcp/default.json index 0b0bebfee..c8404ef93 100644 --- a/template/v3/dirs/etc/sagemaker-ui/sagemaker-mcp/default.json +++ b/template/v3/dirs/etc/sagemaker-ui/sagemaker-mcp/default.json @@ -9,6 +9,16 @@ "/etc/sagemaker-ui/sagemaker-mcp/smus-mcp.py" ], "env": {} + }, + "aws-dp-remote-mcp": { + "command": "aws-data-processing-mcp-proxy", + "args": [], + "env": { + "AWS_PROFILE_NAME": "remote_mcp_profile", + "SELECTED_SERVERS": "*" + }, + "disabled": false, + "timeout": 60000 } }, "tools": [ @@ -18,7 +28,8 @@ "listDirectory", "fileSearch", "executeBash", - "@smus-local-mcp" + "@smus-local-mcp", + "@aws-dp-remote-mcp" ], "allowedTools": [ "fsRead", diff --git a/template/v3/dirs/etc/sagemaker-ui/sagemaker-mcp/mcp.json b/template/v3/dirs/etc/sagemaker-ui/sagemaker-mcp/mcp.json index 587d172ca..77e1eccff 100644 --- a/template/v3/dirs/etc/sagemaker-ui/sagemaker-mcp/mcp.json +++ b/template/v3/dirs/etc/sagemaker-ui/sagemaker-mcp/mcp.json @@ -3,6 +3,16 @@ "smus-local-mcp": { "command": "python", "args": ["/etc/sagemaker-ui/sagemaker-mcp/smus-mcp.py"] + }, + "aws-dp-remote-mcp": { + "command": "aws-data-processing-mcp-proxy", + "args": [], + "env": { + "AWS_PROFILE_NAME": "remote_mcp_profile", + "SELECTED_SERVERS": "*" + }, + "disabled": false, + "timeout": 60000 } } } \ No newline at end of file diff --git a/template/v3/dirs/etc/sagemaker-ui/sagemaker_ui_post_startup.sh b/template/v3/dirs/etc/sagemaker-ui/sagemaker_ui_post_startup.sh index 3cec0b226..cfe2c7a1d 100755 --- a/template/v3/dirs/etc/sagemaker-ui/sagemaker_ui_post_startup.sh +++ b/template/v3/dirs/etc/sagemaker-ui/sagemaker_ui_post_startup.sh @@ -405,6 +405,11 @@ else echo "Warning: Source configuration file not found at $agents_source_file" fi +# Install AWS Data Processing MCP proxy wheel file +echo "Installing AWS Data Processing MCP proxy..." +pip install /etc/sagemaker-ui/sagemaker-mcp/amzn_aws_data_processing_remote_mcp_local_proxy-0.1.0-py3-none-any.whl +echo "Successfully installed AWS Data Processing MCP proxy" + # Generate sagemaker pysdk intelligent default config nohup python /etc/sagemaker/sm_pysdk_default_config.py & # Only run the following commands if SAGEMAKER_APP_TYPE_LOWERCASE is jupyterlab and domain is not in express mode