In this PR, a separate embed_query method is created for AzureOpenAIEmbeddings and it now inherits from BaseOpenAIEmbeddings, which would avoid the openai.OpenAI client being used.
Type of Change
[ ] New feature
[x] Bug fix
[ ] Breaking change
[ ] Documentation update
[ ] Project configuration change
Complexity
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):
Description
This PR addresses the bug in the PR https://github.com/neo4j/neo4j-graphrag-python/pull/195 and this issue https://github.com/neo4j/neo4j-graphrag-python/issues/190 . Thank you @brandon-amg for raising this!
In this PR, a separate
embed_query
method is created for AzureOpenAIEmbeddings and it now inherits from BaseOpenAIEmbeddings, which would avoid the openai.OpenAI client being used.Type of Change
Complexity
Complexity: Low
How Has This Been Tested?
Checklist
The following requirements should have been met (depending on the changes in the branch):