neo4j / neo4j-graphrag-python

Neo4j GraphRAG for Python

https://neo4j.com/docs/neo4j-graphrag-python/current/

Other

231 stars 37 forks source link

Support for Azure OpenAI + Ollama + unit tests for the embeders and LLMs #140

Closed stellasia closed 2 months ago

stellasia commented 2 months ago

Description

Add AzureOpenAI wrappers (LLM and embeddings)
Add LLMs to neo4j_graphrag.llm and embedder to neo4j_graphrag.embedding for consistency
Add tests for OpenAI (LLM + embedder) and VertexAI (embedder)
Add google cloud aiplatform to dev dependencies (for unit tests)
Update docs

Type of Change

[x] New feature
[ ] Bug fix
[ ] Breaking change
[ ] Documentation update
[ ] Project configuration change

Complexity

Note

Please provide an estimated complexity of this PR of either Low, Medium or High

Complexity: Low

How Has This Been Tested?

[x] Unit tests
[ ] E2E tests
[x] Manual tests

Checklist

The following requirements should have been met (depending on the changes in the branch):

[x] Documentation has been updated
[x] Unit tests have been updated
[ ] E2E tests have been updated
[x] Examples have been updated
[x] New files have copyright header
[x] CLA (https://neo4j.com/developer/cla/) has been signed
[ ] CHANGELOG.md updated if appropriate

stellasia commented 2 months ago

@CodiumAI-Agent /update_changelog

CodiumAI-Agent commented 2 months ago

Changelog updates: 🔄

2024-09-18

Added

Support for Azure OpenAI, including new wrappers for LLM and embeddings.
Unit tests for OpenAI and VertexAI embedders and LLMs.
Google Cloud AI Platform added to development dependencies.

Changed

Refactored OpenAILLM import for consistency across modules.

to commit the new content to the CHANGELOG.md file, please type: '/update_changelog --pr_update_changelog.push_changelog_changes=true'

CodiumAI-Agent commented 2 months ago

Changelog updates: 🔄

Next

Added

Added support for Azure OpenAI models with AzureOpenAILLM and AzureOpenAIEmbeddings.
Introduced VertexAILLM for Vertex AI language models.
Added unit tests for OpenAI and Vertex AI embedders and LLMs.
Included google-cloud-aiplatform in development dependencies.

to commit the new content to the CHANGELOG.md file, please type: '/update_changelog --pr_update_changelog.push_changelog_changes=true'