databricks-langchain 0.13.0__tar.gz

databricks_langchain-0.13.0/.gitignore

@@ -0,0 +1,22 @@
+# test
+.pytest_cache/
+mlruns/
+
+# Byte-compiled files
+__pycache__
+*.pyc
+
+# Editor files
+*.vscode
+.idea
+
+# Mac
+.DS_Store
+
+# packaging
+dist/
+*.egg-info/
+Pipfile
+docs/build
+
+**/uv.lock

databricks_langchain-0.13.0/PKG-INFO

@@ -0,0 +1,76 @@
+Metadata-Version: 2.4
+Name: databricks-langchain
+Version: 0.13.0
+Summary: Support for Databricks AI support in LangChain
+Author-email: Databricks <agent-feedback@databricks.com>
+License: Apache-2.0
+Requires-Python: >=3.10
+Requires-Dist: databricks-ai-bridge>=0.4.2
+Requires-Dist: databricks-mcp>=0.5.1
+Requires-Dist: databricks-sdk>=0.65.0
+Requires-Dist: databricks-vectorsearch>=0.50
+Requires-Dist: langchain-mcp-adapters>=0.1.13
+Requires-Dist: langchain>=1.0.0
+Requires-Dist: mlflow>=3.0.0
+Requires-Dist: openai>=1.99.9
+Requires-Dist: pydantic>2.10.0
+Requires-Dist: unitycatalog-langchain[databricks]>=0.3.0
+Provides-Extra: memory
+Requires-Dist: databricks-ai-bridge[memory]>=0.10.0; extra == 'memory'
+Requires-Dist: langgraph-checkpoint-postgres>=2.0.5; extra == 'memory'
+Description-Content-Type: text/markdown
+
+# 🦜🔗 Databricks LangChain Integration
+
+The `databricks-langchain` package provides seamless integration of Databricks AI features into LangChain applications. This repository is now the central hub for all Databricks-related LangChain components, consolidating previous packages such as `langchain-databricks` and `langchain-community`.
+
+## Installation
+
+### From PyPI
+```sh
+pip install databricks-langchain
+```
+
+### From Source
+```sh
+pip install git+https://git@github.com/databricks/databricks-ai-bridge.git#subdirectory=integrations/langchain
+```
+
+## Key Features
+
+- **LLMs Integration:** Use Databricks-hosted large language models (LLMs) like Llama and Mixtral through `ChatDatabricks`.
+- **Vector Search:** Store and query vector representations using `DatabricksVectorSearch`.
+- **Embeddings:** Generate embeddings with `DatabricksEmbeddings`.
+- **Genie:** Use [Genie](https://www.databricks.com/product/ai-bi/genie) in Langchain.
+
+## Getting Started
+
+### Use LLMs on Databricks
+```python
+from databricks_langchain import ChatDatabricks
+
+llm = ChatDatabricks(endpoint="databricks-meta-llama-3-1-70b-instruct")
+```
+
+### Use a Genie Space as an Agent (Preview)
+> **Note:** Requires Genie API Private Preview. Contact your Databricks account team for enablement.
+
+```python
+from databricks_langchain.genie import GenieAgent
+
+genie_agent = GenieAgent(
+    "space-id", "Genie",
+    description="This Genie space has access to sales data in Europe"
+)
+```
+
+---
+
+## Contribution Guide
+We welcome contributions! Please see our [contribution guidelines](https://github.com/databricks/databricks-ai-bridge/tree/main/integrations/langchain) for details.
+
+## License
+This project is licensed under the [MIT License](LICENSE).
+
+Thank you for using Databricks LangChain!
+

databricks_langchain-0.13.0/README.md

@@ -0,0 +1,54 @@
+# 🦜🔗 Databricks LangChain Integration
+
+The `databricks-langchain` package provides seamless integration of Databricks AI features into LangChain applications. This repository is now the central hub for all Databricks-related LangChain components, consolidating previous packages such as `langchain-databricks` and `langchain-community`.
+
+## Installation
+
+### From PyPI
+```sh
+pip install databricks-langchain
+```
+
+### From Source
+```sh
+pip install git+https://git@github.com/databricks/databricks-ai-bridge.git#subdirectory=integrations/langchain
+```
+
+## Key Features
+
+- **LLMs Integration:** Use Databricks-hosted large language models (LLMs) like Llama and Mixtral through `ChatDatabricks`.
+- **Vector Search:** Store and query vector representations using `DatabricksVectorSearch`.
+- **Embeddings:** Generate embeddings with `DatabricksEmbeddings`.
+- **Genie:** Use [Genie](https://www.databricks.com/product/ai-bi/genie) in Langchain.
+
+## Getting Started
+
+### Use LLMs on Databricks
+```python
+from databricks_langchain import ChatDatabricks
+
+llm = ChatDatabricks(endpoint="databricks-meta-llama-3-1-70b-instruct")
+```
+
+### Use a Genie Space as an Agent (Preview)
+> **Note:** Requires Genie API Private Preview. Contact your Databricks account team for enablement.
+
+```python
+from databricks_langchain.genie import GenieAgent
+
+genie_agent = GenieAgent(
+    "space-id", "Genie",
+    description="This Genie space has access to sales data in Europe"
+)
+```
+
+---
+
+## Contribution Guide
+We welcome contributions! Please see our [contribution guidelines](https://github.com/databricks/databricks-ai-bridge/tree/main/integrations/langchain) for details.
+
+## License
+This project is licensed under the [MIT License](LICENSE).
+
+Thank you for using Databricks LangChain!
+

databricks_langchain-0.13.0/pyproject.toml

@@ -0,0 +1,78 @@
+[project]
+name = "databricks-langchain"
+version = "0.13.0"
+description = "Support for Databricks AI support in LangChain"
+authors = [
+    { name="Databricks", email="agent-feedback@databricks.com" },
+]
+readme = "README.md"
+license = { text="Apache-2.0" }
+requires-python = ">=3.10"
+dependencies = [
+    "langchain>=1.0.0",
+    "databricks-vectorsearch>=0.50",
+    "databricks-ai-bridge>=0.4.2",
+    "mlflow>=3.0.0",
+    "pydantic>2.10.0",
+    "unitycatalog-langchain[databricks]>=0.3.0",
+    "databricks-sdk>=0.65.0",
+    "openai>=1.99.9",
+    "langchain-mcp-adapters>=0.1.13",
+    "databricks_mcp>=0.5.1",
+]
+
+[project.optional-dependencies]
+memory = [
+  "langgraph-checkpoint-postgres>=2.0.5",
+  "databricks-ai-bridge[memory]>=0.10.0",
+]
+
+[dependency-groups]
+dev = [
+  "typing_extensions>=4.15.0",
+  "ruff==0.14.10",
+  "ty>=0.0.11",
+  { include-group = "tests" },
+]
+
+tests = [
+  "pytest>=9.0.0",
+  "langgraph>=0.2.27",
+  "pytest-timeout>=2.3.1",
+  "pytest-asyncio>=1.3.0",
+  "anyio>=4.8.0",
+]
+
+
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[tool.uv.sources]
+databricks-ai-bridge = { path = "../../", editable = true }
+
+[tool.hatch.build]
+include = [
+  "src/databricks_langchain/*"
+]
+
+[tool.hatch.build.targets.wheel]
+packages = ["src/databricks_langchain"]
+
+[tool.ruff]
+include = ["pyproject.toml", "src/**/*.py", "tests/**/*.py"]
+extend = "../../pyproject.toml"
+
+[tool.pytest.ini_options]
+filterwarnings = [
+    "ignore::Warning",
+    "default::Warning:databricks_langchain",
+    "default::Warning:tests",
+]
+
+[tool.ty.environment]
+root = ["./src", "./tests"]
+
+[tool.ty.src]
+include = ["./src", "./tests"]
+exclude = ["./tests/integration_tests"]

databricks_langchain-0.13.0/src/databricks_langchain/__init__.py

@@ -0,0 +1,51 @@
+"""
+**Re-exported Unity Catalog Utilities**
+
+This module re-exports selected utilities from the Unity Catalog open source package.
+
+Available aliases:
+
+- :class:`databricks_langchain.UCFunctionToolkit`
+- :class:`databricks_langchain.UnityCatalogTool`
+- :class:`databricks_langchain.DatabricksFunctionClient`
+- :func:`databricks_langchain.set_uc_function_client`
+
+Refer to the Unity Catalog `documentation <https://docs.unitycatalog.io/ai/integrations/langchain/#using-unity-catalog-ai-with-langchain>`_ for more information.
+"""
+
+from unitycatalog.ai.core.base import set_uc_function_client
+from unitycatalog.ai.core.databricks import DatabricksFunctionClient
+from unitycatalog.ai.langchain.toolkit import UCFunctionToolkit, UnityCatalogTool
+
+from databricks_langchain.chat_models import ChatDatabricks
+from databricks_langchain.checkpoint import AsyncCheckpointSaver, CheckpointSaver
+from databricks_langchain.embeddings import DatabricksEmbeddings
+from databricks_langchain.genie import GenieAgent
+from databricks_langchain.multi_server_mcp_client import (
+    DatabricksMCPServer,
+    DatabricksMultiServerMCPClient,
+    MCPServer,
+)
+from databricks_langchain.store import AsyncDatabricksStore, DatabricksStore
+from databricks_langchain.vector_search_retriever_tool import VectorSearchRetrieverTool
+from databricks_langchain.vectorstores import DatabricksVectorSearch
+
+# Expose all integrations to users under databricks-langchain
+__all__ = [
+    "AsyncCheckpointSaver",
+    "AsyncDatabricksStore",
+    "ChatDatabricks",
+    "CheckpointSaver",
+    "DatabricksEmbeddings",
+    "DatabricksStore",
+    "DatabricksVectorSearch",
+    "GenieAgent",
+    "VectorSearchRetrieverTool",
+    "UCFunctionToolkit",
+    "UnityCatalogTool",
+    "DatabricksFunctionClient",
+    "set_uc_function_client",
+    "DatabricksMultiServerMCPClient",
+    "DatabricksMCPServer",
+    "MCPServer",
+]