databricks-solutions/lakebase-online-ml · Archived

databricks-agent-bricks

Create and manage Databricks Agent Bricks: Knowledge Assistants (KA) for document Q&A, Genie Spaces for SQL exploration, and Supervisor Agents (MAS) for multi-agent orchestration. Use when building conversational AI applications on Databricks.

First seen Aug 11, 2026

Installation

$ npx skills add databricks-solutions/lakebase-online-ml --skill databricks-agent-bricks

Stronger alternatives

This repository is archived — consider an actively maintained alternative.

Similar popular skills

Related neighbors and high-traction skills in the same topics — useful to compare before installing.

Also in this package

Other skills from databricks-solutions/lakebase-online-ml · top by installs.

npx skills add databricks-solutions/lakebase-online-ml

Browse all from databricks-solutions/lakebase-online-ml

More details

Agent compatibility

Declared targets from SKILL.md / docs. Unmarked agents are not listed — the skill may still install via the CLI.

Claude Code Not declared
Cursor Not declared
Codex Not declared
GitHub Copilot Not declared
Windsurf Not declared
Gemini CLI Not declared
Cline Not declared
OpenCode Not declared

Repository health

Stars 1
License LICENSE.md
Default branch main
Open issues 0
Status Archived

Package contents

Files included with this skill beyond the listing page.

  • skill md SKILL.md 9,971 B
  • docs SUMMARY.md 274 B

History

  1. First seen on skills.sh
  2. First recorded snapshot · 1 installs

SKILL.md

Agent Bricks

Create and manage Databricks Agent Bricks - pre-built AI components for building conversational applications.

Overview

Agent Bricks are three types of pre-built AI tiles in Databricks:

Brick Purpose Data Source
Knowledge Assistant (KA) Document-based Q&A using RAG PDF/text files in Volumes
Genie Space Natural language to SQL Unity Catalog tables
Supervisor Agent (MAS) Multi-agent orchestration Model serving endpoints

Prerequisites

Before creating Agent Bricks, ensure you have the required data:

For Knowledge Assistants

  • Documents in a Volume: PDF, text, or other files stored in a Unity Catalog volume
  • Generate synthetic documents using the databricks-unstructured-pdf-generation skill if needed

For Genie Spaces

  • See the databricks-genie skill for comprehensive Genie Space guidance
  • Tables in Unity Catalog with the data to explore
  • Generate raw data using the databricks-synthetic-data-gen skill
  • Create tables using the databricks-spark-declarative-pipelines skill

For Supervisor Agents

  • Model Serving Endpoints: Deployed agent endpoints (KA endpoints, custom agents, fine-tuned models)
  • Genie Spaces: Existing Genie spaces can be used directly as agents for SQL-based queries
  • Mix and match endpoint-based and Genie-based agents in the same Supervisor Agent

For Unity Catalog Functions

  • Existing UC Function: Function already registered in Unity Catalog
  • Agent service principal has EXECUTE privilege on the function

For External MCP Servers

  • Existing UC HTTP Connection: Connection configured with ismcpconnection: 'true'
  • Agent service principal has USE CONNECTION privilege on the connection

MCP Tools

Knowledge Assistant Tool

manage_ka - Manage Knowledge Assistants (KA)

  • action: "createorupdate", "get", "findbyname", or "delete"
  • name: Name for the KA (for createorupdate, findbyname)
  • volumepath: Path to documents (e.g., /Volumes/catalog/schema/volume/folder) (for createor_update)
  • description: (optional) What the KA does (for createorupdate)
  • instructions: (optional) How the KA should answer (for createorupdate)
  • tileid: The KA tile ID (for get, delete, or update via createor_update)
  • addexamplesfromvolume: (optional, default: true) Auto-add examples from JSON files (for createor_update)

Actions:

  • createorupdate: Requires name, volumepath. Optionally pass tileid to update.
  • get: Requires tileid. Returns tileid, name, description, endpointstatus, knowledgesources, examples_count.
  • findbyname: Requires name (exact match). Returns found, tileid, name, endpointname, endpointstatus. Use this to look up an existing KA when you know the name but not the tileid.
  • delete: Requires tile_id.

Genie Space Tools

For comprehensive Genie guidance, use the databricks-genie skill.

Basic tools available:

  • createorupdate_genie - Create or update a Genie Space
  • get_genie - Get Genie Space details
  • delete_genie - Delete a Genie Space

See databricks-genie skill for:

  • Table inspection workflow
  • Sample question best practices
  • Curation (instructions, certified queries)

IMPORTANT: There is NO system table for Genie spaces (e.g., system.ai.geniespaces does not exist). To find a Genie space by name, use the findgeniebyname tool.

Supervisor Agent Tool

manage_mas - Manage Supervisor Agents (MAS)

  • action: "createorupdate", "get", "findbyname", or "delete"
  • name: Name for the Supervisor Agent (for createorupdate, findbyname)
  • agents: List of agent configurations (for createorupdate), each with:

