From 51ff5581ff06d8bbc1571afbdc9020cea5fa1084 Mon Sep 17 00:00:00 2001 From: apcho Date: Tue, 8 Jul 2025 14:47:29 -0700 Subject: [PATCH] Disable the SMUS generative AI extension in SMAI and disable jupyter-ai extension in SMUS **Description** SMUS will support a single generative AI chat experience provided by sagemaker_gen_ai_jupyterlab_extension. to accomplish this in SMUS spaces, jupyter-ai will be disabled. In SMAI, jupyter-ai will be the single generative AI chat experience, so the sagemaker_gen_ai_jupyterlab_extension will be disabled in SMAI spaces. **Testing Done** Manually disabled the extensions using the commands included in start-jupyter-server and start-sagemaker-ui-jupyter-server and confirmed only one chat experience exists after running command. --- template/v2/dirs/usr/local/bin/start-jupyter-server | 2 +- .../v2/dirs/usr/local/bin/start-sagemaker-ui-jupyter-server | 3 +++ template/v3/dirs/usr/local/bin/start-jupyter-server | 2 +- .../v3/dirs/usr/local/bin/start-sagemaker-ui-jupyter-server | 3 +++ 4 files changed, 8 insertions(+), 2 deletions(-) diff --git a/template/v2/dirs/usr/local/bin/start-jupyter-server b/template/v2/dirs/usr/local/bin/start-jupyter-server index 7b20d3862..c9aedde69 100755 --- a/template/v2/dirs/usr/local/bin/start-jupyter-server +++ b/template/v2/dirs/usr/local/bin/start-jupyter-server @@ -10,7 +10,7 @@ if [ -n "$SAGEMAKER_RECOVERY_MODE" ]; then else # Activate conda environment 'base' micromamba activate base - jupyter labextension disable @amzn/sagemaker-data-explorer-jl-plugin sagemaker-data-explorer:plugin amzn/sagemaker-ui-theme-jlplugin @amzn/sagemaker-ui-doc-manager-jl-plugin @amzn/sagemaker-connection-magics-jlextension + jupyter labextension disable @amzn/sagemaker-data-explorer-jl-plugin sagemaker-data-explorer:plugin amzn/sagemaker-ui-theme-jlplugin @amzn/sagemaker-ui-doc-manager-jl-plugin @amzn/sagemaker-connection-magics-jlextension sagemaker_gen_ai_jupyterlab_extension fi # Start Jupyter server in rtc mode for shared spaces diff --git a/template/v2/dirs/usr/local/bin/start-sagemaker-ui-jupyter-server b/template/v2/dirs/usr/local/bin/start-sagemaker-ui-jupyter-server index 3c03c50a8..0cd986b58 100755 --- a/template/v2/dirs/usr/local/bin/start-sagemaker-ui-jupyter-server +++ b/template/v2/dirs/usr/local/bin/start-sagemaker-ui-jupyter-server @@ -10,6 +10,9 @@ if [ -n "$SAGEMAKER_RECOVERY_MODE" ]; then else # Activate conda environment 'base' which is the default for Cosmos micromamba activate base + + # Disable jupyter-ai in favor of sagemaker_gen_ai_jupyterlab_extension + jupyter labextension disable @jupyter-ai/core fi sudo cp -r /etc/sagemaker-ui/kernels/. /opt/conda/share/jupyter/kernels/ diff --git a/template/v3/dirs/usr/local/bin/start-jupyter-server b/template/v3/dirs/usr/local/bin/start-jupyter-server index a0242ffb1..575358103 100755 --- a/template/v3/dirs/usr/local/bin/start-jupyter-server +++ b/template/v3/dirs/usr/local/bin/start-jupyter-server @@ -10,7 +10,7 @@ if [ -n "$SAGEMAKER_RECOVERY_MODE" ]; then else # Activate conda environment 'base' micromamba activate base - jupyter labextension disable @amzn/sagemaker-data-explorer-jl-plugin sagemaker-data-explorer:plugin amzn/sagemaker-ui-theme-jlplugin @amzn/sagemaker-ui-doc-manager-jl-plugin @amzn/sagemaker-connection-magics-jlextension + jupyter labextension disable @amzn/sagemaker-data-explorer-jl-plugin sagemaker-data-explorer:plugin amzn/sagemaker-ui-theme-jlplugin @amzn/sagemaker-ui-doc-manager-jl-plugin @amzn/sagemaker-connection-magics-jlextension sagemaker_gen_ai_jupyterlab_extension fi # Start Jupyter server in rtc mode for shared spaces diff --git a/template/v3/dirs/usr/local/bin/start-sagemaker-ui-jupyter-server b/template/v3/dirs/usr/local/bin/start-sagemaker-ui-jupyter-server index dfe780ef2..a2e7cb256 100755 --- a/template/v3/dirs/usr/local/bin/start-sagemaker-ui-jupyter-server +++ b/template/v3/dirs/usr/local/bin/start-sagemaker-ui-jupyter-server @@ -10,6 +10,9 @@ if [ -n "$SAGEMAKER_RECOVERY_MODE" ]; then else # Activate conda environment 'base' which is the default for Cosmos micromamba activate base + + # Disable jupyter-ai in favor of sagemaker_gen_ai_jupyterlab_extension + jupyter labextension disable @jupyter-ai/core fi sudo cp -r /etc/sagemaker-ui/kernels/. /opt/conda/share/jupyter/kernels/