From 58b1a85aebf27d2e2836229b79f554f2b66ff601 Mon Sep 17 00:00:00 2001 From: Arunav Gupta Date: Thu, 25 Sep 2025 11:23:29 -0400 Subject: [PATCH] feat: add AWS Data Processing MCP server proxy --- ...emote_mcp_local_proxy-0.1.0-py3-none-any.whl | Bin 0 -> 18487 bytes .../etc/sagemaker-ui/sagemaker-mcp/default.json | 13 ++++++++++++- .../etc/sagemaker-ui/sagemaker-mcp/mcp.json | 10 ++++++++++ .../sagemaker-ui/sagemaker_ui_post_startup.sh | 5 +++++ ...emote_mcp_local_proxy-0.1.0-py3-none-any.whl | Bin 0 -> 18487 bytes .../etc/sagemaker-ui/sagemaker-mcp/default.json | 13 ++++++++++++- .../etc/sagemaker-ui/sagemaker-mcp/mcp.json | 10 ++++++++++ .../sagemaker-ui/sagemaker_ui_post_startup.sh | 5 +++++ 8 files changed, 54 insertions(+), 2 deletions(-) create mode 100644 template/v2/dirs/etc/sagemaker-ui/sagemaker-mcp/amzn_aws_data_processing_remote_mcp_local_proxy-0.1.0-py3-none-any.whl create mode 100644 template/v3/dirs/etc/sagemaker-ui/sagemaker-mcp/amzn_aws_data_processing_remote_mcp_local_proxy-0.1.0-py3-none-any.whl 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 0000000000000000000000000000000000000000..4654d27189491ffe7fd389e76c27d5f112602f2b GIT binary patch literal 18487 zcmdU1OKc-sTFx*JBugyN49sSPj;hsM=U%%KJI5^gCjvW|=KAsxJ8TWkKbqs@5nYOH#^kpq~n8I}UJO5Mq z)vqrs(C6U8FwG;XZ~6gIBxnH%_Gs*S0UNp}C!jf}{&f9Z_;nBj)fAr)hvjqup#BH(N$?zj1i5v)?fqJM~T>w|HCR_Q_er zuqJ-sjtp}W3_`qZ`{cM%7td>WoXg!~bwSv~6`Lkg7kXr%o8FKUe48-Pcv)*Yo z)Lyq0@K3dz*aopFOsXd|*$P~5$^w^h$EVHVrZoWlZzkZ1**^2_6NmTMxsA;gyPE6q zKC__1Flh|Z{tH#l54hQb6<7CNhZh)WgZldEr>Ezpcj7N@3T@OhP5iD47|e8-FfG02 zPaUf+U?tEAK(Hj+G6QUfo)c?p04Jf5ek;tIY+I5*?AVSqob)(;hurvLCCw+=HW?`< z8=D^VCxBw)!KmwjRnuqZoWU=g*;wyRJm#J|i_zkc8vy9_Y@*2=GO#xpfl8lHk4<0v zo?%PC2-$~4C;>3b9Z#9tXHrZA!aQ^B9vitmyFZ0HV!;3u0W8St*;XI`NwFJhQI)-o zNYwi7aOj>xgL}3IP4Tr&*6}915YHe3awh=>B$XZccfM3!)|U&Q$Z@c+x+eYtoxQ3W zIrr>2qpyPS4#y(gULs$33#E5C59`x z=l4k9;a)Qsh1#NCL2Wh(iZmBI_gKxc_&8A0$$7BQO_K&dhsjV4r8zL5x0+l5iD>3{ zJXHF&^xAiw0t@&>umrE<1Vvx?Z|zFR5>$jBU@K(or2VmGX7K2fF@hrxO+_kIY9C)| z3ZC`_h{`ONa*&Qpx1GK#-6b)k31uP)%xIl7Ch4R@YB$+^U&A&FYf~yjPA(szFR-^) zl0LXY!Vu{z>7rbiHB&J!N#1*@5*{UE;^cn`AvrhO;s!Z%$U>vE0$Z-KiXwuyMp~eJ zDkfmCr!hYC`Mk-+03JHY~_N6dlqYAuf{D)M{7Ccn6guEBR zbs%|_2vUp+^`eFJW1gEfMQq84Zj!Uh$L^kA2tU#kzbGF-ct{vHK~NemP~`jCBgm4Z z?E+|ms!jHmUxm6&%kIG^%gC#QkfKp;VBZ`LyQXy-$1X9+82Kk;5>lNB*o;i%yl`i$ zF@xmL68zEP3(L{2^dZ!$jdkYv6ej+&TVPdW^zYPlmMYltVX=l8>+MaA?Y z30DwCGE~`cRO{uzrxGcNUxk}%3cvK`V`UYeI8qP*p}i{2HdM-9RY;R5c)Uu5&}=0x z>xIoQ6~gwaq)1dz)p;RSL83rAQ+2=teX5BvXzq3+JUcV1V#AobMwK+OERX5aOKbqB-k6ijK{LD@2#_zDb>{!;i35 z%K4;4nlxWBd-}P7;?1LEj(o3yac^jZ0^mfU!%2h%a{-D7*ON0i@vBuz+3O*9=Gl(?vpkgHO{I!5|KT@!yYmVGp= zLoCUV()-_9q89{#7?@E`WOHCC%^XBUs_n8&*R$&OQ~0sDu7Eg9nnPkzFhi)Fmx8XadU7CPtqy0d%d~cPA^}>R^xsD(Bu5XekNwvTV7l zqfv13U@p+oJAhv0t}>ePcey7o{wlYyK%eHVG$q?1zDcXS>jv(MY)OmMCsHNo_^tco zf)KJ8XqRf-3oF9^@aU(#LoG;+dl6Je=Byz#8Z#q#YHP;jxat*%Nm(BODGWJvEkz2G+DlShWwk9s0hnot8Bt>EE_i>~2d zRHgoTrMSIUs~6kbwesqkmPnIT724z7$qCBn9UKnX=p0@~{w`);&`M4S{h{hx_2nwW zSVhrDn3LrwktWVcumTLIC6UpKVYv#U@=JO!VglpQ_gDtxevLkwu1W|>VArvyWp+)E z)-!|>Y9CP8Q9u$^^*welCP&FiNaA^bRz1ZJq`^J%{U&-i!UN;1!uDK;_LiNQo=vAs zaSMTJH52Vo-$!?rwlj0UQLERR?Y8lx`NG&~B#_fOp95Jie_gv-Z#6qcynJRY&I3_# z1@0-IbFR_`QXH%por5P$9Ir^(QVtX`U4AFilpvs=iPkSff|wyP3QEADjU)!tK-g8x zuzuJ&c(k+IH1=zIIFw0}u*HK}ISWTc1<<<%j@&9Ea}lxA;*1|9q!d?6B#4xSXh(Fo zI!n`BQW85sfKt1kCM^;c%}ecedizuJ!X1@U67)y2_)rNUDgSqIC$xJmgFq5(?_0u{ z1evsX;vi`nRVMaJZ_~LnnqbLwqN|&Hut@E2Be6WuR3gAB5hkYT@j%2p%}g1Mxh_sbh;XXQAcCQ*3}bT zJxN-91_A|EWFMCz$C+Iz4!u4}P5NXp32tihAssCULNnR%AK1U~{rWF1EYRoIZ=W4Q zHrU}fF&>!yU}$%vizg$~3cnL`aEdZFMn%u*95+m7!@uQdx(btNgE>^`b1JQFDdv94 zFKNIKjw{j{C^RS2U-?wJs#HcATNb6J(ORBwuF5rTq5QYb7dos7k_MdZp^F>Pc|kxgf23GA34erLv1)+?3^b~59V9SqAa znNgvpy~L5ZHo(c_w=at3=(0$N~Z) zB_hEOmX08ytqHDuEMXP*xP#+ouzgm91I3J@6afwcgg4L!!Dx67cV2M2K@5S{A~I@0 zCaEE15VQp6XC9}VSrrE)EIO(hjfLtgvf}5gI|T)samGpO2e5p9V4_aSV==Bw_$LY7 zAvuR9*_fz1{5+bbgLJeT^V%fn7dJF?zm-$bCA>vl%9>h#1c!bERk9#RAQpF?DjNF? zDu*bLC;%2`k;IIpTIY1;LIs&B8xqh%oK!UubUE+Z#iDadcLHX}6o6tZAv~q=F+kc5 zY}0!IBrz3mBjnt#ghg;7kx}AL0F~klH{S|1I9!{q4mZ(66q&>r zfRbdSymsAFB80UCB`E+23X_0h4B_3HD(eepvl#_`;+x#h)pwoP0d2mG#E9n!cZAv_ zE#bQ*vgR^uR+1Da@;tt%v?Llgp{Y^bc?ia~7bg)1io91Gy04brX3<|2vI3+09#YxHEfKmmIbT!-F-s7P8DTI<+=Ng& zF4G>BR}(qZ4a0`{LF8pboN1|tk;J2WU8aZtqVW{k9rYwcB#v06bO{b6T5U}{gvr1V zc_=QdA+pg|Tu!zeG{fTUj4Tscm|Ua-V4nu_+Wi;zNB4W3``h>T?zdki+$=#A&hdR~ zan%*6M+4K@`I0M~YJpOX9-TzT8^VM>#Fp~2?H;g!t!Ay$WHnZ=?H}y#)N8xAw(<0+ z+3v*q6ngP65^wq&YB1X`?bni2hUw!qrkL!bec0wxeSy@;JbJuUmj? zizFo6X@;(}i&gR!@|v}2@~hyzwp%~ir41JBgXvo5sMU;VOrCEPSaB})aCPs?H^cx6+}2-3Biy4BoowuIsyb+!*$J1=V;^nclQv({*~ zqUG~Ed{TMmji+}hHhifXbah5~ZutP&V(3__(YJ_zL`cOIfX_J-pQf?n-4cH;zC;|tS#AXpKpz%<{}URS*|@9 zjc^Q(w+qnjr`ne;7_{kn5;eUq#D!4`$r9S!>jTjWFSml={EX7n;t z{_`rJ>&uJT3G$Ex!zI1EHfLP+qA2zZMp@Xm)Fhl zYLX-JHlMD}!?mLe!|CF(8S9YJpIBFXopqpBQpwC-Ph1rzO+)t-J;%5UBX1@ntMdNs zH~-Mj9zf!_7fBFlOUWI7X z^e(&o8b7%>>z}rlj?2YE+%2`O;-SaSY<^xO(c+I#wnZu2A~-Ax`|*mY3@gSX#{{_8! zH{(0WX-dt+>HCfSH+A}ETz%?YzfHQ=mf`s9f4{Z)891fS-bdg#aPi(NsT;97k0NP` z+4o~JRQ=x9{;7=XS_|~~7@_PEw&k0BoPZ$pN-8r^!ss@dMz`r3o-cFFEuzTAEOX5;wKynNALe_TI( z(!PAXVg-}lUSW0PLpYwf>=>ftuZ@#u#$a+VUcPKvrM1@M{bj#vb=vDr=S}x%huTq=DyPO7wKUsO&Ac>DR$(W<=~Y`=It*xfpQc=q_^O5gEXPfFIgyHlQa z9~-5;qpeHB^b3{MkH++ZS0=5re6Q)o7&l$iQ@Lb(er-2^Ex;z zzuA3UDZd^*d}^LvIJL{uLZwoAkh`k`4s+O$fQJXQ%iY@V zZx03shP^tfSO@l~>lPlsrtTO#v(YCckK$&&-S?t+VKgRxJywh@5F{hk-7IikC uk5Sv-zB}sBw`Z|t)B70f-+u(G(x)Fj`6`Nsg@s?hfB*1RRE-bmum1=4?Sf_i literal 0 HcmV?d00001 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 0000000000000000000000000000000000000000..4654d27189491ffe7fd389e76c27d5f112602f2b GIT binary patch literal 18487 zcmdU1OKc-sTFx*JBugyN49sSPj;hsM=U%%KJI5^gCjvW|=KAsxJ8TWkKbqs@5nYOH#^kpq~n8I}UJO5Mq z)vqrs(C6U8FwG;XZ~6gIBxnH%_Gs*S0UNp}C!jf}{&f9Z_;nBj)fAr)hvjqup#BH(N$?zj1i5v)?fqJM~T>w|HCR_Q_er zuqJ-sjtp}W3_`qZ`{cM%7td>WoXg!~bwSv~6`Lkg7kXr%o8FKUe48-Pcv)*Yo z)Lyq0@K3dz*aopFOsXd|*$P~5$^w^h$EVHVrZoWlZzkZ1**^2_6NmTMxsA;gyPE6q zKC__1Flh|Z{tH#l54hQb6<7CNhZh)WgZldEr>Ezpcj7N@3T@OhP5iD47|e8-FfG02 zPaUf+U?tEAK(Hj+G6QUfo)c?p04Jf5ek;tIY+I5*?AVSqob)(;hurvLCCw+=HW?`< z8=D^VCxBw)!KmwjRnuqZoWU=g*;wyRJm#J|i_zkc8vy9_Y@*2=GO#xpfl8lHk4<0v zo?%PC2-$~4C;>3b9Z#9tXHrZA!aQ^B9vitmyFZ0HV!;3u0W8St*;XI`NwFJhQI)-o zNYwi7aOj>xgL}3IP4Tr&*6}915YHe3awh=>B$XZccfM3!)|U&Q$Z@c+x+eYtoxQ3W zIrr>2qpyPS4#y(gULs$33#E5C59`x z=l4k9;a)Qsh1#NCL2Wh(iZmBI_gKxc_&8A0$$7BQO_K&dhsjV4r8zL5x0+l5iD>3{ zJXHF&^xAiw0t@&>umrE<1Vvx?Z|zFR5>$jBU@K(or2VmGX7K2fF@hrxO+_kIY9C)| z3ZC`_h{`ONa*&Qpx1GK#-6b)k31uP)%xIl7Ch4R@YB$+^U&A&FYf~yjPA(szFR-^) zl0LXY!Vu{z>7rbiHB&J!N#1*@5*{UE;^cn`AvrhO;s!Z%$U>vE0$Z-KiXwuyMp~eJ zDkfmCr!hYC`Mk-+03JHY~_N6dlqYAuf{D)M{7Ccn6guEBR zbs%|_2vUp+^`eFJW1gEfMQq84Zj!Uh$L^kA2tU#kzbGF-ct{vHK~NemP~`jCBgm4Z z?E+|ms!jHmUxm6&%kIG^%gC#QkfKp;VBZ`LyQXy-$1X9+82Kk;5>lNB*o;i%yl`i$ zF@xmL68zEP3(L{2^dZ!$jdkYv6ej+&TVPdW^zYPlmMYltVX=l8>+MaA?Y z30DwCGE~`cRO{uzrxGcNUxk}%3cvK`V`UYeI8qP*p}i{2HdM-9RY;R5c)Uu5&}=0x z>xIoQ6~gwaq)1dz)p;RSL83rAQ+2=teX5BvXzq3+JUcV1V#AobMwK+OERX5aOKbqB-k6ijK{LD@2#_zDb>{!;i35 z%K4;4nlxWBd-}P7;?1LEj(o3yac^jZ0^mfU!%2h%a{-D7*ON0i@vBuz+3O*9=Gl(?vpkgHO{I!5|KT@!yYmVGp= zLoCUV()-_9q89{#7?@E`WOHCC%^XBUs_n8&*R$&OQ~0sDu7Eg9nnPkzFhi)Fmx8XadU7CPtqy0d%d~cPA^}>R^xsD(Bu5XekNwvTV7l zqfv13U@p+oJAhv0t}>ePcey7o{wlYyK%eHVG$q?1zDcXS>jv(MY)OmMCsHNo_^tco zf)KJ8XqRf-3oF9^@aU(#LoG;+dl6Je=Byz#8Z#q#YHP;jxat*%Nm(BODGWJvEkz2G+DlShWwk9s0hnot8Bt>EE_i>~2d zRHgoTrMSIUs~6kbwesqkmPnIT724z7$qCBn9UKnX=p0@~{w`);&`M4S{h{hx_2nwW zSVhrDn3LrwktWVcumTLIC6UpKVYv#U@=JO!VglpQ_gDtxevLkwu1W|>VArvyWp+)E z)-!|>Y9CP8Q9u$^^*welCP&FiNaA^bRz1ZJq`^J%{U&-i!UN;1!uDK;_LiNQo=vAs zaSMTJH52Vo-$!?rwlj0UQLERR?Y8lx`NG&~B#_fOp95Jie_gv-Z#6qcynJRY&I3_# z1@0-IbFR_`QXH%por5P$9Ir^(QVtX`U4AFilpvs=iPkSff|wyP3QEADjU)!tK-g8x zuzuJ&c(k+IH1=zIIFw0}u*HK}ISWTc1<<<%j@&9Ea}lxA;*1|9q!d?6B#4xSXh(Fo zI!n`BQW85sfKt1kCM^;c%}ecedizuJ!X1@U67)y2_)rNUDgSqIC$xJmgFq5(?_0u{ z1evsX;vi`nRVMaJZ_~LnnqbLwqN|&Hut@E2Be6WuR3gAB5hkYT@j%2p%}g1Mxh_sbh;XXQAcCQ*3}bT zJxN-91_A|EWFMCz$C+Iz4!u4}P5NXp32tihAssCULNnR%AK1U~{rWF1EYRoIZ=W4Q zHrU}fF&>!yU}$%vizg$~3cnL`aEdZFMn%u*95+m7!@uQdx(btNgE>^`b1JQFDdv94 zFKNIKjw{j{C^RS2U-?wJs#HcATNb6J(ORBwuF5rTq5QYb7dos7k_MdZp^F>Pc|kxgf23GA34erLv1)+?3^b~59V9SqAa znNgvpy~L5ZHo(c_w=at3=(0$N~Z) zB_hEOmX08ytqHDuEMXP*xP#+ouzgm91I3J@6afwcgg4L!!Dx67cV2M2K@5S{A~I@0 zCaEE15VQp6XC9}VSrrE)EIO(hjfLtgvf}5gI|T)samGpO2e5p9V4_aSV==Bw_$LY7 zAvuR9*_fz1{5+bbgLJeT^V%fn7dJF?zm-$bCA>vl%9>h#1c!bERk9#RAQpF?DjNF? zDu*bLC;%2`k;IIpTIY1;LIs&B8xqh%oK!UubUE+Z#iDadcLHX}6o6tZAv~q=F+kc5 zY}0!IBrz3mBjnt#ghg;7kx}AL0F~klH{S|1I9!{q4mZ(66q&>r zfRbdSymsAFB80UCB`E+23X_0h4B_3HD(eepvl#_`;+x#h)pwoP0d2mG#E9n!cZAv_ zE#bQ*vgR^uR+1Da@;tt%v?Llgp{Y^bc?ia~7bg)1io91Gy04brX3<|2vI3+09#YxHEfKmmIbT!-F-s7P8DTI<+=Ng& zF4G>BR}(qZ4a0`{LF8pboN1|tk;J2WU8aZtqVW{k9rYwcB#v06bO{b6T5U}{gvr1V zc_=QdA+pg|Tu!zeG{fTUj4Tscm|Ua-V4nu_+Wi;zNB4W3``h>T?zdki+$=#A&hdR~ zan%*6M+4K@`I0M~YJpOX9-TzT8^VM>#Fp~2?H;g!t!Ay$WHnZ=?H}y#)N8xAw(<0+ z+3v*q6ngP65^wq&YB1X`?bni2hUw!qrkL!bec0wxeSy@;JbJuUmj? zizFo6X@;(}i&gR!@|v}2@~hyzwp%~ir41JBgXvo5sMU;VOrCEPSaB})aCPs?H^cx6+}2-3Biy4BoowuIsyb+!*$J1=V;^nclQv({*~ zqUG~Ed{TMmji+}hHhifXbah5~ZutP&V(3__(YJ_zL`cOIfX_J-pQf?n-4cH;zC;|tS#AXpKpz%<{}URS*|@9 zjc^Q(w+qnjr`ne;7_{kn5;eUq#D!4`$r9S!>jTjWFSml={EX7n;t z{_`rJ>&uJT3G$Ex!zI1EHfLP+qA2zZMp@Xm)Fhl zYLX-JHlMD}!?mLe!|CF(8S9YJpIBFXopqpBQpwC-Ph1rzO+)t-J;%5UBX1@ntMdNs zH~-Mj9zf!_7fBFlOUWI7X z^e(&o8b7%>>z}rlj?2YE+%2`O;-SaSY<^xO(c+I#wnZu2A~-Ax`|*mY3@gSX#{{_8! zH{(0WX-dt+>HCfSH+A}ETz%?YzfHQ=mf`s9f4{Z)891fS-bdg#aPi(NsT;97k0NP` z+4o~JRQ=x9{;7=XS_|~~7@_PEw&k0BoPZ$pN-8r^!ss@dMz`r3o-cFFEuzTAEOX5;wKynNALe_TI( z(!PAXVg-}lUSW0PLpYwf>=>ftuZ@#u#$a+VUcPKvrM1@M{bj#vb=vDr=S}x%huTq=DyPO7wKUsO&Ac>DR$(W<=~Y`=It*xfpQc=q_^O5gEXPfFIgyHlQa z9~-5;qpeHB^b3{MkH++ZS0=5re6Q)o7&l$iQ@Lb(er-2^Ex;z zzuA3UDZd^*d}^LvIJL{uLZwoAkh`k`4s+O$fQJXQ%iY@V zZx03shP^tfSO@l~>lPlsrtTO#v(YCckK$&&-S?t+VKgRxJywh@5F{hk-7IikC uk5Sv-zB}sBw`Z|t)B70f-+u(G(x)Fj`6`Nsg@s?hfB*1RRE-bmum1=4?Sf_i literal 0 HcmV?d00001 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