Custom MCP Server for Databricks

Custom Model Context Protocol (MCP) server with OBO (On-Behalf-Of) authentication, designed for deployment as a Databricks app.

Features

The server implements three MCP tools:

• test_connection: Test server connectivity and OBO authentication status

• echo: Echo back a message (useful for testing basic functionality)

• get_user_info: Retrieve current user information via OBO authentication

Authentication

This server uses Databricks OBO (On-Behalf-Of) authentication with ModelServingUserCredentials, allowing it to act on behalf of the authenticated user making requests.

Deployment to Databricks

Prerequisites

1. Install Databricks CLI

2. Authenticate to your workspace:

   databricks auth login --host https://<your-workspace-hostname>

Deploy Steps

1. Create the Databricks app:

   databricks apps create mcp-cust-kat

2. Upload and deploy:

   DATABRICKS_USERNAME=$(databricks current-user me | jq -r .userName) databricks sync . "/Users/$DATABRICKS_USERNAME/mcp-cust-kat" databricks apps deploy mcp-cust-kat --source-code-path "/Workspace/Users/$DATABRICKS_USERNAME/mcp-cust-kat"

3. Access your app:

   • The app URL will be available in the Databricks UI

   • MCP endpoint: https://<app-url>/mcp

   • Health check: https://<app-url>/health

API Endpoints

• GET / - Root endpoint with server information

• GET /health - Health check endpoint with OBO authentication status

• POST /mcp - Main MCP JSON-RPC endpoint

Testing the Server

Once deployed, you can test the connection using the available MCP tools:

1. test_connection: Verifies the server is running and shows OBO authentication status

2. echo: Simple echo test to verify request/response functionality

3. get_user_info: Returns detailed information about the authenticated user

Local Development

Run locally for testing:

The server will start on port 8000 with both HTTP and MCP protocol support.

Project Structure

• server.py - Main FastAPI application with MCP protocol implementation

• app.yaml - Databricks app configuration

• pyproject.toml - Python project configuration and dependencies

• requirements.txt - Direct dependencies for the application