Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Add support for custom embeddings in the embedding folder.
This new class, that inherits from EmbeddingModel, would provide a generic template for calling any custom model that is deployed as an AzureML endpoint.
A new file called
custom_embedding_model.py
is created in which the CustomEmbeddingModel class is defined. This class contains several methods to make an HTTP request, package the chunk to be embedded in a format readable by the endpoint and then retrieve the result from the endpoint.Another case is added in
factory.py
to support this new embedding type.Module is fully tested.
The
azure_model_api_key
andazure_model_api_endpoint
parameters are added theEnvironment
class. UpdatedREADME.md
to reflect these changes.