Databricks AI Bridge library

The Databricks AI Bridge library provides a shared layer of APIs to interact with Databricks AI features, such as Databricks AI/BI Genie and Vector Search. Use these packages to help author agents with Agent Framework on Databricks.

Integration Packages

If you are using LangChain/LangGraph or the OpenAI SDK, we provide these integration packages for seamless integration of Databricks AI features.

• databricks-langchain - For LangChain/LangGraph users
• databricks-openai - For OpenAI SDK users

Installation

If you're using LangChain/LangGraph or OpenAI:

pip install databricks-langchain
pip install databricks-openai

For frameworks without dedicated integration packages:

pip install databricks-ai-bridge

Install from source

With https:

# For LangChain/LangGraph users (recommended):
pip install git+https://git@github.com/databricks/databricks-ai-bridge.git#subdirectory=integrations/langchain
# For OpenAI users (recommended):
pip install git+https://git@github.com/databricks/databricks-ai-bridge.git#subdirectory=integrations/openai
# Generic installation (only if needed):
pip install git+https://git@github.com/databricks/databricks-ai-bridge.git