import os
from langchain_community.embeddings import HuggingFaceInferenceAPIEmbeddings

def get_embedding_function():
    embeddings = HuggingFaceInferenceAPIEmbeddings(
        api_key=os.environ.get("HUGGINGFACE_API_KEY"), model_name="BAAI/bge-small-en-v1.5"
    )
    return embeddings