chatbot/embeddings.py

"""
Modulo embeddings
"""
from langchain_huggingface import HuggingFaceEmbeddings

def init_embeddings( embeddings_model_name="jinaai/jina-embeddings-v3" ):
    """
    Inicializa y devuelve un modelo para embeddings.
    """

    model_kwargs = {"trust_remote_code":True}
    encode_kwargs = {'normalize_embeddings': False}
    embeddings = HuggingFaceEmbeddings(
        model_name=embeddings_model_name,
        model_kwargs=model_kwargs,
        encode_kwargs=encode_kwargs,
        show_progress=True
    )

    return embeddings