Connecting with Databricks

AtScale enables you to connect to Databricks using either OAuth or an API bearer token. Both methods are described below.

Prerequisites

• The AtScale MCP server must be enabled for your AtScale instance.
• You must have the admin or application_admin role assigned in the Identity Broker.
• Your AtScale instance must be available to open ports.

Connect with an API bearer token

Create a connection

First, an AtScale administrator must create a connection in Databricks.

To create a connection:

1. In Design Center, generate an API token:

   1. Click the profile icon in the top-right corner. The Account panel opens.

   2. In the API Token section, click the generate icon. A token is generated.

      

   3. Copy the token.

2. Log in to Databricks and go to the Marketplace.

3. Click View MCP listings.

4. Search for and select AtScale MCP.

5. Click Install.

6. In the Install "AtScale MCP" dialog, complete the following fields:

   • Connection name: Enter a name for the connection.
   • Host: Enter the URL for your AtScale instance. Be sure to remove any trailing slashes.
   • Base path: Enter /mcp.
   • Bearer token: Enter the API token you generated above.

7. Accept the terms and conditions, then click Install.

You can now either add your connection to the Databricks Playground or use it to create a multi-agent supervisor.

Add your connection to the Databricks Playground

Once you have created a connection for the AtScale MCP Server, you can add it to the Databricks Playground and begin querying your model.

To add your connection to the Playground:

1. In the left-side navigation, click Playground.
2. At the top of the page, select the LLM you want to use.
3. In the Add your own tool field, click Add, then click Add tool. The Add tools panel opens.
4. Go to the MCP Servers tab.
5. In the External MCP Servers section, click Select a Unity Catalog Connection and select your AtScale MCP server.
6. Click Save.

You can now begin querying your model data.

Create a multi-agent supervisor

As an alternative to using Genie, you can use your connection to create a multi-agent supervisor. This option can give you more accurate results.

Note

To create a multi-agent supervisor, you must have already created a connection as described in the procedure above.

To create a multi-agent supervisor:

1. Obtain the atscale-mcp client's credentials:

   1. In Design Center, in the left-side navigation, click Security. The Identity Broker opens.

   2. In the left-side navigation, click Clients.

   3. Click the atscale-mcp client.

   4. Go to the Credentials tab and copy the Client Secret.

      Important

      Do not regenerate the Client Secret.

2. In Databricks, in the left-side navigation, click Agents.

3. Under Multi-Agent Supervisor, click Build. The Create Multi-Agent Supervisor page opens.

4. Enter a name for your supervisor.

5. In the Configure Agents section, complete the following fields:

   • Type: Select External MCP Server.
   • Unity Catalog connection: Select your AtScale MCP server.

6. Click Create Agent.

A chat interface appears, where you can begin querying your AtScale models.

Connect using OAuth

Create a connection

First, an AtScale administrator must create a connection in Databricks.

To create a connection:

1. In Design Center, obtain the required connection details:

   1. Obtain the secret for the atscale-mcp client:

      1. In Design Center, in the left-side navigation, click Security. The Identity Broker opens.

      2. In the left-side navigation, click Clients.

      3. Click the atscale-mcp client.

      4. Go to the Credentials tab and copy the Client Secret.

         Important

         Do not regenerate the Client Secret.

   2. Obtain the authorization and token endpoints:

      1. In the left-side navigation, click Realm settings.

      2. On the General tab, scroll down to the Endpoints field and click OpenID Endpoint Configuration.

         

         The OpenID configuration opens in a new browser window.

      3. Enable the Pretty-print option in your browser.

      4. Locate the "authorization_endpoint" line and copy the URL inside the quotes.

      5. Locate the "token_endpoint" line and copy the URL inside the quotes.

2. In Databricks, in the left-side navigation, click Catalog.

3. At the top of the page, click External Data.

4. On the External Data page, click Connections, then click Create Connection. The Set up connection page opens.

5. On the Connection basics screen, complete the following fields:

   • Connection name: Enter a name for your connection.
   • Connection type: Select HTTP.
   • Auth type: Select OAuth User to Machine Per User.

6. Click Next.

7. On the Authentication screen, complete the following fields:

   • Host: Enter the URL for your AtScale instance. Be sure to remove any trailing slashes.
   • Client secret: Enter the atscale-mcp client secret you copied above.
   • Client ID: Enter atscale-mcp.
   • Authorization endpoint: Enter the authorization endpoint URL you copied above.

8. Click Next.

9. On the Connection details screen, complete the following fields:

   • Token endpoint: Enter the token endpoint URL you copied above.
   • Base path: Enter /mcp.
   • Is mcp connection: Check the box.
   • Oauth credential exchange method: Leave set to Header and body.

10. Click Create connection.

11. On the overview page for your connection, click Login. If you are not currently logged in to Design Center, you will be prompted for your credentials.

You can now add your connection to the Databricks Playground.

Add your connection to the Databricks Playground

Once you have created a connection for the AtScale MCP Server, you can add it to the Databricks Playground and begin querying your models.

To add your connection to the Playground:

1. In the left-side navigation, click Playground.
2. At the top of the page, select the LLM you want to use.
3. In the Add your own tool field, click Add, then click Add tool. The Add tools panel opens.
4. Go to the MCP Servers tab.
5. In the External MCP Servers section, click Select a Unity Catalog Connection and select your AtScale MCP server.
6. Click Save.

You can now begin querying your model data.