Databricks Agents

Databricks External Agents allow you to make a Databricks Agent available as a standard Dataiku Agent

Once configured, it can be leveraged anywhere a Dataiku Agent is supported, including:

  • The Dataiku Flow (notably Prompt Recipes)

  • Chat UIs such as Agent Hub

  • The LLM Mesh API

  • Prompt Studios

  • /agents/agent-review

  • /agent/agent-evaluation

It features streaming responses and agent discovery (automatically fetching and listing available agents from your Databricks workspace)

Prerequisites

The Databricks External Agent is provided by the “Databrick AI” plugin, which you need to install.

You need:

  • A Databricks Workspace with Model Serving enabled.

  • A Dataiku Connection to Databricks configured with either a Personal Access Token or OAuth2 credentials.

  • An active Serving Endpoint deployed in Databricks (must have the agent/ prefix task type).

Usage

  1. Navigate to your Dataiku project.

  2. Go to Agents & GenAI Models > + New Agent.

  3. Select Databricks Agent and provide a name.

  4. Configure the Databricks connection

  5. Choose the Databricks serving endpoint from the dropdown.

You can then use the built-in chat interface within the Agent configuration page to test your agent.

Limitations

Image inputs are not supported