skillfed

databricks-langchain

Support for Databricks AI support in LangChain

databricks-langchain v0.20.0 3.5M downloads/30d#2,592 on PyPI
Permissive license Apache-2.0 Active released

What it is and what it does

databricks-langchain bridges Databricks AI capabilities into LangChain applications, consolidating what were previously scattered across multiple packages. It provides LangChain-compatible wrappers for Databricks-hosted LLMs (like Llama and Mixtral via ChatDatabricks), vector search operations, embeddings generation, and experimental Genie integration for AI-powered analytics agents.

The package is designed for teams already using Databricks workspaces who want to build LangChain-based AI applications without managing separate authentication or API layers. It depends on langchain, databricks-sdk, and several Databricks-specific adapters (databricks-ai-bridge, databricks-mcp, databricks-openai, and others), making it a fairly heavy integration layer best suited for applications where Databricks is the primary AI infrastructure.

Use it for:

  • Build LangChain agents that query Databricks-hosted LLMs like Llama 3.1 without direct SDK calls.
  • Create RAG pipelines using DatabricksVectorSearch for semantic search over Databricks-managed vector stores.
  • Generate embeddings for documents or queries using Databricks embedding models within LangChain workflows.
  • Prototype Genie-powered analytics agents that answer business questions over Databricks data (preview feature).
  • Migrate existing langchain-databricks or langchain-community Databricks code to the consolidated package.

Worth the install?

AI-flagged interpretation of the facts on this page — verify before relying

Integrates Databricks AI services—LLMs, vector search, embeddings, and Genie—into LangChain applications through a unified package.

Yes, if you are building LangChain applications on Databricks infrastructure. The package consolidates fragmented Databricks integrations and is actively maintained. The 11-dependency footprint is expected for an integration layer. No known vulnerabilities. Install only if you have a Databricks workspace and LangChain is your chosen framework; it adds no value as a standalone tool.

Install

databricks-langchain on PyPI

pip

pip install databricks-langchain

uv

uv add databricks-langchain

poetry

poetry add databricks-langchain

Installing databricks-langchain

Before you install

Low install friction; pure Python wheel. Active maintenance with recent release (65 days ago). Depends on 11 runtime packages including langchain, databricks-sdk, and several Databricks-specific bridges—a substantial dependency footprint typical of integration packages.

License in practice

Apache-2.0 permissive license allows commercial use, modification, and redistribution with minimal restrictions, suitable for most production scenarios.

Quickstart

pip install databricks-langchain

from databricks_langchain import ChatDatabricks

llm = ChatDatabricks(endpoint="databricks-meta-llama-3-1-70b-instruct")
response = llm.invoke("Hello")

Requires Python >=3.10 and valid Databricks workspace credentials to authenticate with Databricks endpoints.

Verify before relying

  • Whether Genie Agent features are available outside the private preview or require special enablement.
  • Performance characteristics and latency when calling Databricks endpoints through LangChain.
  • Compatibility matrix with specific Databricks workspace versions or LangChain versions beyond the stated requirements.

Package facts

License Apache-2.0 (permissive)
Python support supports the current Python release (>=3.10)
Install friction low — pure-Python wheel
Runtime dependencies 11 — databricks-ai-bridge, databricks-ai-search, databricks-mcp, databricks-openai, databricks-sdk, langchain-mcp-adapters, langchain, mlflow, openai, pydantic, unitycatalog-langchain
Maintenance actively maintained — 65 days since the last release
First released
Downloads 3,527,468/month — #2,592 on PyPI (30-day window, as of 2026-08-14)
Known vulnerabilities none known (OSV.dev, checked 2026-08-14)

Evidence: databricks_langchain-0.20.0-py3-none-any.whl

Tags

databricks langchain integrationdatabricks llm langchaindatabricks vector search langchaindatabricks embeddingslangchain databricks aigenie langchain agentdatabricks ai bridge
databricks-integrationrag-pipelinellm-framework

More Artificial Intelligence packages

litellm

LiteLLM provides a unified Python interface to…

permissive · top 100 on PyPI

huggingface-hub

Client library and CLI tool for downloading,…

permissive · top 100 on PyPI

langchain

LangChain provides a framework for building…

permissive · top 1,000 on PyPI

hf-xet

hf-xet provides chunk-based deduplication and…

permissive · top 1,000 on PyPI

tokenizers

Tokenizers converts raw text into token…

permissive · top 1,000 on PyPI

transformers

Transformers provides a unified framework for…

permissive · top 1,000 on PyPI

databricks-ai-bridge

Provides a shared Python API layer for building…

unclear · top 5,000 on PyPI

langchain-databricks

Connects LangChain applications to Databricks…

permissive · top 15,000 on PyPI

unitycatalog-langchain

Exposes Unity Catalog functions as LangChain…

permissive · top 5,000 on PyPI

langchain-google-vertexai

Connects LangChain applications to Google…

permissive · top 1,000 on PyPI

databricks-openai

Integrates Databricks AI features, particularly…

permissive · top 5,000 on PyPI

databricks-agents

Mosaic AI Agent Framework SDK provides a Python…

unclear · top 5,000 on PyPI

databricks-vectorsearch

A deprecated re-export shim that forwards to…

unclear · top 5,000 on PyPI

databricks-ai-search

Python client for Databricks AI Search, a…

unclear · top 5,000 on PyPI

langchain-elasticsearch

Integrates Elasticsearch with LangChain to…

permissive · top 15,000 on PyPI

langchain-milvus

Integrates LangChain with Milvus vector…

permissive · top 15,000 on PyPI