databricks/databricks-ai-bridge

databricks/databricks-ai-bridge

Language: Python

License: NOASSERTION

Stars: 92

Forks: 61

Open issues: 56

Created: 2024-10-16T05:24:21Z

Pushed: 2026-06-10T01:56:13Z

Default branch: main

Fork: no

Archived: no

README:

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](./integrations/langchain/README.md) - For LangChain/LangGraph users
• [databricks-openai](./integrations/openai/README.md) - 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