Skip to content

Add Support for Custom Embeddings#829

Open
raduburtea wants to merge 3 commits intodevelopmentfrom
raduburtea/819_add_custom_embedding_model
Open

Add Support for Custom Embeddings#829
raduburtea wants to merge 3 commits intodevelopmentfrom
raduburtea/819_add_custom_embedding_model

Conversation

@raduburtea
Copy link

@raduburtea raduburtea commented Nov 18, 2024

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 and azure_model_api_endpoint parameters are added the Environment class. Updated README.md to reflect these changes.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant