From 535fa69a19d6d58b7de61fbdf9703908011dbbdd Mon Sep 17 00:00:00 2001 From: Jenny Date: Mon, 8 Jun 2026 11:53:17 -0700 Subject: [PATCH] agent-langgraph[-advanced]: pin databricks-ai-bridge to fix branch (temporary) Pins databricks-ai-bridge, databricks-openai, and databricks-langchain to databricks/databricks-ai-bridge#435 so the LangGraph templates boot on environments that pull a databricks-vectorsearch / databricks-ai- search version that no longer re-exports VectorSearchIndex from databricks.vector_search.client. The legacy import in databricks_openai was the trigger: ImportError: cannot import name 'VectorSearchIndex' from 'databricks.vector_search.client' Remove this override once a databricks-ai-bridge release containing the fix is published to PyPI. --- agent-langgraph-advanced/pyproject.toml | 10 ++++++++++ agent-langgraph/pyproject.toml | 11 +++++++++++ 2 files changed, 21 insertions(+) diff --git a/agent-langgraph-advanced/pyproject.toml b/agent-langgraph-advanced/pyproject.toml index d76f3544..e089ab50 100644 --- a/agent-langgraph-advanced/pyproject.toml +++ b/agent-langgraph-advanced/pyproject.toml @@ -11,6 +11,7 @@ dependencies = [ "fastapi>=0.129.0", "uvicorn>=0.41.0", "databricks-langchain[memory]>=0.19.0", + "databricks-openai", "databricks-ai-bridge[agent-server]>=0.19.0", "databricks-sdk>=0.79.0", "databricks-agents>=1.9.3", @@ -39,6 +40,15 @@ setup = [ [tool.uv] default-groups = ["dev", "setup"] +# TEMPORARY: pin databricks-ai-bridge and its integrations to the fix branch for +# https://github.com/databricks/databricks-ai-bridge/pull/435 while the fix is +# unreleased. Remove these once a databricks-ai-bridge release containing the +# VectorSearchIndex import fix is published to PyPI. +[tool.uv.sources] +databricks-ai-bridge = { git = "https://github.com/databricks/databricks-ai-bridge.git", branch = "jennsun/fix-vector-search-index-import" } +databricks-openai = { git = "https://github.com/databricks/databricks-ai-bridge.git", branch = "jennsun/fix-vector-search-index-import", subdirectory = "integrations/openai" } +databricks-langchain = { git = "https://github.com/databricks/databricks-ai-bridge.git", branch = "jennsun/fix-vector-search-index-import", subdirectory = "integrations/langchain" } + [tool.pytest.ini_options] base_url = "http://localhost:8000" diff --git a/agent-langgraph/pyproject.toml b/agent-langgraph/pyproject.toml index 554e32b3..f9f9c2c0 100644 --- a/agent-langgraph/pyproject.toml +++ b/agent-langgraph/pyproject.toml @@ -11,6 +11,8 @@ dependencies = [ "fastapi>=0.129.0", "uvicorn>=0.41.0", "databricks-langchain>=0.17.0", + "databricks-openai", + "databricks-ai-bridge", "databricks-agents>=1.9.3", "mlflow>=3.10.0", "langgraph>=1.1.0", @@ -36,6 +38,15 @@ setup = [ [tool.uv] default-groups = ["dev", "setup"] +# TEMPORARY: pin databricks-ai-bridge and its integrations to the fix branch for +# https://github.com/databricks/databricks-ai-bridge/pull/435 while the fix is +# unreleased. Remove these once a databricks-ai-bridge release containing the +# VectorSearchIndex import fix is published to PyPI. +[tool.uv.sources] +databricks-ai-bridge = { git = "https://github.com/databricks/databricks-ai-bridge.git", branch = "jennsun/fix-vector-search-index-import" } +databricks-openai = { git = "https://github.com/databricks/databricks-ai-bridge.git", branch = "jennsun/fix-vector-search-index-import", subdirectory = "integrations/openai" } +databricks-langchain = { git = "https://github.com/databricks/databricks-ai-bridge.git", branch = "jennsun/fix-vector-search-index-import", subdirectory = "integrations/langchain" } + [project.scripts] quickstart = "scripts.quickstart:main"