By Datasource
Databricks Actions
The Databricks action pack contains the following actions:
- Search Databricks with SQL
- Search Databricks with Genie
Overview
Glean enables users to query their Databricks workspaces using natural language queries and direct SQL via Databricks Action Packs. Glean users can combine Databricks data with their other connected apps and documents.
Key Features
- Real-time Querying: The Glean Assistant can query your Databricks Genie space or tables in your Databricks workspace live, without needing to index any additional data.
- Support for User OAuth: Customers can enable User OAuth for querying.
- Combining Structured and Unstructured Data: Glean users can combine Databricks data with any other data Glean has access to.
Usage Instructions
Set Up in Agent Builder
- Ensure your admin has successfully set up the Databricks Genie Action.
- Create a new Glean Agent.
- Add the following steps to the agent:
- Step 1: Select the Trigger type > Task.
- Step 2: Select the “Databricks Genie Query Fetcher” action. In the text box, add:
For Databricks Genie, send the following natural language query to the tool as is without any modification: [[query]].
- Step 3: Select the “Respond” action. In the text box, add:
Show me the results from the Databricks Genie Query Fetcher tool only. If there are more than 3 rows, show them as a table. Use the first row in the response from the tool for column names. Always show citations.
- Done! Use the agent by sharing the link.
Best Practices and Troubleshooting
For Databricks Genie Action
- Specify a Genie Space ID either in the Action Pack configuration section (default space ID) or in the step configuration section in the Agent Builder.
For Databricks SQL Execution Action
- Specify a Databricks Warehouse ID either in the Action Pack configuration section (default warehouse ID) or in the step configuration section in the Agent Builder.
If the Databricks Genie action results are not as expected
- Refer to the SQL created in the intermediate step citation and debug.
To customize Databricks Genie behavior
- Set up custom instructions in the Genie space configuration. Refer to Databricks’ guidelines on curating an effective Genie space.
If your query instructions are sometimes dropped
- Add a specific instruction saying
Send Query: [[Query]] to Databricks
. If you just saySend [[Query]] to Databricks
, the query may sometimes get dropped by the LLM.
Known Limitations
- Multi-turn conversations in the Databricks Genie Query Fetcher action are not supported. For example, if Genie responds with a clarification question, Glean does not pass the response back to Genie.
- Glean does not automatically retry queries on errors from the Genie API.