This release is a pre-release and may not be stable for production use.
DataHub Agent Context
MCP tools for AI agents to search and query your DataHub metadata catalog — works with Claude, Cursor, Copilot, and any MCP-compatible AI assistant.
What you can do
- Search datasets, dashboards, pipelines, and other data assets by name or description
- Retrieve entity details — schema, lineage, ownership, tags, glossary terms, and more
- Trace lineage upstream and downstream across your data assets
- Mutate metadata — update descriptions, tags, owners, domains, and glossary terms
- Build LangChain or Google ADK agents with pre-built tool bindings
- Set up Snowflake AI agents with one CLI command
Installation
pip install datahub-agent-context
# With LangChain support
pip install "datahub-agent-context[langchain]"
Quickstart
LangChain agent
from datahub.sdk.main_client import DataHubClient
from datahub_agent_context.langchain_tools import build_langchain_tools
client = DataHubClient.from_env()
# Read-only tools (search, lineage, entity details)
tools = build_langchain_tools(client, include_mutations=False)
# Include write tools (tags, descriptions, owners, etc.)
tools = build_langchain_tools(client, include_mutations=True)
# DataHub Cloud: add Ask DataHub AI assistant
from datahub_agent_context.langchain_tools import build_langchain_cloud_tools
tools += build_langchain_cloud_tools(client, ask_datahub=True)
Snowflake AI agent setup
datahub agent create snowflake \
--datahub-url https://your-datahub-instance \
--datahub-token your-token
Available tools
Search — search(), search_documents(), grep_documents()
Entities — get_entities(), list_schema_fields()
Lineage — get_lineage(), get_lineage_paths_between()
Queries — get_dataset_queries()
Incidents — list_incidents()
Mutations — add_tags(), remove_tags(), update_description(), set_domains(), add_owners(), add_glossary_terms(), add_structured_properties(), save_document(), raise_incident(), resolve_incident()
Cloud-only — ask_datahub_chat() (DataHub Cloud AI assistant)
Links
1 maintainer
Metadata
1 maintainer
Release files for datahub-agent-context 1.7.0.12rc1
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | |
|---|---|---|---|
| datahub_agent_context-1.7.0.12rc1.tar.gz | 111.3 kB | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| datahub_agent_context-1.7.0.12rc1-py3-none-any.whl | Python 3 | none | any | Details |
Total release size: 263.0 kB
Release files / datahub_agent_context-1.7.0.12rc1.tar.gz
| Download URL | datahub_agent_context-1.7.0.12rc1.tar.gz |
|---|---|
| Size | 111.3 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
95a8a86bc9ee00897388dde6cee9cd085cacce8e563e538093c3f7aea4bcba8b
|
|
BLAKE2b-256 checksum How to use checksums |
75900bf150c1e5dbf6d53e4d3aac62c90989fd40d74cdc114364f95c647d80d1
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/7.0.0 CPython/3.10.21
|
Release files / datahub_agent_context-1.7.0.12rc1-py3-none-any.whl
| Download URL | datahub_agent_context-1.7.0.12rc1-py3-none-any.whl |
|---|---|
| Size | 151.7 kB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
f4f8d221fd2a044d1e0940309fad9ce9532f0e0a5f8fa4ed10d8cc4bbd09ba4a
|
|
BLAKE2b-256 checksum How to use checksums |
6b1d9d333d6434a9a5a89e170000f014c54186555e180a8200c78a38a79a1fbb
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/7.0.0 CPython/3.10.21
|
