banking-rag-ui / app /embeddings /embedder.py
ranjithkumar111's picture
Upload 54 files
0ab392d verified
Raw
History Blame Contribute Delete
647 Bytes
# File : app/embeddings/embedder.py
# Purpose : Sentence transformer embedding model singleton
from sentence_transformers import SentenceTransformer
from app.config import get_settings
from functools import lru_cache
settings = get_settings()
@lru_cache()
def get_embedder() -> SentenceTransformer:
print(f"[EMBEDDER] Loading model: {settings.embedding_model}")
model = SentenceTransformer(settings.embedding_model)
print("[EMBEDDER] Model loaded successfully")
return model
def embed_query(query: str) -> list:
model = get_embedder()
return model.encode(query, normalize_embeddings=True).tolist()