Databricks MCP servers

Table of Contents

• Databricks MCP servers
• Table of Contents
  • Overview
  • Unity Catalog Server
    • Overview
    • Usage
    • Supported tools
  • Developer Tools Server
  • Support
  • Contributing

Overview

An experimental collection of MCP servers to help AI agents fetch enterprise data from Databricks, automate common developer actions on Databricks, etc:

• Databricks Unity Catalog server: Fetch data and run tools registered in from Unity Catalog, making agents aware of your enterprise data
• Databricks developer tools server: Perform common developer actions in Databricks, like creating and updating notebooks, running jobs, etc. This server is not yet usable, but contributions are welcome! Note: if developing from within the Databricks product (e.g. using Databricks notebooks), we recommend using Databricks Assistant

The set of servers and tools in this repo is fluid and will evolve over time. We welcome contributions to this repo - please first read the contributor guidelines to streamline the process and discover areas where help is needed.

Unity Catalog Server

Overview

A Model Context Protocol server that exposes structured and unstructured data in Unity Catalog (vector search indexes, functions, and Genie spaces), as tools.

Usage

1. Install uv
2. Install Python using uv python install 3.12
3. Configure Databricks credentials with access to the required APIs
4. Add the server to your MCP client configuration. For example, to use this server with Claude Desktop, add the following to your claude_desktop_config.json:

{
  "mcpServers": {
    "databricks_unity_catalog": {
      "command": "/path/to/uv/executable/uv",
      "args": [
        "--directory",
        "/path/to/this/repo",
        "run",
        "unitycatalog-mcp",
        "-s",
        "your_catalog.your_schema",
        "-g",
        "genie_space_id_1,genie_space_id_2"
      ]
    }
  }
}

Supported tools

The list of tools supported by this server is dynamically inferred at startup time based on the functions and vector search indexes within the specified Unity Catalog schema, as well as any specified Genie spaces. In particular, the server exposes the following tools:

• UC Functions: for each UC function, the server exposes a tool with the same name, arguments, and return type as the function
• Vector search indexes: for each vector search index, the server exposes a tool for querying that vector search index
• Genie spaces: for each Genie space, the server exposes tools for managing conversations and sending questions to the space

Developer Tools Server

This server is currently under construction. It is not yet usable, but contributions are welcome!

Support

Please note that all projects in the databrickslabs GitHub organization are provided for your exploration only, and are not formally supported by Databricks with Service Level Agreements (SLAs). They are provided AS-IS and we do not make any guarantees of any kind. Please do not submit a support ticket relating to any issues arising from the use of these projects.

Any issues discovered through the use of this project should be filed as GitHub Issues on the Repo. They will be reviewed as time permits, but there are no formal SLAs for support.

Contributing

We welcome contributions :) - see CONTRIBUTING.md for details. Please make sure to read this guide before submitting pull requests, to ensure your contribution has the best chance of being accepted.