skillfed

databricks-agents

Mosaic AI Agent Framework SDK

databricks-agents v1.11.0 8.2M downloads/30d#1,648 on PyPI
License unclear Databricks Proprietary License Active released

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-agents

uv

uv add databricks-agents

poetry

poetry add databricks-agents

Installing 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

Development Status :: 3 - AlphaLicense :: Other/Proprietary LicenseProgramming Language :: Python :: Implementation :: CPythonProgramming Language :: Python :: Implementation :: PyPy

Tags

databricks agent frameworkbuild agents on databricksmosaic ai sdkdatabricks llm integrationagent development platform
databricks-platformagent-frameworkproprietary-license

More Application Frameworks packages

fastapi

FastAPI is a Python web framework for building…

permissive · top 100 on PyPI

annotated-doc

Provides a way to document function parameters,…

permissive · top 100 on PyPI

textual

Textual is a Python framework for building…

permissive · top 100 on PyPI

typer

Typer builds command-line applications from…

permissive · top 1,000 on PyPI

mcp

Build and connect to Model Context Protocol…

permissive · top 1,000 on PyPI

Werkzeug

Werkzeug is a WSGI utility library providing…

permissive · top 1,000 on PyPI

appdynamics

A Python agent that instruments and monitors…

unclear · top 15,000 on PyPI

databricks-feature-engineering

Databricks Feature Engineering client for…

unclear · top 5,000 on PyPI

databricks-ai-search

Python client for Databricks AI Search, a…

unclear · top 5,000 on PyPI

databricks-ai-bridge

Provides a shared Python API layer for building…

unclear · top 5,000 on PyPI

databricks-feature-store

This package is deprecated as of v0.17.0; it…

unclear · top 15,000 on PyPI

databricks-langchain

Integrates Databricks AI services—LLMs, vector…

permissive · top 5,000 on PyPI

databricks-vectorsearch

A deprecated re-export shim that forwards to…

unclear · top 5,000 on PyPI

databricks-dlt

Provides type hints, API specs, and IDE…

unclear · top 5,000 on PyPI

apache-airflow-providers-databricks

Integrates Apache Airflow with Databricks,…

permissive · top 1,000 on PyPI

unitycatalog-ai

Provides Python APIs to create, retrieve, and…

permissive · top 5,000 on PyPI

Further reading