- name: Agent identifier (required) - description: What this agent handles - critical for routing (required) - katileid: Knowledge Assistant tile ID (use for document Q&A agents - recommended for KAs) - geniespaceid: Genie space ID (use for SQL-based data agents) - endpointname: Model serving endpoint name (for custom agents) - ucfunctionname: Unity Catalog function name in format catalog.schema.functionname - connectionname: Unity Catalog connection name (for external MCP servers) - Note: Provide exactly one of: katileid, geniespaceid, endpointname, ucfunctionname, or connection_name

  • description: (optional) What the Supervisor Agent does (for createorupdate)
  • instructions: (optional) Routing instructions for the supervisor (for createorupdate)
  • tileid: The Supervisor Agent tile ID (for get, delete, or update via createor_update)
  • examples: (optional) List of example questions with question and guideline fields (for createorupdate)

Actions:

  • createorupdate: Requires name, agents. Optionally pass tile_id to update.
  • get: Requires tileid. Returns tileid, name, description, endpointstatus, agents, examplescount.
  • findbyname: Requires name (exact match). Returns found, tileid, name, endpointstatus, agentscount. Use this to look up an existing Supervisor Agent when you know the name but not the tileid.
  • delete: Requires tile_id.

Typical Workflow

1. Generate Source Data

Before creating Agent Bricks, generate the required source data:

For KA (document Q&A):

1. Use `databricks-unstructured-pdf-generation` skill to generate PDFs
2. PDFs are saved to a Volume with companion JSON files (question/guideline pairs)

For Genie (SQL exploration):

1. Use `databricks-synthetic-data-gen` skill to create raw parquet data
2. Use `databricks-spark-declarative-pipelines` skill to create bronze/silver/gold tables

2. Create the Agent Brick

Use manageka(action="createorupdate", ...) or managemas(action="createorupdate", ...) with your data sources.

3. Wait for Provisioning

Newly created KA and MAS tiles need time to provision. The endpoint status will progress:

  • PROVISIONING - Being created (can take 2-5 minutes)
  • ONLINE - Ready to use
  • OFFLINE - Not running

4. Add Examples (Automatic)

For KA, if addexamplesfrom_volume=true, examples are automatically extracted from JSON files in the volume and added once the endpoint is ONLINE.

Best Practices

  1. Use meaningful names: Names are sanitized automatically (spaces become underscores)
  2. Provide descriptions: Helps users understand what the brick does
  3. Add instructions: Guide the AI's behavior and tone
  4. Include sample questions: Shows users how to interact with the brick
  5. Use the workflow: Generate data first, then create the brick

Example: Multi-Modal Supervisor Agent

manage_mas(
    action="create_or_update",
    name="Enterprise Support Supervisor",
    agents=[
        {
            "name": "knowledge_base",
            "ka_tile_id": "f32c5f73-466b-...",
            "description": "Answers questions about company policies, procedures, and documentation from indexed files"
        },
        {
            "name": "analytics_engine",
            "genie_space_id": "01abc123...",
            "description": "Runs SQL analytics on usage metrics, performance stats, and operational data"
        },
        {
            "name": "ml_classifier",
            "endpoint_name": "custom-classification-endpoint",
            "description": "Classifies support tickets and predicts resolution time using custom ML model"
        },
        {
            "name": "data_enrichment",
            "uc_function_name": "support.utils.enrich_ticket_data",
            "description": "Enriches support ticket data with customer history and context"
        },
        {
            "name": "ticket_operations",
            "connection_name": "ticket_system_mcp",
            "description": "Creates, updates, assigns, and closes support tickets in external ticketing system"
        }
    ],
    description="Comprehensive enterprise support agent with knowledge retrieval, analytics, ML, data enrichment, and ticketing operations",
    instructions="""
    Route queries as follows:
    1. Policy/procedure questions → knowledge_base
    2. Data analysis requests → analytics_engine
    3. Ticket classification → ml_classifier
    4. Customer context lookups → data_enrichment
    5. Ticket creation/updates → ticket_operations

    If a query spans multiple domains, chain agents:
    - First gather information (analytics_engine or knowledge_base)
    - Then take action (ticket_operations)
    """
)

Related Skills

  • [databricks-genie](../databricks-genie/SKILL.md) - Comprehensive Genie Space creation, curation, and Conversation API guidance
  • [databricks-unstructured-pdf-generation](../databricks-unstructured-pdf-generation/SKILL.md) - Generate synthetic PDFs to feed into Knowledge Assistants
  • [databricks-synthetic-data-gen](../databricks-synthetic-data-gen/SKILL.md) - Create raw data for Genie Space tables
  • [databricks-spark-declarative-pipelines](../databricks-spark-declarative-pipelines/SKILL.md) - Build bronze/silver/gold tables consumed by Genie Spaces
  • [databricks-model-serving](../databricks-model-serving/SKILL.md) - Deploy custom agent endpoints used as MAS agents
  • [databricks-vector-search](../databricks-vector-search/SKILL.md) - Build vector indexes for RAG applications paired with KAs

See Also

  • 1-knowledge-assistants.md - Detailed KA patterns and examples
  • databricks-genie skill - Detailed Genie patterns, curation, and examples
  • 2-supervisor-agents.md - Detailed MAS patterns and examples