Skip to content

Enable MCP server extension for Sherpa#1108

Open
aws-srijach wants to merge 4 commits intoaws:mainfrom
aws-srijach:enable-mcp-main
Open

Enable MCP server extension for Sherpa#1108
aws-srijach wants to merge 4 commits intoaws:mainfrom
aws-srijach:enable-mcp-main

Conversation

@aws-srijach
Copy link
Copy Markdown
Contributor

@aws-srijach aws-srijach commented Mar 16, 2026

Description

This adds the jupyter-server-mcp extension configuration

  • Add jupyter_server_mcp config (jupyter_server_config.py, mcp_settings.json)
  • Add JupyterLab overrides to place chat panel on right side
  • Add smai_post_startup.sh to copy configs at runtime
  • Wire post_startup script in start-jupyter-server entrypoint

Type of Change

  • Image update - Bug fix
  • Image update - New feature
  • Image update - Breaking change
  • SMD image build tool update
  • Documentation update

Release Information

Does this change need to be included in patch version releases? By default, any pull requests will only be added to the next SMD image minor version release once they are merged in template folder. Only critical bug fix or security update should be applied to new patch versions of existed image minor versions.

  • Yes (Critical bug fix or security update)
  • No (New feature or non-critical change)
  • N/A (Not an image update)

If yes, please explain why:
[Explain the criticality of this change and why it should be included in patch releases]

How Has This Been Tested?

  • Built v4.0.0 CPU image locally with docker build
  • Verified jupyter_server_mcp 0.1.2 loads as enabled server extension (jupyter server extension list)
  • Verified MCP configs copied to ~/.jupyter/ at startup via smai_post_startup.sh
  • Verified MCP endpoint responds on localhost:3001/mcp with proper MCP protocol initialization
  • Verified JupyterLab overrides applied

Checklist:

  • My code follows the style guidelines of this project
  • I have commented my code, particularly in hard-to-understand areas
  • I have made corresponding changes to the documentation
  • I have added tests that prove my fix is effective or that my feature works

Test Screenshots (if applicable):

Related Issues

[Link any related issues here]

Additional Notes

  • jupyter-server-mcp conda package needs to be added to cpu.env.in/gpu.env.in in the release branch
  • The MCP server runs on a separate port (3001) from Jupyter (8888)

@aws-srijach aws-srijach requested a review from a team as a code owner March 16, 2026 23:31
Comment thread template/v4/dirs/etc/sagemaker/jupyter/lab/settings/overrides.json

# Copy MCP server config and client settings to user-level Jupyter config
mkdir -p ~/.jupyter
cp /etc/sagemaker/jupyter/server/jupyter_server_config.py ~/.jupyter/
Copy link
Copy Markdown

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

fi

# Run SMAI post startup script
bash /etc/sagemaker/smai_post_startup.sh
Copy link
Copy Markdown

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

ok did a bit of digging- seems that the SMUS post startup script is actually triggered separately from extension and that is why it is called "post startup script". let's actually move back to having the commands here directly- my bad.

Copy link
Copy Markdown

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I believe in SMUS it is triggered via workflows

- Add jupyter_server_mcp config (jupyter_server_config.py, mcp_settings.json)
- Add JupyterLab overrides to place chat panel on right side
- Inline MCP config copy commands in start-jupyter-server entrypoint
Jupyter AI v3.0 auto-detects jupyter_server_mcp and ships with the
notebook toolkit built-in. Remove manual MCP client settings and
server config that are no longer needed.
@aws-pavankks
Copy link
Copy Markdown

General comment: Squash multiple commits into one commit for a CR or merge request.

@joshuatowner
Copy link
Copy Markdown

General comment: Squash multiple commits into one commit for a CR or merge request.

@aws-pavankks
for GH it should be OK to have multiple commits, when merging github squashes for you

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants