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"