from langchain_google_genai import GoogleGenerativeAIEmbeddings
from src.settings import settings

GOOGLE_API_KEY = settings.GOOGLE_API_KEY



def all_minilm_l6_v2():
    """
    Return the  embedding instance.
    """
    embedding_instance = GoogleGenerativeAIEmbeddings(
            google_api_key=GOOGLE_API_KEY, model="models/text-embedding-004",output_dimensionality=384
        )
    return embedding_instance