tests.system.databricks.example_databricks_unity_mcp¶

Example Dag giving an agent the tools of a Unity Gateway MCP Service.

Needs a Databricks connection whose identity has EXECUTE on the MCP Service and USE CATALOG and USE SCHEMA on its catalog and schema, an LLM connection for the agent, and these environment variables:

UNITY_MCP_SERVICE: the service’s three-level name, catalog.schema.service.
UNITY_MCP_READ_ONLY_TOOL: the name of one of its tools that only reads, which the agent calls once.

Attributes¶

`DAG_ID`
`UNITY_MCP_SERVICE`
`UNITY_MCP_READ_ONLY_TOOL`
`ask_unity_mcp_service`
`test_run`

Module Contents¶

tests.system.databricks.example_databricks_unity_mcp.DAG_ID = 'example_databricks_unity_mcp'[source]¶

tests.system.databricks.example_databricks_unity_mcp.UNITY_MCP_SERVICE[source]¶

tests.system.databricks.example_databricks_unity_mcp.UNITY_MCP_READ_ONLY_TOOL[source]¶

tests.system.databricks.example_databricks_unity_mcp.ask_unity_mcp_service[source]¶

tests.system.databricks.example_databricks_unity_mcp.test_run[source]¶