Genie Code

Genie Code is the AI coding and data assistant for developers and technical practitioners in the Databricks workspace. It generates and runs code, builds pipelines and AI/BI dashboards, debugs errors, and works directly with Unity Catalog tables, columns, and lineage to accelerate multi-step data work. It runs in notebooks, the SQL editor, the Lakeflow Pipelines Editor, AI/BI dashboards, and MLflow, and chat threads persist as you navigate between pages.

Genie Code is governed by your Unity Catalog permissions, so it can only access data and perform operations that you have permissions for. It is available at no additional cost to all Databricks customers; you pay only for the compute used to run your notebooks, queries, and jobs. Fair usage limits prevent abuse but do not affect most users.

Genie Code is part of the Genie family of AI experiences. Business users who want a simplified chat interface to ask data questions, explore dashboards, and run apps should use Genie. Data teams configuring trusted data, metrics, and business rules for domain-specific environments should see Genie Spaces. For privacy, trust, and security details, see Databricks AI assistive features trust and safety.

Demo video

See Genie Code in action in this demo video.

Agent mode capabilities by product surface

In Agent mode, Genie Code adapts to the product surface you are using and handles multi-step tasks autonomously. In the Lakeflow Pipelines Editor, it focuses on pipeline editing and data engineering. In notebooks and the SQL Editor, it supports data exploration and analysis. In dashboards, it supports data analysis and dashboard creation. In MLflow, it helps debug and improve GenAI applications.

• • Data science and ML in notebooks
  • Genie Code is integrated with Databricks notebooks and optimized to automate exploratory data analysis and model training.
• • Pipeline development in Lakeflow
  • Genie Code has data engineering capabilities that make it easy to automate ETL workloads and build Lakeflow Spark Declarative Pipelines.
• • Data analysis in dashboards
  • Genie Code can plan and generate production-ready dashboards and help with data analysis.
• • Observability & evaluation in MLflow
  • Genie Code can understand, debug, and improve your GenAI applications.

Built-in features for code assistance

Genie Code uses Unity Catalog metadata (tables, columns, descriptions, and your company's popular data assets) to give personalized responses while you write, debug, and run code. Use the following built-in features:

• • Chat with Genie Code
  • Ask Genie Code questions and it responds with relevant information, including citations from Databricks documentation.
• • Inline suggestions and autocomplete
  • Get help writing code with inline suggestions.
• • Quick fix
  • Genie Code automatically suggests fixes for basic errors that you can choose to accept and run.
• • Diagnose error
  • Genie Code analyzes more complex errors, including environment errors, and attempts to fix them.
• • /slash commands
  • Use /slash commands for common prompts.
• • Command shortcuts
  • Work with Genie Code directly in a notebook cell.
• • Filter data
  • Filter data using natural language in supported surfaces.
• • Explore sample data
  • Explore sample table data in Catalog Explorer using natural language in Catalog Explorer.

Tutorial

To get started, try the Genie Code tutorial:

• • Segmentation analysis tutorial
  • Use Genie Code to run end-to-end customer segmentation directly inside a Databricks notebook.

Geo availability of Genie Code features

Genie Code is a Designated Service that uses Geos to manage data residency when processing customer content. To learn more about the availability of Genie Code features in different geographic areas, see Databricks Designated Services.

tip

If you can't enable Genie Code, you might need to disable Enforce data processing within workspace Geography for AI features. See Enable cross-Geo processing.