File size: 418 Bytes
a6ccbb2 cbeea95 a6ccbb2 425e09d cbeea95 e5edc91 425e09d a6ccbb2 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
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
|