Use Genie in multi-agent systems

Preview

This feature is in Public Preview.

This page describes Genie agent systems and shows how to create a multi-agent system using Mosaic AI Agent Framework and Genie spaces.

Requirements

• You must have a Genie space set up. See Set up and manage an AI/BI Genie space.

Benefits of Genie agents

Multi-agent systems consist of multiple AI agents working together, each with specialized capabilities. A Genie agent allows users to use natural language to query structured data.

Agent Framework allows agents to share answers as context for each other. So users can get answers that pull from multiple sources. You could have a RAG agent that queries an unstructured document corpus and combine it with a Genie agent that queries a database of records.

Multi-agent systems typically consist of a supervisor agent that directs and manages context for specialized worker agents.

Example notebook: Multi-agent system with Genie

The following notebook shows you how to create a multi-agent system using LangGraph and Genie.

Multi-agent using Genie and LangGraph

Open notebook in new tab

Genie authentication using a Personal Access Token (PAT)

Deployed agents that have access to a Genie Space must authenticate to the Genie Space and its underlying resources, including Unity Catalog Functions, Unity Catalog Tables, and SQL Warehouses.

Create a personal access token and store it as a Databricks secret to authenticate the agent to the Genie Space and its underlying resources:

1. Create a PAT either as yourself or as a service principal.
   • PATs have an expiration date, so you must rotate the token to ensure service continuity.
2. Add secrets-based environment variables to the deployed agent's model serving endpoint. See Add secrets-based environment variables.
3. Specify the correct permissions for the PAT. The permissions level depends on the resource. See Automatic authentication passthrough.
   • Provision with CAN RUN on the Genie Space.
   • Provision with CAN USE on the SQL Warehouse powering the Genie Space
   • Provision with SELECT on underlying Unity Catalog Tables
   • Provision with EXECUTE on underlying Unity Catalog Functions

Access Genie spaces with on-behalf-of-user authentication

On-behalf-of-user authentication allows an agent system to access a Genie space and its underlying resources with agent end user's permissions.

See On-behalf-of-user authentication for more information on how to set up on-behalf-of-user authentication with Genie spaces.