| 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 | |