File size: 289 Bytes
66b8b0e
ff311e5
66b8b0e
ff311e5
 
ac97fe2
ff311e5
 
1
2
3
4
5
6
7
8
9
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