Example - custom MCP server on Databricks Apps
This example shows how to create and launch a custom agent on Databricks Apps.
Please note that this example doesn't use any Databricks SDK, and is independent of the mcp package in the root dir of this repo.
Prerequisites
- Databricks CLI installed and configured
uv
Local development
- run
uvsync:
uv sync
- start the server locally. Changes will trigger a reload:
uv run custom-server
This will start the server on http://localhost:8000
Deploying a custom MCP server on Databricks Apps
There are two ways to deploy the server on Databricks Apps: using the databricks apps CLI or using the databricks bundle CLI. Depending on your preference, you can choose either method.
Both approaches require first configuring Databricks authentication:
export DATABRICKS_CONFIG_PROFILE=<your-profile-name> # e.g. custom-mcp-server
databricks auth login --profile "$DATABRICKS_CONFIG_PROFILE"
Using databricks apps CLI
To deploy the server using the databricks apps CLI, follow these steps:
Create a Databricks app to host your MCP server:
databricks apps create mcp-custom-server
Upload the source code to Databricks and deploy the app:
DATABRICKS_USERNAME=$(databricks current-user me | jq -r .userName)
databricks sync . "/Users/$DATABRICKS_USERNAME/my-mcp-server"
databricks apps deploy mcp-custom-server --source-code-path "/Workspace/Users/$DATABRICKS_USERNAME/my-mcp-server"
Using databricks bundle CLI
To deploy the server using the databricks bundle CLI, follow these steps
Update the app.yaml file in this directory to use the following command:
command: ["uvicorn", "custom_server.app:app"]
- In this directory, run the following command to deploy and run the MCP server on Databricks Apps:
uv build --wheel
databricks bundle deploy
databricks bundle run custom-mcp-server
Connecting to the MCP server
To connect to the MCP server, use the Streamable HTTP transport with the following URL:
https://your-app-url.usually.ends.with.databricksapps.com/mcp/
For authentication, you can use the Bearer token from your Databricks profile.
You can get the token by running the following command:
databricks auth token -p <name-of-your-profile>
Please note that the URL should end with /mcp/ (including the trailing slash), as this is required for the server to work correctly.