embeddings.py 408 B

123456789101112131415
  1. from langchain_openai import OpenAIEmbeddings
  2. from app.config import settings
  3. _embeddings = None
  4. def get_embeddings():
  5. global _embeddings
  6. if _embeddings is None:
  7. _embeddings = OpenAIEmbeddings(
  8. model=settings.embedding_model,
  9. api_key=settings.effective_embedding_api_key,
  10. base_url=settings.effective_embedding_base_url,
  11. )
  12. return _embeddings