databricks-agents
Mosaic AI Agent Framework SDK
What it is and what it does
Databricks-agents is the Mosaic AI Agent Framework SDK, a Python library for constructing and deploying agents within the Databricks ecosystem. It integrates with the Databricks platform and LLM services, providing abstractions and utilities for agent development. The package depends on databricks-sdk for platform connectivity, mlflow-skinny for experiment tracking, and standard data-science libraries like pandas, numpy, and pydantic for data handling and validation.
This is a platform-specific tool: it is licensed exclusively for use in connection with Databricks Platform Services under a proprietary agreement. The license explicitly restricts use to that context and terminates if the underlying Databricks agreement ends or is breached. It is not a general-purpose agent framework and cannot be used independently of a Databricks subscription or Community Edition account.
Use it for:
- Build autonomous agents that interact with Databricks data and compute resources within a workspace.
- Integrate LLM-powered workflows with Databricks MLflow for experiment tracking and model management.
- Deploy agents that leverage Databricks SQL, Delta Lake, and other platform services as part of agent logic.
- Develop multi-step agent workflows that coordinate with Databricks infrastructure and governance.
Worth the install?
AI-flagged interpretation of the facts on this page — verify before relying
Mosaic AI Agent Framework SDK provides a Python library for building and deploying agents on the Databricks platform, integrating with Databricks services and LLM capabilities.
Yes, if you have an active Databricks account or Community Edition access and need to build agents within that platform. No, if you need a standalone agent framework or do not have a Databricks agreement in place—the proprietary license explicitly restricts use to Databricks Platform Services and is not transferable to other contexts. Verify your Databricks terms of service before installing.
Install
databricks-agents on PyPI
pip
pip install databricks-agentsuv
uv add databricks-agentspoetry
poetry add databricks-agentsInstalling databricks-agents
Before you install
Low install friction with a pure Python wheel. Actively maintained as of 73 days since last release. Requires Python 3.10 or later and pulls in 15 runtime dependencies including databricks-sdk, mlflow-skinny, and common data-science libraries.
License in practice
Licensed under Databricks Proprietary License with unclear treatment. Use is restricted to connection with Databricks Platform Services under a separate agreement; the license terminates if that agreement ends or is breached. Not suitable for standalone or off-platform use.
Quickstart
pip install databricks-agents
from databricks_agents import Agent
# Requires active Databricks workspace connection and agreement to DB license
Requires Python 3.10+. Use is restricted to Databricks Platform Services under the DB license agreement; standalone use outside that context is not permitted.
Verify before relying
- What specific agent capabilities or frameworks are included in this SDK beyond integration with Databricks services.
- Whether this package can be used with Databricks Community Edition or only paid tiers.
- Documentation and examples availability for getting started with agent development.
Package facts
| License | Databricks Proprietary License (unclear) |
| Python support | supports the current Python release (>=3.10) |
| Install friction | low — pure-Python wheel |
| Runtime dependencies | 15 — dataclasses-json, databricks-sdk, googleapis-common-protos, numpy, pandas, jinja2, mlflow-skinny, tenacity, tiktoken, tqdm, urllib3, pydantic, whenever, boto3, botocore |
| Maintenance | actively maintained — 73 days since the last release |
| First released | |
| Downloads | 8,213,510/month — #1,648 on PyPI (30-day window, as of 2026-08-14) |
| Known vulnerabilities | none known (OSV.dev, checked 2026-08-14) |
Evidence: databricks_agents-1.11.0-py3-none-any.whl
Tags
More Application Frameworks packages
FastAPI is a Python web framework for building…
permissive · top 100 on PyPI
annotated-docProvides a way to document function parameters,…
permissive · top 100 on PyPI
textualTextual is a Python framework for building…
permissive · top 100 on PyPI
typerTyper builds command-line applications from…
permissive · top 1,000 on PyPI
mcpBuild and connect to Model Context Protocol…
permissive · top 1,000 on PyPI
WerkzeugWerkzeug is a WSGI utility library providing…
permissive · top 1,000 on PyPI
appdynamicsA Python agent that instruments and monitors…
unclear · top 15,000 on PyPI
databricks-feature-engineeringDatabricks Feature Engineering client for…
unclear · top 5,000 on PyPI
databricks-ai-searchPython client for Databricks AI Search, a…
unclear · top 5,000 on PyPI
databricks-ai-bridgeProvides a shared Python API layer for building…
unclear · top 5,000 on PyPI
databricks-feature-storeThis package is deprecated as of v0.17.0; it…
unclear · top 15,000 on PyPI
databricks-langchainIntegrates Databricks AI services—LLMs, vector…
permissive · top 5,000 on PyPI
databricks-vectorsearchA deprecated re-export shim that forwards to…
unclear · top 5,000 on PyPI
databricks-dltProvides type hints, API specs, and IDE…
unclear · top 5,000 on PyPI
apache-airflow-providers-databricksIntegrates Apache Airflow with Databricks,…
permissive · top 1,000 on PyPI
unitycatalog-aiProvides Python APIs to create, retrieve, and…
permissive · top 5,000 on PyPI