Skip to main content

Genie

Genie Agents

Give your users a chat box that queries your data. No text-to-SQL, no schema mapping, no custom LLM. A Genie Agent (formerly Genie space) is a Databricks natural-language interface over Unity Catalog tables: curated datasets plus a knowledge store (synonyms, example SQL, column descriptions) plus a compound AI system that turns questions into SQL. Your AppKit app wires it in with one plugin on the server and one component on the page.

Prerequisites

  • Databricks CLI v1.0.0+ with an authenticated profile.
  • A running AppKit app. See Apps quickstart.
  • A Genie Agent configured on Unity Catalog tables. See What is a Genie Agent for setup.

    Attach the agent as a resource in the app's configuration (UI or CLI) with Can run selected, and Databricks grants your app's service principal that permission. app.yaml then binds the resource to an env var. End-user permissions are covered below.

Why Genie

From question to result, Genie:

  • Understands your schema from Unity Catalog tables, synonyms, example SQL, and column descriptions.
  • Generates SQL from natural-language questions, with follow-up clarifications when the prompt is ambiguous.
  • Runs the query against your warehouse and returns tabular results ready to render.

The genie plugin wires all of that to your chat UI with SSE streaming, auth, and conversation replay handled.

Wire the plugin

Register the plugin with one or more space aliases. Alias keys become the alias prop on the frontend component.

server/server.ts
import { createApp, genie, server } from "@databricks/appkit";

await createApp({
  plugins: [
    server(),
    genie({
      spaces: {
        sales: process.env.SALES_GENIE_SPACE_ID!,
      },
    }),
  ],
});

Bind each alias to a Genie Agent resource in app.yaml:

app.yaml
env:
  - name: SALES_GENIE_SPACE_ID
    valueFrom: genie-space

The Databricks Apps runtime injects the space ID from the resource into the env var. Find your space ID in the Settings tab of the Genie Agent page in your workspace.

For a single-agent app, skip the spaces config entirely and bind the plugin's default env var:

app.yaml
env:
  - name: DATABRICKS_GENIE_SPACE_ID
    valueFrom: genie-space

With no spaces passed, the plugin reads DATABRICKS_GENIE_SPACE_ID and registers it under the default alias.

Render the chat component

client/src/pages/ChatPage.tsx
import { GenieChat } from "@databricks/appkit-ui/react";

export function ChatPage() {
  return (
    <div style={{ height: 600 }}>
      <GenieChat alias="sales" />
    </div>
  );
}

The alias prop must match a key in the server's spaces config. <GenieChat> fills its parent, so give it a fixed-height container or it collapses to zero. The component renders messages, handles streaming, persists the conversation ID in the URL, and replays history on reload. See the GenieChat reference for the full prop list.

Custom UI with useGenieChat

For a custom chat UI, use the hook directly. It returns the same message stream plus state for the request lifecycle.

client/src/pages/CustomChat.tsx
import { useGenieChat } from "@databricks/appkit-ui/react";

export function CustomChat() {
  const { messages, status, sendMessage, reset } = useGenieChat({
    alias: "sales",
  });

  return (
    <>
      {messages.map((msg) => (
        <div key={msg.id} data-role={msg.role}>
          {msg.content}
        </div>
      ))}
      <button
        onClick={() => sendMessage("What were total sales last quarter?")}
        disabled={status === "streaming"}
      >
        Ask
      </button>
      <button onClick={reset}>New conversation</button>
    </>
  );
}

status cycles through idle, streaming, loading-history, loading-older, and error. Use it to drive loading states in your UI. The hook also returns error, conversationId, and pagination helpers (hasPreviousPage, isFetchingPreviousPage, fetchPreviousPage). See the AppKit Genie plugin reference for the full return type and the Genie conversation API for the underlying REST API.

Multiple spaces

Register more than one space to let your users switch between domains, for example a sales space and a support space in the same app.

server/server.ts
genie({
  spaces: {
    sales: process.env.SALES_GENIE_SPACE_ID!,
    support: process.env.SUPPORT_GENIE_SPACE_ID!,
  },
}),

Bind each ID to a separate resource in app.yaml. See the Genie Multi-Agent Selector template for a working UI with agent switching, conversation cleanup, and URL sync.

Permissions and data access

The genie plugin calls the Genie API on behalf of the signed-in user. Both the app's service principal and each end user need access for a request to succeed:

  • App service principal: CAN RUN on the Genie Agent, granted when you attach the agent as an app resource (UI or CLI) with Can run selected. Permissions on the underlying data are not auto-provisioned: grant the service principal USE CATALOG, USE SCHEMA, and SELECT on the Unity Catalog tables separately. See Add a Genie Agent resource to an app.
  • End users: access to the Genie Agent (shared with them or via a group) and SELECT on the same tables. If the user doesn't have access, the call returns a 403. You don't write the permission check.

Where to next

Try the Genie Analytics App for a complete wired setup, or explore Custom agent endpoints for Knowledge Assistants and Supervisor Agents.

Databricks Developer Hub

Ready to ship your next agentic app in minutes?

Read docs