Add support for Azure PostgreSQL

[orhankislal]

Add integration support for Azure PostgreSQL with the following additional features:

• Microsoft Entra ID support
• pg_diskann indexing support

The codebase has a common folder that might get refactored out of this repo and to a different package in the future.

Note that this PR contains repurposed code from the following repo: langchain-ai/langchain-azure@401da3f#diff-ed0e9cf3e9e87215e4408f495e0be3ca408796c28ae0428dcde8ca3839f4deb4

Description

Please include a summary of the change and which issue is fixed. Please also include relevant motivation and context. List any dependencies that are required for this change.

Fixes # (issue)

New Package?

Did I fill in the tool.llamahub section in the pyproject.toml and provide a detailed README.md for my new integration or package?

• Yes
• No

Version Bump?

Did I bump the version in the pyproject.toml file of the package I am updating? (Except for the llama-index-core package)

• Yes
• No

Type of Change

Please delete options that are not relevant.

• Bug fix (non-breaking change which fixes an issue)
• New feature (non-breaking change which adds functionality)
• Breaking change (fix or feature that would cause existing functionality to not work as expected)
• This change requires a documentation update

How Has This Been Tested?

Your pull-request will likely not be merged unless it is covered by some form of impactful unit testing.

• I added new unit tests to cover this change
• I believe this change is already covered by existing unit tests

Suggested Checklist:

• I have performed a self-review of my own code
• I have commented my code, particularly in hard-to-understand areas
• I have made corresponding changes to the documentation
• I have added Google Colab support for the newly added notebooks.
• My changes generate no new warnings
• I have added tests that prove my fix is effective or that my feature works
• New and existing unit tests pass locally with my changes
• I ran uv run make format; uv run make lint to appease the lint gods

[orhankislal]

I am still working on the tests, but I wanted to open the PR up for reviews for faster iteration. I noticed that @AstraBert and @logan-markewich reviewed some of the recent large PRs, looking forward to your feedback. Please let me know if I should tag someone else as well.

[review-notebook-app]

Check out this pull request on 

See visual diffs & provide feedback on Jupyter Notebooks.

---

Powered by ReviewNB

[AstraBert]

Hey @orhankislal
I saw you put an src folder in your integration, and we generally do not do that... Just wanted to let you know before starting the review!
The correct pattern would be:

llama-index-integrations/vector-stores/llama-index-vector-stores-azure-postgresql/
|__README, LICENSE, Makefile etc.
|__llama_index/vector_stores/azure_postgresql/
     |__ __init__.py
     |__ subdirectory/
     |__ script.py
     ...

[orhankislal]

Hey @orhankislal I saw you put an src folder in your integration, and we generally do not do that... Just wanted to let you know before starting the review! The correct pattern would be:

Hey @AstraBert thanks for the suggestion. I was following the pattern from this page: https://packaging.python.org/en/latest/discussions/src-layout-vs-flat-layout/
I am happy to move it out as you suggested. Looking forward to your further feedback and suggestions.

[logan-markewich]

@orhankislal since we have a monorepo with namespaced imports, best to follow the structure @AstraBert recommended. All integrations in the repo follow this pattern

[orhankislal]

@orhankislal since we have a monorepo with namespaced imports, best to follow the structure @AstraBert recommended. All integrations in the repo follow this pattern

Sure, I made the change as suggested.

[logan-markewich]

probably this should be >=0.13,<0.15 these days

[logan-markewich]

This looks good, but tests are failing. I think they are trying to connect to an actual DB? We will need to either skip these when a DB is not available or mock out the DB so that this works in CICD