Skip to content

Pin databricks-openai to git main to fix VectorSearchIndex import - #1

Open
jennsun wants to merge 1 commit into
mainfrom
fix/vectorsearch-import-via-git-source
Open

Pin databricks-openai to git main to fix VectorSearchIndex import#1
jennsun wants to merge 1 commit into
mainfrom
fix/vectorsearch-import-via-git-source

Conversation

@jennsun

@jennsun jennsun commented Jun 9, 2026

Copy link
Copy Markdown
Owner

Summary

  • The released databricks-openai (transitive of databricks-langchain) imports VectorSearchIndex from databricks.vector_search.client, which was removed in newer databricks-vectorsearch releases. Fresh deploys 500 on import.
  • databricks/databricks-ai-bridge#435 fixed the import path but is not yet in a tagged release.
  • Override databricks-openai with the merged source on main in all three memory-pipeline templates (agent-stateful-example, agent-stateless-example, agent-database-admin).

Diff (per template)

  • Add databricks-openai as a direct dependency.
  • Add a [tool.uv.sources] block pointing databricks-openai at git+https://github.com/databricks/databricks-ai-bridge.git, branch = "main", subdirectory = "integrations/openai".

Verified

  • Redeployed agent-stateful-example and agent-stateless-example against this config. Both build the patched databricks-openai from main and reach Application startup complete with no ImportError.

Follow-up

  • Once a release of databricks-openai ships with the fix from #435, drop the [tool.uv.sources] block and the direct databricks-openai dep, and bump databricks-langchain to a version that pulls in the fixed databricks-openai.

Test plan

  • databricks bundle deploy --target prod && databricks bundle run agent_stateful_example --target prod
  • databricks bundle deploy --target prod && databricks bundle run agent_stateless_example --target prod
  • databricks bundle deploy --target prod && databricks bundle run agent_database_admin --target prod
  • Confirm no ImportError: cannot import name 'VectorSearchIndex' in app logs

🤖 Generated with Claude Code

The released databricks-openai (transitive of databricks-langchain) imports
VectorSearchIndex from databricks.vector_search.client, which was removed
in newer databricks-vectorsearch releases. databricks-ai-bridge PR #435
fixes the import path but is not yet in a tagged release.

Override databricks-openai with the merged source on the
databricks/databricks-ai-bridge `main` branch in all three memory-pipeline
templates (stateful, stateless, database-admin). Once a release of
databricks-openai ships with the fix, drop the [tool.uv.sources] block
and the direct databricks-openai dep, and bump databricks-langchain.

Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
@jennsun

jennsun commented Jun 10, 2026

Copy link
Copy Markdown
Owner Author

Superseded by direct commit 81b5076 on main, which bumps to databricks-ai-bridge 0.20.0 (released today) and pins to the release commit 827ae737 as a stopgap until the Databricks PyPI proxy syncs the new versions.

@jennsun jennsun closed this Jun 10, 2026
@jennsun jennsun reopened this Jul 1, 2026
]

# Track databricks-openai from main to pick up the merged fix for
# https://github.com/databricks/databricks-ai-bridge/pull/435 (imports

Copy link
Copy Markdown
Owner Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

test

"fastapi>=0.129.0",
"uvicorn>=0.41.0",
"databricks-langchain>=0.17.0",
"databricks-openai",

Copy link
Copy Markdown
Owner Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

test

"uvicorn>=0.41.0",
"databricks-langchain[memory]>=0.19.0",
"databricks-ai-bridge[agent-server]>=0.19.0",
"databricks-openai",

Copy link
Copy Markdown
Owner Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Suggested change
"databricks-openai",
"databricks-openaitest",

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant