Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Client Arguments for DatabricksVectorSearch #27

Merged

Conversation

chutch1122
Copy link
Contributor

This PR allows new instances of DatabricksVectorSearch to pass custom arguments to the underlying VectorSearchClient (imported from databricks.vector_search.client).

This allows users to be able to use Service Principal based authentication by passing in service_principal_client_id or service_principal_client_secret within the new client_args parameter.

@chutch1122
Copy link
Contributor Author

Not sure where the appropriate repository for this change is with the switch over, but I've made it both in this repository and in langchain-ai/langchain-databricks#45

Copy link
Collaborator

@prithvikannan prithvikannan left a comment

Choose a reason for hiding this comment

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

Thanks @chutch1122 for the contribution. I left one small comment to fix the test failures, good to merge once fixed.

chutch1122 and others added 2 commits December 17, 2024 18:46
Resolve weird line duplication from accepting the proposed change
@prithvikannan prithvikannan self-assigned this Dec 19, 2024
Copy link
Collaborator

@prithvikannan prithvikannan left a comment

Choose a reason for hiding this comment

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

LGTM

@prithvikannan prithvikannan merged commit d60e701 into databricks:main Dec 19, 2024
5 checks passed
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.

2 participants