Support for Azure OpenAi embedding model
Opened this issue · 2 comments
I'd like to propose adding support for the Azure OpenAI embedding model. My primary reason for this is to enable the use of the text-embedding-ada-002 model as the embedding model.
I believe this could be beneficial for those who are already using Azure's services or prefer to work within that ecosystem. I'm also happy to contribute to implementing this feature. Would there be any preference on how to integrate Azure OpenAI support? We could potentially add it in the configuration file, similar to other models.
models:
- type: main
engine: openai
model: text-davinci-003
- type: embedding
engine: azure
model: text-embedding-ada-002
Hi @shubhiroy! Thanks for offering to contribute to this. Should be very straightforward to add, very similar to https://github.com/NVIDIA/NeMo-Guardrails/blob/develop/nemoguardrails/embeddings/providers/openai.py.
And all the relevant code is in here: https://github.com/NVIDIA/NeMo-Guardrails/tree/develop/nemoguardrails/embeddings/providers.
Let me know if you need more guidance. Looking forward to your PR!