Build Databricks-Powered Applications in Gemini Code Assist with CData Code Assist MCP

Somya Sharma
Somya Sharma
Technical Marketing Engineer
Use the CData Code Assist MCP for Databricks to explore live Databricks Data in Gemini Code Assist to assist with building Databricks-powered applications.

Gemini Code Assist is an AI-powered coding companion that integrates intelligent code generation into everyday development workflows. With support for MCP, Gemini Code Assist can connect to live enterprise data sources directly from Visual Studio Code, enabling natural language interaction with structured data without switching context or manually writing data access code.

Model Context Protocol (MCP) is an open standard for connecting LLM clients to external services through structured tool interfaces. MCP servers expose capabilities such as schema discovery and live querying, allowing AI agents to retrieve and reason over real-time data safely and consistently.

This guide walks through installing the CData Code Assist MCP for Databricks, configuring the connection to Databricks, connecting the Code Assist MCP add-on to Gemini Code Assist, and querying live Databricks data from within the editor.

About Databricks Data Integration

Accessing and integrating live data from Databricks has never been easier with CData. Customers rely on CData connectivity to:

  • Access all versions of Databricks from Runtime Versions 9.1 - 13.X to both the Pro and Classic Databricks SQL versions.
  • Leave Databricks in their preferred environment thanks to compatibility with any hosting solution.
  • Secure authenticate in a variety of ways, including personal access token, Azure Service Principal, and Azure AD.
  • Upload data to Databricks using Databricks File System, Azure Blog Storage, and AWS S3 Storage.

While many customers are using CData's solutions to migrate data from different systems into their Databricks data lakehouse, several customers use our live connectivity solutions to federate connectivity between their databases and Databricks. These customers are using SQL Server Linked Servers or Polybase to get live access to Databricks from within their existing RDBMs.

Read more about common Databricks use-cases and how CData's solutions help solve data problems in our blog: What is Databricks Used For? 6 Use Cases.


Getting Started


Prerequisites

Before starting, ensure the following requirements are met:

  1. Visual Studio Code is installed on the machine
  2. Gemini Code Assist extension is enabled in Visual Studio Code
  3. CData Code Assist MCP for Databricks has been installed
  4. Access to Databricks

Note: Gemini Code Assist must already be set up and functional in Visual Studio Code before configuring MCP servers. MCP servers are accessed when Gemini Code Assist is running in Agent mode.

Step 1: Download and install the CData Code Assist MCP for Databricks

  1. To begin, download the CData Code Assist MCP for Databricks
  2. Find and double-click the installer to begin the installation
  3. Run the installer and follow the prompts to complete the installation

When the installation is complete, the Code Assist MCP add-on is ready for configuration by connecting to Databricks.

Step 2: Configure the connection to Databricks

  1. After installation, open the CData Code Assist MCP for Databricks configuration wizard

    NOTE: If the wizard does not open automatically, search for "CData Code Assist MCP for Databricks" in the Windows search bar and open the application.

  2. In MCP Configuration > Configuration Name, either select an existing configuration or choose to create a new one
  3. Name the configuration (e.g. "cdata_databricks") and click OK
  4. Enter the appropriate connection properties in the configuration wizard

    To connect to a Databricks cluster, set the properties as described below.

    Note: The needed values can be found in your Databricks instance by navigating to Clusters, and selecting the desired cluster, and selecting the JDBC/ODBC tab under Advanced Options.

    • Server: Set to the Server Hostname of your Databricks cluster.
    • HTTPPath: Set to the HTTP Path of your Databricks cluster.
    • Token: Set to your personal access token (this value can be obtained by navigating to the User Settings page of your Databricks instance and selecting the Access Tokens tab).
  5. Click Connect to authenticate with Databricks
  6. Then, click Save Configuration to save the Code Assist MCP add-on

This process creates a .mcp configuration file that Gemini Code Assist will reference when launching the Code Assist MCP add-on. With the Code Assist MCP add-on configured, it is ready to connect to Gemini Code Assist.

Step 3: Connect the Code Assist MCP add-on to Gemini Code Assist

  1. Ensure Visual Studio Code is installed and the Gemini Code Assist extension is enabled
  2. From the configuration wizard, click Next after saving and testing the connection
  3. Select Gemini Code Assist from the AI MCP Tool dropdown
  4. Click Copy JSON to copy the generated MCP configuration to the clipboard
  5. Paste the copied JSON into the appropriate configuration file based on the desired scope:
    • User-level: Configuration applies across all projects for the current user
    • Workspace-level: Configuration applies only to the current workspace or project

    NOTE: The configuration includes the path to Java 17+ executable and the CData Code Assist MCP add-on JAR file. The final argument must match the MCP configuration name saved in the wizard (e.g. "cdata_databricks").

  6. Save the configuration file and restart Visual Studio Code if necessary

Step 4: Query live Databricks data in Gemini Code Assist

  1. Open Visual Studio Code and select Gemini Code Assist in the activity bar
  2. Enter /mcp in the chat prompt to verify the connection status. The Databricks Code Assist MCP add-on should appear with a green connection indicator
  3. Ask questions about Databricks data using natural language. For example:
    "Provide the list of all tables available in my Databricks data connection."
  4. Generate code that works with live Databricks data. For example:
    "Write a function to retrieve records from the Customers table where City matches a given value."

Gemini Code Assist is now fully integrated with the CData Code Assist MCP add-on and can use the MCP tools exposed to explore schemas, execute live queries against Databricks, and generate data-aware code.

Build with Code Assist MCP. Deploy with CData Drivers.

Download Code Assist MCP for free and give your AI tools schema-aware access to live Databricks data during development. When you're ready to move to production, CData Databricks Drivers deliver the same SQL-based access with enterprise-grade performance, security, and reliability.

Visit the CData Community to share insights, ask questions, and explore what's possible with MCP-powered AI workflows.

Ready to get started?

Download a free Databricks Code Assist MCP to get started:

 Download Now

Learn more:

Databricks Icon Databricks Code Assist MCP

The CData Code Assist MCP for Databricks provides schema-aware context for AI-assisted code generation with live Databricks data.