Skip to content

cosmos db entra id support and fixes #2707

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

Open
wants to merge 3 commits into
base: main
Choose a base branch
from

Conversation

TheovanKraay
Copy link
Contributor

@TheovanKraay TheovanKraay commented Apr 11, 2025

Add support for Entra ID in the auto config for Azure Cosmos DB vector store support.
This PR also adds a few fixes/enhancements and corresponding tests:

  • fix to ensure getMetadata() returns the metadata fields and values
  • allow metadata field to be defined as the Cosmos DB partition key (and not just /id)
  • allow choice between gateway and direct mode in autoconfig

Signed-off-by: Theo van Kraay <[email protected]>
@TheovanKraay TheovanKraay force-pushed the cosmos-db-entra-id-support-v2 branch from e258d1c to 20dafb6 Compare April 12, 2025 10:34
@sobychacko
Copy link
Contributor

@TheovanKraay As @markpollack commented on the other issue, we could not verify certain refactoring changes we made due to the lack of access to a proper CosmosDB environment. Apologies for that bug you mentioned. Quick question: Did you get a chance to verify all the tests including the IT's on yoru CosmosDB environment? Thanks!

@TheovanKraay
Copy link
Contributor Author

@TheovanKraay As @markpollack commented on the other issue, we could not verify certain refactoring changes we made due to the lack of access to a proper CosmosDB environment. Apologies for that bug you mentioned. Quick question: Did you get a chance to verify all the tests including the IT's on yoru CosmosDB environment? Thanks!

For this PR I have run all the ITs and they pass (using Entra ID with appropriate level of access to the Cosmos DB resources).

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