class EmbedData:

    def __init__(...):
        ...
        
    def _load_embed_model(self):
        ...
    
    def generate_embedding(self, context):
        return self.embed_model.get_text_embedding_batch(context)