Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add Support for Custom Embeddings #829

Open
wants to merge 3 commits into
base: development
Choose a base branch
from

Conversation

raduburtea
Copy link
Collaborator

@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