yashshingvi/databricks-genie-MCP

A Model Context Protocol (MCP) server that connects to the Databricks Genie API, allowing LLMs to ask natural language questions, run SQL queries, and interact with Databricks conversational agents.
✨ Features
- List Genie spaces available in your Databricks workspace (Currently Manual/Using Resource)
- Fetch metadata (title, description) of a specific Genie space
- Start new Genie conversations with natural language questions
- Ask follow-up questions in ongoing Genie conversations
- Retrieve SQL and result tables in structured format
🧱 Prerequisites
- Python 3.7+
- Databricks workspace with:
- Personal access token
- Genie API enabled
- Permissions to access Genie spaces and run queries
⚙️ Setup
1. **Clone this repository**
2. **Create and activate a virtual environment** (recommended):
**Install dependencies:**
Create a **.env** file in the root directory with the following variables:
📌 **Manually Adding Genie Space IDs**
**Note:**
At this time, the Databricks Genie API **does not provide a public endpoint to list all available space IDs and titles**. (afaik)
As a workaround, you need to **manually add the Genie space IDs and their titles** in the get_genie_space_id() function in main.py.
🧪 Test the Server
You can test the MCP server using the inspector (optional but recommended):
OR
**You can directly build and run docker to test the server**
💬 Use with Claude Desktop
Download Claude Desktop
**Install Your MCP Server:**
From your project directory, run:
**Once Server Installed**
1. Connect in Claude
2. Open Claude Desktop
3. Click Resources → Add Resource
4. Select your Genie MCP Server
5. Start chatting with your data using natural language! 🎯
🧾 Obtaining Databricks Credentials
**Host**
Your Databricks instance URL (e.g., your-instance.cloud.databricks.com) — do not include https://
**Token**
1. Go to your Databricks workspace
2. Click your username (top right) → User Settings
3. Under the Developer tab, click Manage under "Access tokens"
4. Generate a new token and copy it
🚀 Running the Server
This will start the Genie MCP server over the stdio transport for LLM interaction.
🧰 Available MCP Tools
The following MCP tools are available:
**Tool Description**
1. get_genie_space_id() List available Genie space IDs and titles
2. get_space_info(space_id: str) Retrieve title and description of a Genie space
3. ask_genie(space_id: str, question: str) Start a new Genie conversation and get results
4. follow_up(space_id: str, conversation_id: str, question: str) Continue an existing Genie conversation