rag/embeddings.py

from langchain_openai import OpenAIEmbeddings
from langchain_huggingface import HuggingFaceEmbeddings
import os

def get_embeddings():

    if not os.environ.get("OPENAI_API_KEY"):
        return HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2")
    return OpenAIEmbeddings()